日志系统——日志落地模块设计

一,大致框架

首先我们需要明确模块的功能,将格式化后的日志信息字符串,输出到对应的位置。同时由于用户输出信息的方式是多样的,因此我们日志落地模块也支持拓展的功能,也就是用户自定义落地方式。

日志信息落地的方式大概可以分为下面三种

1.标准输出,指的是将信息直接进行cout打印,程序相当简单,但是不方便分析

2.指定文件, 将日志信息打印到目标文件中,方便后续查看进程运行状况,但也有问题例如单个文件过大,多天的信息堆积一处不方便查询观看。

3.滚动文件,滚动文件分为两种:①按照文件大小进行切换,例如一个文件存储1G的内容,内容超出1G,重新创建文件存储。②按照时间进行切换,例如以天为单位,每一天都开一个新文件存储信息(该方式我们会以用户自定义输出方法的方法实现)。

实现思路:

1.抽象出一个日志落地模块基类

2.不同的落地方式从基类创建派生类实现

3.使用简单工厂模式,将对落地方式进行一层封装,也就是将创建和实现分开。

二,代码实现

2.1 sink.hpp

#ifndef _M_SINK_H_
#define _M_SINK_H_

#include "util.hpp"
#include "format.hpp"
#include <iostream>
#include <fstream>
#include <memory>
#include <cassert>
#include <ctime>

namespace mjwlog
{
    // 基类
    class Sink
    {
    public:
        using ptr = std::shared_ptr<Sink>;
        Sink(){};
        virtual ~Sink(){};
        virtual void log(const char *msg, const size_t len) = 0;
    };

    // 标准输出落地方向
    class StdoutSink : public Sink
    {
    public:
        void log(const char *msg, const size_t len) override
        {
            std::cout.write(msg, len);
        }
    };

    // 文件输出落地方向
    class FileSink : public Sink
    {
    public:
        // 先获得存储位置
        FileSink(const std::string& filepath)
            : _filepath(filepath)
        {
            // 在构造的时候就将文件打开
            // 1.先判断目录是否存在,不存在创建
            util::FileDirectory::createDirectory(util::FileDirectory::getDirectory(_filepath));
            _ofs.open(_filepath, std::ios_base::binary | std::ios_base::app);
            assert(_ofs.is_open()); // 打开失败,直接退出报错
        }

        // 写入内容
        void log(const char *msg, const size_t len) override
        {
            _ofs.write(msg, len);
            if (!_ofs.good())
            {
                std::cout << "日志写入文件失败" << std::endl;
            }
        }

    private:
        std::string _filepath;
        std::ofstream _ofs;
    };

    // 滚动文件输出落地方向(按照大小)
    class RollFileSink : public Sink
    {
    public:
        // 先获得存储位置
        // basepath:"/a/b/log"(自行添加后缀方便辨认) ->"/a/b/log200208051123.log-_filecount"
        RollFileSink(const std::string& basepath,const size_t& max_size)
            : _basepath(basepath),
              _max_size(max_size),
              _cur_size(0),
              _filecount(1)
        {
            // 在构造的时候就将文件打开
            //检查目录"/a/b/log"是否存在
            util::FileDirectory::createDirectory(util::FileDirectory::getDirectory(_basepath));

            //打开文件
            creatfile();
        }

        // 写入内容
        void log(const char *msg, const size_t len) override
        {
            //判断文件是否已满,满了就重新开文件
            if(_cur_size>=_max_size)
            {
                creatfile();
                //注意创建新文件后,_cur_size需要置为0
                _cur_size=0;
            }

            _ofs.write(msg,len);

            if (!_ofs.good())
            {
                std::cout << "日志写入文件失败" << std::endl;
            }
            _cur_size+=len;
        }

    private:
        //辅助接口
        //1.添加后缀
        std::string AddSuffix()
        {
            std::string suffix = _basepath;
            // basepath:"/a/b/log"(自行添加后缀方便辨认) ->"/a/b/log200208051123.log-_filecount"
            time_t time = util::gettime::nowtime();
            // localtime_r 函数获取时间详细属性
            struct tm attr;
            localtime_r(&time, &attr);

            suffix += std::to_string(attr.tm_year+1900); // 年,默认从1900开始算,因此需要+1900
            suffix += std::to_string(attr.tm_mon+1);  // 月,0-11
            suffix += std::to_string(attr.tm_mday); // 日
            suffix += std::to_string(attr.tm_hour); // 时
            suffix += std::to_string(attr.tm_min);  // 分
            suffix += ".log";
            suffix+="-"+std::to_string(_filecount++);

            return suffix;
        }

