为RPC服务增加异步日志模块

在实际开发过程中我们不可能将一些错误信息或是提示信息全部打印到屏幕上,而是将这些信息输送到日志中去。但是存在着一个问题,写日志是在磁盘中写日志,是磁盘的io操作,效率会很慢,致使项目的效率降低。

为了解决这一个问题我们可以先将需要输出的信息,输送到一个中间件当中,再由中间件单独启动一个线程去写磁盘文件,这样不会影响整个项目的效率。这也就是设计一个缓冲队列来先接收日志信息,再由缓冲队列写到磁盘中。

同样,即使这样在很大程度上提高了效率,但是仍有很大的问题,我们的设计是基于muduo库,他具有epoll+多线程的特点。因此在往消息队列写日志的时候一定要考虑线程安全问题,在前面的文章的也提到了为什么要保证线程安全(每个线程分配的CPU时间片是有限的,假设第一个线程读到了数据,但时间片已经到了,第二个线程拿到了CPU的时间片进行了文件的写操作,当再次到第一个线程的时间片时,他会将读到的数据写入,将覆盖第二个线程写的数据,导致了数据的错误),所以在日志模块中我们要加锁,来保证线程的安全。即在往日志队列中放数据时我们要保证线程安全,并且在将日志队列中的数据pop出来时同样需要拿到锁 ,因为这涉及到对同一资源的访问。并且当队列为空时我们不能取数据,这涉及到了线程间的通信,在C++中我们使用条件变量来进行通信,即在队列为空时,需要进行等待操作。

 下面为日志模块:

Logger& Logger::GetInstance()
{
    static Logger logger;
    return logger;
}

Logger::Logger()
{
    // 启动专门的写日志线程
    std::thread writeLogTask([&](){
        for (;;)
        {
            // 获取当前的日期,然后取日志信息,写入相应的日志文件当中 a+
            time_t now = time(nullptr);
            tm *nowtm = localtime(&now);

            char file_name[128];
            sprintf(file_name, "%d-%d-%d-log.txt", nowtm->tm_year+1900, nowtm->tm_mon+1, nowtm->tm_mday);

            FILE *pf = fopen(file_name, "a+");
            if (pf == nullptr)
            {
                std::cout << "logger file : " << file_name << " open error!" << std::endl;
                exit(EXIT_FAILURE);
            }

            std::string msg = m_lckQue.Pop();

            char time_buf[128] = {0};
            sprintf(time_buf, "%d:%d:%d =>[%s] ", 
                    nowtm->tm_hour, 
                    nowtm->tm_min, 
                    nowtm->tm_sec,
                    (m_loglevel == INFO ? "info" : "error"));
            msg.insert(0, time_buf);
            msg.append("\n");

            fputs(msg.c_str(), pf);
            fclose(pf);
        }
    });
    // 设置分离线程,守护线程
    writeLogTask.detach();
}

// 设置日志级别 
void Logger::SetLogLevel(LogLevel level)
{
    m_loglevel = level;
}

// 写日志, 把日志信息写入lockqueue缓冲区当中
void Logger::Log(std::string msg)
{
    m_lckQue.Push(msg);
}
#pragma once
#include <queue>
#include <thread>
#include <mutex> // pthread_mutex_t
#include <condition_variable> // pthread_condition_t

// 异步写日志的日志队列
template<typename T>
class LockQueue
{
public:
    // 多个worker线程都会写日志queue 
    void Push(const T &data)
    {
        std::lock_guard<std::mutex> lock(m_mutex);
        m_queue.push(data);
        m_condvariable.notify_one();
    }

    // 一个线程读日志queue,写日志文件
    T Pop()
    {
        std::unique_lock<std::mutex> lock(m_mutex);
        while (m_queue.empty())
        {
            // 日志队列为空,线程进入wait状态
            m_condvariable.wait(lock);
        }

        T data = m_queue.front();
        m_queue.pop();
        return data;
    }
private:
    std::queue<T> m_queue;
    std::mutex m_mutex;
    std::condition_variable m_condvariable;
};

我们可以通过定义宏来简化使用日志的操作:

#define LOG_INFO(logmsgformat, ...) \

    do \

    {  \

        Logger &logger = Logger::GetInstance(); \

        logger.SetLogLevel(INFO); \

        char c[1024] = {0}; \

        snprintf(c, 1024, logmsgformat, ##__VA_ARGS__); \

        logger.Log(c); \

    } while(0) \

#define LOG_ERR(logmsgformat, ...) \

    do \

    {  \

        Logger &logger = Logger::GetInstance(); \            //得到唯一的单例实例

        logger.SetLogLevel(ERROR); \

        char c[1024] = {0}; \

        snprintf(c, 1024, logmsgformat, ##__VA_ARGS__); \

        logger.Log(c); \

    } while(0) \

  • 22
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值