#ifndef MS_PAYLOAD_CHANNEL_UNIX_STREAM_SOCKET_HPP
#define MS_PAYLOAD_CHANNEL_UNIX_STREAM_SOCKET_HPP
#include "common.hpp"
#include "PayloadChannel/Notification.hpp"
#include "PayloadChannel/PayloadChannelRequest.hpp"
#include "handles/UnixStreamSocket.hpp"
#include <nlohmann/json.hpp>
namespace PayloadChannel
{
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);
~ConsumerSocket();
public:
void UserOnUnixStreamRead() override;
void UserOnUnixStreamSocketClosed() override;
private:
Listener* listener{ nullptr };
};
class ProducerSocket : public ::UnixStreamSocket
{
public:
ProducerSocket(int fd, size_t bufferSize);
public:
void UserOnUnixStreamRead() override
{
}
void UserOnUnixStreamSocketClosed() override
{
}
};
class PayloadChannelSocket : public ConsumerSocket::Listener
{
public:
class Listener
{
public:
virtual ~Listener() = default;
public:
virtual void OnPayloadChannelNotification(
PayloadChannel::PayloadChannelSocket* payloadChannel,
PayloadChannel::Notification* notification) = 0;
virtual void OnPayloadChannelRequest(
PayloadChannel::PayloadChannelSocket* payloadChannel,
PayloadChannel::PayloadChannelRequest* request) = 0;
virtual void OnPayloadChannelClosed(PayloadChannel::PayloadChannelSocket* payloadChannel) = 0;
};
public:
explicit PayloadChannelSocket(int consumerFd, int producerFd);
explicit PayloadChannelSocket(
PayloadChannelReadFn payloadChannelReadFn,
PayloadChannelReadCtx payloadChannelReadCtx,
PayloadChannelWriteFn payloadChannelWriteFn,
PayloadChannelWriteCtx payloadChannelWriteCtx);
virtual ~PayloadChannelSocket();
public:
void Close();
void SetListener(Listener* listener);
bool CallbackRead();
void Send(json& jsonMessage, const uint8_t* payload, size_t payloadLen);
void Send(json& jsonMessage);
private:
void SendImpl(const uint8_t* message, uint32_t messageLen);
void SendImpl(
const uint8_t* message, uint32_t messageLen, const uint8_t* payload, uint32_t payloadLen);
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{ nullptr };
ProducerSocket* producerSocket{ nullptr };
PayloadChannelReadFn payloadChannelReadFn{ nullptr };
PayloadChannelReadCtx payloadChannelReadCtx{ nullptr };
PayloadChannelWriteFn payloadChannelWriteFn{ nullptr };
PayloadChannelWriteCtx payloadChannelWriteCtx{ nullptr };
PayloadChannel::Notification* ongoingNotification{ nullptr };
PayloadChannel::PayloadChannelRequest* ongoingRequest{ nullptr };
uv_async_t* uvReadHandle{ nullptr };
uint8_t* writeBuffer{ nullptr };
};
}
#endif