mirror of https://github.com/gabime/spdlog.git
parent
654dbc5c32
commit
3cab260814
@ -0,0 +1,61 @@
|
|||||||
|
// Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
|
||||||
|
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <spdlog/details/null_mutex.h>
|
||||||
|
#include <spdlog/sinks/base_sink.h>
|
||||||
|
#include <spdlog/details/synchronous_factory.h>
|
||||||
|
|
||||||
|
#include <mutex>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
namespace spdlog {
|
||||||
|
|
||||||
|
// callbacks type
|
||||||
|
typedef std::function<void(const details::log_msg &msg)> custom_log_callback;
|
||||||
|
|
||||||
|
namespace sinks {
|
||||||
|
/*
|
||||||
|
* Trivial callback sink, gets a callback function and calls it on each log
|
||||||
|
*/
|
||||||
|
template<typename Mutex>
|
||||||
|
class callback_sink final : public base_sink<Mutex>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
explicit callback_sink(const custom_log_callback &callback)
|
||||||
|
: callback_{callback}
|
||||||
|
{}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
void sink_it_(const details::log_msg &msg) override
|
||||||
|
{
|
||||||
|
callback_(msg);
|
||||||
|
}
|
||||||
|
void flush_() override{};
|
||||||
|
|
||||||
|
private:
|
||||||
|
custom_log_callback callback_;
|
||||||
|
};
|
||||||
|
|
||||||
|
using callback_sink_mt = callback_sink<std::mutex>;
|
||||||
|
using callback_sink_st = callback_sink<details::null_mutex>;
|
||||||
|
|
||||||
|
} // namespace sinks
|
||||||
|
|
||||||
|
//
|
||||||
|
// factory functions
|
||||||
|
//
|
||||||
|
template<typename Factory = spdlog::synchronous_factory>
|
||||||
|
inline std::shared_ptr<logger> callback_logger_mt(const std::string &logger_name, const custom_log_callback &callback)
|
||||||
|
{
|
||||||
|
return Factory::template create<sinks::callback_sink_mt>(logger_name, callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename Factory = spdlog::synchronous_factory>
|
||||||
|
inline std::shared_ptr<logger> callback_logger_st(const std::string &logger_name, const custom_log_callback &callback)
|
||||||
|
{
|
||||||
|
return Factory::template create<sinks::callback_sink_st>(logger_name, callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace spdlog
|
@ -0,0 +1,34 @@
|
|||||||
|
/*
|
||||||
|
* This content is released under the MIT License as specified in https://raw.githubusercontent.com/gabime/spdlog/master/LICENSE
|
||||||
|
*/
|
||||||
|
#include "includes.h"
|
||||||
|
#include "test_sink.h"
|
||||||
|
#include "spdlog/sinks/callback_sink.h"
|
||||||
|
#include "spdlog/async.h"
|
||||||
|
#include "spdlog/common.h"
|
||||||
|
|
||||||
|
TEST_CASE("custom_callback_logger", "[custom_callback_logger]]")
|
||||||
|
{
|
||||||
|
std::vector<std::string> lines;
|
||||||
|
spdlog::pattern_formatter formatter;
|
||||||
|
auto callback_logger = std::make_shared<spdlog::sinks::callback_sink_st>([&](const spdlog::details::log_msg &msg) {
|
||||||
|
spdlog::memory_buf_t formatted;
|
||||||
|
formatter.format(msg, formatted);
|
||||||
|
auto eol_len = strlen(spdlog::details::os::default_eol);
|
||||||
|
lines.emplace_back(formatted.begin(), formatted.end() - eol_len);
|
||||||
|
});
|
||||||
|
std::shared_ptr<spdlog::sinks::test_sink_st> test_sink(new spdlog::sinks::test_sink_st);
|
||||||
|
|
||||||
|
spdlog::logger logger("test-callback", {callback_logger, test_sink});
|
||||||
|
|
||||||
|
logger.info("test message 1");
|
||||||
|
logger.info("test message 2");
|
||||||
|
logger.info("test message 3");
|
||||||
|
|
||||||
|
std::vector<std::string> ref_lines = test_sink->lines();
|
||||||
|
|
||||||
|
REQUIRE(lines[0] == ref_lines[0]);
|
||||||
|
REQUIRE(lines[1] == ref_lines[1]);
|
||||||
|
REQUIRE(lines[2] == ref_lines[2]);
|
||||||
|
spdlog::drop_all();
|
||||||
|
}
|
Loading…
Reference in New Issue