4 #include <json-c/json.h>
7 #define AFB_BINDING_VERSION 2
8 #include <afb/afb-binding.h>
10 #define DB_FILE "/home/agl/projects/ll-store-binding/build/src/ll-store-binding.lmdb"
14 /// @brief Get a string from a json object.
15 /// @param[in] obj Json object from wich the string is queried.
16 /// @param[in] name Name of the string to get.
18 const char* get_json_string(struct json_object* obj, const char* name)
20 if (!obj || !name || !strlen(name)) return NULL;
22 struct json_object* item = NULL;
23 if (!json_object_object_get_ex(obj, name, &item) || !item) return NULL;
25 return json_object_get_string(item);
28 char* make_key(const char* username, const char* appname, const char* tagname)
30 size_t sz_username = username ? strlen(username) : 0;
31 size_t sz_appname = appname ? strlen(appname) : 0;
32 size_t sz_tagname = tagname ? strlen(tagname) : 0;
33 size_t sz_total = sz_username + sz_appname + sz_tagname + 3;
35 char* result = (char*)malloc(sz_total);
36 memset(result, sz_total, 0);
38 strcpy(result, username);
39 result[sz_username] = '.';
41 strcpy(result + sz_username + 1, appname);
42 result[sz_username + 1 + sz_appname] = '.';
44 strcpy(result + sz_username + 1 + sz_appname + 1, tagname);
49 static void verb_get(struct afb_req req)
52 struct json_object* args = afb_req_json(req);
54 AFB_INFO("args:\n%s\n", json_object_to_json_string_ext(args, JSON_C_TO_STRING_PRETTY));
56 const char* username = get_json_string(args, "username");
57 const char* appname = get_json_string(args, "appname");
58 const char* tagname = get_json_string(args, "tagname");
60 if (!username || !appname || !tagname)
62 AFB_ERROR("[store] username, appname and tagname must be provided!");
63 afb_req_fail(req, "username, appname and tagname must be provided!", NULL);
67 char* keyname = make_key(username, appname, tagname);
70 r = mdb_txn_begin(dbenv, NULL, 0, &txn);
74 AFB_ERROR("Failed to begin a transaction!");
75 afb_req_fail(req, "Failed to begin a transaction!", NULL);
80 r = mdb_dbi_open(txn, NULL, 0, &dbi);
85 AFB_ERROR("Failed to open the database!");
86 afb_req_fail(req, "Failed to open the database!", NULL);
92 k.mv_size = strlen(keyname) + 1;
95 if(mdb_get(txn, dbi, &k, &v))
99 mdb_dbi_close(dbenv, dbi);
100 AFB_ERROR("Failed to get the data!");
101 afb_req_fail(req, "Failed to get the data!", NULL);
105 char* value = strndup(v.mv_data, v.mv_size + 1);
106 if(mdb_txn_commit(txn))
110 mdb_dbi_close(dbenv, dbi);
111 AFB_ERROR("Failed to commit the transaction!");
115 json_object* result = json_object_new_object();
116 json_object_object_add(result, "key", json_object_new_string(keyname));
117 json_object_object_add(result, "value", json_object_new_string(value));
118 afb_req_success(req, result, NULL);
125 static void verb_set(struct afb_req req)
128 struct json_object* args = afb_req_json(req);
130 AFB_INFO("args:\n%s\n", json_object_to_json_string_ext(args, JSON_C_TO_STRING_PRETTY));
132 const char* username = get_json_string(args, "username");
133 const char* appname = get_json_string(args, "appname");
134 const char* tagname = get_json_string(args, "tagname");
135 const char* value = get_json_string(args, "value");
137 if (!username || !appname || !tagname)
139 AFB_ERROR("[store] username, appname and tagname must be provided!");
140 afb_req_fail(req, "username, appname and tagname must be provided!", NULL);
144 char* keyname = make_key(username, appname, tagname);
147 r = mdb_txn_begin(dbenv, NULL, 0, &txn);
151 AFB_ERROR("Failed to begin a transaction!");
152 afb_req_fail(req, "Failed to begin a transaction!", NULL);
157 r = mdb_dbi_open(txn, NULL, 0, &dbi);
162 AFB_ERROR("Failed to open the database!");
163 afb_req_fail(req, "Failed to open the database!", NULL);
169 k.mv_size = strlen(keyname) + 1;
171 v.mv_size = value ? strlen(value) + 1 : 0;
174 if(mdb_put(txn, dbi, &k, &v, 0))
178 mdb_dbi_close(dbenv, dbi);
179 AFB_ERROR("Failed to get the data!");
180 afb_req_fail(req, "Failed to get the data!", NULL);
184 if(mdb_txn_commit(txn))
187 mdb_dbi_close(dbenv, dbi);
188 AFB_ERROR("Failed to commit the transaction!");
192 json_object* result = json_object_new_object();
193 json_object_object_add(result, "key", json_object_new_string(keyname));
194 json_object_object_add(result, "value", json_object_new_string(value));
195 afb_req_success(req, result, NULL);
201 int ll_store_preinit()
203 int r = mdb_env_create(&dbenv);
206 AFB_INFO("Failed to create MDB environment!");
211 r = mdb_env_open(dbenv, "/home/agl/ll-store-binding.lmdb", MDB_NOSUBDIR, 0644);
214 mdb_env_close(dbenv);
216 AFB_INFO("Failed to open MDB environment!");
222 r = mdb_txn_begin(dbenv, NULL, 0, &txn);
225 mdb_env_close(dbenv);
227 AFB_INFO("Failed to begin a transaction!");
232 r = mdb_dbi_open(txn, NULL, 0, &dbi);
235 mdb_env_close(dbenv);
236 AFB_INFO("Failed to open the database!");
243 static const afb_verb_v2 _ll_store_binding_verbs[]= {
246 .callback = verb_get,
249 .session = AFB_SESSION_NONE_V2
253 .callback = verb_set,
256 .session = AFB_SESSION_NONE_V2
261 const struct afb_binding_v2 afbBindingV2 = {
263 .specification = NULL,
264 .verbs = _ll_store_binding_verbs,
265 .preinit = ll_store_preinit,