mirror of
https://github.com/sogou/workflow.git
synced 2026-10-21 14:52:26 +08:00
liyingxin
89 lines
2.1 KiB
C++
89 lines
2.1 KiB
C++
/*
|
|
Copyright (c) 2021 Sogou, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
|
|
Author: Li Yingxin (liyingxin@sogou-inc.com)
|
|
*/
|
|
|
|
#ifndef _WFWEBSOCKETCLIENT_H_
|
|
#define _WFWEBSOCKETCLIENT_H_
|
|
|
|
#include <string>
|
|
#include <functional>
|
|
#include "HttpUtil.h"
|
|
#include "HttpMessage.h"
|
|
#include "WFChannel.h"
|
|
#include "WebSocketMessage.h"
|
|
|
|
struct WFWebSocketParams
|
|
{
|
|
const char *url;
|
|
int idle_timeout;
|
|
int ping_interval;
|
|
size_t size_limit;
|
|
bool random_masking_key;
|
|
const char *sec_protocol;
|
|
const char *sec_version;
|
|
};
|
|
|
|
static constexpr struct WFWebSocketParams WEBSOCKET_PARAMS_DEFAULT =
|
|
{
|
|
.url = NULL,
|
|
.idle_timeout = WS_HANDSHAKE_TIMEOUT,
|
|
.ping_interval = -1,
|
|
.size_limit = (size_t)-1,
|
|
.random_masking_key = true,
|
|
.sec_protocol = NULL,
|
|
.sec_version = NULL,
|
|
};
|
|
|
|
class WebSocketClient
|
|
{
|
|
public:
|
|
using websocket_close_t = std::function<void ()>;
|
|
|
|
int init(const std::string& url);
|
|
int init(const struct WFWebSocketParams *params);
|
|
void deinit();
|
|
|
|
WFWebSocketTask *create_websocket_task(websocket_callback_t cb);
|
|
WFWebSocketTask *create_ping_task(websocket_callback_t cb);
|
|
WFWebSocketTask *create_close_task(websocket_callback_t cb);
|
|
|
|
private:
|
|
void channel_callback(WFChannel<protocol::WebSocketFrame> *channel,
|
|
websocket_close_t close);
|
|
|
|
private:
|
|
ComplexWebSocketChannel *channel;
|
|
websocket_process_t process;
|
|
websocket_close_t close;
|
|
|
|
public:
|
|
WebSocketClient(websocket_process_t process) :
|
|
process(std::move(process)),
|
|
close(nullptr)
|
|
{ }
|
|
|
|
WebSocketClient(websocket_process_t process,
|
|
websocket_close_t close) :
|
|
process(std::move(process)),
|
|
close(std::move(close))
|
|
{ }
|
|
|
|
};
|
|
|
|
#endif
|
|
|