8 #include <netinet/in.h>
13 typedef struct ftsMY {
17 MYSQL_STMT *prepareStmt;
22 execQuery(ftsDB *adb, char **words, int flags) {
23 ftsMY *db = (ftsMY*)adb;
24 static MYSQL_BIND data;
27 if ( db->prepareStmt == NULL ) {
28 db->prepareStmt = mysql_stmt_init(db->conn);
29 if ( db->prepareStmt == NULL ) {
30 fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
34 #define SEARCH_QUERY "SELECT count(*) FROM ftsbench WHERE MATCH(body) AGAINST ( ? IN BOOLEAN MODE);"
35 if ( mysql_stmt_prepare( db->prepareStmt, SEARCH_QUERY, strlen(SEARCH_QUERY) ) != 0 ) {
36 fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
40 if ( mysql_stmt_param_count(db->prepareStmt) != 1 ) {
41 fprintf(stderr,"mysql_stmt_param_count: invalid parameter count\n");
45 memset(&data, 0, sizeof(MYSQL_BIND));
46 data.buffer_type = MYSQL_TYPE_BLOB;
47 data.length = (unsigned long*)&(db->b.strlen);
55 if ( db->flags & FLG_OR)
56 sb_add(&db->b, " ", 1);
58 sb_add(&db->b, " +", 2);
60 sb_add(&db->b, *words, -1);
64 data.buffer = db->b.str;
65 data.buffer_length = db->b.strlen;
67 if ( mysql_stmt_bind_param(db->prepareStmt, &data) ) {
68 fprintf(stderr,"mysql_stmt_bind_param failed: %s\n", mysql_error(db->conn));
72 if ( mysql_stmt_execute( db->prepareStmt ) ) {
73 fprintf(stderr,"mysql_stmt_execute failed: %s\n", mysql_error(db->conn));
77 res = mysql_store_result(db->conn);
79 if ( mysql_errno(db->conn)!=0 ) {
80 fprintf(stderr,"mysql_store_result failed: %s\n", mysql_error(db->conn));
84 mysql_free_result(res);
86 pthread_mutex_lock(&(db->db.nqueryMutex));
88 pthread_mutex_unlock(&(db->db.nqueryMutex));
92 startCreateScheme(ftsDB *adb, int flags) {
93 ftsMY *db = (ftsMY*)adb;
96 if ( flags & FLG_FUNC )
97 printf("Flag 'func' is ignored by MySQL\n");
99 if ( flags & (FLG_GIN | FLG_GIST) )
100 printf("MySQL doesn't distinguish 'gin' and 'gist' flags\n");
102 if ( mysql_query(db->conn, "DROP TABLE IF EXISTS ftsbench CASCADE;")!= 0 ) {
103 fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
107 if ( mysql_query(db->conn, "CREATE TABLE ftsbench (id int not null, body text) ENGINE MyISAM;")!= 0 ) {
108 fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
114 finishCreateScheme(ftsDB *adb) {
115 ftsMY *db = (ftsMY*)adb;
117 if ( db->flags & (FLG_GIN | FLG_GIST) ) {
118 printf("(create index, ");
121 if ( mysql_query(db->conn, "CREATE FULLTEXT INDEX fts ON ftsbench (body);")!= 0 ) {
122 fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
132 if ( mysql_query(db->conn, "OPTIMIZE TABLE ftsbench;")!= 0 ) {
133 fprintf(stderr,"mysql_query failed: %s\n", mysql_error(db->conn));
142 InsertRow(ftsDB *adb, int id, char *txt) {
143 ftsMY *db = (ftsMY*)adb;
144 static MYSQL_BIND data[2];
145 static unsigned long txtlen;
147 if ( db->prepareStmt == NULL ) {
148 db->prepareStmt = mysql_stmt_init(db->conn);
149 if ( db->prepareStmt == NULL ) {
150 fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
154 #define INSERT_QUERY "INSERT INTO ftsbench (id, body) VALUES ( ? , ? );"
155 if ( mysql_stmt_prepare( db->prepareStmt, INSERT_QUERY, strlen(INSERT_QUERY) ) != 0 ) {
156 fprintf(stderr,"mysql_stmt_init failed: %s\n", mysql_error(db->conn));
160 if ( mysql_stmt_param_count(db->prepareStmt) != 2 ) {
161 fprintf(stderr,"mysql_stmt_param_count: invalid parameter count\n");
165 memset(data, 0, sizeof(data));
166 data[0].buffer_type = MYSQL_TYPE_LONG;
167 data[1].buffer_type = MYSQL_TYPE_BLOB;
168 data[1].length = &txtlen;
172 data[0].buffer = &id;
174 txtlen = (unsigned long) strlen(txt);
175 data[1].buffer = txt;
176 data[1].buffer_length = txtlen;
178 if ( mysql_stmt_bind_param(db->prepareStmt, data) ) {
179 fprintf(stderr,"mysql_stmt_bind_param failed: %s\n", mysql_error(db->conn));
183 if ( mysql_stmt_execute( db->prepareStmt ) ) {
184 fprintf(stderr,"mysql_stmt_execute failed: %s\n", mysql_error(db->conn));
190 MYInit(char * connstr) {
191 ftsMY *db = (ftsMY*)malloc(sizeof(ftsMY));
193 memset(db,0,sizeof(ftsMY));
195 db->conn = mysql_init(NULL);
197 if ( !mysql_real_connect(db->conn, NULL, NULL, NULL, connstr, 0, NULL, 0) ) {
198 fprintf(stderr,"mysql_real_connect failed: %s\n", mysql_error(db->conn));
202 db->db.execQuery = execQuery;
203 db->db.startCreateScheme = startCreateScheme;
204 db->db.finishCreateScheme = finishCreateScheme;
205 db->db.InsertRow = InsertRow;