clean code.

connection.hpp

@@ -4,6 +4,7 @@
 #include <boost/asio.hpp>
 #include <boost/asio/deadline_timer.hpp>
 #include "common.h"
+#include "router.hpp"
 
 using boost::asio::ip::tcp;
 

main.cpp

@@ -1,15 +1,6 @@
-#include <iostream>
 #include <string>
-#include <tuple>
-#include <fstream>
-#include <kapok/Kapok.hpp>
-#include <boost/timer.hpp>
-#include "router.hpp"
-#include "test_router.hpp"
 #include "server.hpp"
-#include "base64.hpp"
-#include "client_proxy/client_proxy.hpp"
-#include "log.hpp"
+
 
 int add(int a, int b)
 {
@@ -26,7 +17,7 @@ struct messager
 	}
 };
 
-TEST_CASE(rpc_qps, true)
+int main()
 {
 	messager m;
 	log::get().init("rest_rpc_server.lg");
@@ -35,4 +26,7 @@ TEST_CASE(rpc_qps, true)
 	s.register_handler("translate", &messager::translate, &m);
 
 	s.run();
+
+	getchar();
+	return 0;
 }

server.hpp

@@ -76,8 +76,8 @@ class server : private boost::noncopyable
 
 	void pub(const std::string& topic, const std::string& result)
 	{
-		decltype(conn_map_.equal_range(topic)) temp;
 #ifdef PUB_SUB
+		decltype(conn_map_.equal_range(topic)) temp;
 		std::unique_lock<std::mutex> lock(mtx_);
 
 		auto range = conn_map_.equal_range(topic);
@@ -100,7 +100,7 @@ class server : private boost::noncopyable
 		}
 
 		lock.lock(); //clear invalid connection
-#endif
+
 		for (auto it = conn_map_.cbegin(); it != conn_map_.end();)
 		{
 			auto ptr = it->second.lock();
@@ -113,6 +113,7 @@ class server : private boost::noncopyable
 				++it;
 			}
 		}
+#endif
 	}
 
 	//this callback from router, tell the server which connection sub the topic and the result of handler
@@ -146,14 +147,13 @@ class server : private boost::noncopyable
 #endif
 	}
 
-	//insert
-	std::multimap<std::string, std::weak_ptr<connection>> conn_map_;
 	io_service_pool io_service_pool_;
 	tcp::acceptor acceptor_;
 	std::shared_ptr<connection> conn_;
 	std::shared_ptr<std::thread> thd_;
 	std::size_t timeout_milli_;
 #ifdef PUB_SUB
+	std::multimap<std::string, std::weak_ptr<connection>> conn_map_;
 	std::mutex mtx_;
 #endif
 };