TaskDispatcher/WebSocketObserver.cpp at main · AndroidCoderPeng/TaskDispatcher · GitHub

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
98
99
100
101
102
103
#include "WebSocketObserver.hpp"

#include "Logger.hpp"

WebSocketObserver::WebSocketObserver(QObject *parent) : QObject(parent) {}

void WebSocketObserver::startServer(QString ip, quint16 port) {
  if (server) {
    Logger::Tag("WebSocketObserver").w("Server already running.");
    return;
  }

  server = new QWebSocketServer("TaskDispatcher",
                                QWebSocketServer::NonSecureMode, this);

  if (!server->listen(QHostAddress(ip), port)) {
    Logger::Tag("WebSocketObserver")
        .eFmt("Failed to start WebSocket server on port %d: %s", port,
              server->errorString().toUtf8().constData());
    delete server;
    server = nullptr;
    return;
  }

  connect(server, &QWebSocketServer::newConnection, this,
          &WebSocketObserver::onNewConnection);

  Logger::Tag("WebSocketObserver")
      .dFmt("WebSocket server started on %s:%d", ip.toUtf8().constData(), port);
  emit signalServerStateChanged(WebSocketState::RUNNING);
}

void WebSocketObserver::stopServer() {
  if (!server)
    return;

  server->close();

  for (QWebSocket *client : clients) {
    client->close();
    client->deleteLater();
  }
  clients.clear();

  server->deleteLater();
  server = nullptr;

  Logger::Tag("WebSocketObserver").i("WebSocket server stopped.");
  emit signalServerStateChanged(WebSocketState::SHUTDOWN);
}

bool WebSocketObserver::isServerRunning() const {
  return server != nullptr && server->isListening();
}

void WebSocketObserver::sendMessage(const QString &message) {
  if (clients.isEmpty()) {
    emit signalNoClient();
    return;
  }
  for (QWebSocket *client : clients) {
    if (client->isValid()) {
      client->sendTextMessage(message);
    }
  }
}

void WebSocketObserver::onNewConnection() {
  while (server->hasPendingConnections()) {
    QWebSocket *client = server->nextPendingConnection();
    clients.append(client);

    connect(client, &QWebSocket::textMessageReceived, this,
            &WebSocketObserver::onMessageReceived);
    connect(client, &QWebSocket::disconnected, this,
            &WebSocketObserver::onClientDisconnected);

    const QString clientInfo = QString("%1:%2")
                                   .arg(client->peerAddress().toString())
                                   .arg(client->peerPort());
    Logger::Tag("WebSocketObserver")
        .dFmt("Client connected: %s", clientInfo.toUtf8().constData());
  }
}

void WebSocketObserver::onMessageReceived(const QString &message) {
  emit signalDataReceived(message);
}

void WebSocketObserver::onClientDisconnected() {
  QWebSocket *client = qobject_cast<QWebSocket *>(sender());
  if (client) {
    const QString clientInfo = QString("%1:%2")
                                   .arg(client->peerAddress().toString())
                                   .arg(client->peerPort());
    Logger::Tag("WebSocketObserver")
        .dFmt("Client disconnected: %s", clientInfo.toUtf8().constData());
    clients.removeAll(client);
    client->deleteLater();
  }
}

WebSocketObserver::~WebSocketObserver() { stopServer(); }