updated code to shutdown on tunnel close callback

awslabs · Sep 28, 2023 · 726d4d4 · 726d4d4
1 parent 95db8c9
commit 726d4d4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/source/main.cpp b/source/main.cpp
@@ -237,6 +237,10 @@ namespace Aws
                         case ClientBaseEventNotification::FEATURE_STOPPED:
                         {
                             LOGM_INFO(TAG, "%s has stopped", feature->getName().c_str());
+                            // Stopping DC for ST component
+                            #if defined(DISABLE_MQTT)
+                            shutdown();
+                            #endif
                             break;
                         }
                         default:

diff --git a/source/tunneling/SecureTunnelingFeature.cpp b/source/tunneling/SecureTunnelingFeature.cpp
@@ -272,12 +272,15 @@ namespace Aws
                 void SecureTunnelingFeature::OnConnectionShutdown(SecureTunnelingContext *contextToRemove)
                 {
                     LOG_DEBUG(TAG, "SecureTunnelingFeature::OnConnectionShutdown");
-
+                    #if defined(DISABLE_MQTT)
+                    this->stop();
+                    #else
                     auto it =
                         find_if(mContexts.begin(), mContexts.end(), [&](const unique_ptr<SecureTunnelingContext> &c) {
                             return c.get() == contextToRemove;
                         });
                     mContexts.erase(std::remove(mContexts.begin(), mContexts.end(), *it));
+                    #endif
                 }
 
             } // namespace SecureTunneling