index/main.c

   1 /*
   2  * Copyright (C) 1994-2002, Index Data
   3  * All rights reserved.
   4  *
   5  * $Id: main.c,v 1.82 2002-02-20 17:30:01 adam Exp $
   6  */
   7 #include <stdio.h>
   8 #include <string.h>
   9 #include <assert.h>
  10 #ifdef WIN32
  11 #include <io.h>
  12 #else
  13 #include <unistd.h>
  14 #endif
  15
  16 #include <yaz/data1.h>
  17 #include "zebraapi.h"
  18 #include "zserver.h"
  19 #include "index.h"
  20 #include "recindex.h"
  21
  22 char *prog;
  23
  24 int main (int argc, char **argv)
  25 {
  26     int ret;
  27     int cmd = 0;
  28     char *arg;
  29     char *configName = 0;
  30     int nsections = 0;
  31     int disableCommit = 0;
  32     size_t mem_max = 0;
  33 #ifndef WIN32
  34     char nbuf[100];
  35 #endif
  36     struct recordGroup rGroupDef;
  37     ZebraService zs = 0;
  38     ZebraHandle zh = 0;
  39
  40     nmem_init ();
  41
  42 #ifdef WIN32
  43 #else
  44     sprintf(nbuf, "%.40s(%d)", *argv, getpid());
  45     yaz_log_init_prefix (nbuf);
  46 #endif
  47
  48     rGroupDef.groupName = NULL;
  49     rGroupDef.databaseName = NULL;
  50     rGroupDef.path = NULL;
  51     rGroupDef.recordId = NULL;
  52     rGroupDef.recordType = NULL;
  53     rGroupDef.flagStoreData = -1;
  54     rGroupDef.flagStoreKeys = -1;
  55     rGroupDef.flagRw = 1;
  56     rGroupDef.databaseNamePath = 0;
  57     rGroupDef.explainDatabase = 0;
  58     rGroupDef.fileVerboseLimit = 100000;
  59
  60     prog = *argv;
  61     if (argc < 2)
  62     {
  63         fprintf (stderr, "%s [options] command <dir> ...\n"
  64         "Commands:\n"
  65         " update <dir>  Update index with files below <dir>.\n"
  66         "               If <dir> is empty filenames are read from stdin.\n"
  67         " delete <dir>  Delete index with files below <dir>.\n"
  68         " commit        Commit changes\n"
  69         " clean         Clean shadow files\n"
  70         "Options:\n"
  71         " -t <type>     Index files as <type> (grs or text).\n"
  72         " -c <config>   Read configuration file <config>.\n"
  73         " -g <group>    Index files according to group settings.\n"
  74         " -d <database> Records belong to Z39.50 database <database>.\n"
  75         " -m <mbytes>   Use <mbytes> before flushing keys to disk.\n"
  76         " -n            Don't use shadow system.\n"
  77         " -s            Show analysis on stdout, but do no work.\n"
  78         " -v <level>    Set logging to <level>.\n"
  79         " -l <file>     Write log to <file>.\n"
  80         " -f <n>        Display information for the first <n> records.\n"
  81         " -V            Show version.\n", *argv
  82                  );
  83         exit (1);
  84     }
  85     while ((ret = options ("sVt:c:g:d:m:v:nf:l:"
  86                            , argv, argc, &arg)) != -2)
  87     {
  88         if (ret == 0)
  89         {
  90             if(cmd == 0) /* command */
  91             {
  92                 if (!zs)
  93                 {
  94 #if ZMBOL
  95                     logf (LOG_LOG, "zmbol version %s %s",
  96                           ZEBRAVER, ZEBRADATE);
  97 #else
  98                     logf (LOG_LOG, "zebra version %s %s",
  99                           ZEBRAVER, ZEBRADATE);
 100 #endif
 101                     zs = zebra_start (configName ? configName : FNAME_CONFIG);
 102
 103                     zh = zebra_open (zs);
 104                 }
 105                 if (!strcmp (arg, "update"))
 106                     cmd = 'u';
 107                 else if (!strcmp (arg, "update1"))
 108                     cmd = 'U';
 109                 else if (!strcmp (arg, "update2"))
 110                     cmd = 'm';
 111                 else if (!strcmp (arg, "dump"))
 112                     cmd = 's';
 113                 else if (!strcmp (arg, "del") || !strcmp(arg, "delete"))
 114                     cmd = 'd';
 115                 else if (!strcmp (arg, "init"))
 116                 {
 117                     zebra_init (zh);
 118                 }
 119                 else if (!strcmp (arg, "commit"))
 120                 {
 121                     zebra_commit (zh);
 122                 }
 123                 else if (!strcmp (arg, "clean"))
 124                 {
 125                     assert (!"todo");
 126                 }
 127                 else if (!strcmp (arg, "stat") || !strcmp (arg, "status"))
 128                 {
 129                     assert (!"todo");
 130                 }
 131                 else if (!strcmp (arg, "compact"))
 132                 {
 133                     zebra_compact (zh);
 134                 }
 135                 else
 136                 {
 137                     logf (LOG_FATAL, "unknown command: %s", arg);
 138                     exit (1);
 139                 }
 140             }
 141             else
 142             {
 143                 memcpy (&zh->rGroup, &rGroupDef, sizeof(rGroupDef));
 144                 zebra_begin_trans (zh);
 145
 146                 zh->rGroup.path = arg;
 147                 switch (cmd)
 148                 {
 149                 case 'u':
 150                     zebra_repository_update (zh);
 151                     break;
 152                 case 'd':
 153                     zebra_repository_delete (zh);
 154                     break;
 155                 case 's':
 156                     logf (LOG_LOG, "dumping %s", zh->rGroup.path);
 157                     zebra_repository_show (zh);
 158                     nsections = 0;
 159                     break;
 160                 default:
 161                     nsections = 0;
 162                 }
 163                 cmd = 0;
 164                 zebra_end_trans (zh);
 165                 log_event_end (NULL, NULL);
 166             }
 167         }
 168         else if (ret == 'V')
 169         {
 170 #if ZMBOL
 171             fprintf (stderr, "Z'mbol %s %s\n", ZEBRAVER, ZEBRADATE);
 172 #else
 173             fprintf (stderr, "Zebra %s %s\n", ZEBRAVER, ZEBRADATE);
 174 #endif
 175             fprintf (stderr, " (C) 1994-2002, Index Data ApS\n");
 176 #ifdef WIN32
 177 #ifdef _DEBUG
 178             fprintf (stderr, " WIN32 Debug\n");
 179 #else
 180             fprintf (stderr, " WIN32 Release\n");
 181 #endif
 182 #endif
 183 #if HAVE_BZLIB_H
 184             fprintf (stderr, "libbzip2\n"
 185                      " (C) 1996-1999 Julian R Seward.  All rights reserved.\n");
 186 #endif
 187         }
 188         else if (ret == 'v')
 189             yaz_log_init_level (yaz_log_mask_str(arg));
 190         else if (ret == 'l')
 191             yaz_log_init_file (arg);
 192         else if (ret == 'm')
 193             mem_max = 1024*1024*atoi(arg);
 194         else if (ret == 'd')
 195             rGroupDef.databaseName = arg;
 196         else if (ret == 's')
 197             rGroupDef.flagRw = 0;
 198         else if (ret == 'g')
 199             rGroupDef.groupName = arg;
 200         else if (ret == 'f')
 201             rGroupDef.fileVerboseLimit = atoi(arg);
 202         else if (ret == 'c')
 203             configName = arg;
 204         else if (ret == 't')
 205             rGroupDef.recordType = arg;
 206         else if (ret == 'n')
 207             disableCommit = 1;
 208         else
 209             logf (LOG_WARN, "unknown option '-%s'", arg);
 210     }
 211     zebra_close (zh);
 212     zebra_stop (zs);
 213     exit (0);
 214     return 0;
 215 }
 216