src/logic.c

   1 /* $Id: logic.c,v 1.33 2007-05-24 10:57:38 adam Exp $
   2    Copyright (c) 2006-2007, Index Data.
   3
   4 This file is part of Pazpar2.
   5
   6 Pazpar2 is free software; you can redistribute it and/or modify it under
   7 the terms of the GNU General Public License as published by the Free
   8 Software Foundation; either version 2, or (at your option) any later
   9 version.
  10
  11 Pazpar2 is distributed in the hope that it will be useful, but WITHOUT ANY
  12 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14 for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with Pazpar2; see the file LICENSE.  If not, write to the
  18 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  19 02111-1307, USA.
  20  */
  21
  22 /** \file logic.c
  23     \brief high-level logic; mostly user sessions and settings
  24 */
  25
  26 #include <stdlib.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29 #include <sys/time.h>
  30 #include <unistd.h>
  31 #include <sys/socket.h>
  32 #include <netdb.h>
  33 #include <signal.h>
  34 #include <ctype.h>
  35 #include <assert.h>
  36
  37 #include <yaz/marcdisp.h>
  38 #include <yaz/comstack.h>
  39 #include <yaz/tcpip.h>
  40 #include <yaz/proto.h>
  41 #include <yaz/readconf.h>
  42 #include <yaz/pquery.h>
  43 #include <yaz/otherinfo.h>
  44 #include <yaz/yaz-util.h>
  45 #include <yaz/nmem.h>
  46 #include <yaz/query-charset.h>
  47 #include <yaz/querytowrbuf.h>
  48 #if YAZ_VERSIONL >= 0x020163
  49 #include <yaz/oid_db.h>
  50 #endif
  51
  52 #if HAVE_CONFIG_H
  53 #include "cconfig.h"
  54 #endif
  55
  56 #define USE_TIMING 0
  57 #if USE_TIMING
  58 #include <yaz/timing.h>
  59 #endif
  60
  61 #include <netinet/in.h>
  62
  63 #include "pazpar2.h"
  64 #include "eventl.h"
  65 #include "http.h"
  66 #include "termlists.h"
  67 #include "reclists.h"
  68 #include "relevance.h"
  69 #include "config.h"
  70 #include "database.h"
  71 #include "client.h"
  72 #include "settings.h"
  73 #include "normalize7bit.h"
  74
  75 #define MAX_CHUNK 15
  76
  77 // Note: Some things in this structure will eventually move to configuration
  78 struct parameters global_parameters =
  79 {
  80     "",
  81     "",
  82     "",
  83     "",
  84     0,
  85     0, /* dump_records */
  86     0, /* debug_mode */
  87     30,
  88     "81",
  89     "Index Data PazPar2",
  90     VERSION,
  91     600, // 10 minutes
  92     60,
  93     100,
  94     MAX_CHUNK,
  95     0,
  96     0
  97 };
  98
  99
 100 // Recursively traverse query structure to extract terms.
 101 void pull_terms(NMEM nmem, struct ccl_rpn_node *n, char **termlist, int *num)
 102 {
 103     char **words;
 104     int numwords;
 105     int i;
 106
 107     switch (n->kind)
 108     {
 109         case CCL_RPN_AND:
 110         case CCL_RPN_OR:
 111         case CCL_RPN_NOT:
 112         case CCL_RPN_PROX:
 113             pull_terms(nmem, n->u.p[0], termlist, num);
 114             pull_terms(nmem, n->u.p[1], termlist, num);
 115             break;
 116         case CCL_RPN_TERM:
 117             nmem_strsplit(nmem, " ", n->u.t.term, &words, &numwords);
 118             for (i = 0; i < numwords; i++)
 119                 termlist[(*num)++] = words[i];
 120             break;
 121         default: // NOOP
 122             break;
 123     }
 124 }
 125
 126
 127
 128 static void add_facet(struct session *s, const char *type, const char *value)
 129 {
 130     int i;
 131
 132     if (!*value)
 133         return;
 134     for (i = 0; i < s->num_termlists; i++)
 135         if (!strcmp(s->termlists[i].name, type))
 136             break;
 137     if (i == s->num_termlists)
 138     {
 139         if (i == SESSION_MAX_TERMLISTS)
 140         {
 141             yaz_log(YLOG_FATAL, "Too many termlists");
 142             return;
 143         }
 144
 145         s->termlists[i].name = nmem_strdup(s->nmem, type);
 146         s->termlists[i].termlist
 147             = termlist_create(s->nmem, s->expected_maxrecs, 15);
 148         s->num_termlists = i + 1;
 149     }
 150     termlist_insert(s->termlists[i].termlist, value);
 151 }
 152
 153 xmlDoc *normalize_record(struct session_database *sdb, Z_External *rec)
 154 {
 155     struct database_retrievalmap *m;
 156     struct database *db = sdb->database;
 157     xmlNode *res;
 158     xmlDoc *rdoc;
 159
 160     // First normalize to XML
 161     if (sdb->yaz_marc)
 162     {
 163         char *buf;
 164         int len;
 165         if (rec->which != Z_External_octet)
 166         {
 167             yaz_log(YLOG_WARN, "Unexpected external branch, probably BER %s",
 168                     db->url);
 169             return 0;
 170         }
 171         buf = (char*) rec->u.octet_aligned->buf;
 172         len = rec->u.octet_aligned->len;
 173         if (yaz_marc_read_iso2709(sdb->yaz_marc, buf, len) < 0)
 174         {
 175             yaz_log(YLOG_WARN, "Failed to decode MARC %s", db->url);
 176             return 0;
 177         }
 178
 179         yaz_marc_write_using_libxml2(sdb->yaz_marc, 1);
 180         if (yaz_marc_write_xml(sdb->yaz_marc, &res,
 181                     "http://www.loc.gov/MARC21/slim", 0, 0) < 0)
 182         {
 183             yaz_log(YLOG_WARN, "Failed to encode as XML %s",
 184                     db->url);
 185             return 0;
 186         }
 187         rdoc = xmlNewDoc((xmlChar *) "1.0");
 188         xmlDocSetRootElement(rdoc, res);
 189
 190     }
 191     else
 192     {
 193         yaz_log(YLOG_FATAL,
 194                 "Unknown native_syntax in normalize_record from %s",
 195                 db->url);
 196         return 0;
 197     }
 198
 199     if (global_parameters.dump_records){
 200         fprintf(stderr,
 201                 "Input Record (normalized) from %s\n----------------\n",
 202                 db->url);
 203 #if LIBXML_VERSION >= 20600
 204         xmlDocFormatDump(stderr, rdoc, 1);
 205 #else
 206         xmlDocDump(stderr, rdoc);
 207 #endif
 208     }
 209
 210     for (m = sdb->map; m; m = m->next){
 211         xmlDoc *new = 0;
 212
 213         {
 214             xmlNodePtr root = 0;
 215             new = xsltApplyStylesheet(m->stylesheet, rdoc, 0);
 216             root= xmlDocGetRootElement(new);
 217         if (!new || !root || !(root->children))
 218         {
 219             yaz_log(YLOG_WARN, "XSLT transformation failed from %s",
 220                     sdb->database->url);
 221             xmlFreeDoc(new);
 222             xmlFreeDoc(rdoc);
 223             return 0;
 224         }
 225         }
 226
 227         xmlFreeDoc(rdoc);
 228         rdoc = new;
 229     }
 230     if (global_parameters.dump_records)
 231     {
 232         fprintf(stderr, "Record from %s\n----------------\n",
 233                 sdb->database->url);
 234 #if LIBXML_VERSION >= 20600
 235         xmlDocFormatDump(stderr, rdoc, 1);
 236 #else
 237         xmlDocDump(stderr, rdoc);
 238 #endif
 239     }
 240     return rdoc;
 241 }
 242
 243 // Retrieve first defined value for 'name' for given database.
 244 // Will be extended to take into account user associated with session
 245 char *session_setting_oneval(struct session_database *db, int offset)
 246 {
 247     if (!db->settings[offset])
 248         return "";
 249     return db->settings[offset]->value;
 250 }
 251
 252
 253
 254 // Initialize YAZ Map structures for MARC-based targets
 255 static int prepare_yazmarc(struct session_database *sdb)
 256 {
 257     char *s;
 258
 259     if (!sdb->settings)
 260     {
 261         yaz_log(YLOG_WARN, "No settings for %s", sdb->database->url);
 262         return -1;
 263     }
 264     if ((s = session_setting_oneval(sdb, PZ_NATIVESYNTAX))
 265         && !strncmp(s, "iso2709", 7))
 266     {
 267         char *encoding = "marc-8s", *e;
 268         yaz_iconv_t cm;
 269
 270         // See if a native encoding is specified
 271         if ((e = strchr(s, ';')))
 272             encoding = e + 1;
 273
 274         sdb->yaz_marc = yaz_marc_create();
 275         yaz_marc_subfield_str(sdb->yaz_marc, "\t");
 276
 277         cm = yaz_iconv_open("utf-8", encoding);
 278         if (!cm)
 279         {
 280             yaz_log(YLOG_FATAL,
 281                     "Unable to map from %s to UTF-8 for target %s",
 282                     encoding, sdb->database->url);
 283             return -1;
 284         }
 285         yaz_marc_iconv(sdb->yaz_marc, cm);
 286     }
 287     return 0;
 288 }
 289
 290 // Prepare XSLT stylesheets for record normalization
 291 // Structures are allocated on the session_wide nmem to avoid having
 292 // to recompute this for every search. This would lead
 293 // to leaking if a single session was to repeatedly change the PZ_XSLT
 294 // setting. However, this is not a realistic use scenario.
 295 static int prepare_map(struct session *se, struct session_database *sdb)
 296 {
 297    char *s;
 298
 299     if (!sdb->settings)
 300     {
 301         yaz_log(YLOG_WARN, "No settings on %s", sdb->database->url);
 302         return -1;
 303     }
 304     if ((s = session_setting_oneval(sdb, PZ_XSLT)))
 305     {
 306         char **stylesheets;
 307         struct database_retrievalmap **m = &sdb->map;
 308         int num, i;
 309
 310         nmem_strsplit(se->session_nmem, ",", s, &stylesheets, &num);
 311         for (i = 0; i < num; i++)
 312         {
 313             (*m) = nmem_malloc(se->session_nmem, sizeof(**m));
 314             (*m)->next = 0;
 315             if (!((*m)->stylesheet = conf_load_stylesheet(stylesheets[i])))
 316             {
 317                 yaz_log(YLOG_FATAL, "Unable to load stylesheet: %s",
 318                         stylesheets[i]);
 319                 return -1;
 320             }
 321             m = &(*m)->next;
 322         }
 323     }
 324     if (!sdb->map)
 325         yaz_log(YLOG_WARN, "No Normalization stylesheet for target %s",
 326                 sdb->database->url);
 327     return 0;
 328 }
 329
 330 // This analyzes settings and recomputes any supporting data structures
 331 // if necessary.
 332 static int prepare_session_database(struct session *se,
 333                                     struct session_database *sdb)
 334 {
 335     if (!sdb->settings)
 336     {
 337         yaz_log(YLOG_WARN,
 338                 "No settings associated with %s", sdb->database->url);
 339         return -1;
 340     }
 341     if (sdb->settings[PZ_NATIVESYNTAX] && !sdb->yaz_marc)
 342     {
 343         if (prepare_yazmarc(sdb) < 0)
 344             return -1;
 345     }
 346     if (sdb->settings[PZ_XSLT] && !sdb->map)
 347     {
 348         if (prepare_map(se, sdb) < 0)
 349             return -1;
 350     }
 351     return 0;
 352 }
 353
 354
 355 void session_set_watch(struct session *s, int what,
 356                        session_watchfun fun, void *data)
 357 {
 358     s->watchlist[what].fun = fun;
 359     s->watchlist[what].data = data;
 360 }
 361
 362 void session_alert_watch(struct session *s, int what)
 363 {
 364     if (!s->watchlist[what].fun)
 365         return;
 366     (*s->watchlist[what].fun)(s->watchlist[what].data);
 367     s->watchlist[what].fun = 0;
 368     s->watchlist[what].data = 0;
 369 }
 370
 371 //callback for grep_databases
 372 static void select_targets_callback(void *context, struct session_database *db)
 373 {
 374     struct session *se = (struct session*) context;
 375     struct client *cl = client_create();
 376     client_set_database(cl, db);
 377     client_set_session(cl, se);
 378 }
 379
 380 // Associates a set of clients with a session;
 381 // Note: Session-databases represent databases with per-session
 382 // setting overrides
 383 int select_targets(struct session *se, struct database_criterion *crit)
 384 {
 385     while (se->clients)
 386         client_destroy(se->clients);
 387
 388     return session_grep_databases(se, crit, select_targets_callback);
 389 }
 390
 391 int session_active_clients(struct session *s)
 392 {
 393     struct client *c;
 394     int res = 0;
 395
 396     for (c = s->clients; c; c = client_next_in_session(c))
 397         if (client_is_active(c))
 398             res++;
 399
 400     return res;
 401 }
 402
 403 // parses crit1=val1,crit2=val2|val3,...
 404 static struct database_criterion *parse_filter(NMEM m, const char *buf)
 405 {
 406     struct database_criterion *res = 0;
 407     char **values;
 408     int num;
 409     int i;
 410
 411     if (!buf || !*buf)
 412         return 0;
 413     nmem_strsplit(m, ",", buf,  &values, &num);
 414     for (i = 0; i < num; i++)
 415     {
 416         char **subvalues;
 417         int subnum;
 418         int subi;
 419         struct database_criterion *new = nmem_malloc(m, sizeof(*new));
 420         char *eq = strchr(values[i], '=');
 421         if (!eq)
 422         {
 423             yaz_log(YLOG_WARN, "Missing equal-sign in filter");
 424             return 0;
 425         }
 426         *(eq++) = '\0';
 427         new->name = values[i];
 428         nmem_strsplit(m, "|", eq, &subvalues, &subnum);
 429         new->values = 0;
 430         for (subi = 0; subi < subnum; subi++)
 431         {
 432             struct database_criterion_value *newv
 433                 = nmem_malloc(m, sizeof(*newv));
 434             newv->value = subvalues[subi];
 435             newv->next = new->values;
 436             new->values = newv;
 437         }
 438         new->next = res;
 439         res = new;
 440     }
 441     return res;
 442 }
 443
 444 char *search(struct session *se, char *query, char *filter)
 445 {
 446     int live_channels = 0;
 447     struct client *cl;
 448     struct database_criterion *criteria;
 449
 450     yaz_log(YLOG_DEBUG, "Search");
 451
 452     nmem_reset(se->nmem);
 453     criteria = parse_filter(se->nmem, filter);
 454     se->requestid++;
 455     live_channels = select_targets(se, criteria);
 456     if (live_channels)
 457     {
 458         int maxrecs = live_channels * global_parameters.toget;
 459         se->num_termlists = 0;
 460         se->reclist = reclist_create(se->nmem, maxrecs);
 461         // This will be initialized in send_search()
 462         se->total_records = se->total_hits = se->total_merged = 0;
 463         se->expected_maxrecs = maxrecs;
 464     }
 465     else
 466         return "NOTARGETS";
 467
 468     se->relevance = 0;
 469
 470     for (cl = se->clients; cl; cl = client_next_in_session(cl))
 471     {
 472         if (prepare_session_database(se, client_get_database(cl)) < 0)
 473             return "CONFIG_ERROR";
 474         // Query must parse for all targets
 475         if (client_parse_query(cl, query) < 0)
 476             return "QUERY";
 477     }
 478
 479     for (cl = se->clients; cl; cl = client_next_in_session(cl))
 480     {
 481         client_prep_connection(cl);
 482     }
 483
 484     return 0;
 485 }
 486
 487 // Creates a new session_database object for a database
 488 static void session_init_databases_fun(void *context, struct database *db)
 489 {
 490     struct session *se = (struct session *) context;
 491     struct session_database *new = nmem_malloc(se->session_nmem, sizeof(*new));
 492     int num = settings_num();
 493     int i;
 494
 495     new->database = db;
 496     new->yaz_marc = 0;
 497
 498 #ifdef HAVE_ICU
 499     if (global_parameters.server && global_parameters.server->icu_chn)
 500         new->pct = pp2_charset_create(global_parameters.server->icu_chn);
 501     else
 502         new->pct = pp2_charset_create(0);
 503 #else // HAVE_ICU
 504     new->pct = pp2_charset_create(0);
 505 #endif // HAVE_ICU
 506
 507     new->map = 0;
 508     new->settings
 509         = nmem_malloc(se->session_nmem, sizeof(struct settings *) * num);
 510     memset(new->settings, 0, sizeof(struct settings*) * num);
 511
 512     if (db->settings)
 513     {
 514         for (i = 0; i < num; i++)
 515             new->settings[i] = db->settings[i];
 516     }
 517     new->next = se->databases;
 518     se->databases = new;
 519 }
 520
 521 // Doesn't free memory associated with sdb -- nmem takes care of that
 522 static void session_database_destroy(struct session_database *sdb)
 523 {
 524     struct database_retrievalmap *m;
 525
 526     for (m = sdb->map; m; m = m->next)
 527         xsltFreeStylesheet(m->stylesheet);
 528     if (sdb->yaz_marc)
 529         yaz_marc_destroy(sdb->yaz_marc);
 530     if (sdb->pct)
 531         pp2_charset_destroy(sdb->pct);
 532 }
 533
 534 // Initialize session_database list -- this represents this session's view
 535 // of the database list -- subject to modification by the settings ws command
 536 void session_init_databases(struct session *se)
 537 {
 538     se->databases = 0;
 539     grep_databases(se, 0, session_init_databases_fun);
 540 }
 541
 542 // Probably session_init_databases_fun should be refactored instead of
 543 // called here.
 544 static struct session_database *load_session_database(struct session *se,
 545                                                       char *id)
 546 {
 547     struct database *db = find_database(id, 0);
 548
 549     session_init_databases_fun((void*) se, db);
 550     // New sdb is head of se->databases list
 551     return se->databases;
 552 }
 553
 554 // Find an existing session database. If not found, load it
 555 static struct session_database *find_session_database(struct session *se,
 556                                                       char *id)
 557 {
 558     struct session_database *sdb;
 559
 560     for (sdb = se->databases; sdb; sdb = sdb->next)
 561         if (!strcmp(sdb->database->url, id))
 562             return sdb;
 563     return load_session_database(se, id);
 564 }
 565
 566 // Apply a session override to a database
 567 void session_apply_setting(struct session *se, char *dbname, char *setting,
 568                            char *value)
 569 {
 570     struct session_database *sdb = find_session_database(se, dbname);
 571     struct setting *new = nmem_malloc(se->session_nmem, sizeof(*new));
 572     int offset = settings_offset_cprefix(setting);
 573
 574     if (offset < 0)
 575     {
 576         yaz_log(YLOG_WARN, "Unknown setting %s", setting);
 577         return;
 578     }
 579     // Jakub: This breaks the filter setting.
 580     /*if (offset == PZ_ID)
 581     {
 582         yaz_log(YLOG_WARN, "No need to set pz:id setting. Ignoring");
 583         return;
 584     }*/
 585     new->precedence = 0;
 586     new->target = dbname;
 587     new->name = setting;
 588     new->value = value;
 589     new->next = sdb->settings[offset];
 590     sdb->settings[offset] = new;
 591
 592     // Force later recompute of settings-driven data structures
 593     // (happens when a search starts and client connections are prepared)
 594     switch (offset)
 595     {
 596         case PZ_NATIVESYNTAX:
 597             if (sdb->yaz_marc)
 598             {
 599                 yaz_marc_destroy(sdb->yaz_marc);
 600                 sdb->yaz_marc = 0;
 601             }
 602             break;
 603         case PZ_XSLT:
 604             if (sdb->map)
 605             {
 606                 struct database_retrievalmap *m;
 607                 // We don't worry about the map structure -- it's in nmem
 608                 for (m = sdb->map; m; m = m->next)
 609                     xsltFreeStylesheet(m->stylesheet);
 610                 sdb->map = 0;
 611             }
 612             break;
 613     }
 614 }
 615
 616 void destroy_session(struct session *s)
 617 {
 618     struct session_database *sdb;
 619
 620     yaz_log(YLOG_LOG, "Destroying session");
 621     while (s->clients)
 622         client_destroy(s->clients);
 623     for (sdb = s->databases; sdb; sdb = sdb->next)
 624         session_database_destroy(sdb);
 625     nmem_destroy(s->nmem);
 626     wrbuf_destroy(s->wrbuf);
 627 }
 628
 629 struct session *new_session(NMEM nmem)
 630 {
 631     int i;
 632     struct session *session = nmem_malloc(nmem, sizeof(*session));
 633
 634     yaz_log(YLOG_DEBUG, "New Pazpar2 session");
 635
 636     session->total_hits = 0;
 637     session->total_records = 0;
 638     session->num_termlists = 0;
 639     session->reclist = 0;
 640     session->requestid = -1;
 641     session->clients = 0;
 642     session->expected_maxrecs = 0;
 643     session->session_nmem = nmem;
 644     session->nmem = nmem_create();
 645     session->wrbuf = wrbuf_alloc();
 646     session_init_databases(session);
 647     for (i = 0; i <= SESSION_WATCH_MAX; i++)
 648     {
 649         session->watchlist[i].data = 0;
 650         session->watchlist[i].fun = 0;
 651     }
 652
 653     return session;
 654 }
 655
 656 struct hitsbytarget *hitsbytarget(struct session *se, int *count)
 657 {
 658     static struct hitsbytarget res[1000]; // FIXME MM
 659     struct client *cl;
 660
 661     *count = 0;
 662     for (cl = se->clients; cl; cl = client_next_in_session(cl))
 663     {
 664         char *name = session_setting_oneval(client_get_database(cl), PZ_NAME);
 665
 666         res[*count].id = client_get_database(cl)->database->url;
 667         res[*count].name = *name ? name : "Unknown";
 668         res[*count].hits = client_get_hits(cl);
 669         res[*count].records = client_get_num_records(cl);
 670         res[*count].diagnostic = client_get_diagnostic(cl);
 671         res[*count].state = client_get_state_str(cl);
 672         res[*count].connected  = client_get_connection(cl) ? 1 : 0;
 673         (*count)++;
 674     }
 675
 676     return res;
 677 }
 678
 679 struct termlist_score **termlist(struct session *s, const char *name, int *num)
 680 {
 681     int i;
 682
 683     for (i = 0; i < s->num_termlists; i++)
 684         if (!strcmp((const char *) s->termlists[i].name, name))
 685             return termlist_highscore(s->termlists[i].termlist, num);
 686     return 0;
 687 }
 688
 689 #ifdef MISSING_HEADERS
 690 void report_nmem_stats(void)
 691 {
 692     size_t in_use, is_free;
 693
 694     nmem_get_memory_in_use(&in_use);
 695     nmem_get_memory_free(&is_free);
 696
 697     yaz_log(YLOG_LOG, "nmem stat: use=%ld free=%ld",
 698             (long) in_use, (long) is_free);
 699 }
 700 #endif
 701
 702 struct record_cluster *show_single(struct session *s, int id)
 703 {
 704     struct record_cluster *r;
 705
 706     reclist_rewind(s->reclist);
 707     while ((r = reclist_read_record(s->reclist)))
 708         if (r->recid == id)
 709             return r;
 710     return 0;
 711 }
 712
 713 struct record_cluster **show(struct session *s, struct reclist_sortparms *sp,
 714                              int start, int *num, int *total, int *sumhits,
 715                              NMEM nmem_show)
 716 {
 717     struct record_cluster **recs = nmem_malloc(nmem_show, *num
 718                                        * sizeof(struct record_cluster *));
 719     struct reclist_sortparms *spp;
 720     int i;
 721 #if USE_TIMING
 722     yaz_timing_t t = yaz_timing_create();
 723 #endif
 724
 725     for (spp = sp; spp; spp = spp->next)
 726         if (spp->type == Metadata_sortkey_relevance)
 727         {
 728             relevance_prepare_read(s->relevance, s->reclist);
 729             break;
 730         }
 731     reclist_sort(s->reclist, sp);
 732
 733     *total = s->reclist->num_records;
 734     *sumhits = s->total_hits;
 735
 736     for (i = 0; i < start; i++)
 737         if (!reclist_read_record(s->reclist))
 738         {
 739             *num = 0;
 740             recs = 0;
 741             break;
 742         }
 743
 744     for (i = 0; i < *num; i++)
 745     {
 746         struct record_cluster *r = reclist_read_record(s->reclist);
 747         if (!r)
 748         {
 749             *num = i;
 750             break;
 751         }
 752         recs[i] = r;
 753     }
 754 #if USE_TIMING
 755     yaz_timing_stop(t);
 756     yaz_log(YLOG_LOG, "show %6.5f %3.2f %3.2f",
 757             yaz_timing_get_real(t), yaz_timing_get_user(t),
 758             yaz_timing_get_sys(t));
 759     yaz_timing_destroy(&t);
 760 #endif
 761     return recs;
 762 }
 763
 764 void statistics(struct session *se, struct statistics *stat)
 765 {
 766     struct client *cl;
 767     int count = 0;
 768
 769     memset(stat, 0, sizeof(*stat));
 770     for (cl = se->clients; cl; cl = client_next_in_session(cl))
 771     {
 772         if (!client_get_connection(cl))
 773             stat->num_no_connection++;
 774         switch (client_get_state(cl))
 775         {
 776             case Client_Connecting: stat->num_connecting++; break;
 777             case Client_Initializing: stat->num_initializing++; break;
 778             case Client_Searching: stat->num_searching++; break;
 779             case Client_Presenting: stat->num_presenting++; break;
 780             case Client_Idle: stat->num_idle++; break;
 781             case Client_Failed: stat->num_failed++; break;
 782             case Client_Error: stat->num_error++; break;
 783             default: break;
 784         }
 785         count++;
 786     }
 787     stat->num_hits = se->total_hits;
 788     stat->num_records = se->total_records;
 789
 790     stat->num_clients = count;
 791 }
 792
 793 void start_http_listener(void)
 794 {
 795     char hp[128] = "";
 796     struct conf_server *ser = global_parameters.server;
 797
 798     if (*global_parameters.listener_override)
 799         strcpy(hp, global_parameters.listener_override);
 800     else
 801     {
 802         strcpy(hp, ser->host ? ser->host : "");
 803         if (ser->port)
 804         {
 805             if (*hp)
 806                 strcat(hp, ":");
 807             sprintf(hp + strlen(hp), "%d", ser->port);
 808         }
 809     }
 810     http_init(hp);
 811 }
 812
 813 void start_proxy(void)
 814 {
 815     char hp[128] = "";
 816     struct conf_server *ser = global_parameters.server;
 817
 818     if (*global_parameters.proxy_override)
 819         strcpy(hp, global_parameters.proxy_override);
 820     else if (ser->proxy_host || ser->proxy_port)
 821     {
 822         strcpy(hp, ser->proxy_host ? ser->proxy_host : "");
 823         if (ser->proxy_port)
 824         {
 825             if (*hp)
 826                 strcat(hp, ":");
 827             sprintf(hp + strlen(hp), "%d", ser->proxy_port);
 828         }
 829     }
 830     else
 831         return;
 832
 833     http_set_proxyaddr(hp, ser->myurl ? ser->myurl : "");
 834 }
 835
 836 void start_zproxy(void)
 837 {
 838     struct conf_server *ser = global_parameters.server;
 839
 840     if (*global_parameters.zproxy_override){
 841         yaz_log(YLOG_LOG, "Z39.50 proxy  %s",
 842                 global_parameters.zproxy_override);
 843         return;
 844     }
 845
 846     else if (ser->zproxy_host || ser->zproxy_port)
 847     {
 848         char hp[128] = "";
 849
 850         strcpy(hp, ser->zproxy_host ? ser->zproxy_host : "");
 851         if (ser->zproxy_port)
 852         {
 853             if (*hp)
 854                 strcat(hp, ":");
 855             else
 856                 strcat(hp, "@:");
 857
 858             sprintf(hp + strlen(hp), "%d", ser->zproxy_port);
 859         }
 860         strcpy(global_parameters.zproxy_override, hp);
 861         yaz_log(YLOG_LOG, "Z39.50 proxy  %s",
 862                 global_parameters.zproxy_override);
 863
 864     }
 865     else
 866         return;
 867 }
 868
 869 // Master list of connections we're handling events to
 870 static IOCHAN channel_list = 0;
 871 void pazpar2_add_channel(IOCHAN chan)
 872 {
 873     chan->next = channel_list;
 874     channel_list = chan;
 875 }
 876
 877 void pazpar2_event_loop()
 878 {
 879     event_loop(&channel_list);
 880 }
 881
 882 struct record *ingest_record(struct client *cl, Z_External *rec,
 883                              int record_no)
 884 {
 885     xmlDoc *xdoc = normalize_record(client_get_database(cl), rec);
 886     xmlNode *root, *n;
 887     struct record *record;
 888     struct record_cluster *cluster;
 889     struct session *se = client_get_session(cl);
 890     xmlChar *mergekey, *mergekey_norm;
 891     xmlChar *type = 0;
 892     xmlChar *value = 0;
 893     struct conf_service *service = global_parameters.server->service;
 894
 895     if (!xdoc)
 896         return 0;
 897
 898     root = xmlDocGetRootElement(xdoc);
 899     if (!(mergekey = xmlGetProp(root, (xmlChar *) "mergekey")))
 900     {
 901         yaz_log(YLOG_WARN, "No mergekey found in record");
 902         xmlFreeDoc(xdoc);
 903         return 0;
 904     }
 905
 906     record = record_create(se->nmem,
 907                            service->num_metadata, service->num_sortkeys);
 908     record_assign_client(record, cl);
 909
 910     mergekey_norm = (xmlChar *) nmem_strdup(se->nmem, (char*) mergekey);
 911     xmlFree(mergekey);
 912     normalize7bit_mergekey((char *) mergekey_norm, 0);
 913
 914     cluster = reclist_insert(se->reclist,
 915                              global_parameters.server->service,
 916                              record, (char *) mergekey_norm,
 917                              &se->total_merged);
 918     if (global_parameters.dump_records)
 919         yaz_log(YLOG_LOG, "Cluster id %d from %s (#%d)", cluster->recid,
 920                 client_get_database(cl)->database->url, record_no);
 921     if (!cluster)
 922     {
 923         /* no room for record */
 924         xmlFreeDoc(xdoc);
 925         return 0;
 926     }
 927     relevance_newrec(se->relevance, cluster);
 928
 929
 930     // now parsing XML record and adding data to cluster or record metadata
 931     for (n = root->children; n; n = n->next)
 932     {
 933         if (type)
 934             xmlFree(type);
 935         if (value)
 936             xmlFree(value);
 937         type = value = 0;
 938
 939         if (n->type != XML_ELEMENT_NODE)
 940             continue;
 941         if (!strcmp((const char *) n->name, "metadata"))
 942         {
 943             struct conf_metadata *ser_md = 0;
 944             struct conf_sortkey *ser_sk = 0;
 945             struct record_metadata **wheretoput = 0;
 946             struct record_metadata *rec_md = 0;
 947             int md_field_id = -1;
 948             int sk_field_id = -1;
 949             int first, last;
 950
 951             type = xmlGetProp(n, (xmlChar *) "type");
 952             value = xmlNodeListGetString(xdoc, n->children, 0);
 953
 954             if (!type || !value)
 955                 continue;
 956
 957             md_field_id
 958                 = conf_service_metadata_field_id(service, (const char *) type);
 959             if (md_field_id < 0)
 960             {
 961                 yaz_log(YLOG_WARN,
 962                         "Ignoring unknown metadata element: %s", type);
 963                 continue;
 964             }
 965
 966             ser_md = &service->metadata[md_field_id];
 967
 968             if (ser_md->sortkey_offset >= 0){
 969                 sk_field_id = ser_md->sortkey_offset;
 970                 ser_sk = &service->sortkeys[sk_field_id];
 971             }
 972
 973             // Find out where we are putting it - based on merge or not
 974             if (ser_md->merge == Metadata_merge_no)
 975                 wheretoput = &record->metadata[md_field_id];
 976             else
 977                 wheretoput = &cluster->metadata[md_field_id];
 978
 979             // create new record_metadata
 980             rec_md = record_metadata_create(se->nmem);
 981
 982             // and polulate with data:
 983             // type based charmapping decisions follow here
 984             if (ser_md->type == Metadata_type_generic)
 985             {
 986
 987                 char * p = (char *) value;
 988                 p = normalize7bit_generic(p, " ,/.:([");
 989
 990                 rec_md->data.text = nmem_strdup(se->nmem, p);
 991
 992             }
 993             else if (ser_md->type == Metadata_type_year)
 994             {
 995                 if (extract7bit_years((char *) value, &first, &last) < 0)
 996                     continue;
 997             }
 998             else
 999             {
1000                 yaz_log(YLOG_WARN,
1001                         "Unknown type in metadata element %s", type);
1002                 continue;
1003             }
1004
1005             // and polulate with data:
1006             // assign cluster or record based on merge action
1007             if (ser_md->merge == Metadata_merge_unique)
1008             {
1009                 struct record_metadata *mnode;
1010                 for (mnode = *wheretoput; mnode; mnode = mnode->next)
1011                     if (!strcmp((const char *) mnode->data.text,
1012                                 rec_md->data.text))
1013                         break;
1014                 if (!mnode)
1015                 {
1016                     rec_md->next = *wheretoput;
1017                     *wheretoput = rec_md;
1018                 }
1019             }
1020             else if (ser_md->merge == Metadata_merge_longest)
1021             {
1022                 if (!*wheretoput
1023                     || strlen(rec_md->data.text)
1024                        > strlen((*wheretoput)->data.text))
1025                 {
1026                     *wheretoput = rec_md;
1027                     if (ser_sk)
1028                     {
1029                         char *s = nmem_strdup(se->nmem, rec_md->data.text);
1030                         if (!cluster->sortkeys[sk_field_id])
1031                             cluster->sortkeys[sk_field_id] =
1032                                 nmem_malloc(se->nmem,
1033                                             sizeof(union data_types));
1034                         normalize7bit_mergekey(s,
1035                              (ser_sk->type == Metadata_sortkey_skiparticle));
1036                         cluster->sortkeys[sk_field_id]->text = s;
1037                     }
1038                 }
1039             }
1040             else if (ser_md->merge == Metadata_merge_all
1041                      || ser_md->merge == Metadata_merge_no)
1042             {
1043                 rec_md->next = *wheretoput;
1044                 *wheretoput = rec_md;
1045             }
1046             else if (ser_md->merge == Metadata_merge_range)
1047             {
1048                 if (!*wheretoput)
1049                 {
1050                     *wheretoput = rec_md;
1051                     (*wheretoput)->data.number.min = first;
1052                     (*wheretoput)->data.number.max = last;
1053                     if (ser_sk)
1054                         cluster->sortkeys[sk_field_id]
1055                             = &rec_md->data;
1056                 }
1057                 else
1058                 {
1059                     if (first < (*wheretoput)->data.number.min)
1060                         (*wheretoput)->data.number.min = first;
1061                     if (last > (*wheretoput)->data.number.max)
1062                         (*wheretoput)->data.number.max = last;
1063                 }
1064 #ifdef GAGA
1065                 if (ser_sk)
1066                 {
1067                     union data_types *sdata
1068                         = cluster->sortkeys[sk_field_id];
1069                     yaz_log(YLOG_LOG, "SK range: %d-%d",
1070                             sdata->number.min, sdata->number.max);
1071                 }
1072 #endif
1073             }
1074
1075
1076             // ranking of _all_ fields enabled ...
1077             if (ser_md->rank)
1078                 relevance_countwords(se->relevance, cluster,
1079                                      (char *) value, ser_md->rank);
1080
1081             // construct facets ...
1082             if (ser_md->termlist)
1083             {
1084                 if (ser_md->type == Metadata_type_year)
1085                 {
1086                     char year[64];
1087                     sprintf(year, "%d", last);
1088                     add_facet(se, (char *) type, year);
1089                     if (first != last)
1090                     {
1091                         sprintf(year, "%d", first);
1092                         add_facet(se, (char *) type, year);
1093                     }
1094                 }
1095                 else
1096                     add_facet(se, (char *) type, (char *) value);
1097             }
1098
1099             // cleaning up
1100             xmlFree(type);
1101             xmlFree(value);
1102             type = value = 0;
1103         }
1104         else
1105             yaz_log(YLOG_WARN,
1106                     "Unexpected element %s in internal record", n->name);
1107     }
1108     if (type)
1109         xmlFree(type);
1110     if (value)
1111         xmlFree(value);
1112
1113     xmlFreeDoc(xdoc);
1114
1115     relevance_donerecord(se->relevance, cluster);
1116     se->total_records++;
1117
1118     return record;
1119 }
1120
1121
1122
1123 /*
1124  * Local variables:
1125  * c-basic-offset: 4
1126  * indent-tabs-mode: nil
1127  * End:
1128  * vim: shiftwidth=4 tabstop=8 expandtab
1129  */