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