summaryrefslogtreecommitdiffstats
path: root/src/dictionary.c
diff options
context:
space:
mode:
authorFederico Ceratto <federico.ceratto@gmail.com>2016-03-30 21:40:42 +0000
committerFederico Ceratto <federico.ceratto@gmail.com>2016-03-30 21:40:42 +0000
commit9ce153ce7167c11adba8ac225edc7a707e97c6eb (patch)
tree9f6e849cce2dcc7e5b4e9e6252c843dc2d0787a2 /src/dictionary.c
downloadnetdata-9ce153ce7167c11adba8ac225edc7a707e97c6eb.tar.xz
netdata-9ce153ce7167c11adba8ac225edc7a707e97c6eb.zip
Imported Upstream version 1.0.0upstream/1.0.0
Diffstat (limited to 'src/dictionary.c')
-rwxr-xr-xsrc/dictionary.c160
1 files changed, 160 insertions, 0 deletions
diff --git a/src/dictionary.c b/src/dictionary.c
new file mode 100755
index 000000000..31f4d52e1
--- /dev/null
+++ b/src/dictionary.c
@@ -0,0 +1,160 @@
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+#include <pthread.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "avl.h"
+#include "common.h"
+#include "log.h"
+
+#include "dictionary.h"
+
+// ----------------------------------------------------------------------------
+// name_value index
+
+static int name_value_iterator(avl *a) { if(a) {}; return 0; }
+
+static int name_value_compare(void* a, void* b) {
+ if(((NAME_VALUE *)a)->hash < ((NAME_VALUE *)b)->hash) return -1;
+ else if(((NAME_VALUE *)a)->hash > ((NAME_VALUE *)b)->hash) return 1;
+ else return strcmp(((NAME_VALUE *)a)->name, ((NAME_VALUE *)b)->name);
+}
+
+#define name_value_index_add(dict, cv) avl_insert(&((dict)->values_index), (avl *)(cv))
+#define name_value_index_del(dict, cv) avl_remove(&((dict)->values_index), (avl *)(cv))
+
+static NAME_VALUE *dictionary_name_value_index_find(DICTIONARY *dict, const char *name, uint32_t hash) {
+ NAME_VALUE *result = NULL, tmp;
+ tmp.hash = (hash)?hash:simple_hash(name);
+ tmp.name = (char *)name;
+
+ avl_search(&(dict->values_index), (avl *)&tmp, name_value_iterator, (avl **)&result);
+ return result;
+}
+
+// ----------------------------------------------------------------------------
+
+static NAME_VALUE *dictionary_name_value_create(DICTIONARY *dict, const char *name, void *value, size_t value_len) {
+ debug(D_DICTIONARY, "Creating name value entry for name '%s', value '%s'.", name, value);
+
+ NAME_VALUE *nv = calloc(1, sizeof(NAME_VALUE));
+ if(!nv) {
+ fatal("Cannot allocate name_value of size %z", sizeof(NAME_VALUE));
+ exit(1);
+ }
+
+ nv->name = strdup(name);
+ if(!nv->name) fatal("Cannot allocate name_value.name of size %z", strlen(name));
+ nv->hash = simple_hash(nv->name);
+
+ nv->value = malloc(value_len);
+ if(!nv->value) fatal("Cannot allocate name_value.value of size %z", value_len);
+ memcpy(nv->value, value, value_len);
+
+ // link it
+ pthread_rwlock_wrlock(&dict->rwlock);
+ nv->next = dict->values;
+ dict->values = nv;
+ pthread_rwlock_unlock(&dict->rwlock);
+
+ // index it
+ name_value_index_add(dict, nv);
+
+ return nv;
+}
+
+static void dictionary_name_value_destroy(DICTIONARY *dict, NAME_VALUE *nv) {
+ debug(D_DICTIONARY, "Destroying name value entry for name '%s'.", nv->name);
+
+ pthread_rwlock_wrlock(&dict->rwlock);
+ if(dict->values == nv) dict->values = nv->next;
+ else {
+ NAME_VALUE *n = dict->values;
+ while(n && n->next && n->next != nv) nv = nv->next;
+ if(!n || n->next != nv) {
+ fatal("Cannot find name_value with name '%s' in dictionary.", nv->name);
+ exit(1);
+ }
+ n->next = nv->next;
+ nv->next = NULL;
+ }
+ pthread_rwlock_unlock(&dict->rwlock);
+
+ free(nv->value);
+ free(nv);
+}
+
+// ----------------------------------------------------------------------------
+
+DICTIONARY *dictionary_create(void) {
+ debug(D_DICTIONARY, "Creating dictionary.");
+
+ DICTIONARY *dict = calloc(1, sizeof(DICTIONARY));
+ if(!dict) {
+ fatal("Cannot allocate DICTIONARY");
+ exit(1);
+ }
+
+ avl_init(&dict->values_index, name_value_compare);
+ pthread_rwlock_init(&dict->rwlock, NULL);
+
+ return dict;
+}
+
+void dictionary_destroy(DICTIONARY *dict) {
+ debug(D_DICTIONARY, "Destroying dictionary.");
+
+ pthread_rwlock_wrlock(&dict->rwlock);
+ while(dict->values) dictionary_name_value_destroy(dict, dict->values);
+ pthread_rwlock_unlock(&dict->rwlock);
+
+ free(dict);
+}
+
+// ----------------------------------------------------------------------------
+
+void *dictionary_set(DICTIONARY *dict, const char *name, void *value, size_t value_len) {
+ debug(D_DICTIONARY, "SET dictionary entry with name '%s'.", name);
+
+ pthread_rwlock_rdlock(&dict->rwlock);
+ NAME_VALUE *nv = dictionary_name_value_index_find(dict, name, 0);
+ pthread_rwlock_unlock(&dict->rwlock);
+ if(!nv) {
+ debug(D_DICTIONARY, "Dictionary entry with name '%s' not found. Creating a new one.", name);
+ nv = dictionary_name_value_create(dict, name, value, value_len);
+ if(!nv) {
+ fatal("Cannot create name_value.");
+ exit(1);
+ }
+ return nv->value;
+ }
+ else {
+ debug(D_DICTIONARY, "Dictionary entry with name '%s' found. Changing its value.", name);
+ pthread_rwlock_wrlock(&dict->rwlock);
+ void *old = nv->value;
+ nv->value = malloc(value_len);
+ if(!nv->value) fatal("Cannot allocate value of size %z", value_len);
+ memcpy(nv->value, value, value_len);
+ pthread_rwlock_unlock(&dict->rwlock);
+ free(old);
+ }
+
+ return nv->value;
+}
+
+void *dictionary_get(DICTIONARY *dict, const char *name) {
+ debug(D_DICTIONARY, "GET dictionary entry with name '%s'.", name);
+
+ pthread_rwlock_rdlock(&dict->rwlock);
+ NAME_VALUE *nv = dictionary_name_value_index_find(dict, name, 0);
+ pthread_rwlock_unlock(&dict->rwlock);
+ if(!nv) {
+ debug(D_DICTIONARY, "Not found dictionary entry with name '%s'.", name);
+ return NULL;
+ }
+
+ debug(D_DICTIONARY, "Found dictionary entry with name '%s'.", name);
+ return nv->value;
+}