From 7cf8cc7567ab612b55f150c6bb4f71395256084a Mon Sep 17 00:00:00 2001
From: yang li
Date: Wed, 9 Jan 2019 16:22:34 +0800
Subject: [PATCH] Trim codes.
---
examples/client/client.cpp | 31 ++++-----------------
examples/echo_server/echo_server.cpp | 41 ++++++++--------------------
2 files changed, 17 insertions(+), 55 deletions(-)
diff --git a/examples/client/client.cpp b/examples/client/client.cpp
index 5f69fb0..60b44cd 100644
--- a/examples/client/client.cpp
+++ b/examples/client/client.cpp
@@ -90,24 +90,13 @@ private:
class short_client : public multi_client_base, protected i_controller
{
public:
- short_client(service_pump& service_pump_) : multi_client_base(service_pump_) {}
-
- void set_server_addr(unsigned short _port, const std::string& _ip = ASCS_SERVER_IP)
- {
- port = _port;
- ip = _ip;
- }
+ short_client(service_pump& service_pump_) : multi_client_base(service_pump_) {}
+ void set_server_addr(unsigned short _port, const std::string& _ip = ASCS_SERVER_IP) {port = _port; ip = _ip;}
bool send_msg(const std::string& msg)
{
auto socket_ptr = add_socket(port, ip);
- if (socket_ptr)
- {
- socket_ptr->set_controller(this);
- return socket_ptr->send_msg(msg);
- }
-
- return false;
+ return socket_ptr ? socket_ptr->set_controller(this), socket_ptr->send_msg(msg) : false;
}
protected:
@@ -151,22 +140,14 @@ int main(int argc, const char* argv[])
// argv[2] = "::1" //ipv6
// argv[2] = "127.0.0.1" //ipv4
unsigned short port = ASCS_SERVER_PORT + 100;
- std::string ip;
+ std::string ip = ASCS_SERVER_IP;
if (argc > 1)
port = (unsigned short) atoi(argv[1]);
if (argc > 2)
ip = argv[2];
- if (!ip.empty())
- {
- client.set_server_addr(port, ip);
- client2.set_server_addr(port + 1, ip);
- }
- else
- {
- client.set_server_addr(port);
- client2.set_server_addr(port + 1);
- }
+ client.set_server_addr(port, ip);
+ client2.set_server_addr(port + 1, ip);
sp.start_service();
auto t = create_sync_recv_thread(client);
diff --git a/examples/echo_server/echo_server.cpp b/examples/echo_server/echo_server.cpp
index 51d5572..63e8b71 100644
--- a/examples/echo_server/echo_server.cpp
+++ b/examples/echo_server/echo_server.cpp
@@ -185,24 +185,12 @@ protected:
//msg handling
#ifdef ASCS_SYNC_DISPATCH
//do not hold msg_can for further using, return from on_msg as quickly as possible
- virtual size_t on_msg(std::list& msg_can)
- {
- auto re = server_socket_base::on_msg(msg_can);
- force_shutdown();
-
- return re;
- }
+ virtual size_t on_msg(std::list& msg_can) {auto re = server_socket_base::on_msg(msg_can); force_shutdown(); return re;}
#endif
#ifdef ASCS_DISPATCH_BATCH_MSG
//do not hold msg_can for further using, access msg_can and return from on_msg_handle as quickly as possible
- virtual size_t on_msg_handle(out_queue_type& msg_can)
- {
- auto re = server_socket_base::on_msg_handle(msg_can);
- force_shutdown();
-
- return re;
- }
+ virtual size_t on_msg_handle(out_queue_type& msg_can) {auto re = server_socket_base::on_msg_handle(msg_can); force_shutdown(); return re;}
#else
virtual bool on_msg_handle(out_msg_type& msg) {auto re = server_socket_base::on_msg_handle(msg); force_shutdown(); return re;}
#endif
@@ -227,23 +215,16 @@ int main(int argc, const char* argv[])
server_base short_server(sp);
echo_server echo_server_(sp); //echo server
+ unsigned short port = ASCS_SERVER_PORT;
+ std::string ip;
+ if (argc > 2)
+ port = (unsigned short) atoi(argv[2]);
if (argc > 3)
- {
- normal_server.set_server_addr(atoi(argv[2]) + 100, argv[3]);
- short_server.set_server_addr(atoi(argv[2]) + 101, argv[3]);
- echo_server_.set_server_addr(atoi(argv[2]), argv[3]);
- }
- else if (argc > 2)
- {
- normal_server.set_server_addr(atoi(argv[2]) + 100);
- short_server.set_server_addr(atoi(argv[2]) + 101);
- echo_server_.set_server_addr(atoi(argv[2]));
- }
- else
- {
- normal_server.set_server_addr(ASCS_SERVER_PORT + 100);
- short_server.set_server_addr(ASCS_SERVER_PORT + 101);
- }
+ ip = argv[3];
+
+ normal_server.set_server_addr(port + 100, ip);
+ short_server.set_server_addr(port + 101, ip);
+ echo_server_.set_server_addr(port, ip);
auto thread_num = 1;
if (argc > 1)
--
GitLab