-/* $Id: reckeys.c,v 1.1 2005-10-28 09:22:50 adam Exp $
- Copyright (C) 1995-2005
+/* $Id: reckeys.c,v 1.6 2006-06-07 10:14:41 adam Exp $
+ Copyright (C) 1995-2006
Index Data ApS
This file is part of the Zebra server.
#include <assert.h>
#include <ctype.h>
+#include <yaz/nmem.h>
#include "index.h"
#include "reckeys.h"
+struct zebra_rec_key_entry {
+ char *buf;
+ size_t len;
+ struct it_key key;
+ struct zebra_rec_key_entry *next;
+};
+
struct zebra_rec_keys_t_ {
size_t buf_used;
size_t buf_max;
void *encode_handle;
void *decode_handle;
char owner_of_buffer;
+
+ NMEM nmem;
+ size_t hash_size;
+ struct zebra_rec_key_entry **entries;
};
+
+struct zebra_rec_key_entry **zebra_rec_keys_mk_hash(zebra_rec_keys_t p,
+ const char *buf,
+ size_t len,
+ const struct it_key *key)
+{
+ unsigned h = 0;
+ size_t i;
+ int j;
+ for (i = 0; i<len; i++)
+ h = h * 65509 + buf[i];
+ for (j = 0; j<key->len; j++)
+ h = h * 65509 + CAST_ZINT_TO_INT(key->mem[j]);
+ return &p->entries[h % (unsigned) p->hash_size];
+}
+
+static void init_hash(zebra_rec_keys_t p)
+{
+ p->entries = 0;
+ nmem_reset(p->nmem);
+ if (p->hash_size)
+ {
+ size_t i;
+ p->entries = nmem_malloc(p->nmem, p->hash_size * sizeof(*p->entries));
+ for (i = 0; i<p->hash_size; i++)
+ p->entries[i] = 0;
+ }
+}
+
zebra_rec_keys_t zebra_rec_keys_open()
{
zebra_rec_keys_t p = xmalloc(sizeof(*p));
p->owner_of_buffer = 1;
p->encode_handle = iscz1_start();
p->decode_handle = iscz1_start();
+
+ p->nmem = nmem_create();
+ p->hash_size = 1023;
+ p->entries = 0;
+
+ init_hash(p);
+
return p;
}
-
+
void zebra_rec_keys_set_buf(zebra_rec_keys_t p, char *buf, size_t sz,
- int owner)
+ int copy_buf)
{
if (p->owner_of_buffer)
xfree(p->buf);
p->buf_used = sz;
p->buf_max = sz;
- if (!owner)
+ if (!copy_buf)
{
p->buf = buf;
}
memcpy(p->buf, buf, sz);
}
}
- p->owner_of_buffer = owner;
+ p->owner_of_buffer = copy_buf;
}
void zebra_rec_keys_get_buf(zebra_rec_keys_t p, char **buf, size_t *sz)
iscz1_stop(p->encode_handle);
if (p->decode_handle)
iscz1_stop(p->decode_handle);
+ nmem_destroy(p->nmem);
xfree(p);
}
+int zebra_rec_keys_add_hash(zebra_rec_keys_t keys,
+ const char *str, size_t slen,
+ const struct it_key *key)
+{
+ struct zebra_rec_key_entry **kep = zebra_rec_keys_mk_hash(keys,
+ str, slen, key);
+ while (*kep)
+ {
+ struct zebra_rec_key_entry *e = *kep;
+ if (slen == e->len && !memcmp(str, e->buf, slen) &&
+ !key_compare(key, &e->key))
+ {
+ return 0;
+ }
+ kep = &(*kep)->next;
+ }
+ *kep = nmem_malloc(keys->nmem, sizeof(**kep));
+ (*kep)->next = 0;
+ (*kep)->len = slen;
+ memcpy(&(*kep)->key, key, sizeof(*key));
+ (*kep)->buf = nmem_malloc(keys->nmem, slen);
+ memcpy((*kep)->buf, str, slen);
+ return 1;
+}
+
void zebra_rec_keys_write(zebra_rec_keys_t keys,
- int reg_type,
const char *str, size_t slen,
const struct it_key *key)
{
assert(keys->owner_of_buffer);
+ if (!zebra_rec_keys_add_hash(keys, str, slen, key))
+ return; /* key already there . Omit it */
if (keys->buf_used+1024 > keys->buf_max)
{
char *b = (char *) xmalloc (keys->buf_max += 128000);
iscz1_encode(keys->encode_handle, &dst, &src);
-#if REG_TYPE_PREFIX
- *dst++ = reg_type;
-#endif
memcpy (dst, str, slen);
dst += slen;
*dst++ = '\0';
keys->buf_used = 0;
iscz1_reset(keys->encode_handle);
+
+ init_hash(keys);
+
}
int zebra_rec_keys_rewind(zebra_rec_keys_t keys)
}
return 1;
}
+/*
+ * Local variables:
+ * c-basic-offset: 4
+ * indent-tabs-mode: nil
+ * End:
+ * vim: shiftwidth=4 tabstop=8 expandtab
+ */
+