Added enabled check while creating log files and logging. Do not allow to create log file or log if the log type is disabled.
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package elektronLogging
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type PCPLogger struct {
|
|
LoggerImpl
|
|
}
|
|
|
|
func NewPCPLogger(logType int, prefix string) *PCPLogger {
|
|
pLog := &PCPLogger{}
|
|
pLog.Type = logType
|
|
pLog.CreateLogFile(prefix)
|
|
return pLog
|
|
}
|
|
|
|
func (pLog PCPLogger) Log(logType int, level log.Level, logData log.Fields, message string) {
|
|
if config.PCPConfig.Enabled {
|
|
if pLog.Type == logType {
|
|
|
|
logger.SetLevel(level)
|
|
|
|
if pLog.AllowOnConsole {
|
|
logger.SetOutput(os.Stdout)
|
|
logger.WithFields(logData).Println(message)
|
|
}
|
|
|
|
logger.SetOutput(pLog.LogFile)
|
|
logger.WithFields(logData).Println(message)
|
|
}
|
|
if pLog.next != nil {
|
|
pLog.next.Log(logType, level, logData, message)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (pLog *PCPLogger) CreateLogFile(prefix string) {
|
|
if config.PCPConfig.Enabled {
|
|
filename := strings.Join([]string{prefix, config.PCPConfig.FilenameExtension}, "")
|
|
dirName := logDir.getDirName()
|
|
if dirName != "" {
|
|
if logFile, err := os.Create(filepath.Join(dirName, filename)); err != nil {
|
|
log.Fatal("Unable to create logFile: ", err)
|
|
} else {
|
|
pLog.LogFile = logFile
|
|
pLog.AllowOnConsole = config.PCPConfig.AllowOnConsole
|
|
}
|
|
}
|
|
}
|
|
}
|