1 /* Blue Sky: File Systems in the Cloud
3 * Copyright (C) 2009 The Regents of the University of California
4 * Written by Michael Vrable <mvrable@cs.ucsd.edu>
9 /* Interface to John McCullough's simple key/value store. */
16 #include "bluesky-private.h"
17 #include "kvservice.h"
20 using namespace boost;
21 using namespace kvstore;
24 static GThreadPool *thread_pool = NULL;
26 static void kvstore_task(gpointer a, gpointer b)
28 BlueSkyStoreAsync *async = (BlueSkyStoreAsync *)a;
29 KeyValueClient *client = (KeyValueClient *)bluesky_store_async_get_handle(async);
31 async->status = ASYNC_RUNNING;
37 if (client->Get(async->key, &value)) {
38 async->data = bluesky_string_new(g_memdup(value.c_str(),
48 string value(async->data->data, async->data->len);
49 client->Put(async->key, value);
57 bluesky_store_async_mark_complete(async);
58 bluesky_store_async_unref(async);
61 static gpointer kvstore_new(const gchar *path)
63 /* TODO: Right now we leak this memory. We should probably clean up in
64 * kvstore_destroy, but it's not a big deal. */
65 gchar **target = g_strsplit(path, ":", 0);
66 const gchar *host = "127.0.0.1", *port = "9090";
67 if (target[0] != NULL) {
69 if (target[1] != NULL) {
74 static volatile gsize once = 0;
75 if (g_once_init_enter(&once)) {
76 thread_pool = g_thread_pool_new(kvstore_task, NULL, -1, FALSE, NULL);
77 g_once_init_leave(&once, 1);
80 g_print("kvstore: %s port %s\n", host, port);
81 KeyValueClient *client = new KeyValueClient(host, port);
85 static void kvstore_destroy(gpointer store)
87 KeyValueClient *client = (KeyValueClient *)store;
91 static void kvstore_submit(gpointer store, BlueSkyStoreAsync *async)
93 KeyValueClient *client = (KeyValueClient *)store;
95 g_return_if_fail(async->status == ASYNC_NEW);
96 g_return_if_fail(async->op != STORE_OP_NONE);
101 async->status = ASYNC_PENDING;
102 bluesky_store_async_ref(async);
103 g_thread_pool_push(thread_pool, async, NULL);
107 g_warning("Uknown operation type for kvstore: %d\n", async->op);
108 bluesky_store_async_mark_complete(async);
113 static void kvstore_cleanup(gpointer store, BlueSkyStoreAsync *async)
115 KeyValueClient *client = (KeyValueClient *)store;
118 static BlueSkyStoreImplementation store_impl = {
125 extern "C" void bluesky_store_init_kv(void)
127 bluesky_store_register(&store_impl, "kv");