TcpLogging.hxx

Go to the documentation of this file.

 
#ifndef _UTILS_TCPLOGGING_HXX_
#define _UTILS_TCPLOGGING_HXX_
 
#include "executor/StateFlow.hxx"
#include "utils/logging.h"
#include "utils/Singleton.hxx"
 
struct LogEntry
{
    string data;
};
 
using FdLoggingServerBase = StateFlow<Buffer<LogEntry>, QList<1>>;
 
class FdLoggingServer : public Singleton<FdLoggingServer>,
                        public FdLoggingServerBase
{
public:
    FdLoggingServer(Service *service)
        : FdLoggingServerBase(service)
    {
    }
 
private:
    Action entry() override
    {
        if (fd_ < 0)
        {
            return release_and_exit();
        }
        return write_repeated(&selectHelper_, fd_, input()->data.data(),
            input()->data.size(), STATE(write_done));
    }
 
    Action write_done()
    {
        if (selectHelper_.hasError_)
        {
            ::close(fd_);
            fd_ = -1;
        }
        return release_and_exit();
    }
 
    LogEntry *input()
    {
        return message()->data();
    }
 
protected:
    int fd_{-1};
 
private:
    StateFlowSelectHelper selectHelper_{this};
};
 
void log_output(char *buf, int size)
{
    if (size <= 0)
        return;
    if (!Singleton<FdLoggingServer>::exists())
        return;
    auto *b = Singleton<FdLoggingServer>::instance()->alloc();
    b->data()->data.reserve(size + 1);
    b->data()->data.assign(buf, size);
    b->data()->data.push_back('\n');
    Singleton<FdLoggingServer>::instance()->send(b);
}
 
class TcpLoggingServer : public FdLoggingServer
{
public:
    TcpLoggingServer(Service *service, int port)
        : FdLoggingServer(service)
        , listener_(port, std::bind(&TcpLoggingServer::on_new_connection, this,
                              std::placeholders::_1))
    {
    }
 
private:
    void on_new_connection(int fd)
    {
        if (fd_ >= 0)
        {
            ::close(fd_);
        }
        fd_ = fd;
    }
 
    SocketListener listener_;
};
 
class SerialLoggingServer : public FdLoggingServer
{
public:
    SerialLoggingServer(Service *service, const char *name)
        : FdLoggingServer(service)
    {
        fd_ = ::open(name, O_WRONLY);
        HASSERT(fd_ >= 0);
    }
};
 
#endif // _UTILS_TCPLOGGING_HXX_