1 files changed, 130 insertions, 0 deletions
diff --git a/src/messages/MRecoveryReserve.h b/src/messages/MRecoveryReserve.h
new file mode 100644
index 00000000..2dc7d6c3
--- /dev/null
+++ b/src/messages/MRecoveryReserve.h
@@ -0,0 +1,130 @@
+// -*- 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) 2004-2006 Sage Weil <sage@newdream.net>
+ *
+ * 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 CEPH_MRECOVERY_H
+#define CEPH_MRECOVERY_H
+
+#include "msg/Message.h"
+#include "messages/MOSDPeeringOp.h"
+
+class MRecoveryReserve : public MessageInstance<MRecoveryReserve, MOSDPeeringOp> {
+public:
+  friend factory;
+private:
+  static constexpr int HEAD_VERSION = 3;
+  static constexpr int COMPAT_VERSION = 2;
+public:
+  spg_t pgid;
+  epoch_t query_epoch;
+  enum {
+    REQUEST = 0,   // primary->replica: please reserve slot
+    GRANT = 1,     // replica->primary: ok, i reserved it
+    RELEASE = 2,   // primary->replica: release the slot i reserved before
+    REVOKE = 3,    // replica->primary: i'm taking back the slot i gave you
+  };
+  uint32_t type;
+  uint32_t priority = 0;
+
+  spg_t get_spg() const {
+    return pgid;
+  }
+  epoch_t get_map_epoch() const {
+    return query_epoch;
+  }
+  epoch_t get_min_epoch() const {
+    return query_epoch;
+  }
+
+  PGPeeringEvent *get_event() override {
+    switch (type) {
+    case REQUEST:
+      return new PGPeeringEvent(
+	query_epoch,
+	query_epoch,
+	RequestRecoveryPrio(priority));
+    case GRANT:
+      return new PGPeeringEvent(
+	query_epoch,
+	query_epoch,
+	RemoteRecoveryReserved());
+    case RELEASE:
+      return new PGPeeringEvent(
+	query_epoch,
+	query_epoch,
+	RecoveryDone());
+    case REVOKE:
+      return new PGPeeringEvent(
+	query_epoch,
+	query_epoch,
+	DeferRecovery(0.0));
+    default:
+      ceph_abort();
+    }
+  }
+
+  MRecoveryReserve()
+    : MessageInstance(MSG_OSD_RECOVERY_RESERVE, HEAD_VERSION, COMPAT_VERSION),
+      query_epoch(0), type(-1) {}
+  MRecoveryReserve(int type,
+		   spg_t pgid,
+		   epoch_t query_epoch,
+		   unsigned prio = 0)
+    : MessageInstance(MSG_OSD_RECOVERY_RESERVE, HEAD_VERSION, COMPAT_VERSION),
+      pgid(pgid), query_epoch(query_epoch),
+      type(type), priority(prio) {}
+
+  std::string_view get_type_name() const override {
+    return "MRecoveryReserve";
+  }
+
+  void inner_print(ostream& out) const override {
+    switch (type) {
+    case REQUEST:
+      out << "REQUEST";
+      break;
+    case GRANT:
+      out << "GRANT";
+      break;
+    case RELEASE:
+      out << "RELEASE";
+      break;
+    case REVOKE:
+      out << "REVOKE";
+      break;
+    }
+    if (type == REQUEST) out << " prio: " << priority;
+  }
+
+  void decode_payload() override {
+    auto p = payload.cbegin();
+    decode(pgid.pgid, p);
+    decode(query_epoch, p);
+    decode(type, p);
+    decode(pgid.shard, p);
+    if (header.version >= 3) {
+      decode(priority, p);
+    }
+  }
+
+  void encode_payload(uint64_t features) override {
+    using ceph::encode;
+    encode(pgid.pgid, payload);
+    encode(query_epoch, payload);
+    encode(type, payload);
+    encode(pgid.shard, payload);
+    encode(priority, payload);
+  }
+};
+
+#endif