Infrastructure to generate PDF figures for the paper
[bluesky.git] / bluesky / store-multi.c
1 /* Blue Sky: File Systems in the Cloud
2  *
3  * Copyright (C) 2009  The Regents of the University of California
4  * Written by Michael Vrable <mvrable@cs.ucsd.edu>
5  *
6  * TODO: Licensing
7  */
8
9 /* A stacked storage implementation which tries to improve performance by
10  * duplicating GET requests.  Create using a name like "multi:s3" and each GET
11  * request will be translated into two GET requests to the "s3" backend.  The
12  * first to complete will have its results returned. */
13
14 #include <stdint.h>
15 #include <stdlib.h>
16 #include <glib.h>
17 #include <string.h>
18
19 #include "bluesky-private.h"
20
21 struct MultiRequest {
22 };
23
24 static gpointer multistore_new(const gchar *path)
25 {
26     BlueSkyStore *base = bluesky_store_new(path);
27     if (base == NULL) {
28         g_warning("Unable to create base store %s for multirequest stacking.",
29                   path);
30     }
31
32     return base;
33 }
34
35 static void multistore_destroy(gpointer store)
36 {
37     bluesky_store_free(store);
38 }
39
40 static void multistore_completion_handler(BlueSkyStoreAsync *async,
41                                           BlueSkyStoreAsync *top_async)
42 {
43     g_mutex_lock(top_async->lock);
44
45     /* This might be the second request to finish; in that case we don't do
46      * anything. */
47     if (top_async->status == ASYNC_RUNNING) {
48         if (top_async->op == STORE_OP_GET) {
49             bluesky_string_unref(top_async->data);
50             top_async->data = async->data;
51             bluesky_string_ref(top_async->data);
52         }
53         top_async->result = async->result;
54         bluesky_store_async_mark_complete(top_async);
55     }
56
57     g_mutex_unlock(top_async->lock);
58     bluesky_store_async_unref(top_async);
59 }
60
61 static void multistore_submit(gpointer store, BlueSkyStoreAsync *async)
62 {
63     BlueSkyStore *base = (BlueSkyStore *)store;
64
65     g_return_if_fail(async->status == ASYNC_NEW);
66     g_return_if_fail(async->op != STORE_OP_NONE);
67
68     switch (async->op) {
69     case STORE_OP_GET:
70         async->status = ASYNC_RUNNING;
71         async->exec_time = bluesky_now_hires();
72         for (int i = 0; i < 2; i++) {
73             BlueSkyStoreAsync *a = bluesky_store_async_new(base);
74             a->op = STORE_OP_GET;
75             a->key = g_strdup(async->key);
76             bluesky_store_async_ref(async);
77             bluesky_store_async_add_notifier(a, (GFunc)multistore_completion_handler, async);
78             bluesky_store_async_submit(a);
79             bluesky_store_async_unref(a);
80         }
81         break;
82
83     case STORE_OP_PUT:
84     {
85         async->status = ASYNC_RUNNING;
86         async->exec_time = bluesky_now_hires();
87
88         bluesky_store_async_ref(async);
89         BlueSkyStoreAsync *a = bluesky_store_async_new(base);
90         a->op = STORE_OP_PUT;
91         a->key = g_strdup(async->key);
92         a->data = async->data;
93         bluesky_string_ref(a->data);
94         bluesky_store_async_add_notifier(a, (GFunc)multistore_completion_handler, async);
95         bluesky_store_async_submit(a);
96         bluesky_store_async_unref(a);
97         break;
98     }
99
100     default:
101         g_warning("Uknown operation type for multistore: %d\n", async->op);
102         bluesky_store_async_mark_complete(async);
103         break;
104     }
105 }
106
107 static char *multistore_lookup_last(gpointer store, const char *prefix)
108 {
109     BlueSkyStore *base = (BlueSkyStore *)store;
110     return bluesky_store_lookup_last(base, prefix);
111 }
112
113 static void multistore_cleanup(gpointer store, BlueSkyStoreAsync *async)
114 {
115 }
116
117 static BlueSkyStoreImplementation store_impl = {
118     .create = multistore_new,
119     .destroy = multistore_destroy,
120     .submit = multistore_submit,
121     .cleanup = multistore_cleanup,
122     .lookup_last = multistore_lookup_last,
123 };
124
125 void bluesky_store_init_multi(void)
126 {
127     bluesky_store_register(&store_impl, "multi");
128 }