TCPServer.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <functional>
#include <string>
#include <vector>
 
#include "Network/Codec/MessageFrame.h"
 
namespace Ember {
namespace Monitor {

class TCPServer {
  public:
    using MessageCallback = std::function<void(const Network::MessageFrame &)>;
    using ConnectCallback = std::function<void(const std::string &address)>;
    using DisconnectCallback = std::function<void()>;
 
    TCPServer();
    ~TCPServer();

    bool Start(uint16_t port);
    void Stop();
    bool IsRunning() const { return m_serverSocket >= 0; }
    bool IsClientConnected() const { return m_clientSocket >= 0; }

    std::string GetClientAddress() const { return m_clientAddress; }
    std::string GetClientName() const { return m_clientName; }
    void SetClientName(const std::string &name) { m_clientName = name; }
    uint16_t GetPort() const { return m_port; }

    void SetMessageCallback(MessageCallback cb) { m_messageCallback = std::move(cb); }
    void SetConnectCallback(ConnectCallback cb) { m_connectCallback = std::move(cb); }
    void SetDisconnectCallback(DisconnectCallback cb) { m_disconnectCallback = std::move(cb); }

    void Poll();

    bool Send(const Network::MessageFrame &frame);

    void DisconnectClient();
 
  private:
    void AcceptConnection();
    void ReadFromClient();
    bool SetNonBlocking(int socket);
 
    int m_serverSocket = -1;
    int m_clientSocket = -1;
    std::string m_clientAddress;
    std::string m_clientName;
    uint16_t m_port = 0;
 
    std::vector<uint8_t> m_recvBuffer;
 
    MessageCallback m_messageCallback;
    ConnectCallback m_connectCallback;
    DisconnectCallback m_disconnectCallback;
};
 
} // namespace Monitor
} // namespace Ember