Initial commit

Lts/Services/ExerciseEventManagement/WebSocketServer/CMakeLists.txt

@@ -0,0 +1,22 @@
+#
+# $Id$
+# $HeadURL$
+# $Author$
+#
+
+SET(websocket_SRCS
+  main.cpp
+  web_socket_handler.cpp
+  web_socket_handler.h
+)
+
+NGL_ADD_EXECUTABLE(AAR SV webSocketServer ${websocket_SRCS})
+NGL_TARGET_LINK_LIBRARIES(webSocketServer
+    Qt5::WebSockets
+    Qt5::Core
+    Qt5::Network
+    NGLLogger
+    nglcoreapplication
+    svcEemCommon
+)
+

Lts/Services/ExerciseEventManagement/WebSocketServer/main.cpp

@@ -0,0 +1,39 @@
+/**
+  * @RDE_HEADER@
+  * main for TimeLineManagement
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  */
+
+#include <nglapplication/nglcoreapplication.h>
+
+
+#include "web_socket_handler.h"
+#include "../logger.h"
+
+
+
+int main(int argc, char **argv)
+{
+  NGLCoreApplication app(argc, argv);
+
+  app.createLogger("WebSocketServer", -1);
+
+
+  int iRet = -3;
+  {
+
+    ExerciseEventManagement::WebSocketHandler webSocketServer;
+
+    webSocketServer.init(7020);
+
+    STARTUP_LOG(app.baseLogger());
+
+    iRet = app.exec();
+
+    SHUTDOWN_LOG(app.baseLogger());
+  }
+
+  return iRet;
+}

Lts/Services/ExerciseEventManagement/WebSocketServer/web_socket_handler.cpp

@@ -0,0 +1,115 @@
+/**
+  * Websocket interface for building control unit
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Christian Ehrlicher, SLSE, 2017-01-30
+  */
+
+#include "web_socket_handler.h"
+#include <QWebSocket>
+#include <iostream>
+
+#include <QJsonObject>
+#include <QJsonValue>
+#include <QJsonDocument>
+
+namespace ExerciseEventManagement
+{
+
+WebSocketHandler::WebSocketHandler(QObject *parent)
+  : QObject(parent)
+  , m_webSocketServer(new QWebSocketServer(QStringLiteral("AVS-Server"), QWebSocketServer::NonSecureMode, this))
+  , m_timer(new QTimer())
+  , m_logger(getLogger("Json.WebSocketHandler"))
+{
+  connect(m_timer, &QTimer::timeout,
+          this, &WebSocketHandler::onSendMessage);
+
+  m_timer->start(1000);
+
+}
+
+WebSocketHandler::~WebSocketHandler()
+{
+  for (auto 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 QString &jsonStr)
+{
+  if (!isConnected())
+    return false;
+
+  ERROR_LOG(m_logger) << "jsonStr" << jsonStr;
+
+  bool bOk = false;
+  for (QWebSocket *ws : qAsConst(m_webSockets))
+  {
+    if (ws->sendTextMessage(jsonStr))
+      bOk = true;
+  }
+
+  return bOk;
+}
+
+
+void WebSocketHandler::onNewConnection()
+{
+  QWebSocket *ws = m_webSocketServer->nextPendingConnection();
+  if (!ws)
+    return;
+  INFO_LOG(m_logger) << "New websocket connection from " << ws->peerAddress().toString() << ":" << ws->peerPort();
+
+  connect(ws, &QWebSocket::disconnected,
+          this, &WebSocketHandler::onWebSocketDisconnected);
+
+  m_webSockets += ws;
+
+  emit webSocketConnected(ws);
+}
+
+void WebSocketHandler::onSendMessage()
+{
+  ERROR_LOG(m_logger) << "Send message: ";
+
+  QJsonObject obj;
+
+  obj[QStringLiteral("eventTime")] = QJsonValue(QDateTime::currentMSecsSinceEpoch());
+
+  QString jsonStr = QString::fromUtf8(QJsonDocument(obj).toJson());
+  sendTelegram(jsonStr);
+
+}
+
+void WebSocketHandler::onWebSocketDisconnected()
+{
+  QWebSocket *ws = qobject_cast<QWebSocket *>(sender());
+  if (!ws)
+    return;
+  m_webSockets.remove(ws);
+  ws->deleteLater();
+}
+
+
+} // namespace ExerciseEventManagement

Lts/Services/ExerciseEventManagement/WebSocketServer/web_socket_handler.h

@@ -0,0 +1,50 @@
+/**
+  * Websocket interface for building control unit
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Christian Ehrlicher, SLSE, 2017-01-30
+  */
+
+#pragma once
+
+#include <QObject>
+#include <QQueue>
+#include <QWebSocketServer>
+#include <memory>
+#include <QTimer>
+#include "../logger.h"
+
+namespace ExerciseEventManagement
+{
+
+class WebSocketHandler : public QObject
+{
+  Q_OBJECT
+public:
+  explicit WebSocketHandler(QObject *parent = nullptr);
+  ~WebSocketHandler();
+
+  bool init(quint16 listenPort);
+
+  bool isConnected() const
+  {
+    return !m_webSockets.isEmpty();
+  }
+  bool sendTelegram(const QString &jsonStr);
+
+Q_SIGNALS:  
+  void webSocketConnected(QWebSocket *ws);
+private Q_SLOTS:
+  void onNewConnection();
+  void onWebSocketDisconnected();
+  void onSendMessage();
+private:
+  QWebSocketServer *m_webSocketServer;
+  QSet<QWebSocket *> m_webSockets;
+  QTimer           *m_timer;
+  LOGGER_T m_logger;
+};
+
+
+} // namespace ExerciseEventManagement