C++ compilation.
[idzebra-moved-to-github.git] / isam / memory.c
index ba39802..d6e3a62 100644 (file)
@@ -1,10 +1,57 @@
 /*
- * Copyright (C) 1994, Index Data I/S 
+ * Copyright (C) 1994-1999, Index Data
  * All rights reserved.
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: memory.c,v $
- * Revision 1.2  1994-09-26 17:06:35  quinn
+ * Revision 1.17  1999-05-26 07:49:14  adam
+ * C++ compilation.
+ *
+ * Revision 1.16  1999/02/02 14:51:20  adam
+ * Updated WIN32 code specific sections. Changed header.
+ *
+ * Revision 1.15  1997/09/09 13:38:11  adam
+ * Partial port to WIN95/NT.
+ *
+ * Revision 1.14  1996/10/29 13:56:56  adam
+ * Include of zebrautl.h instead of alexutil.h.
+ *
+ * Revision 1.13  1996/03/20 13:29:16  quinn
+ * Bug-fix
+ *
+ * Revision 1.12  1996/03/11  14:52:23  quinn
+ * Fixed update bug. Repeated insertion in the same area sometimes caused
+ * problems.
+ *
+ * Revision 1.11  1996/02/10  12:20:58  quinn
+ * *** empty log message ***
+ *
+ * Revision 1.10  1995/12/12  14:12:47  quinn
+ * *** empty log message ***
+ *
+ * Revision 1.9  1995/12/06  15:48:46  quinn
+ * Fixed update-problem.
+ *
+ * Revision 1.8  1995/12/06  14:48:27  quinn
+ * Fixed some strange bugs.
+ *
+ * Revision 1.7  1995/12/06  09:59:46  quinn
+ * Fixed memory-consumption bug in memory.c
+ * Added more blocksizes to the default ISAM configuration.
+ *
+ * Revision 1.6  1995/09/04  12:33:47  adam
+ * Various cleanup. YAZ util used instead.
+ *
+ * Revision 1.5  1994/09/28  16:58:33  quinn
+ * Small mod.
+ *
+ * Revision 1.4  1994/09/27  20:03:52  quinn
+ * Seems relatively bug-free.
+ *
+ * Revision 1.3  1994/09/26  17:11:30  quinn
+ * Trivial
+ *
+ * Revision 1.2  1994/09/26  17:06:35  quinn
  * Back again...
  *
  * Revision 1.1  1994/09/26  16:07:56  quinn
  */
 
 #include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
 
-#include <util.h>
-#include <ismemory.h>
-#include "physical.h"
+#include <zebrautl.h>
 #include <isam.h>
 
 int is_mbuf_size[3] = { 0, 1024, 4096 };
 
-/*
- * TODO: make internal memory-management scheme for these units.
- */
+static is_mblock *mblock_tmplist = 0, *mblock_freelist = 0;
+static is_mbuf *mbuf_freelist[3] = {0, 0, 0};
+
+#define MALLOC_CHUNK 20
 
 is_mblock *xmalloc_mblock()
 {
     is_mblock *tmp;
+    int i;
 
-    tmp = xmalloc(sizeof(is_mblock));
+    if (!mblock_freelist)
+    {
+       mblock_freelist = (is_mblock *)
+           xmalloc(sizeof(is_mblock) * MALLOC_CHUNK);
+       for (i = 0; i < MALLOC_CHUNK - 1; i++)
+           mblock_freelist[i].next = &mblock_freelist[i+1];
+       mblock_freelist[i].next = 0;
+    }
+    tmp = mblock_freelist;
+    mblock_freelist = mblock_freelist->next;
     tmp->next = 0;
+    tmp->state = IS_MBSTATE_UNREAD;
+    tmp->data = 0;
     return tmp;
 }
 
@@ -42,8 +103,16 @@ is_mbuf *xmalloc_mbuf(int type)
 {
     is_mbuf *tmp;
 
-    tmp = xmalloc(sizeof(is_mbuf) + is_mbuf_size[type]);
-    tmp->type = type;
+    if (mbuf_freelist[type])
+    {
+       tmp = mbuf_freelist[type];
+       mbuf_freelist[type] = tmp->next;
+    }
+    else
+    {
+       tmp = (is_mbuf*) xmalloc(sizeof(is_mbuf) + is_mbuf_size[type]);
+       tmp->type = type;
+    }
     tmp->refcount = type ? 1 : 0;
     tmp->offset = tmp->num = tmp->cur_record = 0;
     tmp->data = (char*) tmp + sizeof(is_mbuf);
@@ -51,19 +120,48 @@ is_mbuf *xmalloc_mbuf(int type)
     return tmp;
 }
 
