From 03818d91fe9f98d27fe8033a7cc70c81e08c770b Mon Sep 17 00:00:00 2001 From: Adam Dickmeiss Date: Fri, 6 Jun 2014 14:54:25 +0200 Subject: [PATCH] Parse configure settings for xquery module --- src/metaproxy_filter_xquery.cpp | 149 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 149 insertions(+) diff --git a/src/metaproxy_filter_xquery.cpp b/src/metaproxy_filter_xquery.cpp index c7fe809..76067e3 100644 --- a/src/metaproxy_filter_xquery.cpp +++ b/src/metaproxy_filter_xquery.cpp @@ -18,9 +18,12 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #include #include +#include +#include namespace mp = metaproxy_1; namespace yf = mp::filter; +namespace mp_util = metaproxy_1::util; using namespace mp; namespace metaproxy_1 { @@ -35,10 +38,24 @@ namespace metaproxy_1 { void start() const; void stop(int signo) const; private: + std::map zorba_variables; + std::string zorba_filename; + std::string zorba_script; + std::string zorba_record_variable; }; } } +yf::XQuery::XQuery() +{ +} + +yf::XQuery::~XQuery() +{ + +} + + void yf::XQuery::start() const { } @@ -49,12 +66,144 @@ void yf::XQuery::stop(int signo) const void yf::XQuery::process(Package &package) const { + Z_GDU *gdu_req = package.request().get(); + Z_PresentRequest *pr_req = 0; + Z_SearchRequest *sr_req = 0; + + const char *input_schema = 0; + Odr_oid *input_syntax = 0; + + if (gdu_req && gdu_req->which == Z_GDU_Z3950 && + gdu_req->u.z3950->which == Z_APDU_presentRequest) + { + pr_req = gdu_req->u.z3950->u.presentRequest; + + input_schema = + mp_util::record_composition_to_esn(pr_req->recordComposition); + input_syntax = pr_req->preferredRecordSyntax; + } + else if (gdu_req && gdu_req->which == Z_GDU_Z3950 && + gdu_req->u.z3950->which == Z_APDU_searchRequest) + { + sr_req = gdu_req->u.z3950->u.searchRequest; + + input_syntax = sr_req->preferredRecordSyntax; + + // we don't know how many hits we're going to get and therefore + // the effective element set name.. Therefore we can only allow + // two cases.. Both equal or absent.. If not, we'll just have to + // disable the piggyback! + if (sr_req->smallSetElementSetNames + && + sr_req->mediumSetElementSetNames + && + sr_req->smallSetElementSetNames->which == Z_ElementSetNames_generic + && + sr_req->mediumSetElementSetNames->which == Z_ElementSetNames_generic + && + !strcmp(sr_req->smallSetElementSetNames->u.generic, + sr_req->mediumSetElementSetNames->u.generic)) + { + input_schema = sr_req->smallSetElementSetNames->u.generic; + } + else if (!sr_req->smallSetElementSetNames && + !sr_req->mediumSetElementSetNames) + ; // input_schema is 0 already + else + { + // disable piggyback (perhaps it was disabled already) + *sr_req->smallSetUpperBound = 0; + *sr_req->largeSetLowerBound = 0; + *sr_req->mediumSetPresentNumber = 0; + package.move(); + return; + } + // we can handle it in record_transform. + } + else + { + package.move(); + return; + } package.move(); } void yf::XQuery::configure(const xmlNode * ptr, bool test_only, const char *path) { + for (ptr = ptr->children; ptr; ptr = ptr->next) + { + if (ptr->type != XML_ELEMENT_NODE) + continue; + if (!strcmp((const char *) ptr->name, "setVariable")) + { + std::string name; + std::string value; + struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + if (!strcmp((const char *) attr->name, "name")) + name = mp::xml::get_text(attr->children); + else if (!strcmp((const char *) attr->name, "value")) + value = mp::xml::get_text(attr->children); + else + throw mp::filter::FilterException( + "Bad attribute " + std::string((const char *) + attr->name)); + if (name.length() > 0) + zorba_variables[name] = value; + } + else if (!strcmp((const char *) ptr->name, "filename")) + { + std::string value; + struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + if (!strcmp((const char *) attr->name, "value")) + value = mp::xml::get_text(attr->children); + else + throw mp::filter::FilterException( + "Bad attribute " + std::string((const char *) + attr->name)); + zorba_filename = value; + } + else if (!strcmp((const char *) ptr->name, "script")) + { + std::string value; + struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + if (!strcmp((const char *) attr->name, "value")) + value = mp::xml::get_text(attr->children); + else + throw mp::filter::FilterException( + "Bad attribute " + std::string((const char *) + attr->name)); + zorba_script = value; + } + else if (!strcmp((const char *) ptr->name, "record")) + { + std::string value; + struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + if (!strcmp((const char *) attr->name, "value")) + value = mp::xml::get_text(attr->children); + else + throw mp::filter::FilterException( + "Bad attribute " + std::string((const char *) + attr->name)); + zorba_record_variable = value; + } + else + { + throw mp::filter::FilterException("Bad element " + + std::string((const char *) + ptr->name)); + } + } + if (zorba_script.length() == 0) + throw mp::filter::FilterException("Missing element script"); + if (zorba_record_variable.length() == 0) + throw mp::filter::FilterException("Missing element record"); + if (zorba_filename.length() == 0) + throw mp::filter::FilterException("Missing element filename"); } static yf::Base* filter_creator() -- 1.7.10.4