        //2.重新开文件
        void creatfile()
        {
            //打开前需要将原文件关闭
            _ofs.close();
            // 1.先判断目录是否存在,不存在创建
            std::string filepath=AddSuffix();
            _ofs.open(filepath, std::ios_base::binary | std::ios_base::app);
            assert(_ofs.is_open()); // 打开失败,直接退出报错
        }

    private:
        std::string _basepath;
        std::ofstream _ofs;
        size_t _max_size; // 文件空间上限
        size_t _cur_size; // 文件当前空间大小
        int _filecount;
    };

    //简单工厂模式,将创建和实现方法分开
    class SinkFactory
    {
    public: 
        template<typename SinkDirection,typename ...Args>
        static Sink::ptr  LogSink(Args &&...args)
        {
            return std::make_shared<SinkDirection>(std::forward<Args>(args)...);
        }

    };



}

#endif

测试

#include "util.hpp"
#include "level.hpp"
#include "format.hpp"
#include "sink.hpp"

int main()
{
    mjwlog::message msg(53,mjwlog::LogLevel::level::DEBUG,"test","root","日志格式化输出调试");
    mjwlog::Formatter ft;
    std::cout<<ft.format(msg)<<std::endl;

    mjwlog::Sink::ptr stdout_ptr=mjwlog::SinkFactory::LogSink<mjwlog::StdoutSink>();
    mjwlog::Sink::ptr file_ptr=mjwlog::SinkFactory::LogSink<mjwlog::FileSink>("./logfile/test.log");
    //以1m为分界线,进行滚动文件输出
    mjwlog::Sink::ptr rollfile_ptr=mjwlog::SinkFactory::LogSink<mjwlog::RollFileSink>("./logfile/test",1024*1024);

    stdout_ptr->log(ft.format(msg).c_str(),ft.format(msg).size());
    file_ptr->log(ft.format(msg).c_str(),ft.format(msg).size());

    size_t testsize=0;
    //我们写入10m的内容
    while(testsize<1024*1024*10)
    {
        rollfile_ptr->log(ft.format(msg).c_str(),ft.format(msg).size());
        testsize+=ft.format(msg).size();
    }




    //std::cout<<"abc\taaa"<<std::endl;
    /* std::cout<<mjwlog::LogLevel::LeveltoString(mjwlog::LogLevel::level::DEBUG)<<std::endl;
    std::cout<<mjwlog::LogLevel::LeveltoString(mjwlog::LogLevel::level::INFO)<<std::endl;
    std::cout<<mjwlog::LogLevel::LeveltoString(mjwlog::LogLevel::level::WARN)<<std::endl;
    std::cout<<mjwlog::LogLevel::LeveltoString(mjwlog::LogLevel::level::ERROR)<<std::endl;
    std::cout<<mjwlog::LogLevel::LeveltoString(mjwlog::LogLevel::level::FATAL)<<std::endl;
    std::cout<<mjwlog::LogLevel::LeveltoString(mjwlog::LogLevel::level::OFF)<<std::endl; */


    /* std::cout<<mjwlog::util::gettime::nowtime()<<std::endl;
    mjwlog::util::FileDirectory::createDirectory("./abc/abbbb/xxxx/vvvv/c.txt"); */
    return 0;
}

结果

2.2 自定义日志落地方式

 这里我们以写入滚动文件(按照时间)为例

#include "util.hpp"
#include "level.hpp"
#include "format.hpp"
#include "sink.hpp"

// 滚动文件输出落地方向(按照时间)
enum class time_seg
{
    SECOND = 0,
    MINUTE,
    HOUR,
    DAY
};
//注意:当_seg以秒为单位时,getTimeSecond(_seg)返回1的话,任何数%1都为0
//这样就导致_create_time和_cur_time永远为0,这样就永远没办法创新文件
//因此当_seg以秒为单位时,_create_time和_cur_time不%上任何值
class TimeFileSink : public mjwlog::Sink
{
public:
    // 先获得存储位置
    // basepath:"/a/b/log"(自行添加后缀方便辨认) ->"/a/b/log200208051123.log-_filecount"
    TimeFileSink(const std::string &basepath, const time_seg &seg)
        : _basepath(basepath),
          _seg(seg),
          _create_time(0),
          _cur_time(0),
          _filecount(1)
    {
        // 在构造的时候就将文件打开
        // 检查目录"/a/b/log"是否存在
        mjwlog::util::FileDirectory::createDirectory(mjwlog::util::FileDirectory::getDirectory(_basepath));
        // 打开文件
        creatfile();

        //确定创建时的时间段,注意每次重新创建文件都要重置_create_time
        
        _create_time=_seg==time_seg::SECOND?mjwlog::util::gettime::nowtime():mjwlog::util::gettime::nowtime()%getTimeSecond(_seg);

    }

