]> code.citadel.org Git - citadel.git/blobdiff - citadel/serv_fulltext.c
mk_module_init.sh now tests to see if echo supports -e and -E
[citadel.git] / citadel / serv_fulltext.c
index 0d5da3e672a679b6b37188106ef82a33f9f61c80..f6e2c672a8902f0a07be23f3a4c18b7a39e27fa9 100644 (file)
 #include <limits.h>
 #include "citadel.h"
 #include "server.h"
-#include "sysdep_decls.h"
 #include "citserver.h"
 #include "support.h"
 #include "config.h"
-#include "serv_extensions.h"
 #include "database.h"
 #include "msgbase.h"
 #include "control.h"
 #include "ft_wordbreaker.h"
 
 
+#include "ctdl_module.h"
+
+
+
 long ft_newhighest = 0L;
 long *ft_newmsgs = NULL;
 int ft_num_msgs = 0;
@@ -79,7 +81,8 @@ void ft_flush_cache(void) {
 
        for (i=0; i<65536; ++i) {
                if ((time(NULL) - last_update) >= 10) {
-                       lprintf(5, "Flushing index cache to disk (%d%% complete)\n",
+                       lprintf(CTDL_INFO,
+                               "Flushing index cache to disk (%d%% complete)\n",
                                (i * 100 / 65536)
                        );
                        last_update = time(NULL);
@@ -92,7 +95,7 @@ void ft_flush_cache(void) {
                        ftc_msgs[i] = NULL;
                }
        }
-       lprintf(5, "Cache flush has been completed\n");
+       lprintf(CTDL_INFO, "Flushed index cache to disk (100%% complete)\n");
 }
 
 
@@ -117,7 +120,7 @@ void ft_index_message(long msgnum, int op) {
        CC->redirect_buffer = malloc(SIZ);
        CC->redirect_len = 0;
        CC->redirect_alloc = SIZ;
-       CtdlOutputMsg(msgnum, MT_CITADEL, HEADERS_ALL, 0, 1);
+       CtdlOutputMsg(msgnum, MT_CITADEL, HEADERS_ALL, 0, 1, NULL);
        msgtext = CC->redirect_buffer;
        CC->redirect_buffer = NULL;
        CC->redirect_len = 0;
@@ -126,7 +129,7 @@ void ft_index_message(long msgnum, int op) {
        wordbreaker(msgtext, &num_tokens, &tokens);
        free(msgtext);
 
-       lprintf(CTDL_DEBUG, "Indexing message %ld...\n", msgnum);
+       lprintf(CTDL_DEBUG, "Indexing message %ld [%d tokens]\n", msgnum, num_tokens);
        if (num_tokens > 0) {
                for (i=0; i<num_tokens; ++i) {
 
@@ -135,42 +138,45 @@ void ft_index_message(long msgnum, int op) {
                        /* search for tokens[i] */
                        tok = tokens[i];
 
-                       /* fetch the bucket, Liza */
-                       if (ftc_msgs[tok] == NULL) {
-                               cdb_bucket = cdb_fetch(CDB_FULLTEXT, &tok, sizeof(int));
-                               if (cdb_bucket != NULL) {
-                                       ftc_num_msgs[tok] = cdb_bucket->len / sizeof(long);
-                                       ftc_msgs[tok] = (long *)cdb_bucket->ptr;
-                                       cdb_bucket->ptr = NULL;
-                                       cdb_free(cdb_bucket);
+                       if ( (tok >= 0) && (tok <= 65535) ) {
+                               /* fetch the bucket, Liza */
+                               if (ftc_msgs[tok] == NULL) {
+                                       cdb_bucket = cdb_fetch(CDB_FULLTEXT, &tok, sizeof(int));
+                                       if (cdb_bucket != NULL) {
+                                               ftc_num_msgs[tok] = cdb_bucket->len / sizeof(long);
+                                               ftc_msgs[tok] = (long *)cdb_bucket->ptr;
+                                               cdb_bucket->ptr = NULL;
+                                               cdb_free(cdb_bucket);
+                                       }
+                                       else {
+                                               ftc_num_msgs[tok] = 0;
+                                               ftc_msgs[tok] = malloc(sizeof(long));
+                                       }
                                }
-                               else {
-                                       ftc_num_msgs[tok] = 0;
-                                       ftc_msgs[tok] = malloc(sizeof(long));
+       
+       
+                               if (op == 1) {  /* add to index */
+                                       ++ftc_num_msgs[tok];
+                                       ftc_msgs[tok] = realloc(ftc_msgs[tok],
+                                                               ftc_num_msgs[tok]*sizeof(long));
+                                       ftc_msgs[tok][ftc_num_msgs[tok] - 1] = msgnum;
                                }
-                       }
-
-
-                       if (op == 1) {  /* add to index */
-                               ++ftc_num_msgs[tok];
-                               ftc_msgs[tok] = realloc(ftc_msgs[tok],
-                                                       ftc_num_msgs[tok]*sizeof(long));
-                               ftc_msgs[tok][ftc_num_msgs[tok] - 1] = msgnum;
-                       }
-
-                       if (op == 0) {  /* remove from index */
-                               if (ftc_num_msgs[tok] >= 1) {
-                                       for (j=0; j<ftc_num_msgs[tok]; ++j) {
-                                               if (ftc_msgs[tok][j] == msgnum) {
-                                                       memmove(&ftc_msgs[tok][j], &ftc_msgs[tok][j+1], ((ftc_num_msgs[tok] - j - 1)*sizeof(long)));
-                                                       --ftc_num_msgs[tok];
-                                                       --j;
+       
+                               if (op == 0) {  /* remove from index */
+                                       if (ftc_num_msgs[tok] >= 1) {
+                                               for (j=0; j<ftc_num_msgs[tok]; ++j) {
+                                                       if (ftc_msgs[tok][j] == msgnum) {
+                                                               memmove(&ftc_msgs[tok][j], &ftc_msgs[tok][j+1], ((ftc_num_msgs[tok] - j - 1)*sizeof(long)));
+                                                               --ftc_num_msgs[tok];
+                                                               --j;
+                                                       }
                                                }
                                        }
                                }
                        }
-
-                       /* FIXME do we need to unlock the file here? */
+                       else {
+                               lprintf(CTDL_ALERT, "Invalid token %d !!\n", tok);
+                       }
                }
 
                free(tokens);
@@ -202,7 +208,7 @@ void ft_index_msg(long msgnum, void *userdata) {
 void ft_index_room(struct ctdlroom *qrbuf, void *data)
 {
        getroom(&CC->room, qrbuf->QRname);
-       CtdlForEachMessage(MSGS_ALL, 0L, NULL, NULL, ft_index_msg, NULL);
+       CtdlForEachMessage(MSGS_ALL, 0L, NULL, NULL, NULL, ft_index_msg, NULL);
 }
 
 
@@ -243,12 +249,16 @@ void do_fulltext_indexing(void) {
         * If we've switched wordbreaker modules, burn the index and start
         * over.
         */
+       begin_critical_section(S_CONTROL);
+       lprintf(CTDL_DEBUG, "wb ver on disk = %d, code ver = %d\n",
+                       CitControl.fulltext_wordbreaker, FT_WORDBREAKER_ID);
        if (CitControl.fulltext_wordbreaker != FT_WORDBREAKER_ID) {
                lprintf(CTDL_INFO, "(re)initializing full text index\n");
                cdb_trunc(CDB_FULLTEXT);
                CitControl.MMfulltext = 0L;
                put_control();
        }
+       end_critical_section(S_CONTROL);
 
        /*
         * Now go through each room and find messages to index.
@@ -269,7 +279,7 @@ void do_fulltext_indexing(void) {
 
                /* Here it is ... do each message! */
                for (i=0; i<ft_num_msgs; ++i) {
-                       if ((time(NULL) - last_progress) > 10) {
+                       if (time(NULL) != last_progress) {
                                lprintf(CTDL_DEBUG,
                                        "Indexed %d of %d messages (%d%%)\n",
                                                i, ft_num_msgs,
@@ -286,6 +296,13 @@ void do_fulltext_indexing(void) {
                                break;
                        }
 
+                       /* Check to see if we have to maybe flush to disk */
+                       if (i >= FT_MAX_CACHE) {
+                               lprintf(CTDL_DEBUG, "Time to flush.\n");
+                               ft_newhighest = ft_newmsgs[i];
+                               break;
+                       }
+
                }
 
                free(ft_newmsgs);
@@ -296,9 +313,11 @@ void do_fulltext_indexing(void) {
 
        /* Save our place so we don't have to do this again */
        ft_flush_cache();
+       begin_critical_section(S_CONTROL);
        CitControl.MMfulltext = ft_newhighest;
        CitControl.fulltext_wordbreaker = FT_WORDBREAKER_ID;
        put_control();
+       end_critical_section(S_CONTROL);
        last_index = time(NULL);
 
        lprintf(CTDL_DEBUG, "do_fulltext_indexing() finished\n");
@@ -409,7 +428,7 @@ void ft_search(int *fts_num_msgs, long **fts_msgs, char *search_string) {
 
 
 /*
- * Tentative form of a search command
+ * This search command is for diagnostic purposes and may be removed or replaced.
  */
 void cmd_srch(char *argbuf) {
        int num_msgs = 0;
@@ -422,6 +441,7 @@ void cmd_srch(char *argbuf) {
        if (!config.c_enable_fulltext) {
                cprintf("%d Full text index is not enabled on this server.\n",
                        ERROR + CMD_NOT_SUPPORTED);
+               return;
        }
 
        extract_token(search_string, argbuf, 0, '|', sizeof search_string);
@@ -449,9 +469,11 @@ void initialize_ft_cache(void) {
 
 /*****************************************************************************/
 
-char *serv_fulltext_init(void)
+CTDL_MODULE_INIT(fulltext)
 {
        initialize_ft_cache();
        CtdlRegisterProtoHook(cmd_srch, "SRCH", "Full text search");
+
+       /* return our Subversion id for the Log */
        return "$Id$";
 }