1 /* This file is part of Metaproxy.
2 Copyright (C) 2005-2011 Index Data
4 Metaproxy is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
9 Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20 #include "filter_zoom.hpp"
22 #include <metaproxy/package.hpp>
23 #include <metaproxy/util.hpp>
25 #include <boost/thread/mutex.hpp>
26 #include <boost/thread/condition.hpp>
27 #include <yaz/diagbib1.h>
31 namespace mp = metaproxy_1;
32 namespace yf = mp::filter;
34 namespace metaproxy_1 {
38 friend class Frontend;
40 ZOOM_connection m_connection;
41 ZOOM_resultset m_resultset;
42 std::string m_frontend_database;
46 void connect(std::string zurl);
48 class Zoom::Frontend {
53 yazpp_1::GDU m_init_gdu;
54 std::list<BackendPtr> m_backend_list;
55 void handle_package(mp::Package &package);
56 void handle_search(mp::Package &package);
57 void handle_present(mp::Package &package);
58 BackendPtr get_backend_from_databases(std::string database);
67 void process(metaproxy_1::Package & package);
68 void configure(const xmlNode * ptr);
70 FrontendPtr get_frontend(mp::Package &package);
71 void release_frontend(mp::Package &package);
73 std::map<mp::Session, FrontendPtr> m_clients;
75 boost::condition m_cond_session_ready;
80 // define Pimpl wrapper forwarding to Impl
82 yf::Zoom::Zoom() : m_p(new Impl)
87 { // must have a destructor because of boost::scoped_ptr
90 void yf::Zoom::configure(const xmlNode *xmlnode, bool test_only)
92 m_p->configure(xmlnode);
95 void yf::Zoom::process(mp::Package &package) const
97 m_p->process(package);
101 // define Implementation stuff
103 yf::Zoom::Backend::Backend()
105 m_connection = ZOOM_connection_create(0);
109 yf::Zoom::Backend::~Backend()
111 ZOOM_connection_destroy(m_connection);
112 ZOOM_resultset_destroy(m_resultset);
115 void yf::Zoom::Backend::connect(std::string zurl)
117 ZOOM_connection_connect(m_connection, zurl.c_str(), 0);
120 yf::Zoom::Frontend::Frontend(Impl *impl) :
121 m_p(impl), m_is_virtual(false), m_in_use(true)
125 yf::Zoom::Frontend::~Frontend()
129 yf::Zoom::FrontendPtr yf::Zoom::Impl::get_frontend(mp::Package &package)
131 boost::mutex::scoped_lock lock(m_mutex);
133 std::map<mp::Session,yf::Zoom::FrontendPtr>::iterator it;
137 it = m_clients.find(package.session());
138 if (it == m_clients.end())
141 if (!it->second->m_in_use)
143 it->second->m_in_use = true;
146 m_cond_session_ready.wait(lock);
148 FrontendPtr f(new Frontend(this));
149 m_clients[package.session()] = f;
154 void yf::Zoom::Impl::release_frontend(mp::Package &package)
156 boost::mutex::scoped_lock lock(m_mutex);
157 std::map<mp::Session,yf::Zoom::FrontendPtr>::iterator it;
159 it = m_clients.find(package.session());
160 if (it != m_clients.end())
162 if (package.session().is_closed())
168 it->second->m_in_use = false;
170 m_cond_session_ready.notify_all();
174 yf::Zoom::Impl::Impl()
178 yf::Zoom::Impl::~Impl()
182 void yf::Zoom::Impl::configure(const xmlNode *xmlnode)
186 yf::Zoom::BackendPtr yf::Zoom::Frontend::get_backend_from_databases(
187 std::string database)
189 std::list<BackendPtr>::const_iterator map_it;
190 map_it = m_backend_list.begin();
191 for (; map_it != m_backend_list.end(); map_it++)
192 if ((*map_it)->m_frontend_database == database)
195 BackendPtr b(new Backend);
197 std::string url = "localhost:9999/" + database;
198 yaz_log(YLOG_LOG, "new backend url=%s", url.c_str());
203 void yf::Zoom::Frontend::handle_search(mp::Package &package)
205 Z_GDU *gdu = package.request().get();
206 Z_APDU *apdu_req = gdu->u.z3950;
207 Z_APDU *apdu_res = 0;
209 Z_SearchRequest *sr = apdu_req->u.searchRequest;
210 if (sr->num_databaseNames != 1)
212 apdu_res = odr.create_searchResponse(
213 apdu_req, YAZ_BIB1_TOO_MANY_DATABASES_SPECIFIED, 0);
214 package.response() = apdu_res;
217 BackendPtr b = get_backend_from_databases(sr->databaseNames[0]);
218 switch (sr->query->which)
221 case Z_Query_type_101:
223 odr.create_searchResponse(
225 YAZ_BIB1_TEMPORARY_SYSTEM_ERROR,
226 "search filter do not handle type-1/type-101 yet");
227 package.response() = apdu_res;
231 odr.create_searchResponse(
233 YAZ_BIB1_QUERY_TYPE_UNSUPP, 0);
234 package.response() = apdu_res;
239 void yf::Zoom::Frontend::handle_present(mp::Package &package)
241 Z_GDU *gdu = package.request().get();
242 Z_APDU *apdu_req = gdu->u.z3950;
244 package.response() = odr.create_close(
246 Z_Close_protocolError,
247 "zoom filter has not implemented present request yet");
248 package.session().close();
251 void yf::Zoom::Frontend::handle_package(mp::Package &package)
253 Z_GDU *gdu = package.request().get();
256 else if (gdu->which == Z_GDU_Z3950)
258 Z_APDU *apdu_req = gdu->u.z3950;
259 if (apdu_req->which == Z_APDU_initRequest)
262 package.response() = odr.create_close(
264 Z_Close_protocolError,
267 else if (apdu_req->which == Z_APDU_searchRequest)
269 handle_search(package);
271 else if (apdu_req->which == Z_APDU_presentRequest)
273 handle_present(package);
278 package.response() = odr.create_close(
280 Z_Close_protocolError,
281 "zoom filter cannot handle this APDU");
282 package.session().close();
287 package.session().close();
291 void yf::Zoom::Impl::process(mp::Package &package)
293 FrontendPtr f = get_frontend(package);
294 Z_GDU *gdu = package.request().get();
298 f->handle_package(package);
300 else if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
303 Z_InitRequest *req = gdu->u.z3950->u.initRequest;
307 Z_APDU *apdu = odr.create_initResponse(gdu->u.z3950, 0, 0);
308 Z_InitResponse *resp = apdu->u.initResponse;
311 static const int masks[] = {
316 for (i = 0; masks[i] != -1; i++)
317 if (ODR_MASK_GET(req->options, masks[i]))
318 ODR_MASK_SET(resp->options, masks[i]);
320 static const int versions[] = {
326 for (i = 0; versions[i] != -1; i++)
327 if (ODR_MASK_GET(req->protocolVersion, versions[i]))
328 ODR_MASK_SET(resp->protocolVersion, versions[i]);
332 *resp->preferredMessageSize = *req->preferredMessageSize;
333 *resp->maximumRecordSize = *req->maximumRecordSize;
335 package.response() = apdu;
336 f->m_is_virtual = true;
341 release_frontend(package);
345 static mp::filter::Base* filter_creator()
347 return new mp::filter::Zoom;
351 struct metaproxy_1_filter_struct metaproxy_1_filter_zoom = {
362 * c-file-style: "Stroustrup"
363 * indent-tabs-mode: nil
365 * vim: shiftwidth=4 tabstop=8 expandtab