95 lines
2.5 KiB
C++
Executable File
95 lines
2.5 KiB
C++
Executable File
|
|
|
|
|
|
#pragma once
|
|
|
|
#include <QtCore/QObject>
|
|
#include <QtCore/QQueue>
|
|
#include <QtWebSockets/QWebSocketServer>
|
|
#include <memory>
|
|
#include <TrainingAreaBase/training_area_telegrams.h>
|
|
|
|
#include <Common/logger.h>
|
|
|
|
namespace HighLevelArchitecture
|
|
{
|
|
class MainHandler;
|
|
|
|
|
|
class WebSocketWrapper;
|
|
|
|
class WebSocketHandler : public QObject
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit WebSocketHandler(int exerciseId, MainHandler *mainHandler, QObject *parent = nullptr);
|
|
~WebSocketHandler();
|
|
|
|
bool init(quint16 listenPort);
|
|
|
|
bool isConnected() const
|
|
{
|
|
return !m_webSockets.isEmpty();
|
|
}
|
|
template <typename T>
|
|
bool sendTelegram(const T *data, int exerciseId)
|
|
{
|
|
if (!isConnected() || !data || m_webSockets.isEmpty())
|
|
return false;
|
|
|
|
return sendTelegramInternal(data,
|
|
TrainingAreaBase::TetraIcd::Constraints<T>::type(),
|
|
exerciseId);
|
|
}
|
|
|
|
Q_SIGNALS:
|
|
void initRequest(const TrainingAreaBase::TetraIcd::Init &request);
|
|
void biteRequest(const TrainingAreaBase::TetraIcd::DemandBite &request);
|
|
void vulnerabilityStateRequest(const TrainingAreaBase::TetraIcd::DemandVulnerabilityStateChange &request);
|
|
public Q_SLOTS:
|
|
void sendHlaConnectionState();
|
|
void sendParticipantConnectionStates();
|
|
void onExternalEquipmentNumberAdded(int equipNo);
|
|
void onExternalEquipmentNumberRemoved(int equipNo);
|
|
|
|
private:
|
|
NGL::INGLLogger &m_logger;
|
|
void sendDataCommunicationState(int equipNo, bool bIsAvailable);
|
|
bool sendTelegramInternal(const TrainingAreaBase::TetraIcd::TrainingAreaTelegram *data,
|
|
const QString &type, int exerciseId);
|
|
private Q_SLOTS:
|
|
void onNewConnection();
|
|
void onWebSocketDisconnected();
|
|
void onTextMessageReceived(const QByteArray &baJson);
|
|
private:
|
|
NGL::INGLLogger &m_logger;
|
|
QWebSocketServer *m_webSocketServer;
|
|
QVector<std::shared_ptr<WebSocketWrapper>> m_webSockets;
|
|
MainHandler *m_mainHandler;
|
|
int m_exerciseId;
|
|
};
|
|
|
|
class WebSocketWrapper : public QObject
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
explicit WebSocketWrapper(QWebSocket *ws);
|
|
~WebSocketWrapper();
|
|
|
|
bool sendTextMessage(const QString &msg, int exerciseId);
|
|
|
|
void setExerciseId(int exerciseId);
|
|
int exerciseId() const { return m_exerciseId; }
|
|
QWebSocket *webSocket() const { return m_webSocket; }
|
|
Q_SIGNALS:
|
|
void textMessageReceived(const QByteArray &data);
|
|
private Q_SLOTS:
|
|
void onTextMessageReceived(const QString &data);
|
|
private:
|
|
NGL::INGLLogger &m_logger;
|
|
QWebSocket *m_webSocket;
|
|
int m_exerciseId;
|
|
};
|
|
|
|
}
|