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