forked from jwcpp/jwEngine
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathLua_WebSockServer.cpp
More file actions
54 lines (44 loc) · 1.37 KB
/
Copy pathLua_WebSockServer.cpp
File metadata and controls
54 lines (44 loc) · 1.37 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include <string>
#include "sol/sol.hpp"
#include "WebSocketEvent.h"
#include "WebSocketServer.h"
#include "WebSocketPacket.h"
#include "WebSocketConnect.h"
#include "EventLoop.h"
class Lua_WebSocketServer : public WebSocketServer, public WebSocketEvent
{
public:
Lua_WebSocketServer() :
WebSocketServer(EventLoop::Instance(), this)
{
}
protected:
virtual void onAccept(WebSocketConnect * conn) {
on_accept(conn);
};
virtual void onClose(WebSocketConnect * conn) {
on_close(conn);
};
virtual void onMsg(WebSocketConnect * conn, WebSocketPacket * pack) {
on_msg(conn, pack);
};
public:
std::function<void(WebSocketConnect *)> on_accept;
std::function<void(WebSocketConnect *)> on_close;
std::function<void(WebSocketConnect *, WebSocketPacket *)> on_msg;
};
void luabind_websocket(sol::state & lua)
{
lua.new_usertype<WebSocketConnect>("WebSocketConnect",
sol::base_classes, sol::bases<TcpSocket>(),
"sendPacket", &WebSocketConnect::sendPacket,
"sendMsg", &WebSocketConnect::sendData);
lua.new_usertype<Lua_WebSocketServer>("WebSocketServer",
"listen", &Lua_WebSocketServer::listen,
"shutdown", &Lua_WebSocketServer::shutdown,
"on_accept", &Lua_WebSocketServer::on_accept,
"on_close", &Lua_WebSocketServer::on_close,
"on_msg", &Lua_WebSocketServer::on_msg);
lua.new_usertype<WebSocketPacket>("WebSocketPacket",
sol::base_classes, sol::bases<BasePacket>());
}