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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
#include "netwebsocketconnection.h"
#include "networkmanager.h"
#include <QNetworkRequest>
#include <QUrl>
NetWebSocketConnection::NetWebSocketConnection(NetworkManager *networkManager)
: NetConnection(networkManager)
, m_network_manager(networkManager)
, m_socket(new QWebSocket(QString(), QWebSocketProtocol::VersionLatest, this))
, m_last_state(QAbstractSocket::UnconnectedState)
{
connect(m_socket, QOverload<QAbstractSocket::SocketError>::of(&QWebSocket::error), this, &NetWebSocketConnection::onError);
connect(m_socket, &QWebSocket::stateChanged, this, &NetWebSocketConnection::onStateChanged);
connect(m_socket, &QWebSocket::textMessageReceived, this, &NetWebSocketConnection::onTextMessageReceived);
}
NetWebSocketConnection::~NetWebSocketConnection()
{
m_socket->disconnect(this);
disconnectFromServer();
}
bool NetWebSocketConnection::isConnected()
{
return m_last_state == QAbstractSocket::ConnectedState;
}
void NetWebSocketConnection::connectToServer(ServerInfo &server)
{
disconnectFromServer();
QUrl url;
url.setScheme("ws");
url.setHost(server.ip);
url.setPort(server.port);
QNetworkRequest req(url);
req.setHeader(QNetworkRequest::UserAgentHeader, m_network_manager->get_user_agent());
m_socket->open(req);
}
void NetWebSocketConnection::disconnectFromServer()
{
if (isConnected())
{
m_socket->close(QWebSocketProtocol::CloseCodeGoingAway);
}
}
void NetWebSocketConnection::sendPacket(AOPacket packet)
{
m_socket->sendTextMessage(packet.toString(true));
}
void NetWebSocketConnection::onError()
{
Q_EMIT errorOccurred(m_socket->errorString());
}
void NetWebSocketConnection::onStateChanged(QAbstractSocket::SocketState state)
{
m_last_state = state;
switch (state)
{
default:
break;
case QAbstractSocket::ConnectedState:
Q_EMIT connectedToServer();
break;
case QAbstractSocket::UnconnectedState:
Q_EMIT disconnectFromServer();
break;
}
}
void NetWebSocketConnection::onTextMessageReceived(QString message)
{
if (!message.endsWith("#%"))
{
return;
}
message.chop(2);
QStringList raw_content = message.split('#');
const QString header = raw_content.takeFirst();
for (QString &data : raw_content)
{
data = AOPacket::decode(data);
}
Q_EMIT receivedPacket(AOPacket(header, raw_content));
}
|