-#if 0
+void xfree_mbuf(is_mbuf *p)
+{
+    p->next = mbuf_freelist[p->type];
+    mbuf_freelist[p->type] = p;
+}
+
+void xfree_mbufs(is_mbuf *l)
+{
+    is_mbuf *p;
+
+    while (l)
+    {
+       p = l->next;
+       xfree_mbuf(l);
+       l = p;
+    }
+}
+
 void xfree_mblock(is_mblock *p)
 {
-    xfree(p);
+    xfree_mbufs(p->data);
+    p->next = mblock_freelist;
+    mblock_freelist = p;
+}
+
+void xrelease_mblock(is_mblock *p)
+{
+    p->next = mblock_tmplist;
+    mblock_tmplist = p;
 }
-#endif
 
-#if 0
-void xfree_mbuf(is_mblock *p)
+void xfree_mblocks(is_mblock *l)
 {
-    xfree(p);
+    is_mblock *p;
+
+    while (l)
+    {
+       p = l->next;
+       xfree_mblock(l);
+       l = p;
+    }
 }
-#endif
 
 void is_m_establish_tab(ISAM is, is_mtable *tab, ISAM_P pos)
 {
@@ -78,6 +176,7 @@ void is_m_establish_tab(ISAM is, is_mtable *tab, ISAM_P pos)
        tab->data->data = 0;
        tab->cur_mblock = tab->data;
        tab->cur_mblock->cur_mbuf = 0;
+       tab->last_mbuf = 0;
     }
     else /* new block */
     {
@@ -90,10 +189,18 @@ void is_m_establish_tab(ISAM is, is_mtable *tab, ISAM_P pos)
        tab->cur_mblock = tab->data;
        tab->cur_mblock->cur_mbuf = tab->data->data;
        tab->cur_mblock->cur_mbuf->cur_record = 0;
+       tab->last_mbuf = 0;
     }
     tab->is = is;
 }
 
+void is_m_release_tab(is_mtable *tab)
+{
+    xfree_mblocks(tab->data);
+    xfree_mblocks(mblock_tmplist);
+    mblock_tmplist = 0;
+}
+
 void is_m_rewind(is_mtable *tab)
 {
     tab->cur_mblock = tab->data;
@@ -145,7 +252,7 @@ void is_m_replace_record(is_mtable *tab, const void *rec)
  */
 void is_m_delete_record(is_mtable *tab)
 {
-    is_mbuf *mbuf, *new;
+    is_mbuf *mbuf, *inew;
 
     mbuf = tab->cur_mblock->cur_mbuf;
     if (mbuf->cur_record >= mbuf->num)  /* top of mbuf */
@@ -153,18 +260,28 @@ void is_m_delete_record(is_mtable *tab)
        mbuf->num--;
        mbuf->cur_record--;
     }
-    else /* middle of a block */
+    else if (mbuf->cur_record == 1) /* beginning of mbuf */
+    {
+       mbuf->num--;
+       mbuf->offset +=is_keysize(tab->is);
+       mbuf->cur_record = 0;
+    }
+    else /* middle of mbuf */
     {
-       new = xmalloc_mbuf(IS_MBUF_TYPE_SMALL);
-       new->next = mbuf->next;
-       mbuf->next = new;
-       new->data = mbuf->data;
+       /* insert block after current one */
+       inew = xmalloc_mbuf(IS_MBUF_TYPE_SMALL);
+       inew->next = mbuf->next;
+       mbuf->next = inew;
+
+       /* virtually transfer everything after current record to new one. */
+       inew->data = mbuf->data;
        mbuf->refcount++;
-       new->offset = mbuf->offset + mbuf->cur_record * is_keysize(tab->is);
-       new->num = mbuf->num - mbuf->cur_record;
+       inew->offset = mbuf->offset + mbuf->cur_record * is_keysize(tab->is);
+       inew->num = mbuf->num - mbuf->cur_record;
+       
+       /* old buf now only contains stuff before current record */
        mbuf->num = mbuf->cur_record -1;
-       mbuf = mbuf->next;
-       mbuf->cur_record = 0;
+       tab->cur_mblock->cur_mbuf = inew;
     }
     tab->num_records--;
     tab->cur_mblock->num_records--;
@@ -208,7 +325,9 @@ int is_m_write_record(is_mtable *tab, const void *rec)
        mbuf = tab->cur_mblock->cur_mbuf = mbuf->next;
        mbuf->cur_record = 0;
     }
-    log(LOG_DEBUG, "is_m_write_rec(rec == %d)", mbuf->cur_record);
+    /*
+    logf (LOG_DEBUG, "is_m_write_rec(rec == %d)", mbuf->cur_record);
+    */
     memcpy(mbuf->data + mbuf->offset + mbuf->cur_record * is_keysize(tab->is),
        rec, is_keysize(tab->is));
     mbuf->num++;
