/*
- * $Id: http_command.c,v 1.3 2007-01-03 06:23:44 quinn Exp $
+ * $Id: http_command.c,v 1.15 2007-01-09 22:27:10 quinn Exp $
*/
#include <stdio.h>
#include <ctype.h>
#include <sys/time.h>
+#if HAVE_CONFIG_H
+#include <cconfig.h>
+#endif
+
#include <yaz/yaz-util.h>
-#include "command.h"
#include "util.h"
#include "eventl.h"
#include "pazpar2.h"
http_send_response(c);
}
+// Compares two hitsbytarget nodes by hitcount
+static int cmp_ht(const void *p1, const void *p2)
+{
+ const struct hitsbytarget *h1 = p1;
+ const struct hitsbytarget *h2 = p2;
+ return h2->hits - h1->hits;
+}
+
+// This implements functionality somewhat similar to 'bytarget', but in a termlist form
+static void targets_termlist(WRBUF wrbuf, struct session *se)
+{
+ struct hitsbytarget *ht;
+ int count, i;
+
+ if (!(ht = hitsbytarget(se, &count)))
+ return;
+ qsort(ht, count, sizeof(struct hitsbytarget), cmp_ht);
+ for (i = 0; i < count && i < 15; i++)
+ {
+ wrbuf_puts(wrbuf, "\n<term>\n");
+ wrbuf_printf(wrbuf, "<name>%s</name>\n", ht[i].id);
+ wrbuf_printf(wrbuf, "<frequency>%d</frequency>\n", ht[i].hits);
+ wrbuf_printf(wrbuf, "<state>%s</state>\n", ht[i].state);
+ wrbuf_printf(wrbuf, "<diagnostic>%d</diagnostic>\n", ht[i].diagnostic);
+ wrbuf_puts(wrbuf, "\n</term>\n");
+ }
+}
+
static void cmd_termlist(struct http_channel *c)
{
struct http_response *rs = c->response;
int len;
int i;
char *name = http_argbyname(rq, "name");
+ int status;
if (!s)
return;
+ status = session_active_clients(s->psession);
+
if (!name)
name = "subject";
if (strlen(name) > 255)
wrbuf_rewind(c->wrbuf);
wrbuf_puts(c->wrbuf, "<termlist>");
+ wrbuf_printf(c->wrbuf, "\n<activeclients>%d</activeclients>", status);
while (*name)
{
char tname[256];
strncpy(tname, name, tp - name);
tname[tp - name] = '\0';
- p = termlist(s->psession, tname, &len);
wrbuf_printf(c->wrbuf, "\n<list name=\"%s\">\n", tname);
- if (p)
- for (i = 0; i < len; i++)
- {
- wrbuf_puts(c->wrbuf, "\n<term>");
- wrbuf_printf(c->wrbuf, "<name>%s</name>", p[i]->term);
- wrbuf_printf(c->wrbuf, "<frequency>%d</frequency>", p[i]->frequency);
- wrbuf_puts(c->wrbuf, "</term>");
- }
+ if (!strcmp(tname, "xtargets"))
+ targets_termlist(c->wrbuf, s->psession);
+ else
+ {
+ p = termlist(s->psession, tname, &len);
+ if (p)
+ for (i = 0; i < len; i++)
+ {
+ wrbuf_puts(c->wrbuf, "\n<term>");
+ wrbuf_printf(c->wrbuf, "<name>%s</name>", p[i]->term);
+ wrbuf_printf(c->wrbuf, "<frequency>%d</frequency>", p[i]->frequency);
+ wrbuf_puts(c->wrbuf, "</term>");
+ }
+ }
wrbuf_puts(c->wrbuf, "\n</list>");
name = tp;
if (*name == ',')
http_send_response(c);
}
-static void show_records(struct http_channel *c)
+static void write_metadata(WRBUF w, struct conf_service *service,
+ struct record_metadata **ml, int full)
+{
+ int imeta;
+
+ for (imeta = 0; imeta < service->num_metadata; imeta++)
+ {
+ struct conf_metadata *cmd = &service->metadata[imeta];
+ struct record_metadata *md;
+ if (!cmd->brief && !full)
+ continue;
+ for (md = ml[imeta]; md; md = md->next)
+ {
+ wrbuf_printf(w, "<md-%s>", cmd->name);
+ switch (cmd->type)
+ {
+ case Metadata_type_generic:
+ wrbuf_puts(w, md->data.text);
+ break;
+ case Metadata_type_year:
+ wrbuf_printf(w, "%d", md->data.year.year1);
+ if (md->data.year.year1 != md->data.year.year2)
+ wrbuf_printf(w, "-%d", md->data.year.year2);
+ break;
+ default:
+ wrbuf_puts(w, "[can't represent]");
+ }
+ wrbuf_printf(w, "</md-%s>", cmd->name);
+ }
+ }
+}
+
+static void cmd_record(struct http_channel *c)
+{
+ struct http_response *rs = c->response;
+ struct http_request *rq = c->request;
+ struct http_session *s = locate_session(rq, rs);
+ struct record_cluster *rec;
+ struct conf_service *service = global_parameters.server->service;
+ char *idstr = http_argbyname(rq, "id");
+ int id;
+
+ if (!s)
+ return;
+ if (!idstr)
+ {
+ error(rs, "417", "Must supply id", 0);
+ return;
+ }
+ wrbuf_rewind(c->wrbuf);
+ id = atoi(idstr);
+ if (!(rec = show_single(s->psession, id)))
+ {
+ error(rs, "500", "Record missing", 0);
+ return;
+ }
+ wrbuf_puts(c->wrbuf, "<record>\n");
+ wrbuf_printf(c->wrbuf, "<recid>%d</recid>", rec->recid);
+ write_metadata(c->wrbuf, service, rec->metadata, 1);
+ wrbuf_puts(c->wrbuf, "</record>\n");
+ rs->payload = nmem_strdup(c->nmem, wrbuf_buf(c->wrbuf));
+ http_send_response(c);
+}
+
+static void show_records(struct http_channel *c, int active)
{
struct http_request *rq = c->request;
struct http_response *rs = c->response;
struct http_session *s = locate_session(rq, rs);
- struct record **rl;
+ struct record_cluster **rl;
NMEM nmem_show;
char *start = http_argbyname(rq, "start");
char *num = http_argbyname(rq, "num");
if (!s)
return;
+ // We haven't counted clients yet if we're called on a block release
+ if (active < 0)
+ active = session_active_clients(s->psession);
+
if (start)
startn = atoi(start);
if (num)
wrbuf_rewind(c->wrbuf);
wrbuf_puts(c->wrbuf, "<show>\n<status>OK</status>\n");
+ wrbuf_printf(c->wrbuf, "<activeclients>%d</activeclients>\n", active);
wrbuf_printf(c->wrbuf, "<merged>%d</merged>\n", total);
wrbuf_printf(c->wrbuf, "<total>%d</total>\n", total_hits);
wrbuf_printf(c->wrbuf, "<start>%d</start>\n", startn);
{
int ccount;
struct record *p;
+ struct record_cluster *rec = rl[i];
+ struct conf_service *service = global_parameters.server->service;
wrbuf_puts(c->wrbuf, "<hit>\n");
- wrbuf_printf(c->wrbuf, "<title>%s</title>\n", rl[i]->title);
- for (ccount = 1, p = rl[i]->next_cluster; p; p = p->next_cluster, ccount++)
+ write_metadata(c->wrbuf, service, rec->metadata, 0);
+ for (ccount = 0, p = rl[i]->records; p; p = p->next, ccount++)
;
if (ccount > 1)
wrbuf_printf(c->wrbuf, "<count>%d</count>\n", ccount);
+ wrbuf_printf(c->wrbuf, "<recid>%d</recid>\n", rec->recid);
wrbuf_puts(c->wrbuf, "</hit>\n");
}
{
struct http_channel *c = (struct http_channel *) data;
- show_records(c);
+ show_records(c, -1);
}
static void cmd_show(struct http_channel *c)
struct http_response *rs = c->response;
struct http_session *s = locate_session(rq, rs);
char *block = http_argbyname(rq, "block");
+ int status;
if (!s)
return;
+ status = session_active_clients(s->psession);
+
if (block)
{
- if (!s->psession->reclist || !s->psession->reclist->num_records)
+ if (status && (!s->psession->reclist || !s->psession->reclist->num_records))
{
session_set_watch(s->psession, SESSION_WATCH_RECORDS, show_records_ready, c);
yaz_log(YLOG_DEBUG, "Blocking on cmd_show");
}
}
- show_records(c);
+ show_records(c, status);
}
static void cmd_ping(struct http_channel *c)
struct http_response *rs = c->response;
struct http_session *s = locate_session(rq, rs);
struct statistics stat;
+ int clients;
if (!s)
return;
+ clients = session_active_clients(s->psession);
statistics(s->psession, &stat);
wrbuf_rewind(c->wrbuf);
wrbuf_puts(c->wrbuf, "<stat>");
+ wrbuf_printf(c->wrbuf, "<activeclients>%d</activeclients>\n", clients);
wrbuf_printf(c->wrbuf, "<hits>%d</hits>\n", stat.num_hits);
wrbuf_printf(c->wrbuf, "<records>%d</records>\n", stat.num_records);
wrbuf_printf(c->wrbuf, "<clients>%d</clients>\n", stat.num_clients);
{ "termlist", cmd_termlist },
{ "exit", cmd_exit },
{ "ping", cmd_ping },
+ { "record", cmd_record },
{0,0}
};