caching datetime string for Date header

1 files changed, 18 insertions, 3 deletions

diff --git a/http_server.h b/http_server.h
index e4893ad..387f2e5 100644
--- a/http_server.h
+++ b/http_server.h
@@ -2,8 +2,10 @@
 
 #include <boost/asio.hpp>
 #include <stdint.h>
+#include <atomic>
 
 #include "http_connection.h"
+#include "datetime.h"
 
 // TEST
 #include <iostream>
@@ -17,7 +19,11 @@ namespace crow
     {
     public:
         Server(Handler* handler, uint16_t port, uint16_t concurrency = 1)
-            : acceptor_(io_service_, tcp::endpoint(asio::ip::address(), port)), socket_(io_service_), handler_(handler), concurrency_(concurrency)
+            : acceptor_(io_service_, tcp::endpoint(asio::ip::address(), port)), 
+            socket_(io_service_), 
+            signals_(io_service_, SIGINT, SIGTERM),
+            handler_(handler), 
+            concurrency_(concurrency)
         {
             do_accept();
         }
@@ -29,6 +35,12 @@ namespace crow
                 v.push_back(
                         std::async(std::launch::async, [this]{io_service_.run();})
                         );
+
+            signals_.async_wait(
+                [&](const boost::system::error_code& error, int signal_number){
+                    io_service_.stop();
+                });
+            
         }
 
         void stop()
@@ -52,8 +64,11 @@ namespace crow
         asio::io_service io_service_;
         tcp::acceptor acceptor_;
         tcp::socket socket_;
+        boost::asio::signal_set signals_;
+
         Handler* handler_;
-        uint16_t concurrency_ = 1;
-        std::string server_name_ = "Flask++/0.1";
+
+        uint16_t concurrency_ = {1};
+        std::string server_name_ = "Crow/0.1";
     };
 }