ceph/src/common/CommandTable.h

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3 /*
   4  * Ceph - scalable distributed file system
   5  *
   6  * Copyright (C) 2016 Red Hat Inc
   7  *
   8  * This is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License version 2.1, as published by the Free Software
  11  * Foundation.  See file COPYING.
  12  *
  13  */
  14
  15 #ifndef COMMAND_TABLE_H_
  16 #define COMMAND_TABLE_H_
  17
  18 #include "messages/MCommand.h"
  19
  20 class CommandOp
  21 {
  22   public:
  23   ConnectionRef con;
  24   ceph_tid_t tid;
  25
  26   std::vector<std::string> cmd;
  27   bufferlist    inbl;
  28   Context      *on_finish;
  29   bufferlist   *outbl;
  30   std::string  *outs;
  31
  32   MCommand::ref get_message(const uuid_d &fsid) const
  33   {
  34     auto m = MCommand::create(fsid);
  35     m->cmd = cmd;
  36     m->set_data(inbl);
  37     m->set_tid(tid);
  38
  39     return m;
  40   }
  41
  42   CommandOp(const ceph_tid_t t) : tid(t), on_finish(nullptr),
  43                                   outbl(nullptr), outs(nullptr) {}
  44   CommandOp() : tid(0), on_finish(nullptr), outbl(nullptr), outs(nullptr) {}
  45 };
  46
  47 /**
  48  * Hold client-side state for a collection of in-flight commands
  49  * to a remote service.
  50  */
  51 template<typename T>
  52 class CommandTable
  53 {
  54 protected:
  55   ceph_tid_t last_tid;
  56   std::map<ceph_tid_t, T> commands;
  57
  58 public:
  59
  60   CommandTable()
  61     : last_tid(0)
  62   {}
  63
  64   ~CommandTable()
  65   {
  66     ceph_assert(commands.empty());
  67   }
  68
  69   T& start_command()
  70   {
  71     ceph_tid_t tid = last_tid++;
  72     commands.insert(std::make_pair(tid, T(tid)) );
  73
  74     return commands.at(tid);
  75   }
  76
  77   const std::map<ceph_tid_t, T> &get_commands() const
  78   {
  79     return commands;
  80   }
  81
  82   bool exists(ceph_tid_t tid) const
  83   {
  84     return commands.count(tid) > 0;
  85   }
  86
  87   T& get_command(ceph_tid_t tid)
  88   {
  89     return commands.at(tid);
  90   }
  91
  92   void erase(ceph_tid_t tid)
  93   {
  94     commands.erase(tid);
  95   }
  96
  97   void clear() {
  98     commands.clear();
  99   }
 100 };
 101
 102 #endif
 103