123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475 |
- #pragma once
- #include <spdlog/common.h>
- #include <spdlog/details/null_mutex.h>
- #include <spdlog/sinks/base_sink.h>
- #ifdef _WIN32
- #include <spdlog/details/tcp_client-windows.h>
- #else
- #include <spdlog/details/tcp_client.h>
- #endif
- #include <chrono>
- #include <functional>
- #include <mutex>
- #include <string>
- #pragma once
- namespace spdlog {
- namespace sinks {
- struct tcp_sink_config {
- std::string server_host;
- int server_port;
- bool lazy_connect = false;
- tcp_sink_config(std::string host, int port)
- : server_host{std::move(host)},
- server_port{port} {}
- };
- template <typename Mutex>
- class tcp_sink : public spdlog::sinks::base_sink<Mutex> {
- public:
-
-
- explicit tcp_sink(tcp_sink_config sink_config)
- : config_{std::move(sink_config)} {
- if (!config_.lazy_connect) {
- this->client_.connect(config_.server_host, config_.server_port);
- }
- }
- ~tcp_sink() override = default;
- 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);
- if (!client_.is_connected()) {
- client_.connect(config_.server_host, config_.server_port);
- }
- client_.send(formatted.data(), formatted.size());
- }
- void flush_() override {}
- tcp_sink_config config_;
- details::tcp_client client_;
- };
- using tcp_sink_mt = tcp_sink<std::mutex>;
- using tcp_sink_st = tcp_sink<spdlog::details::null_mutex>;
- }
- }
|