mirror of
https://github.com/open-policy-agent/opa.git
synced 2026-08-12 19:32:48 -06:00
Anders Eknert
e03ac2f200
- Bump golangci-lint -> 2.6.2 - Fix all `deprecatedComment` "notices should be in a dedicated paragraph, separated from the rest" reports - Enable `appendCombine` and fix all "appendCombine: can combine chain of X appends into one" notices - Enable `preferFprint` and fix the few reported issues - Fix various issues reported only once or twice, like `zeroByteRepeat` Signed-off-by: Anders Eknert <anders.eknert@apple.com>
106 lines
2.7 KiB
Go
106 lines
2.7 KiB
Go
// Copyright 2021 The OPA Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache2
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/open-policy-agent/opa/v1/logging"
|
|
)
|
|
|
|
func GetLevel(level string) (logging.Level, error) {
|
|
switch strings.ToLower(level) {
|
|
case "debug":
|
|
return logging.Debug, nil
|
|
case "", "info":
|
|
return logging.Info, nil
|
|
case "warn":
|
|
return logging.Warn, nil
|
|
case "error":
|
|
return logging.Error, nil
|
|
default:
|
|
return logging.Debug, fmt.Errorf("invalid log level: %v", level)
|
|
}
|
|
}
|
|
|
|
func GetFormatter(format, timestampFormat string) logrus.Formatter {
|
|
switch format {
|
|
case "text":
|
|
return &prettyFormatter{}
|
|
case "json-pretty":
|
|
return &logrus.JSONFormatter{PrettyPrint: true, TimestampFormat: timestampFormat}
|
|
default:
|
|
return &logrus.JSONFormatter{TimestampFormat: timestampFormat}
|
|
}
|
|
}
|
|
|
|
// prettyFormatter implements the Logrus Formatter interface
|
|
// and provides a more simple, but easier to read, text formatter
|
|
// option than the default logrus.TextFormatter.
|
|
type prettyFormatter struct{}
|
|
|
|
func spaces(num int) string {
|
|
return strings.Repeat(" ", num)
|
|
}
|
|
|
|
func (*prettyFormatter) Format(e *logrus.Entry) ([]byte, error) {
|
|
b := new(bytes.Buffer)
|
|
|
|
level := strings.ToUpper(e.Level.String())
|
|
fmt.Fprintf(b, "[%s] %s\n", level, e.Message)
|
|
|
|
// Format each key for optimal ease of human reading
|
|
fieldIndent := 2
|
|
multiLineIndent := 6
|
|
for k, v := range e.Data {
|
|
// Special case for multi-line strings, keep them as-is
|
|
// but indent them. Everything else gets json'd
|
|
stringVal, ok := v.(string)
|
|
if ok && strings.Contains(stringVal, "\n") {
|
|
sb := strings.Builder{}
|
|
for i, line := range strings.Split(stringVal, "\n") {
|
|
// match the json indent helper by not indenting the first value
|
|
if i != 0 {
|
|
sb.WriteString(spaces(multiLineIndent))
|
|
}
|
|
sb.WriteString(line)
|
|
sb.WriteByte('\n')
|
|
stringVal = sb.String()
|
|
}
|
|
} else if ok && json.Valid([]byte(stringVal)) {
|
|
var tmp bytes.Buffer
|
|
err := json.Indent(&tmp, []byte(stringVal), spaces(multiLineIndent), spaces(2))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stringVal = tmp.String()
|
|
} else {
|
|
jsonVal, err := json.MarshalIndent(v, spaces(multiLineIndent), spaces(2))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stringVal = string(jsonVal)
|
|
}
|
|
|
|
b.WriteString(spaces(fieldIndent))
|
|
b.WriteString(k)
|
|
if strings.Contains(stringVal, "\n") {
|
|
b.WriteString(" = |\n")
|
|
b.WriteString(spaces(multiLineIndent))
|
|
} else {
|
|
b.WriteString(" = ")
|
|
}
|
|
b.WriteString(stringVal)
|
|
b.WriteString("\n")
|
|
}
|
|
b.WriteByte('\n')
|
|
return b.Bytes(), nil
|
|
}
|