Shutdown V2

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

Hide file tree

Showing 54 changed files with 1,280 additions and 403 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;