    // 写入内容
    void log(const char *msg, const size_t len) override
    {
        // 判断文件是否已满,满了就重新开文件
        _cur_time=_seg==time_seg::SECOND?mjwlog::util::gettime::nowtime():mjwlog::util::gettime::nowtime()%getTimeSecond(_seg);
        
        if (_cur_time != _create_time)
        {
            creatfile();
            // 注意创建新文件后,_cur_size需要置为0
            _create_time=_seg==time_seg::SECOND?mjwlog::util::gettime::nowtime():mjwlog::util::gettime::nowtime()%getTimeSecond(_seg);

        }

        _ofs.write(msg, len);

        if (!_ofs.good())
        {
            std::cout << "日志写入文件失败" << std::endl;
        }
    }

private:
    // 辅助接口
    // 1.添加后缀
    std::string AddSuffix()
    {
        std::string suffix = _basepath;
        // basepath:"/a/b/log"(自行添加后缀方便辨认) ->"/a/b/log200208051123.log-_filecount"
        time_t time = mjwlog::util::gettime::nowtime();
        // localtime_r 函数获取时间详细属性
        struct tm attr;
        localtime_r(&time, &attr);

        suffix += std::to_string(attr.tm_year + 1900); // 年,默认从1900开始算,因此需要+1900
        suffix += std::to_string(attr.tm_mon + 1);     // 月,0-11
        suffix += std::to_string(attr.tm_mday);        // 日
        suffix += std::to_string(attr.tm_hour);        // 时
        suffix += std::to_string(attr.tm_min);         // 分
        suffix += ".log";
        suffix += "-" + std::to_string(_filecount++);

        return suffix;
    }

    // 2.重新开文件
    void creatfile()
    {
        // 打开前需要将原文件关闭
        _ofs.close();
        // 1.先判断目录是否存在,不存在创建
        std::string filepath = AddSuffix();
        _ofs.open(filepath, std::ios_base::binary | std::ios_base::app);
        assert(_ofs.is_open()); // 打开失败,直接退出报错
    }

    // 3.获取时间对应的秒数,SECOND——1
    int getTimeSecond(time_seg seg)
    {
        switch (seg)
        {
        case time_seg::SECOND:
            return 1;
        case time_seg::HOUR:
            return 60;
        case time_seg::MINUTE:
            return 2400;
        case time_seg::DAY:
            return 2400 * 24;
        default:
            std::cout << "暂时不支持当前间隔时间" << std::endl;
            abort();
        }

    }

private:
    std::string _basepath;
    std::ofstream _ofs;
    // 时间间隔,由用户确定
    time_seg _seg;
    // 例如我们以1秒为间隔分文件存储
    // 创建文件时为第31秒,32秒存入内容时检测,发现两个时间段对不上
    // 此时就需要重新创建文件写入日志
    size_t _create_time; // 文件创建时的时间段
    size_t _cur_time;    // 现在的时间段
    int _filecount;
};

int main()
{
    mjwlog::message msg(53, mjwlog::LogLevel::level::DEBUG, "test", "root", "日志格式化输出调试");
    mjwlog::Formatter ft;
    std::cout << ft.format(msg) << std::endl;

    mjwlog::Sink::ptr timefile_ptr=mjwlog::SinkFactory::LogSink<TimeFileSink>("./logfile/test",time_seg::SECOND);

    time_t old=mjwlog::util::gettime::nowtime();
    while(mjwlog::util::gettime::nowtime()<old+5)
    {
        timefile_ptr->log(ft.format(msg).c_str(), ft.format(msg).size());
        usleep(1000);
    }

    return 0;
}

 结果

 写入滚动文件(按照时间)落地方向,主要用来证明用户可以根据自己需要自定义日志落地,完成证明后,我们将其并入到sink.hpp中。

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值