允许日志文件中添加颜色

This commit is contained in:
wanna
2021-07-13 22:45:35 +08:00
parent bc724deb5e
commit 84aa30d9c2

View File

@ -47,7 +47,12 @@ func (i *HandlerInput) addStringToBuffer(buffer *bytes.Buffer, s string) {
func (i *HandlerInput) Buffer() *bytes.Buffer {
buffer := bytes.NewBuffer(nil)
buffer.WriteString(i.TimeFormat)
i.addStringToBuffer(buffer, i.LevelFormat)
levelString := i.LevelFormat
if i.logger.config.FileColorEnable {
fg := i.getLevelFormatColor()
levelString = color.New(fg).Sprintf(i.LevelFormat)
}
i.addStringToBuffer(buffer, levelString)
msg := i.GetContent()
i.addStringToBuffer(buffer, msg.String())
return buffer