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, -1, FALSE, NULL);
79 g_once_init_leave(&once, 1);
82 g_print("kvstore: %s port %s\n", host, port);
83 KeyValueClient *client = new KeyValueClient(host, port);
87 static void kvstore_destroy(gpointer store)
89 KeyValueClient *client = (KeyValueClient *)store;
93 static void kvstore_submit(gpointer store, BlueSkyStoreAsync *async)
95 KeyValueClient *client = (KeyValueClient *)store;
97 g_return_if_fail(async->status == ASYNC_NEW);
98 g_return_if_fail(async->op != STORE_OP_NONE);
103 async->status = ASYNC_PENDING;
104 bluesky_store_async_ref(async);
105 g_thread_pool_push(thread_pool, async, NULL);
109 g_warning("Uknown operation type for kvstore: %d\n", async->op);
110 bluesky_store_async_mark_complete(async);
115 static void kvstore_cleanup(gpointer store, BlueSkyStoreAsync *async)
117 KeyValueClient *client = (KeyValueClient *)store;
120 static BlueSkyStoreImplementation store_impl = {
127 extern "C" void bluesky_store_init_kv(void)
129 bluesky_store_register(&store_impl, "kv");