Handle exit signals.

Author: MartinRixham

src/main.cpp

@@ -1,12 +1,27 @@
 #include <memory>
+#include <signal.h>
 
 #include <boost/json/src.hpp>
 
 #include "server/server.h"
 
+std::shared_ptr<server::server> database_server;
+
+void handle_signal(int signum)
+{
+	database_server->close();
+
+	exit(signum);
+}
+
 int main(void)
 {
-	std::make_shared<server::server>(8080)->serve();
+	database_server = std::make_shared<server::server>(8080);
+
+	signal(SIGINT, handle_signal);
+	signal(SIGTERM, handle_signal);
+
+	database_server->serve();
 
 	return EXIT_SUCCESS;
 }

src/server/server.cpp

@@ -15,7 +15,7 @@ server::server::server(
 		acceptor(boost::asio::make_strand(io_context)),
 		repository(repository::rocksdb_repository()),
 		router(router::router(repository))
-{   
+{
 	boost::beast::error_code error;
 	boost::asio::ip::tcp::endpoint endpoint { boost::asio::ip::address_v4::any(), port };
 
@@ -81,3 +81,8 @@ void server::server::accept()
 		boost::asio::make_strand(io_context),
 		boost::beast::bind_front_handler(&server::on_accept, shared_from_this()));
 }
+
+void server::server::close()
+{
+	acceptor.close();
+}

src/server/server.h

@@ -37,6 +37,8 @@ namespace server
 
 		boost::asio::ip::port_type port() const;
 
+		void close();
+
 	private:
 		void accept();
 	};