ceph/src/test/librados_test_stub/TestMemIoCtxImpl.h

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3
   4 #ifndef CEPH_TEST_MEM_IO_CTX_IMPL_H
   5 #define CEPH_TEST_MEM_IO_CTX_IMPL_H
   6
   7 #include "test/librados_test_stub/TestIoCtxImpl.h"
   8 #include "test/librados_test_stub/TestMemCluster.h"
   9
  10 namespace librados {
  11
  12 class TestMemRadosClient;
  13
  14 class TestMemIoCtxImpl : public TestIoCtxImpl {
  15 public:
  16   TestMemIoCtxImpl();
  17   TestMemIoCtxImpl(TestMemRadosClient *client, int64_t m_pool_id,
  18                    const std::string& pool_name,
  19                    TestMemCluster::Pool *pool);
  20   ~TestMemIoCtxImpl() override;
  21
  22   TestIoCtxImpl *clone() override;
  23
  24   int aio_remove(const std::string& oid, AioCompletionImpl *c) override;
  25
  26   int append(const std::string& oid, const bufferlist &bl,
  27              const SnapContext &snapc) override;
  28
  29   int assert_exists(const std::string &oid) override;
  30
  31   int create(const std::string& oid, bool exclusive) override;
  32   int list_snaps(const std::string& o, snap_set_t *out_snaps) override;
  33   int omap_get_vals(const std::string& oid,
  34                     const std::string& start_after,
  35                     const std::string &filter_prefix,
  36                     uint64_t max_return,
  37                     std::map<std::string, bufferlist> *out_vals) override;
  38   int omap_rm_keys(const std::string& oid,
  39                    const std::set<std::string>& keys) override;
  40   int omap_set(const std::string& oid, const std::map<std::string,
  41                bufferlist> &map) override;
  42   int read(const std::string& oid, size_t len, uint64_t off,
  43            bufferlist *bl) override;
  44   int remove(const std::string& oid, const SnapContext &snapc) override;
  45   int selfmanaged_snap_create(uint64_t *snapid) override;
  46   int selfmanaged_snap_remove(uint64_t snapid) override;
  47   int selfmanaged_snap_rollback(const std::string& oid,
  48                                 uint64_t snapid) override;
  49   int sparse_read(const std::string& oid, uint64_t off, uint64_t len,
  50                   std::map<uint64_t,uint64_t> *m, bufferlist *data_bl) override;
  51   int stat(const std::string& oid, uint64_t *psize, time_t *pmtime) override;
  52   int truncate(const std::string& oid, uint64_t size,
  53                const SnapContext &snapc) override;
  54   int write(const std::string& oid, bufferlist& bl, size_t len,
  55             uint64_t off, const SnapContext &snapc) override;
  56   int write_full(const std::string& oid, bufferlist& bl,
  57                  const SnapContext &snapc) override;
  58   int writesame(const std::string& oid, bufferlist& bl, size_t len,
  59                 uint64_t off, const SnapContext &snapc) override;
  60   int xattr_get(const std::string& oid,
  61                 std::map<std::string, bufferlist>* attrset) override;
  62   int xattr_set(const std::string& oid, const std::string &name,
  63                 bufferlist& bl) override;
  64   int zero(const std::string& oid, uint64_t off, uint64_t len) override;
  65
  66 protected:
  67   TestMemCluster::Pool *get_pool() {
  68     return m_pool;
  69   }
  70
  71 private:
  72   TestMemIoCtxImpl(const TestMemIoCtxImpl&);
  73
  74   TestMemRadosClient *m_client;
  75   TestMemCluster::Pool *m_pool;
  76
  77   void append_clone(bufferlist& src, bufferlist* dest);
  78   size_t clip_io(size_t off, size_t len, size_t bl_len);
  79   void ensure_minimum_length(size_t len, bufferlist *bl);
  80
  81   TestMemCluster::SharedFile get_file(const std::string &oid, bool write,
  82                                       const SnapContext &snapc);
  83
  84 };
  85
  86 } // namespace librados
  87
  88 #endif // CEPH_TEST_MEM_IO_CTX_IMPL_H