Adding upstream version 14.2.21.upstream/14.2.21upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 145 insertions, 0 deletions

diff --git a/src/test/librados_test_stub/TestRadosClient.h b/src/test/librados_test_stub/TestRadosClient.h
new file mode 100644
index 00000000..993382f7
--- /dev/null
+++ b/src/test/librados_test_stub/TestRadosClient.h
@@ -0,0 +1,145 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef CEPH_TEST_RADOS_CLIENT_H
+#define CEPH_TEST_RADOS_CLIENT_H
+
+#include <map>
+#include <list>
+#include <string>
+#include <vector>
+#include <atomic>
+
+#include <boost/function.hpp>
+#include <boost/functional/hash.hpp>
+
+#include "include/rados/librados.hpp"
+#include "common/config.h"
+#include "include/buffer_fwd.h"
+#include "test/librados_test_stub/TestWatchNotify.h"
+
+class Finisher;
+
+namespace librados {
+
+class TestIoCtxImpl;
+
+class TestRadosClient {
+public:
+
+  static void Deallocate(librados::TestRadosClient* client)
+  {
+    client->put();
+  }
+
+  typedef boost::function<int()> AioFunction;
+
+  struct Object {
+    std::string oid;
+    std::string locator;
+    std::string nspace;
+  };
+
+  class Transaction {
+  public:
+    Transaction(TestRadosClient *rados_client, const std::string& nspace,
+                const std::string &oid)
+      : rados_client(rados_client), nspace(nspace), oid(oid) {
+      rados_client->transaction_start(nspace, oid);
+    }
+    ~Transaction() {
+      rados_client->transaction_finish(nspace, oid);
+    }
+  private:
+    TestRadosClient *rados_client;
+    std::string nspace;
+    std::string oid;
+  };
+
+  TestRadosClient(CephContext *cct, TestWatchNotify *watch_notify);
+
+  void get();
+  void put();
+
+  virtual CephContext *cct();
+
+  virtual uint32_t get_nonce() = 0;
+  virtual uint64_t get_instance_id() = 0;
+
+  virtual int get_min_compatible_osd(int8_t* require_osd_release) = 0;
+  virtual int get_min_compatible_client(int8_t* min_compat_client,
+                                        int8_t* require_min_compat_client) = 0;
+
+  virtual int connect();
+  virtual void shutdown();
+  virtual int wait_for_latest_osdmap();
+
+  virtual TestIoCtxImpl *create_ioctx(int64_t pool_id,
+                                      const std::string &pool_name) = 0;
+
+  virtual int mon_command(const std::vector<std::string>& cmd,
+                          const bufferlist &inbl,
+                          bufferlist *outbl, std::string *outs);
+
+  virtual void object_list(int64_t pool_id,
+			   std::list<librados::TestRadosClient::Object> *list) = 0;
+
+  virtual int service_daemon_register(const std::string& service,
+                                      const std::string& name,
+                                      const std::map<std::string,std::string>& metadata) = 0;
+  virtual int service_daemon_update_status(std::map<std::string,std::string>&& status) = 0;
+
+  virtual int pool_create(const std::string &pool_name) = 0;
+  virtual int pool_delete(const std::string &pool_name) = 0;
+  virtual int pool_get_base_tier(int64_t pool_id, int64_t* base_tier) = 0;
+  virtual int pool_list(std::list<std::pair<int64_t, std::string> >& v) = 0;
+  virtual int64_t pool_lookup(const std::string &name) = 0;
+  virtual int pool_reverse_lookup(int64_t id, std::string *name) = 0;
+
+  virtual int aio_watch_flush(AioCompletionImpl *c);
+  virtual int watch_flush() = 0;
+
+  virtual bool is_blacklisted() const = 0;
+  virtual int blacklist_add(const std::string& client_address,
+			    uint32_t expire_seconds) = 0;
+
+  Finisher *get_aio_finisher() {
+    return m_aio_finisher;
+  }
+  TestWatchNotify *get_watch_notify() {
+    return m_watch_notify;
+  }
+
+  void add_aio_operation(const std::string& oid, bool queue_callback,
+			 const AioFunction &aio_function, AioCompletionImpl *c);
+  void flush_aio_operations();
+  void flush_aio_operations(AioCompletionImpl *c);
+
+  void finish_aio_completion(AioCompletionImpl *c, int r);
+
+protected:
+  virtual ~TestRadosClient();
+
+  virtual void transaction_start(const std::string& nspace,
+                                 const std::string &oid) = 0;
+  virtual void transaction_finish(const std::string& nspace,
+                                  const std::string &oid) = 0;
+
+private:
+
+  CephContext *m_cct;
+  std::atomic<uint64_t> m_refcount = { 0 };
+
+  TestWatchNotify *m_watch_notify;
+
+  Finisher *get_finisher(const std::string& oid);
+
+  Finisher *m_aio_finisher;
+  std::vector<Finisher *> m_finishers;
+  boost::hash<std::string> m_hash;
+
+};
+
+} // namespace librados
+
+#endif // CEPH_TEST_RADOS_CLIENT_H