X-Git-Url: http://git.vrable.net/?a=blobdiff_plain;f=bluesky%2Fstore-kv.cc;h=de16832d50d30718f1b72b967728144419dca6c1;hb=8a2da0db5d5f4f599486876396883f6207ebec6a;hp=7a6ac714ae7ec259b24a258f11419fac8cc41595;hpb=25ddfddd324493ee87cc3770245104420b35fd75;p=bluesky.git diff --git a/bluesky/store-kv.cc b/bluesky/store-kv.cc index 7a6ac71..de16832 100644 --- a/bluesky/store-kv.cc +++ b/bluesky/store-kv.cc @@ -21,24 +21,14 @@ using namespace boost; using namespace kvstore; using namespace std; -static gpointer kvstore_new() -{ - KeyValueClient *client = new KeyValueClient("127.0.0.1", "9090"); - return client; -} +static GThreadPool *thread_pool = NULL; -static void kvstore_destroy(gpointer store) +static void kvstore_task(gpointer a, gpointer b) { - KeyValueClient *client = (KeyValueClient *)store; - delete client; -} + BlueSkyStoreAsync *async = (BlueSkyStoreAsync *)a; + KeyValueClient *client = (KeyValueClient *)bluesky_store_async_get_handle(async); -static void kvstore_submit(gpointer store, BlueSkyStoreAsync *async) -{ - KeyValueClient *client = (KeyValueClient *)store; - - g_return_if_fail(async->status == ASYNC_NEW); - g_return_if_fail(async->op != STORE_OP_NONE); + async->status = ASYNC_RUNNING; switch (async->op) { case STORE_OP_GET: @@ -61,11 +51,65 @@ static void kvstore_submit(gpointer store, BlueSkyStoreAsync *async) } default: - g_warning("Uknown operation type for MemStore: %d\n", async->op); - return; + break; } bluesky_store_async_mark_complete(async); + bluesky_store_async_unref(async); +} + +static gpointer kvstore_new(const gchar *path) +{ + /* TODO: Right now we leak this memory. We should probably clean up in + * kvstore_destroy, but it's not a big deal. */ + const gchar *host = "127.0.0.1", *port = "9090"; + if (path != NULL) { + gchar **target = g_strsplit(path, ":", 0); + if (target[0] != NULL) { + host = target[0]; + if (target[1] != NULL) { + port = target[1]; + } + } + } + + static volatile gsize once = 0; + if (g_once_init_enter(&once)) { + thread_pool = g_thread_pool_new(kvstore_task, NULL, -1, FALSE, NULL); + g_once_init_leave(&once, 1); + } + + g_print("kvstore: %s port %s\n", host, port); + KeyValueClient *client = new KeyValueClient(host, port); + return client; +} + +static void kvstore_destroy(gpointer store) +{ + KeyValueClient *client = (KeyValueClient *)store; + delete client; +} + +static void kvstore_submit(gpointer store, BlueSkyStoreAsync *async) +{ + KeyValueClient *client = (KeyValueClient *)store; + + g_return_if_fail(async->status == ASYNC_NEW); + g_return_if_fail(async->op != STORE_OP_NONE); + + switch (async->op) { + case STORE_OP_GET: + case STORE_OP_PUT: + async->status = ASYNC_PENDING; + bluesky_store_async_ref(async); + g_thread_pool_push(thread_pool, async, NULL); + break; + + default: + g_warning("Uknown operation type for kvstore: %d\n", async->op); + bluesky_store_async_mark_complete(async); + break; + } } static void kvstore_cleanup(gpointer store, BlueSkyStoreAsync *async)