simple_rpc_server/CommandHandler.cpp at master · qdore/simple_rpc_server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/*********************************************************************************
*     File Name           :     CommandHandler.cpp
*     Created By          :     QiangWei
*     Creation Date       :     [2015-04-15 13:35]
*     Last Modified       :     [2015-04-15 15:14]
*     Description         :
**********************************************************************************/

#include "CommandHandler.h"
#include <cctype>
#include <iostream>
#include <string>
#include <vector>

using std::string;

namespace chl {

//去尾部空白字符
inline string rtrim(string ss)
{
    int (*pf)(int) = std::iscntrl;
    string::reverse_iterator p = find_if(ss.rbegin(), ss.rend(),
            not1(std::ptr_fun(pf)));
    ss.erase(p.base(), ss.end());
    return ss;
}

CommandHandler::CommandHandler(boost::asio::io_service& io, int port_number) :
    __io_service(io), __acceptor(io, boost::asio::ip::tcp::endpoint(
                boost::asio::ip::tcp::v4(), port_number))
{
    start_listen();
}

void CommandHandler::bind_command_with_func(const std::string& command,
        const boost::function<std::string()>& func)
{
    __comand_and_funcs.insert(std::make_pair(command, func));
}

void CommandHandler::start_listen()
{
    boost::asio::ip::tcp::socket* new_socket =
            new boost::asio::ip::tcp::socket(__io_service);
    __acceptor.async_accept(*new_socket,
            boost::bind(&CommandHandler::handle_accept, this, new_socket,
                boost::asio::placeholders::error));
}

void CommandHandler::handle_accept(boost::asio::ip::tcp::socket* socket,
        const boost::system::error_code& error)
{
    if (!error) {
        char* __data = new char[max_length];
        socket->async_read_some(boost::asio::buffer(__data, max_length),
                boost::bind(&CommandHandler::handle_read, this,
                    socket, __data, boost::asio::placeholders::error,
                    boost::asio::placeholders::bytes_transferred));
    }
    else {
        delete socket;
    }
    this->start_listen();
}

void CommandHandler::process_command(const std::string& command,
        boost::asio::ip::tcp::socket* socket) {
    std::string ret(__comand_and_funcs[command]());
    std::cout << "scker" << std::endl;
    boost::system::error_code ignored_error;
    socket->write_some(boost::asio::buffer(ret), ignored_error);
    if (socket != NULL) {
        delete socket;
        socket = NULL;
    }
}

void CommandHandler::handle_read(boost::asio::ip::tcp::socket* socket,
        char* __data, const boost::system::error_code& error,
        size_t bytes)
{
    if (!error) {
        std::string command(rtrim(__data));
        if (__data != NULL) {
            delete[] __data;
            __data = NULL;
        }
        if (__comand_and_funcs.find(command) != __comand_and_funcs.end())
        {
            boost::thread t1(boost::bind(&CommandHandler::process_command, this, command, socket));
            t1.detach();
        }
        else {
            boost::system::error_code ignored_error;
            socket->write_some(boost::asio::buffer(std::string("error: this command [")
                        + command + "] not found!"), ignored_error);
            if (socket != NULL) {
                delete socket;
                socket = NULL;
            }
        }
    }
    else {
        if (__data != NULL) {
            delete[] __data;
            __data = NULL;
        }
        if (socket != NULL) {
            delete socket;
            socket = NULL;
        }
    }
}

}