Updated MSVC project files.
[idzebra-moved-to-github.git] / index / recindex.c
index 893fa05..2fe16b8 100644 (file)
@@ -1,10 +1,61 @@
 /*
- * Copyright (C) 1994-1995, Index Data I/S 
+ * Copyright (C) 1994-1999, Index Data
  * All rights reserved.
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: recindex.c,v $
- * Revision 1.11  1995-12-06 13:58:26  adam
+ * Revision 1.24  1999-06-25 13:48:02  adam
+ * Updated MSVC project files.
+ * Added BZIP2 record compression (not very well tested).
+ *
+ * Revision 1.23  1999/05/26 07:49:13  adam
+ * C++ compilation.
+ *
+ * Revision 1.22  1999/02/18 12:49:34  adam
+ * Changed file naming scheme for register files as well as record
+ * store/index files.
+ *
+ * Revision 1.21  1999/02/02 14:51:03  adam
+ * Updated WIN32 code specific sections. Changed header.
+ *
+ * Revision 1.20  1998/01/12 15:04:08  adam
+ * The test option (-s) only uses read-lock (and not write lock).
+ *
+ * Revision 1.19  1997/09/17 12:19:16  adam
+ * Zebra version corresponds to YAZ version 1.4.
+ * Changed Zebra server so that it doesn't depend on global common_resource.
+ *
+ * Revision 1.18  1997/07/15 16:28:42  adam
+ * Bug fix: storeData didn't work with files with multiple records.
+ * Bug fix: fixed memory management with records; not really well
+ *  thought through.
+ *
+ * Revision 1.17  1997/02/12 20:39:46  adam
+ * Implemented options -f <n> that limits the log to the first <n>
+ * records.
+ * Changed some log messages also.
+ *
+ * Revision 1.16  1996/06/04 10:19:00  adam
+ * Minor changes - removed include of ctype.h.
+ *
+ * Revision 1.15  1996/05/13  14:23:06  adam
+ * Work on compaction of set/use bytes in dictionary.
+ *
+ * Revision 1.14  1996/02/01  20:48:15  adam
+ * The total size of records are always checked in rec_cache_insert to
+ * reduce memory usage.
+ *
+ * Revision 1.13  1995/12/11  09:12:49  adam
+ * The rec_get function returns NULL if record doesn't exist - will
+ * happen in the server if the result set records have been deleted since
+ * the creation of the set (i.e. the search).
+ * The server saves a result temporarily if it is 'volatile', i.e. the
+ * set is register dependent.
+ *
+ * Revision 1.12  1995/12/07  17:38:47  adam
+ * Work locking mechanisms for concurrent updates/commit.
+ *
+ * Revision 1.11  1995/12/06  13:58:26  adam
  * Improved flushing of records - all flushes except the last one
  * don't write the last accessed. Also flush takes place if record
  * info occupy more than about 256k.
  * Started work on better record management system.
  *
  */
+
+
+/*
+ *  Format of first block
+ *      next       (4 bytes)
+ *      ref_count  (4 bytes)
+ *      block      (504 bytes)
+ *
+ *  Format of subsequent blocks 
+ *      next  (4 bytes)
+ *      block (508 bytes)
+ *
+ *  Format of each record
+ *      sysno
+ *      (length, data) - pairs
+ *      length = 0 if same as previous
+ */
 #include <stdio.h>
 #include <assert.h>
 #include <string.h>
-#include <ctype.h>
 
 #include "recindxp.h"
 
+#if HAVE_BZLIB_H
+#include <bzlib.h>
+#endif
 static void rec_write_head (Records p)
 {
     int r;
@@ -72,14 +142,14 @@ static void rec_write_head (Records p)
     }
 }
 
