Shutdown V2

Signed-off-by: dorjesinpo <129227380+dorjesinpo@users.noreply.github.com>
bloomberg · Aug 15, 2024 · d7971bb · d7971bb
1 parent 196d0b8
commit d7971bb
Show file tree

Hide file tree

Showing 54 changed files with 1,277 additions and 397 deletions.
diff --git a/src/groups/bmq/bmqp/bmqp_ctrlmsg.xsd b/src/groups/bmq/bmqp/bmqp_ctrlmsg.xsd
@@ -1404,6 +1404,7 @@
  </annotation>
  <sequence>
  <element name='clusterName' type='string'/>
+ <element name='version' type='int' default='1'/>
  </sequence>
  </complexType>
 

diff --git a/src/groups/bmq/bmqp/bmqp_ctrlmsg_messages.cpp b/src/groups/bmq/bmqp/bmqp_ctrlmsg_messages.cpp
@@ -4017,19 +4017,26 @@ const char* StatusCategory::toString(StatusCategory::Value value)
 
 const char StopRequest::CLASS_NAME[] = "StopRequest";
 
+const int StopRequest::DEFAULT_INITIALIZER_VERSION = 1;
+
 const bdlat_AttributeInfo StopRequest::ATTRIBUTE_INFO_ARRAY[] = {
  {ATTRIBUTE_ID_CLUSTER_NAME,
  "clusterName",
  sizeof("clusterName") - 1,
  "",
- bdlat_FormattingMode::e_TEXT}};
+ bdlat_FormattingMode::e_TEXT},
+ {ATTRIBUTE_ID_VERSION,
+ "version",
+ sizeof("version") - 1,
+ "",
+ bdlat_FormattingMode::e_DEC}};
 
 // CLASS METHODS
 
 const bdlat_AttributeInfo* StopRequest::lookupAttributeInfo(const char* name,
  int nameLength)
 {
- for (int i = 0; i < 1; ++i) {
+ for (int i = 0; i < 2; ++i) {
  const bdlat_AttributeInfo& attributeInfo =
  StopRequest::ATTRIBUTE_INFO_ARRAY[i];
 
@@ -4047,6 +4054,8 @@ const bdlat_AttributeInfo* StopRequest::lookupAttributeInfo(int id)
  switch (id) {
  case ATTRIBUTE_ID_CLUSTER_NAME:
  return &ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_CLUSTER_NAME];
+ case ATTRIBUTE_ID_VERSION:
+ return &ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_VERSION];
  default: return 0;
  }
 }
@@ -4055,25 +4064,29 @@ const bdlat_AttributeInfo* StopRequest::lookupAttributeInfo(int id)
 
 StopRequest::StopRequest(bslma::Allocator* basicAllocator)
 : d_clusterName(basicAllocator)
+, d_version(DEFAULT_INITIALIZER_VERSION)
 {
 }
 
 StopRequest::StopRequest(const StopRequest& original,
  bslma::Allocator* basicAllocator)
 : d_clusterName(original.d_clusterName, basicAllocator)
+, d_version(original.d_version)
 {
 }
 
 #if defined(BSLS_COMPILERFEATURES_SUPPORT_RVALUE_REFERENCES) && \
  defined(BSLS_COMPILERFEATURES_SUPPORT_NOEXCEPT)
 StopRequest::StopRequest(StopRequest&& original) noexcept
-: d_clusterName(bsl::move(original.d_clusterName))
+: d_clusterName(bsl::move(original.d_clusterName)),
+ d_version(bsl::move(original.d_version))
 {
 }
 
 StopRequest::StopRequest(StopRequest&& original,
  bslma::Allocator* basicAllocator)
 : d_clusterName(bsl::move(original.d_clusterName), basicAllocator)
+, d_version(bsl::move(original.d_version))
 {
 }
 #endif
