devbox
weixin_40725706
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

【Linux】写一个日志类

作者：weixin_40725706 | 2024-06-02 08:25:13

赞

踩

【Linux】写一个日志类

文章目录

在这里插入图片描述

1. 源代码

下面代码定义了一个 Log 类，用于记录日志信息。这个类支持将日志信息输出到屏幕、单个文件或者按日志级别分类的多个文件。

主要功能：

  • 日志级别转换：将日志级别从整数转换为字符串。
  • 日志输出方式：支持屏幕输出、单个文件输出和分类文件输出。
  • 日志格式化：支持格式化日志信息，包括时间戳和自定义日志内容。
  • 文件操作：使用系统调用进行文件操作，确保日志写入文件中。
#pragma once

#include <iostream>
#include <time.h>
#include <stdarg.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>

#define SIZE 1024

#define Info 0
#define Debug 1
#define Warning 2
#define Error 3
#define Fatal 4

#define Screen 1
#define Onefile 2
#define Classfile 3

#define LogFile "log.txt"

class Log
{
public:
    Log()
    {
        printMethod = Screen;
        path = "./log/";
    }
    void Enable(int method)
    {
        printMethod = method;
    }
    std::string levelToString(int level)
    {
        switch (level)
        {
        case Info:
            return "Info";
        case Debug:
            return "Debug";
        case Warning:
            return "Warning";
        case Error:
            return "Error";
        case Fatal:
            return "Fatal";
        default:
            return "None";
        }
    }
    void printLog(int level, const std::string &logtxt)
    {
        switch (printMethod)
        {
        case Screen:
            std::cout << logtxt << std::endl;
            break;
        case Onefile:
            printOneFile(LogFile, logtxt);
            break;
        case Classfile:
            printClassFile(level, logtxt);
            break;
        default:
            break;
        }
    }
    void printOneFile(const std::string &logname, const std::string &logtxt)
    {
        // ./log/ + log.txt
        std::string _logname = path + logname;
        int fd = open(_logname.c_str(), O_WRONLY | O_CREAT | O_APPEND, 0666);
        if (fd < 0)
            return;
        write(fd, logtxt.c_str(), logtxt.size());
        close(fd);
    }
    void printClassFile(int level, const std::string &logtxt)
    {
        std::string filename = LogFile;
        filename += ".";
        // log.txt.Debug/Warning/Fatal
        filename += levelToString(level);
        printOneFile(filename, logtxt);
    }
    ~Log()
    {
    }
    void operator()(int level, const char *format, ...)
    {
        time_t t = time(nullptr);
        struct tm *ctime = localtime(&t);
        char leftbuffer[SIZE];
        snprintf(leftbuffer, sizeof(leftbuffer), "[%s][%d-%d-%d %d:%d:%d]", levelToString(level).c_str(),
                 ctime->tm_year + 1900, ctime->tm_mon + 1, ctime->tm_mday,
                 ctime->tm_hour, ctime->tm_min, ctime->tm_sec);

        va_list s;
        va_start(s, format);
        char rightbuffer[SIZE];
        vsnprintf(rightbuffer, sizeof(rightbuffer), format, s);
        va_end(s);

        // 格式: 默认部分 + 自定义部分
        char logtxt[SIZE * 2];
        snprintf(logtxt, sizeof(logtxt), "%s %s", leftbuffer, rightbuffer);

        printLog(level, logtxt);
    }

private:
    int printMethod;
    std::string path;
};

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
  • 51
  • 52
  • 53
  • 54
  • 55
  • 56
  • 57
  • 58
  • 59
  • 60
  • 61
  • 62
  • 63
  • 64
  • 65
  • 66
  • 67
  • 68
  • 69
  • 70
  • 71
  • 72
  • 73
  • 74
  • 75
  • 76
  • 77
  • 78
  • 79
  • 80
  • 81
  • 82
  • 83
  • 84
  • 85
  • 86
  • 87
  • 88
  • 89
  • 90
  • 91
  • 92
  • 93
  • 94
  • 95
  • 96
  • 97
  • 98
  • 99
  • 100
  • 101
  • 102
  • 103
  • 104
  • 105
  • 106
  • 107
  • 108
  • 109
  • 110
  • 111
  • 112
  • 113
  • 114
  • 115
  • 116
  • 117
  • 118
  • 119

2. 函数功能概览

  1. 宏定义：

    • SIZE: 定义缓冲区大小为1024。
    • Info, Debug, Warning, Error, Fatal: 定义日志级别。
    • Screen, Onefile, Classfile: 定义日志输出方式。
    • LogFile: 定义默认日志文件名为 “log.txt”。

  2. 类的构造函数 Log()：

    • 初始化日志输出方式为 Screen（输出到屏幕）。
    • 设置日志文件路径为 “./log/”。

  3. Enable 函数：

    • 设置日志输出方式。

  4. levelToString 函数：

    • 将日志级别转换为对应的字符串表示。

  5. printLog 函数：

    • 根据设置的输出方式调用对应的日志打印函数（Screen、Onefile、Classfile）。

  6. printOneFile 函数：

    • 将日志写入单个文件。文件名为传入的 logname，实际路径为 path + logname。

  7. printClassFile 函数：

    • 将日志按级别分类写入不同的文件，文件名为 “log.txt.日志级别”。

  8. 析构函数 ~Log()：

    • 没有特殊操作。

  9. 重载的 operator()：

    • 使用可变参数处理日志信息，格式化后调用 printLog 打印日志。

3. 代码详细解释

