websocket_connection.hpp

Go to the documentation of this file.

#ifndef CPP_WEB_SERVER_WEBSOCKET_CONNECTION_HPP
#define CPP_WEB_SERVER_WEBSOCKET_CONNECTION_HPP

#include "async_web_server_cpp/http_connection.hpp"
#include "async_web_server_cpp/websocket_message.hpp"

#include <boost/asio.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/noncopyable.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/thread/mutex.hpp>

namespace async_web_server_cpp
{

class WebsocketHttpRequestHandler;

class WebsocketConnection;
typedef boost::shared_ptr<WebsocketConnection> WebsocketConnectionPtr;
typedef boost::weak_ptr<WebsocketConnection> WebsocketConnectionWeakPtr;

class WebsocketConnection
    : public boost::enable_shared_from_this<WebsocketConnection>,
      private boost::noncopyable
{
public:
    explicit WebsocketConnection(HttpConnectionPtr connection);

    typedef boost::function<void(const WebsocketMessage& message)>
        MessageHandler;

    bool sendTextMessage(const std::string& content);
    bool sendPingMessage(const std::string& content = "");

    bool sendMessage(const WebsocketMessage& message);
    bool sendFrame(WebsocketFrame& frame);

private:
    static void static_handle_read(WebsocketConnectionWeakPtr weak_this,
                                   const char* begin, const char* end);
    void handle_read(const char* begin, const char* end);
    HttpConnectionPtr connection_;

    void set_message_handler(MessageHandler& handler);
    MessageHandler handler_;

    WebsocketFrame frame_;
    WebsocketMessage message_;
    WebsocketFrameParser frame_parser_;
    WebsocketFrameBuffer frame_buffer_;

    friend class WebsocketHttpRequestHandler;
};

}  // namespace async_web_server_cpp

#endif