mirrors/spdlog
1
0
Fork 0
mirror of https://github.com/gabime/spdlog.git synced 2024-12-25 10:01:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #1426 from Proheeler/v1.x

Browse Source 
tcp_sink implementation for fluentbit
This commit is contained in:
Gabi Melman 2020-02-10 14:19:21 +02:00 committed by GitHub
commit 38584a1fca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -0,0 +1,51 @@
// Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
#include <spdlog/spdlog.h>
#include <spdlog/sinks/base_sink.h>
#include <spdlog/common-inl.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
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)
{
SPDLOG_THROW(spdlog::spdlog_ex("Socket creation error", errno));
}
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)
{
SPDLOG_THROW(spdlog::spdlog_ex("Invalid address/ Address not supported", errno));
}
if (connect(sock_, (struct sockaddr *)&serv_addr_, sizeof(serv_addr_)) < 0)
{
SPDLOG_THROW(spdlog::spdlog_ex("Connection Failed", errno));
}
}
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);
int res = send(sock_ , formatted.data() , formatted.size() , 0 );
if(res < 0)
SPDLOG_THROW(spdlog::spdlog_ex("Message Send Failed", errno));
}
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>;