X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=ztest%2Fztest.c;h=e5b7ebede0fd5ec12b19260b043da445d02e0860;hp=bbbb402c871f817a0e8cd83f5eae7c1625782abc;hb=836e12246ca0e84c73f552d5aa5a0ca8d4978de9;hpb=4e627aa761e0b91b240f703173eeb82462ee8409 diff --git a/ztest/ztest.c b/ztest/ztest.c index bbbb402..e5b7ebe 100644 --- a/ztest/ztest.c +++ b/ztest/ztest.c @@ -29,6 +29,7 @@ #include #include #include +#include #include "ztest.h" @@ -133,12 +134,16 @@ static Odr_int get_term_hit(Z_RPNStructure *s) */ static Odr_int get_hit_count(Z_Query *q) { - Odr_int h = -1; if (q->which == Z_Query_type_1 || q->which == Z_Query_type_101) + { + Odr_int h = -1; h = get_term_hit(q->u.type_1->RPNStructure); - if (h == -1) - h = rand() % 24; - return h; + if (h == -1) + h = rand() % 24; + return h; + } + else + return 24; } /** \brief checks if it's a dummy Slow database @@ -222,6 +227,52 @@ static void do_delay(const struct delay *delayp) } } +Z_FacetList *extract_facet_request(ODR odr, Z_OtherInformation *search_input) { + Z_OtherInformation **oi; + Z_FacetList *facet_list = yaz_oi_get_facetlist_oid(oi, odr, yaz_oid_userinfo_facet_1, 1, 0); + + return facet_list; +} + +Z_Term *term_new(ODR odr, const char *cstr) { + Z_Term *term = odr_malloc(odr, sizeof(*term)); + term->which = Z_Term_characterString; + term->u.characterString = odr_strdup(odr, cstr); + return term; +} + +static void addterms(ODR odr, Z_FacetField *facet_field) { + int index; + int count = 100; + facet_field->num_terms = 3; + facet_field->terms = odr_malloc(odr, facet_field->num_terms * sizeof(*facet_field->terms)); + for (index = 0; index < facet_field->num_terms; index++) { + Z_FacetTerm *facet_term = odr_malloc(odr, sizeof(*facet_term)); + facet_term->count = odr_malloc(odr, sizeof(*facet_term->count)); + *facet_term->count = count; + facet_term->term = term_new(odr, "key"); + count = count - 10 ; + facet_field->terms[index] = facet_term; + } +} +Z_OtherInformation *build_facet_response(ODR odr, Z_FacetList *facet_list) { + int index; + Z_OtherInformation *oi = odr_malloc(odr, sizeof(*oi)); + Z_OtherInformationUnit *oiu = odr_malloc(odr, sizeof(*oiu)); + for (index = 0; index < facet_list->num; index++) { + addterms(odr, facet_list->elements[index]); + } + oi->list = odr_malloc(odr, sizeof(*oi->list)); + oiu->category = 0; + oiu->which = Z_OtherInfo_externallyDefinedInfo; + oiu->information.externallyDefinedInfo = odr_malloc(odr, sizeof(*oiu->information.externallyDefinedInfo)); + oiu->information.externallyDefinedInfo->direct_reference = odr_oiddup(odr, yaz_oid_userinfo_facet_1); + + oiu->information.externallyDefinedInfo->u.facetList = facet_list; + oi->list[0] = oiu; + return oi; +} + int ztest_search(void *handle, bend_search_rr *rr) { struct session_handle *sh = (struct session_handle*) handle; @@ -327,6 +378,14 @@ int ztest_search(void *handle, bend_search_rr *rr) } rr->hits = get_hit_count(rr->query); + if (1) + { + /* TODO Not general. Only handles one (Facet) OtherInformation. Overwrite */ + Z_FacetList *facet_list = extract_facet_request(rr->stream, rr->search_input); + if (facet_list) { + rr->search_info = build_facet_response(rr->stream, facet_list); + } + } do_delay(&new_set->search_delay); new_set->hits = rr->hits;