#ifndef MS_CHANNEL_UNIX_STREAM_SOCKET_HPP
#define MS_CHANNEL_UNIX_STREAM_SOCKET_HPP
#include "common.hpp"
#include "Channel/ChannelRequest.hpp"
#include "handles/UnixStreamSocket.hpp"
#include <nlohmann/json.hpp>
namespace Channel
{
class ConsumerSocket : public ::UnixStreamSocket
{
public:
class Listener
{
public:
virtual ~Listener() = default;
public:
virtual void OnConsumerSocketMessage(ConsumerSocket* consumerSocket, char* msg, size_t msgLen) = 0;
virtual void OnConsumerSocketClosed(ConsumerSocket* consumerSocket) = 0;
};
public:
ConsumerSocket(int fd, size_t bufferSize, Listener* listener);
public:
void UserOnUnixStreamRead() override;
void UserOnUnixStreamSocketClosed() override;
private:
Listener* listener{ nullptr };
size_t msgStart{ 0u }; };
class ProducerSocket : public ::UnixStreamSocket
{
public:
ProducerSocket(int fd, size_t bufferSize);
public:
void UserOnUnixStreamRead() override
{
}
void UserOnUnixStreamSocketClosed() override
{
}
};
class ChannelSocket : public ConsumerSocket::Listener
{
public:
class Listener
{
public:
virtual ~Listener() = default;
public:
virtual void OnChannelRequest(
Channel::ChannelSocket* channel, Channel::ChannelRequest* request) = 0;
virtual void OnChannelClosed(Channel::ChannelSocket* channel) = 0;
};
public:
explicit ChannelSocket(int consumerFd, int producerFd);
virtual ~ChannelSocket();
public:
void Close();
void SetListener(Listener* listener);
void Send(json& jsonMessage);
void SendLog(char* message, size_t messageLen);
private:
void SendImpl(const void* nsPayload, size_t nsPayloadLen);
public:
void OnConsumerSocketMessage(ConsumerSocket* consumerSocket, char* msg, size_t msgLen) override;
void OnConsumerSocketClosed(ConsumerSocket* consumerSocket) override;
private:
Listener* listener{ nullptr };
bool closed{ false };
ConsumerSocket consumerSocket;
ProducerSocket producerSocket;
uint8_t* writeBuffer;
};
}
#endif