doveadm-fts-lucene.c revision 39ed514f9d401b3cb589595c6a2f532050254d77
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen/* Copyright (c) 2011 Dovecot authors, see the included COPYING file */
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include "lib.h"
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include "hex-binary.h"
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include "doveadm-dump.h"
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include "doveadm-fts.h"
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include "lucene-wrapper.h"
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include <stdio.h>
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen#include <sys/stat.h>
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenconst char *doveadm_fts_lucene_plugin_version = DOVECOT_VERSION;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenvoid doveadm_fts_lucene_plugin_init(struct module *module);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenvoid doveadm_fts_lucene_plugin_deinit(void);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenstatic void cmd_dump_fts_lucene(int argc ATTR_UNUSED, char *argv[])
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen{
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen struct lucene_index *index;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen struct lucene_index_iter *iter;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen mail_guid_128_t prev_guid;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen const struct lucene_index_record *rec;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen bool first = TRUE;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen memset(&prev_guid, 0, sizeof(prev_guid));
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen index = lucene_index_init(argv[1], NULL, NULL);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen iter = lucene_index_iter_init(index);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen while ((rec = lucene_index_iter_next(iter)) != NULL) {
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen if (memcmp(prev_guid, rec->mailbox_guid,
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen sizeof(prev_guid)) != 0) {
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen if (first)
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen first = FALSE;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen else
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen printf("\n");
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen memcpy(prev_guid, rec->mailbox_guid, sizeof(prev_guid));
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen printf("%s: ", binary_to_hex(prev_guid, sizeof(prev_guid)));
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen }
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen printf("%u,", rec->uid);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen }
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen printf("\n");
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen if (lucene_index_iter_deinit(&iter) < 0)
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen i_error("Lucene index iteration failed");
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen lucene_index_deinit(index);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen}
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenstatic bool test_dump_fts_lucene(const char *path)
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen{
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen const char *p;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen struct stat st;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen p = strrchr(path, '/');
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen if (p++ == NULL)
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen p = path;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen p = t_strconcat(p, "/segments.gen", NULL);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen return stat(p, &st) == 0;
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen}
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenstruct doveadm_cmd_dump doveadm_cmd_dump_fts_lucene = {
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen "fts-lucene",
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen test_dump_fts_lucene,
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen cmd_dump_fts_lucene
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen};
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenvoid doveadm_fts_lucene_plugin_init(struct module *module ATTR_UNUSED)
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen{
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen doveadm_dump_register(&doveadm_cmd_dump_fts_lucene);
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen}
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainenvoid doveadm_fts_lucene_plugin_deinit(void)
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen{
39ed514f9d401b3cb589595c6a2f532050254d77Timo Sirainen}