X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=zoom%2Fzoomsh.c;h=37f5452223f3572d20e9da37e3eafdae9ef17493;hp=a8ec761434e90a38f34ce67149871178e622a3bd;hb=d84e4c3cd444d04aee7beeadb5bb0ae061ee24c9;hpb=77c5a4fca8b516fd39b8ba213daed17a465a6b2a diff --git a/zoom/zoomsh.c b/zoom/zoomsh.c index a8ec761..37f5452 100644 --- a/zoom/zoomsh.c +++ b/zoom/zoomsh.c @@ -1,5 +1,5 @@ /* This file is part of the YAZ toolkit. - * Copyright (C) 1995-2012 Index Data + * Copyright (C) Index Data * See the file LICENSE for details. */ /** \file zoomsh.c @@ -9,15 +9,20 @@ #include #endif +#if HAVE_UNISTD_H +#include +#endif + #include #include #include #include #include +#include #include #if HAVE_READLINE_READLINE_H -#include +#include #endif #if HAVE_READLINE_HISTORY_H #include @@ -26,14 +31,38 @@ #include #include +struct zoom_sh { + WRBUF strategy; + WRBUF criteria; + ZOOM_options options; + struct zoom_db *list; +}; + +struct zoom_db { + ZOOM_connection con; + ZOOM_resultset res; + struct zoom_db *next; +}; + #define MAX_CON 100 -static void process_events(ZOOM_connection *c) +static void process_events(struct zoom_sh *sh) { - int i; + ZOOM_connection *c; + int i, number; + struct zoom_db *db; + + for (number = 0, db = sh->list; db; db = db->next) + if (db->con) + number++; + c = xmalloc(sizeof(*c) * number); + + for (i = 0, db = sh->list; db; db = db->next) + if (db->con) + c[i++] = db->con; yaz_log(YLOG_DEBUG, "process_events"); - while ((i = ZOOM_event(MAX_CON, c)) != 0) + while ((i = ZOOM_event(number, c)) != 0) { int peek = ZOOM_connection_peek_event(c[i-1]); int event = ZOOM_connection_last_event(c[i-1]); @@ -42,6 +71,7 @@ static void process_events(ZOOM_connection *c) event, ZOOM_get_event_str(event)); } + xfree(c); } static int next_token_chars(const char **cpp, const char **t_start, @@ -107,9 +137,7 @@ static int is_command(const char *cmd_str, const char *this_str, int this_len) return 1; } -static int cmd_set(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_set(struct zoom_sh *sh, const char **args) { WRBUF key; const char *val_buf; @@ -122,16 +150,14 @@ static int cmd_set(ZOOM_connection *c, ZOOM_resultset *r, } val_len = next_token_chars(args, &val_buf, ""); if (val_len != -1) - ZOOM_options_setl(options, wrbuf_cstr(key), val_buf, val_len); + ZOOM_options_setl(sh->options, wrbuf_cstr(key), val_buf, val_len); else - ZOOM_options_set(options, wrbuf_cstr(key), 0); + ZOOM_options_set(sh->options, wrbuf_cstr(key), 0); wrbuf_destroy(key); return 0; } -static int cmd_get(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_get(struct zoom_sh *sh, const char **args) { WRBUF key; if (!(key = next_token_new_wrbuf(args))) @@ -141,16 +167,22 @@ static int cmd_get(ZOOM_connection *c, ZOOM_resultset *r, } else { - const char *val = ZOOM_options_get(options, wrbuf_cstr(key)); + const char *val = ZOOM_options_get(sh->options, wrbuf_cstr(key)); printf("%s = %s\n", wrbuf_cstr(key), val ? val : ""); wrbuf_destroy(key); } return 0; } -static int cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_shell(struct zoom_sh *sh, const char **args) +{ + int ret = system(*args); + if (ret) + printf("system command returned %d\n", ret); + return 0; +} + +static int cmd_rget(struct zoom_sh *sh, const char **args) { WRBUF key; if (!(key = next_token_new_wrbuf(args))) @@ -160,44 +192,42 @@ static int cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, } else { - int i; - for (i = 0; ilist; + for (; db; db = db->next) { - const char *val; - if (!r[i]) - continue; - - val = ZOOM_resultset_option_get(r[i], wrbuf_cstr(key)); - printf("%s = %s\n", wrbuf_cstr(key), val ? val : ""); + if (db->res) + { + const char *val = + ZOOM_resultset_option_get(db->res, wrbuf_cstr(key)); + printf("%s = %s\n", wrbuf_cstr(key), val ? val : ""); + } } wrbuf_destroy(key); } return 0; } -static int cmd_close(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_close(struct zoom_sh *sh, const char **args) { + struct zoom_db **dbp; WRBUF host; - int i; host = next_token_new_wrbuf(args); - for (i = 0; ilist; *dbp; ) { const char *h; - if (!c[i]) - continue; - if (!host) - { - ZOOM_connection_destroy(c[i]); - c[i] = 0; - } - else if ((h = ZOOM_connection_option_get(c[i], "host")) - && !strcmp(h, wrbuf_cstr(host))) + struct zoom_db *db = *dbp; + if (!db->con || !host || + ((h = ZOOM_connection_option_get(db->con, "host")) + && !strcmp(h, wrbuf_cstr(host)))) { - ZOOM_connection_destroy(c[i]); - c[i] = 0; + *dbp = (*dbp)->next; + ZOOM_connection_destroy(db->con); + ZOOM_resultset_destroy(db->res); + xfree(db); } + else + dbp = &(*dbp)->next; } if (host) wrbuf_destroy(host); @@ -214,14 +244,14 @@ static void display_records(ZOOM_connection c, size_t pos = i + start; ZOOM_record rec = ZOOM_resultset_record(r, pos); const char *db = ZOOM_record_get(rec, "database", 0); - + if (ZOOM_record_error(rec, 0, 0, 0)) { const char *msg; const char *addinfo; const char *diagset; int error = ZOOM_record_error(rec, &msg, &addinfo, &diagset); - + printf("%lld %s: %s (%s:%d) %s\n", (long long) pos, (db ? db : "unknown"), msg, diagset, error, addinfo ? addinfo : "none"); @@ -251,15 +281,13 @@ static void display_records(ZOOM_connection c, } } -static int cmd_show(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_show(struct zoom_sh *sh, const char **args) { - int i; size_t start = 0, count = 1; const char *type = "render"; WRBUF render_str = 0; int ret = 0; + struct zoom_db *db; { WRBUF tmp; @@ -280,28 +308,29 @@ static int cmd_show(ZOOM_connection *c, ZOOM_resultset *r, if (render_str) type = wrbuf_cstr(render_str); - for (i = 0; i < MAX_CON; i++) - ZOOM_resultset_records(r[i], 0, start, count); - process_events(c); + for (db = sh->list; db; db = db->next) + if (db->res) + ZOOM_resultset_records(db->res, 0, start, count); + process_events(sh); - for (i = 0; i < MAX_CON; i++) + for (db = sh->list; db; db = db->next) { int error; const char *errmsg, *addinfo, *dset; /* display errors if any */ - if (!c[i]) + if (!db->con) continue; - if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, &dset))) { printf("%s error: %s (%s:%d) %s\n", - ZOOM_connection_option_get(c[i], "host"), errmsg, + ZOOM_connection_option_get(db->con, "host"), errmsg, dset, error, addinfo); ret = 1; } - else if (r[i]) + else if (db->res) { /* OK, no major errors. Display records... */ - display_records(c[i], r[i], start, count, type); + display_records(db->con, db->res, start, count, type); } } if (render_str) @@ -309,49 +338,51 @@ static int cmd_show(ZOOM_connection *c, ZOOM_resultset *r, return ret; } -static void display_facets(ZOOM_facet_field *facets, int count) { - int index; +static void display_facets(ZOOM_facet_field *facets, int count) +{ + int i; printf("Facets: \n"); - for (index = 0; index < count; index++) { - int term_index; - const char *facet_name = ZOOM_facet_field_name(facets[index]); + for (i = 0; i < count; i++) + { + int j; + const char *facet_name = ZOOM_facet_field_name(facets[i]); printf(" %s: \n", facet_name); - for (term_index = 0; term_index < ZOOM_facet_field_term_count(facets[index]); term_index++) { + for (j = 0; j < ZOOM_facet_field_term_count(facets[i]); j++) + { int freq = 0; - const char *term = ZOOM_facet_field_get_term(facets[index], term_index, &freq); + const char *term = ZOOM_facet_field_get_term(facets[i], j, &freq); printf(" %s(%d) \n", term, freq); } } } -static int cmd_facets(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_facets(struct zoom_sh *sh, const char **args) { - int i; + struct zoom_db *db; int ret = 0; - process_events(c); + process_events(sh); - for (i = 0; i < MAX_CON; i++) + for (db = sh->list; db; db = db->next) { int error; const char *errmsg, *addinfo, *dset; /* display errors if any */ - if (!c[i]) + if (!db->con) continue; - if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { printf("%s error: %s (%s:%d) %s\n", - ZOOM_connection_option_get(c[i], "host"), errmsg, + ZOOM_connection_option_get(db->con, "host"), errmsg, dset, error, addinfo); ret = 1; } - else if (r[i]) + else if (db->res) { - int num_facets = ZOOM_resultset_facets_size(r[i]); + int num_facets = ZOOM_resultset_facets_size(db->res); if (num_facets) { - ZOOM_facet_field *facets = ZOOM_resultset_facets(r[i]); + ZOOM_facet_field *facets = ZOOM_resultset_facets(db->res); display_facets(facets, num_facets); } } @@ -359,72 +390,81 @@ static int cmd_facets(ZOOM_connection *c, ZOOM_resultset *r, return ret; } -static int cmd_suggestions(ZOOM_connection *c, ZOOM_resultset *r, ZOOM_options options, const char **args) +static int cmd_suggestions(struct zoom_sh *sh, const char **args) { - int i; + struct zoom_db *db; int ret = 0; - process_events(c); + process_events(sh); - for (i = 0; i < MAX_CON; i++) + for (db = sh->list; db; db = db->next) { int error; const char *errmsg, *addinfo, *dset; /* display errors if any */ - if (!c[i]) + if (!db->con) continue; - if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { printf("%s error: %s (%s:%d) %s\n", - ZOOM_connection_option_get(c[i], "host"), errmsg, + ZOOM_connection_option_get(db->con, "host"), errmsg, dset, error, addinfo); ret = 1; } - else if (r[i]) + else if (db->res) { - const char *suggestions = ZOOM_resultset_option_get(r[i], "suggestions"); - if (suggestions) { + const char *suggestions = + ZOOM_resultset_option_get(db->res, "suggestions"); + if (suggestions) printf("Suggestions: \n%s\n", suggestions); - } } } return ret; } - -static int cmd_ext(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_ext(struct zoom_sh *sh, const char **args) { - ZOOM_package p[MAX_CON]; - int i; int ret = 0; + ZOOM_package *p = 0; + struct zoom_db *db; + int i, number; WRBUF ext_type_str = next_token_new_wrbuf(args); - - for (i = 0; ilist; db; db = db->next) + if (db->con) + number++; + + p = xmalloc(sizeof(*p) * number); + + for (i = 0, db = sh->list; db; db = db->next) + if (db->con) { - p[i] = ZOOM_connection_package(c[i], 0); + p[i] = ZOOM_connection_package(db->con, 0); ZOOM_package_send(p[i], ext_type_str ? wrbuf_cstr(ext_type_str):0); + i++; } - else - p[i] = 0; - } - process_events(c); + process_events(sh); - for (i = 0; ilist; db; db = db->next) { int error; const char *errmsg, *addinfo, *dset; /* display errors if any */ - if (!p[i]) + if (!db->con) continue; - if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { printf("%s error: %s (%s:%d) %s\n", - ZOOM_connection_option_get(c[i], "host"), errmsg, + ZOOM_connection_option_get(db->con, "host"), errmsg, dset, error, addinfo); ret = 1; } @@ -440,29 +480,64 @@ static int cmd_ext(ZOOM_connection *c, ZOOM_resultset *r, printf("xmlUpdateDoc: %s\n", v); } ZOOM_package_destroy(p[i]); + i++; } if (ext_type_str) wrbuf_destroy(ext_type_str); + xfree(p); return ret; } -static int cmd_debug(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_debug(struct zoom_sh *sh, const char **args) { yaz_log_init_level(YLOG_ALL); return 0; } -static int cmd_search(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static void display_search_result(struct zoom_db *db) +{ + const char *v; + int num; + + v = ZOOM_resultset_option_get(db->res, "searchresult.size"); + if (v && (num = atoi(v))) + { + int i; + printf("SearchResult-1:"); + for (i = 0; i < num; i++) + { + const char *v; + char str[60]; + + if (i) + printf(","); + + sprintf(str, "searchresult.%d.id", i); + v = ZOOM_resultset_option_get(db->res, str); + if (v) + printf(" id=%s", v); + + sprintf(str, "searchresult.%d.subquery.term", i); + v = ZOOM_resultset_option_get(db->res, str); + if (v) + printf(" term=%s", v); + + sprintf(str, "searchresult.%d.count", i); + v = ZOOM_resultset_option_get(db->res, str); + if (v) + printf(" cnt=%s", v); + } + printf("\n"); + } +} + +static int cmd_search(struct zoom_sh *sh, const char **args) { ZOOM_query s; const char *query_str = *args; - int i; int ret = 0; - + struct zoom_db *db; + s = ZOOM_query_create(); while (*query_str == ' ') query_str++; @@ -476,49 +551,63 @@ static int cmd_search(ZOOM_connection *c, ZOOM_resultset *r, ZOOM_query_destroy(s); return 1; } - for (i = 0; istrategy && wrbuf_len(sh->strategy) && wrbuf_len(sh->criteria)) { - - if (c[i]) + int r = ZOOM_query_sortby2(s, wrbuf_cstr(sh->strategy), + wrbuf_cstr(sh->criteria)); + if (r) + { + if (r == -1) + printf("Bad sortby strategy: %s\n", wrbuf_cstr(sh->strategy)); + else + printf("Bad sortby criteria: %s\n", wrbuf_cstr(sh->criteria)); + ZOOM_query_destroy(s); + return 1; + } + printf("sortby added\n"); + } + for (db = sh->list; db; db = db->next) + { + if (db->con) { - ZOOM_resultset_destroy(r[i]); - r[i] = 0; + ZOOM_resultset_destroy(db->res); + db->res = ZOOM_connection_search(db->con, s); } - if (c[i]) - r[i] = ZOOM_connection_search(c[i], s); } ZOOM_query_destroy(s); - process_events(c); + process_events(sh); - for (i = 0; ilist; db; db = db->next) { int error; const char *errmsg, *addinfo, *dset; /* display errors if any */ - if (!c[i]) + if (!db->con) continue; - if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { printf("%s error: %s (%s:%d) %s\n", - ZOOM_connection_option_get(c[i], "host"), errmsg, + ZOOM_connection_option_get(db->con, "host"), errmsg, dset, error, addinfo); ret = 1; } - else if (r[i]) + else if (db->res) { /* OK, no major errors. Look at the result count */ - int start = ZOOM_options_get_int(options, "start", 0); - int count = ZOOM_options_get_int(options, "count", 0); + int start = ZOOM_options_get_int(sh->options, "start", 0); + int count = ZOOM_options_get_int(sh->options, "count", 0); int facet_num; - printf("%s: %lld hits\n", ZOOM_connection_option_get(c[i], "host"), - (long long int) ZOOM_resultset_size(r[i])); - - facet_num = ZOOM_resultset_facets_size(r[i]); + printf("%s: %lld hits\n", ZOOM_connection_option_get(db->con, + "host"), + (long long int) ZOOM_resultset_size(db->res)); + + facet_num = ZOOM_resultset_facets_size(db->res); if (facet_num) { - ZOOM_facet_field *facets = ZOOM_resultset_facets(r[i]); + ZOOM_facet_field *facets = ZOOM_resultset_facets(db->res); int facet_idx; for (facet_idx = 0; facet_idx < facet_num; facet_idx++) { @@ -535,23 +624,23 @@ static int cmd_search(ZOOM_connection *c, ZOOM_resultset *r, } } } + display_search_result(db); /* and display */ - display_records(c[i], r[i], start, count, "render"); + display_records(db->con, db->res, start, count, "render"); } } return ret; } -static int cmd_scan(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_scan(struct zoom_sh *sh, const char **args) { const char *query_str = *args; ZOOM_query query = ZOOM_query_create(); - int i; + int i, number; int ret = 0; - ZOOM_scanset s[MAX_CON]; - + ZOOM_scanset *s = 0; + struct zoom_db *db; + while (*query_str == ' ') query_str++; @@ -566,28 +655,32 @@ static int cmd_scan(ZOOM_connection *c, ZOOM_resultset *r, return 1; } - for (i = 0; ilist; db; db = db->next) + if (db->con) + number++; + + s = xmalloc(sizeof(*s) * number); + + for (i = 0, db = sh->list; db; db = db->next) + if (db->con) + s[i++] = ZOOM_connection_scan1(db->con, query); + ZOOM_query_destroy(query); - process_events(c); + process_events(sh); - for (i = 0; ilist; db; db = db->next) { int error; const char *errmsg, *addinfo, *dset; /* display errors if any */ - if (!c[i]) + if (!db->con) continue; - if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { printf("%s error: %s (%s:%d) %s\n", - ZOOM_connection_option_get(c[i], "host"), errmsg, + ZOOM_connection_option_get(db->con, "host"), errmsg, dset, error, addinfo); ret = 1; } @@ -601,39 +694,56 @@ static int cmd_scan(ZOOM_connection *c, ZOOM_resultset *r, const char *term = ZOOM_scanset_display_term(s[i], p, &occ, &len); printf("%.*s %lld\n", (int) len, term, (long long int) occ); - } + } ZOOM_scanset_destroy(s[i]); } + i++; } + xfree(s); return ret; } -static int cmd_sort(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_sortby(struct zoom_sh *sh, const char **args) +{ + WRBUF strategy; + const char *criteria; + if (!(strategy = next_token_new_wrbuf(args))) + { + printf("missing argument argument: strategy and criteria\n"); + return 1; + } + criteria = *args; + while (*criteria == ' ') + criteria++; + wrbuf_destroy(sh->strategy); + sh->strategy = strategy; + + wrbuf_rewind(sh->criteria); + wrbuf_puts(sh->criteria, criteria); + return 0; +} + +static int cmd_sort(struct zoom_sh *sh, const char **args) { const char *sort_spec = *args; - int i; int ret = 0; - + struct zoom_db *db; + while (*sort_spec == ' ') sort_spec++; - - for (i = 0; ilist; db; db = db->next) + if (db->res) + ZOOM_resultset_sort(db->res, "yaz", sort_spec); + process_events(sh); return ret; } -static int cmd_help(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_help(struct zoom_sh *sh, const char **args) { printf("connect \n"); printf("search \n"); + printf("sortby \n"); printf("show [ [ [\n"); @@ -642,6 +752,7 @@ static int cmd_help(ZOOM_connection *c, ZOOM_resultset *r, printf("ext \n"); printf("set