-/*
- * Copyright (C) 1994-1995, Index Data I/S
- * All rights reserved.
- * Sebastian Hammer, Adam Dickmeiss
- *
- * $Log: main.c,v $
- * Revision 1.11 1995-09-29 14:01:45 adam
- * Bug fixes.
- *
- * Revision 1.10 1995/09/28 14:22:57 adam
- * Sort uses smaller temporary files.
- *
- * Revision 1.9 1995/09/14 07:48:24 adam
- * Record control management.
- *
- * Revision 1.8 1995/09/06 16:11:18 adam
- * Option: only one word key per file.
- *
- * Revision 1.7 1995/09/05 15:28:39 adam
- * More work on search engine.
- *
- * Revision 1.6 1995/09/04 12:33:43 adam
- * Various cleanup. YAZ util used instead.
- *
- * Revision 1.5 1995/09/04 09:10:39 adam
- * More work on index add/del/update.
- * Merge sort implemented.
- * Initial work on z39 server.
- *
- * Revision 1.4 1995/09/01 14:06:36 adam
- * Split of work into more files.
- *
- * Revision 1.3 1995/09/01 10:57:07 adam
- * Minor changes.
- *
- * Revision 1.2 1995/09/01 10:30:24 adam
- * More work on indexing. Not working yet.
- *
- * Revision 1.1 1995/08/31 14:50:24 adam
- * New simple file index tool.
- *
- */
+/* $Id: main.c,v 1.128 2006-03-31 15:58:04 adam Exp $
+ Copyright (C) 1995-2005
+ Index Data ApS
+
+This file is part of the Zebra server.
+
+Zebra is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 2, or (at your option) any later
+version.
+
+Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with Zebra; see the file LICENSE.zebra. If not, write to the
+Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
+02111-1307, USA.
+*/
+
#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
#include <assert.h>
+#ifdef WIN32
+#include <io.h>
+#endif
+#if HAVE_UNISTD_H
#include <unistd.h>
+#endif
+#if HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
+#include <time.h>
+#if HAVE_SYS_TIMES_H
+#include <sys/times.h>
+#endif
-#include <alexutil.h>
-#include "index.h"
+#include <yaz/log.h>
+#include <yaz/options.h>
+#include <idzebra/api.h>
char *prog;
-size_t mem_max = 4*1024*1024;
+
+static void filter_cb(void *cd, const char *name)
+{
+ puts (name);
+}
+
+static void show_filters(ZebraService zs)
+{
+ zebra_filter_info(zs, 0, filter_cb);
+}
int main (int argc, char **argv)
{
int ret;
int cmd = 0;
char *arg;
- char *base_name = NULL;
- char *base_path = NULL;
- int nsections;
- char **mbuf;
+ char *configName = 0;
+ int nsections = 0;
+ int enable_commit = 1;
+ char *database = 0;
+ Res res = res_open(0, 0);
+
+ int trans_started=0;
+#if HAVE_SYS_TIMES_H
+ struct tms tms1, tms2;
+ double usec;
+#endif
+#if HAVE_SYS_TIME_H
+ struct timeval start_time, end_time;
+#endif
+#ifndef WIN32
+ char nbuf[100];
+#endif
+ ZebraService zs = 0;
+ ZebraHandle zh = 0;
+ nmem_init ();
+
+#ifdef WIN32
+#else
+ sprintf(nbuf, "%.40s(%ld)", *argv, (long) getpid());
+ yaz_log_init_prefix (nbuf);
+#endif
+#if HAVE_SYS_TIMES_H
+ times(&tms1);
+#endif
+#if HAVE_SYS_TIME_H
+ gettimeofday(&start_time, 0);
+#endif
prog = *argv;
- while ((ret = options ("r:v:m:", argv, argc, &arg)) != -2)
+ if (argc < 2)
+ {
+ fprintf (stderr, "%s [options] command <dir> ...\n"
+ "Commands:\n"
+ " update <dir> Update index with files below <dir>.\n"
+ " If <dir> is empty filenames are read from stdin.\n"
+ " delete <dir> Delete index with files below <dir>.\n"
+ " create <db> Create database <db>\n"
+ " drop <db> Drop database <db>\n"
+ " commit Commit changes\n"
+ " clean Clean shadow files\n"
+ "Options:\n"
+ " -t <type> Index files as <type> (grs or text).\n"
+ " -c <config> Read configuration file <config>.\n"
+ " -g <group> Index files according to group settings.\n"
+ " -d <database> Records belong to Z39.50 database <database>.\n"
+ " -m <mbytes> Use <mbytes> before flushing keys to disk.\n"
+ " -n Don't use shadow system.\n"
+ " -s Show analysis on stdout, but do no work.\n"
+ " -v <level> Set logging to <level>.\n"
+ " -l <file> Write log to <file>.\n"
+ " -L Don't follow symbolic links.\n"
+ " -f <n> Display information for the first <n> records.\n"
+ " -V Show version.\n", *argv
+ );
+ exit (1);
+ }
+ while ((ret = options("sVt:c:g:d:m:v:nf:l:L", argv, argc, &arg)) != -2)
{
if (ret == 0)
{
- if (!base_name)
+ if(cmd == 0) /* command */
{
- base_name = arg;
+ if (!zs)
+ {
+ const char *config = configName ? configName : "zebra.cfg";
+ zs = zebra_start_res (config, 0, res);
+ if (!zs)
+ {
+ yaz_log (YLOG_FATAL, "Cannot read config %s", config);
+ exit (1);
+ }
+ zh = zebra_open (zs, 0);
+ zebra_shadow_enable (zh, enable_commit);
+ }
- common_resource = res_open (base_name);
- if (!common_resource)
+ if (database &&
+ zebra_select_database (zh, database) == ZEBRA_FAIL)
+ {
+ yaz_log(YLOG_FATAL, "Could not select database %s "
+ "errCode=%d",
+ database, zebra_errCode(zh) );
+ exit (1);
+ }
+ if (!strcmp (arg, "update"))
+ cmd = 'u';
+ else if (!strcmp (arg, "update1"))
+ cmd = 'U';
+ else if (!strcmp (arg, "update2"))
+ cmd = 'm';
+ else if (!strcmp (arg, "dump"))
+ cmd = 's';
+ else if (!strcmp (arg, "del") || !strcmp(arg, "delete"))
+ cmd = 'd';
+ else if (!strcmp (arg, "init"))
+ {
+ zebra_init (zh);
+ }
+ else if (!strcmp(arg, "drop"))
+ {
+ cmd = 'D';
+ }
+ else if (!strcmp(arg, "create"))
+ {
+ cmd = 'C';
+ }
+ else if (!strcmp (arg, "commit"))
{
- logf (LOG_FATAL, "Cannot open resource `%s'", base_name);
- exit (1);
+ zebra_commit (zh);
}
- }
- else if(cmd == 0) /* command */
- {
- if (!strcmp (arg, "add"))
+ else if (!strcmp (arg, "clean"))
{
- cmd = 'a';
+ zebra_clean (zh);
}
- else if (!strcmp (arg, "del"))
+ else if (!strcmp (arg, "stat") || !strcmp (arg, "status"))
{
- cmd = 'd';
+ zebra_register_statistics (zh,0);
+ }
+ else if (!strcmp (arg, "dumpdict"))
+ {
+ zebra_register_statistics (zh,1);
+ }
+ else if (!strcmp (arg, "compact"))
+ {
+ zebra_compact (zh);
+ }
+ else if (!strcmp (arg, "filters"))
+ {
+ show_filters(zs);
}
else
{
- logf (LOG_FATAL, "Unknown command: %s", arg);
+ yaz_log (YLOG_FATAL, "unknown command: %s", arg);
exit (1);
}
}
- else
+ else
{
- unlink ("keys.tmp");
- key_open (mem_max);
- repository (cmd, arg, base_path);
- cmd = 0;
+ ZEBRA_RES res = ZEBRA_OK;
+ if (!trans_started)
+ {
+ trans_started=1;
+ if (zebra_begin_trans (zh, 1) != ZEBRA_OK)
+ exit(1);
+ }
+ switch (cmd)
+ {
+ case 'u':
+ res = zebra_repository_update (zh, arg);
+ break;
+ case 'd':
+ res = zebra_repository_delete (zh, arg);
+ break;
+ case 's':
+ res = zebra_repository_show (zh, arg);
+ nsections = 0;
+ break;
+ case 'C':
+ res = zebra_create_database(zh, arg);
+ break;
+ case 'D':
+ res = zebra_drop_database(zh, arg);
+ break;
+ default:
+ nsections = 0;
+ }
+ if (res != ZEBRA_OK)
+ {
+ yaz_log(YLOG_WARN, "Operation failed");
+ exit(1);
+ }
+ log_event_end (NULL, NULL);
}
}
- else if (ret == 'v')
+ else if (ret == 'V')
{
- log_init (log_mask_str(arg), prog, NULL);
- }
- else if (ret == 'r')
- {
- base_path = arg;
+ printf("Zebra %s %s\n", ZEBRAVER, ZEBRADATE);
+ printf(" (C) 1994-2005, Index Data ApS\n");
+#ifdef WIN32
+#ifdef _DEBUG
+ printf(" WIN32 Debug\n");
+#else
+ printf(" WIN32 Release\n");
+#endif
+#endif
+#if HAVE_BZLIB_H
+ printf("Using: libbzip2, (C) 1996-1999 Julian R Seward. All rights reserved.\n");
+#endif
}
+ else if (ret == 'v')
+ yaz_log_init_level (yaz_log_mask_str(arg));
+ else if (ret == 'l')
+ yaz_log_init_file (arg);
else if (ret == 'm')
- {
- mem_max = 1024*1024*atoi(arg);
- }
+ res_set(res, "memMax", arg);
+ else if (ret == 'd')
+ database = arg;
+ else if (ret == 's')
+ res_set(res, "openRW", "0");
+ else if (ret == 'g')
+ res_set(res, "group", arg);
+ else if (ret == 'f')
+ res_set(res, "fileVerboseLimit", arg);
+ else if (ret == 'c')
+ configName = arg;
+ else if (ret == 't')
+ res_set(res, "recordType", arg);
+ else if (ret == 'n')
+ enable_commit = 0;
+ else if (ret == 'L')
+ res_set(res, "followLinks", "0");
else
- {
- logf (LOG_FATAL, "Unknown option '-%s'", arg);
- exit (1);
- }
- }
- if (!base_name)
+ yaz_log (YLOG_WARN, "unknown option '-%s'", arg);
+ } /* while arg */
+
+ if (trans_started)
+ if (zebra_end_trans (zh) != ZEBRA_OK)
+ yaz_log (YLOG_WARN, "zebra_end_trans failed");
+
+ zebra_close (zh);
+ zebra_stop (zs);
+#if HAVE_SYS_TIMES_H
+#if HAVE_SYS_TIME_H
+ if (trans_started)
{
- fprintf (stderr, "index [-v log] [-r repository] "
- "base cmd1 dir1 cmd2 dir2 ...\n");
- exit (1);
+ gettimeofday(&end_time, 0);
+ usec = (end_time.tv_sec - start_time.tv_sec) * 1000000.0 +
+ end_time.tv_usec - start_time.tv_usec;
+ times(&tms2);
+ yaz_log (YLOG_LOG, "zebraidx times: %5.2f %5.2f %5.2f",
+ usec / 1000000,
+ (double) (tms2.tms_utime - tms1.tms_utime)/100,
+ (double) (tms2.tms_stime - tms1.tms_stime)/100);
}
- nsections = key_close ();
- if (!nsections)
- exit (0);
- logf (LOG_LOG, "Merge sorting");
- mbuf = xmalloc (100000);
- merge_sort (mbuf, 1, nsections+1);
- xfree (mbuf);
- logf (LOG_LOG, "Input");
- key_input (FNAME_WORD_DICT, FNAME_WORD_ISAM, "keys1.tmp", 60);
+#endif
+#endif
+ nmem_exit();
exit (0);
+ return 0;
}