138 lines
4.1 KiB
C++
138 lines
4.1 KiB
C++
#ifndef TCPSERVERPROTOCOL_H
|
||
#define TCPSERVERPROTOCOL_H
|
||
|
||
#include <atomic>
|
||
#include <functional>
|
||
#include <map>
|
||
#include <mutex>
|
||
#include <set>
|
||
#include <vector>
|
||
#include <string>
|
||
#include <memory>
|
||
#include <QByteArray>
|
||
#include "IYTCPServer.h"
|
||
#include "ProtocolCommon.h"
|
||
|
||
/**
|
||
* @brief TCP服务器协议封装类
|
||
* 根据协议文档实现定子工件定位 TCP/IP 通信协议
|
||
*
|
||
* 协议特点：
|
||
* - 服务端模式，等待客户端连接
|
||
* - 请求和结果均使用简单文本帧
|
||
* - 字段使用 '_' 分隔，帧尾使用 "\r\n"
|
||
*/
|
||
class TCPServerProtocol
|
||
{
|
||
public:
|
||
/**
|
||
* @brief 连接状态枚举（使用公共状态定义）
|
||
*/
|
||
using TCPStatus = ConnectionStatus;
|
||
|
||
/**
|
||
* @brief 检测结果回调函数类型
|
||
* @param success 是否成功触发检测
|
||
* @param cameraIndex 相机索引（从开始检测指令中解析，-1表示所有相机）
|
||
* @param timestamp 请求时间戳
|
||
*/
|
||
using DetectionTriggerCallback = std::function<bool(bool success, int cameraIndex, qint64 timestamp)>;
|
||
using BinDetectionTriggerCallback = std::function<bool(int cameraIndex, qint64 timestamp)>;
|
||
|
||
public:
|
||
TCPServerProtocol();
|
||
~TCPServerProtocol();
|
||
|
||
/**
|
||
* @brief 初始化TCP服务器
|
||
* @param port TCP/IP 协议端口号，默认7800
|
||
* @return 0-成功，其他-错误码
|
||
*/
|
||
int Initialize(uint16_t port = 7800);
|
||
|
||
/**
|
||
* @brief 反初始化，停止服务
|
||
*/
|
||
void Deinitialize();
|
||
|
||
/**
|
||
* @brief 发送简单文本帧给客户端，函数会自动补 "\r\n"
|
||
*/
|
||
int SendTextFrame(const QByteArray& frame, const TCPClient* pClient = nullptr);
|
||
|
||
/**
|
||
* @brief 获取当前连接状态
|
||
* @return 当前状态
|
||
*/
|
||
TCPStatus GetConnectionStatus() const;
|
||
|
||
/**
|
||
* @brief 设置连接状态回调
|
||
* @param callback 回调函数
|
||
*/
|
||
void SetConnectionCallback(const ConnectionCallback& callback);
|
||
|
||
/**
|
||
* @brief 设置检测触发回调
|
||
* @param callback 回调函数
|
||
*/
|
||
void SetDetectionTriggerCallback(const DetectionTriggerCallback& callback);
|
||
void SetBinDetectionTriggerCallback(const BinDetectionTriggerCallback& callback);
|
||
|
||
/**
|
||
* @brief 获取服务运行状态
|
||
* @return true-运行中，false-已停止
|
||
*/
|
||
bool IsRunning() const;
|
||
|
||
private:
|
||
/**
|
||
* @brief TCP客户端连接事件处理
|
||
* @param pClient 客户端对象
|
||
* @param eventType 事件类型
|
||
*/
|
||
void OnTCPEvent(const TCPClient* pClient, TCPServerEventType eventType);
|
||
|
||
/**
|
||
* @brief TCP数据接收处理
|
||
* @param pClient 客户端对象
|
||
* @param pData 数据指针
|
||
* @param nLen 数据长度
|
||
*/
|
||
void OnTCPDataReceived(const TCPClient* pClient, const char* pData, unsigned int nLen);
|
||
|
||
void ParseTextCommand(const TCPClient* pClient, const QByteArray& frame);
|
||
|
||
/**
|
||
* @brief 发送错误响应
|
||
* @param pClient 客户端对象
|
||
* @param code 错误码
|
||
* @param message 错误消息
|
||
* @param timestamp 时间戳
|
||
*/
|
||
void SendErrorResponse(const TCPClient* pClient, int code, const QString& message, qint64 timestamp);
|
||
|
||
private:
|
||
// TCP服务器相关
|
||
IYTCPServer* m_pTCPServer; // TCP服务器实例
|
||
std::atomic<bool> m_bServerRunning; // 服务器运行状态
|
||
uint16_t m_nPort; // TCP端口
|
||
|
||
// 连接状态
|
||
std::atomic<TCPStatus> m_connectionStatus; // TCP连接状态
|
||
|
||
// 回调函数
|
||
ConnectionCallback m_connectionCallback; // 连接状态回调
|
||
DetectionTriggerCallback m_detectionTriggerCallback; // 检测触发回调
|
||
BinDetectionTriggerCallback m_binDetectionTriggerCallback;
|
||
|
||
// TCP receive callbacks may run concurrently for different clients.
|
||
std::mutex m_sendMutex;
|
||
std::mutex m_connectionEventMutex;
|
||
std::mutex m_clientBuffersMutex;
|
||
std::set<const TCPClient*> m_connectedClients;
|
||
std::map<const TCPClient*, QByteArray> m_clientBuffers;
|
||
};
|
||
|
||
#endif // TCPSERVERPROTOCOL_H
|