projects / ftsbench.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
add .gitignore
[ftsbench.git] / mysqldriver.c
diff --git a/mysqldriver.c b/mysqldriver.c
index be64aa4..f815dd9 100644 (file)
--- a/mysqldriver.c
+++ b/mysqldriver.c
@@ -50,30 +50,34 @@ typedef struct ftsMY {
 static void
 execQuery(ftsDB *adb, char **words, int flags) {
        ftsMY *db = (ftsMY*)adb;
-       static MYSQL_BIND       data;
+       static MYSQL_BIND       datain, dataout;
+       static int intout;
+       static my_bool  isnull, my_error;
+       static unsigned long    length;
        MYSQL_RES       *res;
 
        if ( db->prepareStmt == NULL ) {
                db->prepareStmt = mysql_stmt_init(db->conn);
-               if ( db->prepareStmt == NULL ) {
-                       fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
-                       exit(1);
-               }
+               if ( db->prepareStmt == NULL ) 
+                       fatal("mysql_stmt_init failed: %s\n", mysql_error(db->conn));
 
 #define        SEARCH_QUERY "SELECT count(*) FROM ftsbench WHERE MATCH(body) AGAINST ( ? IN BOOLEAN MODE);"
-               if ( mysql_stmt_prepare( db->prepareStmt, SEARCH_QUERY, strlen(SEARCH_QUERY) ) != 0 ) {
-                       fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
-                       exit(1);
-               }
+               if ( mysql_stmt_prepare( db->prepareStmt, SEARCH_QUERY, strlen(SEARCH_QUERY) ) != 0 ) 
+                       fatal("mysql_stmt_init failed: %s\n", mysql_error(db->conn));
 
-               if ( mysql_stmt_param_count(db->prepareStmt) != 1 ) {
-                       fprintf(stderr,"mysql_stmt_param_count: invalid parameter count\n");
-                       exit(1);
-               }
+               if ( mysql_stmt_param_count(db->prepareStmt) != 1 ) 
+                       fatal("mysql_stmt_param_count: invalid parameter count\n");
 
-               memset(&data, 0, sizeof(MYSQL_BIND));
-               data.buffer_type = MYSQL_TYPE_BLOB;
-               data.length = (unsigned long*)&(db->b.strlen);
+               memset(&datain, 0, sizeof(MYSQL_BIND));
+               datain.buffer_type = MYSQL_TYPE_BLOB;
+               datain.length = (unsigned long*)&(db->b.strlen);
+
+               memset(&dataout, 0, sizeof(MYSQL_BIND));
+               dataout.buffer_type= MYSQL_TYPE_LONG;
+               dataout.buffer= (char *)&intout;
+               dataout.is_null= &isnull;
+               dataout.length= &length;
+               dataout.error= &my_error;
 
                db->flags = flags;
        }
@@ -81,36 +85,40 @@ execQuery(ftsDB *adb, char **words, int flags) {
     db->b.strlen = 0;
                 
        while( *words ) {
-               if ( db->flags & FLG_OR)
-                       sb_add(&db->b, " ", 1);
-               else
-                       sb_add(&db->b, " +", 2);
+               sb_addchar(&db->b, ' ');
+               if ( db->flags & FLG_AND)
+                       sb_addchar(&db->b, '+');
 
                sb_add(&db->b, *words, -1);
                words++;
        }
 
-       data.buffer = db->b.str;
-       data.buffer_length = db->b.strlen;
+       datain.buffer = db->b.str;
+       datain.buffer_length = db->b.strlen;
 
-       if ( mysql_stmt_bind_param(db->prepareStmt, &data) ) {
-               fprintf(stderr,"mysql_stmt_bind_param failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       if ( mysql_stmt_bind_param(db->prepareStmt, &datain) ) 
+               fatal("mysql_stmt_bind_param failed: %s\n", mysql_error(db->conn));
+
+       res = mysql_stmt_result_metadata(db->prepareStmt);
+       if ( !res ) 
+               fatal("mysql_stmt_result_metadata failed: %s\n", mysql_error(db->conn));
+
+       if ( mysql_stmt_execute( db->prepareStmt ) ) 
+               fatal("mysql_stmt_execute failed: %s\n", mysql_error(db->conn));
+
+       if (mysql_stmt_bind_result( db->prepareStmt, &dataout)) 
+               fatal("mysql_stmt_bind_result failed: %s\n", mysql_error(db->conn));
 
-       if ( mysql_stmt_execute( db->prepareStmt ) ) {
-               fprintf(stderr,"mysql_stmt_execute failed: %s\n", mysql_error(db->conn));
-               exit(1);
+       if ( mysql_stmt_store_result( db->prepareStmt ) ) 
+               fatal("mysql_stmt_store_result failed: %s\n", mysql_error(db->conn));
+
+       if ( !mysql_stmt_fetch( db->prepareStmt) ) {
+               db->db.nres += intout;
+       } else {
+               fatal("mysql_stmt_fetch returns void result\n");
        }
 
-       res = mysql_store_result(db->conn);
-       if ( res == NULL ) {
-               if  ( mysql_errno(db->conn)!=0 ) {
-                       fprintf(stderr,"mysql_store_result failed: %s\n", mysql_error(db->conn));
-                       exit(1);
-               }
-       } else
-               mysql_free_result(res);
+       mysql_free_result(res);
 
        pthread_mutex_lock(&(db->db.nqueryMutex));
        db->db.nquery ++;
@@ -123,20 +131,16 @@ startCreateScheme(ftsDB *adb, int flags) {
 
        db->flags = flags;
        if ( flags & FLG_FUNC )
-               printf("Flag 'func' is ignored by MySQL\n");
+               report("Flag 'func' is ignored by MySQL\n");
 
        if ( flags & (FLG_GIN | FLG_GIST) )
-               printf("MySQL doesn't distinguish 'gin' and 'gist' flags\n");
+               report("MySQL doesn't distinguish 'gin' and 'gist' flags\n");
 
-       if ( mysql_query(db->conn, "DROP TABLE IF EXISTS ftsbench CASCADE;")!= 0 ) {
-               fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       if ( mysql_query(db->conn, "DROP TABLE IF EXISTS ftsbench CASCADE;")!= 0 ) 
+               fatal("mysql_query failed: %s\n", mysql_error(db->conn));
 
-       if ( mysql_query(db->conn, "CREATE TABLE ftsbench (id int not null, body text) ENGINE MyISAM;")!= 0 ) {
-               fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       if ( mysql_query(db->conn, "CREATE TABLE ftsbench (id int not null, body text) ENGINE MyISAM;")!= 0 ) 
+               fatal("mysql_query failed: %s\n", mysql_error(db->conn));
 }
 
 static void
@@ -144,27 +148,19 @@ finishCreateScheme(ftsDB *adb) {
        ftsMY *db = (ftsMY*)adb;
 
        if  ( db->flags & (FLG_GIN | FLG_GIST) ) {
-               printf("(create index, ");
-               fflush(stdout);
+               report("(create index, ");
 
-               if ( mysql_query(db->conn, "CREATE FULLTEXT INDEX fts ON ftsbench (body);")!= 0 ) {
-                       fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
-                       exit(1);
-               }
-       } else {
-               printf("(");
-               fflush(stdout);
-       }
+               if ( mysql_query(db->conn, "CREATE FULLTEXT INDEX fts ON ftsbench (body);")!= 0 ) 
+                       fatal("mysql_query failed: %s\n", mysql_error(db->conn));
+       } else 
+               report("(");
 
-       printf("optimize");
-       fflush(stdout);
-       if ( mysql_query(db->conn, "OPTIMIZE TABLE ftsbench;")!= 0 ) {
-               fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       report("optimize");
+       
+       if ( mysql_query(db->conn, "OPTIMIZE TABLE ftsbench;")!= 0 ) 
+               fatal("mysql_query failed: %s\n", mysql_error(db->conn));
 
-       printf(") ");
-       fflush(stdout);
+       report(") ");
 }
 
 static void
@@ -175,21 +171,15 @@ InsertRow(ftsDB *adb, int id, char *txt) {
 
        if ( db->prepareStmt == NULL ) {
                db->prepareStmt = mysql_stmt_init(db->conn);
-               if ( db->prepareStmt == NULL ) {
-                       fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
-                       exit(1);
-               }
+               if ( db->prepareStmt == NULL ) 
+                       fatal("mysql_stmt_init failed: %s\n", mysql_error(db->conn));
 
 #define        INSERT_QUERY "INSERT INTO ftsbench (id, body) VALUES ( ? , ? );"
-               if ( mysql_stmt_prepare( db->prepareStmt, INSERT_QUERY, strlen(INSERT_QUERY) ) != 0 ) {
-                       fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
-                       exit(1);
-               }
+               if ( mysql_stmt_prepare( db->prepareStmt, INSERT_QUERY, strlen(INSERT_QUERY) ) != 0 ) 
+                       fatal("mysql_stmt_init failed: %s\n", mysql_error(db->conn));
 
-               if ( mysql_stmt_param_count(db->prepareStmt) != 2 ) {
-                       fprintf(stderr,"mysql_stmt_param_count: invalid parameter count\n");
-                       exit(1);
-               }
+               if ( mysql_stmt_param_count(db->prepareStmt) != 2 ) 
+                       fatal("mysql_stmt_param_count: invalid parameter count\n");
 
                memset(data, 0, sizeof(data));
                data[0].buffer_type = MYSQL_TYPE_LONG;
@@ -204,15 +194,18 @@ InsertRow(ftsDB *adb, int id, char *txt) {
        data[1].buffer = txt;
        data[1].buffer_length = txtlen;
 
-       if ( mysql_stmt_bind_param(db->prepareStmt, data) ) {
-               fprintf(stderr,"mysql_stmt_bind_param failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       if ( mysql_stmt_bind_param(db->prepareStmt, data) ) 
+               fatal("mysql_stmt_bind_param failed: %s\n", mysql_error(db->conn));
 
-       if ( mysql_stmt_execute( db->prepareStmt ) ) {
-               fprintf(stderr,"mysql_stmt_execute failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       if ( mysql_stmt_execute( db->prepareStmt ) ) 
+               fatal("mysql_stmt_execute failed: %s\n", mysql_error(db->conn));
+}
+
+static void
+Close(ftsDB* adb) {  
+    ftsMY *db = (ftsMY*)adb;
+
+       mysql_close(db->conn);
 }
 
 ftsDB* 
@@ -223,15 +216,14 @@ MYInit(char * connstr) {
 
        db->conn = mysql_init(NULL);
 
-       if ( !mysql_real_connect(db->conn, NULL, NULL, NULL, connstr, 0, NULL, 0) ) {
-               fprintf(stderr,"mysql_real_connect failed: %s\n", mysql_error(db->conn));
-               exit(1);
-       }
+       if ( !mysql_real_connect(db->conn, NULL, NULL, NULL, connstr, 0, NULL, 0) ) 
+               fatal("mysql_real_connect failed: %s\n", mysql_error(db->conn));
 
        db->db.execQuery = execQuery;
        db->db.startCreateScheme = startCreateScheme;
        db->db.finishCreateScheme = finishCreateScheme;
        db->db.InsertRow = InsertRow;
+       db->db.Close = Close;
        
        return (ftsDB*)db;
 }
Unnamed repository; edit this file 'description' to name the repository.