diff --git a/include/spdlog/details/spdlog_impl.h b/include/spdlog/details/spdlog_impl.h index 7fe9ab40..1f880bbe 100644 --- a/include/spdlog/details/spdlog_impl.h +++ b/include/spdlog/details/spdlog_impl.h @@ -59,25 +59,25 @@ inline std::shared_ptr spdlog::basic_logger_st(const std::string } // Create multi/single threaded rotating file logger -inline std::shared_ptr spdlog::rotating_logger_mt(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files) +inline std::shared_ptr spdlog::rotating_logger_mt(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files, std::function cb /*= nullptr*/) { - return create(logger_name, filename, max_file_size, max_files); + return create(logger_name, filename, max_file_size, max_files, cb); } -inline std::shared_ptr spdlog::rotating_logger_st(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files) +inline std::shared_ptr spdlog::rotating_logger_st(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files, std::function cb /*= nullptr*/) { - return create(logger_name, filename, max_file_size, max_files); + return create(logger_name, filename, max_file_size, max_files, cb); } // Create file logger which creates new file at midnight): -inline std::shared_ptr spdlog::daily_logger_mt(const std::string& logger_name, const filename_t& filename, int hour, int minute) +inline std::shared_ptr spdlog::daily_logger_mt(const std::string& logger_name, const filename_t& filename, int hour, int minute, std::function cb /*= nullptr*/) { - return create(logger_name, filename, hour, minute); + return create(logger_name, filename, hour, minute, cb); } -inline std::shared_ptr spdlog::daily_logger_st(const std::string& logger_name, const filename_t& filename, int hour, int minute) +inline std::shared_ptr spdlog::daily_logger_st(const std::string& logger_name, const filename_t& filename, int hour, int minute, std::function cb /*= nullptr*/) { - return create(logger_name, filename, hour, minute); + return create(logger_name, filename, hour, minute, cb); } diff --git a/include/spdlog/sinks/file_sinks.h b/include/spdlog/sinks/file_sinks.h index 421acc8a..acbf9266 100644 --- a/include/spdlog/sinks/file_sinks.h +++ b/include/spdlog/sinks/file_sinks.h @@ -66,12 +66,13 @@ class rotating_file_sink SPDLOG_FINAL : public base_sink < Mutex > { public: rotating_file_sink(const filename_t &base_filename, - std::size_t max_size, std::size_t max_files) : + std::size_t max_size, std::size_t max_files, std::function cb) : _base_filename(base_filename), _max_size(max_size), _max_files(max_files), _current_size(0), - _file_helper() + _file_helper(), + _on_finish_cb(cb) { _file_helper.open(calc_filename(_base_filename, 0)); _current_size = _file_helper.size(); //expensive. called only once @@ -134,12 +135,18 @@ private: } } _file_helper.reopen(true); + + if (_on_finish_cb) + { + _on_finish_cb(); + } } filename_t _base_filename; std::size_t _max_size; std::size_t _max_files; std::size_t _current_size; details::file_helper _file_helper; + std::function _on_finish_cb; }; typedef rotating_file_sink rotating_file_sink_mt; @@ -186,9 +193,11 @@ public: daily_file_sink( const filename_t& base_filename, int rotation_hour, - int rotation_minute) : _base_filename(base_filename), + int rotation_minute, + std::function cb) : _base_filename(base_filename), _rotation_h(rotation_hour), - _rotation_m(rotation_minute) + _rotation_m(rotation_minute), + _rotate_cb(cb) { if (rotation_hour < 0 || rotation_hour > 23 || rotation_minute < 0 || rotation_minute > 59) throw spdlog_ex("daily_file_sink: Invalid rotation time in ctor"); @@ -204,6 +213,10 @@ protected: { _file_helper.open(FileNameCalc::calc_filename(_base_filename)); _rotation_tp = _next_rotation_tp(); + if (_rotate_cb) + { + _rotate_cb(); + } } _file_helper.write(msg); } @@ -234,6 +247,7 @@ private: int _rotation_m; std::chrono::system_clock::time_point _rotation_tp; details::file_helper _file_helper; + std::function _rotate_cb; }; typedef daily_file_sink daily_file_sink_mt; diff --git a/include/spdlog/spdlog.h b/include/spdlog/spdlog.h index 5da20621..dada9398 100644 --- a/include/spdlog/spdlog.h +++ b/include/spdlog/spdlog.h @@ -77,14 +77,14 @@ std::shared_ptr basic_logger_st(const std::string& logger_name, const fi // // Create and register multi/single threaded rotating file logger // -std::shared_ptr rotating_logger_mt(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files); -std::shared_ptr rotating_logger_st(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files); +std::shared_ptr rotating_logger_mt(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files, std::function cb = nullptr); +std::shared_ptr rotating_logger_st(const std::string& logger_name, const filename_t& filename, size_t max_file_size, size_t max_files, std::function cb = nullptr); // // Create file logger which creates new file on the given time (default in midnight): // -std::shared_ptr daily_logger_mt(const std::string& logger_name, const filename_t& filename, int hour=0, int minute=0); -std::shared_ptr daily_logger_st(const std::string& logger_name, const filename_t& filename, int hour=0, int minute=0); +std::shared_ptr daily_logger_mt(const std::string& logger_name, const filename_t& filename, int hour=0, int minute=0, std::function cb = nullptr); +std::shared_ptr daily_logger_st(const std::string& logger_name, const filename_t& filename, int hour=0, int minute=0, std::function cb = nullptr); // // Create and register stdout/stderr loggers diff --git a/tests/file_log.cpp b/tests/file_log.cpp index 45f6e8c1..dcf3371d 100644 --- a/tests/file_log.cpp +++ b/tests/file_log.cpp @@ -108,7 +108,7 @@ TEST_CASE("daily_logger with dateonly calculator", "[daily_logger_dateonly]]") fmt::MemoryWriter w; w.write("{}_{:04d}-{:02d}-{:02d}", basename, tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday); - auto logger = spdlog::create("logger", basename, 0, 0); + auto logger = spdlog::create("logger", basename, 0, 0, nullptr); for (int i = 0; i < 10; ++i) logger->info("Test message {}", i); logger->flush(); @@ -140,7 +140,7 @@ TEST_CASE("daily_logger with custom calculator", "[daily_logger_custom]]") fmt::MemoryWriter w; w.write("{}{:04d}{:02d}{:02d}", basename, tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday); - auto logger = spdlog::create("logger", basename, 0, 0); + auto logger = spdlog::create("logger", basename, 0, 0, nullptr); for (int i = 0; i < 10; ++i) logger->info("Test message {}", i);