websocket_session.h

Go to the documentation of this file.

#ifndef CPPSERVER_ASIO_WEBSOCKET_SESSION_H
#define CPPSERVER_ASIO_WEBSOCKET_SESSION_H

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

#include "system/uuid.h"

namespace CppServer {
namespace Asio {

template <class TServer, class TSession>
class WebSocketServer;


template <class TServer, class TSession>
class WebSocketSession : public std::enable_shared_from_this<WebSocketSession<TServer, TSession>>
{
    template <class TSomeServer, class TSomeSession>
    friend class WebSocketServer;

public:

    explicit WebSocketSession(std::shared_ptr<WebSocketServer<TServer, TSession>> server);
    WebSocketSession(const WebSocketSession&) = delete;
    WebSocketSession(WebSocketSession&&) = default;
    virtual ~WebSocketSession() = default;

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

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

    std::shared_ptr<Service>& service() noexcept { return _server->service(); }
    std::shared_ptr<WebSocketServer<TServer, TSession>>& server() noexcept { return _server; }
    websocketpp::connection_hdl& connection() noexcept { return _connection; }

    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 Disconnect(websocketpp::close::status::value code = websocketpp::close::status::normal, const std::string& reason = "") { return Disconnect(false, code, reason); }


    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 WebSocketMessage& message);

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


    virtual void onReceived(const WebSocketMessage& message) {}


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

private:
    // Session Id
    CppCommon::UUID _id;
    // Session server & connection
    std::shared_ptr<WebSocketServer<TServer, TSession>> _server;
    websocketpp::connection_hdl _connection;
    std::atomic<bool> _connected;
    // Session statistic
    uint64_t _messages_sent;
    uint64_t _messages_received;
    uint64_t _bytes_sent;
    uint64_t _bytes_received;


    void Connect(websocketpp::connection_hdl connection);

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

    void Disconnected();

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

} // namespace Asio
} // namespace CppServer

#endif // CPPSERVER_ASIO_WEBSOCKET_SESSION_H