thashtable

Author: Lior Amram

include/ribs.h

@@ -26,6 +26,7 @@
 #include "context.h"
 #include "ctx_pool.h"
 #include "hashtable.h"
+#include "thashtable.h"
 #include "lhashtable.h"
 #include "epoll_worker.h"
 #include "logger.h"

include/thashtable.h

@@ -0,0 +1,48 @@
+/*
+    This file is part of RIBS2.0 (Robust Infrastructure for Backend Systems).
+    RIBS is an infrastructure for building great SaaS applications (but not
+    limited to).
+
+    Copyright (C) 2014 Adap.tv, Inc.
+
+    RIBS is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, version 2.1 of the License.
+
+    RIBS is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with RIBS.  If not, see <http://www.gnu.org/licenses/>.
+*/
+#ifndef _THASHTABLE__H_
+#define _THASHTABLE__H_
+
+#include "ribs_defs.h"
+#include <string.h>
+
+struct thashtable;
+struct thashtable_internal_rec;
+
+typedef struct thashtable_internal_rec thashtable_rec_t;
+
+struct thashtable *thashtable_create(void);
+thashtable_rec_t *thashtable_insert(struct thashtable *tht, const void *key, size_t key_len, const void *val, size_t val_len, int *inserted);
+thashtable_rec_t *thashtable_put(struct thashtable *tht, const void *key, size_t key_len, const void *val, size_t val_len);
+thashtable_rec_t *thashtable_insert_alloc(struct thashtable *tht, const void *key, size_t key_len, size_t val_len);
+thashtable_rec_t *thashtable_lookup(struct thashtable *tht, const void *key, size_t key_len);
+thashtable_rec_t *thashtable_remove(struct thashtable *tht, const void *key, size_t key_len);
+int thashtable_foreach(struct thashtable *tht, int (*func)(thashtable_rec_t *rec));
+
+static inline void *thashtable_get_key(thashtable_rec_t *rec);
+static inline uint32_t thashtable_get_key_size(thashtable_rec_t *rec);
+static inline void *thashtable_get_val(thashtable_rec_t *rec);
+static inline uint32_t thashtable_get_val_size(thashtable_rec_t *rec);
+static inline uint32_t thashtable_get_size(struct thashtable *tht);
+static inline const char *thashtable_lookup_str(struct thashtable *tht, const char *key, const char *default_val);
+
+#include "../src/_thashtable.c"
+
+#endif // _THASHTABLE__H_

src/_thashtable.c

@@ -0,0 +1,60 @@
+/*
+    This file is part of RIBS2.0 (Robust Infrastructure for Backend Systems).
+    RIBS is an infrastructure for building great SaaS applications (but not
+    limited to).
+
+    Copyright (C) 2014 Adap.tv, Inc.
+
+    RIBS is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, version 2.1 of the License.
+
+    RIBS is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with RIBS.  If not, see <http://www.gnu.org/licenses/>.
+*/
+struct tht_entry {
+    uint32_t hashcode;
+    thashtable_rec_t *rec;
+};
+
+struct thashtable {
+    uint32_t mask;
+    uint32_t size;
+    struct tht_entry *buckets[32];
+};
+
+struct thashtable_internal_rec {
+    uint32_t key_size;
+    uint32_t val_size;
+    char data[];
+};
+
+static inline void *thashtable_get_key(thashtable_rec_t *rec) {
+    return ((struct thashtable_internal_rec *)rec)->data;
+}
+
+static inline uint32_t thashtable_get_key_size(thashtable_rec_t *rec) {
+    return ((struct thashtable_internal_rec *)rec)->key_size;
+}
+
+static inline void *thashtable_get_val(thashtable_rec_t *rec) {
+    return ((struct thashtable_internal_rec *)rec)->data + ((struct thashtable_internal_rec *)rec)->key_size;
+}
+
+static inline uint32_t thashtable_get_val_size(thashtable_rec_t *rec) {
+    return ((struct thashtable_internal_rec *)rec)->val_size;
+}
+
+static inline uint32_t thashtable_get_size(struct thashtable *tht) {
+    return tht->size;
+}
+
+static inline const char *thashtable_lookup_str(struct thashtable *tht, const char *key, const char *default_val) {
+    thashtable_rec_t *rec = thashtable_lookup(tht, key, strlen(key));
+    return (rec ? thashtable_get_val(rec) : default_val);
+}

src/ribs2.mk

@@ -1,6 +1,6 @@
 TARGET=ribs2.a
 
-SRC=context.c epoll_worker.c ctx_pool.c http_server.c hashtable.c mime_types.c http_client_pool.c timeout_handler.c ribify.c logger.c daemonize.c http_headers.c http_cookies.c file_mapper.c ds_var_field.c file_utils.c lhashtable.c search.c json.c memalloc.c mempool.c sleep.c timer.c ringbuf.c ringfile.c sendemail.c ds_loader.c heap.c vmallocator.c base64.c http_file_server.c http_vhost.c 
+SRC=context.c epoll_worker.c ctx_pool.c http_server.c hashtable.c mime_types.c http_client_pool.c timeout_handler.c ribify.c logger.c daemonize.c http_headers.c http_cookies.c file_mapper.c ds_var_field.c file_utils.c lhashtable.c search.c json.c memalloc.c mempool.c sleep.c timer.c ringbuf.c ringfile.c sendemail.c ds_loader.c heap.c vmallocator.c base64.c http_file_server.c http_vhost.c thashtable.c
 ASM=context_asm.S
 
 CFLAGS+= -I ../include