--- /dev/null
+/*
+ * Key-value store using librados
+ *
+ * September 2, 2012
+ * Eleanor Cawthon
+ * eleanor.cawthon@inktank.com
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation. See file COPYING.
+ */
+
+#include "key_value_store/key_value_structure.h"
+#include "key_value_store/kv_flat_btree_async.h"
+#include "key_value_store/kvs_arg_types.h"
+#include "include/rados/librados.hpp"
+#include "/usr/include/asm-generic/errno.h"
+#include "/usr/include/asm-generic/errno-base.h"
+#include "common/ceph_context.h"
+#include "common/Clock.h"
+#include "include/types.h"
+
+
+#include <string>
+#include <iostream>
+#include <cassert>
+#include <climits>
+#include <cmath>
+#include <sstream>
+#include <stdlib.h>
+#include <iterator>
+
+using namespace std;
+using ceph::bufferlist;
+
+bool index_data::is_timed_out(utime_t now, utime_t timeout) const {
+ return prefix != "" && now - ts > timeout;
+}
+
+void IndexCache::clear() {
+ k2itmap.clear();
+ t2kmap.clear();
+}
+
+void IndexCache::push(const string &key, const index_data &idata) {
+ if (cache_size == 0) {
+ return;
+ }
+ index_data old_idata;
+ map<key_data, pair<index_data, utime_t> >::iterator old_it =
+ k2itmap.lower_bound(key_data(key));
+ if (old_it != k2itmap.end()) {
+ t2kmap.erase(old_it->second.second);
+ k2itmap.erase(old_it);
+ }
+ map<key_data, pair<index_data, utime_t> >::iterator new_it =
+ k2itmap.find(idata.kdata);
+ if (new_it != k2itmap.end()) {
+ utime_t old_time = new_it->second.second;
+ t2kmap.erase(old_time);
+ }
+ utime_t time = ceph_clock_now();
+ k2itmap[idata.kdata] = make_pair(idata, time);
+ t2kmap[time] = idata.kdata;
+ if ((int)k2itmap.size() > cache_size) {
+ pop();
+ }
+
+}
+
+void IndexCache::push(const index_data &idata) {
+ if (cache_size == 0) {
+ return;
+ }
+ if (k2itmap.count(idata.kdata) > 0) {
+ utime_t old_time = k2itmap[idata.kdata].second;
+ t2kmap.erase(old_time);
+ k2itmap.erase(idata.kdata);
+ }
+ utime_t time = ceph_clock_now();
+ k2itmap[idata.kdata] = make_pair(idata, time);
+ t2kmap[time] = idata.kdata;
+ if ((int)k2itmap.size() > cache_size) {
+ pop();
+ }
+}
+
+void IndexCache::pop() {
+ if (cache_size == 0) {
+ return;
+ }
+ map<utime_t, key_data>::iterator it = t2kmap.begin();
+ utime_t time = it->first;
+ key_data kdata = it->second;
+ k2itmap.erase(kdata);
+ t2kmap.erase(time);
+}
+
+void IndexCache::erase(key_data kdata) {
+ if (cache_size == 0) {
+ return;
+ }
+ if (k2itmap.count(kdata) > 0) {
+ utime_t c = k2itmap[kdata].second;
+ k2itmap.erase(kdata);
+ t2kmap.erase(c);
+ }
+}
+
+int IndexCache::get(const string &key, index_data *idata) const {
+ if (cache_size == 0) {
+ return -ENODATA;
+ }
+ if ((int)k2itmap.size() == 0) {
+ return -ENODATA;
+ }
+ map<key_data, pair<index_data, utime_t> >::const_iterator it =
+ k2itmap.lower_bound(key_data(key));
+ if (it == k2itmap.end() || !(it->second.first.min_kdata < key_data(key))) {
+ return -ENODATA;
+ } else {
+ *idata = it->second.first;
+ }
+ return 0;
+}
+
+int IndexCache::get(const string &key, index_data *idata,
+ index_data *next_idata) const {
+ if (cache_size == 0) {
+ return -ENODATA;
+ }
+ map<key_data, pair<index_data, utime_t> >::const_iterator it =
+ k2itmap.lower_bound(key_data(key));
+ if (it == k2itmap.end() || ++it == k2itmap.end()) {
+ return -ENODATA;
+ } else {
+ --it;
+ if (!(it->second.first.min_kdata < key_data(key))){
+ //stale, should be reread.
+ return -ENODATA;
+ } else {
+ *idata = it->second.first;
+ ++it;
+ if (it != k2itmap.end()) {
+ *next_idata = it->second.first;
+ }
+ }
+ }
+ return 0;
+}
+
+int KvFlatBtreeAsync::nothing() {
+ return 0;
+}
+
+int KvFlatBtreeAsync::wait() {
+ if (rand() % 10 == 0) {
+ usleep(wait_ms);
+ }
+ return 0;
+}
+
+int KvFlatBtreeAsync::suicide() {
+ if (rand() % 10 == 0) {
+ if (verbose) cout << client_name << " is suiciding" << std::endl;
+ return 1;
+ }
+ return 0;
+}
+
+int KvFlatBtreeAsync::next(const index_data &idata, index_data * out_data)
+{
+ if (verbose) cout << "\t\t" << client_name << "-next: finding next of "
+ << idata.str()
+ << std::endl;
+ int err = 0;
+ librados::ObjectReadOperation oro;
+ std::map<std::string, bufferlist> kvs;
+ oro.omap_get_vals2(idata.kdata.encoded(),1,&kvs, nullptr, &err);
+ err = io_ctx.operate(index_name, &oro, NULL);
+ if (err < 0){
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-next: getting index failed with error "
+ << err << std::endl;
+ return err;
+ }
+ if (!kvs.empty()) {
+ out_data->kdata.parse(kvs.begin()->first);
+ bufferlist::iterator b = kvs.begin()->second.begin();
+ out_data->decode(b);
+ if (idata.is_timed_out(ceph_clock_now(), timeout)) {
+ if (verbose) cout << client_name << " THINKS THE OTHER CLIENT DIED."
+ << std::endl;
+ //the client died after deleting the object. clean up.
+ cleanup(idata, err);
+ }
+ } else {
+ err = -EOVERFLOW;
+ }
+ return err;
+}
+
+int KvFlatBtreeAsync::prev(const index_data &idata, index_data * out_data)
+{
+ if (verbose) cout << "\t\t" << client_name << "-prev: finding prev of "
+ << idata.str() << std::endl;
+ int err = 0;
+ bufferlist inbl;
+ idata_from_idata_args in_args;
+ in_args.idata = idata;
+ in_args.encode(inbl);
+ bufferlist outbl;
+ err = io_ctx.exec(index_name,"kvs", "get_prev_idata", inbl, outbl);
+ if (err < 0){
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-prev: getting index failed with error "
+ << err << std::endl;
+ if (idata.is_timed_out(ceph_clock_now(), timeout)) {
+ if (verbose) cout << client_name << " THINKS THE OTHER CLIENT DIED."
+ << std::endl;
+ //the client died after deleting the object. clean up.
+ err = cleanup(idata, err);
+ if (err == -ESUICIDE) {
+ return err;
+ } else {
+ err = 0;
+ }
+ }
+ return err;
+ }
+ bufferlist::iterator it = outbl.begin();
+ in_args.decode(it);
+ *out_data = in_args.next_idata;
+ if (verbose) cout << "\t\t" << client_name << "-prev: prev is "
+ << out_data->str()
+ << std::endl;
+ return err;
+}
+
+int KvFlatBtreeAsync::read_index(const string &key, index_data * idata,
+ index_data * next_idata, bool force_update) {
+ int err = 0;
+ if (!force_update) {
+ if (verbose) cout << "\t" << client_name
+ << "-read_index: getting index_data for " << key
+ << " from cache" << std::endl;
+ icache_lock.Lock();
+ if (next_idata != NULL) {
+ err = icache.get(key, idata, next_idata);
+ } else {
+ err = icache.get(key, idata);
+ }
+ icache_lock.Unlock();
+
+ if (err == 0) {
+ //if (verbose) cout << "CACHE SUCCESS" << std::endl;
+ return err;
+ } else {
+ if (verbose) cout << "NOT IN CACHE" << std::endl;
+ }
+ }
+
+ if (verbose) cout << "\t" << client_name
+ << "-read_index: getting index_data for " << key
+ << " from object" << std::endl;
+ librados::ObjectReadOperation oro;
+ bufferlist raw_val;
+ std::set<std::string> key_set;
+ key_set.insert(key_data(key).encoded());
+ std::map<std::string, bufferlist> kvmap;
+ std::map<std::string, bufferlist> dupmap;
+ oro.omap_get_vals_by_keys(key_set, &dupmap, &err);
+ oro.omap_get_vals2(key_data(key).encoded(),
+ (cache_size / cache_refresh >= 2? cache_size / cache_refresh: 2),
+ &kvmap, nullptr, &err);
+ err = io_ctx.operate(index_name, &oro, NULL);
+ utime_t mytime = ceph_clock_now();
+ if (err < 0){
+ cerr << "\t" << client_name
+ << "-read_index: getting keys failed with "
+ << err << std::endl;
+ assert(0 == client_name + "-read_index: reading index failed");
+ return err;
+ }
+ kvmap.insert(dupmap.begin(), dupmap.end());
+ for (map<string, bufferlist>::iterator it = ++kvmap.begin();
+ it != kvmap.end();
+ ++it) {
+ bufferlist bl = it->second;
+ bufferlist::iterator blit = bl.begin();
+ index_data this_idata;
+ this_idata.decode(blit);
+ if (this_idata.is_timed_out(mytime, timeout)) {
+ if (verbose) cout << client_name
+ << " THINKS THE OTHER CLIENT DIED. (mytime is "
+ << mytime.sec() << "." << mytime.usec() << ", idata.ts is "
+ << this_idata.ts.sec() << "." << this_idata.ts.usec()
+ << ", it has been " << (mytime - this_idata.ts).sec()
+ << '.' << (mytime - this_idata.ts).usec()
+ << ", timeout is " << timeout << ")" << std::endl;
+ //the client died after deleting the object. clean up.
+ if (cleanup(this_idata, -EPREFIX) == -ESUICIDE) {
+ return -ESUICIDE;
+ }
+ return read_index(key, idata, next_idata, force_update);
+ }
+ icache_lock.Lock();
+ icache.push(this_idata);
+ icache_lock.Unlock();
+ }
+ bufferlist::iterator b = kvmap.begin()->second.begin();
+ idata->decode(b);
+ idata->kdata.parse(kvmap.begin()->first);
+ if (verbose) cout << "\t" << client_name << "-read_index: kvmap_size is "
+ << kvmap.size()
+ << ", idata is " << idata->str() << std::endl;
+
+ assert(idata->obj != "");
+ icache_lock.Lock();
+ icache.push(key, *idata);
+ icache_lock.Unlock();
+
+ if (next_idata != NULL && idata->kdata.prefix != "1") {
+ next_idata->kdata.parse((++kvmap.begin())->first);
+ bufferlist::iterator nb = (++kvmap.begin())->second.begin();
+ next_idata->decode(nb);
+ icache_lock.Lock();
+ icache.push(*next_idata);
+ icache_lock.Unlock();
+ }
+ return err;
+}
+
+int KvFlatBtreeAsync::split(const index_data &idata) {
+ int err = 0;
+ opmap['l']++;
+
+ if (idata.prefix != "") {
+ return -EPREFIX;
+ }
+
+ rebalance_args args;
+ args.bound = 2 * k - 1;
+ args.comparator = CEPH_OSD_CMPXATTR_OP_GT;
+ err = read_object(idata.obj, &args);
+ args.odata.max_kdata = idata.kdata;
+ if (err < 0) {
+ if (verbose) cout << "\t\t" << client_name << "-split: read object "
+ << args.odata.name
+ << " got " << err << std::endl;
+ return err;
+ }
+
+ if (verbose) cout << "\t\t" << client_name << "-split: splitting "
+ << idata.obj
+ << ", which has size " << args.odata.size
+ << " and actual size " << args.odata.omap.size() << std::endl;
+
+ ///////preparations that happen outside the critical section
+ //for prefix index
+ vector<object_data> to_create;
+ vector<object_data> to_delete;
+ to_delete.push_back(object_data(idata.min_kdata,
+ args.odata.max_kdata, args.odata.name, args.odata.version));
+
+ //for lower half object
+ map<std::string, bufferlist>::const_iterator it = args.odata.omap.begin();
+ client_index_lock.Lock();
+ to_create.push_back(object_data(to_string(client_name, client_index++)));
+ client_index_lock.Unlock();
+ for (int i = 0; i < k; i++) {
+ to_create[0].omap.insert(*it);
+ ++it;
+ }
+ to_create[0].min_kdata = idata.min_kdata;
+ to_create[0].max_kdata = key_data(to_create[0].omap.rbegin()->first);
+
+ //for upper half object
+ client_index_lock.Lock();
+ to_create.push_back(object_data(to_create[0].max_kdata,
+ args.odata.max_kdata,
+ to_string(client_name, client_index++)));
+ client_index_lock.Unlock();
+ to_create[1].omap.insert(
+ ++args.odata.omap.find(to_create[0].omap.rbegin()->first),
+ args.odata.omap.end());
+
+ //setting up operations
+ librados::ObjectWriteOperation owos[6];
+ vector<pair<pair<int, string>, librados::ObjectWriteOperation*> > ops;
+ index_data out_data;
+ set_up_prefix_index(to_create, to_delete, &owos[0], &out_data, &err);
+ ops.push_back(make_pair(
+ pair<int, string>(ADD_PREFIX, index_name),
+ &owos[0]));
+ for (int i = 1; i < 6; i++) {
+ ops.push_back(make_pair(make_pair(0,""), &owos[i]));
+ }
+ set_up_ops(to_create, to_delete, &ops, out_data, &err);
+
+ /////BEGIN CRITICAL SECTION/////
+ //put prefix on index entry for idata.val
+ err = perform_ops("\t\t" + client_name + "-split:", out_data, &ops);
+ if (err < 0) {
+ return err;
+ }
+ if (verbose) cout << "\t\t" << client_name << "-split: done splitting."
+ << std::endl;
+ /////END CRITICAL SECTION/////
+ icache_lock.Lock();
+ for (vector<delete_data>::iterator it = out_data.to_delete.begin();
+ it != out_data.to_delete.end(); ++it) {
+ icache.erase(it->max);
+ }
+ for (vector<create_data>::iterator it = out_data.to_create.begin();
+ it != out_data.to_create.end(); ++it) {
+ icache.push(index_data(*it));
+ }
+ icache_lock.Unlock();
+ return err;
+}
+
+int KvFlatBtreeAsync::rebalance(const index_data &idata1,
+ const index_data &next_idata){
+ opmap['m']++;
+ int err = 0;
+
+ if (idata1.prefix != "") {
+ return -EPREFIX;
+ }
+
+ rebalance_args args1;
+ args1.bound = k + 1;
+ args1.comparator = CEPH_OSD_CMPXATTR_OP_LT;
+ index_data idata2 = next_idata;
+
+ rebalance_args args2;
+ args2.bound = k + 1;
+ args2.comparator = CEPH_OSD_CMPXATTR_OP_LT;
+
+ if (idata1.kdata.prefix == "1") {
+ //this is the highest key in the index, so it doesn't have a next.
+
+ //read the index for the previous entry
+ err = prev(idata1, &idata2);
+ if (err == -ERANGE) {
+ if (verbose) cout << "\t\t" << client_name
+ << "-rebalance: this is the only node, "
+ << "so aborting" << std::endl;
+ return -EUCLEAN;
+ } else if (err < 0) {
+ return err;
+ }
+
+ //read the first object
+ err = read_object(idata1.obj, &args2);
+ if (err < 0) {
+ if (verbose) cout << "reading " << idata1.obj << " failed with " << err
+ << std::endl;
+ if (err == -ENOENT) {
+ return -ECANCELED;
+ }
+ return err;
+ }
+ args2.odata.min_kdata = idata1.min_kdata;
+ args2.odata.max_kdata = idata1.kdata;
+
+ //read the second object
+ args1.bound = 2 * k + 1;
+ err = read_object(idata2.obj, &args1);
+ if (err < 0) {
+ if (verbose) cout << "reading " << idata1.obj << " failed with " << err
+ << std::endl;
+ return err;
+ }
+ args1.odata.min_kdata = idata2.min_kdata;
+ args1.odata.max_kdata = idata2.kdata;
+
+ if (verbose) cout << "\t\t" << client_name << "-rebalance: read "
+ << idata2.obj
+ << ". size: " << args1.odata.size << " version: "
+ << args1.odata.version
+ << std::endl;
+ } else {
+ assert (next_idata.obj != "");
+ //there is a next key, so get it.
+ err = read_object(idata1.obj, &args1);
+ if (err < 0) {
+ if (verbose) cout << "reading " << idata1.obj << " failed with " << err
+ << std::endl;
+ return err;
+ }
+ args1.odata.min_kdata = idata1.min_kdata;
+ args1.odata.max_kdata = idata1.kdata;
+
+ args2.bound = 2 * k + 1;
+ err = read_object(idata2.obj, &args2);
+ if (err < 0) {
+ if (verbose) cout << "reading " << idata1.obj << " failed with " << err
+ << std::endl;
+ if (err == -ENOENT) {
+ return -ECANCELED;
+ }
+ return err;
+ }
+ args2.odata.min_kdata = idata2.min_kdata;
+ args2.odata.max_kdata = idata2.kdata;
+
+ if (verbose) cout << "\t\t" << client_name << "-rebalance: read "
+ << idata2.obj
+ << ". size: " << args2.odata.size << " version: "
+ << args2.odata.version
+ << std::endl;
+ }
+
+ if (verbose) cout << "\t\t" << client_name << "-rebalance: o1 is "
+ << args1.odata.max_kdata.encoded() << ","
+ << args1.odata.name << " with size " << args1.odata.size
+ << " , o2 is " << args2.odata.max_kdata.encoded()
+ << "," << args2.odata.name << " with size " << args2.odata.size
+ << std::endl;
+
+ //calculations
+ if ((int)args1.odata.size > k && (int)args1.odata.size <= 2*k
+ && (int)args2.odata.size > k
+ && (int)args2.odata.size <= 2*k) {
+ //nothing to do
+ if (verbose) cout << "\t\t" << client_name
+ << "-rebalance: both sizes in range, so"
+ << " aborting " << std::endl;
+ return -EBALANCE;
+ } else if (idata1.prefix != "" || idata2.prefix != "") {
+ return -EPREFIX;
+ }
+
+ //this is the high object. it gets created regardless of rebalance or merge.
+ client_index_lock.Lock();
+ string o2w = to_string(client_name, client_index++);
+ client_index_lock.Unlock();
+ index_data idata;
+ vector<object_data> to_create;
+ vector<object_data> to_delete;
+ librados::ObjectWriteOperation create[2];//possibly only 1 will be used
+ librados::ObjectWriteOperation other_ops[6];
+ vector<pair<pair<int, string>, librados::ObjectWriteOperation*> > ops;
+ ops.push_back(make_pair(
+ pair<int, string>(ADD_PREFIX, index_name),
+ &other_ops[0]));
+
+ if ((int)args1.odata.size + (int)args2.odata.size <= 2*k) {
+ //merge
+ if (verbose) cout << "\t\t" << client_name << "-rebalance: merging "
+ << args1.odata.name
+ << " and " << args2.odata.name << " to get " << o2w
+ << std::endl;
+ map<string, bufferlist> write2_map;
+ write2_map.insert(args1.odata.omap.begin(), args1.odata.omap.end());
+ write2_map.insert(args2.odata.omap.begin(), args2.odata.omap.end());
+ to_create.push_back(object_data(args1.odata.min_kdata,
+ args2.odata.max_kdata, o2w, write2_map));
+ ops.push_back(make_pair(
+ pair<int, string>(MAKE_OBJECT, o2w),
+ &create[0]));
+ assert((int)write2_map.size() <= 2*k);
+ } else {
+ //rebalance
+ if (verbose) cout << "\t\t" << client_name << "-rebalance: rebalancing "
+ << args1.odata.name
+ << " and " << args2.odata.name << std::endl;
+ map<std::string, bufferlist> write1_map;
+ map<std::string, bufferlist> write2_map;
+ map<std::string, bufferlist>::iterator it;
+ client_index_lock.Lock();
+ string o1w = to_string(client_name, client_index++);
+ client_index_lock.Unlock();
+ int target_size_1 = ceil(((int)args1.odata.size + (int)args2.odata.size)
+ / 2.0);
+ if (args1.odata.max_kdata != idata1.kdata) {
+ //this should be true if idata1 is the high object
+ target_size_1 = floor(((int)args1.odata.size + (int)args2.odata.size)
+ / 2.0);
+ }
+ for (it = args1.odata.omap.begin();
+ it != args1.odata.omap.end() && (int)write1_map.size()
+ < target_size_1;
+ ++it) {
+ write1_map.insert(*it);
+ }
+ if (it != args1.odata.omap.end()){
+ //write1_map is full, so put the rest in write2_map
+ write2_map.insert(it, args1.odata.omap.end());
+ write2_map.insert(args2.odata.omap.begin(), args2.odata.omap.end());
+ } else {
+ //args1.odata.omap was small, and write2_map still needs more
+ map<std::string, bufferlist>::iterator it2;
+ for(it2 = args2.odata.omap.begin();
+ (it2 != args2.odata.omap.end()) && ((int)write1_map.size()
+ < target_size_1);
+ ++it2) {
+ write1_map.insert(*it2);
+ }
+ write2_map.insert(it2, args2.odata.omap.end());
+ }
+ if (verbose) cout << "\t\t" << client_name
+ << "-rebalance: write1_map has size "
+ << write1_map.size() << ", write2_map.size() is " << write2_map.size()
+ << std::endl;
+ //at this point, write1_map and write2_map should have the correct pairs
+ to_create.push_back(object_data(args1.odata.min_kdata,
+ key_data(write1_map.rbegin()->first),
+ o1w,write1_map));
+ to_create.push_back(object_data( key_data(write1_map.rbegin()->first),
+ args2.odata.max_kdata, o2w, write2_map));
+ ops.push_back(make_pair(
+ pair<int, string>(MAKE_OBJECT, o1w),
+ &create[0]));
+ ops.push_back(make_pair(
+ pair<int, string>(MAKE_OBJECT, o2w),
+ &create[1]));
+ }
+
+ to_delete.push_back(object_data(args1.odata.min_kdata,
+ args1.odata.max_kdata, args1.odata.name, args1.odata.version));
+ to_delete.push_back(object_data(args2.odata.min_kdata,
+ args2.odata.max_kdata, args2.odata.name, args2.odata.version));
+ for (int i = 1; i < 6; i++) {
+ ops.push_back(make_pair(make_pair(0,""), &other_ops[i]));
+ }
+
+ index_data out_data;
+ set_up_prefix_index(to_create, to_delete, &other_ops[0], &out_data, &err);
+ set_up_ops(to_create, to_delete, &ops, out_data, &err);
+
+ //at this point, all operations should be completely set up.
+ /////BEGIN CRITICAL SECTION/////
+ err = perform_ops("\t\t" + client_name + "-rebalance:", out_data, &ops);
+ if (err < 0) {
+ return err;
+ }
+ icache_lock.Lock();
+ for (vector<delete_data>::iterator it = out_data.to_delete.begin();
+ it != out_data.to_delete.end(); ++it) {
+ icache.erase(it->max);
+ }
+ for (vector<create_data>::iterator it = out_data.to_create.begin();
+ it != out_data.to_create.end(); ++it) {
+ icache.push(index_data(*it));
+ }
+ icache_lock.Unlock();
+ if (verbose) cout << "\t\t" << client_name << "-rebalance: done rebalancing."
+ << std::endl;
+ /////END CRITICAL SECTION/////
+ return err;
+}
+
+int KvFlatBtreeAsync::read_object(const string &obj, object_data * odata) {
+ librados::ObjectReadOperation get_obj;
+ librados::AioCompletion * obj_aioc = rados.aio_create_completion();
+ int err;
+ bufferlist unw_bl;
+ odata->name = obj;
+ get_obj.omap_get_vals2("", LONG_MAX, &odata->omap, nullptr, &err);
+ get_obj.getxattr("unwritable", &unw_bl, &err);
+ io_ctx.aio_operate(obj, obj_aioc, &get_obj, NULL);
+ obj_aioc->wait_for_safe();
+ err = obj_aioc->get_return_value();
+ if (err < 0){
+ //possibly -ENOENT, meaning someone else deleted it.
+ obj_aioc->release();
+ return err;
+ }
+ odata->unwritable = string(unw_bl.c_str(), unw_bl.length()) == "1";
+ odata->version = obj_aioc->get_version64();
+ odata->size = odata->omap.size();
+ obj_aioc->release();
+ return 0;
+}
+
+int KvFlatBtreeAsync::read_object(const string &obj, rebalance_args * args) {
+ bufferlist inbl;
+ args->encode(inbl);
+ bufferlist outbl;
+ int err;
+ librados::AioCompletion * a = rados.aio_create_completion();
+ io_ctx.aio_exec(obj, a, "kvs", "maybe_read_for_balance", inbl, &outbl);
+ a->wait_for_safe();
+ err = a->get_return_value();
+ if (err < 0) {
+ if (verbose) cout << "\t\t" << client_name
+ << "-read_object: reading failed with "
+ << err << std::endl;
+ a->release();
+ return err;
+ }
+ bufferlist::iterator it = outbl.begin();
+ args->decode(it);
+ args->odata.name = obj;
+ args->odata.version = a->get_version64();
+ a->release();
+ return err;
+}
+
+void KvFlatBtreeAsync::set_up_prefix_index(
+ const vector<object_data> &to_create,
+ const vector<object_data> &to_delete,
+ librados::ObjectWriteOperation * owo,
+ index_data * idata,
+ int * err) {
+ std::map<std::string, pair<bufferlist, int> > assertions;
+ map<string, bufferlist> to_insert;
+ idata->prefix = "1";
+ idata->ts = ceph_clock_now();
+ for(vector<object_data>::const_iterator it = to_create.begin();
+ it != to_create.end();
+ ++it) {
+ create_data c(it->min_kdata, it->max_kdata, it->name);
+ idata->to_create.push_back(c);
+ }
+ for(vector<object_data>::const_iterator it = to_delete.begin();
+ it != to_delete.end();
+ ++it) {
+ delete_data d(it->min_kdata, it->max_kdata, it->name, it->version);
+ idata->to_delete.push_back(d);
+ }
+ for(vector<object_data>::const_iterator it = to_delete.begin();
+ it != to_delete.end();
+ ++it) {
+ idata->obj = it->name;
+ idata->min_kdata = it->min_kdata;
+ idata->kdata = it->max_kdata;
+ bufferlist insert;
+ idata->encode(insert);
+ to_insert[it->max_kdata.encoded()] = insert;
+ index_data this_entry;
+ this_entry.min_kdata = idata->min_kdata;
+ this_entry.kdata = idata->kdata;
+ this_entry.obj = idata->obj;
+ assertions[it->max_kdata.encoded()] = pair<bufferlist, int>
+ (to_bl(this_entry), CEPH_OSD_CMPXATTR_OP_EQ);
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-setup_prefix: will assert "
+ << this_entry.str() << std::endl;
+ }
+ assert(*err == 0);
+ owo->omap_cmp(assertions, err);
+ if (to_create.size() <= 2) {
+ owo->omap_set(to_insert);
+ }
+}
+
+//some args can be null if there are no corresponding entries in p
+void KvFlatBtreeAsync::set_up_ops(
+ const vector<object_data> &create_vector,
+ const vector<object_data> &delete_vector,
+ vector<pair<pair<int, string>, librados::ObjectWriteOperation*> > * ops,
+ const index_data &idata,
+ int * err) {
+ vector<pair<pair<int, string>,
+ librados::ObjectWriteOperation* > >::iterator it;
+
+ //skip the prefixing part
+ for(it = ops->begin(); it->first.first == ADD_PREFIX; ++it) {}
+ map<string, bufferlist> to_insert;
+ std::set<string> to_remove;
+ map<string, pair<bufferlist, int> > assertions;
+ if (create_vector.size() > 0) {
+ for (int i = 0; i < (int)idata.to_delete.size(); ++i) {
+ it->first = pair<int, string>(UNWRITE_OBJECT, idata.to_delete[i].obj);
+ set_up_unwrite_object(delete_vector[i].version, it->second);
+ ++it;
+ }
+ }
+ for (int i = 0; i < (int)idata.to_create.size(); ++i) {
+ index_data this_entry(idata.to_create[i].max, idata.to_create[i].min,
+ idata.to_create[i].obj);
+ to_insert[idata.to_create[i].max.encoded()] = to_bl(this_entry);
+ if (idata.to_create.size() <= 2) {
+ it->first = pair<int, string>(MAKE_OBJECT, idata.to_create[i].obj);
+ } else {
+ it->first = pair<int, string>(AIO_MAKE_OBJECT, idata.to_create[i].obj);
+ }
+ set_up_make_object(create_vector[i].omap, it->second);
+ ++it;
+ }
+ for (int i = 0; i < (int)idata.to_delete.size(); ++i) {
+ index_data this_entry = idata;
+ this_entry.obj = idata.to_delete[i].obj;
+ this_entry.min_kdata = idata.to_delete[i].min;
+ this_entry.kdata = idata.to_delete[i].max;
+ if (verbose) cout << "\t\t\t" << client_name << "-setup_ops: will assert "
+ << this_entry.str() << std::endl;
+ assertions[idata.to_delete[i].max.encoded()] = pair<bufferlist, int>(
+ to_bl(this_entry), CEPH_OSD_CMPXATTR_OP_EQ);
+ to_remove.insert(idata.to_delete[i].max.encoded());
+ it->first = pair<int, string>(REMOVE_OBJECT, idata.to_delete[i].obj);
+ set_up_delete_object(it->second);
+ ++it;
+ }
+ if ((int)idata.to_create.size() <= 2) {
+ it->second->omap_cmp(assertions, err);
+ }
+ it->second->omap_rm_keys(to_remove);
+ it->second->omap_set(to_insert);
+
+
+ it->first = pair<int, string>(REMOVE_PREFIX, index_name);
+}
+
+void KvFlatBtreeAsync::set_up_make_object(
+ const map<std::string, bufferlist> &to_set,
+ librados::ObjectWriteOperation *owo) {
+ bufferlist inbl;
+ ::encode(to_set, inbl);
+ owo->exec("kvs", "create_with_omap", inbl);
+}
+
+void KvFlatBtreeAsync::set_up_unwrite_object(
+ const int &ver, librados::ObjectWriteOperation *owo) {
+ if (ver > 0) {
+ owo->assert_version(ver);
+ }
+ owo->cmpxattr("unwritable", CEPH_OSD_CMPXATTR_OP_EQ, to_bl("0"));
+ owo->setxattr("unwritable", to_bl("1"));
+}
+
+void KvFlatBtreeAsync::set_up_restore_object(
+ librados::ObjectWriteOperation *owo) {
+ owo->cmpxattr("unwritable", CEPH_OSD_CMPXATTR_OP_EQ, to_bl("1"));
+ owo->setxattr("unwritable", to_bl("0"));
+}
+
+void KvFlatBtreeAsync::set_up_delete_object(
+ librados::ObjectWriteOperation *owo) {
+ owo->cmpxattr("unwritable", CEPH_OSD_CMPXATTR_OP_EQ, to_bl("1"));
+ owo->remove();
+}
+
+int KvFlatBtreeAsync::perform_ops(const string &debug_prefix,
+ const index_data &idata,
+ vector<pair<pair<int, string>, librados::ObjectWriteOperation*> > *ops) {
+ int err = 0;
+ vector<librados::AioCompletion*> aiocs(idata.to_create.size());
+ int count = 0;
+ for (vector<pair<pair<int, string>,
+ librados::ObjectWriteOperation*> >::iterator it = ops->begin();
+ it != ops->end(); ++it) {
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ return -ESUICIDE;
+ }
+ switch (it->first.first) {
+ case ADD_PREFIX://prefixing
+ if (verbose) cout << debug_prefix << " adding prefix" << std::endl;
+ err = io_ctx.operate(index_name, it->second);
+ if (err < 0) {
+ if (verbose) cout << debug_prefix << " prefixing the index failed with "
+ << err << std::endl;
+ return -EPREFIX;
+ }
+ if (verbose) cout << debug_prefix << " prefix added." << std::endl;
+ break;
+ case UNWRITE_OBJECT://marking
+ if (verbose) cout << debug_prefix << " marking " << it->first.second
+ << std::endl;
+ err = io_ctx.operate(it->first.second, it->second);
+ if (err < 0) {
+ //most likely because it changed, in which case it will be -ERANGE
+ if (verbose) cout << debug_prefix << " marking " << it->first.second
+ << "failed with code" << err << std::endl;
+ if (it->first.second == (*idata.to_delete.begin()).max.encoded()) {
+ if (cleanup(idata, -EFIRSTOBJ) == -ESUICIDE) {
+ return -ESUICIDE;
+ }
+ } else {
+ if (cleanup(idata, -ERANGE) == -ESUICIDE) {
+ return -ESUICIDE;
+ }
+ }
+ return err;
+ }
+ if (verbose) cout << debug_prefix << " marked " << it->first.second
+ << std::endl;
+ break;
+ case MAKE_OBJECT://creating
+ if (verbose) cout << debug_prefix << " creating " << it->first.second
+ << std::endl;
+ err = io_ctx.operate(it->first.second, it->second);
+ if (err < 0) {
+ //this can happen if someone else was cleaning up after us.
+ if (verbose) cout << debug_prefix << " creating " << it->first.second
+ << " failed"
+ << " with code " << err << std::endl;
+ if (err == -EEXIST) {
+ //someone thinks we died, so die
+ if (verbose) cout << client_name << " is suiciding!" << std::endl;
+ return -ESUICIDE;
+ } else {
+ assert(false);
+ }
+ return err;
+ }
+ if (verbose || idata.to_create.size() > 2) {
+ cout << debug_prefix << " created object " << it->first.second
+ << std::endl;
+ }
+ break;
+ case AIO_MAKE_OBJECT:
+ cout << debug_prefix << " launching asynchronous create "
+ << it->first.second << std::endl;
+ aiocs[count] = rados.aio_create_completion();
+ io_ctx.aio_operate(it->first.second, aiocs[count], it->second);
+ count++;
+ if ((int)idata.to_create.size() == count) {
+ cout << "starting aiowrite waiting loop" << std::endl;
+ for (count -= 1; count >= 0; count--) {
+ aiocs[count]->wait_for_safe();
+ err = aiocs[count]->get_return_value();
+ if (err < 0) {
+ //this can happen if someone else was cleaning up after us.
+ cerr << debug_prefix << " a create failed"
+ << " with code " << err << std::endl;
+ if (err == -EEXIST) {
+ //someone thinks we died, so die
+ cerr << client_name << " is suiciding!" << std::endl;
+ return -ESUICIDE;
+ } else {
+ assert(false);
+ }
+ return err;
+ }
+ if (verbose || idata.to_create.size() > 2) {
+ cout << debug_prefix << " completed aio " << aiocs.size() - count
+ << "/" << aiocs.size() << std::endl;
+ }
+ }
+ }
+ break;
+ case REMOVE_OBJECT://deleting
+ if (verbose) cout << debug_prefix << " deleting " << it->first.second
+ << std::endl;
+ err = io_ctx.operate(it->first.second, it->second);
+ if (err < 0) {
+ //if someone else called cleanup on this prefix first
+ if (verbose) cout << debug_prefix << " deleting " << it->first.second
+ << "failed with code" << err << std::endl;
+ }
+ if (verbose) cout << debug_prefix << " deleted " << it->first.second
+ << std::endl;
+ break;
+ case REMOVE_PREFIX://rewriting index
+ if (verbose) cout << debug_prefix << " updating index " << std::endl;
+ err = io_ctx.operate(index_name, it->second);
+ if (err < 0) {
+ if (verbose) cout << debug_prefix
+ << " rewriting the index failed with code " << err
+ << ". someone else must have thought we died, so dying" << std::endl;
+ return -ETIMEDOUT;
+ }
+ if (verbose) cout << debug_prefix << " updated index." << std::endl;
+ break;
+ case RESTORE_OBJECT:
+ if (verbose) cout << debug_prefix << " restoring " << it->first.second
+ << std::endl;
+ err = io_ctx.operate(it->first.second, it->second);
+ if (err < 0) {
+ if (verbose) cout << debug_prefix << "restoring " << it->first.second
+ << " failed"
+ << " with " << err << std::endl;
+ return err;
+ }
+ if (verbose) cout << debug_prefix << " restored " << it->first.second
+ << std::endl;
+ break;
+ default:
+ if (verbose) cout << debug_prefix << " performing unknown op on "
+ << it->first.second
+ << std::endl;
+ err = io_ctx.operate(index_name, it->second);
+ if (err < 0) {
+ if (verbose) cout << debug_prefix << " unknown op on "
+ << it->first.second
+ << " failed with " << err << std::endl;
+ return err;
+ }
+ if (verbose) cout << debug_prefix << " unknown op on "
+ << it->first.second
+ << " succeeded." << std::endl;
+ break;
+ }
+ }
+
+ return err;
+}
+
+int KvFlatBtreeAsync::cleanup(const index_data &idata, const int &error) {
+ if (verbose) cout << "\t\t" << client_name << ": cleaning up after "
+ << idata.str()
+ << std::endl;
+ int err = 0;
+ assert(idata.prefix != "");
+ map<std::string,bufferlist> new_index;
+ map<std::string, pair<bufferlist, int> > assertions;
+ switch (error) {
+ case -EFIRSTOBJ: {
+ //this happens if the split or rebalance failed to mark the first object,
+ //meaning only the index needs to be changed.
+ //restore objects that had been marked unwritable.
+ for(vector<delete_data >::const_iterator it =
+ idata.to_delete.begin();
+ it != idata.to_delete.end(); ++it) {
+ index_data this_entry;
+ this_entry.obj = (*it).obj;
+ this_entry.min_kdata = it->min;
+ this_entry.kdata = it->max;
+ new_index[it->max.encoded()] = to_bl(this_entry);
+ this_entry = idata;
+ this_entry.obj = it->obj;
+ this_entry.min_kdata = it->min;
+ this_entry.kdata = it->max;
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: will assert index contains "
+ << this_entry.str() << std::endl;
+ assertions[it->max.encoded()] =
+ pair<bufferlist, int>(to_bl(this_entry),
+ CEPH_OSD_CMPXATTR_OP_EQ);
+ }
+
+ //update the index
+ librados::ObjectWriteOperation update_index;
+ update_index.omap_cmp(assertions, &err);
+ update_index.omap_set(new_index);
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: updating index"
+ << std::endl;
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ return -ESUICIDE;
+ }
+ err = io_ctx.operate(index_name, &update_index);
+ if (err < 0) {
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: rewriting failed with "
+ << err << ". returning -ECANCELED" << std::endl;
+ return -ECANCELED;
+ }
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: updated index. cleanup done."
+ << std::endl;
+ break;
+ }
+ case -ERANGE: {
+ //this happens if a split or rebalance fails to mark an object. It is a
+ //special case of rolling back that does not have to deal with new objects.
+
+ //restore objects that had been marked unwritable.
+ vector<delete_data >::const_iterator it;
+ for(it = idata.to_delete.begin();
+ it != idata.to_delete.end(); ++it) {
+ index_data this_entry;
+ this_entry.obj = (*it).obj;
+ this_entry.min_kdata = it->min;
+ this_entry.kdata = it->max;
+ new_index[it->max.encoded()] = to_bl(this_entry);
+ this_entry = idata;
+ this_entry.obj = it->obj;
+ this_entry.min_kdata = it->min;
+ this_entry.kdata = it->max;
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: will assert index contains "
+ << this_entry.str() << std::endl;
+ assertions[it->max.encoded()] =
+ pair<bufferlist, int>(to_bl(this_entry),
+ CEPH_OSD_CMPXATTR_OP_EQ);
+ }
+ it = idata.to_delete.begin();
+ librados::ObjectWriteOperation restore;
+ set_up_restore_object(&restore);
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: restoring "
+ << it->obj
+ << std::endl;
+ err = io_ctx.operate(it->obj, &restore);
+ if (err < 0) {
+ //i.e., -ECANCELED because the object was already restored by someone
+ //else
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: restoring "
+ << it->obj
+ << " failed with " << err << std::endl;
+ } else {
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: restored "
+ << it->obj
+ << std::endl;
+ }
+
+ //update the index
+ librados::ObjectWriteOperation update_index;
+ update_index.omap_cmp(assertions, &err);
+ update_index.omap_set(new_index);
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: updating index"
+ << std::endl;
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ return -ESUICIDE;
+ }
+ err = io_ctx.operate(index_name, &update_index);
+ if (err < 0) {
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: rewriting failed with "
+ << err << ". returning -ECANCELED" << std::endl;
+ return -ECANCELED;
+ }
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: updated index. cleanup done."
+ << std::endl;
+ break;
+ }
+ case -ENOENT: {
+ if (verbose) cout << "\t\t" << client_name << "-cleanup: rolling forward"
+ << std::endl;
+ //all changes were created except for updating the index and possibly
+ //deleting the objects. roll forward.
+ vector<pair<pair<int, string>, librados::ObjectWriteOperation*> > ops;
+ vector<librados::ObjectWriteOperation> owos(idata.to_delete.size() + 1);
+ for (int i = 0; i <= (int)idata.to_delete.size(); ++i) {
+ ops.push_back(make_pair(pair<int, string>(0, ""), &owos[i]));
+ }
+ set_up_ops(vector<object_data>(),
+ vector<object_data>(), &ops, idata, &err);
+ err = perform_ops("\t\t" + client_name + "-cleanup:", idata, &ops);
+ if (err < 0) {
+ if (err == -ESUICIDE) {
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: rewriting failed with "
+ << err << ". returning -ECANCELED" << std::endl;
+ return -ECANCELED;
+ }
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: updated index"
+ << std::endl;
+ break;
+ }
+ default: {
+ //roll back all changes.
+ if (verbose) cout << "\t\t" << client_name << "-cleanup: rolling back"
+ << std::endl;
+ map<std::string,bufferlist> new_index;
+ std::set<string> to_remove;
+ map<std::string, pair<bufferlist, int> > assertions;
+
+ //mark the objects to be created. if someone else already has, die.
+ for(vector<create_data >::const_reverse_iterator it =
+ idata.to_create.rbegin();
+ it != idata.to_create.rend(); ++it) {
+ librados::ObjectWriteOperation rm;
+ set_up_unwrite_object(0, &rm);
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 )
+ {
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: marking "
+ << it->obj
+ << std::endl;
+ err = io_ctx.operate(it->obj, &rm);
+ if (err < 0) {
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: marking "
+ << it->obj
+ << " failed with " << err << std::endl;
+ } else {
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: marked "
+ << it->obj
+ << std::endl;
+ }
+ }
+
+ //restore objects that had been marked unwritable.
+ for(vector<delete_data >::const_iterator it =
+ idata.to_delete.begin();
+ it != idata.to_delete.end(); ++it) {
+ index_data this_entry;
+ this_entry.obj = (*it).obj;
+ this_entry.min_kdata = it->min;
+ this_entry.kdata = it->max;
+ new_index[it->max.encoded()] = to_bl(this_entry);
+ this_entry = idata;
+ this_entry.obj = it->obj;
+ this_entry.min_kdata = it->min;
+ this_entry.kdata = it->max;
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: will assert index contains "
+ << this_entry.str() << std::endl;
+ assertions[it->max.encoded()] =
+ pair<bufferlist, int>(to_bl(this_entry),
+ CEPH_OSD_CMPXATTR_OP_EQ);
+ librados::ObjectWriteOperation restore;
+ set_up_restore_object(&restore);
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: will assert index contains "
+ << this_entry.str() << std::endl;
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 )
+ {
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: restoring "
+ << it->obj
+ << std::endl;
+ err = io_ctx.operate(it->obj, &restore);
+ if (err == -ENOENT) {
+ //it had gotten far enough to be rolled forward - unmark the objects
+ //and roll forward.
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: roll forward instead"
+ << std::endl;
+ for(vector<create_data >::const_iterator cit =
+ idata.to_create.begin();
+ cit != idata.to_create.end(); ++cit) {
+ librados::ObjectWriteOperation res;
+ set_up_restore_object(&res);
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)()
+ == 1 ) {
+ return -ECANCELED;
+ }
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: restoring " << cit->obj
+ << std::endl;
+ err = io_ctx.operate(cit->obj, &res);
+ if (err < 0) {
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: restoring "
+ << cit->obj << " failed with " << err << std::endl;
+ }
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: restored "
+ << cit->obj
+ << std::endl;
+ }
+ return cleanup(idata, -ENOENT);
+ } else if (err < 0) {
+ //i.e., -ECANCELED because the object was already restored by someone
+ //else
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: restoring " << it->obj
+ << " failed with " << err << std::endl;
+ } else {
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: restored "
+ << it->obj
+ << std::endl;
+ }
+ }
+
+ //remove the new objects
+ for(vector<create_data >::const_reverse_iterator it =
+ idata.to_create.rbegin();
+ it != idata.to_create.rend(); ++it) {
+ to_remove.insert(it->max.encoded());
+ librados::ObjectWriteOperation rm;
+ rm.remove();
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 )
+ {
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: removing "
+ << it->obj
+ << std::endl;
+ err = io_ctx.operate(it->obj, &rm);
+ if (err < 0) {
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: failed to remove "
+ << it->obj << std::endl;
+ } else {
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: removed "
+ << it->obj
+ << std::endl;
+ }
+ }
+
+ //update the index
+ librados::ObjectWriteOperation update_index;
+ update_index.omap_cmp(assertions, &err);
+ update_index.omap_rm_keys(to_remove);
+ update_index.omap_set(new_index);
+ if (verbose) cout << "\t\t\t" << client_name << "-cleanup: updating index"
+ << std::endl;
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ return -ESUICIDE;
+ }
+ err = io_ctx.operate(index_name, &update_index);
+ if (err < 0) {
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: rewriting failed with "
+ << err << ". returning -ECANCELED" << std::endl;
+ return -ECANCELED;
+ }
+ if (verbose) cout << "\t\t\t" << client_name
+ << "-cleanup: updated index. cleanup done."
+ << std::endl;
+ break;
+ }
+ }
+ return err;
+}
+
+string KvFlatBtreeAsync::to_string(string s, int i) {
+ stringstream ret;
+ ret << s << i;
+ return ret.str();
+}
+
+string KvFlatBtreeAsync::get_name() {
+ return rados_id;
+}
+
+void KvFlatBtreeAsync::set_inject(injection_t inject, int wait_time) {
+ interrupt = inject;
+ wait_ms = wait_time;
+}
+
+int KvFlatBtreeAsync::setup(int argc, const char** argv) {
+ int r = rados.init(rados_id.c_str());
+ if (r < 0) {
+ cerr << "error during init" << r << std::endl;
+ return r;
+ }
+ r = rados.conf_parse_argv(argc, argv);
+ if (r < 0) {
+ cerr << "error during parsing args" << r << std::endl;
+ return r;
+ }
+ r = rados.conf_parse_env(NULL);
+ if (r < 0) {
+ cerr << "error during parsing env" << r << std::endl;
+ return r;
+ }
+ r = rados.conf_read_file(NULL);
+ if (r < 0) {
+ cerr << "error during read file: " << r << std::endl;
+ return r;
+ }
+ r = rados.connect();
+ if (r < 0) {
+ cerr << "error during connect: " << r << std::endl;
+ return r;
+ }
+ r = rados.ioctx_create(pool_name.c_str(), io_ctx);
+ if (r < 0) {
+ cerr << "error creating io ctx: " << r << std::endl;
+ rados.shutdown();
+ return r;
+ }
+
+ librados::ObjectWriteOperation make_index;
+ make_index.create(true);
+ map<std::string,bufferlist> index_map;
+ index_data idata;
+ idata.obj = client_name;
+ idata.min_kdata.raw_key = "";
+ idata.kdata = key_data("");
+ index_map["1"] = to_bl(idata);
+ make_index.omap_set(index_map);
+ r = io_ctx.operate(index_name, &make_index);
+ if (r < 0) {
+ if (verbose) cout << client_name << ": Making the index failed with code "
+ << r
+ << std::endl;
+ return 0;
+ }
+ if (verbose) cout << client_name << ": created index object" << std::endl;
+
+ librados::ObjectWriteOperation make_max_obj;
+ make_max_obj.create(true);
+ make_max_obj.setxattr("unwritable", to_bl("0"));
+ make_max_obj.setxattr("size", to_bl("0"));
+ r = io_ctx.operate(client_name, &make_max_obj);
+ if (r < 0) {
+ if (verbose) cout << client_name << ": Setting xattr failed with code "
+ << r
+ << std::endl;
+ }
+
+ return 0;
+}
+
+int KvFlatBtreeAsync::set(const string &key, const bufferlist &val,
+ bool update_on_existing) {
+ if (verbose) cout << client_name << " is "
+ << (update_on_existing? "updating " : "setting ")
+ << key << std::endl;
+ int err = 0;
+ utime_t mytime;
+ index_data idata(key);
+
+ if (verbose) cout << "\t" << client_name << ": finding oid" << std::endl;
+ err = read_index(key, &idata, NULL, false);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ if (verbose) cout << "\t" << client_name << ": index data is " << idata.str()
+ << ", object is " << idata.obj << std::endl;
+
+ err = set_op(key, val, update_on_existing, idata);
+
+ if (verbose) cout << "\t" << client_name << ": finished set with " << err
+ << std::endl;
+ return err;
+}
+
+int KvFlatBtreeAsync::set_op(const string &key, const bufferlist &val,
+ bool update_on_existing, index_data &idata) {
+ //write
+
+ bufferlist inbl;
+ omap_set_args args;
+ args.bound = 2 * k;
+ args.exclusive = !update_on_existing;
+ args.omap[key] = val;
+ args.encode(inbl);
+
+ librados::ObjectWriteOperation owo;
+ owo.exec("kvs", "omap_insert", inbl);
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ if (verbose) cout << client_name << " IS SUICIDING!" << std::endl;
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t" << client_name << ": inserting " << key
+ << " into object "
+ << idata.obj << std::endl;
+ int err = io_ctx.operate(idata.obj, &owo);
+ if (err < 0) {
+ switch (err) {
+ case -EEXIST: {
+ //the key already exists and this is an exclusive insert.
+ cerr << "\t" << client_name << ": writing key failed with "
+ << err << std::endl;
+ return err;
+ }
+ case -EKEYREJECTED: {
+ //the object needs to be split.
+ do {
+ if (verbose) cout << "\t" << client_name << ": running split on "
+ << idata.obj
+ << std::endl;
+ err = read_index(key, &idata, NULL, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ err = split(idata);
+ if (err < 0 && err != -ENOENT && err != -EBALANCE) {
+ if (verbose) cerr << "\t" << client_name << ": split failed with "
+ << err << std::endl;
+ int ret = handle_set_rm_errors(err, idata.obj, key, &idata, NULL);
+ switch (ret) {
+ case -ESUICIDE:
+ if (verbose) cout << client_name << " IS SUICIDING!" << std::endl;
+ return ret;
+ break;
+ case 1:
+ return set_op(key, val, update_on_existing, idata);
+ break;
+ case 2:
+ return err;
+ break;
+ }
+ }
+ } while (err < 0 && err != -EBALANCE && err != -ENOENT);
+ err = read_index(key, &idata, NULL, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ return set_op(key, val, update_on_existing, idata);
+ }
+ default:
+ if (verbose) cerr << "\t" << client_name << ": writing obj failed with "
+ << err << std::endl;
+ if (err == -ENOENT || err == -EACCES) {
+ if (err == -ENOENT) {
+ if (verbose) cout << "CACHE FAILURE" << std::endl;
+ }
+ err = read_index(key, &idata, NULL, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ if (verbose) cout << "\t" << client_name << ": index data is "
+ << idata.str()
+ << ", object is " << idata.obj << std::endl;
+ return set_op(key, val, update_on_existing, idata);
+ } else {
+ return err;
+ }
+ }
+ }
+ return 0;
+}
+
+int KvFlatBtreeAsync::remove(const string &key) {
+ if (verbose) cout << client_name << ": removing " << key << std::endl;
+ int err = 0;
+ string obj;
+ utime_t mytime;
+ index_data idata;
+ index_data next_idata;
+
+ if (verbose) cout << "\t" << client_name << ": finding oid" << std::endl;
+ err = read_index(key, &idata, &next_idata, false);
+ if (err < 0) {
+ if (verbose) cout << "getting oid failed with code " << err << std::endl;
+ return err;
+ }
+ obj = idata.obj;
+ if (verbose) cout << "\t" << client_name << ": idata is " << idata.str()
+ << ", next_idata is " << next_idata.str()
+ << ", obj is " << obj << std::endl;
+
+ err = remove_op(key, idata, next_idata);
+
+ if (verbose) cout << "\t" << client_name << ": finished remove with " << err
+ << " and exiting" << std::endl;
+ return err;
+}
+
+int KvFlatBtreeAsync::remove_op(const string &key, index_data &idata,
+ index_data &next_idata) {
+ //write
+ bufferlist inbl;
+ omap_rm_args args;
+ args.bound = k;
+ args.omap.insert(key);
+ args.encode(inbl);
+
+ librados::ObjectWriteOperation owo;
+ owo.exec("kvs", "omap_remove", inbl);
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ if (verbose) cout << client_name << " IS SUICIDING!" << std::endl;
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t" << client_name << ": removing " << key << " from "
+ << idata.obj
+ << std::endl;
+ int err = io_ctx.operate(idata.obj, &owo);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name << ": writing obj failed with "
+ << err << std::endl;
+ switch (err) {
+ case -ENODATA: {
+ //the key does not exist in the object
+ return err;
+ }
+ case -EKEYREJECTED: {
+ //the object needs to be split.
+ do {
+ if (verbose) cerr << "\t" << client_name << ": running rebalance on "
+ << idata.obj << std::endl;
+ err = read_index(key, &idata, &next_idata, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ err = rebalance(idata, next_idata);
+ if (err < 0 && err != -ENOENT && err != -EBALANCE) {
+ if (verbose) cerr << "\t" << client_name << ": rebalance returned "
+ << err << std::endl;
+ int ret = handle_set_rm_errors(err, idata.obj, key, &idata,
+ &next_idata);
+ switch (ret) {
+ case -ESUICIDE:
+ if (verbose) cout << client_name << " IS SUICIDING!" << std::endl;
+ return err;
+ break;
+ case 1:
+ return remove_op(key, idata, next_idata);
+ break;
+ case 2:
+ return err;
+ break;
+ case -EUCLEAN:
+ //this is the only node, so it's ok to go below k.
+ librados::ObjectWriteOperation owo;
+ bufferlist inbl;
+ omap_rm_args args;
+ args.bound = 0;
+ args.omap.insert(key);
+ args.encode(inbl);
+ owo.exec("kvs", "omap_remove", inbl);
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)()
+ == 1 ) {
+ if (verbose) cout << client_name << " IS SUICIDING!"
+ << std::endl;
+ return -ESUICIDE;
+ }
+ if (verbose) cout << "\t" << client_name << ": removing " << key
+ << " from "
+ << idata.obj
+ << std::endl;
+ int err = io_ctx.operate(idata.obj, &owo);
+ if (err == 0) {
+ return 0;
+ }
+ }
+ }
+ } while (err < 0 && err != -EBALANCE && err != -ENOENT);
+ err = read_index(key, &idata, &next_idata, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ return remove(key);
+ }
+ default:
+ if (err == -ENOENT || err == -EACCES) {
+ err = read_index(key, &idata, &next_idata, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ if (verbose) cout << "\t" << client_name << ": index data is "
+ << idata.str()
+ << ", object is " << idata.obj << std::endl;
+ //idea: we read the time every time we read the index anyway - store it.
+ return remove_op(key, idata, next_idata);
+ } else {
+ return err;
+ }
+ }
+ }
+ return 0;
+}
+
+int KvFlatBtreeAsync::handle_set_rm_errors(int &err, string obj,
+ string key,
+ index_data * idata, index_data * next_idata) {
+ if (err == -ESUICIDE) {
+ return err;
+ } else if (err == -ECANCELED //if an object was unwritable or index changed
+ || err == -EPREFIX //if there is currently a prefix
+ || err == -ETIMEDOUT// if the index changes during the op - i.e. cleanup
+ || err == -EACCES) //possible if we were acting on old index data
+ {
+ err = read_index(key, idata, next_idata, true);
+ if (err < 0) {
+ return err;
+ }
+ if (verbose) cout << "\t" << client_name << ": prefix is " << idata->str()
+ << std::endl;
+ if (idata->obj != obj) {
+ //someone else has split or cleaned up or something. start over.
+ return 1;//meaning repeat
+ }
+ } else if (err != -ETIMEDOUT && err != -ERANGE && err != -EACCES
+ && err != -EUCLEAN){
+ if (verbose) cout << "\t" << client_name
+ << ": split encountered an unexpected error: " << err
+ << std::endl;
+ return 2;
+ }
+ return err;
+}
+
+int KvFlatBtreeAsync::get(const string &key, bufferlist *val) {
+ opmap['g']++;
+ if (verbose) cout << client_name << ": getting " << key << std::endl;
+ int err = 0;
+ index_data idata;
+ utime_t mytime;
+
+ if ((((KeyValueStructure *)this)->*KvFlatBtreeAsync::interrupt)() == 1 ) {
+ return -ESUICIDE;
+ }
+ err = read_index(key, &idata, NULL, false);
+ mytime = ceph_clock_now();
+ if (err < 0) {
+ if (verbose) cout << "getting oid failed with code " << err << std::endl;
+ return err;
+ }
+
+ err = get_op(key, val, idata);
+
+ if (verbose) cout << client_name << ": got " << key << " with " << err
+ << std::endl;
+
+ return err;
+}
+
+int KvFlatBtreeAsync::get_op(const string &key, bufferlist *val,
+ index_data &idata) {
+ int err = 0;
+ std::set<std::string> key_set;
+ key_set.insert(key);
+ map<std::string,bufferlist> omap;
+ librados::ObjectReadOperation read;
+ read.omap_get_vals_by_keys(key_set, &omap, &err);
+ err = io_ctx.operate(idata.obj, &read, NULL);
+ if (err < 0) {
+ if (err == -ENOENT) {
+ err = read_index(key, &idata, NULL, true);
+ if (err < 0) {
+ if (verbose) cout << "\t" << client_name
+ << ": getting oid failed with code "
+ << err << std::endl;
+ return err;
+ }
+ if (verbose) cout << "\t" << client_name << ": index data is "
+ << idata.str()
+ << ", object is " << idata.obj << std::endl;
+ return get_op(key, val, idata);
+ } else {
+ if (verbose) cout << client_name
+ << ": get encountered an unexpected error: " << err
+ << std::endl;
+ return err;
+ }
+ }
+
+ *val = omap[key];
+ return err;
+}
+
+void *KvFlatBtreeAsync::pset(void *ptr) {
+ struct aio_set_args *args = (struct aio_set_args *)ptr;
+ *args->err =
+ args->kvba->KvFlatBtreeAsync::set((string)args->key,
+ (bufferlist)args->val, (bool)args->exc);
+ args->cb(args->err, args->cb_args);
+ delete args;
+ return NULL;
+}
+
+void KvFlatBtreeAsync::aio_set(const string &key, const bufferlist &val,
+ bool exclusive, callback cb, void * cb_args, int * err) {
+ aio_set_args *args = new aio_set_args();
+ args->kvba = this;
+ args->key = key;
+ args->val = val;
+ args->exc = exclusive;
+ args->cb = cb;
+ args->cb_args = cb_args;
+ args->err = err;
+ pthread_t t;
+ int r = pthread_create(&t, NULL, pset, (void*)args);
+ if (r < 0) {
+ *args->err = r;
+ return;
+ }
+ pthread_detach(t);
+}
+
+void *KvFlatBtreeAsync::prm(void *ptr) {
+ struct aio_rm_args *args = (struct aio_rm_args *)ptr;
+ *args->err =
+ args->kvba->KvFlatBtreeAsync::remove((string)args->key);
+ args->cb(args->err, args->cb_args);
+ delete args;
+ return NULL;
+}
+
+void KvFlatBtreeAsync::aio_remove(const string &key,
+ callback cb, void * cb_args, int * err) {
+ aio_rm_args * args = new aio_rm_args();
+ args->kvba = this;
+ args->key = key;
+ args->cb = cb;
+ args->cb_args = cb_args;
+ args->err = err;
+ pthread_t t;
+ int r = pthread_create(&t, NULL, prm, (void*)args);
+ if (r < 0) {
+ *args->err = r;
+ return;
+ }
+ pthread_detach(t);
+}
+
+void *KvFlatBtreeAsync::pget(void *ptr) {
+ struct aio_get_args *args = (struct aio_get_args *)ptr;
+ *args->err =
+ args->kvba->KvFlatBtreeAsync::get((string)args->key,
+ (bufferlist *)args->val);
+ args->cb(args->err, args->cb_args);
+ delete args;
+ return NULL;
+}
+
+void KvFlatBtreeAsync::aio_get(const string &key, bufferlist *val,
+ callback cb, void * cb_args, int * err) {
+ aio_get_args * args = new aio_get_args();
+ args->kvba = this;
+ args->key = key;
+ args->val = val;
+ args->cb = cb;
+ args->cb_args = cb_args;
+ args->err = err;
+ pthread_t t;
+ int r = pthread_create(&t, NULL, pget, (void*)args);
+ if (r < 0) {
+ *args->err = r;
+ return;
+ }
+ pthread_detach(t);
+}
+
+int KvFlatBtreeAsync::set_many(const map<string, bufferlist> &in_map) {
+ int err = 0;
+ bufferlist inbl;
+ bufferlist outbl;
+ std::set<string> keys;
+
+ map<string, bufferlist> big_map;
+ for (map<string, bufferlist>::const_iterator it = in_map.begin();
+ it != in_map.end(); ++it) {
+ keys.insert(it->first);
+ big_map.insert(*it);
+ }
+
+ if (verbose) cout << "created key set and big_map" << std::endl;
+
+ ::encode(keys, inbl);
+ librados::AioCompletion * aioc = rados.aio_create_completion();
+ io_ctx.aio_exec(index_name, aioc, "kvs", "read_many", inbl, &outbl);
+ aioc->wait_for_safe();
+ err = aioc->get_return_value();
+ aioc->release();
+ if (err < 0) {
+ cerr << "getting index failed with " << err << std::endl;
+ return err;
+ }
+
+ map<string, bufferlist> imap;//read from the index
+ bufferlist::iterator blit = outbl.begin();
+ ::decode(imap, blit);
+
+ if (verbose) cout << "finished reading index for objects. there are "
+ << imap.size() << " entries that need to be changed. " << std::endl;
+
+
+ vector<object_data> to_delete;
+
+ vector<object_data> to_create;
+
+ if (verbose) cout << "setting up to_delete and to_create vectors from index "
+ << "map" << std::endl;
+ //set up to_delete from index map
+ for (map<string, bufferlist>::iterator it = imap.begin(); it != imap.end();
+ ++it){
+ index_data idata;
+ blit = it->second.begin();
+ idata.decode(blit);
+ to_delete.push_back(object_data(idata.min_kdata, idata.kdata, idata.obj));
+ err = read_object(idata.obj, &to_delete[to_delete.size() - 1]);
+ if (err < 0) {
+ if (verbose) cout << "reading " << idata.obj << " failed with " << err
+ << std::endl;
+ return set_many(in_map);
+ }
+
+ big_map.insert(to_delete[to_delete.size() - 1].omap.begin(),
+ to_delete[to_delete.size() - 1].omap.end());
+ }
+
+ to_create.push_back(object_data(
+ to_string(client_name, client_index++)));
+ to_create[0].min_kdata = to_delete[0].min_kdata;
+
+ for(map<string, bufferlist>::iterator it = big_map.begin();
+ it != big_map.end(); ++it) {
+ if (to_create[to_create.size() - 1].omap.size() == 1.5 * k) {
+ to_create[to_create.size() - 1].max_kdata =
+ key_data(to_create[to_create.size() - 1]
+ .omap.rbegin()->first);
+
+ to_create.push_back(object_data(
+ to_string(client_name, client_index++)));
+ to_create[to_create.size() - 1].min_kdata =
+ to_create[to_create.size() - 2].max_kdata;
+ }
+
+ to_create[to_create.size() - 1].omap.insert(*it);
+ }
+ to_create[to_create.size() - 1].max_kdata =
+ to_delete[to_delete.size() - 1].max_kdata;
+
+ vector<librados::ObjectWriteOperation> owos(2 + 2 * to_delete.size()
+ + to_create.size());
+ vector<pair<pair<int, string>, librados::ObjectWriteOperation*> > ops;
+
+
+ index_data idata;
+ set_up_prefix_index(to_create, to_delete, &owos[0], &idata, &err);
+
+ if (verbose) cout << "finished making to_create and to_delete. "
+ << std::endl;
+
+ ops.push_back(make_pair(
+ pair<int, string>(ADD_PREFIX, index_name),
+ &owos[0]));
+ for (int i = 1; i < 2 + 2 * (int)to_delete.size() + (int)to_create.size();
+ i++) {
+ ops.push_back(make_pair(make_pair(0,""), &owos[i]));
+ }
+
+ set_up_ops(to_create, to_delete, &ops, idata, &err);
+
+ cout << "finished setting up ops. Starting critical section..." << std::endl;
+
+ /////BEGIN CRITICAL SECTION/////
+ //put prefix on index entry for idata.val
+ err = perform_ops("\t\t" + client_name + "-set_many:", idata, &ops);
+ if (err < 0) {
+ return set_many(in_map);
+ }
+ if (verbose) cout << "\t\t" << client_name << "-split: done splitting."
+ << std::endl;
+ /////END CRITICAL SECTION/////
+ icache_lock.Lock();
+ for (vector<delete_data>::iterator it = idata.to_delete.begin();
+ it != idata.to_delete.end(); ++it) {
+ icache.erase(it->max);
+ }
+ for (vector<create_data>::iterator it = idata.to_create.begin();
+ it != idata.to_create.end(); ++it) {
+ icache.push(index_data(*it));
+ }
+ icache_lock.Unlock();
+ return err;
+}
+
+int KvFlatBtreeAsync::remove_all() {
+ if (verbose) cout << client_name << ": removing all" << std::endl;
+ int err = 0;
+ librados::ObjectReadOperation oro;
+ librados::AioCompletion * oro_aioc = rados.aio_create_completion();
+ std::map<std::string, bufferlist> index_set;
+ oro.omap_get_vals2("",LONG_MAX,&index_set, nullptr, &err);
+ err = io_ctx.aio_operate(index_name, oro_aioc, &oro, NULL);
+ if (err < 0){
+ if (err == -ENOENT) {
+ return 0;
+ }
+ if (verbose) cout << "getting keys failed with error " << err << std::endl;
+ return err;
+ }
+ oro_aioc->wait_for_safe();
+ oro_aioc->release();
+
+ librados::ObjectWriteOperation rm_index;
+ librados::AioCompletion * rm_index_aioc = rados.aio_create_completion();
+ map<std::string,bufferlist> new_index;
+ new_index["1"] = index_set["1"];
+ rm_index.omap_clear();
+ rm_index.omap_set(new_index);
+ io_ctx.aio_operate(index_name, rm_index_aioc, &rm_index);
+ err = rm_index_aioc->get_return_value();
+ rm_index_aioc->release();
+ if (err < 0) {
+ if (verbose) cout << "rm index aioc failed with " << err
+ << std::endl;
+ return err;
+ }
+
+ if (!index_set.empty()) {
+ for (std::map<std::string,bufferlist>::iterator it = index_set.begin();
+ it != index_set.end(); ++it){
+ librados::ObjectWriteOperation sub;
+ if (it->first == "1") {
+ sub.omap_clear();
+ } else {
+ sub.remove();
+ }
+ index_data idata;
+ bufferlist::iterator b = it->second.begin();
+ idata.decode(b);
+ io_ctx.operate(idata.obj, &sub);
+ }
+ }
+
+ icache.clear();
+
+ return 0;
+}
+
+int KvFlatBtreeAsync::get_all_keys(std::set<std::string> *keys) {
+ if (verbose) cout << client_name << ": getting all keys" << std::endl;
+ int err = 0;
+ librados::ObjectReadOperation oro;
+ std::map<std::string,bufferlist> index_set;
+ oro.omap_get_vals2("",LONG_MAX,&index_set, nullptr, &err);
+ io_ctx.operate(index_name, &oro, NULL);
+ if (err < 0){
+ if (verbose) cout << "getting keys failed with error " << err << std::endl;
+ return err;
+ }
+ for (std::map<std::string,bufferlist>::iterator it = index_set.begin();
+ it != index_set.end(); ++it){
+ librados::ObjectReadOperation sub;
+ std::set<std::string> ret;
+ sub.omap_get_keys2("",LONG_MAX,&ret, nullptr, &err);
+ index_data idata;
+ bufferlist::iterator b = it->second.begin();
+ idata.decode(b);
+ io_ctx.operate(idata.obj, &sub, NULL);
+ keys->insert(ret.begin(), ret.end());
+ }
+ return err;
+}
+
+int KvFlatBtreeAsync::get_all_keys_and_values(
+ map<std::string,bufferlist> *kv_map) {
+ if (verbose) cout << client_name << ": getting all keys and values"
+ << std::endl;
+ int err = 0;
+ librados::ObjectReadOperation first_read;
+ std::set<std::string> index_set;
+ first_read.omap_get_keys2("",LONG_MAX,&index_set, nullptr, &err);
+ io_ctx.operate(index_name, &first_read, NULL);
+ if (err < 0){
+ if (verbose) cout << "getting keys failed with error " << err << std::endl;
+ return err;
+ }
+ for (std::set<std::string>::iterator it = index_set.begin();
+ it != index_set.end(); ++it){
+ librados::ObjectReadOperation sub;
+ map<std::string, bufferlist> ret;
+ sub.omap_get_vals2("",LONG_MAX,&ret, nullptr, &err);
+ io_ctx.operate(*it, &sub, NULL);
+ kv_map->insert(ret.begin(), ret.end());
+ }
+ return err;
+}
+
+bool KvFlatBtreeAsync::is_consistent() {
+ int err;
+ bool ret = true;
+ if (verbose) cout << client_name << ": checking consistency" << std::endl;
+ std::map<std::string,bufferlist> index;
+ map<std::string, std::set<std::string> > sub_objs;
+ librados::ObjectReadOperation oro;
+ oro.omap_get_vals2("",LONG_MAX,&index, nullptr, &err);
+ io_ctx.operate(index_name, &oro, NULL);
+ if (err < 0){
+ //probably because the index doesn't exist - this might be ok.
+ for (librados::NObjectIterator oit = io_ctx.nobjects_begin();
+ oit != io_ctx.nobjects_end(); ++oit) {
+ //if this executes, there are floating objects.
+ cerr << "Not consistent! found floating object " << oit->get_oid()
+ << std::endl;
+ ret = false;
+ }
+ return ret;
+ }
+
+ std::map<std::string, string> parsed_index;
+ std::set<std::string> onames;
+ std::set<std::string> special_names;
+ for (map<std::string,bufferlist>::iterator it = index.begin();
+ it != index.end(); ++it) {
+ if (it->first != "") {
+ index_data idata;
+ bufferlist::iterator b = it->second.begin();
+ idata.decode(b);
+ if (idata.prefix != "") {
+ for(vector<delete_data>::iterator dit = idata.to_delete.begin();
+ dit != idata.to_delete.end(); ++dit) {
+ librados::ObjectReadOperation oro;
+ librados::AioCompletion * aioc = rados.aio_create_completion();
+ bufferlist un;
+ oro.getxattr("unwritable", &un, &err);
+ io_ctx.aio_operate(dit->obj, aioc, &oro, NULL);
+ aioc->wait_for_safe();
+ err = aioc->get_return_value();
+ if (ceph_clock_now() - idata.ts > timeout) {
+ if (err < 0) {
+ aioc->release();
+ if (err == -ENOENT) {
+ continue;
+ } else {
+ cerr << "Not consistent! reading object " << dit->obj
+ << "returned " << err << std::endl;
+ ret = false;
+ break;
+ }
+ }
+ if (atoi(string(un.c_str(), un.length()).c_str()) != 1 &&
+ aioc->get_version64() != dit->version) {
+ cerr << "Not consistent! object " << dit->obj << " has been "
+ << " modified since the client died was not cleaned up."
+ << std::endl;
+ ret = false;
+ }
+ }
+ special_names.insert(dit->obj);
+ aioc->release();
+ }
+ for(vector<create_data >::iterator cit = idata.to_create.begin();
+ cit != idata.to_create.end(); ++cit) {
+ special_names.insert(cit->obj);
+ }
+ }
+ parsed_index.insert(make_pair(it->first, idata.obj));
+ onames.insert(idata.obj);
+ }
+ }
+
+ //make sure that an object exists iff it either is the index
+ //or is listed in the index
+ for (librados::NObjectIterator oit = io_ctx.nobjects_begin();
+ oit != io_ctx.nobjects_end(); ++oit) {
+ string name = oit->get_oid();
+ if (name != index_name && onames.count(name) == 0
+ && special_names.count(name) == 0) {
+ cerr << "Not consistent! found floating object " << name << std::endl;
+ ret = false;
+ }
+ }
+
+ //check objects
+ string prev = "";
+ for (std::map<std::string, string>::iterator it = parsed_index.begin();
+ it != parsed_index.end();
+ ++it) {
+ librados::ObjectReadOperation read;
+ read.omap_get_keys2("", LONG_MAX, &sub_objs[it->second], nullptr, &err);
+ err = io_ctx.operate(it->second, &read, NULL);
+ int size_int = (int)sub_objs[it->second].size();
+
+ //check that size is in the right range
+ if (it->first != "1" && special_names.count(it->second) == 0 &&
+ err != -ENOENT && (size_int > 2*k|| size_int < k)
+ && parsed_index.size() > 1) {
+ cerr << "Not consistent! Object " << *it << " has size " << size_int
+ << ", which is outside the acceptable range." << std::endl;
+ ret = false;
+ }
+
+ //check that all keys belong in that object
+ for(std::set<std::string>::iterator subit = sub_objs[it->second].begin();
+ subit != sub_objs[it->second].end(); ++subit) {
+ if ((it->first != "1"
+ && *subit > it->first.substr(1,it->first.length()))
+ || *subit <= prev) {
+ cerr << "Not consistent! key " << *subit << " does not belong in "
+ << *it << std::endl;
+ cerr << "not last element, i.e. " << it->first << " not equal to 1? "
+ << (it->first != "1") << std::endl
+ << "greater than " << it->first.substr(1,it->first.length())
+ <<"? " << (*subit > it->first.substr(1,it->first.length()))
+ << std::endl
+ << "less than or equal to " << prev << "? "
+ << (*subit <= prev) << std::endl;
+ ret = false;
+ }
+ }
+
+ prev = it->first.substr(1,it->first.length());
+ }
+
+ if (!ret) {
+ if (verbose) cout << "failed consistency test - see error log"
+ << std::endl;
+ cerr << str();
+ } else {
+ if (verbose) cout << "passed consistency test" << std::endl;
+ }
+ return ret;
+}
+
+string KvFlatBtreeAsync::str() {
+ stringstream ret;
+ ret << "Top-level map:" << std::endl;
+ int err = 0;
+ std::set<std::string> keys;
+ std::map<std::string,bufferlist> index;
+ librados::ObjectReadOperation oro;
+ librados::AioCompletion * top_aioc = rados.aio_create_completion();
+ oro.omap_get_vals2("",LONG_MAX,&index, nullptr, &err);
+ io_ctx.aio_operate(index_name, top_aioc, &oro, NULL);
+ top_aioc->wait_for_safe();
+ err = top_aioc->get_return_value();
+ top_aioc->release();
+ if (err < 0 && err != -5){
+ if (verbose) cout << "getting keys failed with error " << err << std::endl;
+ return ret.str();
+ }
+ if(index.empty()) {
+ ret << "There are no objects!" << std::endl;
+ return ret.str();
+ }
+
+ for (map<std::string,bufferlist>::iterator it = index.begin();
+ it != index.end(); ++it) {
+ keys.insert(string(it->second.c_str(), it->second.length())
+ .substr(1,it->second.length()));
+ }
+
+ vector<std::string> all_names;
+ vector<int> all_sizes(index.size());
+ vector<int> all_versions(index.size());
+ vector<bufferlist> all_unwrit(index.size());
+ vector<map<std::string,bufferlist> > all_maps(keys.size());
+ vector<map<std::string,bufferlist>::iterator> its(keys.size());
+ unsigned done = 0;
+ vector<bool> dones(keys.size());
+ ret << std::endl << string(150,'-') << std::endl;
+
+ for (map<std::string,bufferlist>::iterator it = index.begin();
+ it != index.end(); ++it){
+ index_data idata;
+ bufferlist::iterator b = it->second.begin();
+ idata.decode(b);
+ string s = idata.str();
+ ret << "|" << string((148 -
+ ((*it).first.length()+s.length()+3))/2,' ');
+ ret << (*it).first;
+ ret << " | ";
+ ret << string(idata.str());
+ ret << string((148 -
+ ((*it).first.length()+s.length()+3))/2,' ');
+ ret << "|\t";
+ all_names.push_back(idata.obj);
+ ret << std::endl << string(150,'-') << std::endl;
+ }
+
+ int indexer = 0;
+
+ //get the object names and sizes
+ for(vector<std::string>::iterator it = all_names.begin(); it
+ != all_names.end();
+ ++it) {
+ librados::ObjectReadOperation oro;
+ librados::AioCompletion *aioc = rados.aio_create_completion();
+ oro.omap_get_vals2("", LONG_MAX, &all_maps[indexer], nullptr, &err);
+ oro.getxattr("unwritable", &all_unwrit[indexer], &err);
+ io_ctx.aio_operate(*it, aioc, &oro, NULL);
+ aioc->wait_for_safe();
+ if (aioc->get_return_value() < 0) {
+ ret << "reading" << *it << "failed: " << err << std::endl;
+ //return ret.str();
+ }
+ all_sizes[indexer] = all_maps[indexer].size();
+ all_versions[indexer] = aioc->get_version64();
+ indexer++;
+ aioc->release();
+ }
+
+ ret << "///////////////////OBJECT NAMES////////////////" << std::endl;
+ //HEADERS
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ ret << "---------------------------\t";
+ }
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ ret << "|" << string((25 -
+ (string("Bucket: ").length() + all_names[i].length()))/2, ' ');
+ ret << "Bucket: " << all_names[i];
+ ret << string((25 -
+ (string("Bucket: ").length() + all_names[i].length()))/2, ' ') << "|\t";
+ }
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ its[i] = all_maps[i].begin();
+ ret << "|" << string((25 - (string("size: ").length()
+ + to_string("",all_sizes[i]).length()))/2, ' ');
+ ret << "size: " << all_sizes[i];
+ ret << string((25 - (string("size: ").length()
+ + to_string("",all_sizes[i]).length()))/2, ' ') << "|\t";
+ }
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ its[i] = all_maps[i].begin();
+ ret << "|" << string((25 - (string("version: ").length()
+ + to_string("",all_versions[i]).length()))/2, ' ');
+ ret << "version: " << all_versions[i];
+ ret << string((25 - (string("version: ").length()
+ + to_string("",all_versions[i]).length()))/2, ' ') << "|\t";
+ }
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ its[i] = all_maps[i].begin();
+ ret << "|" << string((25 - (string("unwritable? ").length()
+ + 1))/2, ' ');
+ ret << "unwritable? " << string(all_unwrit[i].c_str(),
+ all_unwrit[i].length());
+ ret << string((25 - (string("unwritable? ").length()
+ + 1))/2, ' ') << "|\t";
+ }
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ ret << "---------------------------\t";
+ }
+ ret << std::endl;
+ ret << "///////////////////THE ACTUAL BLOCKS////////////////" << std::endl;
+
+
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ ret << "---------------------------\t";
+ }
+ ret << std::endl;
+ //each time through this part is two lines
+ while(done < keys.size()) {
+ for(int i = 0; i < indexer; i++) {
+ if(dones[i]){
+ ret << " \t";
+ } else {
+ if (its[i] == all_maps[i].end()){
+ done++;
+ dones[i] = true;
+ ret << " \t";
+ } else {
+ ret << "|" << string((25 -
+ ((*its[i]).first.length()+its[i]->second.length()+3))/2,' ');
+ ret << (*its[i]).first;
+ ret << " | ";
+ ret << string(its[i]->second.c_str(), its[i]->second.length());
+ ret << string((25 -
+ ((*its[i]).first.length()+its[i]->second.length()+3))/2,' ');
+ ret << "|\t";
+ ++(its[i]);
+ }
+
+ }
+ }
+ ret << std::endl;
+ for (int i = 0; i < indexer; i++) {
+ if(dones[i]){
+ ret << " \t";
+ } else {
+ ret << "---------------------------\t";
+ }
+ }
+ ret << std::endl;
+
+ }
+ return ret.str();
+}
Code Review / stor4nfv.git / blobdiff