Add the reset() method to HttpAppFramework

drogonframework · Mar 11, 2024 · df5882e · df5882e
1 parent 4cbac30
commit df5882e
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/lib/inc/drogon/HttpAppFramework.h b/lib/inc/drogon/HttpAppFramework.h
@@ -1558,6 +1558,7 @@ class DROGON_EXPORT HttpAppFramework : public trantor::NonCopyable
      * @brief get the number of active connections.
      */
     virtual int64_t getConnectionCount() const = 0;
+    virtual HttpAppFramework &reset() = 0;
 
   private:
     virtual void registerHttpController(

diff --git a/lib/src/HttpAppFrameworkImpl.cc b/lib/src/HttpAppFrameworkImpl.cc
@@ -87,6 +87,15 @@ HttpAppFrameworkImpl::HttpAppFrameworkImpl()
 {
 }
 
+HttpAppFramework &HttpAppFrameworkImpl::reset()
+{
+    listenerManagerPtr_ = std::make_unique<ListenerManager>();
+    pluginsManagerPtr_ = std::make_unique<PluginsManager>();
+    dbClientManagerPtr_ = std::make_unique<orm::DbClientManager>();
+    redisClientManagerPtr_ = std::make_unique<nosql::RedisClientManager>();
+    return *this;
+}
+
 static std::function<void()> f = [] {
     LOG_TRACE << "Initialize the main event loop in the main thread";
 };

diff --git a/lib/src/HttpAppFrameworkImpl.h b/lib/src/HttpAppFrameworkImpl.h
@@ -637,6 +637,7 @@ class HttpAppFrameworkImpl final : public HttpAppFramework
     void findSessionForRequest(const HttpRequestImplPtr &req);
     HttpResponsePtr handleSessionForResponse(const HttpRequestImplPtr &req,
                                              const HttpResponsePtr &resp);
+    HttpAppFramework &reset() override;
 
   private:
     void registerHttpController(const std::string &pathPattern,