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.14  1995-10-09 16:18:37  adam
   8  * Function dict_lookup_grep got extra client data parameter.
   9  *
  10  * Revision 1.13  1995/10/06  14:38:00  adam
  11  * New result set method: r_score.
  12  * Local no (sysno) and score is transferred to retrieveCtrl.
  13  *
  14  * Revision 1.12  1995/10/06  13:52:06  adam
  15  * Bug fixes. Handler may abort further scanning.
  16  *
  17  * Revision 1.11  1995/10/06  10:43:57  adam
  18  * Scan added. 'occurrences' in scan entries not set yet.
  19  *
  20  * Revision 1.10  1995/10/02  16:43:32  quinn
  21  * Set default resulting record type in fetch.
  22  *
  23  * Revision 1.9  1995/10/02  15:18:52  adam
  24  * New member in recRetrieveCtrl: diagnostic.
  25  *
  26  * Revision 1.8  1995/09/28  09:19:47  adam
  27  * xfree/xmalloc used everywhere.
  28  * Extract/retrieve method seems to work for text records.
  29  *
  30  * Revision 1.7  1995/09/27  16:17:32  adam
  31  * More work on retrieve.
  32  *
  33  * Revision 1.6  1995/09/08  08:53:22  adam
  34  * Record buffer maintained in server_info.
  35  *
  36  * Revision 1.5  1995/09/06  16:11:18  adam
  37  * Option: only one word key per file.
  38  *
  39  * Revision 1.4  1995/09/06  10:33:04  adam
  40  * More work on present. Some log messages removed.
  41  *
  42  * Revision 1.3  1995/09/05  15:28:40  adam
  43  * More work on search engine.
  44  *
  45  * Revision 1.2  1995/09/04  12:33:43  adam
  46  * Various cleanup. YAZ util used instead.
  47  *
  48  * Revision 1.1  1995/09/04  09:10:41  adam
  49  * More work on index add/del/update.
  50  * Merge sort implemented.
  51  * Initial work on z39 server.
  52  *
  53  */
  54 #include <stdio.h>
  55 #include <assert.h>
  56 #include <unistd.h>
  57 #include <fcntl.h>
  58
  59 #include <recctrl.h>
  60 #include <dmalloc.h>
  61 #include "zserver.h"
  62
  63 ZServerInfo server_info;
  64
  65 bend_initresult *bend_init (bend_initrequest *q)
  66 {
  67     static bend_initresult r;
  68     static char *name = "zserver";
  69
  70     r.errcode = 0;
  71     r.errstring = 0;
  72     r.handle = name;
  73
  74     server_info.sets = NULL;
  75     if (!(server_info.sys_idx_fd = open (FNAME_SYS_IDX, O_RDONLY)))
  76     {
  77         r.errcode = 1;
  78         r.errstring = "dict_open fail: filedict";
  79         return &r;
  80     }
  81     if (!(server_info.fileDict = dict_open (FNAME_FILE_DICT, 10, 0)))
  82     {
  83         r.errcode = 1;
  84         r.errstring = "dict_open fail: filedict";
  85         return &r;
  86     }
  87     if (!(server_info.wordDict = dict_open (FNAME_WORD_DICT, 40, 0)))
  88     {
  89         dict_close (server_info.fileDict);
  90         r.errcode = 1;
  91         r.errstring = "dict_open fail: worddict";
  92         return &r;
  93     }
  94     if (!(server_info.wordIsam = is_open (FNAME_WORD_ISAM, key_compare, 0,
  95                                           sizeof (struct it_key))))
  96     {
  97         dict_close (server_info.wordDict);
  98         dict_close (server_info.fileDict);
  99         r.errcode = 1;
 100         r.errstring = "is_open fail: wordisam";
 101         return &r;
 102     }
 103     server_info.odr = odr_createmem (ODR_ENCODE);
 104     return &r;
 105 }
 106
 107 bend_searchresult *bend_search (void *handle, bend_searchrequest *q, int *fd)
 108 {
 109     static bend_searchresult r;
 110
 111     r.errcode = 0;
 112     r.errstring = 0;
 113     r.hits = 0;
 114
 115     odr_reset (server_info.odr);
 116     server_info.errCode = 0;
 117     switch (q->query->which)
 118     {
 119     case Z_Query_type_1:
 120         r.errcode = rpn_search (&server_info, q->query->u.type_1,
 121                                 q->num_bases, q->basenames, q->setname,
 122                                 &r.hits);
 123         break;
 124     default:
 125         r.errcode = 107;
 126     }
 127     return &r;
 128 }
 129
 130 static int record_read (int fd, char *buf, size_t count)
 131 {
 132     return read (fd, buf, count);
 133 }
 134
 135 static int record_fetch (ZServerInfo *zi, int sysno, int score, ODR stream,
 136                           oid_value input_format, oid_value *output_format,
 137                           char **rec_bufp, int *rec_lenp)
 138 {
 139     char record_info[SYS_IDX_ENTRY_LEN];
 140     char *fname, *file_type;
 141     RecType rt;
 142     struct recRetrieveCtrl retrieveCtrl;
 143
 144     if (lseek (zi->sys_idx_fd, sysno * SYS_IDX_ENTRY_LEN,
 145                SEEK_SET) == -1)
 146     {
 147         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: lseek of sys_idx");
 148         exit (1);
 149     }
 150     if (read (zi->sys_idx_fd, record_info, SYS_IDX_ENTRY_LEN) == -1)
 151     {
 152         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: read of sys_idx");
 153         exit (1);
 154     }
 155     file_type = record_info;
 156     fname = record_info + strlen(record_info) + 1;
 157     if (!(rt = recType_byName (file_type)))
 158     {
 159         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: Cannot handle type %s",
 160               file_type);
 161         exit (1);
 162     }
 163     if ((retrieveCtrl.fd = open (fname, O_RDONLY)) == -1)
 164     {
 165         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: Open record file %s", fname);
 166         exit (1);
 167     }
 168     logf (LOG_DEBUG, "retrieve localno=%d score=%d", sysno, score);
 169     retrieveCtrl.localno = sysno;
 170     retrieveCtrl.score = score;
 171     retrieveCtrl.odr = stream;
 172     retrieveCtrl.readf = record_read;
 173     retrieveCtrl.input_format = retrieveCtrl.output_format = input_format;
 174     retrieveCtrl.diagnostic = 0;
 175     (*rt->retrieve)(&retrieveCtrl);
 176     *output_format = retrieveCtrl.output_format;
 177     *rec_bufp = retrieveCtrl.rec_buf;
 178     *rec_lenp = retrieveCtrl.rec_len;
 179     close (retrieveCtrl.fd);
 180     return retrieveCtrl.diagnostic;
 181 }
 182
 183 bend_fetchresult *bend_fetch (void *handle, bend_fetchrequest *q, int *num)
 184 {
 185     static bend_fetchresult r;
 186     int positions[2];
 187     ZServerSetSysno *records;
 188
 189     r.errstring = 0;
 190     r.last_in_set = 0;
 191     r.basename = "base";
 192
 193     odr_reset (server_info.odr);
 194     server_info.errCode = 0;
 195
 196     positions[0] = q->number;
 197     records = resultSetSysnoGet (&server_info, q->setname, 1, positions);
 198     if (!records)
 199     {
 200         logf (LOG_DEBUG, "resultSetRecordGet, error");
 201         r.errcode = 13;
 202         return &r;
 203     }
 204     if (!records[0].sysno)
 205     {
 206         r.errcode = 13;
 207         logf (LOG_DEBUG, "Out of range. pos=%d", q->number);
 208         return &r;
 209     }
 210     r.errcode = record_fetch (&server_info, records[0].sysno,
 211                               records[0].score, q->stream,
 212                               q->format, &r.format, &r.record, &r.len);
 213     return &r;
 214 }
 215
 216 bend_deleteresult *bend_delete (void *handle, bend_deleterequest *q, int *num)
 217 {
 218     return 0;
 219 }
 220
 221 bend_scanresult *bend_scan (void *handle, bend_scanrequest *q, int *num)
 222 {
 223     static bend_scanresult r;
 224     int status;
 225
 226     odr_reset (server_info.odr);
 227     server_info.errCode = 0;
 228
 229     r.errstring = 0;
 230     r.term_position = q->term_position;
 231     r.num_entries = q->num_entries;
 232     r.errcode = rpn_scan (&server_info, server_info.odr, q->term,
 233                           &r.term_position,
 234                           &r.num_entries, &r.entries, &status);
 235     r.status = status;
 236     return &r;
 237 }
 238
 239 void bend_close (void *handle)
 240 {
 241     dict_close (server_info.fileDict);
 242     dict_close (server_info.wordDict);
 243     is_close (server_info.wordIsam);
 244     close (server_info.sys_idx_fd);
 245     return;
 246 }
 247
 248 int main (int argc, char **argv)
 249 {
 250     char *base_name = "base";
 251
 252     if (!(common_resource = res_open (base_name)))
 253     {
 254         logf (LOG_FATAL, "Cannot open resource `%s'", base_name);
 255         exit (1);
 256     }
 257     return statserv_main (argc, argv);
 258 }