Logrus是go语言开发中应用十分广泛的开源日志组件,具有日志分级打印、日志文件轮转分割、自动删除过期日志文件等功能,在github上拥有14.2K的star;
众所周知,Logrus的日志文件轮转规则为定期生成一个带日期格式的日志文件,类似XXX_2020.03.19.log、XXX_2020.03.20.log等形式。但在某些场景下我们需要一个固定名称的最新日志文件,且不能是软连接(emm,这个需求也挺奇葩的,奈何我们也左右不了)。这时如果还想用logrus做工程的日志组件的话(比如你已经用这个写了一坨坨代码了,实在不想回去搅屎了。事实上,我也没找到很契合支持这个的其他日志组件,可能也是我懒吧),那么就该在logrus源码上动刀了。
在手术之前确保你足够了解你的牛,成不了庖丁也别差太远。logrus整体包含三部分:logrus本身、本地文件系统lfsHook、轮转file-rotatelogs。logrus本身只负责写日志,当我们调用logger.Debug等语句时,logrus去获取一个writer,获取成功了就写,别的不管。lfsHook是写入处理的钩子,因为写的时候我们只传入了内容message,至于这个message最后记录在文件中用什么格式,是就一句话,还是用个类似json的字段分割,加不加上时间戳等等这些都有钩子决定,所以这个是你定制化日志格式的东西。最后,file-rotatelogs负责日志文件轮转,它不管你要写什么,只管到时候了就创建新文件,然后把新文件的writer返回给logrus,如果有软连就把软件指向新文件,完成轮转。大体上就这么简单,就简要说下,具体的还是你去看代码去。这里有一点就是,我们可能以为轮转是定时器控制的,但其实不是,而是每次调用一次logger.Debug类函数就触发一次是否要轮转的流程 ,而且进入流程必须加锁来包含并发。所以效率其实有点低,虽然做了一些空间换时间的技巧,在不需要轮转的情况下不用走完整个流程,但加锁就注定并发性能高不到哪儿去。
背景就说这么多,下边上代码。以下代码是在按天分割日志的需求下改的,有点偏硬,你理解了后可以自己重写。思路就是在logrus获取日志文件writer的轮转判断流程里,不生成带日期名称的当天日志文件,而是先把昨天的固定名称日志文件重命名成带日期的文件名(日期是文件的创建日期),然后再生成一个固定的日志名(我这里是pai-customer.log,是上层应用传过来的),再把writer返回。在不需要轮转的情况下,直接返回当前固定名称的日志文件writer,结束流程。
首先扩展了interface.go文件中的RotateLogs结构体&