40 lines
672 B
Go
40 lines
672 B
Go
package logger
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
// Mask is a passthrough for MaskIf
|
|
func Mask(key string, val interface{}) zap.Field {
|
|
return MaskIf(key, val, true)
|
|
}
|
|
|
|
// MaskIf conditionally masks (replaces string with * of equal length input string/stringer
|
|
func MaskIf(key string, val interface{}, mask bool) zap.Field {
|
|
var str string
|
|
|
|
switch v := val.(type) {
|
|
case fmt.Stringer:
|
|
str = v.String()
|
|
case string:
|
|
str = v
|
|
default:
|
|
// empty string for other types
|
|
str = ""
|
|
}
|
|
|
|
if mask {
|
|
str = strings.Repeat("*", len(str))
|
|
}
|
|
|
|
return zap.Field{
|
|
Key: key,
|
|
Type: zapcore.StringType,
|
|
String: str,
|
|
}
|
|
}
|