index/index.h

   1 /* $Id: index.h,v 1.132 2005-04-13 13:03:47 adam Exp $
   2    Copyright (C) 1995-2005
   3    Index Data ApS
   4
   5 This file is part of the Zebra server.
   6
   7 Zebra is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
  19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  20 02111-1307, USA.
  21 */
  22
  23 #ifndef INDEX_H
  24 #define INDEX_H
  25
  26 #include <time.h>
  27 #include <stdlib.h>
  28 #include <idzebra/version.h>
  29 #include <idzebra/util.h>
  30 #include <sortidx.h>
  31
  32 #if HAVE_SYS_TIMES_H
  33 #include <sys/times.h>
  34 #endif
  35 #include <sys/stat.h>
  36
  37 #include <idzebra/dict.h>
  38 #include <idzebra/isams.h>
  39 #include <idzebra/isamc.h>
  40 #include <idzebra/isamb.h>
  41 #include <d1_absyn.h>
  42 #include <idzebra/recgrs.h>
  43 #include "recindex.h"
  44 #include <idzebra/api.h>
  45 #include "zinfo.h"
  46 #include <passwddb.h>
  47 #include <rset.h>
  48
  49 YAZ_BEGIN_CDECL
  50
  51 #define ISAM_DEFAULT "b"
  52
  53 #define SU_SCHEME 1
  54
  55 #define IT_MAX_WORD 256
  56
  57 #define IT_KEY_LEVEL_MAX 4
  58 struct it_key {
  59     int  len;
  60     zint mem[IT_KEY_LEVEL_MAX];
  61 };
  62
  63 enum dirsKind { dirs_dir, dirs_file };
  64
  65 struct dir_entry {
  66     enum dirsKind kind;
  67     char *name;
  68     time_t mtime;
  69 };
  70
  71 struct dirs_entry {
  72     enum dirsKind kind;
  73     char path[256];
  74     SYSNO sysno;
  75     time_t mtime;
  76 };
  77
  78 void getFnameTmp (Res res, char *fname, int no);
  79
  80 struct dirs_info *dirs_open (Dict dict, const char *rep, int rw);
  81 struct dirs_info *dirs_fopen (Dict dict, const char *path);
  82 struct dirs_entry *dirs_read (struct dirs_info *p);
  83 struct dirs_entry *dirs_last (struct dirs_info *p);
  84 void dirs_mkdir (struct dirs_info *p, const char *src, time_t mtime);
  85 void dirs_rmdir (struct dirs_info *p, const char *src);
  86 void dirs_add (struct dirs_info *p, const char *src, SYSNO sysno, time_t mtime);
  87 void dirs_del (struct dirs_info *p, const char *src);
  88 void dirs_free (struct dirs_info **pp);
  89
  90 struct dir_entry *dir_open (const char *rep, const char *base,
  91                             int follow_links);
  92 void dir_sort (struct dir_entry *e);
  93 void dir_free (struct dir_entry **e_p);
  94
  95 void repositoryUpdate (ZebraHandle zh, const char *path);
  96 void repositoryAdd (ZebraHandle zh, const char *path);
  97 void repositoryDelete (ZebraHandle zh, const char *path);
  98 void repositoryShow (ZebraHandle zh, const char *path);
  99
 100 int key_open (ZebraHandle zh, int mem);
 101 int key_close (ZebraHandle zh);
 102 int key_compare (const void *p1, const void *p2);
 103 void key_init(struct it_key *k);
 104 char *key_print_it (const void *p, char *buf);
 105 zint key_get_seq (const void *p);
 106 int key_compare_it (const void *p1, const void *p2);
 107 int key_qsort_compare (const void *p1, const void *p2);
 108 void key_logdump (int mask, const void *p);
 109 void key_logdump_txt (int logmask, const void *p, const char *txt);
 110 void inv_prstat (ZebraHandle zh);
 111 void inv_compact (BFiles bfs);
 112 void key_input (ZebraHandle zh, int nkeys, int cache, Res res);
 113 ISAMS_M *key_isams_m (Res res, ISAMS_M *me);
 114 ISAMC_M *key_isamc_m (Res res, ISAMC_M *me);
 115 int merge_sort (char **buf, int from, int to);
 116 int key_SU_code (int ch, char *out);
 117
 118 #define FNAME_DICT "dict"
 119 #define FNAME_ISAM "isam"
 120 #define FNAME_ISAMC "isamc"
 121 #define FNAME_ISAMS "isams"
 122 #define FNAME_ISAMH "isamh"
 123 #define FNAME_ISAMD "isamd"
 124 #define FNAME_CONFIG "zebra.cfg"
 125
 126 #define GMATCH_DICT "gmatch"
 127 #define FMATCH_DICT "fmatch"
 128
 129 struct strtab *strtab_mk (void);
 130 int strtab_src (struct strtab *t, const char *name, void ***infop);
 131 void strtab_del (struct strtab *t,
 132                  void (*func)(const char *name, void *info, void *data),
 133                  void *data);
 134 int index_char_cvt (int c);
 135 int index_word_prefix (char *string, int attset_ordinal,
 136                        int local_attribute, const char *databaseName);
 137
 138
 139 void zebraIndexLockMsg (ZebraHandle zh, const char *str);
 140 void zebraIndexUnlock (ZebraHandle zh);
 141 int zebraIndexLock (BFiles bfs, ZebraHandle zh, int commitNow, const char *rval);
 142 int zebraIndexWait (ZebraHandle zh, int commitPhase);
 143
 144 #define FNAME_MAIN_LOCK   "zebraidx.LCK"
 145 #define FNAME_COMMIT_LOCK "zebracmt.LCK"
 146 #define FNAME_ORG_LOCK    "zebraorg.LCK"
 147 #define FNAME_TOUCH_TIME  "zebraidx.time"
 148
 149 typedef struct zebra_lock_info *ZebraLockHandle;
 150 ZebraLockHandle zebra_lock_create(const char *dir,
 151                                   const char *file, int excl_flag);
 152 void zebra_lock_destroy (ZebraLockHandle h);
 153 int zebra_lock (ZebraLockHandle h);
 154 int zebra_lock_nb (ZebraLockHandle h);
 155 int zebra_unlock (ZebraLockHandle h);
 156 int zebra_lock_fd (ZebraLockHandle h);
 157 void zebra_lock_prefix (Res res, char *dst);
 158 char *zebra_mk_fname (const char *dir, const char *name);
 159
 160 int zebra_lock_w (ZebraLockHandle h);
 161 int zebra_lock_r (ZebraLockHandle h);
 162
 163 void zebra_load_atts (data1_handle dh, Res res);
 164
 165 int key_SU_decode (int *ch, const unsigned char *out);
 166 int key_SU_encode (int ch, char *out);
 167
 168 #define ENCODE_BUFLEN 768
 169 struct encode_info {
 170     int  sysno;  /* previously written values for delta-compress */
 171     int  seqno;
 172     int  cmd;
 173     int prevsys; /* buffer for skipping insert/delete pairs */
 174     int prevseq;
 175     int prevcmd;
 176     int keylen; /* tells if we have an unwritten key in buf, and how long*/
 177     void *encode_handle;
 178     char buf[ENCODE_BUFLEN];
 179 };
 180
 181 void encode_key_init (struct encode_info *i);
 182 char *encode_key_int (int d, char *bp);
 183 void encode_key_write (char *k, struct encode_info *i, FILE *outf);
 184 void encode_key_flush (struct encode_info *i, FILE *outf);
 185
 186 typedef struct zebra_set *ZebraSet;
 187
 188 typedef struct zebra_rank_class {
 189     struct rank_control *control;
 190     int init_flag;
 191     void *class_handle;
 192     struct zebra_rank_class *next;
 193 } *ZebraRankClass;
 194
 195 struct recKeys {
 196     int buf_used;
 197     int buf_max;
 198     char *buf;
 199     void *codec_handle;
 200 };
 201
 202 struct sortKeys {
 203     int buf_used;
 204     int buf_max;
 205     char *buf;
 206 };
 207
 208 struct zebra_register {
 209     char *name;
 210
 211     ISAMS isams;
 212     ISAMC isamc;
 213     ISAMB isamb;
 214     Dict dict;
 215     Dict matchDict;
 216     SortIdx sortIdx;
 217     int registerState; /* 0 (no commit pages), 1 (use commit pages) */
 218     time_t registerChange;
 219     BFiles bfs;
 220     Records records;
 221     ZebraExplainInfo zei;
 222
 223     char *server_path_prefix;
 224     data1_handle dh;
 225     ZebraMaps zebra_maps;
 226     ZebraRankClass rank_classes;
 227     RecTypes recTypes;
 228     int seqno;
 229     int last_val;
 230     int stop_flag;
 231     int active; /* 0=shutdown, 1=enabled and inactive, 2=activated */
 232
 233     struct recKeys keys;
 234 #if 1
 235     struct sortKeys sortKeys;
 236 #else
 237     struct sortKey *sortKeys;
 238 #endif
 239     char **key_buf;
 240     size_t ptr_top;
 241     size_t ptr_i;
 242     size_t key_buf_used;
 243     int key_file_no;
 244 };
 245
 246 struct zebra_service {
 247     int stop_flag;
 248     Res global_res;
 249     char *configName;
 250     struct zebra_session *sessions;
 251     struct zebra_register *regs;
 252     Zebra_mutex_cond session_lock;
 253     Passwd_db passwd_db;
 254     const char *path_root;
 255     RecTypeClass record_classes;
 256     NMEM nmem;
 257 };
 258
 259
 260 struct zebra_session {
 261     struct zebra_session *next;
 262     struct zebra_service *service;
 263     struct zebra_register *reg;
 264
 265     char *xadmin_databaseName;
 266
 267     char **basenames;
 268     int num_basenames;
 269     char *reg_name;
 270     char *path_reg;
 271
 272     ZebraLockHandle lock_normal;
 273     ZebraLockHandle lock_shadow;
 274
 275     int trans_no;
 276     int trans_w_no;
 277
 278     int destroyed;
 279     ZebraSet sets;
 280     Res res;
 281     char *user_perm;
 282     int errCode;
 283     zint hits;
 284     char *errString;
 285 #if HAVE_SYS_TIMES_H
 286     struct tms tms1;
 287     struct tms tms2;
 288 #endif
 289     int  shadow_enable;
 290
 291     zint records_inserted;
 292     zint records_updated;
 293     zint records_deleted;
 294     zint records_processed;
 295     char *record_encoding;
 296
 297     yaz_iconv_t iconv_to_utf8;
 298     yaz_iconv_t iconv_from_utf8;
 299
 300     int m_follow_links;
 301     const char *m_group;
 302     const char *m_record_id;
 303     const char *m_record_type;
 304     int m_store_data;
 305     int m_store_keys;
 306     int m_explain_database;
 307     int m_flag_rw;
 308     int m_file_verbose_limit;
 309 };
 310
 311 struct rank_control {
 312     char *name;
 313     void *(*create)(ZebraHandle zh);
 314     void (*destroy)(struct zebra_register *reg, void *class_handle);
 315     void *(*begin)(struct zebra_register *reg,
 316                    void *class_handle, RSET rset, NMEM nmem,
 317                    TERMID *terms, int numterms);
 318     /* ### Could add parameters to begin:
 319      *  char *index;    // author, title, etc.
 320      *  int dbsize;     // number of records in database
 321      *  int rssize;     // number of records in result set (estimate?)
 322      */
 323     void (*end)(struct zebra_register *reg, void *set_handle);
 324     int (*calc)(void *set_handle, zint sysno);
 325     void (*add)(void *set_handle, int seqno, TERMID term);
 326 };
 327
 328 struct term_set_entry {
 329     char *term;
 330     struct term_set_entry *next;
 331 };
 332
 333 struct term_set_list {
 334     struct term_set_entry *first;
 335     struct term_set_entry *last;
 336 };
 337
 338 RSET rpn_search (ZebraHandle zh, NMEM mem, NMEM rset_nmem,
 339                  Z_RPNQuery *rpn, int num_bases, char **basenames,
 340                  const char *setname, ZebraSet sset);
 341
 342
 343 void rpn_scan (ZebraHandle zh, ODR stream, Z_AttributesPlusTerm *zapt,
 344                oid_value attributeset,
 345                int num_bases, char **basenames,
 346                int *position, int *num_entries, ZebraScanEntry **list,
 347                int *is_partial, RSET limit_set, int return_zero);
 348
 349 RSET rset_trunc (ZebraHandle zh, ISAM_P *isam_p, int no,
 350                  const char *term, int length_term, const char *flags,
 351                  int preserve_position, int term_type, NMEM rset_nmem,
 352                  const struct key_control *kctrl, int scope);
 353
 354 void resultSetAddTerm (ZebraHandle zh, ZebraSet s, int reg_type,
 355                        const char *db, int set,
 356                        int use, const char *term);
 357 ZebraSet resultSetAdd (ZebraHandle zh, const char *name, int ov);
 358 ZebraSet resultSetGet (ZebraHandle zh, const char *name);
 359 ZebraSet resultSetAddRPN (ZebraHandle zh, NMEM m, Z_RPNQuery *rpn,
 360                           int num_bases, char **basenames,
 361                           const char *setname);
 362 RSET resultSetRef (ZebraHandle zh, const char *resultSetId);
 363 void resultSetDestroy (ZebraHandle zh, int num_names, char **names,
 364                        int *statuses);
 365 void resultSetSort (ZebraHandle zh, NMEM nmem,
 366                     int num_input_setnames, const char **input_setnames,
 367                     const char *output_setname,
 368                     Z_SortKeySpecList *sort_sequence, int *sort_status);
 369 void resultSetSortSingle (ZebraHandle zh, NMEM nmem,
 370                           ZebraSet sset, RSET rset,
 371                           Z_SortKeySpecList *sort_sequence, int *sort_status);
 372 void resultSetRank (ZebraHandle zh, ZebraSet zebraSet, RSET rset, NMEM nmem);
 373 void resultSetInvalidate (ZebraHandle zh);
 374
 375 int zebra_server_lock_init (ZebraService zh);
 376 int zebra_server_lock_destroy (ZebraService zh);
 377 int zebra_server_lock (ZebraService zh, int lockCommit);
 378 void zebra_server_unlock (ZebraService zh, int commitPhase);
 379 int zebra_server_lock_get_state (ZebraService zh, time_t *timep);
 380
 381 typedef struct attent
 382 {
 383     int attset_ordinal;
 384     data1_local_attribute *local_attributes;
 385 } attent;
 386
 387 void zebraRankInstall (struct zebra_register *reg, struct rank_control *ctrl);
 388 ZebraRankClass zebraRankLookup (ZebraHandle zh, const char *name);
 389 void zebraRankDestroy (struct zebra_register *reg);
 390
 391 int att_getentbyatt(ZebraHandle zh, attent *res, oid_value set, int att,
 392                 const char *sattr);
 393
 394 extern struct rank_control *rank1_class;
 395 extern struct rank_control *rankzv_class;
 396 extern struct rank_control *rankliv_class;
 397
 398 int zebra_record_fetch (ZebraHandle zh, SYSNO sysno, int score, ODR stream,
 399                         oid_value input_format, Z_RecordComposition *comp,
 400                         oid_value *output_format, char **rec_bufp,
 401                         int *rec_lenp, char **basenamep,
 402                         char **addinfo);
 403
 404 void extract_get_fname_tmp (ZebraHandle zh, char *fname, int no);
 405
 406 void zebra_index_merge (ZebraHandle zh);
 407
 408 int buffer_extract_record (ZebraHandle zh,
 409                            const char *buf, size_t buf_size,
 410                            int delete_flag,
 411                            int test_mode,
 412                            const char *recordType,
 413                            SYSNO *sysno,
 414                            const char *match_criteria,
 415                            const char *fname,
 416                            int force_update,
 417                            int allow_update);
 418
 419 #if 0
 420 int extract_rec_in_mem (ZebraHandle zh, const char *recordType,
 421                         const char *buf, size_t buf_size,
 422                         const char *databaseName, int delete_flag,
 423                         int test_mode, int *sysno,
 424                         int store_keys, int store_data,
 425                         const char *match_criteria);
 426 #endif
 427 void extract_flushWriteKeys (ZebraHandle zh, int final);
 428
 429 struct zebra_fetch_control {
 430     off_t offset_end;
 431     off_t record_offset;
 432     off_t record_int_pos;
 433     const char *record_int_buf;
 434     int record_int_len;
 435     int fd;
 436 };
 437
 438 int zebra_record_ext_read (void *fh, char *buf, size_t count);
 439 off_t zebra_record_ext_seek (void *fh, off_t offset);
 440 off_t zebra_record_ext_tell (void *fh);
 441 off_t zebra_record_int_seek (void *fh, off_t offset);
 442 off_t zebra_record_int_tell (void *fh);
 443 int zebra_record_int_read (void *fh, char *buf, size_t count);
 444 void zebra_record_int_end (void *fh, off_t offset);
 445
 446 void extract_flushRecordKeys (ZebraHandle zh, SYSNO sysno,
 447                               int cmd, struct recKeys *reckeys);
 448 void extract_flushSortKeys (ZebraHandle zh, SYSNO sysno,
 449                             int cmd, struct sortKeys *skp);
 450 void extract_schema_add (struct recExtractCtrl *p, Odr_oid *oid);
 451 void extract_token_add (RecWord *p);
 452 int explain_extract (void *handle, Record rec, data1_node *n);
 453
 454 int fileExtract (ZebraHandle zh, SYSNO *sysno, const char *fname,
 455                  int deleteFlag);
 456
 457 int zebra_begin_read (ZebraHandle zh);
 458 int zebra_end_read (ZebraHandle zh);
 459
 460 int zebra_file_stat (const char *file_name, struct stat *buf,
 461                      int follow_links);
 462
 463 void zebra_livcode_transform(ZebraHandle zh, Z_RPNQuery *query);
 464
 465 void *iscz1_start ();
 466 void iscz1_reset (void *vp);
 467 void iscz1_stop (void *p);
 468 void iscz1_decode (void *vp, char **dst, const char **src);
 469 void iscz1_encode (void *vp, char **dst, const char **src);
 470
 471 Dict dict_open_res (BFiles bfs, const char *name, int cache, int rw,
 472                     int compact_flag, Res res);
 473
 474 YAZ_END_CDECL
 475
 476 #endif