summaryrefslogtreecommitdiffstats
path: root/src/common/io_exerciser/RadosIo.h
blob: a5c66ad4768abbcbe362956932ba4242a00a4d5a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#pragma once

#include "ObjectModel.h"

/* Overview
 *
 * class RadosIo
 *   An IoExerciser. A simple RADOS client that generates I/Os
 *   from IoOps. Uses an ObjectModel to track the data stored
 *   in the object. Uses DataBuffer to create and validate
 *   data buffers. When there are not barrier I/Os this may
 *   issue multiple async I/Os in parallel.
 *
 */

namespace ceph {
namespace io_exerciser {
namespace data_generation {
class DataGenerator;
}

class RadosIo : public Model {
 protected:
  librados::Rados& rados;
  boost::asio::io_context& asio;
  std::unique_ptr<ObjectModel> om;
  std::unique_ptr<ceph::io_exerciser::data_generation::DataGenerator> db;
  std::string pool;
  std::optional<std::vector<int>> cached_shard_order;
  int threads;
  ceph::mutex& lock;
  ceph::condition_variable& cond;
  librados::IoCtx io;
  int outstanding_io;

  void start_io();
  void finish_io();
  void wait_for_io(int count);

 public:
  RadosIo(librados::Rados& rados, boost::asio::io_context& asio,
          const std::string& pool, const std::string& oid,
          const std::optional<std::vector<int>>& cached_shard_order,
          uint64_t block_size, int seed, int threads, ceph::mutex& lock,
          ceph::condition_variable& cond);

  ~RadosIo();

  void allow_ec_overwrites(bool allow);

  template <int N>
  class AsyncOpInfo {
   public:
    librados::ObjectReadOperation rop;
    librados::ObjectWriteOperation wop;
    std::array<ceph::bufferlist, N> bufferlist;
    std::array<uint64_t, N> offset;
    std::array<uint64_t, N> length;

    AsyncOpInfo(const std::array<uint64_t, N>& offset = {},
                const std::array<uint64_t, N>& length = {});
    ~AsyncOpInfo() = default;
  };

  // Must be called with lock held
  bool readyForIoOp(IoOp& op);
  void applyIoOp(IoOp& op);

 private:
  void applyReadWriteOp(IoOp& op);
  void applyInjectOp(IoOp& op);
};
}  // namespace io_exerciser
}  // namespace ceph