tcp_sink implementation for fluentbit

This commit is contained in:
v.reshetnikov 2020-02-10 14:24:17 +03:00
parent 12f36debae
commit 346b9ae5a1

47
include/spdlog/tcp_sink.h Normal file
View File

@ -0,0 +1,47 @@
#include <spdlog/spdlog.h>
#include <spdlog/sinks/base_sink.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
using namespace std;
template<typename Mutex>
class tcp_sink : public spdlog::sinks::base_sink <Mutex>
{
public:
tcp_sink(std::string address,int port)
{
if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
printf("\n Socket creation error \n");
}
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(port);
if(inet_pton(AF_INET, address.c_str(), &serv_addr.sin_addr)<=0)
{
printf("\nInvalid address/ Address not supported \n");
}
if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
{
printf("\nConnection Failed \n");
}
}
protected:
void sink_it_(const spdlog::details::log_msg& msg) override
{
spdlog::memory_buf_t formatted;
spdlog::sinks::base_sink<Mutex>::formatter_->format(msg, formatted);
std::string message = fmt::to_string(formatted);
send(sock , message.c_str() , message.size() , 0 );
}
void flush_() override
{
}
private:
int sock;
struct sockaddr_in serv_addr;
};
using tcp_sink_mt = tcp_sink<std::mutex>;
using tcp_sink_st = tcp_sink<spdlog::details::null_mutex>;