123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118 |
- // Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
- // Distributed under the MIT License (http://opensource.org/licenses/MIT)
- #pragma once
- #include <spdlog/common.h>
- #include <spdlog/details/log_msg.h>
- #include <spdlog/details/os.h>
- #include <spdlog/formatter.h>
- #include <chrono>
- #include <ctime>
- #include <memory>
- #include <string>
- #include <unordered_map>
- #include <vector>
- namespace spdlog {
- namespace details {
- // padding information.
- struct padding_info {
- enum class pad_side { left, right, center };
- padding_info() = default;
- padding_info(size_t width, padding_info::pad_side side, bool truncate)
- : width_(width),
- side_(side),
- truncate_(truncate),
- enabled_(true) {}
- bool enabled() const { return enabled_; }
- size_t width_ = 0;
- pad_side side_ = pad_side::left;
- bool truncate_ = false;
- bool enabled_ = false;
- };
- class SPDLOG_API flag_formatter {
- public:
- explicit flag_formatter(padding_info padinfo)
- : padinfo_(padinfo) {}
- flag_formatter() = default;
- virtual ~flag_formatter() = default;
- virtual void format(const details::log_msg &msg,
- const std::tm &tm_time,
- memory_buf_t &dest) = 0;
- protected:
- padding_info padinfo_;
- };
- } // namespace details
- class SPDLOG_API custom_flag_formatter : public details::flag_formatter {
- public:
- virtual std::unique_ptr<custom_flag_formatter> clone() const = 0;
- void set_padding_info(const details::padding_info &padding) {
- flag_formatter::padinfo_ = padding;
- }
- };
- class SPDLOG_API pattern_formatter final : public formatter {
- public:
- using custom_flags = std::unordered_map<char, std::unique_ptr<custom_flag_formatter>>;
- explicit pattern_formatter(std::string pattern,
- pattern_time_type time_type = pattern_time_type::local,
- std::string eol = spdlog::details::os::default_eol,
- custom_flags custom_user_flags = custom_flags());
- // use default pattern is not given
- explicit pattern_formatter(pattern_time_type time_type = pattern_time_type::local,
- std::string eol = spdlog::details::os::default_eol);
- pattern_formatter(const pattern_formatter &other) = delete;
- pattern_formatter &operator=(const pattern_formatter &other) = delete;
- std::unique_ptr<formatter> clone() const override;
- void format(const details::log_msg &msg, memory_buf_t &dest) override;
- template <typename T, typename... Args>
- pattern_formatter &add_flag(char flag, Args &&...args) {
- custom_handlers_[flag] = details::make_unique<T>(std::forward<Args>(args)...);
- return *this;
- }
- void set_pattern(std::string pattern);
- void need_localtime(bool need = true);
- private:
- std::string pattern_;
- std::string eol_;
- pattern_time_type pattern_time_type_;
- bool need_localtime_;
- std::tm cached_tm_;
- std::chrono::seconds last_log_secs_;
- std::vector<std::unique_ptr<details::flag_formatter>> formatters_;
- custom_flags custom_handlers_;
- std::tm get_time_(const details::log_msg &msg);
- template <typename Padder>
- void handle_flag_(char flag, details::padding_info padding);
- // Extract given pad spec (e.g. %8X)
- // Advance the given it pass the end of the padding spec found (if any)
- // Return padding.
- static details::padding_info handle_padspec_(std::string::const_iterator &it,
- std::string::const_iterator end);
- void compile_pattern_(const std::string &pattern);
- };
- } // namespace spdlog
- #ifdef SPDLOG_HEADER_ONLY
- #include "pattern_formatter-inl.h"
- #endif
|