c++ websocket 客户端
本次创建一个c++ 的websocket客户端，不依赖于其他库
头文件
#ifndef _WS_CLIENT_H
#define _WS_CLIENT_H
#ifdef _WIN32
#define _CRT_SECURE_NO_WARNINGS
#define WIN32_LEAN_AND_MEAN
#include 实现cpp
#include "wsclient.h"
#include 调用
封装一个class，实现线程
class ws_class //:public TThreadRunable
{
thread _thread;
std::mutex _mutex;
std::condition_variable _cond;
WebSocket _ws;
int _stop = 0;
string _url;
public:
static int InitSock()
{
#ifdef _WIN32
INT rc;
WSADATA wsaData;
rc = WSAStartup(MAKEWORD(2, 2), &wsaData);
if (rc) {
printf("WSAStartup Failed.\n");
return -1;
}
#endif
return 0;
}
static void UnInitSock()
{
WSACleanup();
}
ws_class()
{}
~ws_class()
{
_ws.close();
}
callback_message_recv _recv = NULL;
public:
void set_url(const char * url)
{
_url = url;
}
int connect()
{
if (_url.empty())
return -1;
return _ws.connect(_url);
}
void Start(callback_message_recv recv)
{
_ws.initSize(0, 0);
_recv = recv;
_thread = std::thread(std::bind(&ws_class::Run, this));
//return 0;
}
void Join()
{
if (_thread.joinable())
_thread.join();
}
void send(char * str)
{
if (str != NULL)
{
if (_ws.getReadyState() != CLOSED)
{
_ws.send(str);
//_ws.pollSend(10);
}
}
}
void sendBinary(uint8_t *data, int len)
{
if (_ws.getReadyState() != CLOSED)
{
_ws.sendBinary(data, len);
//_ws.pollSend(0);
}
}
void Stop()
{
_stop = 1;
}
void Run()
{
while (_stop == 0) {
//WebSocket::pointer wsp = &*ws; // <-- because a unique_ptr cannot be copied into a lambda
if (_stop == 1)
break;
if (_ws.getReadyState() == CLOSED)
{
//断线重连
if (connect() != 0)
{
for (int i = 0; i < 20; i++)
{
std::this_thread::sleep_for(std::chrono::milliseconds(100));
if (_stop == 1)
break;
}
}
}
else
{
_ws.pollRecv(10);
_ws.dispatch(_recv);
//_ws.dispatch([](void * pUser, const char * message, int len, int type) {
//});
}
//std::this_thread::sleep_for(std::chrono::milliseconds(10));
}
_ws.close();
//std::cout << "server exit" << endl;
_stop = 1;
}
void WaitForSignal()
{
std::unique_lock<std::mutex> ul(_mutex);
_cond.wait(ul);
}
void Notify()
{
_cond.notify_one();
}
};
总结
实现websocket服务端和客户端并不是很难，只是很多知识点，本次的实现还有很多不理想之处，会在应用中修正。