diff options
Diffstat (limited to 'utils')
-rw-r--r-- | utils/config_test.go | 19 | ||||
-rw-r--r-- | utils/redirect_std_log.go | 65 |
2 files changed, 84 insertions, 0 deletions
diff --git a/utils/config_test.go b/utils/config_test.go index bce85d2ae..3032766ec 100644 --- a/utils/config_test.go +++ b/utils/config_test.go @@ -6,6 +6,7 @@ package utils import ( "os" "testing" + "time" "github.com/mattermost/platform/model" ) @@ -59,7 +60,25 @@ func TestConfigFromEnviroVars(t *testing.T) { if Cfg.TeamSettings.SiteName != "Mattermost" { t.Fatal("should have been reset") } +} + +func TestRedirectStdLog(t *testing.T) { + TranslationsPreInit() + LoadConfig("config.json") + InitTranslations(Cfg.LocalizationSettings) + + log := NewRedirectStdLog("test", false) + + log.Println("[DEBUG] this is a message") + log.Println("[DEBG] this is a message") + log.Println("[WARN] this is a message") + log.Println("[ERROR] this is a message") + log.Println("[EROR] this is a message") + log.Println("[ERR] this is a message") + log.Println("[INFO] this is a message") + log.Println("this is a message") + time.Sleep(time.Second * 1) } func TestAddRemoveConfigListener(t *testing.T) { diff --git a/utils/redirect_std_log.go b/utils/redirect_std_log.go new file mode 100644 index 000000000..4fbfcf8ec --- /dev/null +++ b/utils/redirect_std_log.go @@ -0,0 +1,65 @@ +// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved. +// See License.txt for license information. + +package utils + +import ( + "bufio" + "log" + "os" + "strings" + + l4g "github.com/alecthomas/log4go" +) + +type RedirectStdLog struct { + reader *os.File + writer *os.File + system string + ignoreDebug bool +} + +func NewRedirectStdLog(system string, ignoreDebug bool) *log.Logger { + r, w, _ := os.Pipe() + logger := &RedirectStdLog{ + reader: r, + writer: w, + system: system, + ignoreDebug: ignoreDebug, + } + + go func(l *RedirectStdLog) { + scanner := bufio.NewScanner(l.reader) + for scanner.Scan() { + line := scanner.Text() + + if strings.Index(line, "[DEBUG]") == 0 { + if !ignoreDebug { + l4g.Debug("%v%v", system, line[7:]) + } + } else if strings.Index(line, "[DEBG]") == 0 { + if !ignoreDebug { + l4g.Debug("%v%v", system, line[6:]) + } + } else if strings.Index(line, "[WARN]") == 0 { + l4g.Info("%v%v", system, line[6:]) + } else if strings.Index(line, "[ERROR]") == 0 { + l4g.Error("%v%v", system, line[7:]) + } else if strings.Index(line, "[EROR]") == 0 { + l4g.Error("%v%v", system, line[6:]) + } else if strings.Index(line, "[ERR]") == 0 { + l4g.Error("%v%v", system, line[5:]) + } else if strings.Index(line, "[INFO]") == 0 { + l4g.Info("%v%v", system, line[6:]) + } else { + l4g.Info("%v %v", system, line) + } + } + }(logger) + + return log.New(logger.writer, "", 0) +} + +func (l *RedirectStdLog) Write(p []byte) (n int, err error) { + return l.writer.Write(p) +} |