@@ -4088,6 +4101,7 @@ StopRequest& StopRequest::operator=(const StopRequest& rhs)
 {
  if (this != &rhs) {
  d_clusterName = rhs.d_clusterName;
+ d_version = rhs.d_version;
  }
 
  return *this;
@@ -4099,6 +4113,7 @@ StopRequest& StopRequest::operator=(StopRequest&& rhs)
 {
  if (this != &rhs) {
  d_clusterName = bsl::move(rhs.d_clusterName);
+ d_version = bsl::move(rhs.d_version);
  }
 
  return *this;
@@ -4108,6 +4123,7 @@ StopRequest& StopRequest::operator=(StopRequest&& rhs)
 void StopRequest::reset()
 {
  bdlat_ValueTypeFunctions::reset(&d_clusterName);
+ d_version = DEFAULT_INITIALIZER_VERSION;
 }
 
 // ACCESSORS
@@ -4118,6 +4134,7 @@ StopRequest::print(bsl::ostream& stream, int level, int spacesPerLevel) const
  bslim::Printer printer(&stream, level, spacesPerLevel);
  printer.start();
  printer.printAttribute("clusterName", this->clusterName());
+ printer.printAttribute("version", this->version());
  printer.end();
  return stream;
 }

diff --git a/src/groups/bmq/bmqp/bmqp_ctrlmsg_messages.h b/src/groups/bmq/bmqp/bmqp_ctrlmsg_messages.h
@@ -7429,18 +7429,21 @@ namespace bmqp_ctrlmsg {
 class StopRequest {
  // INSTANCE DATA
  bsl::string d_clusterName;
+ int d_version;
 
  public:
  // TYPES
- enum { ATTRIBUTE_ID_CLUSTER_NAME = 0 };
+ enum { ATTRIBUTE_ID_CLUSTER_NAME = 0, ATTRIBUTE_ID_VERSION = 1 };
 
- enum { NUM_ATTRIBUTES = 1 };
+ enum { NUM_ATTRIBUTES = 2 };
 
- enum { ATTRIBUTE_INDEX_CLUSTER_NAME = 0 };
+ enum { ATTRIBUTE_INDEX_CLUSTER_NAME = 0, ATTRIBUTE_INDEX_VERSION = 1 };
 
  // CONSTANTS
  static const char CLASS_NAME[];
 
+ static const int DEFAULT_INITIALIZER_VERSION;
+
  static const bdlat_AttributeInfo ATTRIBUTE_INFO_ARRAY[];
 
  public:
@@ -7540,6 +7543,10 @@ class StopRequest {
  /// object.
  bsl::string& clusterName();
 
+ /// Return a reference to the non-modifiable "Version" attribute of this
+ /// object.
+ int& version();
+
  // ACCESSORS
 
  /// Format this object to the specified output `stream` at the
@@ -7587,6 +7594,10 @@ class StopRequest {
  /// Return a reference to the non-modifiable "ClusterName" attribute of
  /// this object.
  const bsl::string& clusterName() const;
+
+ /// Return a reference to the non-modifiable "Version" attribute of this
+ /// object.
+ int version() const;
 };
 
 // FREE OPERATORS
@@ -27150,6 +27161,12 @@ int StopRequest::manipulateAttributes(MANIPULATOR& manipulator)
  return ret;
  }
 
+ ret = manipulator(&d_version,
+ ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_VERSION]);
+ if (ret) {
+ return ret;
+ }
+
  return ret;
 }
 
@@ -27163,6 +27180,10 @@ int StopRequest::manipulateAttribute(MANIPULATOR& manipulator, int id)
  return manipulator(&d_clusterName,
  ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_CLUSTER_NAME]);
  }
+ case ATTRIBUTE_ID_VERSION: {
+ return manipulator(&d_version,
+ ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_VERSION]);
+ }
  default: return NOT_FOUND;
  }
 }
@@ -27188,6 +27209,11 @@ inline bsl::string& StopRequest::clusterName()
  return d_clusterName;
 }
 
+inline int& StopRequest::version()
+{
+ return d_version;
+}
+
 // ACCESSORS
 template <class ACCESSOR>
 int StopRequest::accessAttributes(ACCESSOR& accessor) const
@@ -27200,6 +27226,10 @@ int StopRequest::accessAttributes(ACCESSOR& accessor) const
  return ret;
  }
 
+ ret = accessor(d_version, ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_VERSION]);
+ if (ret) {
+ return ret;
+ }
  return ret;
 }
 
