From 59b925737db9248d2949afe1c9194eace9b352bc Mon Sep 17 00:00:00 2001
From: an-tao <20741618@qq.com>
Date: Thu, 31 May 2018 13:55:33 +0800
Subject: [PATCH] addListener method

---
 examples/static_link_example/main.cc |  3 ++-
 lib/inc/drogon/HttpAppFramework.h    |  2 +-
 lib/src/HttpAppFramework.cc          | 22 +++++++++++++---------
 3 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/examples/static_link_example/main.cc b/examples/static_link_example/main.cc
index d2f6bb01..df7cb4b4 100755
--- a/examples/static_link_example/main.cc
+++ b/examples/static_link_example/main.cc
@@ -28,7 +28,8 @@ int main()
 {
 
     std::cout<<banner<<std::endl;
-    drogon::HttpAppFramework::instance().setListening("0.0.0.0",12345);
+    drogon::HttpAppFramework::instance().addListener("0.0.0.0",12345);
+    drogon::HttpAppFramework::instance().addListener("0.0.0.0",8080);
     trantor::Logger::setLogLevel(trantor::Logger::INFO);
     drogon::HttpAppFramework::instance().run();
 }
diff --git a/lib/inc/drogon/HttpAppFramework.h b/lib/inc/drogon/HttpAppFramework.h
index e7902e59..761bfbc3 100755
--- a/lib/inc/drogon/HttpAppFramework.h
+++ b/lib/inc/drogon/HttpAppFramework.h
@@ -36,7 +36,7 @@ namespace drogon
     {
     public:
         static HttpAppFramework &instance();
-        virtual void setListening(const std::string &ip,uint16_t port)=0;
+        virtual void addListener(const std::string &ip,uint16_t port)=0;
         virtual void run()=0;
         virtual ~HttpAppFramework();
         virtual void registerHttpSimpleController(const std::string &pathName,const std::string &crtlName,const std::vector<std::string> &filters=
diff --git a/lib/src/HttpAppFramework.cc b/lib/src/HttpAppFramework.cc
index d7385759..2f41a786 100755
--- a/lib/src/HttpAppFramework.cc
+++ b/lib/src/HttpAppFramework.cc
@@ -19,13 +19,14 @@
 #include <algorithm>
 #include <drogon/HttpSimpleController.h>
 #include <drogon/version.h>
+#include <memory>
 namespace drogon
 {
 
     class HttpAppFrameworkImpl:public HttpAppFramework
     {
     public:
-        virtual void setListening(const std::string &ip,uint16_t port) override;
+        virtual void addListener(const std::string &ip,uint16_t port) override;
         virtual void run() override ;
         virtual void registerHttpSimpleController(const std::string &pathName,const std::string &crtlName,const std::vector<std::string> &filters=
         std::vector<std::string>())override ;
@@ -33,8 +34,7 @@ namespace drogon
         virtual void disableSession() override { _useSession=false;}
         ~HttpAppFrameworkImpl(){}
     private:
-        std::string _ip;
-        uint16_t _port;
+        std::vector<std::pair<std::string,uint16_t>> _listeners;
         void onAsyncRequest(const HttpRequest& req,std::function<void (HttpResponse &)>callback);
         void readSendFile(const std::string& filePath,const HttpRequest& req, HttpResponse* resp);
 
@@ -84,19 +84,23 @@ void HttpAppFrameworkImpl::registerHttpSimpleController(const std::string &pathN
     _simpCtrlMap[path].controllerName=crtlName;
     _simpCtrlMap[path].filtersName=filters;
 }
-void HttpAppFrameworkImpl::setListening(const std::string &ip,uint16_t port)
+void HttpAppFrameworkImpl::addListener(const std::string &ip, uint16_t port)
 {
-    _ip=ip;
-    _port=port;
+    _listeners.push_back(std::make_pair(ip,port));
 }
 
 void HttpAppFrameworkImpl::run()
 {
     trantor::EventLoop loop;
-    HttpServer httpServer(&loop,InetAddress(_ip,_port),"");
-    httpServer.setHttpAsyncCallback(std::bind(&HttpAppFrameworkImpl::onAsyncRequest,this,_1,_2));
+    std::vector<std::shared_ptr<HttpServer>> servers;
+    for(auto listener:_listeners)
+    {
+        auto serverPtr=std::make_shared<HttpServer>(&loop,InetAddress(listener.first,listener.second),"drogon");
+        serverPtr->setHttpAsyncCallback(std::bind(&HttpAppFrameworkImpl::onAsyncRequest,this,_1,_2));
+        serverPtr->start();
+        servers.push_back(serverPtr);
+    }
     _sessionMapPtr=std::unique_ptr<CacheMap<std::string,SessionPtr>>(new CacheMap<std::string,SessionPtr>(&loop,1,1200));
-    httpServer.start();
     loop.loop();
 }