mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-05 05:18:53 +08:00
166 lines
4.3 KiB
C++
166 lines
4.3 KiB
C++
#pragma once
|
||
#include "../acl_cpp_define.hpp"
|
||
#include "HttpServlet.hpp"
|
||
|
||
namespace acl
|
||
{
|
||
|
||
class websocket;
|
||
class session;
|
||
|
||
class WebSocketServlet: public HttpServlet
|
||
{
|
||
public:
|
||
WebSocketServlet(void);
|
||
|
||
/**
|
||
* 构造函数
|
||
* @param stream {socket_stream*} 当在 acl_master 服务器框架控制下
|
||
* 运行时,该参数必须非空;当在 apache 下以 CGI 方式运行时,该参数
|
||
* 设为 NULL;另外,该函数内部不会关闭流连接,应用应自行处理流对象
|
||
* 的关闭情况,这样可以方便与 acl_master 架构结合
|
||
* @param session {session*} 每一个 HttpServlet 对象一个 session 对象
|
||
*/
|
||
WebSocketServlet(socket_stream* stream, session* session);
|
||
|
||
/**
|
||
* 构造函数
|
||
* @param stream {socket_stream*} 当在 acl_master 服务器框架控制下
|
||
* 运行时,该参数必须非空;当在 apache 下以 CGI 方式运行时,该参数
|
||
* 设为 NULL;另外,该函数内部不会关闭流连接,应用应自行处理流对象
|
||
* 的关闭情况,这样可以方便与 acl_master 架构结合
|
||
* @param memcache_addr {const char*}
|
||
*/
|
||
WebSocketServlet(socket_stream* stream,
|
||
const char* memcache_addr = "127.0.0.1:11211");
|
||
|
||
/**
|
||
* HttpServlet 对象开始运行,接收 HTTP 请求,并回调以下 doXXX 虚函数,
|
||
* 该函数首先会调用 start 过程,然后根据 start 的返回结果及请求/响应
|
||
* 对象是否要求保持长连接来决定是否需要与客户端保持长连接.
|
||
* @return {bool} 返回处理结果,返回 false 表示处理失败或处理成功且
|
||
* 不保持长连接,应关闭连接
|
||
*/
|
||
|
||
virtual ~WebSocketServlet(void);
|
||
|
||
/**
|
||
* HttpServlet 对象开始运行,接收 HTTP 请求,并回调以下 doXXX 虚函数
|
||
* @param session {session&} 存储 session 数据的对象
|
||
* @param stream {socket_stream*} 当在 acl_master 服务器框架控制下
|
||
* 运行时,该参数必须非空;当在 apache 下以 CGI 方式运行时,该参数
|
||
* 设为 NULL;另外,该函数内部不会关闭流连接,应用应自行处理流对象
|
||
* 的关闭情况,这样可以方便与 acl_master 架构结合
|
||
* @return {bool} 返回处理结果
|
||
*/
|
||
virtual bool doRun(session& session, socket_stream* stream = NULL)
|
||
{
|
||
return HttpServlet::doRun(session, stream);
|
||
}
|
||
|
||
/**
|
||
* HttpServlet 对象开始运行,接收 HTTP 请求,并回调以下 doXXX 虚函数,
|
||
* 调用本函数意味着采用 memcached 来存储 session 数据
|
||
* @param memcached_addr {const char*} memcached 服务器地址,格式:IP:PORT
|
||
* @param stream {socket_stream*} 含义同上
|
||
* @return {bool} 返回处理结果
|
||
*/
|
||
virtual bool doRun(const char* memcached_addr, socket_stream* stream)
|
||
{
|
||
return HttpServlet::doRun(memcached_addr, stream);
|
||
}
|
||
|
||
virtual bool doRun(void);
|
||
|
||
/**
|
||
* 发送二进制数据.
|
||
* @param rw_timeout {const char *} 发送的数据
|
||
* @return {bool} 错误 false.否则 true
|
||
*/
|
||
|
||
bool send_binary(const char *buf, int len);
|
||
|
||
/**
|
||
* 发送文本数据.
|
||
* @param rw_timeout {const char *} 发送的数据
|
||
* @return {bool} 错误 false.否则 true
|
||
*/
|
||
|
||
bool send_text(const char *text);
|
||
|
||
/**
|
||
* 发送pong 消息.
|
||
* @param rw_timeout {const char *} 发送的数据
|
||
* @return {bool} 错误 false.否则 true
|
||
*/
|
||
bool send_pong(const char *buffer = NULL);
|
||
|
||
/**
|
||
* 发送pong 消息.
|
||
* @param rw_timeout {const char *} 发送的数据
|
||
* @return {bool} 错误 false.否则 true
|
||
*/
|
||
bool send_ping(const char *buffer = NULL);
|
||
|
||
unsigned long long get_max_msg_len(void) const
|
||
{
|
||
return max_msg_len_;
|
||
}
|
||
|
||
/**
|
||
* 设置最大消息长度,当websocket 消息大于这个值,将断开websocket连接.
|
||
* @param unsigned long long{len} 新的长度
|
||
*/
|
||
void set_max_msg_len(unsigned long long len)
|
||
{
|
||
max_msg_len_ = len;
|
||
}
|
||
|
||
protected:
|
||
/**
|
||
* websocket 关闭消息回调
|
||
* @return {void}
|
||
*/
|
||
virtual void on_close(void);
|
||
|
||
/**
|
||
* websocket ping 消息回调.
|
||
* @param {const char *} buf 消息数据
|
||
* @param {int} len 消息数据长度
|
||
* @return {bool} false 断开连接。
|
||
*/
|
||
virtual bool on_ping(const char *buf, unsigned long long len) = 0;
|
||
|
||
/**
|
||
* websocket pong 消息回调.
|
||
* @param {const char *} buf 消息数据
|
||
* @param {int} len 消息数据长度
|
||
* @return {bool} false 断开连接。
|
||
*/
|
||
virtual bool on_pong(const char *buf, unsigned long long len) = 0;
|
||
|
||
/**
|
||
* websocket ping 消息回调.
|
||
* @param data{char *} 回调数据缓存区。
|
||
* @param len{unsigned long long}回调数据缓存区长度。
|
||
* @param text{bool } true 为文本数据。否则是 二进制数据。
|
||
* @return {bool} false 断开连接。
|
||
*/
|
||
virtual bool on_message(char *data, unsigned long long len, bool text) = 0;
|
||
|
||
private:
|
||
// @override
|
||
bool doWebsocket(HttpServletRequest&, HttpServletResponse&);
|
||
|
||
private:
|
||
|
||
unsigned long long max_msg_len_;
|
||
websocket *ws_;
|
||
|
||
char *recv_buffer_;
|
||
int write_pos_;
|
||
int opcode_;
|
||
};
|
||
|
||
} // namespace acl
|