/
logger.go
57 lines (45 loc) · 1.09 KB
/
logger.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package logger
import (
"os"
"github.com/batx-dev/batproxy"
"golang.org/x/exp/slog"
)
type Logger struct {
*slog.Logger
}
type Options struct {
// slog.Logger set for return a source parent logger
*slog.Logger
// slog.HandlerOptions set for slog options
slog.HandlerOptions
// JsonHandler default false, means default log as `Text` format,
// non-effective when *slog.Logger is not nil
JsonHandler bool
}
func New(opts Options) *Logger {
if opts.Logger != nil {
return &Logger{opts.Logger}
}
if &opts.HandlerOptions == nil {
opts.HandlerOptions = slog.HandlerOptions{
AddSource: true,
Level: slog.LevelDebug,
ReplaceAttr: nil,
}
}
if opts.JsonHandler {
return &Logger{slog.New(opts.NewJSONHandler(os.Stdout))}
}
return &Logger{slog.New(opts.NewTextHandler(os.Stdout))}
}
func logErr(logger *slog.Logger, msg string, err error) {
code, message := batproxy.ErrorCode(err), batproxy.ErrorMessage(err)
switch code {
case "":
logger.Info(msg)
case batproxy.EINTERNAL:
logger.Error(msg, "err", err)
default:
logger.Error(msg, "code", code, "err", message)
}
}