1 /* $Id: filter_multi.cpp,v 1.10 2006-01-18 14:36:15 adam Exp $
2 Copyright (c) 2005, Index Data.
10 #include "package.hpp"
12 #include <boost/thread/thread.hpp>
13 #include <boost/thread/mutex.hpp>
14 #include <boost/thread/condition.hpp>
15 #include <boost/shared_ptr.hpp>
18 #include "filter_multi.hpp"
21 #include <yaz/otherinfo.h>
22 #include <yaz/diagbib1.h>
29 namespace yf = yp2::filter;
34 struct Multi::BackendSet {
37 bool operator < (const BackendSet &k) const;
38 bool operator == (const BackendSet &k) const;
40 struct Multi::ScanTermInfo {
41 std::string m_norm_term;
42 std::string m_display_term;
44 bool operator < (const ScanTermInfo &) const;
45 bool operator == (const ScanTermInfo &) const;
46 Z_Entry *get_entry(ODR odr);
48 struct Multi::FrontendSet {
54 FrontendSet(std::string setname);
58 void round_robin(int pos, int number, std::list<PresentJob> &job);
60 std::list<BackendSet> m_backend_sets;
61 std::string m_setname;
63 struct Multi::Backend {
65 std::string m_backend_database;
68 void operator() (void); // thread operation
70 struct Multi::Frontend {
75 std::list<BackendPtr> m_backend_list;
76 std::map<std::string,Multi::FrontendSet> m_sets;
78 void multi_move(std::list<BackendPtr> &blist);
79 void init(Package &package, Z_GDU *gdu);
80 void close(Package &package);
81 void search(Package &package, Z_APDU *apdu);
82 void present(Package &package, Z_APDU *apdu);
83 void scan1(Package &package, Z_APDU *apdu);
84 void scan2(Package &package, Z_APDU *apdu);
88 Map(std::list<std::string> hosts, std::string route);
90 std::list<std::string> m_hosts;
95 friend class Frontend;
97 FrontendPtr get_frontend(Package &package);
98 void release_frontend(Package &package);
100 boost::mutex m_sessions_mutex;
101 std::map<std::string, Multi::Map>m_maps;
102 std::map<std::string,std::string> m_target_route;
103 boost::mutex m_mutex;
104 boost::condition m_cond_session_ready;
105 std::map<yp2::Session, FrontendPtr> m_clients;
112 bool yf::Multi::BackendSet::operator < (const BackendSet &k) const
114 return m_count < k.m_count;
117 yf::Multi::Frontend::Frontend(Rep *rep)
123 yf::Multi::Frontend::~Frontend()
127 yf::Multi::FrontendPtr yf::Multi::Rep::get_frontend(Package &package)
129 boost::mutex::scoped_lock lock(m_mutex);
131 std::map<yp2::Session,yf::Multi::FrontendPtr>::iterator it;
135 it = m_clients.find(package.session());
136 if (it == m_clients.end())
139 if (!it->second->m_in_use)
141 it->second->m_in_use = true;
144 m_cond_session_ready.wait(lock);
146 FrontendPtr f(new Frontend(this));
147 m_clients[package.session()] = f;
152 void yf::Multi::Rep::release_frontend(Package &package)
154 boost::mutex::scoped_lock lock(m_mutex);
155 std::map<yp2::Session,yf::Multi::FrontendPtr>::iterator it;
157 it = m_clients.find(package.session());
158 if (it != m_clients.end())
160 if (package.session().is_closed())
162 it->second->close(package);
167 it->second->m_in_use = false;
169 m_cond_session_ready.notify_all();
173 yf::Multi::FrontendSet::FrontendSet(std::string setname)
179 yf::Multi::FrontendSet::FrontendSet()
184 yf::Multi::FrontendSet::~FrontendSet()
188 yf::Multi::Map::Map(std::list<std::string> hosts, std::string route)
189 : m_hosts(hosts), m_route(route)
193 yf::Multi::Map::Map()
197 yf::Multi::Multi() : m_p(new Multi::Rep)
201 yf::Multi::~Multi() {
205 void yf::Multi::add_map_host2hosts(std::string host,
206 std::list<std::string> hosts,
209 m_p->m_maps[host] = Multi::Map(hosts, route);
212 void yf::Multi::Backend::operator() (void)
214 m_package->move(m_route);
218 void yf::Multi::Frontend::close(Package &package)
220 std::list<BackendPtr>::const_iterator bit;
221 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
225 b->m_package->copy_filter(package);
226 b->m_package->request() = (Z_GDU *) 0;
227 b->m_package->session().close();
228 b->m_package->move(b->m_route);
232 void yf::Multi::Frontend::multi_move(std::list<BackendPtr> &blist)
234 std::list<BackendPtr>::const_iterator bit;
235 boost::thread_group g;
236 for (bit = blist.begin(); bit != blist.end(); bit++)
238 g.add_thread(new boost::thread(**bit));
244 void yf::Multi::FrontendSet::round_robin(int start, int number,
245 std::list<PresentJob> &jobs)
252 std::list<int> inside_pos;
253 std::list<BackendSet>::const_iterator bsit;
254 for (bsit = m_backend_sets.begin(); bsit != m_backend_sets.end(); bsit++)
257 inside_pos.push_back(0);
260 std::list<int>::iterator psit = pos.begin();
261 std::list<int>::iterator esit = inside_pos.begin();
262 bsit = m_backend_sets.begin();
263 while (fetched < number)
265 if (bsit == m_backend_sets.end())
268 esit = inside_pos.begin();
269 bsit = m_backend_sets.begin();
274 if (*psit <= bsit->m_count)
279 job.m_backend = bsit->m_backend;
281 job.m_inside_pos = *esit;
296 void yf::Multi::Frontend::init(Package &package, Z_GDU *gdu)
298 Z_InitRequest *req = gdu->u.z3950->u.initRequest;
300 std::list<std::string> targets;
302 yp2::util::get_vhost_otherinfo(&req->otherInfo, false, targets);
304 if (targets.size() < 1)
310 std::list<std::string>::const_iterator t_it = targets.begin();
311 for (; t_it != targets.end(); t_it++)
314 Backend *b = new Backend;
317 b->m_route = m_p->m_target_route[*t_it];
319 b->m_package = PackagePtr(new Package(s, package.origin()));
321 m_backend_list.push_back(BackendPtr(b));
325 // create init request
326 std::list<BackendPtr>::const_iterator bit;
327 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
331 Z_APDU *init_apdu = zget_APDU(odr, Z_APDU_initRequest);
333 std::list<std::string>vhost_one;
334 vhost_one.push_back(b->m_vhost);
335 yp2::util::set_vhost_otherinfo(&init_apdu->u.initRequest->otherInfo,
338 Z_InitRequest *req = init_apdu->u.initRequest;
340 ODR_MASK_SET(req->options, Z_Options_search);
341 ODR_MASK_SET(req->options, Z_Options_present);
342 ODR_MASK_SET(req->options, Z_Options_namedResultSets);
343 ODR_MASK_SET(req->options, Z_Options_scan);
345 ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_1);
346 ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_2);
347 ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_3);
349 b->m_package->request() = init_apdu;
351 b->m_package->copy_filter(package);
353 multi_move(m_backend_list);
355 // create the frontend init response based on each backend init response
358 Z_APDU *f_apdu = odr.create_initResponse(gdu->u.z3950, 0, 0);
359 Z_InitResponse *f_resp = f_apdu->u.initResponse;
361 ODR_MASK_SET(f_resp->options, Z_Options_search);
362 ODR_MASK_SET(f_resp->options, Z_Options_present);
363 ODR_MASK_SET(f_resp->options, Z_Options_namedResultSets);
365 ODR_MASK_SET(f_resp->protocolVersion, Z_ProtocolVersion_1);
366 ODR_MASK_SET(f_resp->protocolVersion, Z_ProtocolVersion_2);
367 ODR_MASK_SET(f_resp->protocolVersion, Z_ProtocolVersion_3);
369 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
371 PackagePtr p = (*bit)->m_package;
373 if (p->session().is_closed()) // if any backend closes, close frontend
374 package.session().close();
375 Z_GDU *gdu = p->response().get();
376 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
380 Z_APDU *b_apdu = gdu->u.z3950;
381 Z_InitResponse *b_resp = b_apdu->u.initResponse;
383 // common options for all backends
384 for (i = 0; i <= Z_Options_stringSchema; i++)
386 if (!ODR_MASK_GET(b_resp->options, i))
387 ODR_MASK_CLEAR(f_resp->options, i);
389 // common protocol version
390 for (i = 0; i <= Z_ProtocolVersion_3; i++)
391 if (!ODR_MASK_GET(b_resp->protocolVersion, i))
392 ODR_MASK_CLEAR(f_resp->protocolVersion, i);
393 // reject if any of the backends reject
394 if (!*b_resp->result)
399 // if any target does not return init return that (close or
401 package.response() = p->response();
405 package.response() = f_apdu;
408 void yf::Multi::Frontend::search(Package &package, Z_APDU *apdu_req)
410 // create search request
411 Z_SearchRequest *req = apdu_req->u.searchRequest;
413 // save these for later
414 int smallSetUpperBound = *req->smallSetUpperBound;
415 int largeSetLowerBound = *req->largeSetLowerBound;
416 int mediumSetPresentNumber = *req->mediumSetPresentNumber;
418 // they are altered now - to disable piggyback
419 *req->smallSetUpperBound = 0;
420 *req->largeSetLowerBound = 1;
421 *req->mediumSetPresentNumber = 1;
423 int default_num_db = req->num_databaseNames;
424 char **default_db = req->databaseNames;
426 std::list<BackendPtr>::const_iterator bit;
427 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
429 PackagePtr p = (*bit)->m_package;
432 if (!yp2::util::set_databases_from_zurl(odr, (*bit)->m_vhost,
433 &req->num_databaseNames,
434 &req->databaseNames))
436 req->num_databaseNames = default_num_db;
437 req->databaseNames = default_db;
439 p->request() = apdu_req;
440 p->copy_filter(package);
442 multi_move(m_backend_list);
444 // look at each response
445 FrontendSet resultSet(std::string(req->resultSetName));
447 int result_set_size = 0;
448 Z_Records *z_records_diag = 0; // no diagnostics (yet)
449 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
451 PackagePtr p = (*bit)->m_package;
453 if (p->session().is_closed()) // if any backend closes, close frontend
454 package.session().close();
456 Z_GDU *gdu = p->response().get();
457 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
458 Z_APDU_searchResponse)
460 Z_APDU *b_apdu = gdu->u.z3950;
461 Z_SearchResponse *b_resp = b_apdu->u.searchResponse;
463 // see we get any errors (AKA diagnstics)
466 if (b_resp->records->which == Z_Records_NSD
467 || b_resp->records->which == Z_Records_multipleNSD)
468 z_records_diag = b_resp->records;
469 // we may set this multiple times (TOO BAD!)
471 BackendSet backendSet;
472 backendSet.m_backend = *bit;
473 backendSet.m_count = *b_resp->resultCount;
474 result_set_size += *b_resp->resultCount;
475 resultSet.m_backend_sets.push_back(backendSet);
479 // if any target does not return search response - return that
480 package.response() = p->response();
486 Z_APDU *f_apdu = odr.create_searchResponse(apdu_req, 0, 0);
487 Z_SearchResponse *f_resp = f_apdu->u.searchResponse;
489 *f_resp->resultCount = result_set_size;
493 f_resp->records = z_records_diag;
494 package.response() = f_apdu;
498 m_sets[resultSet.m_setname] = resultSet;
501 yp2::util::piggyback(smallSetUpperBound,
503 mediumSetPresentNumber,
506 Package pp(package.session(), package.origin());
509 pp.copy_filter(package);
510 Z_APDU *p_apdu = zget_APDU(odr, Z_APDU_presentRequest);
511 Z_PresentRequest *p_req = p_apdu->u.presentRequest;
512 p_req->preferredRecordSyntax = req->preferredRecordSyntax;
513 p_req->resultSetId = req->resultSetName;
514 *p_req->resultSetStartPoint = 1;
515 *p_req->numberOfRecordsRequested = number;
516 pp.request() = p_apdu;
519 if (pp.session().is_closed())
520 package.session().close();
522 Z_GDU *gdu = pp.response().get();
523 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
524 Z_APDU_presentResponse)
526 Z_PresentResponse *p_res = gdu->u.z3950->u.presentResponse;
527 f_resp->records = p_res->records;
528 *f_resp->numberOfRecordsReturned =
529 *p_res->numberOfRecordsReturned;
530 *f_resp->nextResultSetPosition =
531 *p_res->nextResultSetPosition;
535 package.response() = pp.response();
539 package.response() = f_apdu; // in this scope because of p
542 void yf::Multi::Frontend::present(Package &package, Z_APDU *apdu_req)
544 // create present request
545 Z_PresentRequest *req = apdu_req->u.presentRequest;
548 it = m_sets.find(std::string(req->resultSetId));
549 if (it == m_sets.end())
553 odr.create_presentResponse(
555 YAZ_BIB1_SPECIFIED_RESULT_SET_DOES_NOT_EXIST,
557 package.response() = apdu;
560 std::list<Multi::FrontendSet::PresentJob> jobs;
561 int start = *req->resultSetStartPoint;
562 int number = *req->numberOfRecordsRequested;
563 it->second.round_robin(start, number, jobs);
565 std::list<BackendPtr> present_backend_list;
567 std::list<BackendSet>::const_iterator bsit;
568 bsit = it->second.m_backend_sets.begin();
569 for (; bsit != it->second.m_backend_sets.end(); bsit++)
571 std::list<Multi::FrontendSet::PresentJob>::const_iterator jit;
575 for (jit = jobs.begin(); jit != jobs.end(); jit++)
577 if (jit->m_backend == bsit->m_backend)
579 if (start == -1 || jit->m_pos < start)
581 if (end == -1 || jit->m_pos > end)
587 PackagePtr p = bsit->m_backend->m_package;
589 *req->resultSetStartPoint = start;
590 *req->numberOfRecordsRequested = end - start + 1;
592 p->request() = apdu_req;
593 p->copy_filter(package);
595 present_backend_list.push_back(bsit->m_backend);
598 multi_move(present_backend_list);
600 // look at each response
601 Z_Records *z_records_diag = 0;
603 std::list<BackendPtr>::const_iterator pbit = present_backend_list.begin();
604 for (; pbit != present_backend_list.end(); pbit++)
606 PackagePtr p = (*pbit)->m_package;
608 if (p->session().is_closed()) // if any backend closes, close frontend
609 package.session().close();
611 Z_GDU *gdu = p->response().get();
612 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
613 Z_APDU_presentResponse)
615 Z_APDU *b_apdu = gdu->u.z3950;
616 Z_PresentResponse *b_resp = b_apdu->u.presentResponse;
618 // see we get any errors (AKA diagnstics)
621 if (b_resp->records->which != Z_Records_DBOSD)
622 z_records_diag = b_resp->records;
623 // we may set this multiple times (TOO BAD!)
628 // if any target does not return present response - return that
629 package.response() = p->response();
635 Z_APDU *f_apdu = odr.create_presentResponse(apdu_req, 0, 0);
636 Z_PresentResponse *f_resp = f_apdu->u.presentResponse;
640 f_resp->records = z_records_diag;
641 *f_resp->presentStatus = Z_PresentStatus_failure;
645 f_resp->records = (Z_Records *) odr_malloc(odr, sizeof(Z_Records));
646 Z_Records * records = f_resp->records;
647 records->which = Z_Records_DBOSD;
648 records->u.databaseOrSurDiagnostics =
649 (Z_NamePlusRecordList *)
650 odr_malloc(odr, sizeof(Z_NamePlusRecordList));
651 Z_NamePlusRecordList *nprl = records->u.databaseOrSurDiagnostics;
652 nprl->num_records = jobs.size();
653 nprl->records = (Z_NamePlusRecord**)
654 odr_malloc(odr, sizeof(Z_NamePlusRecord *) * nprl->num_records);
656 std::list<Multi::FrontendSet::PresentJob>::const_iterator jit;
657 for (jit = jobs.begin(); jit != jobs.end(); jit++)
659 PackagePtr p = jit->m_backend->m_package;
661 Z_GDU *gdu = p->response().get();
662 Z_APDU *b_apdu = gdu->u.z3950;
663 Z_PresentResponse *b_resp = b_apdu->u.presentResponse;
666 b_resp->records->u.databaseOrSurDiagnostics->
667 records[jit->m_inside_pos];
669 *f_resp->nextResultSetPosition = start + i;
670 *f_resp->numberOfRecordsReturned = i;
672 package.response() = f_apdu;
675 void yf::Multi::Frontend::scan1(Package &package, Z_APDU *apdu_req)
677 if (m_backend_list.size() > 1)
681 odr.create_scanResponse(
682 apdu_req, YAZ_BIB1_COMBI_OF_SPECIFIED_DATABASES_UNSUPP, 0);
683 package.response() = f_apdu;
686 Z_ScanRequest *req = apdu_req->u.scanRequest;
688 int default_num_db = req->num_databaseNames;
689 char **default_db = req->databaseNames;
691 std::list<BackendPtr>::const_iterator bit;
692 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
694 PackagePtr p = (*bit)->m_package;
697 if (!yp2::util::set_databases_from_zurl(odr, (*bit)->m_vhost,
698 &req->num_databaseNames,
699 &req->databaseNames))
701 req->num_databaseNames = default_num_db;
702 req->databaseNames = default_db;
704 p->request() = apdu_req;
705 p->copy_filter(package);
707 multi_move(m_backend_list);
709 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
711 PackagePtr p = (*bit)->m_package;
713 if (p->session().is_closed()) // if any backend closes, close frontend
714 package.session().close();
716 Z_GDU *gdu = p->response().get();
717 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
720 package.response() = p->response();
725 // if any target does not return scan response - return that
726 package.response() = p->response();
732 bool yf::Multi::ScanTermInfo::operator < (const ScanTermInfo &k) const
734 return m_norm_term < k.m_norm_term;
737 bool yf::Multi::ScanTermInfo::operator == (const ScanTermInfo &k) const
739 return m_norm_term == k.m_norm_term;
742 Z_Entry *yf::Multi::ScanTermInfo::get_entry(ODR odr)
744 Z_Entry *e = (Z_Entry *)odr_malloc(odr, sizeof(*e));
745 e->which = Z_Entry_termInfo;
747 t = e->u.termInfo = (Z_TermInfo *) odr_malloc(odr, sizeof(*t));
748 t->suggestedAttributes = 0;
750 t->alternativeTerm = 0;
752 t->otherTermInfo = 0;
753 t->globalOccurrences = odr_intdup(odr, m_count);
755 odr_malloc(odr, sizeof(*t->term));
756 t->term->which = Z_Term_general;
758 t->term->u.general = o = (Odr_oct *)odr_malloc(odr, sizeof(Odr_oct));
760 o->len = o->size = m_norm_term.size();
761 o->buf = (unsigned char *) odr_malloc(odr, o->len);
762 memcpy(o->buf, m_norm_term.c_str(), o->len);
766 void yf::Multi::Frontend::scan2(Package &package, Z_APDU *apdu_req)
768 Z_ScanRequest *req = apdu_req->u.scanRequest;
771 int default_num_db = req->num_databaseNames;
772 char **default_db = req->databaseNames;
774 std::list<BackendPtr>::const_iterator bit;
775 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
777 PackagePtr p = (*bit)->m_package;
780 if (!yp2::util::set_databases_from_zurl(odr, (*bit)->m_vhost,
781 &req->num_databaseNames,
782 &req->databaseNames))
784 req->num_databaseNames = default_num_db;
785 req->databaseNames = default_db;
787 p->request() = apdu_req;
788 p->copy_filter(package);
791 multi_move(m_backend_list);
793 ScanTermInfoList entries_before;
794 ScanTermInfoList entries_after;
798 for (bit = m_backend_list.begin(); bit != m_backend_list.end(); bit++)
800 PackagePtr p = (*bit)->m_package;
802 if (p->session().is_closed()) // if any backend closes, close frontend
803 package.session().close();
805 Z_GDU *gdu = p->response().get();
806 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
809 Z_ScanResponse *res = gdu->u.z3950->u.scanResponse;
811 if (res->entries && res->entries->nonsurrogateDiagnostics)
815 Z_APDU *f_apdu = odr.create_scanResponse(apdu_req, 1, 0);
816 Z_ScanResponse *f_res = f_apdu->u.scanResponse;
818 f_res->entries->nonsurrogateDiagnostics =
819 res->entries->nonsurrogateDiagnostics;
820 f_res->entries->num_nonsurrogateDiagnostics =
821 res->entries->num_nonsurrogateDiagnostics;
823 package.response() = f_apdu;
827 if (res->entries && res->entries->entries)
829 Z_Entry **entries = res->entries->entries;
830 int num_entries = res->entries->num_entries;
832 if (req->preferredPositionInResponse)
833 position = *req->preferredPositionInResponse;
834 if (res->positionOfTerm)
835 position = *res->positionOfTerm;
839 for (i = 0; i<position-1 && i<num_entries; i++)
841 Z_Entry *ent = entries[i];
843 if (ent->which == Z_Entry_termInfo)
847 int *occur = ent->u.termInfo->globalOccurrences;
848 my.m_count = occur ? *occur : 0;
850 if (ent->u.termInfo->term->which == Z_Term_general)
852 my.m_norm_term = std::string(
854 ent->u.termInfo->term->u.general->buf,
855 ent->u.termInfo->term->u.general->len);
857 if (my.m_norm_term.length())
859 ScanTermInfoList::iterator it =
860 entries_before.begin();
861 while (it != entries_before.end() && my <*it)
865 it->m_count += my.m_count;
869 entries_before.insert(it, my);
876 for (i = position-1; i<num_entries; i++)
878 Z_Entry *ent = entries[i];
880 if (ent->which == Z_Entry_termInfo)
884 int *occur = ent->u.termInfo->globalOccurrences;
885 my.m_count = occur ? *occur : 0;
887 if (ent->u.termInfo->term->which == Z_Term_general)
889 my.m_norm_term = std::string(
891 ent->u.termInfo->term->u.general->buf,
892 ent->u.termInfo->term->u.general->len);
894 if (my.m_norm_term.length())
896 ScanTermInfoList::iterator it =
897 entries_after.begin();
898 while (it != entries_after.end() && *it < my)
902 it->m_count += my.m_count;
906 entries_after.insert(it, my);
917 // if any target does not return scan response - return that
918 package.response() = p->response();
926 std::cout << "BEFORE\n";
927 ScanTermInfoList::iterator it = entries_before.begin();
928 for(; it != entries_before.end(); it++)
930 std::cout << " " << it->m_norm_term << " " << it->m_count << "\n";
933 std::cout << "AFTER\n";
934 it = entries_after.begin();
935 for(; it != entries_after.end(); it++)
937 std::cout << " " << it->m_norm_term << " " << it->m_count << "\n";
944 Z_APDU *f_apdu = odr.create_scanResponse(apdu_req, 1, "not implemented");
945 package.response() = f_apdu;
950 Z_APDU *f_apdu = odr.create_scanResponse(apdu_req, 0, 0);
951 Z_ScanResponse *resp = f_apdu->u.scanResponse;
953 int number_returned = *req->numberOfTermsRequested;
954 int position_returned = *req->preferredPositionInResponse;
956 resp->positionOfTerm = odr_intdup(odr, position_returned);
957 resp->numberOfEntriesReturned = odr_intdup(odr, number_returned);
959 resp->entries->num_entries = number_returned;
960 resp->entries->entries = (Z_Entry**)
961 odr_malloc(odr, sizeof(Z_Entry*) * number_returned);
964 ScanTermInfoList::iterator it = entries_before.begin();
965 for (i = 0; i<position_returned-1; i++, it++)
967 resp->entries->entries[i] = it->get_entry(odr);
969 it = entries_after.begin();
970 for (i = position_returned-1; i<number_returned; i++, it++)
972 resp->entries->entries[i] = it->get_entry(odr);
974 package.response() = f_apdu;
979 void yf::Multi::process(Package &package) const
981 FrontendPtr f = m_p->get_frontend(package);
983 Z_GDU *gdu = package.request().get();
985 if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
986 Z_APDU_initRequest && !f->m_is_multi)
988 f->init(package, gdu);
990 else if (!f->m_is_multi)
992 else if (gdu && gdu->which == Z_GDU_Z3950)
994 Z_APDU *apdu = gdu->u.z3950;
995 if (apdu->which == Z_APDU_initRequest)
999 package.response() = odr.create_close(
1001 Z_Close_protocolError,
1004 package.session().close();
1006 else if (apdu->which == Z_APDU_searchRequest)
1008 f->search(package, apdu);
1010 else if (apdu->which == Z_APDU_presentRequest)
1012 f->present(package, apdu);
1014 else if (apdu->which == Z_APDU_scanRequest)
1016 f->scan2(package, apdu);
1022 package.response() = odr.create_close(
1023 apdu, Z_Close_protocolError,
1024 "unsupported APDU in filter multi");
1026 package.session().close();
1029 m_p->release_frontend(package);
1032 void yp2::filter::Multi::configure(const xmlNode * ptr)
1034 for (ptr = ptr->children; ptr; ptr = ptr->next)
1036 if (ptr->type != XML_ELEMENT_NODE)
1038 if (!strcmp((const char *) ptr->name, "target"))
1040 std::string route = yp2::xml::get_route(ptr);
1041 std::string target = yp2::xml::get_text(ptr);
1042 std::cout << "route=" << route << " target=" << target << "\n";
1043 m_p->m_target_route[target] = route;
1045 else if (!strcmp((const char *) ptr->name, "virtual"))
1047 std::list<std::string> targets;
1049 xmlNode *v_node = ptr->children;
1050 for (; v_node; v_node = v_node->next)
1052 if (v_node->type != XML_ELEMENT_NODE)
1055 if (yp2::xml::is_element_yp2(v_node, "vhost"))
1056 vhost = yp2::xml::get_text(v_node);
1057 else if (yp2::xml::is_element_yp2(v_node, "target"))
1058 targets.push_back(yp2::xml::get_text(v_node));
1060 throw yp2::filter::FilterException
1062 + std::string((const char *) v_node->name)
1063 + " in virtual section"
1066 std::string route = yp2::xml::get_route(ptr);
1067 add_map_host2hosts(vhost, targets, route);
1068 std::list<std::string>::const_iterator it;
1069 for (it = targets.begin(); it != targets.end(); it++)
1071 std::cout << "Add " << vhost << "->" << *it
1072 << "," << route << "\n";
1077 throw yp2::filter::FilterException
1079 + std::string((const char *) ptr->name)
1080 + " in virt_db filter");
1085 static yp2::filter::Base* filter_creator()
1087 return new yp2::filter::Multi;
1091 struct yp2_filter_struct yp2_filter_multi = {
1102 * indent-tabs-mode: nil
1103 * c-file-style: "stroustrup"
1105 * vim: shiftwidth=4 tabstop=8 expandtab