#include <math.h>
#include <stdlib.h>
#include <ctype.h>
+
+#if HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
#if HAVE_UNISTD_H
#include <unistd.h>
#endif
+#if HAVE_SYS_SELECT_H
+#include <sys/select.h>
+#endif
+#ifdef WIN32
+#include <windows.h>
+#endif
#include <yaz/log.h>
#include <yaz/backend.h>
#include <yaz/ill.h>
#include <yaz/diagbib1.h>
+#include <yaz/otherinfo.h>
+#include <yaz/facet.h>
#include "ztest.h"
*/
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
static int parse_delay(struct delay *delayp, const char *value)
{
- delayp->d1 = atof(value);
- delayp->d2 = 0.0;
+ if (sscanf(value, "%lf:%lf", &delayp->d1, &delayp->d2) == 2)
+ ;
+ else if (sscanf(value, "%lf", &delayp->d1) == 1)
+ delayp->d2 = 0.0;
+ else
+ return -1;
return 0;
}
-static void do_delay(const struct delay *delayp)
+static void ztest_sleep(double d)
{
+#ifdef WIN32
+ Sleep( (DWORD) (d * 1000));
+#else
struct timeval tv;
-
- tv.tv_sec = floor(delayp->d1);
- tv.tv_usec = (delayp->d1 - floor(delayp->d1)) * 1000000;
+ tv.tv_sec = floor(d);
+ tv.tv_usec = (d - floor(d)) * 1000000;
select(0, 0, 0, 0, &tv);
+#endif
+}
+
+static void do_delay(const struct delay *delayp)
+{
+ double d = delayp->d1;
+
+ if (d > 0.0)
+ {
+ if (delayp->d2 > d)
+ d += (rand()) * (delayp->d2 - d) / RAND_MAX;
+ ztest_sleep(d);
+ }
+}
+
+static void addterms(ODR odr, Z_FacetField *facet_field, const char *facet_name) {
+ int index;
+ 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_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, new_index = 0;
+ Z_FacetList *new_list = facet_list_create(odr, facet_list->num);
+
+ for (index = 0; index < facet_list->num; 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++;
+ }
+ }
+ 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)
new_set->db = xstrdup(db);
init_delay(&new_set->search_delay);
init_delay(&new_set->present_delay);
+ init_delay(&new_set->fetch_delay);
db_sep = strchr(db, '?');
if (db_sep)
const char *name = names[i];
const char *value = values[i];
if (!strcmp(name, "seed"))
- srandom(atoi(value));
+ srand(atoi(value));
else if (!strcmp(name, "search-delay"))
parse_delay(&new_set->search_delay, value);
else if (!strcmp(name, "present-delay"))
}
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) {
+ 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;
/* present request handler */
int ztest_present(void *handle, bend_present_rr *rr)
{
+ struct session_handle *sh = (struct session_handle*) handle;
+ struct result_set *set = get_set(sh, rr->setname);
+
+ if (!set)
+ {
+ rr->errcode = YAZ_BIB1_SPECIFIED_RESULT_SET_DOES_NOT_EXIST;
+ rr->errstring = odr_strdup(rr->stream, rr->setname);
+ return 0;
+ }
+ do_delay(&set->present_delay);
return 0;
}
r->errstring = odr_strdup(r->stream, r->setname);
return 0;
}
+ do_delay(&set->fetch_delay);
r->last_in_set = 0;
r->basename = set->db;
r->output_format = r->request_format;
projects / yaz-moved-to-github.git / blobdiff