Initial commit

Lts/Proxies/Buildings/BuildingControlUnit/Json/web_socket_handler.cpp

@@ -0,0 +1,110 @@
+/**
+  * Websocket interface for building control unit
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Christian Ehrlicher, SLSE, 2017-01-30
+  */
+
+#include "web_socket_handler.h"
+#include "web_socket_telegrams.h"
+#include <QWebSocket>
+#include <iostream>
+
+namespace BuildingControlUnit
+{
+namespace Json
+{
+WebSocketHandler::WebSocketHandler(QObject *parent)
+  : QObject(parent)
+  , m_webSocketServer(new QWebSocketServer(QStringLiteral("BCU"), QWebSocketServer::NonSecureMode, this))
+  , m_logger(getLogger("Json.WebSocketHandler"))
+{}
+
+WebSocketHandler::~WebSocketHandler()
+{
+  for (QWebSocket *s : qAsConst(m_webSockets))
+  {
+    s->disconnect(this);
+    s->close();
+  }
+  qDeleteAll(m_webSockets);
+}
+
+bool WebSocketHandler::init(quint16 listenPort)
+{
+  if (!m_webSocketServer->listen(QHostAddress::Any, listenPort))
+  {
+    WARNING_LOG(m_logger) << "Can not listen on port " << listenPort << " - aborting";
+    return false;
+  }
+  // inform calling process via stdout about the port/url
+  std::cout << "web socket listen port: " << m_webSocketServer->serverPort() << std::endl;
+  std::cout << "web socket url: " << m_webSocketServer->serverUrl().toString() << std::endl;
+  connect(m_webSocketServer, &QWebSocketServer::newConnection,
+          this, &WebSocketHandler::onNewConnection);
+
+  return true;
+}
+
+bool WebSocketHandler::sendTelegram(const Telegrams::BaseTelegram *tele, QWebSocket *wsDirect)
+{
+  if (!isConnected() || !tele)
+    return false;
+
+  const QString jsonStr = QString::fromUtf8(tele->toJson());
+  DEBUG_LOG(m_logger) << "jsonStr" << jsonStr;
+
+  bool bOk = false;
+  for (QWebSocket *ws : qAsConst(m_webSockets))
+  {
+    if (wsDirect != nullptr && wsDirect != ws)
+      continue;
+    if (ws->sendTextMessage(jsonStr))
+      bOk = true;
+  }
+
+  return bOk;
+}
+
+
+void WebSocketHandler::onNewConnection()
+{
+  QWebSocket *ws = m_webSocketServer->nextPendingConnection();
+  if (!ws)
+    return;
+  DEBUG_LOG(m_logger) << "New websocket connection from " << ws->peerAddress().toString() << ":" << ws->peerPort();
+
+  connect(ws, &QWebSocket::textMessageReceived,
+          this, &WebSocketHandler::onTextMessageReceived);
+  connect(ws, &QWebSocket::disconnected,
+          this, &WebSocketHandler::onWebSocketDisconnected);
+
+  m_webSockets += ws;
+
+  emit webSocketConnected(ws);
+}
+
+void WebSocketHandler::onTextMessageReceived(const QString &msg)
+{
+  DEBUG_LOG(m_logger) << "new text message: " << msg;
+  auto tele = std::shared_ptr<Telegrams::BaseTelegram>(Telegrams::BaseTelegram::telegramFromJson(msg.toUtf8()));
+  if (!tele)
+  {
+    WARNING_LOG(m_logger) << "Error parsing incoming json data: '" << msg << "'";
+    return;
+  }
+  emit telegramReceived(tele);
+}
+
+void WebSocketHandler::onWebSocketDisconnected()
+{
+  QWebSocket *ws = qobject_cast<QWebSocket *>(sender());
+  if (!ws)
+    return;
+  m_webSockets.remove(ws);
+  ws->deleteLater();
+}
+
+} // namespace Json
+} // namespace BuildingControlUnit