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