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 const gchar *host = "127.0.0.1", *port = "9090";
67 gchar **target = g_strsplit(path, ":", 0);
68 if (target[0] != NULL) {
70 if (target[1] != NULL) {
76 static volatile gsize once = 0;
77 if (g_once_init_enter(&once)) {
78 thread_pool = g_thread_pool_new(kvstore_task, NULL,
79 bluesky_max_threads, FALSE, NULL);
80 g_once_init_leave(&once, 1);
83 g_print("kvstore: %s port %s\n", host, port);
84 KeyValueClient *client = new KeyValueClient(host, port);
88 static void kvstore_destroy(gpointer store)
90 KeyValueClient *client = (KeyValueClient *)store;
94 static void kvstore_submit(gpointer store, BlueSkyStoreAsync *async)
96 KeyValueClient *client = (KeyValueClient *)store;
98 g_return_if_fail(async->status == ASYNC_NEW);
99 g_return_if_fail(async->op != STORE_OP_NONE);
104 async->status = ASYNC_PENDING;
105 bluesky_store_async_ref(async);
106 g_thread_pool_push(thread_pool, async, NULL);
110 g_warning("Uknown operation type for kvstore: %d\n", async->op);
111 bluesky_store_async_mark_complete(async);
116 static void kvstore_cleanup(gpointer store, BlueSkyStoreAsync *async)
118 KeyValueClient *client = (KeyValueClient *)store;
121 static BlueSkyStoreImplementation store_impl = {
128 extern "C" void bluesky_store_init_kv(void)
130 bluesky_store_register(&store_impl, "kv");