ceph/src/rocksdb/trace_replay/trace_record_handler.h

   1 //  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
   2 //  This source code is licensed under both the GPLv2 (found in the
   3 //  COPYING file in the root directory) and Apache 2.0 License
   4 //  (found in the LICENSE.Apache file in the root directory).
   5
   6 #pragma once
   7
   8 #include <memory>
   9 #include <unordered_map>
  10 #include <vector>
  11
  12 #include "rocksdb/db.h"
  13 #include "rocksdb/options.h"
  14 #include "rocksdb/status.h"
  15 #include "rocksdb/system_clock.h"
  16 #include "rocksdb/trace_record.h"
  17
  18 namespace ROCKSDB_NAMESPACE {
  19
  20 // Handler to execute TraceRecord.
  21 class TraceExecutionHandler : public TraceRecord::Handler {
  22  public:
  23   TraceExecutionHandler(DB* db,
  24                         const std::vector<ColumnFamilyHandle*>& handles);
  25   virtual ~TraceExecutionHandler() override;
  26
  27   virtual Status Handle(const WriteQueryTraceRecord& record,
  28                         std::unique_ptr<TraceRecordResult>* result) override;
  29   virtual Status Handle(const GetQueryTraceRecord& record,
  30                         std::unique_ptr<TraceRecordResult>* result) override;
  31   virtual Status Handle(const IteratorSeekQueryTraceRecord& record,
  32                         std::unique_ptr<TraceRecordResult>* result) override;
  33   virtual Status Handle(const MultiGetQueryTraceRecord& record,
  34                         std::unique_ptr<TraceRecordResult>* result) override;
  35
  36  private:
  37   DB* db_;
  38   std::unordered_map<uint32_t, ColumnFamilyHandle*> cf_map_;
  39   WriteOptions write_opts_;
  40   ReadOptions read_opts_;
  41   SystemClock* clock_;
  42 };
  43
  44 // To do: Handler for trace_analyzer.
  45
  46 }  // namespace ROCKSDB_NAMESPACE