Class EchoClient

performance/ws_echo_client.cpp:32–90

std::atomic<uint64_t> total_messages(0);

class EchoClient : public WSClient
{
public:
    EchoClient(const std::shared_ptr<Service>& service, const std::string& address, int port, int messages)
        : WSClient(service, address, port),
          _messages(messages)
    {
    }

    void SendMessage() { SendBinaryAsync(message_to_send.data(), message_to_send.size()); }

protected:
    void onWSConnecting(CppServer::HTTP::HTTPRequest& request) override
    {
        request.SetBegin("GET", "/");
        request.SetHeader("Host", "localhost");
        request.SetHeader("Origin", "http://localhost");
        request.SetHeader("Upgrade", "websocket");
        request.SetHeader("Connection", "Upgrade");
        request.SetHeader("Sec-WebSocket-Key", CppCommon::Encoding::Base64Encode(ws_nonce()));
        request.SetHeader("Sec-WebSocket-Protocol", "chat, superchat");
        request.SetHeader("Sec-WebSocket-Version", "13");
    }

    void onWSConnected(const CppServer::HTTP::HTTPResponse& response) override
    {
        for (size_t i = _messages; i > 0; --i)
            SendMessage();
    }

    void onWSReceived(const void* buffer, size_t size) override
    {
        _received += size;
        while (_received >= message_to_send.size())
        {
            SendMessage();
            _received -= message_to_send.size();
        }

        timestamp_stop = Timestamp::nano();
        total_bytes += size;
    }

    void onSent(size_t sent, size_t pending) override
    {
        _sent += sent;
    }

    void onError(int error, const std::string& category, const std::string& message) override
    {
        std::cout << "WebSocket client caught an error with code " << error << " and category '" << category << "': " << message << std::endl;
        ++total_errors;
    }

private:
    size_t _sent{0};
    size_t _received{0};
    size_t _messages{0};