@@ -222,7 +341,10 @@ int is_m_write_record(is_mtable *tab, const void *rec)
 void is_m_unread_record(is_mtable *tab)
 {
     assert(tab->cur_mblock->cur_mbuf->cur_record);
-    tab->cur_mblock->cur_mbuf->cur_record--;
+    if (tab->last_mbuf)
+       tab->cur_mblock->cur_mbuf = tab->last_mbuf;
+    else
+       tab->cur_mblock->cur_mbuf->cur_record--;
 }
 
 /*
@@ -231,23 +353,25 @@ void is_m_unread_record(is_mtable *tab)
 int is_m_peek_record(is_mtable *tab, void *rec)
 {
     is_mbuf *mbuf;
+    is_mblock *mblock;
 
     /* make sure block is all in memory */
     if (tab->cur_mblock->state <= IS_MBSTATE_PARTIAL)
        if (read_current_full(tab, tab->cur_mblock) < 0)
            return -1;
-    mbuf = tab->cur_mblock->cur_mbuf;
+    mblock = tab->cur_mblock;
+    mbuf = mblock->cur_mbuf;
     if (mbuf->cur_record >= mbuf->num) /* are we at end of mbuf? */
     {
        if (!mbuf->next) /* end of mblock */
        {
-           if (tab->cur_mblock->next)
+           if (mblock->next)
            {
-               tab->cur_mblock = tab->cur_mblock->next;
-               if (tab->cur_mblock->next->state <= IS_MBSTATE_PARTIAL)
-                   if (read_current_full(tab, tab->cur_mblock->next) < 0)
+               mblock = mblock->next;
+               if (mblock->state <= IS_MBSTATE_PARTIAL)
+                   if (read_current_full(tab, mblock) < 0)
                        return -1;
-               mbuf = tab->cur_mblock->next->data;
+               mbuf = mblock->data;
            }
            else
                return 0;   /* EOTable */
@@ -261,7 +385,7 @@ int is_m_peek_record(is_mtable *tab, void *rec)
     return 1;
 }
 
-int is_m_read_record(is_mtable *tab, void *buf)
+int is_m_read_record(is_mtable *tab, void *buf, int keep)
 {
     is_mbuf *mbuf;
 
@@ -274,6 +398,13 @@ int is_m_read_record(is_mtable *tab, void *buf)
     {
        if (!mbuf->next) /* end of mblock */
        {
+           if (!keep && tab->cur_mblock->state == IS_MBSTATE_CLEAN &&
+               tab->cur_mblock->diskpos > 0)
+           {
+               xfree_mbufs(tab->cur_mblock->data);
+               tab->cur_mblock->data = 0;
+               tab->cur_mblock->state = IS_MBSTATE_UNREAD;
+           }
            if (tab->cur_mblock->next)
            {
                tab->cur_mblock = tab->cur_mblock->next;
@@ -281,14 +412,20 @@ int is_m_read_record(is_mtable *tab, void *buf)
                    if (read_current_full(tab, tab->cur_mblock) < 0)
                        return -1;
                tab->cur_mblock->cur_mbuf = mbuf = tab->cur_mblock->data;
+               tab->last_mbuf = 0;
            }
            else
                return 0;   /* EOTable */
        }
        else
+       {
+           tab->last_mbuf = mbuf;
            tab->cur_mblock->cur_mbuf = mbuf = mbuf->next;
+       }
        mbuf->cur_record = 0;
     }
+    else
+       tab->last_mbuf = 0;
     memcpy(buf, mbuf->data + mbuf->offset + mbuf->cur_record *
        is_keysize(tab->is), is_keysize(tab->is));
     mbuf->cur_record++;
@@ -305,14 +442,25 @@ int is_m_seek_record(is_mtable *tab, const void *rec)
 
     for (;;)
     {
-       if (is_m_read_record(tab, &peek) <= 0)
+       if (is_m_read_record(tab, &peek, 1) <= 0)
            return 1;
        if ((rs = (*tab->is->cmp)(peek, rec)) > 0)
        {
            is_m_unread_record(tab);
-           return 1;
+           return rs;
        }
        else if (rs == 0)
            return 0;
     }
 }
+
+int is_m_num_records(is_mtable *tab)
+{
+    if (tab->data->state < IS_MBSTATE_PARTIAL)
+       if (read_current_full(tab, tab->data) < 0)
+       {
+           logf (LOG_FATAL, "read full failed");
+           exit(1);
+       }
+    return tab->num_records;
+}