X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=ztest%2Fztest.c;h=d5e5d2769fef96796f2858917b02b949ce2d65ca;hp=e5b7ebede0fd5ec12b19260b043da445d02e0860;hb=be042bc51e825a53074dc1b586f1fff0a002ab0b;hpb=836e12246ca0e84c73f552d5aa5a0ca8d4978de9 diff --git a/ztest/ztest.c b/ztest/ztest.c index e5b7ebe..d5e5d27 100644 --- a/ztest/ztest.c +++ b/ztest/ztest.c @@ -30,6 +30,7 @@ #include #include #include +#include #include "ztest.h" @@ -227,50 +228,59 @@ 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) { +static void addterms(ODR odr, Z_FacetField *facet_field, const char *facet_name) { int index; - int count = 100; - facet_field->num_terms = 3; - facet_field->terms = odr_malloc(odr, facet_field->num_terms * sizeof(*facet_field->terms)); + int freq = 100; + int length = strlen(facet_name) + 10; + char key[length]; + key[0] = '\0'; 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_Term *term; + Z_FacetTerm *facet_term; + // sprintf(key, "%s%d", facet_name, index); + //yaz_log(YLOG_DEBUG, "facet add term %s %d %s", facet_name, index, key); + term = term_create(odr, key); + facet_term = facet_term_create(odr, term, freq); + freq = freq - 10 ; + facet_field_term_set(odr, facet_field, facet_term, index); } } + 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)); + int index, new_index = 0; + Z_FacetList *new_list = facet_list_create(odr, facet_list->num); + for (index = 0; index < facet_list->num; index++) { - addterms(odr, facet_list->elements[index]); + struct attrvalues attrvalues; + facet_struct_init(&attrvalues); + attrvalues.limit = 10; + yaz_log(YLOG_LOG, "Attributes: %s %s %d ", attrvalues.useattr, attrvalues.useattrbuff, attrvalues.limit); + facetattrs(facet_list->elements[index]->attributes, &attrvalues); + yaz_log(YLOG_LOG, "Attributes: %s %s %d ", attrvalues.useattr, attrvalues.useattrbuff, attrvalues.limit); + if (attrvalues.errstring) + yaz_log(YLOG_LOG, "Error parsing attributes: %s", attrvalues.errstring); + if (attrvalues.limit > 0) { + new_list->elements[new_index] = facet_field_create(odr, facet_list->elements[index]->attributes, attrvalues.limit); + addterms(odr, new_list->elements[new_index], attrvalues.useattr); + new_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; + new_list->num = new_index; + if (new_index > 0) { + Z_OtherInformation *oi = odr_malloc(odr, sizeof(*oi)); + Z_OtherInformationUnit *oiu = odr_malloc(odr, sizeof(*oiu)); + oi->num_elements = 1; + oi->list = odr_malloc(odr, oi->num_elements * 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->which = Z_External_userFacets; + oiu->information.externallyDefinedInfo->u.facetList = new_list; + oi->list[0] = oiu; + return oi; + } + return 0; } int ztest_search(void *handle, bend_search_rr *rr) @@ -382,9 +392,14 @@ int ztest_search(void *handle, bend_search_rr *rr) { /* TODO Not general. Only handles one (Facet) OtherInformation. Overwrite */ Z_FacetList *facet_list = extract_facet_request(rr->stream, rr->search_input); + if (facet_list) { + yaz_log(YLOG_LOG, "%d Facets in search request.", facet_list->num); rr->search_info = build_facet_response(rr->stream, facet_list); } + else + yaz_log(YLOG_DEBUG, "No facets parsed search request."); + } do_delay(&new_set->search_delay); new_set->hits = rr->hits;