2a64c5fc1ae00d4f6f948923b2985eb550fd9c05
[yaz-moved-to-github.git] / src / zoom-p.h
1 /* This file is part of the YAZ toolkit.
2  * Copyright (C) Index Data.
3  * All rights reserved.
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are met:
6  *
7  *     * Redistributions of source code must retain the above copyright
8  *       notice, this list of conditions and the following disclaimer.
9  *     * Redistributions in binary form must reproduce the above copyright
10  *       notice, this list of conditions and the following disclaimer in the
11  *       documentation and/or other materials provided with the distribution.
12  *     * Neither the name of Index Data nor the names of its contributors
13  *       may be used to endorse or promote products derived from this
14  *       software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 /**
28  * \file zoom-p.h
29  * \brief Internal header for ZOOM implementation
30  */
31 #include <yaz/proto.h>
32 #include <yaz/oid_db.h>
33 #include <yaz/comstack.h>
34 #include <yaz/wrbuf.h>
35 #include <yaz/zoom.h>
36 #include <yaz/srw.h>
37 #include <yaz/cookie.h>
38 #include <yaz/mutex.h>
39 #if HAVE_LIBMEMCACHED_MEMCACHED_H
40 #include <libmemcached/memcached.h>
41 #endif
42
43 #define SHPTR 1
44
45 typedef struct ZOOM_Event_p *ZOOM_Event;
46
47 typedef enum {
48     zoom_sru_error,
49     zoom_sru_soap,
50     zoom_sru_get,
51     zoom_sru_post,
52     zoom_sru_solr
53 } zoom_sru_mode;
54
55
56 typedef struct ZOOM_task_p *ZOOM_task;
57
58 #define STATE_IDLE 0
59 #define STATE_CONNECTING 1
60 #define STATE_ESTABLISHED 2
61
62 struct ZOOM_connection_p {
63     enum oid_proto proto;
64     COMSTACK cs;
65     char *host_port;
66     int error;
67     char *addinfo;
68     char *diagset;
69     int state;
70     int mask;
71     int reconnect_ok;
72     ODR odr_in;
73     ODR odr_out;
74     ODR odr_print;
75     ODR odr_save;
76
77     char *buf_in;
78     int len_in;
79     char *buf_out;
80     int len_out;
81     char *proxy;
82     char *tproxy;
83     char *charset;
84     char *lang;
85     char *cookie_out;
86     char *cookie_in;
87     char *client_IP;
88     char *sru_version;
89
90     char *user;
91     char *group;
92     char *password;
93     int url_authentication;
94
95     int async;
96     int support_named_resultsets;
97     int last_event;
98
99     int maximum_record_size;
100     int preferred_message_size;
101
102     ZOOM_task tasks;
103     ZOOM_options options;
104     ZOOM_resultset resultsets;
105     ZOOM_Event m_queue_front;
106     ZOOM_Event m_queue_back;
107     zoom_sru_mode sru_mode;
108     int no_redirects; /* 0 for no redirects. >0 for number of redirects */
109     yaz_cookies_t cookies;
110
111     int log_details;
112     int log_api;
113     WRBUF saveAPDU_wrbuf;
114 #if HAVE_LIBMEMCACHED_MEMCACHED_H
115     memcached_st *mc_st;
116 #endif
117 };
118
119 typedef struct ZOOM_record_cache_p *ZOOM_record_cache;
120
121 #define RECORD_HASH_SIZE  131
122
123 struct ZOOM_resultset_p {
124     Z_SortKeySpecList *r_sort_spec;
125     ZOOM_query query;
126     int refcount;
127     Odr_int size;
128     int step;
129     int piggyback;
130     char *setname;
131     ODR odr;
132     ZOOM_record_cache record_hash[RECORD_HASH_SIZE];
133     ZOOM_options options;
134     ZOOM_connection connection;
135     char **databaseNames;
136     int num_databaseNames;
137     YAZ_MUTEX mutex;
138 #if SHPTR
139     struct WRBUF_shptr *record_wrbuf;
140 #endif
141     ZOOM_resultset next;
142     ZOOM_facet_field *facets;
143     int num_facets;
144     char **facets_names;
145     WRBUF mc_key;
146     int live_set; /* 0=no hit count, 1=cached hit, 2=hits + real set */
147 };
148
149 struct facet_term_p {
150     char *term;
151     int frequency;
152 };
153
154 struct ZOOM_facet_field_p {
155     char *facet_name;
156     int num_terms;
157     struct facet_term_p *facet_terms;
158 };
159
160 struct ZOOM_scanset_p {
161     int refcount;
162     ODR odr;
163     ZOOM_options options;
164     ZOOM_connection connection;
165     ZOOM_query query;
166     Z_ScanResponse *scan_response;
167     Z_SRW_scanResponse *srw_scan_response;
168
169     char **databaseNames;
170     int num_databaseNames;
171 };
172
173 struct ZOOM_package_p {
174     int refcount;
175     ODR odr_out;
176     ZOOM_options options;
177     ZOOM_connection connection;
178     char *buf_out;
179     int len_out;
180 };
181
182 struct ZOOM_task_p {
183     int running;
184     int which;
185     union {
186 #define ZOOM_TASK_SEARCH 1
187         struct {
188             int count;
189             int start;
190             ZOOM_resultset resultset;
191             char *syntax;
192             char *elementSetName;
193             char *schema;
194         } search;
195 #define ZOOM_TASK_CONNECT 3
196 #define ZOOM_TASK_SCAN 4
197         struct {
198             ZOOM_scanset scan;
199         } scan;
200 #define ZOOM_TASK_PACKAGE 5
201         ZOOM_package package;
202 #define ZOOM_TASK_SORT 6
203         struct {
204             ZOOM_resultset resultset;
205             ZOOM_query q;
206         } sort;
207     } u;
208     ZOOM_task next;
209 };
210
211 typedef enum {
212     zoom_pending,
213     zoom_complete
214 } zoom_ret;
215
216 void ZOOM_options_addref (ZOOM_options opt);
217
218 void ZOOM_handle_Z3950_apdu(ZOOM_connection c, Z_APDU *apdu);
219
220 void ZOOM_set_dset_error(ZOOM_connection c, int error,
221                          const char *dset,
222                          const char *addinfo, const char *addinfo2);
223
224 void ZOOM_set_error(ZOOM_connection c, int error, const char *addinfo);
225
226 ZOOM_Event ZOOM_Event_create(int kind);
227 void ZOOM_connection_put_event(ZOOM_connection c, ZOOM_Event event);
228
229 zoom_ret ZOOM_connection_Z3950_search(ZOOM_connection c);
230 zoom_ret ZOOM_connection_Z3950_send_scan(ZOOM_connection c);
231 zoom_ret ZOOM_send_buf(ZOOM_connection c);
232 zoom_ret send_Z3950_sort(ZOOM_connection c, ZOOM_resultset resultset);
233 char **ZOOM_connection_get_databases(ZOOM_connection con, ZOOM_options options,
234                                      int *num, ODR odr);
235 zoom_ret ZOOM_connection_Z3950_send_init(ZOOM_connection c);
236
237 ZOOM_task ZOOM_connection_add_task(ZOOM_connection c, int which);
238 void ZOOM_connection_remove_task(ZOOM_connection c);
239 int ZOOM_test_reconnect(ZOOM_connection c);
240
241 ZOOM_record ZOOM_record_cache_lookup(ZOOM_resultset r, int pos,
242                                      const char *syntax,
243                                      const char *elementSetName,
244                                      const char *schema);
245 void ZOOM_record_cache_add(ZOOM_resultset r, Z_NamePlusRecord *npr,
246                            int pos,
247                            const char *syntax, const char *elementSetName,
248                            const char *schema,
249                            Z_SRW_diagnostic *diag);
250
251 Z_Query *ZOOM_query_get_Z_Query(ZOOM_query s);
252 Z_SortKeySpecList *ZOOM_query_get_sortspec(ZOOM_query s);
253 const char *ZOOM_query_get_query_string(ZOOM_query s);
254 const char *ZOOM_query_get_sru11(ZOOM_query s);
255
256 int ZOOM_uri_to_code(const char *uri);
257
258 zoom_ret ZOOM_connection_srw_send_search(ZOOM_connection c);
259 zoom_ret ZOOM_connection_srw_send_scan(ZOOM_connection c);
260
261 int ZOOM_handle_sru(ZOOM_connection c, Z_HTTP_Response *hres,
262                     zoom_ret *cret, char **addinfo);
263
264 void ZOOM_set_HTTP_error(ZOOM_connection c, int error,
265                          const char *addinfo, const char *addinfo2);
266
267 ZOOM_Event ZOOM_connection_get_event(ZOOM_connection c);
268 void ZOOM_connection_remove_events(ZOOM_connection c);
269 void ZOOM_Event_destroy(ZOOM_Event event);
270 zoom_ret ZOOM_send_GDU(ZOOM_connection c, Z_GDU *gdu);
271 void ZOOM_handle_facet_list(ZOOM_resultset r, Z_FacetList *fl);
272
273 /*
274  * Local variables:
275  * c-basic-offset: 4
276  * c-file-style: "Stroustrup"
277  * indent-tabs-mode: nil
278  * End:
279  * vim: shiftwidth=4 tabstop=8 expandtab
280  */
281