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