2019-07-27 22:44:32 +08:00
|
|
|
|
#pragma once
|
2018-12-12 11:16:12 +08:00
|
|
|
|
|
|
|
|
|
|
class WebsocketServlet_impl : public acl::WebSocketServlet
|
|
|
|
|
|
{
|
|
|
|
|
|
public:
|
|
|
|
|
|
WebsocketServlet_impl(acl::redis_client_cluster& cluster, size_t max_conns);
|
|
|
|
|
|
~WebsocketServlet_impl(void);
|
|
|
|
|
|
|
|
|
|
|
|
acl::session& get_session(void) const
|
|
|
|
|
|
{
|
|
|
|
|
|
return *session_;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
|
|
// @override
|
|
|
|
|
|
bool doUnknown(acl::HttpServletRequest&, acl::HttpServletResponse&);
|
|
|
|
|
|
// @override
|
|
|
|
|
|
bool doGet(acl::HttpServletRequest&, acl::HttpServletResponse&);
|
|
|
|
|
|
// @override
|
|
|
|
|
|
bool doPost(acl::HttpServletRequest&, acl::HttpServletResponse&);
|
|
|
|
|
|
|
|
|
|
|
|
//for websocket
|
|
|
|
|
|
/**
|
|
|
|
|
|
* @override
|
2019-07-27 22:44:32 +08:00
|
|
|
|
* websocket 关闭消息回调
|
2018-12-12 11:16:12 +08:00
|
|
|
|
* @return {void}
|
|
|
|
|
|
*/
|
|
|
|
|
|
void onClose(void) {}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
|
* @override
|
|
|
|
|
|
*/
|
|
|
|
|
|
bool onPing(unsigned long long len, bool finish);
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
|
* @override
|
|
|
|
|
|
*/
|
|
|
|
|
|
bool onPong(unsigned long long len, bool finish);
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
|
* @override
|
|
|
|
|
|
*/
|
|
|
|
|
|
bool onMessage(unsigned long long len, bool text, bool finish);
|
|
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
acl::session* session_;
|
|
|
|
|
|
int step_;
|
|
|
|
|
|
acl::string filename_;
|
|
|
|
|
|
long long filesize_;
|
|
|
|
|
|
long long nread_;
|
|
|
|
|
|
acl::ofstream fp_;
|
|
|
|
|
|
|
|
|
|
|
|
bool getFilename(unsigned long long payload_len);
|
|
|
|
|
|
bool getFilesize(unsigned long long payload_len);
|
|
|
|
|
|
bool readData(unsigned long long len, acl::string& path);
|
|
|
|
|
|
bool saveFile(unsigned long long len);
|
|
|
|
|
|
};
|
