mirror of
https://github.com/gabime/spdlog.git
synced 2024-12-25 01:51:38 +08:00
async_sink use unique_ptr instead of shared
This commit is contained in:
parent
3a1ac8b4d6
commit
28e1c65963
@ -18,22 +18,24 @@ using namespace utils;
|
|||||||
int main(int argc, char* argv[])
|
int main(int argc, char* argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
const unsigned int howmany = argc <= 1 ? 1000:atoi(argv[1]);
|
const unsigned int howmany = argc <= 1 ? 1000000:atoi(argv[1]);
|
||||||
|
|
||||||
logger cout_logger ("", sinks::stdout_sink());
|
logger cout_logger ("", sinks::stdout_sink());
|
||||||
cout_logger.info() << "Hello " << "man";
|
cout_logger.info() << "Hello " << "man";
|
||||||
|
|
||||||
//auto fsink = std::make_shared<sinks::rotating_file_sink>("log", "txt", 1024*1024*50 , 5, 0);
|
auto fsink = std::make_shared<sinks::rotating_file_sink>("log", "txt", 1024*1024*50 , 5, 0);
|
||||||
|
auto fsink2 = std::make_shared<sinks::rotating_file_sink>("lllog", "txt", 1024*1024*50 , 5, 0);
|
||||||
|
|
||||||
auto as = std::make_shared<sinks::async_sink>(1000);
|
auto as = std::make_shared<sinks::async_sink>(1000);
|
||||||
as->add_sink(sinks::null_sink::get());
|
as->add_sink(sinks::null_sink::get());
|
||||||
|
|
||||||
logger my_logger ("my_logger", as);
|
logger my_logger ("my_logger", as);
|
||||||
|
|
||||||
auto start = system_clock::now();
|
auto start = system_clock::now();
|
||||||
for(unsigned int i = 0; i < howmany ; i++)
|
for(unsigned int i = 1; i <= howmany ; ++i)
|
||||||
my_logger.info() << "Hello logger";
|
my_logger.info() << "Hello logger: " << i;
|
||||||
|
|
||||||
as->shutdown(milliseconds(5000));
|
as->shutdown(milliseconds(500));
|
||||||
auto delta = system_clock::now() - start;
|
auto delta = system_clock::now() - start;
|
||||||
auto delta_d = duration_cast<duration<double>> (delta).count();
|
auto delta_d = duration_cast<duration<double>> (delta).count();
|
||||||
|
|
||||||
|
@ -19,8 +19,9 @@ template<typename T>
|
|||||||
class blocking_queue
|
class blocking_queue
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
using queue_t = std::queue<T>;
|
using queue_type = std::queue<T>;
|
||||||
using size_type = typename queue_t::size_type;
|
using item_type = T;
|
||||||
|
using size_type = typename queue_type::size_type;
|
||||||
using clock = std::chrono::system_clock;
|
using clock = std::chrono::system_clock;
|
||||||
|
|
||||||
explicit blocking_queue(size_type max_size) :
|
explicit blocking_queue(size_type max_size) :
|
||||||
@ -108,7 +109,7 @@ public:
|
|||||||
{
|
{
|
||||||
{
|
{
|
||||||
std::unique_lock<std::mutex> ul(_mutex);
|
std::unique_lock<std::mutex> ul(_mutex);
|
||||||
queue_t().swap(_q);
|
queue_type().swap(_q);
|
||||||
}
|
}
|
||||||
_item_popped_cond.notify_all();
|
_item_popped_cond.notify_all();
|
||||||
}
|
}
|
||||||
|
@ -3,27 +3,35 @@
|
|||||||
#include <thread>
|
#include <thread>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
#include <functional>
|
||||||
|
|
||||||
#include "base_sink.h"
|
#include "base_sink.h"
|
||||||
#include "../logger.h"
|
#include "../logger.h"
|
||||||
#include "../details/blocking_queue.h"
|
#include "../details/blocking_queue.h"
|
||||||
#include "../details/log_msg.h"
|
#include "../details/log_msg.h"
|
||||||
|
|
||||||
|
#include<iostream>
|
||||||
|
|
||||||
using namespace std;
|
|
||||||
namespace c11log
|
namespace c11log
|
||||||
{
|
{
|
||||||
namespace sinks
|
namespace sinks
|
||||||
{
|
{
|
||||||
|
|
||||||
|
|
||||||
|
static void msg_deleter(details::log_msg* msg_to_delete)
|
||||||
|
{
|
||||||
|
delete []msg_to_delete->msg_buf.first;
|
||||||
|
delete msg_to_delete;
|
||||||
|
}
|
||||||
|
|
||||||
class async_sink : public base_sink
|
class async_sink : public base_sink
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
|
||||||
using queue_t = c11log::details::blocking_queue<std::shared_ptr<details::log_msg>>;
|
|
||||||
using size_type = queue_t::size_type;
|
|
||||||
|
|
||||||
explicit async_sink(const size_type max_queue_size);
|
using queue_type = c11log::details::blocking_queue<std::unique_ptr<details::log_msg, std::function<void(details::log_msg*)>>>;
|
||||||
|
|
||||||
|
explicit async_sink(const queue_type::size_type max_queue_size);
|
||||||
|
|
||||||
//Stop logging and join the back thread
|
//Stop logging and join the back thread
|
||||||
// TODO: limit with timeout of the join and kill it afterwards?
|
// TODO: limit with timeout of the join and kill it afterwards?
|
||||||
@ -42,7 +50,7 @@ protected:
|
|||||||
private:
|
private:
|
||||||
c11log::logger::sinks_vector_t _sinks;
|
c11log::logger::sinks_vector_t _sinks;
|
||||||
std::atomic<bool> _active;
|
std::atomic<bool> _active;
|
||||||
queue_t _q;
|
queue_type _q;
|
||||||
std::thread _back_thread;
|
std::thread _back_thread;
|
||||||
//Clear all remaining messages(if any), stop the _back_thread and join it
|
//Clear all remaining messages(if any), stop the _back_thread and join it
|
||||||
void _shutdown();
|
void _shutdown();
|
||||||
@ -55,7 +63,7 @@ private:
|
|||||||
// async_sink class implementation
|
// async_sink class implementation
|
||||||
///////////////////////////////////////////////////////////////////////////////
|
///////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
inline c11log::sinks::async_sink::async_sink(const size_type max_queue_size)
|
inline c11log::sinks::async_sink::async_sink(const queue_type::size_type max_queue_size)
|
||||||
:_sinks(),
|
:_sinks(),
|
||||||
_active(true),
|
_active(true),
|
||||||
_q(max_queue_size),
|
_q(max_queue_size),
|
||||||
@ -66,6 +74,8 @@ inline c11log::sinks::async_sink::~async_sink()
|
|||||||
{
|
{
|
||||||
_shutdown();
|
_shutdown();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
inline void c11log::sinks::async_sink::_sink_it(const details::log_msg& msg)
|
inline void c11log::sinks::async_sink::_sink_it(const details::log_msg& msg)
|
||||||
{
|
{
|
||||||
auto msg_size = msg.msg_buf.second;
|
auto msg_size = msg.msg_buf.second;
|
||||||
@ -78,13 +88,15 @@ inline void c11log::sinks::async_sink::_sink_it(const details::log_msg& msg)
|
|||||||
char *buf = new char[msg_size];
|
char *buf = new char[msg_size];
|
||||||
std::memcpy(buf, msg.msg_buf.first, msg_size);
|
std::memcpy(buf, msg.msg_buf.first, msg_size);
|
||||||
new_msg->msg_buf = bufpair_t(buf, msg_size);
|
new_msg->msg_buf = bufpair_t(buf, msg_size);
|
||||||
|
/*
|
||||||
auto new_shared_msg = std::shared_ptr<details::log_msg>(new_msg, [](details::log_msg* msg_to_delete)
|
auto new_shared_msg = queue_type::item_type(new_msg, [](const details::log_msg* msg_to_delete)
|
||||||
{
|
{
|
||||||
delete []msg_to_delete->msg_buf.first;
|
delete []msg_to_delete->msg_buf.first;
|
||||||
delete msg_to_delete;
|
delete msg_to_delete;
|
||||||
});
|
});
|
||||||
_q.push(new_shared_msg);
|
* */
|
||||||
|
queue_type::item_type new_shared_msg(new_msg, msg_deleter);
|
||||||
|
_q.push(std::move(new_shared_msg));
|
||||||
}
|
}
|
||||||
|
|
||||||
inline void c11log::sinks::async_sink::_thread_loop()
|
inline void c11log::sinks::async_sink::_thread_loop()
|
||||||
@ -92,12 +104,14 @@ inline void c11log::sinks::async_sink::_thread_loop()
|
|||||||
static std::chrono::seconds pop_timeout { 1 };
|
static std::chrono::seconds pop_timeout { 1 };
|
||||||
while (_active)
|
while (_active)
|
||||||
{
|
{
|
||||||
std::shared_ptr<details::log_msg> msg;
|
queue_type::item_type msg;
|
||||||
if (_q.pop(msg, pop_timeout))
|
if (_q.pop(msg, pop_timeout))
|
||||||
{
|
{
|
||||||
for (auto &sink : _sinks)
|
for (auto &sink : _sinks)
|
||||||
{
|
{
|
||||||
sink->log(*msg);
|
sink->log(*msg);
|
||||||
|
if(!_active)
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user