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