62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
|
package dns
|
||
|
|
|
||
|
|
import (
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/qdm12/gluetun/internal/constants"
|
||
|
|
"github.com/qdm12/golibs/logging"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (l *looper) collectLines(wg *sync.WaitGroup, stdout, stderr <-chan string) {
|
||
|
|
defer wg.Done()
|
||
|
|
var line string
|
||
|
|
var ok bool
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case line, ok = <-stderr:
|
||
|
|
case line, ok = <-stdout:
|
||
|
|
}
|
||
|
|
if !ok {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
line, level := processLogLine(line)
|
||
|
|
switch level {
|
||
|
|
case logging.DebugLevel:
|
||
|
|
l.logger.Debug(line)
|
||
|
|
case logging.InfoLevel:
|
||
|
|
l.logger.Info(line)
|
||
|
|
case logging.WarnLevel:
|
||
|
|
l.logger.Warn(line)
|
||
|
|
case logging.ErrorLevel:
|
||
|
|
l.logger.Error(line)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var unboundPrefix = regexp.MustCompile(`\[[0-9]{10}\] unbound\[[0-9]+:[0|1]\] `)
|
||
|
|
|
||
|
|
func processLogLine(s string) (filtered string, level logging.Level) {
|
||
|
|
prefix := unboundPrefix.FindString(s)
|
||
|
|
filtered = s[len(prefix):]
|
||
|
|
switch {
|
||
|
|
case strings.HasPrefix(filtered, "notice: "):
|
||
|
|
filtered = strings.TrimPrefix(filtered, "notice: ")
|
||
|
|
level = logging.InfoLevel
|
||
|
|
case strings.HasPrefix(filtered, "info: "):
|
||
|
|
filtered = strings.TrimPrefix(filtered, "info: ")
|
||
|
|
level = logging.InfoLevel
|
||
|
|
case strings.HasPrefix(filtered, "warn: "):
|
||
|
|
filtered = strings.TrimPrefix(filtered, "warn: ")
|
||
|
|
level = logging.WarnLevel
|
||
|
|
case strings.HasPrefix(filtered, "error: "):
|
||
|
|
filtered = strings.TrimPrefix(filtered, "error: ")
|
||
|
|
level = logging.ErrorLevel
|
||
|
|
default:
|
||
|
|
level = logging.InfoLevel
|
||
|
|
}
|
||
|
|
filtered = constants.ColorUnbound().Sprintf(filtered)
|
||
|
|
return filtered, level
|
||
|
|
}
|