Function main

examples/BenchWebsocket.cpp:41–120

}

int main(int argc, char** argv)
{
    if (argc < 3)
    {
        std::cout << "./benchwebsocket host port [ connections workers ]";
        return 1;
    }

    const char* host = argv[1];
    int port = std::atoi(argv[2]);
    int connections = argc > 3 ? std::atoi(argv[3]) : 200;
    size_t workers = argc > 4 ? std::atoi(argv[4]) : std::thread::hardware_concurrency();

    std::cout << "host: " << host << ':' << port << " | connections: " << connections << " | workers: " << workers << std::endl;

    auto enterCallback = [host](const HttpSession::Ptr& httpSession, HttpSessionHandlers& handlers) {
        HttpRequest request;
        request.setMethod(HttpRequest::HTTP_METHOD::HTTP_METHOD_GET);
        request.setUrl("/ws");
        request.addHeadValue("Host", host);
        request.addHeadValue("Upgrade", "websocket");
        request.addHeadValue("Connection", "Upgrade");
        request.addHeadValue("Sec-WebSocket-Key", "dGhlIHNhbXBsZSBub25jZQ==");
        request.addHeadValue("Sec-WebSocket-Version", "13");

        std::string requestStr = request.getResult();
        httpSession->send(requestStr.c_str(), requestStr.size());

        handlers.setWSConnected([](const HttpSession::Ptr& session, const HTTPParser&) {
            for (int i = 0; i < 200; i++)
            {
                sendPacket(session, "hello, world!", 13);
            }
        });

        handlers.setWSCallback([](const HttpSession::Ptr& session,
                                  WebSocketFormat::WebSocketFrameType, const std::string& payload) {
            std::cout << payload << std::endl;
            sendPacket(session, "hello, world!", 13);
            count += 1;
        });
    };

    auto service = IOThreadTcpService::Create();
    service->startWorkerThread(workers);

    auto connector = AsyncConnector::Create();
    connector->startWorkerThread();

    wrapper::HttpConnectionBuilder connectionBuilder;
    connectionBuilder.WithService(service)
            .WithConnector(connector)
            .WithMaxRecvBufferSize(1024 * 1024);

    for (int i = 0; i < connections; i++)
    {
        connectionBuilder
                .WithAddr(host, port)