Add CMake config

[dragonstd.git] / dragonstd / map.c

diff --git a/dragonstd/map.c b/dragonstd/map.c
new file mode 100644 (file)
index 0000000..dabed94
--- /dev/null
+++ b/dragonstd/map.c
@@ -0,0 +1,86 @@
+#include "bits/callback.h" // for Transformer
+#include "map.h"
+
+static bool get_lock(Map *map, bool write)
+{
+       bool success;
+
+       pthread_rwlock_rdlock(&map->clk);
+
+       if ((success = !map->cnl)) {
+               if (write)
+                       pthread_rwlock_wrlock(&map->tlk);
+               else
+                       pthread_rwlock_rdlock(&map->tlk);
+       }
+
+       pthread_rwlock_unlock(&map->clk);
+
+       return success;
+}
+
+void map_ini(Map *map)
+{
+       tree_ini(&map->tre);
+       pthread_rwlock_init(&map->tlk, NULL);
+       map->cnl = false;
+       pthread_rwlock_init(&map->clk, NULL);
+}
+
+void map_dst(Map *map)
+{
+       pthread_rwlock_destroy(&map->tlk);
+       pthread_rwlock_destroy(&map->clk);
+}
+
+void map_cnl(Map *map, void *iter, void *arg, void *trans, TreeTraversionOrder order)
+{
+       pthread_rwlock_wrlock(&map->clk);
+       map->cnl = true;
+
+       pthread_rwlock_wrlock(&map->tlk);
+       pthread_rwlock_unlock(&map->clk);
+
+       tree_clr(&map->tre, iter, arg, trans, order);
+
+       pthread_rwlock_unlock(&map->tlk);
+}
+
+bool map_add(Map *map, void *key, void *dat, void *cmp, void *trans)
+{
+       if (!get_lock(map, true))
+               return false;
+
+       bool ret = tree_add(&map->tre, key, dat, cmp, trans);
+       pthread_rwlock_unlock(&map->tlk);
+       return ret;
+}
+
+void *map_get(Map *map, void *key, void *cmp, void *trans)
+{
+       if (!get_lock(map, false))
+               return NULL;
+
+       void *ret = tree_get(&map->tre, key, cmp, trans);
+       pthread_rwlock_unlock(&map->tlk);
+       return ret;
+}
+
+bool map_del(Map *map, void *key, void *cmp, void *call, void *arg, void *trans)
+{
+       if (!get_lock(map, true))
+               return false;
+
+       bool ret = tree_del(&map->tre, key, cmp, call, arg, trans);
+       pthread_rwlock_unlock(&map->tlk);
+       return ret;
+}
+
+void map_trv(Map *map, void *iter, void *arg, void *trans, TreeTraversionOrder order)
+{
+       if (!get_lock(map, false))
+               return;
+
+       tree_trv(&map->tre, iter, arg, trans, order);
+       pthread_rwlock_unlock(&map->tlk);
+}