72 lines
1.7 KiB
C
72 lines
1.7 KiB
C
|
#ifndef HV_WEBSOCKET_CLIENT_H_
|
||
|
|
#define HV_WEBSOCKET_CLIENT_H_
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @demo examples/websocket_client_test.cpp
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "hexport.h"
|
||
|
|
|
||
|
|
#include "TcpClient.h"
|
||
|
|
#include "WebSocketChannel.h"
|
||
|
|
|
||
|
|
#include "HttpParser.h"
|
||
|
|
#include "WebSocketParser.h"
|
||
|
|
|
||
|
|
namespace hv {
|
||
|
|
|
||
|
|
class HV_EXPORT WebSocketClient : public TcpClientTmpl<WebSocketChannel> {
|
||
|
|
public:
|
||
|
|
std::string url;
|
||
|
|
std::function<void()> onopen;
|
||
|
|
std::function<void()> onclose;
|
||
|
|
std::function<void(const std::string& msg)> onmessage;
|
||
|
|
// PATCH: onmessage not given opcode
|
||
|
|
enum ws_opcode opcode() { return channel ? channel->opcode : WS_OPCODE_CLOSE; }
|
||
|
|
|
||
|
|
WebSocketClient(EventLoopPtr loop = NULL);
|
||
|
|
virtual ~WebSocketClient();
|
||
|
|
|
||
|
|
// url = ws://ip:port/path
|
||
|
|
// url = wss://ip:port/path
|
||
|
|
int open(const char* url, const http_headers& headers = DefaultHeaders);
|
||
|
|
int close();
|
||
|
|
int send(const std::string& msg);
|
||
|
|
int send(const char* buf, int len, enum ws_opcode opcode = WS_OPCODE_BINARY);
|
||
|
|
|
||
|
|
// setConnectTimeout / setPingInterval / setReconnect
|
||
|
|
void setPingInterval(int ms) {
|
||
|
|
ping_interval = ms;
|
||
|
|
}
|
||
|
|
|
||
|
|
// NOTE: call before open
|
||
|
|
void setHttpRequest(const HttpRequestPtr& req) {
|
||
|
|
http_req_ = req;
|
||
|
|
}
|
||
|
|
|
||
|
|
// NOTE: call when onopen
|
||
|
|
const HttpResponsePtr& getHttpResponse() {
|
||
|
|
return http_resp_;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
enum State {
|
||
|
|
CONNECTING,
|
||
|
|
CONNECTED,
|
||
|
|
WS_UPGRADING,
|
||
|
|
WS_OPENED,
|
||
|
|
WS_CLOSED,
|
||
|
|
} state;
|
||
|
|
HttpParserPtr http_parser_;
|
||
|
|
HttpRequestPtr http_req_;
|
||
|
|
HttpResponsePtr http_resp_;
|
||
|
|
WebSocketParserPtr ws_parser_;
|
||
|
|
// ping/pong
|
||
|
|
int ping_interval;
|
||
|
|
int ping_cnt;
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif // HV_WEBSOCKET_CLIENT_H_
|