index/zserver.c

   1 /*
   2  * Copyright (C) 1994-1995, Index Data I/S
   3  * All rights reserved.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: zserver.c,v $
   7  * Revision 1.30  1995-12-07 17:38:48  adam
   8  * Work locking mechanisms for concurrent updates/commit.
   9  *
  10  * Revision 1.29  1995/12/04  14:22:32  adam
  11  * Extra arg to recType_byName.
  12  * Started work on new regular expression parsed input to
  13  * structured records.
  14  *
  15  * Revision 1.28  1995/11/28  09:09:48  adam
  16  * Zebra config renamed.
  17  * Use setting 'recordId' to identify record now.
  18  * Bug fix in recindex.c: rec_release_blocks was invokeded even
  19  * though the blocks were already released.
  20  * File traversal properly deletes records when needed.
  21  *
  22  * Revision 1.27  1995/11/27  13:58:54  adam
  23  * New option -t. storeStore data implemented in server.
  24  *
  25  * Revision 1.26  1995/11/25  10:24:07  adam
  26  * More record fields - they are enumerated now.
  27  * New options: flagStoreData flagStoreKey.
  28  *
  29  * Revision 1.25  1995/11/21  15:29:13  adam
  30  * Config file 'base' read by default by both indexer and server.
  31  *
  32  * Revision 1.24  1995/11/20  16:59:47  adam
  33  * New update method: the 'old' keys are saved for each records.
  34  *
  35  * Revision 1.23  1995/11/16  17:00:56  adam
  36  * Better logging of rpn query.
  37  *
  38  * Revision 1.22  1995/11/16  15:34:55  adam
  39  * Uses new record management system in both indexer and server.
  40  *
  41  * Revision 1.21  1995/11/01  16:25:52  quinn
  42  * *** empty log message ***
  43  *
  44  * Revision 1.20  1995/10/27  14:00:12  adam
  45  * Implemented detection of database availability.
  46  *
  47  * Revision 1.19  1995/10/17  18:02:11  adam
  48  * New feature: databases. Implemented as prefix to words in dictionary.
  49  *
  50  * Revision 1.18  1995/10/16  14:03:09  quinn
  51  * Changes to support element set names and espec1
  52  *
  53  * Revision 1.17  1995/10/16  09:32:40  adam
  54  * More work on relational op.
  55  *
  56  * Revision 1.16  1995/10/13  12:26:44  adam
  57  * Optimization of truncation.
  58  *
  59  * Revision 1.15  1995/10/12  12:40:55  adam
  60  * Bug fixes in rpn_prox.
  61  *
  62  * Revision 1.14  1995/10/09  16:18:37  adam
  63  * Function dict_lookup_grep got extra client data parameter.
  64  *
  65  * Revision 1.13  1995/10/06  14:38:00  adam
  66  * New result set method: r_score.
  67  * Local no (sysno) and score is transferred to retrieveCtrl.
  68  *
  69  * Revision 1.12  1995/10/06  13:52:06  adam
  70  * Bug fixes. Handler may abort further scanning.
  71  *
  72  * Revision 1.11  1995/10/06  10:43:57  adam
  73  * Scan added. 'occurrences' in scan entries not set yet.
  74  *
  75  * Revision 1.10  1995/10/02  16:43:32  quinn
  76  * Set default resulting record type in fetch.
  77  *
  78  * Revision 1.9  1995/10/02  15:18:52  adam
  79  * New member in recRetrieveCtrl: diagnostic.
  80  *
  81  * Revision 1.8  1995/09/28  09:19:47  adam
  82  * xfree/xmalloc used everywhere.
  83  * Extract/retrieve method seems to work for text records.
  84  *
  85  * Revision 1.7  1995/09/27  16:17:32  adam
  86  * More work on retrieve.
  87  *
  88  * Revision 1.6  1995/09/08  08:53:22  adam
  89  * Record buffer maintained in server_info.
  90  *
  91  * Revision 1.5  1995/09/06  16:11:18  adam
  92  * Option: only one word key per file.
  93  *
  94  * Revision 1.4  1995/09/06  10:33:04  adam
  95  * More work on present. Some log messages removed.
  96  *
  97  * Revision 1.3  1995/09/05  15:28:40  adam
  98  * More work on search engine.
  99  *
 100  * Revision 1.2  1995/09/04  12:33:43  adam
 101  * Various cleanup. YAZ util used instead.
 102  *
 103  * Revision 1.1  1995/09/04  09:10:41  adam
 104  * More work on index add/del/update.
 105  * Merge sort implemented.
 106  * Initial work on z39 server.
 107  *
 108  */
 109 #include <stdio.h>
 110 #include <assert.h>
 111 #include <unistd.h>
 112 #include <fcntl.h>
 113
 114 #include <common.h>
 115 #include <data1.h>
 116 #include <recctrl.h>
 117 #include <dmalloc.h>
 118 #include "zserver.h"
 119
 120 ZServerInfo server_info;
 121
 122 static int register_check (ZServerInfo *zi)
 123 {
 124     int state = zebraServerLockGetState();
 125
 126     switch (state)
 127     {
 128     case 'c':
 129         state = 1;
 130         break;
 131     default:
 132         state = 0;
 133     }
 134     if (zi->registerState == state)
 135         return 0;
 136
 137     zi->registerState = state;
 138     if (server_info.records)
 139     {
 140         dict_close (server_info.wordDict);
 141         is_close (server_info.wordIsam);
 142         rec_close (&server_info.records);
 143     }
 144     /* enable commit if state is 1 */
 145     server_info.records = rec_open (0);
 146     if (!(server_info.wordDict = dict_open (FNAME_WORD_DICT, 40, 0)))
 147         return -1;
 148     if (!(server_info.wordIsam = is_open (FNAME_WORD_ISAM, key_compare, 0,
 149                                           sizeof (struct it_key))))
 150         return -1;
 151     return 0;
 152 }
 153
 154 bend_initresult *bend_init (bend_initrequest *q)
 155 {
 156     static bend_initresult r;
 157     static char *name = "zserver";
 158
 159     r.errcode = 0;
 160     r.errstring = 0;
 161     r.handle = name;
 162
 163     logf (LOG_DEBUG, "bend_init");
 164
 165     if (!common_resource)
 166     {
 167         struct statserv_options_block *sob;
 168
 169         sob = statserv_getcontrol ();
 170         logf (LOG_LOG, "Reading resources from %s", sob->configname);
 171         if (!(common_resource = res_open (sob->configname)))
 172         {
 173             logf (LOG_FATAL, "Cannot open resource `%s'", sob->configname);
 174             exit (1);
 175         }
 176     }
 177     zebraServerLock ();
 178
 179     data1_tabpath = res_get(common_resource, "profilePath");
 180     server_info.sets = NULL;
 181     server_info.registerState = -1;  /* trigger open of registers! */
 182 #if 1
 183     server_info.records = NULL;
 184     server_info.wordDict = NULL;
 185     server_info.wordIsam = NULL;
 186 #else
 187     server_info.records = rec_open (0);
 188     if (!(server_info.wordDict = dict_open (FNAME_WORD_DICT, 40, 0)))
 189     {
 190         logf (LOG_WARN, "dict_open fail: word dict");
 191         r.errcode = 1;
 192         r.errstring = "dict_open fail: word dict";
 193         return &r;
 194     }
 195     if (!(server_info.wordIsam = is_open (FNAME_WORD_ISAM, key_compare, 0,
 196                                           sizeof (struct it_key))))
 197     {
 198         logf (LOG_WARN, "is_open fail: word isam");
 199         dict_close (server_info.wordDict);
 200         r.errcode = 1;
 201         r.errstring = "is_open fail: word isam";
 202         return &r;
 203     }
 204 #endif
 205     server_info.odr = odr_createmem (ODR_ENCODE);
 206     return &r;
 207 }
 208
 209 bend_searchresult *bend_search (void *handle, bend_searchrequest *q, int *fd)
 210 {
 211     static bend_searchresult r;
 212
 213     r.errcode = 0;
 214     r.errstring = 0;
 215     r.hits = 0;
 216
 217     register_check (&server_info);
 218     odr_reset (server_info.odr);
 219     server_info.errCode = 0;
 220     server_info.errString = NULL;
 221
 222     logf (LOG_LOG, "ResultSet '%s'", q->setname);
 223     switch (q->query->which)
 224     {
 225     case Z_Query_type_1:
 226         r.errcode = rpn_search (&server_info, q->query->u.type_1,
 227                                 q->num_bases, q->basenames, q->setname,
 228                                 &r.hits);
 229         r.errstring = server_info.errString;
 230         break;
 231     default:
 232         r.errcode = 107;
 233     }
 234     return &r;
 235 }
 236
 237 static int record_ext_read (int fd, char *buf, size_t count)
 238 {
 239     return read (fd, buf, count);
 240 }
 241
 242 static int record_int_pos;
 243 static char *record_int_buf;
 244 static int record_int_len;
 245
 246 static int record_int_read (int fd, char *buf, size_t count)
 247 {
 248     int l = record_int_len - record_int_pos;
 249     if (l <= 0)
 250         return 0;
 251     l = (l < count) ? l : count;
 252     memcpy (buf, record_int_buf + record_int_pos, l);
 253     record_int_pos += l;
 254     return l;
 255 }
 256
 257 static int record_fetch (ZServerInfo *zi, int sysno, int score, ODR stream,
 258                           oid_value input_format, Z_RecordComposition *comp,
 259                           oid_value *output_format, char **rec_bufp,
 260                           int *rec_lenp)
 261 {
 262     Record rec;
 263     char *fname, *file_type;
 264     RecType rt;
 265     struct recRetrieveCtrl retrieveCtrl;
 266     char subType[128];
 267
 268     rec = rec_get (zi->records, sysno);
 269     file_type = rec->info[recInfo_fileType];
 270     fname = rec->info[recInfo_filename];
 271
 272     if (!(rt = recType_byName (file_type, subType)))
 273     {
 274         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: Cannot handle type %s",
 275               file_type);
 276         exit (1);
 277     }
 278     logf (LOG_DEBUG, "retrieve localno=%d score=%d", sysno, score);
 279     if (rec->size[recInfo_storeData] > 0)
 280     {
 281         retrieveCtrl.readf = record_int_read;
 282         record_int_len = rec->size[recInfo_storeData];
 283         record_int_buf = rec->info[recInfo_storeData];
 284         record_int_pos = 0;
 285         logf (LOG_DEBUG, "Internal retrieve. %d bytes", record_int_len);
 286     }
 287     else
 288     {
 289         if ((retrieveCtrl.fd = open (fname, O_RDONLY)) == -1)
 290         {
 291             char *msg = "Record doesn't exist";
 292             logf (LOG_WARN|LOG_ERRNO, "Retrieve: Open record file %s", fname);
 293             *output_format = VAL_SUTRS;
 294             *rec_bufp = msg;
 295             *rec_lenp = strlen (msg);
 296             rec_rm (&rec);
 297             return 0;     /* or 14: System error in presenting records */
 298         }
 299         retrieveCtrl.readf = record_ext_read;
 300     }
 301     retrieveCtrl.subType = subType;
 302     retrieveCtrl.localno = sysno;
 303     retrieveCtrl.score = score;
 304     retrieveCtrl.odr = stream;
 305     retrieveCtrl.input_format = retrieveCtrl.output_format = input_format;
 306     retrieveCtrl.comp = comp;
 307     retrieveCtrl.diagnostic = 0;
 308     (*rt->retrieve)(&retrieveCtrl);
 309     *output_format = retrieveCtrl.output_format;
 310     *rec_bufp = retrieveCtrl.rec_buf;
 311     *rec_lenp = retrieveCtrl.rec_len;
 312     close (retrieveCtrl.fd);
 313     rec_rm (&rec);
 314
 315     return retrieveCtrl.diagnostic;
 316 }
 317
 318 bend_fetchresult *bend_fetch (void *handle, bend_fetchrequest *q, int *num)
 319 {
 320     static bend_fetchresult r;
 321     int positions[2];
 322     ZServerSetSysno *records;
 323
 324     register_check (&server_info);
 325
 326     r.errstring = 0;
 327     r.last_in_set = 0;
 328     r.basename = "base";
 329
 330     odr_reset (server_info.odr);
 331     server_info.errCode = 0;
 332
 333     positions[0] = q->number;
 334     records = resultSetSysnoGet (&server_info, q->setname, 1, positions);
 335     if (!records)
 336     {
 337         logf (LOG_DEBUG, "resultSetRecordGet, error");
 338         r.errcode = 13;
 339         return &r;
 340     }
 341     if (!records[0].sysno)
 342     {
 343         r.errcode = 13;
 344         logf (LOG_DEBUG, "Out of range. pos=%d", q->number);
 345         return &r;
 346     }
 347     r.errcode = record_fetch (&server_info, records[0].sysno,
 348                               records[0].score, q->stream, q->format,
 349                               q->comp, &r.format, &r.record, &r.len);
 350     resultSetSysnoDel (&server_info, records, 1);
 351     return &r;
 352 }
 353
 354 bend_deleteresult *bend_delete (void *handle, bend_deleterequest *q, int *num)
 355 {
 356     register_check (&server_info);
 357     return 0;
 358 }
 359
 360 bend_scanresult *bend_scan (void *handle, bend_scanrequest *q, int *num)
 361 {
 362     static bend_scanresult r;
 363     int status;
 364
 365     register_check (&server_info);
 366     odr_reset (server_info.odr);
 367     server_info.errCode = 0;
 368     server_info.errString = 0;
 369
 370     r.term_position = q->term_position;
 371     r.num_entries = q->num_entries;
 372     r.errcode = rpn_scan (&server_info, q->term,
 373                           q->num_bases, q->basenames,
 374                           &r.term_position,
 375                           &r.num_entries, &r.entries, &status);
 376     r.errstring = server_info.errString;
 377     r.status = status;
 378     return &r;
 379 }
 380
 381 void bend_close (void *handle)
 382 {
 383     if (server_info.records)
 384     {
 385         dict_close (server_info.wordDict);
 386         is_close (server_info.wordIsam);
 387         rec_close (&server_info.records);
 388     }
 389     zebraServerUnlock ();
 390     return;
 391 }
 392
 393 int main (int argc, char **argv)
 394 {
 395     struct statserv_options_block *sob;
 396
 397     sob = statserv_getcontrol ();
 398     strcpy (sob->configname, FNAME_CONFIG);
 399     statserv_setcontrol (sob);
 400
 401     return statserv_main (argc, argv);
 402 }