1 /* $Id: reckeys.c,v 1.9 2006-11-20 13:59:35 adam Exp $
2 Copyright (C) 1995-2006
5 This file is part of the Zebra server.
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
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
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
32 struct zebra_rec_key_entry {
36 struct zebra_rec_key_entry *next;
39 struct zebra_rec_keys_t_ {
50 struct zebra_rec_key_entry **entries;
54 struct zebra_rec_key_entry **zebra_rec_keys_mk_hash(zebra_rec_keys_t p,
57 const struct it_key *key)
63 h = key->mem[key->len-1];
65 for (i = 0; i<len; i++)
66 h = h * 65509 + buf[i];
67 for (j = 0; j<key->len; j++)
68 h = h * 65509 + CAST_ZINT_TO_INT(key->mem[j]);
70 return &p->entries[h % (unsigned) p->hash_size];
73 static void init_hash(zebra_rec_keys_t p)
80 p->entries = nmem_malloc(p->nmem, p->hash_size * sizeof(*p->entries));
81 for (i = 0; i<p->hash_size; i++)
86 zebra_rec_keys_t zebra_rec_keys_open(void)
88 zebra_rec_keys_t p = xmalloc(sizeof(*p));
93 p->owner_of_buffer = 1;
94 p->encode_handle = iscz1_start();
95 p->decode_handle = iscz1_start();
97 p->nmem = nmem_create();
106 void zebra_rec_keys_set_buf(zebra_rec_keys_t p, char *buf, size_t sz,
109 if (p->owner_of_buffer)
123 p->buf = xmalloc(sz);
124 memcpy(p->buf, buf, sz);
127 p->owner_of_buffer = copy_buf;
130 void zebra_rec_keys_get_buf(zebra_rec_keys_t p, char **buf, size_t *sz)
140 void zebra_rec_keys_close(zebra_rec_keys_t p)
145 if (p->owner_of_buffer)
147 if (p->encode_handle)
148 iscz1_stop(p->encode_handle);
149 if (p->decode_handle)
150 iscz1_stop(p->decode_handle);
151 nmem_destroy(p->nmem);
155 int zebra_rec_keys_add_hash(zebra_rec_keys_t keys,
156 const char *str, size_t slen,
157 const struct it_key *key)
159 struct zebra_rec_key_entry **kep_first
160 = zebra_rec_keys_mk_hash(keys, str, slen, key);
161 struct zebra_rec_key_entry **kep = kep_first;
164 struct zebra_rec_key_entry *e = *kep;
165 if (slen == e->len && !memcmp(str, e->buf, slen) &&
166 !key_compare(key, &e->key))
168 *kep = (*kep)->next; /* out of queue */
169 e->next = *kep_first; /* move to front */
176 *kep = nmem_malloc(keys->nmem, sizeof(**kep));
179 memcpy(&(*kep)->key, key, sizeof(*key));
180 (*kep)->buf = nmem_malloc(keys->nmem, slen);
181 memcpy((*kep)->buf, str, slen);
185 void zebra_rec_keys_write(zebra_rec_keys_t keys,
186 const char *str, size_t slen,
187 const struct it_key *key)
190 const char *src = (char*) key;
192 assert(keys->owner_of_buffer);
195 if (!zebra_rec_keys_add_hash(keys, str, slen, key))
198 yaz_log(YLOG_LOG, "dup key slen=%d %.*s "
199 "ord=" ZINT_FORMAT " seq=" ZINT_FORMAT,
200 slen, slen, str, key->mem[0], key->mem[key->len-1]);
202 return; /* key already there . Omit it */
205 if (keys->buf_used+1024 > keys->buf_max)
207 char *b = (char *) xmalloc (keys->buf_max += 128000);
208 if (keys->buf_used > 0)
209 memcpy (b, keys->buf, keys->buf_used);
213 dst = keys->buf + keys->buf_used;
215 iscz1_encode(keys->encode_handle, &dst, &src);
217 memcpy (dst, str, slen);
220 keys->buf_used = dst - keys->buf;
223 void zebra_rec_keys_reset(zebra_rec_keys_t keys)
228 iscz1_reset(keys->encode_handle);
234 int zebra_rec_keys_rewind(zebra_rec_keys_t keys)
237 iscz1_reset(keys->decode_handle);
238 keys->fetch_offset = 0;
239 if (keys->buf_used == 0)
244 int zebra_rec_keys_empty(zebra_rec_keys_t keys)
246 if (keys->buf_used == 0)
251 int zebra_rec_keys_read(zebra_rec_keys_t keys,
252 const char **str, size_t *slen,
256 if (keys->fetch_offset == keys->buf_used)
260 const char *src = keys->buf + keys->fetch_offset;
261 char *dst = (char*) key;
263 assert (keys->fetch_offset < keys->buf_used);
265 /* store the destination key */
266 iscz1_decode(keys->decode_handle, &dst, &src);
268 /* store pointer to string and length of it */
273 keys->fetch_offset = src - keys->buf;
280 * indent-tabs-mode: nil
282 * vim: shiftwidth=4 tabstop=8 expandtab