@@ -27213,6 +27243,10 @@ int StopRequest::accessAttribute(ACCESSOR& accessor, int id) const
  return accessor(d_clusterName,
  ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_CLUSTER_NAME]);
  }
+ case ATTRIBUTE_ID_VERSION: {
+ return accessor(d_version,
+ ATTRIBUTE_INFO_ARRAY[ATTRIBUTE_INDEX_VERSION]);
+ }
  default: return NOT_FOUND;
  }
 }
@@ -27238,6 +27272,11 @@ inline const bsl::string& StopRequest::clusterName() const
  return d_clusterName;
 }
 
+inline int StopRequest::version() const
+{
+ return d_version;
+}
+
 template <typename HASH_ALGORITHM>
 void hashAppend(HASH_ALGORITHM& hashAlg,
  const bmqp_ctrlmsg::StopRequest& object)

diff --git a/src/groups/bmq/bmqp/bmqp_protocol.cpp b/src/groups/bmq/bmqp/bmqp_protocol.cpp
@@ -264,6 +264,8 @@ const char HighAvailabilityFeatures::k_BROADCAST_TO_PROXIES[] =
  "BROADCAST_TO_PROXIES";
 const char HighAvailabilityFeatures::k_GRACEFUL_SHUTDOWN[] =
  "GRACEFUL_SHUTDOWN";
+const char HighAvailabilityFeatures::k_GRACEFUL_SHUTDOWN_V2[] =
+ "GRACEFUL_SHUTDOWN_V2";
 
 // --------------------------------
 // struct MessagePropertiesFeatures

diff --git a/src/groups/bmq/bmqp/bmqp_protocol.h b/src/groups/bmq/bmqp/bmqp_protocol.h
@@ -633,6 +633,8 @@ struct HighAvailabilityFeatures {
  static const char k_BROADCAST_TO_PROXIES[];
 
  static const char k_GRACEFUL_SHUTDOWN[];
+
+ static const char k_GRACEFUL_SHUTDOWN_V2[];
 };
 
 /// This struct defines feature names related to MessageProperties

diff --git a/src/groups/bmq/bmqp/bmqp_requestmanager.h b/src/groups/bmq/bmqp/bmqp_requestmanager.h
@@ -1091,6 +1091,10 @@ void RequestManager<REQUEST, RESPONSE>::onRequestTimeout(int requestId)
 
  // Explicitly invalidate the timeout since we processed it
  request->d_timeoutSchedulerHandle.release();
+
+ if (!d_lateResponseMode) {
+ d_requests.erase(it);
+ }
  } // close guard scope
 
  BALL_LOG_ERROR << "Request with '" << request->nodeDescription()

diff --git a/src/groups/mqb/mqba/mqba_adminsession.cpp b/src/groups/mqb/mqba/mqba_adminsession.cpp
@@ -419,10 +419,12 @@ void AdminSession::tearDown(const bsl::shared_ptr<void>& session,
 }
 
 void AdminSession::initiateShutdown(const ShutdownCb& callback,
- const bsls::TimeInterval& timeout)
+ const bsls::TimeInterval& timeout,
+ bool suppportShutdownV2)
 {
  // executed by the *ANY* thread
  (void)timeout;
+ (void)suppportShutdownV2;
 
  dispatcher()->execute(
  bdlf::BindUtil::bind(&AdminSession::initiateShutdownDispatched,

diff --git a/src/groups/mqb/mqba/mqba_adminsession.h b/src/groups/mqb/mqba/mqba_adminsession.h
@@ -264,7 +264,8 @@ class AdminSession : public mqbnet::Session, public mqbi::DispatcherClient {
  /// if the specified `timeout` is expired.
  void
  initiateShutdown(const ShutdownCb& callback,
- const bsls::TimeInterval& timeout) BSLS_KEYWORD_OVERRIDE;
+ const bsls::TimeInterval& timeout,
+ bool suppportShutdownV2 = false) BSLS_KEYWORD_OVERRIDE;
 
  /// Make the session abandon any work it has.
  void invalidate() BSLS_KEYWORD_OVERRIDE;