重构网络库

之前的网络库以id作为会话对象的标示，但很多应用服务器的逻辑比较复杂，每个会话对象需要携带（保存）很多状态，那么开发人员总是会构建一个对象管理器。

所以我在之前网路库封装成以C++对象为标示，作为网络事件回调以及网络接口的参数。

这里展示的是一个pingpong协议测试程序（未分包，分包不影响此协议的效率测试）

 

 工程（需要VS2013）下载地址：http://files.cnblogs.com/files/irons/PingPangDemo.zip

 

TcpService类就是设计的以id为对象标示的网络接口层。

而WrapServer则是封装TCPService的，以C++对象为标示的网络接口， 其C++对象为TCPSession。

 

通常使用其智能指针。

 

为了博客园规矩（其实是凑数），只有贴代码了（具体代码设计，还请各位花点时间看源代码了--我这里就不多做解释了哈，有问题还请指教）

 

Server：

#include <iostream>
#include <mutex>

#include "../Net/eventloop.h"
#include "../Net/WrapTCPService.h"

std::mutex g_mutex;
int total_recv = 0;
int total_client_num = 0;

void onSessionClose(TCPSession::PTR session)
{
    g_mutex.lock();
    total_client_num--;
    g_mutex.unlock();
}

int onSessionMsg(TCPSession::PTR session, const char* buffer, int len)
{
    session->send(buffer, len);
    g_mutex.lock();
    total_recv += len;
    g_mutex.unlock();
    return len;
}

int main(int argc, char **argv)
{
    int thread_num = atoi(argv[1]);
    int port_num = atoi(argv[2]);

    WrapServer::PTR server = std::make_shared<WrapServer>();

    server->setDefaultEnterCallback([](TCPSession::PTR session){
        session->setCloseCallback(onSessionClose);
        session->setDataCallback(onSessionMsg);

        g_mutex.lock();
        total_client_num++;
        g_mutex.unlock();
    });

    server->startListen(port_num);
    server->startWorkThread(thread_num);

    EventLoop mainLoop;
    mainLoop.restoreThreadID();

    while (true)
    {
        mainLoop.loop(1000);
        g_mutex.lock();
        std::cout << "total recv : " << (total_recv / 1024) / 1024 << " M /s, of client num:" << total_client_num << std::endl;
        total_recv = 0;
        g_mutex.unlock();
    }
}

 

Client：

#include <iostream>
#include <string>

#include "../Net/socketlibfunction.h"
#include "../Net/WrapTCPService.h"

void onSessionClose(TCPSession::PTR session)
{
}

int onSessionMsg(TCPSession::PTR session, const char* buffer, int len)
{
    session->send(buffer, len);
    return len;
}

int main(int argc, char **argv)
{
    int thread_num = atoi(argv[1]);
    int port_num = atoi(argv[2]);
    int num = atoi(argv[3]);
    int packet_len = atoi(argv[4]);

    std::string tmp(packet_len, 'a');

    WrapServer::PTR server = std::make_shared<WrapServer>();

    server->startWorkThread(thread_num);

    for (int i = 0; i < num; i++)
    {
        sock fd = ox_socket_connect("127.0.0.1", port_num);
        server->addSession(fd, [&](TCPSession::PTR session){
            session->setCloseCallback(onSessionClose);
            session->setDataCallback(onSessionMsg);
            session->send(tmp.c_str(), tmp.size());
        });
    }

    std::cin.get();
}

 

tks。