Fix option -f
[idzebra-moved-to-github.git] / index / main.c
1 /* $Id: main.c,v 1.111 2004-05-30 18:05:30 adam Exp $
2    Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
3    Index Data Aps
4
5 This file is part of the Zebra server.
6
7 Zebra is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 2, or (at your option) any later
10 version.
11
12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
15 for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
20 02111-1307, USA.
21 */
22
23
24 #include <stdio.h>
25 #include <string.h>
26 #include <assert.h>
27 #ifdef WIN32
28 #include <io.h>
29 #else
30 #include <unistd.h>
31 #include <sys/time.h>
32 #endif
33 #include <time.h>
34 #if HAVE_SYS_TIMES_H
35 #include <sys/times.h>
36 #endif
37
38 #if HAVE_TCL_H
39 #include <tcl.h>
40 #endif
41
42 #if HAVE_EXPAT_H
43 #include <expat.h>
44 #endif
45
46 #include <data1.h>
47 #include "zebraapi.h"
48
49 char *prog;
50
51 int main (int argc, char **argv)
52 {
53     int ret;
54     int cmd = 0;
55     char *arg;
56     char *configName = 0;
57     int nsections = 0;
58     int disableCommit = 0;
59     char *database = "Default";
60     Res res = res_open(0, 0, 0);
61     
62     int trans_started=0;
63 #if HAVE_SYS_TIMES_H
64     struct tms tms1, tms2;
65     struct timeval start_time, end_time;
66     double usec;
67 #endif
68 #ifndef WIN32
69     char nbuf[100];
70 #endif
71     ZebraService zs = 0;
72     ZebraHandle zh = 0;
73
74     nmem_init ();
75
76 #ifdef WIN32
77 #else
78     sprintf(nbuf, "%.40s(%d)", *argv, getpid());
79     yaz_log_init_prefix (nbuf);
80 #endif
81 #if HAVE_SYS_TIMES_H
82     times(&tms1);
83     gettimeofday(&start_time, 0);
84 #endif
85     prog = *argv;
86     if (argc < 2)
87     {
88         fprintf (stderr, "%s [options] command <dir> ...\n"
89         "Commands:\n"
90         " update <dir>  Update index with files below <dir>.\n"
91         "               If <dir> is empty filenames are read from stdin.\n"
92         " delete <dir>  Delete index with files below <dir>.\n"
93         " create <db>   Create database <db>\n"
94         " drop <db>     Drop database <db>\n"
95         " commit        Commit changes\n"
96         " clean         Clean shadow files\n"
97         "Options:\n"
98         " -t <type>     Index files as <type> (grs or text).\n"
99         " -c <config>   Read configuration file <config>.\n"
100         " -g <group>    Index files according to group settings.\n"
101         " -d <database> Records belong to Z39.50 database <database>.\n"
102         " -m <mbytes>   Use <mbytes> before flushing keys to disk.\n"
103         " -n            Don't use shadow system.\n"
104         " -s            Show analysis on stdout, but do no work.\n"
105         " -v <level>    Set logging to <level>.\n"
106         " -l <file>     Write log to <file>.\n"
107         " -L            Don't follow symbolic links.\n"
108         " -f <n>        Display information for the first <n> records.\n"
109         " -V            Show version.\n", *argv
110                  );
111         exit (1);
112     }
113     while ((ret = options ("sVt:c:g:d:m:v:nf:l:L"
114                            , argv, argc, &arg)) != -2)
115     {
116         if (ret == 0)
117         {
118             if(cmd == 0) /* command */
119             {
120                 if (!zs)
121                 {
122                     const char *config = configName ? configName : "zebra.cfg";
123                     logf (LOG_LOG, "Zebra version %s %s",
124                           ZEBRAVER, ZEBRADATE);
125                     zs = zebra_start (config, 0, res);
126                     if (!zs)
127                     {
128                         yaz_log (LOG_FATAL, "Cannot read config %s", config);
129                         exit (1);
130                     }   
131                     zh = zebra_open (zs);
132                     if (disableCommit)
133                         zebra_shadow_enable (zh, 0);
134                 }
135
136                 if (zebra_select_database (zh, database))
137                 {
138                     logf(LOG_FATAL, "Could not select database %s errCode=%d",
139                          database, zebra_errCode(zh) );
140                     exit (1);
141                 }
142                 if (!strcmp (arg, "update"))
143                     cmd = 'u';
144                 else if (!strcmp (arg, "update1"))
145                     cmd = 'U';
146                 else if (!strcmp (arg, "update2"))
147                     cmd = 'm';
148                 else if (!strcmp (arg, "dump"))
149                     cmd = 's';
150                 else if (!strcmp (arg, "del") || !strcmp(arg, "delete"))
151                     cmd = 'd';
152                 else if (!strcmp (arg, "init"))
153                 {
154                     zebra_init (zh);
155                 }
156                 else if (!strcmp(arg, "drop"))
157                 {
158                     cmd = 'D';
159                 }
160                 else if (!strcmp(arg, "create"))
161                 {
162                     cmd = 'C';
163                 }
164                 else if (!strcmp (arg, "commit"))
165                 {
166                     zebra_commit (zh);
167                 }
168                 else if (!strcmp (arg, "clean"))
169                 {
170                     zebra_clean (zh);
171                 }
172                 else if (!strcmp (arg, "stat") || !strcmp (arg, "status"))
173                 {
174                     zebra_register_statistics (zh,0);
175                 }
176                 else if (!strcmp (arg, "dumpdict"))
177                 {
178                     zebra_register_statistics (zh,1);
179                 }
180                 else if (!strcmp (arg, "compact"))
181                 {
182                     zebra_compact (zh);
183                 }
184                 else
185                 {
186                     logf (LOG_FATAL, "unknown command: %s", arg);
187                     exit (1);
188                 }
189             }
190             else
191             {
192                 if (!trans_started)
193                 {
194                     trans_started=1;
195                     if (zebra_begin_trans (zh, 1))
196                         exit(1);
197                 }
198                 switch (cmd)
199                 {
200                 case 'u':
201                     zebra_repository_update (zh, arg);
202                     break;
203                 case 'd':
204                     zebra_repository_delete (zh, arg);
205                     break;
206                 case 's':
207                     zebra_repository_show (zh, arg);
208                     nsections = 0;
209                     break;
210                 case 'C':
211                     zebra_create_database(zh, arg);
212                     break;
213                 case 'D':
214                     zebra_drop_database(zh, arg);
215                     break;
216                 default:
217                     nsections = 0;
218                 }
219                 log_event_end (NULL, NULL);
220             }
221         }
222         else if (ret == 'V')
223         {
224             printf("Zebra %s %s\n", ZEBRAVER, ZEBRADATE);
225             printf(" (C) 1994-2004, Index Data ApS\n");
226 #ifdef WIN32
227 #ifdef _DEBUG
228             printf(" WIN32 Debug\n");
229 #else
230             printf(" WIN32 Release\n");
231 #endif
232 #endif
233 #if HAVE_BZLIB_H
234             printf("Using: libbzip2, (C) 1996-1999 Julian R Seward.  All rights reserved.\n");
235 #endif
236 #if HAVE_TCL_H
237             printf("Using: Tcl %s\n", TCL_VERSION);
238 #endif
239 #if HAVE_EXPAT_H
240             if (1)
241             {
242                 XML_Expat_Version v = XML_ExpatVersionInfo();
243                 printf ("Using: Expat %d.%d.%d\n",
244                         v.major, v.minor, v.micro);
245             }
246 #endif
247         }
248         else if (ret == 'v')
249             yaz_log_init_level (yaz_log_mask_str(arg));
250         else if (ret == 'l')
251             yaz_log_init_file (arg);
252         else if (ret == 'm')
253             res_set(res, "memMax", arg);
254         else if (ret == 'd')
255             database = arg;
256         else if (ret == 's')
257             res_set(res, "openRW", "0");
258         else if (ret == 'g')
259             res_set(res, "group", arg);
260         else if (ret == 'f')
261             res_set(res, "fileVerboseLimit", arg);
262         else if (ret == 'c')
263             configName = arg;
264         else if (ret == 't')
265             res_set(res, "recordType", arg);
266         else if (ret == 'n')
267             res_set(res, "disableCommit", "1");
268         else if (ret == 'L')
269             res_set(res, "followLinks", "0");
270         else
271             logf (LOG_WARN, "unknown option '-%s'", arg);
272     } /* while arg */
273
274     if (trans_started)
275         zebra_end_trans (zh);
276
277     zebra_close (zh);
278     zebra_stop (zs);
279 #if HAVE_SYS_TIMES_H
280     if (trans_started)
281     {
282         gettimeofday(&end_time, 0);
283         usec = (end_time.tv_sec - start_time.tv_sec) * 1000000.0 +
284             end_time.tv_usec - start_time.tv_usec;
285         times(&tms2);
286         yaz_log (LOG_LOG, "zebraidx times: %5.2f %5.2f %5.2f",
287                 usec / 1000000,
288                 (double) (tms2.tms_utime - tms1.tms_utime)/100,
289                 (double) (tms2.tms_stime - tms1.tms_stime)/100);
290     }
291 #endif
292     nmem_exit();
293     exit (0);
294     return 0;
295 }
296