websocket_ssl_client.h

Go to the documentation of this file.

#ifndef CPPSERVER_ASIO_WEBSOCKET_SSL_CLIENT_H
#define CPPSERVER_ASIO_WEBSOCKET_SSL_CLIENT_H

#include "service.h"
#include "websocket.h"

#include "system/uuid.h"

namespace CppServer {
namespace Asio {


class WebSocketSSLClient : public std::enable_shared_from_this<WebSocketSSLClient>
{
public:

    explicit WebSocketSSLClient(std::shared_ptr<Service> service, std::shared_ptr<asio::ssl::context> context, const std::string& uri);
    WebSocketSSLClient(const WebSocketSSLClient&) = delete;
    WebSocketSSLClient(WebSocketSSLClient&&) = default;
    virtual ~WebSocketSSLClient() = default;

    WebSocketSSLClient& operator=(const WebSocketSSLClient&) = delete;
    WebSocketSSLClient& operator=(WebSocketSSLClient&&) = default;

    const CppCommon::UUID& id() const noexcept { return _id; }

    std::shared_ptr<Service>& service() noexcept { return _service; }
    std::shared_ptr<asio::ssl::context>& context() noexcept;
    const std::string& uri() const noexcept { return _uri; }
    WebSocketSSLClientCore& core() noexcept { return _core; }

    uint64_t messages_sent() const noexcept { return _messages_sent; }
    uint64_t messages_received() const noexcept { return _messages_received; }
    uint64_t bytes_sent() const noexcept { return _bytes_sent; }
    uint64_t bytes_received() const noexcept { return _bytes_received; }

    bool IsConnected() const noexcept { return _connected; }


    bool Connect();

    bool Disconnect(websocketpp::close::status::value code = websocketpp::close::status::normal, const std::string& reason = "") { return Disconnect(false, code, reason); }

    bool Reconnect();


    size_t Send(const void* buffer, size_t size, websocketpp::frame::opcode::value opcode = websocketpp::frame::opcode::binary);

    size_t Send(const std::string& text, websocketpp::frame::opcode::value opcode = websocketpp::frame::opcode::text);

    size_t Send(const WebSocketSSLMessage& message);

protected:
    virtual void onConnected() {}
    virtual void onDisconnected() {}


    virtual void onReceived(const WebSocketSSLMessage& message) {}


    virtual void onError(int error, const std::string& category, const std::string& message) {}

private:
    // Client Id
    CppCommon::UUID _id;
    // Asio service
    std::shared_ptr<Service> _service;
    // Server SSL context, URI address, client core and client connection
    std::shared_ptr<asio::ssl::context> _context;
    std::string _uri;
    WebSocketSSLClientCore _core;
    websocketpp::connection_hdl _connection;
    std::atomic<bool> _initialized;
    std::atomic<bool> _connected;
    // Client statistic
    uint64_t _messages_sent;
    uint64_t _messages_received;
    uint64_t _bytes_sent;
    uint64_t _bytes_received;

    void InitAsio();


    bool Disconnect(bool dispatch, websocketpp::close::status::value code = websocketpp::close::status::normal, const std::string& reason = "");

    void Connected(websocketpp::connection_hdl connection);
    void Disconnected(websocketpp::connection_hdl connection);

    void SendError(std::error_code ec);
};

} // namespace Asio
} // namespace CppServer

#endif // CPPSERVER_ASIO_WEBSOCKET_SSL_CLIENT_H