X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=zoom%2Fzoomsh.c;h=37f5452223f3572d20e9da37e3eafdae9ef17493;hp=530e08e231ac60b81672955aec69a29edfd7a57f;hb=11ed1645174d8ca8d8dcf5ef2bbe91e70a4a3e00;hpb=4f3bcae93d51a26709c12b51261c3d95af610cb2 diff --git a/zoom/zoomsh.c b/zoom/zoomsh.c index 530e08e..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-2011 Index Data + * Copyright (C) Index Data * See the file LICENSE for details. */ /** \file zoomsh.c @@ -9,14 +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 @@ -25,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]); @@ -41,9 +71,11 @@ static void process_events(ZOOM_connection *c) event, ZOOM_get_event_str(event)); } + xfree(c); } -static int next_token(const char **cpp, const char **t_start) +static int next_token_chars(const char **cpp, const char **t_start, + const char *tok_chars) { int len = 0; const char *cp = *cpp; @@ -64,7 +96,7 @@ static int next_token(const char **cpp, const char **t_start) else { *t_start = cp; - while (*cp && *cp != ' ' && *cp != '\r' && *cp != '\n') + while (*cp && !strchr(tok_chars, *cp)) { cp++; len++; @@ -76,6 +108,12 @@ static int next_token(const char **cpp, const char **t_start) return len; /* return -1 if no token was read .. */ } +static int next_token(const char **cpp, const char **t_start) +{ + return next_token_chars(cpp, t_start, "\r\n "); +} + + static WRBUF next_token_new_wrbuf(const char **cpp) { WRBUF w = 0; @@ -99,95 +137,101 @@ static int is_command(const char *cmd_str, const char *this_str, int this_len) return 1; } -static void 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, val; + WRBUF key; + const char *val_buf; + int val_len; if (!(key = next_token_new_wrbuf(args))) { printf("missing argument for set\n"); - return ; - } - if ((val = next_token_new_wrbuf(args))) - { - ZOOM_options_set(options, wrbuf_cstr(key), wrbuf_cstr(val)); - wrbuf_destroy(val); + return 1; } + val_len = next_token_chars(args, &val_buf, ""); + if (val_len != -1) + 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 void 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))) { printf("missing argument for get\n"); + return 1; } 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_shell(struct zoom_sh *sh, const char **args) +{ + int ret = system(*args); + if (ret) + printf("system command returned %d\n", ret); + return 0; } -static void cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_rget(struct zoom_sh *sh, const char **args) { WRBUF key; if (!(key = next_token_new_wrbuf(args))) { printf("missing argument for get\n"); + return 1; } 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 void 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) + 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; - } - else if ((h = ZOOM_connection_option_get(c[i], "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); + return 0; } static void display_records(ZOOM_connection c, @@ -200,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"); @@ -237,14 +281,13 @@ static void display_records(ZOOM_connection c, } } -static void 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; @@ -265,135 +308,166 @@ static void 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); - else if (r[i]) + ret = 1; + } + 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) wrbuf_destroy(render_str); - + 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 void 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; - size_t start = 0, count = 1; - const char *type = "render"; - WRBUF render_str = 0; + struct zoom_db *db; + int ret = 0; - if (0) - { - WRBUF tmp; + process_events(sh); - if ((tmp = next_token_new_wrbuf(args))) + for (db = sh->list; db; db = db->next) + { + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { - start = atoi(wrbuf_cstr(tmp)); - wrbuf_destroy(tmp); + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; } - - if ((tmp = next_token_new_wrbuf(args))) + else if (db->res) { - count = atoi(wrbuf_cstr(tmp)); - wrbuf_destroy(tmp); + int num_facets = ZOOM_resultset_facets_size(db->res); + if (num_facets) { + ZOOM_facet_field *facets = ZOOM_resultset_facets(db->res); + display_facets(facets, num_facets); + } } - render_str = next_token_new_wrbuf(args); } - if (render_str) - type = wrbuf_cstr(render_str); + return ret; +} + +static int cmd_suggestions(struct zoom_sh *sh, const char **args) +{ + 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); - else if (r[i]) + ret = 1; + } + else if (db->res) { - int num_facets = ZOOM_resultset_facets_size(r[i]); - if (num_facets) { - ZOOM_facet_field *facets = ZOOM_resultset_facets(r[i]); - display_facets(facets, num_facets); - } + const char *suggestions = + ZOOM_resultset_option_get(db->res, "suggestions"); + if (suggestions) + printf("Suggestions: \n%s\n", suggestions); } } - if (render_str) - wrbuf_destroy(render_str); - + return ret; } -static void 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; + } else if (p[i]) { const char *v; @@ -406,26 +480,64 @@ static void 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 void 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 void 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++; @@ -436,57 +548,99 @@ static void cmd_search(ZOOM_connection *c, ZOOM_resultset *r, else if (ZOOM_query_prefix(s, query_str)) { printf("Bad PQF: %s\n", query_str); - return; + 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) { - ZOOM_resultset_destroy(r[i]); - r[i] = 0; + 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(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); - else if (r[i]) + ret = 1; + } + 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])); + 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(db->res); + int facet_idx; + for (facet_idx = 0; facet_idx < facet_num; facet_idx++) + { + const char *name = ZOOM_facet_field_name(facets[facet_idx]); + size_t term_idx; + size_t term_num = ZOOM_facet_field_term_count(facets[facet_idx]); + printf("facet: %s\n", name); + for (term_idx = 0; term_idx < term_num; term_idx++ ) + { + int freq; + const char *term = + ZOOM_facet_field_get_term(facets[facet_idx], term_idx, &freq); + printf("term: %s %d\n", term, freq); + } + } + } + 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 void 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; - ZOOM_scanset s[MAX_CON]; - + int i, number; + int ret = 0; + ZOOM_scanset *s = 0; + struct zoom_db *db; + while (*query_str == ' ') query_str++; @@ -497,33 +651,41 @@ static void cmd_scan(ZOOM_connection *c, ZOOM_resultset *r, else if (ZOOM_query_prefix(query, query_str)) { printf("Bad PQF: %s\n", query_str); - return; + ZOOM_query_destroy(query); + 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); - - if (s[i]) { + ret = 1; + } + if (s[i]) + { size_t p, sz = ZOOM_scanset_size(s[i]); for (p = 0; p < sz; p++) { @@ -532,36 +694,56 @@ static void 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 void 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 void 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"); @@ -570,6 +752,7 @@ static void cmd_help(ZOOM_connection *c, ZOOM_resultset *r, printf("ext \n"); printf("set