forked from Mirrors/privacore-open-source-search-engine
79 lines
1.7 KiB
C++
79 lines
1.7 KiB
C++
#include "BigFile.h"
|
|
#include "RdbIndex.h"
|
|
#include "Posdb.h"
|
|
#include "Log.h"
|
|
#include <libgen.h>
|
|
|
|
static void print_usage(const char *argv0) {
|
|
fprintf(stdout, "Usage: %s [-h] FILE\n", argv0);
|
|
fprintf(stdout, "Generate RDB index from FILE\n");
|
|
fprintf(stdout, "\n");
|
|
fprintf(stdout, " -h, --help display this help and exit\n");
|
|
}
|
|
|
|
static bool starts_with(const char *haystack, const char *needle) {
|
|
size_t haystackLen = strlen(haystack);
|
|
size_t needleLen = strlen(needle);
|
|
if (haystackLen < needleLen) {
|
|
return false;
|
|
}
|
|
|
|
return (memcmp(haystack, needle, needleLen) == 0);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
if (argc < 2) {
|
|
print_usage(argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
if (strcmp(argv[1], "--h") == 0 || strcmp(argv[1], "--help") == 0 ) {
|
|
print_usage(argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
char filepath[PATH_MAX];
|
|
|
|
char dir[PATH_MAX];
|
|
strcpy(filepath, argv[1]);
|
|
strcpy(dir, dirname(filepath));
|
|
|
|
char filename[PATH_MAX];
|
|
strcpy(filepath, argv[1]);
|
|
strcpy(filename, basename(filepath));
|
|
|
|
char indexFilename[PATH_MAX];
|
|
strcpy(indexFilename, filename);
|
|
strcpy(strrchr(indexFilename, '.'), ".idx");
|
|
|
|
// initialize library
|
|
g_mem.init();
|
|
hashinit();
|
|
|
|
g_conf.init(NULL);
|
|
|
|
BigFile bigFile;
|
|
bigFile.set(dir, filename);
|
|
|
|
g_conf.m_logTraceRdbIndex = true;
|
|
|
|
RdbIndex index;
|
|
if (starts_with(filename, "posdb")) {
|
|
index.set(dir, indexFilename, 0, true, sizeof(key144_t), RDB_POSDB);
|
|
if (!index.generateIndex(&bigFile)) {
|
|
fprintf(stdout, "Unable to generate index for %s\n", filename);
|
|
return 1;
|
|
}
|
|
|
|
if (!index.writeIndex()) {
|
|
fprintf(stdout, "Unable to save index for %s\n", filename);
|
|
return 1;
|
|
}
|
|
} else {
|
|
fprintf(stdout, "Unsupported rdb type\n");
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|