diff options
Diffstat (limited to '')
-rw-r--r-- | src/common/CommandTable.h | 112 |
1 files changed, 112 insertions, 0 deletions
diff --git a/src/common/CommandTable.h b/src/common/CommandTable.h new file mode 100644 index 000000000..53218d653 --- /dev/null +++ b/src/common/CommandTable.h @@ -0,0 +1,112 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2016 Red Hat Inc + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ + +#ifndef COMMAND_TABLE_H_ +#define COMMAND_TABLE_H_ + +#include "messages/MCommand.h" +#include "messages/MMgrCommand.h" + +class CommandOp +{ + public: + ConnectionRef con; + ceph_tid_t tid; + + std::vector<std::string> cmd; + ceph::buffer::list inbl; + Context *on_finish; + ceph::buffer::list *outbl; + std::string *outs; + + MessageRef get_message(const uuid_d &fsid, + bool mgr=false) const + { + if (mgr) { + auto m = ceph::make_message<MMgrCommand>(fsid); + m->cmd = cmd; + m->set_data(inbl); + m->set_tid(tid); + return m; + } else { + auto m = ceph::make_message<MCommand>(fsid); + m->cmd = cmd; + m->set_data(inbl); + m->set_tid(tid); + return m; + } + } + + CommandOp(const ceph_tid_t t) : tid(t), on_finish(nullptr), + outbl(nullptr), outs(nullptr) {} + CommandOp() : tid(0), on_finish(nullptr), outbl(nullptr), outs(nullptr) {} +}; + +/** + * Hold client-side state for a collection of in-flight commands + * to a remote service. + */ +template<typename T> +class CommandTable +{ +protected: + ceph_tid_t last_tid; + std::map<ceph_tid_t, T> commands; + +public: + + CommandTable() + : last_tid(0) + {} + + ~CommandTable() + { + ceph_assert(commands.empty()); + } + + T& start_command() + { + ceph_tid_t tid = last_tid++; + commands.insert(std::make_pair(tid, T(tid)) ); + + return commands.at(tid); + } + + const std::map<ceph_tid_t, T> &get_commands() const + { + return commands; + } + + bool exists(ceph_tid_t tid) const + { + return commands.count(tid) > 0; + } + + T& get_command(ceph_tid_t tid) + { + return commands.at(tid); + } + + void erase(ceph_tid_t tid) + { + commands.erase(tid); + } + + void clear() { + commands.clear(); + } +}; + +#endif + |