X-Git-Url: http://jsfdemo.indexdata.com/?a=blobdiff_plain;f=src%2Ffilter_zoom.cpp;h=586328528b873c19e565973a57a40cd54eab1f3d;hb=e871b8cd25b12cd9944201319d1b52c4472d8a5f;hp=a42d9f54f57b5dda46d07ed2bf86b7fda2627b55;hpb=0f92fb88a4e9fec525392fb665043c91b9268f24;p=metaproxy-moved-to-github.git diff --git a/src/filter_zoom.cpp b/src/filter_zoom.cpp index a42d9f5..5863285 100644 --- a/src/filter_zoom.cpp +++ b/src/filter_zoom.cpp @@ -19,6 +19,8 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #include "config.hpp" #include "filter_zoom.hpp" #include +#include +#include #include #include #include "torus.hpp" @@ -28,7 +30,11 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #include #include +#include #include +#include +#include +#include #include #include #include @@ -42,7 +48,11 @@ namespace yf = mp::filter; namespace metaproxy_1 { namespace filter { struct Zoom::Searchable : boost::noncopyable { - std::string database; + std::string authentication; + std::string cfAuth; + std::string cfProxy; + std::string cfSubDb; + std::string udb; std::string target; std::string query_encoding; std::string sru; @@ -53,7 +63,7 @@ namespace metaproxy_1 { bool use_turbomarc; bool piggyback; CCL_bibset ccl_bibset; - Searchable(); + Searchable(CCL_bibset base); ~Searchable(); }; class Zoom::Backend : boost::noncopyable { @@ -71,10 +81,13 @@ namespace metaproxy_1 { void connect(std::string zurl, int *error, const char **addinfo); void search_pqf(const char *pqf, Odr_int *hits, int *error, const char **addinfo); + void search_cql(const char *cql, Odr_int *hits, + int *error, const char **addinfo); void present(Odr_int start, Odr_int number, ZOOM_record *recs, int *error, const char **addinfo); void set_option(const char *name, const char *value); - int get_error(const char **addinfo); + const char *get_option(const char *name); + void get_zoom_error(int *error, const char **addinfo); }; class Zoom::Frontend : boost::noncopyable { friend class Impl; @@ -109,16 +122,19 @@ namespace metaproxy_1 { void process(metaproxy_1::Package & package); void configure(const xmlNode * ptr, bool test_only); private: + void configure_local_records(const xmlNode * ptr, bool test_only); FrontendPtr get_frontend(mp::Package &package); void release_frontend(mp::Package &package); - void parse_torus(const xmlNode *ptr); - - std::listm_searchables; - + SearchablePtr parse_torus_record(const xmlNode *ptr); + struct cql_node *convert_cql_fields(struct cql_node *cn, ODR odr); std::map m_clients; boost::mutex m_mutex; boost::condition m_cond_session_ready; - mp::Torus torus; + std::string torus_url; + std::map fieldmap; + std::string xsldir; + CCL_bibset bibset; + std::map s_map; }; } } @@ -161,18 +177,51 @@ yf::Zoom::Backend::~Backend() ZOOM_resultset_destroy(m_resultset); } + +void yf::Zoom::Backend::get_zoom_error(int *error, const char **addinfo) +{ + const char *msg = 0; + *error = ZOOM_connection_error(m_connection, &msg, addinfo); + if (*error) + { + if (*error >= ZOOM_ERROR_CONNECT) + { + // turn ZOOM diagnostic into a Bib-1 2: with addinfo=zoom err msg + *error = YAZ_BIB1_TEMPORARY_SYSTEM_ERROR; + if (addinfo) + *addinfo = msg; + } + } +} + void yf::Zoom::Backend::connect(std::string zurl, int *error, const char **addinfo) { ZOOM_connection_connect(m_connection, zurl.c_str(), 0); - *error = ZOOM_connection_error(m_connection, 0, addinfo); + get_zoom_error(error, addinfo); } void yf::Zoom::Backend::search_pqf(const char *pqf, Odr_int *hits, int *error, const char **addinfo) { m_resultset = ZOOM_connection_search_pqf(m_connection, pqf); - *error = ZOOM_connection_error(m_connection, 0, addinfo); + get_zoom_error(error, addinfo); + if (*error == 0) + *hits = ZOOM_resultset_size(m_resultset); + else + *hits = 0; +} + +void yf::Zoom::Backend::search_cql(const char *cql, Odr_int *hits, + int *error, const char **addinfo) +{ + ZOOM_query q = ZOOM_query_create(); + + ZOOM_query_cql(q, cql); + + m_resultset = ZOOM_connection_search(m_connection, q); + ZOOM_query_destroy(q); + get_zoom_error(error, addinfo); if (*error == 0) *hits = ZOOM_resultset_size(m_resultset); else @@ -184,7 +233,7 @@ void yf::Zoom::Backend::present(Odr_int start, Odr_int number, int *error, const char **addinfo) { ZOOM_resultset_records(m_resultset, recs, start, number); - *error = ZOOM_connection_error(m_connection, 0, addinfo); + get_zoom_error(error, addinfo); } void yf::Zoom::Backend::set_option(const char *name, const char *value) @@ -194,16 +243,16 @@ void yf::Zoom::Backend::set_option(const char *name, const char *value) ZOOM_resultset_option_set(m_resultset, name, value); } -int yf::Zoom::Backend::get_error(const char **addinfo) +const char *yf::Zoom::Backend::get_option(const char *name) { - return ZOOM_connection_error(m_connection, 0, addinfo); + return ZOOM_connection_option_get(m_connection, name); } -yf::Zoom::Searchable::Searchable() +yf::Zoom::Searchable::Searchable(CCL_bibset base) { piggyback = true; use_turbomarc = true; - ccl_bibset = ccl_qual_mk(); + ccl_bibset = ccl_qual_dup(base); } yf::Zoom::Searchable::~Searchable() @@ -267,101 +316,147 @@ void yf::Zoom::Impl::release_frontend(mp::Package &package) yf::Zoom::Impl::Impl() { + bibset = ccl_qual_mk(); } yf::Zoom::Impl::~Impl() { + ccl_qual_rm(&bibset); } -void yf::Zoom::Impl::parse_torus(const xmlNode *ptr1) +yf::Zoom::SearchablePtr yf::Zoom::Impl::parse_torus_record(const xmlNode *ptr) { - if (!ptr1) - return ; - for (ptr1 = ptr1->children; ptr1; ptr1 = ptr1->next) + Zoom::SearchablePtr s(new Searchable(bibset)); + + for (ptr = ptr->children; ptr; ptr = ptr->next) { - if (ptr1->type != XML_ELEMENT_NODE) + if (ptr->type != XML_ELEMENT_NODE) continue; - if (!strcmp((const char *) ptr1->name, "record")) + if (!strcmp((const char *) ptr->name, "layer")) + ptr = ptr->children; + else if (!strcmp((const char *) ptr->name, + "authentication")) + { + s->authentication = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "cfAuth")) + { + s->cfAuth = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "cfProxy")) + { + s->cfProxy = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "cfSubDb")) + { + s->cfSubDb = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, "udb")) + { + s->udb = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, "zurl")) + { + s->target = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, "sru")) + { + s->sru = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "queryEncoding")) + { + s->query_encoding = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "piggyback")) + { + s->piggyback = mp::xml::get_bool(ptr, true); + } + else if (!strcmp((const char *) ptr->name, + "requestSyntax")) + { + s->request_syntax = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "elementSet")) + { + s->element_set = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "recordEncoding")) + { + s->record_encoding = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "transform")) + { + s->transform_xsl_fname = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, + "useTurboMarc")) + { + ; // useTurboMarc is ignored + } + else if (!strncmp((const char *) ptr->name, + "cclmap_", 7)) + { + std::string value = mp::xml::get_text(ptr); + ccl_qual_fitem(s->ccl_bibset, value.c_str(), + (const char *) ptr->name + 7); + } + } + return s; +} + +void yf::Zoom::Impl::configure_local_records(const xmlNode *ptr, bool test_only) +{ + while (ptr && ptr->type != XML_ELEMENT_NODE) + ptr = ptr->next; + + if (ptr) + { + if (!strcmp((const char *) ptr->name, "records")) { - const xmlNode *ptr2 = ptr1; - for (ptr2 = ptr2->children; ptr2; ptr2 = ptr2->next) + for (ptr = ptr->children; ptr; ptr = ptr->next) { - if (ptr2->type != XML_ELEMENT_NODE) + if (ptr->type != XML_ELEMENT_NODE) continue; - if (!strcmp((const char *) ptr2->name, "layer")) + if (!strcmp((const char *) ptr->name, "record")) { - Zoom::SearchablePtr s(new Searchable); - - const xmlNode *ptr3 = ptr2; - for (ptr3 = ptr3->children; ptr3; ptr3 = ptr3->next) + SearchablePtr s = parse_torus_record(ptr); + if (s) { - if (ptr3->type != XML_ELEMENT_NODE) - continue; - if (!strcmp((const char *) ptr3->name, "id")) - { - s->database = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, "zurl")) - { - s->target = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, "sru")) - { - s->sru = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, - "queryEncoding")) - { - s->query_encoding = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, - "piggyback")) + std::string udb = s->udb; + if (udb.length()) + s_map[s->udb] = s; + else { - s->piggyback = mp::xml::get_bool(ptr3, true); - } - else if (!strcmp((const char *) ptr3->name, - "requestSyntax")) - { - s->request_syntax = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, - "elementSet")) - { - s->element_set = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, - "recordEncoding")) - { - s->record_encoding = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, - "transform")) - { - s->transform_xsl_fname = mp::xml::get_text(ptr3); - } - else if (!strcmp((const char *) ptr3->name, - "useTurboMarc")) - { - ; // useTurboMarc is ignored - } - else if (!strncmp((const char *) ptr3->name, - "cclmap_", 7)) - { - std::string value = mp::xml::get_text(ptr3); - ccl_qual_fitem(s->ccl_bibset, value.c_str(), - (const char *) ptr3->name + 7); + throw mp::filter::FilterException + ("No udb for local torus record"); } } - if (s->database.length() && s->target.length()) - { - yaz_log(YLOG_LOG, "add db=%s target=%s turbomarc=%s", - s->database.c_str(), s->target.c_str(), - s->use_turbomarc ? "1" : "0"); - m_searchables.push_back(s); - } + } + else + { + throw mp::filter::FilterException + ("Bad element " + + std::string((const char *) ptr->name) + + " in zoom filter inside element " + ""); } } } + else + { + throw mp::filter::FilterException + ("Bad element " + + std::string((const char *) ptr->name) + + " in zoom filter inside element "); + } } } @@ -371,30 +466,45 @@ void yf::Zoom::Impl::configure(const xmlNode *ptr, bool test_only) { if (ptr->type != XML_ELEMENT_NODE) continue; - if (!strcmp((const char *) ptr->name, "records")) - { - parse_torus(ptr); - } else if (!strcmp((const char *) ptr->name, "torus")) { - std::string url; const struct _xmlAttr *attr; for (attr = ptr->properties; attr; attr = attr->next) { if (!strcmp((const char *) attr->name, "url")) - url = mp::xml::get_text(attr->children); + torus_url = mp::xml::get_text(attr->children); + else if (!strcmp((const char *) attr->name, "xsldir")) + xsldir = mp::xml::get_text(attr->children); else throw mp::filter::FilterException( "Bad attribute " + std::string((const char *) attr->name)); } - torus.read_searchables(url); - xmlDoc *doc = torus.get_doc(); - if (doc) + configure_local_records(ptr->children, test_only); + } + else if (!strcmp((const char *) ptr->name, "cclmap")) + { + const char *addinfo = 0; + ccl_xml_config(bibset, ptr, &addinfo); + } + else if (!strcmp((const char *) ptr->name, "fieldmap")) + { + const struct _xmlAttr *attr; + std::string ccl_field; + std::string cql_field; + for (attr = ptr->properties; attr; attr = attr->next) { - xmlNode *ptr = xmlDocGetRootElement(doc); - parse_torus(ptr); + if (!strcmp((const char *) attr->name, "ccl")) + ccl_field = mp::xml::get_text(attr->children); + else if (!strcmp((const char *) attr->name, "cql")) + cql_field = mp::xml::get_text(attr->children); + else + throw mp::filter::FilterException( + "Bad attribute " + std::string((const char *) + attr->name)); } + if (cql_field.length()) + fieldmap[cql_field] = ccl_field; } else { @@ -413,29 +523,69 @@ yf::Zoom::BackendPtr yf::Zoom::Frontend::get_backend_from_databases( if (m_backend && m_backend->m_frontend_database == database) return m_backend; - std::list::iterator map_s = - m_p->m_searchables.begin(); - - std::string c_db = mp::util::database_name_normalize(database); + std::string db_args; + std::string cf_parm; + std::string torus_db; + size_t db_arg_pos = database.find(','); + if (db_arg_pos != std::string::npos) + { + torus_db = database.substr(0, db_arg_pos); + db_args = database.substr(db_arg_pos+1); + } + else + torus_db = database; + + SearchablePtr sptr; - while (map_s != m_p->m_searchables.end()) + std::map::iterator it; + it = m_p->s_map.find(torus_db); + if (it != m_p->s_map.end()) + sptr = it->second; + else { - if (c_db.compare((*map_s)->database) == 0) - break; - map_s++; + xmlDoc *doc = mp::get_searchable(m_p->torus_url, torus_db); + if (!doc) + { + *error = YAZ_BIB1_DATABASE_DOES_NOT_EXIST; + *addinfo = database.c_str(); + BackendPtr b; + return b; + } + const xmlNode *ptr = xmlDocGetRootElement(doc); + if (ptr) + { // presumably ptr is a records element node + // parse first record in document + for (ptr = ptr->children; ptr; ptr = ptr->next) + { + if (ptr->type == XML_ELEMENT_NODE + && !strcmp((const char *) ptr->name, "record")) + { + sptr = m_p->parse_torus_record(ptr); + break; + } + } + } + xmlFreeDoc(doc); } - if (map_s == m_p->m_searchables.end()) + + if (!sptr) { *error = YAZ_BIB1_DATABASE_DOES_NOT_EXIST; *addinfo = database.c_str(); BackendPtr b; return b; } - + xsltStylesheetPtr xsp = 0; - if ((*map_s)->transform_xsl_fname.length()) + if (sptr->transform_xsl_fname.length()) { - xmlDoc *xsp_doc = xmlParseFile((*map_s)->transform_xsl_fname.c_str()); + std::string fname; + + if (m_p->xsldir.length()) + fname = m_p->xsldir + "/" + sptr->transform_xsl_fname; + else + fname = sptr->transform_xsl_fname; + xmlDoc *xsp_doc = xmlParseFile(fname.c_str()); if (!xsp_doc) { *error = YAZ_BIB1_TEMPORARY_SYSTEM_ERROR; @@ -454,18 +604,50 @@ yf::Zoom::BackendPtr yf::Zoom::Frontend::get_backend_from_databases( } } - SearchablePtr sptr = *map_s; - m_backend.reset(); BackendPtr b(new Backend(sptr)); b->xsp = xsp; b->m_frontend_database = database; + std::string authentication = sptr->authentication; + + b->set_option("timeout", "40"); if (sptr->query_encoding.length()) b->set_option("rpnCharset", sptr->query_encoding.c_str()); + if (sptr->cfAuth.length()) + { + b->set_option("user", sptr->cfAuth.c_str()); + if (authentication.length()) + { + size_t found = authentication.find('/'); + if (found != std::string::npos) + { + cf_parm += "user=" + mp::util::uri_encode(authentication.substr(0, found)) + + "&password=" + mp::util::uri_encode(authentication.substr(found+1)); + } + else + cf_parm += "user=" + mp::util::uri_encode(authentication); + } + } + else if (authentication.length()) + b->set_option("user", authentication.c_str()); + + if (sptr->cfProxy.length()) + { + if (cf_parm.length()) + cf_parm += "&"; + cf_parm += "proxy=" + mp::util::uri_encode(sptr->cfProxy); + } + if (sptr->cfSubDb.length()) + { + if (cf_parm.length()) + cf_parm += "&"; + cf_parm += "subdatabase=" + mp::util::uri_encode(sptr->cfSubDb); + } + std::string url; if (sptr->sru.length()) { @@ -473,8 +655,14 @@ yf::Zoom::BackendPtr yf::Zoom::Frontend::get_backend_from_databases( b->set_option("sru", sptr->sru.c_str()); } else + { url = sptr->target; - + } + if (db_args.length()) + url += "," + db_args; + else if (cf_parm.length()) + url += "," + cf_parm; + yaz_log(YLOG_LOG, "url=%s", url.c_str()); b->connect(url, error, addinfo); if (*error == 0) { @@ -512,13 +700,12 @@ Z_Records *yf::Zoom::Frontend::get_records(Odr_int start, if (preferredRecordSyntax) { if (!oid_oidcmp(preferredRecordSyntax, yaz_oid_recsyn_xml) - && !strcmp(element_set_name, "pz2")) + && element_set_name && + !strcmp(element_set_name, "pz2")) { if (b->sptr->request_syntax.length()) - { syntax_name = b->sptr->request_syntax.c_str(); - enable_pz2_transform = true; - } + enable_pz2_transform = true; } else { @@ -575,7 +762,6 @@ Z_Records *yf::Zoom::Frontend::get_records(Odr_int start, strcpy(rec_type_str, b->sptr->use_turbomarc ? "txml" : "xml"); - // prevent buffer overflow ... if (b->sptr->record_encoding.length() > 0 && b->sptr->record_encoding.length() < @@ -646,6 +832,38 @@ Z_Records *yf::Zoom::Frontend::get_records(Odr_int start, return records; } +struct cql_node *yf::Zoom::Impl::convert_cql_fields(struct cql_node *cn, + ODR odr) +{ + struct cql_node *r = 0; + if (!cn) + return 0; + switch (cn->which) + { + case CQL_NODE_ST: + if (cn->u.st.index) + { + std::map::const_iterator it; + it = fieldmap.find(cn->u.st.index); + if (it == fieldmap.end()) + return cn; + if (it->second.length()) + cn->u.st.index = odr_strdup(odr, it->second.c_str()); + else + cn->u.st.index = 0; + } + break; + case CQL_NODE_BOOL: + r = convert_cql_fields(cn->u.boolean.left, odr); + if (!r) + r = convert_cql_fields(cn->u.boolean.right, odr); + break; + case CQL_NODE_SORT: + r = convert_cql_fields(cn->u.sort.search, odr); + break; + } + return r; +} void yf::Zoom::Frontend::handle_search(mp::Package &package) { @@ -713,10 +931,24 @@ void yf::Zoom::Frontend::handle_search(mp::Package &package) return; } struct cql_node *cn = cql_parser_result(cp); + struct cql_node *cn_error = m_p->convert_cql_fields(cn, odr); + if (cn_error) + { + // hopefully we are getting a ptr to a index+relation+term node + addinfo = 0; + if (cn_error->which == CQL_NODE_ST) + addinfo = cn_error->u.st.index; + + apdu_res = + odr.create_searchResponse(apdu_req, + YAZ_BIB1_UNSUPP_USE_ATTRIBUTE, + addinfo); + package.response() = apdu_res; + return; + } char ccl_buf[1024]; r = cql_to_ccl_buf(cn, ccl_buf, sizeof(ccl_buf)); - yaz_log(YLOG_LOG, "cql_to_ccl_buf returned %d", r); if (r == 0) { ccl_wrbuf = wrbuf_alloc(); @@ -747,17 +979,28 @@ void yf::Zoom::Frontend::handle_search(mp::Package &package) assert(pqf_wrbuf == 0); int cerror, cpos; struct ccl_rpn_node *cn; + yaz_log(YLOG_LOG, "CCL: %s", wrbuf_cstr(ccl_wrbuf)); cn = ccl_find_str(b->sptr->ccl_bibset, wrbuf_cstr(ccl_wrbuf), &cerror, &cpos); wrbuf_destroy(ccl_wrbuf); if (!cn) { char *addinfo = odr_strdup(odr, ccl_err_msg(cerror)); + int z3950_diag = YAZ_BIB1_MALFORMED_QUERY; + switch (cerror) + { + case CCL_ERR_UNKNOWN_QUAL: + z3950_diag = YAZ_BIB1_UNSUPP_USE_ATTRIBUTE; + break; + case CCL_ERR_TRUNC_NOT_LEFT: + case CCL_ERR_TRUNC_NOT_RIGHT: + case CCL_ERR_TRUNC_NOT_BOTH: + z3950_diag = YAZ_BIB1_UNSUPP_TRUNCATION_ATTRIBUTE; + break; + } apdu_res = - odr.create_searchResponse(apdu_req, - YAZ_BIB1_MALFORMED_QUERY, - addinfo); + odr.create_searchResponse(apdu_req, z3950_diag, addinfo); package.response() = apdu_res; return; } @@ -767,9 +1010,53 @@ void yf::Zoom::Frontend::handle_search(mp::Package &package) } assert(pqf_wrbuf); - b->search_pqf(wrbuf_cstr(pqf_wrbuf), &hits, &error, &addinfo); + if (b->get_option("sru")) + { + int status = 0; + Z_RPNQuery *zquery; + zquery = p_query_rpn(odr, wrbuf_cstr(pqf_wrbuf)); + WRBUF wrb = wrbuf_alloc(); + + if (!strcmp(b->get_option("sru"), "solr")) + { + solr_transform_t cqlt = solr_transform_create(); + + status = solr_transform_rpn2solr_wrbuf(cqlt, wrb, zquery); + + solr_transform_close(cqlt); + } + else + { + cql_transform_t cqlt = cql_transform_create(); + + status = cql_transform_rpn2cql_wrbuf(cqlt, wrb, zquery); + + cql_transform_close(cqlt); + } + if (status == 0) + { + yaz_log(YLOG_LOG, "search CQL: %s", wrbuf_cstr(wrb)); + b->search_cql(wrbuf_cstr(wrb), &hits, &error, &addinfo); + } + + wrbuf_destroy(wrb); + wrbuf_destroy(pqf_wrbuf); + if (status) + { + apdu_res = + odr.create_searchResponse(apdu_req, YAZ_BIB1_MALFORMED_QUERY, + "can not convert from RPN to CQL/SOLR"); + package.response() = apdu_res; + return; + } + } + else + { + yaz_log(YLOG_LOG, "search PQF: %s", wrbuf_cstr(pqf_wrbuf)); + b->search_pqf(wrbuf_cstr(pqf_wrbuf), &hits, &error, &addinfo); + wrbuf_destroy(pqf_wrbuf); + } - wrbuf_destroy(pqf_wrbuf); const char *element_set_name = 0; Odr_int number_to_present = 0;