--- /dev/null
+// -*- mode:C; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#include <errno.h>
+
+#include "objclass/objclass.h"
+
+#include "cls_statelog_ops.h"
+
+
+CLS_VER(1,0)
+CLS_NAME(statelog)
+
+static string statelog_index_by_client_prefix = "1_";
+static string statelog_index_by_object_prefix = "2_";
+
+
+static int write_statelog_entry(cls_method_context_t hctx, const string& index, const cls_statelog_entry& entry)
+{
+ bufferlist bl;
+ ::encode(entry, bl);
+
+ int ret = cls_cxx_map_set_val(hctx, index, &bl);
+ if (ret < 0)
+ return ret;
+
+ return 0;
+}
+
+static void get_index_by_client(const string& client_id, const string& op_id, string& index)
+{
+ index = statelog_index_by_client_prefix;
+ index.append(client_id + "_" + op_id);
+}
+
+static void get_index_by_client(cls_statelog_entry& entry, string& index)
+{
+ get_index_by_client(entry.client_id, entry.op_id, index);
+}
+
+static void get_index_by_object(const string& object, const string& op_id, string& index)
+{
+ char buf[16];
+ snprintf(buf, sizeof(buf), "%d_", (int)object.size());
+
+ index = statelog_index_by_object_prefix + buf; /* append object length to ensure uniqueness */
+ index.append(object + "_" + op_id);
+}
+
+static void get_index_by_object(cls_statelog_entry& entry, string& index)
+{
+ get_index_by_object(entry.object, entry.op_id, index);
+}
+
+static int get_existing_entry(cls_method_context_t hctx, const string& client_id,
+ const string& op_id, const string& object,
+ cls_statelog_entry& entry)
+{
+ if ((object.empty() && client_id.empty()) || op_id.empty()) {
+ return -EINVAL;
+ }
+
+ string obj_index;
+ if (!object.empty()) {
+ get_index_by_object(object, op_id, obj_index);
+ } else {
+ get_index_by_client(client_id, op_id, obj_index);
+ }
+
+ bufferlist bl;
+ int rc = cls_cxx_map_get_val(hctx, obj_index, &bl);
+ if (rc < 0) {
+ CLS_LOG(0, "could not find entry %s", obj_index.c_str());
+ return rc;
+ }
+ try {
+ bufferlist::iterator iter = bl.begin();
+ ::decode(entry, iter);
+ } catch (buffer::error& err) {
+ CLS_LOG(0, "ERROR: failed to decode entry %s", obj_index.c_str());
+ return -EIO;
+ }
+
+ if ((!object.empty() && entry.object != object) ||
+ (!client_id.empty() && entry.client_id != client_id)){
+ /* ouch, we were passed inconsistent client_id / object */
+ CLS_LOG(0, "data mismatch: object=%s client_id=%s entry: object=%s client_id=%s",
+ object.c_str(), client_id.c_str(), entry.object.c_str(), entry.client_id.c_str());
+ return -EINVAL;
+ }
+
+ return 0;
+}
+
+static int cls_statelog_add(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
+{
+ bufferlist::iterator in_iter = in->begin();
+
+ cls_statelog_add_op op;
+ try {
+ ::decode(op, in_iter);
+ } catch (buffer::error& err) {
+ CLS_LOG(1, "ERROR: cls_statelog_add_op(): failed to decode op");
+ return -EINVAL;
+ }
+
+ for (list<cls_statelog_entry>::iterator iter = op.entries.begin();
+ iter != op.entries.end(); ++iter) {
+ cls_statelog_entry& entry = *iter;
+
+ string index_by_client;
+
+ get_index_by_client(entry, index_by_client);
+
+ CLS_LOG(0, "storing entry by client/op at %s", index_by_client.c_str());
+
+ int ret = write_statelog_entry(hctx, index_by_client, entry);
+ if (ret < 0)
+ return ret;
+
+ string index_by_obj;
+
+ get_index_by_object(entry, index_by_obj);
+
+ CLS_LOG(0, "storing entry by object at %s", index_by_obj.c_str());
+ ret = write_statelog_entry(hctx, index_by_obj, entry);
+ if (ret < 0)
+ return ret;
+
+ }
+
+ return 0;
+}
+
+static int cls_statelog_list(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
+{
+ bufferlist::iterator in_iter = in->begin();
+
+ cls_statelog_list_op op;
+ try {
+ ::decode(op, in_iter);
+ } catch (buffer::error& err) {
+ CLS_LOG(1, "ERROR: cls_statelog_list_op(): failed to decode op");
+ return -EINVAL;
+ }
+
+ map<string, bufferlist> keys;
+
+ string from_index;
+ string match_prefix;
+
+ if (!op.client_id.empty()) {
+ get_index_by_client(op.client_id, op.op_id, match_prefix);
+ } else if (!op.object.empty()) {
+ get_index_by_object(op.object, op.op_id, match_prefix);
+ } else {
+ match_prefix = statelog_index_by_object_prefix;
+ }
+
+ if (op.marker.empty()) {
+ from_index = match_prefix;
+ } else {
+ from_index = op.marker;
+ }
+
+#define MAX_ENTRIES 1000
+ size_t max_entries = op.max_entries;
+ if (!max_entries || max_entries > MAX_ENTRIES)
+ max_entries = MAX_ENTRIES;
+
+ cls_statelog_list_ret ret;
+
+ int rc = cls_cxx_map_get_vals(hctx, from_index, match_prefix, max_entries, &keys, &ret.truncated);
+ if (rc < 0)
+ return rc;
+
+ CLS_LOG(20, "from_index=%s match_prefix=%s", from_index.c_str(), match_prefix.c_str());
+
+ list<cls_statelog_entry>& entries = ret.entries;
+ map<string, bufferlist>::iterator iter = keys.begin();
+
+ string marker;
+
+ for (; iter != keys.end(); ++iter) {
+ const string& index = iter->first;
+ marker = index;
+
+ bufferlist& bl = iter->second;
+ bufferlist::iterator biter = bl.begin();
+ try {
+ cls_statelog_entry e;
+ ::decode(e, biter);
+ entries.push_back(e);
+ } catch (buffer::error& err) {
+ CLS_LOG(0, "ERROR: cls_statelog_list: could not decode entry, index=%s", index.c_str());
+ }
+ }
+
+ if (ret.truncated) {
+ ret.marker = marker;
+ }
+
+ ::encode(ret, *out);
+
+ return 0;
+}
+
+static int cls_statelog_remove(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
+{
+ bufferlist::iterator in_iter = in->begin();
+
+ cls_statelog_remove_op op;
+ try {
+ ::decode(op, in_iter);
+ } catch (buffer::error& err) {
+ CLS_LOG(1, "ERROR: cls_statelog_remove_op(): failed to decode op");
+ return -EINVAL;
+ }
+
+ cls_statelog_entry entry;
+
+ int rc = get_existing_entry(hctx, op.client_id, op.op_id, op.object, entry);
+ if (rc < 0)
+ return rc;
+
+ string obj_index;
+ get_index_by_object(entry.object, entry.op_id, obj_index);
+
+ rc = cls_cxx_map_remove_key(hctx, obj_index);
+ if (rc < 0) {
+ CLS_LOG(0, "ERROR: failed to remove key");
+ return rc;
+ }
+
+ string client_index;
+ get_index_by_client(entry.client_id, entry.op_id, client_index);
+
+ rc = cls_cxx_map_remove_key(hctx, client_index);
+ if (rc < 0) {
+ CLS_LOG(0, "ERROR: failed to remove key");
+ return rc;
+ }
+
+ return 0;
+}
+
+static int cls_statelog_check_state(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
+{
+ bufferlist::iterator in_iter = in->begin();
+
+ cls_statelog_check_state_op op;
+ try {
+ ::decode(op, in_iter);
+ } catch (buffer::error& err) {
+ CLS_LOG(1, "ERROR: cls_statelog_check_state_op(): failed to decode op");
+ return -EINVAL;
+ }
+
+ if (op.object.empty() || op.op_id.empty()) {
+ CLS_LOG(0, "object name or op id not specified");
+ return -EINVAL;
+ }
+
+
+ cls_statelog_entry entry;
+
+ int rc = get_existing_entry(hctx, op.client_id, op.op_id, op.object, entry);
+ if (rc < 0)
+ return rc;
+
+ if (entry.state != op.state)
+ return -ECANCELED;
+
+ return 0;
+}
+
+CLS_INIT(statelog)
+{
+ CLS_LOG(1, "Loaded log class!");
+
+ cls_handle_t h_class;
+ cls_method_handle_t h_statelog_add;
+ cls_method_handle_t h_statelog_list;
+ cls_method_handle_t h_statelog_remove;
+ cls_method_handle_t h_statelog_check_state;
+
+ cls_register("statelog", &h_class);
+
+ /* log */
+ cls_register_cxx_method(h_class, "add", CLS_METHOD_RD | CLS_METHOD_WR, cls_statelog_add, &h_statelog_add);
+ cls_register_cxx_method(h_class, "list", CLS_METHOD_RD, cls_statelog_list, &h_statelog_list);
+ cls_register_cxx_method(h_class, "remove", CLS_METHOD_RD | CLS_METHOD_WR, cls_statelog_remove, &h_statelog_remove);
+ cls_register_cxx_method(h_class, "check_state", CLS_METHOD_RD, cls_statelog_check_state, &h_statelog_check_state);
+
+ return;
+}
+