-static void rec_tmp_expand (Records p, int size, int dst_type)
+static void rec_tmp_expand (Records p, int size)
 {
-    if (p->tmp_size < size + 256 ||
-        p->tmp_size < p->head.block_size[dst_type]*2)
+    if (p->tmp_size < size + 2048 ||
+        p->tmp_size < p->head.block_size[REC_BLOCK_TYPES-1]*2)
     {
         xfree (p->tmp_buf);
-        p->tmp_size = size + p->head.block_size[dst_type]*2 + 256;
-        p->tmp_buf = xmalloc (p->tmp_size);
+        p->tmp_size = size + p->head.block_size[REC_BLOCK_TYPES-1]*2 + 2048;
+        p->tmp_buf = (char *) xmalloc (p->tmp_size);
     }
 }
 
@@ -94,7 +164,6 @@ static int read_indx (Records p, int sysno, void *buf, int itemsize,
     {
         logf (LOG_FATAL|LOG_ERRNO, "read in %s at pos %ld",
               p->index_fname, (long) pos);
-        abort ();
         exit (1);
     }
     return r;
@@ -110,35 +179,56 @@ static void write_indx (Records p, int sysno, void *buf, int itemsize)
 static void rec_release_blocks (Records p, int sysno)
 {
     struct record_index_entry entry;
-    int freeblock, freenext;
+    int freeblock;
+    int block_and_ref[2];
     int dst_type;
+    int first = 1;
 
+    logf (LOG_LOG, "release_blocks for sysno=%d", sysno);
     if (read_indx (p, sysno, &entry, sizeof(entry), 1) != 1)
         return ;
-    p->head.total_bytes -= entry.u.used.size;
-    freeblock = entry.u.used.next;
+
+    freeblock = entry.next;
     assert (freeblock > 0);
     dst_type = freeblock & 7;
     assert (dst_type < REC_BLOCK_TYPES);
     freeblock = freeblock / 8;
     while (freeblock)
     {
-        if (bf_read (p->data_BFile[dst_type], freeblock, 0, sizeof(freenext),
-                     &freenext) != 1)
+        if (bf_read (p->data_BFile[dst_type], freeblock, 0,
+                    sizeof(block_and_ref), block_and_ref) != 1)
         {
             logf (LOG_FATAL|LOG_ERRNO, "read in rec_del_single");
             exit (1);
         }
-        if (bf_write (p->data_BFile[dst_type], freeblock, 0, sizeof(freenext),
+       if (first)
+       {
+           block_and_ref[1]--;
+           if (block_and_ref[1])
+           {
+               if (bf_write (p->data_BFile[dst_type], freeblock, 0,
+                             sizeof(block_and_ref), block_and_ref))
+               {
+                   logf (LOG_FATAL|LOG_ERRNO, "write in rec_del_single");
+                   exit (1);
+               }
+               return;
+           }
+           first = 0;
+       }
+       
+        if (bf_write (p->data_BFile[dst_type], freeblock, 0, sizeof(freeblock),
                       &p->head.block_free[dst_type]))
         {
             logf (LOG_FATAL|LOG_ERRNO, "write in rec_del_single");
             exit (1);
         }
         p->head.block_free[dst_type] = freeblock;
-        freeblock = freenext;
+        freeblock = block_and_ref[0];
+
         p->head.block_used[dst_type]--;
     }
+    p->head.total_bytes -= entry.size;
 }
 
 static void rec_delete_single (Records p, Record rec)
@@ -147,32 +237,29 @@ static void rec_delete_single (Records p, Record rec)
 
     rec_release_blocks (p, rec->sysno);
 
-    entry.u.free.next = p->head.index_free;
+    entry.next = p->head.index_free;
+    entry.size = 0;
     p->head.index_free = rec->sysno;
     write_indx (p, rec->sysno, &entry, sizeof(entry));
 }
 
+static void rec_write_tmp_buf (Records p, int size, int *sysnos);
 
 static void rec_write_single (Records p, Record rec)
 {
+
+    int sysnos[2];
     int i, size = 0;
     char *cptr;
-    int dst_type = 0;
-    int no_written = 0;
-    int block_prev = -1, block_free;
-    struct record_index_entry entry;
 
+    logf (LOG_LOG, " rec_write_single !!!!!!!!!!!!!!!!!!!!!!!!!!!!");
     for (i = 0; i < REC_NO_INFO; i++)
         if (!rec->info[i])
             size += sizeof(*rec->size);
         else
             size += sizeof(*rec->size) + rec->size[i];
 
-    for (i = 1; i<REC_BLOCK_TYPES; i++)
-        if (size >= p->head.block_move[i])
-            dst_type = i;
-
-    rec_tmp_expand (p, size, dst_type);
+    rec_tmp_expand (p, size);
 
     cptr = p->tmp_buf + sizeof(int);           /* a hack! */
     for (i = 0; i < REC_NO_INFO; i++)
@@ -185,7 +272,23 @@ static void rec_write_single (Records p, Record rec)
             cptr += rec->size[i];
         }
     }
-    cptr = p->tmp_buf;
+    sysnos[0] = rec->sysno;
+    sysnos[1] = -1;
+    rec_write_tmp_buf (p, size, sysnos);
+}
+
+static void rec_write_tmp_buf (Records p, int size, int *sysnos)
+{
+    struct record_index_entry entry;
+    int no_written = 0;
+    char *cptr = p->tmp_buf;
+    int block_prev = -1, block_free;
+    int dst_type = 0;
+    int i;
+
+    for (i = 1; i<REC_BLOCK_TYPES; i++)
+        if (size >= p->head.block_move[i])
+            dst_type = i;
     while (no_written < size)
     {
         block_free = p->head.block_free[dst_type];
@@ -197,16 +300,21 @@ static void rec_write_single (Records p, Record rec)
             {
                 logf (LOG_FATAL|LOG_ERRNO, "read in %s at free block %d",
                       p->data_fname[dst_type], block_free);
+               exit (1);
             }
         }
         else
             block_free = p->head.block_last[dst_type]++;
         if (block_prev == -1)
         {
-            entry.u.used.next = block_free*8 + dst_type;
-            entry.u.used.size = size;
+            entry.next = block_free*8 + dst_type;
+            entry.size = size;
             p->head.total_bytes += size;
-            write_indx (p, rec->sysno, &entry, sizeof(entry));
+           while (*sysnos > 0)
+           {
+               write_indx (p, *sysnos, &entry, sizeof(entry));
+               sysnos++;
+           }
         }
         else
         {
@@ -231,17 +339,17 @@ static void rec_update_single (Records p, Record rec)
     rec_write_single (p, rec);
 }
 
-Records rec_open (int rw)
+Records rec_open (BFiles bfs, int rw)
 {
     Records p;
     int i, r;
 
-    p = xmalloc (sizeof(*p));
+    p = (Records) xmalloc (sizeof(*p));
     p->rw = rw;
     p->tmp_size = 1024;
-    p->tmp_buf = xmalloc (p->tmp_size);
-    p->index_fname = "recindex";
-    p->index_BFile = bf_open (p->index_fname, 128, rw);
+    p->tmp_buf = (char *) xmalloc (p->tmp_size);
+    p->index_fname = "reci";
+    p->index_BFile = bf_open (bfs, p->index_fname, 128, rw);
     if (p->index_BFile == NULL)
     {
         logf (LOG_FATAL|LOG_ERRNO, "open %s", p->index_fname);
@@ -284,14 +392,14 @@ Records rec_open (int rw)
     for (i = 0; i<REC_BLOCK_TYPES; i++)
     {
         char str[80];
-        sprintf (str, "recdata%c", i + 'A');
-        p->data_fname[i] = xmalloc (strlen(str)+1);
+        sprintf (str, "recd%c", i + 'A');
+        p->data_fname[i] = (char *) xmalloc (strlen(str)+1);
         strcpy (p->data_fname[i], str);
         p->data_BFile[i] = NULL;
     }
     for (i = 0; i<REC_BLOCK_TYPES; i++)
     {
-        if (!(p->data_BFile[i] = bf_open (p->data_fname[i],
+        if (!(p->data_BFile[i] = bf_open (bfs, p->data_fname[i],
                                           p->head.block_size[i],
                                           rw)))
         {
@@ -299,18 +407,158 @@ Records rec_open (int rw)
             exit (1);
         }
     }
-    p->cache_max = 10;
+    p->cache_max = 400;
     p->cache_cur = 0;
-    p->record_cache = xmalloc (sizeof(*p->record_cache)*p->cache_max);
+    p->record_cache = (struct record_cache_entry *)
+       xmalloc (sizeof(*p->record_cache)*p->cache_max);
     return p;
 }
 
+static void rec_encode_unsigned (unsigned n, unsigned char *buf, int *len)
+{
+    (*len) = 0;
+    while (n > 127)
+    {
+       buf[*len] = 128 + (n & 127);
+       n = n >> 7;
+       (*len)++;
+    }
+    buf[*len] = n;
+    (*len)++;
+}
+
+static void rec_decode_unsigned(unsigned *np, unsigned char *buf, int *len)
+{
+    unsigned n = 0;
+    unsigned w = 1;
+    (*len) = 0;
+
+    while (buf[*len] > 127)
+    {
+       n += w*(buf[*len] & 127);
+       w = w << 7;
+       (*len)++;
+    }
+    n += w * buf[*len];
+    (*len)++;
+    *np = n;
+}
+static void rec_cache_flush_block1 (Records p, Record rec, Record last_rec,
+                                   char **out_buf, int *out_size,
+                                   int *out_offset)
+{
+    int i;
+    int len;
+
+    for (i = 0; i<REC_NO_INFO; i++)
+    {
+       if (*out_offset + (int) rec->size[i] + 20 > *out_size)
+       {
+           int new_size = *out_offset + rec->size[i] + 65536;
+           char *np = (char *) xmalloc (new_size);
+           if (*out_offset)
+               memcpy (np, *out_buf, *out_offset);
+           xfree (*out_buf);
+           *out_size = new_size;
+           *out_buf = np;
+       }
+       if (i == 0)
+       {
+           rec_encode_unsigned (rec->sysno, *out_buf + *out_offset, &len);
+           (*out_offset) += len;
+       }
+       if (rec->size[i] == 0)
+       {
+           rec_encode_unsigned (1, *out_buf + *out_offset, &len);
+           (*out_offset) += len;
+       }
+       else if (last_rec && rec->size[i] == last_rec->size[i] &&
+                !memcmp (rec->info[i], last_rec->info[i], rec->size[i]))
+       {
+           rec_encode_unsigned (0, *out_buf + *out_offset, &len);
+           (*out_offset) += len;
+       }
+       else
+       {
+           rec_encode_unsigned (rec->size[i]+1, *out_buf + *out_offset, &len);
+           (*out_offset) += len;
+           memcpy (*out_buf + *out_offset, rec->info[i], rec->size[i]);
+           (*out_offset) += rec->size[i];
+       }
+    }
+}
+
+static void rec_write_multiple (Records p, int saveCount)
+{
+    int i;
+    int ref_count = 0;
+    Record last_rec = 0;
+    int out_size = 1000;
+    int out_offset = 0;
+    char *out_buf = (char *) xmalloc (out_size);
+    int *sysnos = (int *) xmalloc (sizeof(*sysnos) * (p->cache_cur + 1));
+    int *sysnop = sysnos;
+
+    for (i = 0; i<p->cache_cur - saveCount; i++)
+    {
+        struct record_cache_entry *e = p->record_cache + i;
+        switch (e->flag)
+        {
+        case recordFlagNew:
+            rec_cache_flush_block1 (p, e->rec, last_rec, &out_buf,
+                                   &out_size, &out_offset);
+           *sysnop++ = e->rec->sysno;
+           ref_count++;
+           e->flag = recordFlagNop;
+           last_rec = e->rec;
+            break;
+        case recordFlagWrite:
+           rec_release_blocks (p, e->rec->sysno);
+            rec_cache_flush_block1 (p, e->rec, last_rec, &out_buf,
+                                   &out_size, &out_offset);
+           *sysnop++ = e->rec->sysno;
+           ref_count++;
+           e->flag = recordFlagNop;
+           last_rec = e->rec;
+            break;
+       default:
+           break;
+        }
+    }
+    *sysnop = -1;
+    if (ref_count)
+    {
+       int csize = out_offset + (out_offset >> 6) + 620;
+
+       rec_tmp_expand (p, csize);
+#if HAVE_BZLIB_H       
+       i = bzBuffToBuffCompress (p->tmp_buf+2*sizeof(int), &csize,
+                                 out_buf, out_offset, 9, 0, 30);
+       if (i != BZ_OK)
+       {
+           logf (LOG_FATAL, "bzBuffToCompress error code=%d", i);
+           exit (1);
+       }
+#else
+       memcpy (p->tmp_buf + 2*sizeof(int), out_buf, out_offset);
+       csize = out_offset;
+#endif
+       memcpy (p->tmp_buf + sizeof(int), &ref_count, sizeof(ref_count));
+       /* -------- compression */
+       rec_write_tmp_buf (p, csize + sizeof(int), sysnos);
+    }
+    xfree (out_buf);
+    xfree (sysnos);
+}
+
 static void rec_cache_flush (Records p, int saveCount)
 {
     int i, j;
 
     if (saveCount >= p->cache_cur)
         saveCount = 0;
+
+    rec_write_multiple (p, saveCount);
     for (i = 0; i<p->cache_cur - saveCount; i++)
     {
         struct record_cache_entry *e = p->record_cache + i;
@@ -359,7 +607,7 @@ static void rec_cache_insert (Records p, Record rec, enum recordCacheFlag flag)
 
     if (p->cache_cur == p->cache_max)
         rec_cache_flush (p, 1);
-    else if (p->cache_cur > 2)
+    else if (p->cache_cur > 0)
     {
         int i, j;
         int used = 0;
@@ -409,11 +657,12 @@ void rec_close (Records *pp)
 
 Record rec_get (Records p, int sysno)
 {
-    int i;
+    int i, in_size;
     Record rec, *recp;
     struct record_index_entry entry;
     int freeblock, dst_type;
     char *nptr, *cptr;
+    char *in_buf = 0;
 
     assert (sysno > 0);
     assert (p);
@@ -421,16 +670,20 @@ Record rec_get (Records p, int sysno)
     if ((recp = rec_cache_lookup (p, sysno, recordFlagNop)))
         return rec_cp (*recp);
 
-    read_indx (p, sysno, &entry, sizeof(entry), 0);
+    if (!read_indx (p, sysno, &entry, sizeof(entry), 1))
+        return NULL;       /* record is not there! */
+
+    if (!entry.size)
+        return NULL;       /* record is deleted */
 
-    dst_type = entry.u.used.next & 7;
+    dst_type = entry.next & 7;
     assert (dst_type < REC_BLOCK_TYPES);
-    freeblock = entry.u.used.next / 8;
+    freeblock = entry.next / 8;
 
     assert (freeblock > 0);
     
-    rec = xmalloc (sizeof(*rec));
-    rec_tmp_expand (p, entry.u.used.size, dst_type);
+    rec = (Record) xmalloc (sizeof(*rec));
+    rec_tmp_expand (p, entry.size);
 
     cptr = p->tmp_buf;
     bf_read (p->data_BFile[dst_type], freeblock, 0, 0, cptr);
@@ -449,20 +702,51 @@ Record rec_get (Records p, int sysno)
     }
 
     rec->sysno = sysno;
-    nptr = p->tmp_buf + sizeof(freeblock);
-    for (i = 0; i < REC_NO_INFO; i++)
+#if HAVE_BZLIB_H
+    in_size = entry.size * 30+100;
+    in_buf = (char *) xmalloc (in_size);
+    i = bzBuffToBuffDecompress (in_buf, &in_size, p->tmp_buf+2*sizeof(int),
+                               entry.size-sizeof(int), 0, 4);
+    if (i != BZ_OK)
     {
-        memcpy (&rec->size[i], nptr, sizeof(*rec->size));
-        nptr += sizeof(*rec->size);
-        if (rec->size[i])
-        {
-            rec->info[i] = xmalloc (rec->size[i]);
-            memcpy (rec->info[i], nptr, rec->size[i]);
-            nptr += rec->size[i];
-        }
-        else
-            rec->info[i] = NULL;
+       logf (LOG_FATAL, "bzBuffToDecompress error code=%d", i);
+       exit (1);
+    }
+#else
+    in_buf = p->tmp_buf + 2*sizeof(int);
+    in_size = entry.size - sizeof(int);
+#endif
+    nptr = in_buf;                /* skip ref count */
+    while (nptr < in_buf + in_size)
+    {
+       int this_sysno;
+       int len;
+       rec_decode_unsigned (&this_sysno, nptr, &len);
+       nptr += len;
+
+       for (i = 0; i < REC_NO_INFO; i++)
+       {
+           int this_size;
+           rec_decode_unsigned (&this_size, nptr, &len);
+           nptr += len;
+
+           if (this_size == 0)
+               continue;
+           rec->size[i] = this_size-1;
+
+           if (rec->size[i])
+           {
+               rec->info[i] = (char *) xmalloc (rec->size[i]);
+               memcpy (rec->info[i], nptr, rec->size[i]);
+               nptr += rec->size[i];
+           }
+           else
+               rec->info[i] = NULL;
+       }
+       if (this_sysno == sysno)
+           break;
     }
+    xfree (in_buf);
     rec_cache_insert (p, rec, recordFlagNop);
     return rec;
 }
@@ -473,8 +757,8 @@ Record rec_new (Records p)
     Record rec;
 
     assert (p);
-    rec = xmalloc (sizeof(*rec));
-    if (p->head.index_free == 0)
+    rec = (Record) xmalloc (sizeof(*rec));
+    if (1 || p->head.index_free == 0)
         sysno = (p->head.index_last)++;
     else
     {
@@ -482,7 +766,7 @@ Record rec_new (Records p)
 
         read_indx (p, p->head.index_free, &entry, sizeof(entry), 0);
         sysno = p->head.index_free;
-        p->head.index_free = entry.u.free.next;
+        p->head.index_free = entry.next;
     }
     (p->head.no_records)++;
     rec->sysno = sysno;
@@ -499,6 +783,7 @@ void rec_del (Records p, Record *recpp)
 {
     Record *recp;
 
+    (p->head.no_records)--;
     if ((recp = rec_cache_lookup (p, (*recpp)->sysno, recordFlagDelete)))
     {
         rec_rm (recp);
@@ -532,6 +817,9 @@ void rec_put (Records p, Record *recpp)
 void rec_rm (Record *recpp)
 {
     int i;
+
+    if (!*recpp)
+        return ;
     for (i = 0; i < REC_NO_INFO; i++)
         xfree ((*recpp)->info[i]);
     xfree (*recpp);
@@ -543,7 +831,7 @@ Record rec_cp (Record rec)
     Record n;
     int i;
 
-    n = xmalloc (sizeof(*n));
+    n = (Record) xmalloc (sizeof(*n));
     n->sysno = rec->sysno;
     for (i = 0; i < REC_NO_INFO; i++)
         if (!rec->info[i])
@@ -554,7 +842,7 @@ Record rec_cp (Record rec)
         else
         {
             n->size[i] = rec->size[i];
-            n->info[i] = xmalloc (rec->size[i]);
+            n->info[i] = (char *) xmalloc (rec->size[i]);
             memcpy (n->info[i], rec->info[i], rec->size[i]);
         }
     return n;
@@ -571,7 +859,7 @@ char *rec_strdup (const char *s, size_t *len)
         return NULL;
     }
     *len = strlen(s)+1;
-    p = xmalloc (*len);
+    p = (char *) xmalloc (*len);
     strcpy (p, s);
     return p;
 }