3.1 头文件和宏定义

#pragma once

#include <iostream>
#include <time.h>
#include <stdarg.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>

#define SIZE 1024

#define Info 0
#define Debug 1
#define Warning 2
#define Error 3
#define Fatal 4

#define Screen 1
#define Onefile 2
#define Classfile 3

#define LogFile "log.txt"

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • #pragma once：确保头文件只被编译一次。
  • #include：引入了标准库和系统头文件。
  • #define：定义了一些常量，用于日志级别和输出方式。

3.2 Log类定义

class Log
{
public:
    Log()
    {
        printMethod = Screen;
        path = "./log/";
    }
    void Enable(int method)
    {
        printMethod = method;
    }
    std::string levelToString(int level)
    {
        switch (level)
        {
        case Info:
            return "Info";
        case Debug:
            return "Debug";
        case Warning:
            return "Warning";
        case Error:
            return "Error";
        case Fatal:
            return "Fatal";
        default:
            return "None";
        }
    }

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • Log()：构造函数，初始化日志输出方式为屏幕输出（Screen），并设置日志文件路径为./log/。
  • Enable(int method)：设置日志输出方法。
  • levelToString(int level)：将日志级别转换为字符串。

3.3 打印日志的方法

    void printLog(int level, const std::string &logtxt)
    {
        switch (printMethod)
        {
        case Screen:
            std::cout << logtxt << std::endl;
            break;
        case Onefile:
            printOneFile(LogFile, logtxt);
            break;
        case Classfile:
            printClassFile(level, logtxt);
            break;
        default:
            break;
        }
    }
    void printOneFile(const std::string &logname, const std::string &logtxt)
    {
        std::string _logname = path + logname;
        int fd = open(_logname.c_str(), O_WRONLY | O_CREAT | O_APPEND, 0666);
        if (fd < 0)
            return;
        write(fd, logtxt.c_str(), logtxt.size());
        close(fd);
    }
    void printClassFile(int level, const std::string &logtxt)
    {
        std::string filename = LogFile;
        filename += ".";
        filename += levelToString(level);
        printOneFile(filename, logtxt);
    }

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • printLog(int level, const std::string &logtxt)：根据不同的输出方式打印日志。
  • printOneFile(const std::string &logname, const std::string &logtxt)：将日志写入单个文件。
  • printClassFile(int level, const std::string &logtxt)：根据日志级别写入分类文件。

3.4 操作符重载和析构函数

    ~Log()
    {
    }
    void operator()(int level, const char *format, ...)
    {
        time_t t = time(nullptr);
        struct tm *ctime = localtime(&t);
        char leftbuffer[SIZE];
        snprintf(leftbuffer, sizeof(leftbuffer), "[%s][%d-%d-%d %d:%d:%d]", levelToString(level).c_str(),
                 ctime->tm_year + 1900, ctime->tm_mon + 1, ctime->tm_mday,
                 ctime->tm_hour, ctime->tm_min, ctime->tm_sec);

        va_list s;
        va_start(s, format);
        char rightbuffer[SIZE];
        vsnprintf(rightbuffer, sizeof(rightbuffer), format, s);
        va_end(s);

        char logtxt[SIZE * 2];
        snprintf(logtxt, sizeof(logtxt), "%s %s", leftbuffer, rightbuffer);

        printLog(level, logtxt);
    }

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23

operator()：重载了函数调用操作符，使得Log类的对象可以像函数一样被调用。

  • time_t t = time(nullptr); 获取当前时间。
  • struct tm *ctime = localtime(&t); 将时间转换为本地时间结构。
  • snprintf()格式化时间和日志级别。
  • va_list s; 定义一个可变参数列表。
  • va_start(s, format); 初始化可变参数列表。
  • vsnprintf(rightbuffer, sizeof(rightbuffer), format, s); 将可变参数格式化为字符串。
  • va_end(args); 结束可变参数处理。
  • 将 leftbuffer 和 rightbuffer 拼接到 logtxt 中。
  • printLog(level, logtxt)输出日志。

3.5 可变参数函数的原理

C++ 的可变参数函数使用 stdarg.h 中定义的一组宏来处理任意数量的参数。这些宏包括：

  • va_list: 声明一个变量来存储参数列表。
  • va_start: 初始化参数列表。
  • va_arg: 访问参数列表中的下一个参数。
  • va_end: 结束参数列表的访问。

4. 测试用例

下面是一些测试用例，用于测试多参数输入和不同的日志输出方式。

#include "log.hpp"

int main()
{
    Log logger;

    // 测试屏幕输出
    logger.Enable(Screen);
    logger(Info, "This is an info log with no parameters");
    logger(Warning, "This is a warning log with one parameter: %d", 42);
    logger(Error, "This is an error log with two parameters: %s and %d", "error_code", 404);

    // 测试单文件输出
    logger.Enable(Onefile);
    logger(Debug, "Debug log to one file with one parameter: %f", 3.14);
    logger(Fatal, "Fatal log to one file with no parameters");

    // 测试分类文件输出
    logger.Enable(Classfile);
    logger(Info, "Info log to class file");
    logger(Warning, "Warning log to class file with two parameters: %s and %d", "test", 123);

    return 0;
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24

这些测试用例将测试以下功能：

  1. 不同日志级别的信息输出。
  2. 带有不同数量参数的日志格式化。
  3. 日志输出到屏幕、单文件和分类文件。

注意：要保证日志目录存在（默认为 ./log/ ），日志文件才会自动创建在该目录内。

END
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/661949
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号