1818package logger
1919
2020import (
21- "io"
22- "os"
23-
24- "github.com/go-logr/logr"
25- "github.com/go-logr/zapr"
26- "go.uber.org/zap"
27- "go.uber.org/zap/zapcore"
28- "hertzbeat.apache.org/hertzbeat-collector-go/internal/collector/common/types/logger"
21+ "io"
22+ "os"
23+
24+ "github.com/go-logr/logr"
25+ "github.com/go-logr/zapr"
26+ "go.uber.org/zap"
27+ "go.uber.org/zap/zapcore"
28+
29+ "hertzbeat.apache.org/hertzbeat-collector-go/internal/collector/common/types/logger"
2930)
3031
3132type Logger struct {
32- logr.Logger
33- out io.Writer
34- logging * logger.HertzBeatLogging
35- sugaredLogger * zap.SugaredLogger
33+ logr.Logger
34+ out io.Writer
35+ logging * logger.HertzBeatLogging
36+ sugaredLogger * zap.SugaredLogger
3637}
3738
3839func NewLogger (w io.Writer , logging * logger.HertzBeatLogging ) Logger {
3940
40- logger := initZapLogger (w , logging , logging .Level [logger .LogComponentHertzbeatDefault ])
41+ logger := initZapLogger (w , logging , logging .Level [logger .LogComponentHertzbeatDefault ])
4142
42- return Logger {
43- Logger : zapr .NewLogger (logger ),
44- out : w ,
45- logging : logging ,
46- sugaredLogger : logger .Sugar (),
47- }
43+ return Logger {
44+ Logger : zapr .NewLogger (logger ),
45+ out : w ,
46+ logging : logging ,
47+ sugaredLogger : logger .Sugar (),
48+ }
4849}
4950
5051func FileLogger (file , name string , level logger.LogLevel ) Logger {
5152
52- writer , err := os .OpenFile (file , os .O_WRONLY , 0o666 )
53- if err != nil {
54- panic (err )
55- }
53+ writer , err := os .OpenFile (file , os .O_WRONLY , 0o666 )
54+ if err != nil {
55+ panic (err )
56+ }
5657
57- logging := logger .DefaultHertzbeatLogging ()
58- logger := initZapLogger (writer , logging , level )
58+ logging := logger .DefaultHertzbeatLogging ()
59+ logger := initZapLogger (writer , logging , level )
5960
60- return Logger {
61- Logger : zapr .NewLogger (logger ).WithName (name ),
62- logging : logging ,
63- out : writer ,
64- sugaredLogger : logger .Sugar (),
65- }
61+ return Logger {
62+ Logger : zapr .NewLogger (logger ).WithName (name ),
63+ logging : logging ,
64+ out : writer ,
65+ sugaredLogger : logger .Sugar (),
66+ }
6667}
6768
6869func DefaultLogger (out io.Writer , level logger.LogLevel ) Logger {
6970
70- logging := logger .DefaultHertzbeatLogging ()
71- logger := initZapLogger (out , logging , level )
71+ logging := logger .DefaultHertzbeatLogging ()
72+ logger := initZapLogger (out , logging , level )
7273
73- return Logger {
74- Logger : zapr .NewLogger (logger ),
75- out : out ,
76- logging : logging ,
77- sugaredLogger : logger .Sugar (),
78- }
74+ return Logger {
75+ Logger : zapr .NewLogger (logger ),
76+ out : out ,
77+ logging : logging ,
78+ sugaredLogger : logger .Sugar (),
79+ }
7980}
8081
8182// WithName returns a new Logger instance with the specified name element added
@@ -85,23 +86,23 @@ func DefaultLogger(out io.Writer, level logger.LogLevel) Logger {
8586// more information).
8687func (l Logger ) WithName (name string ) Logger {
8788
88- logLevel := l .logging .Level [logger .HertzbeatLogComponent (name )]
89- logger := initZapLogger (l .out , l .logging , logLevel )
89+ logLevel := l .logging .Level [logger .HertzbeatLogComponent (name )]
90+ logger := initZapLogger (l .out , l .logging , logLevel )
9091
91- return Logger {
92- Logger : zapr .NewLogger (logger ).WithName (name ),
93- logging : l .logging ,
94- out : l .out ,
95- sugaredLogger : logger .Sugar ().Named (name ),
96- }
92+ return Logger {
93+ Logger : zapr .NewLogger (logger ).WithName (name ),
94+ logging : l .logging ,
95+ out : l .out ,
96+ sugaredLogger : logger .Sugar ().Named (name ),
97+ }
9798}
9899
99100// WithValues returns a new Logger instance with additional key/value pairs.
100101// See Info for documentation on how key/value pairs work.
101102func (l Logger ) WithValues (keysAndValues ... interface {}) Logger {
102103
103- l .Logger = l .Logger .WithValues (keysAndValues ... )
104- return l
104+ l .Logger = l .Logger .WithValues (keysAndValues ... )
105+ return l
105106}
106107
107108// A Sugar wraps the base Logger functionality in a slower, but less
@@ -124,13 +125,13 @@ func (l Logger) WithValues(keysAndValues ...interface{}) Logger {
124125// Infoln(...any) Println-style logger
125126func (l Logger ) Sugar () * zap.SugaredLogger {
126127
127- return l .sugaredLogger
128+ return l .sugaredLogger
128129}
129130
130131func initZapLogger (w io.Writer , logging * logger.HertzBeatLogging , level logger.LogLevel ) * zap.Logger {
131132
132- parseLevel , _ := zapcore .ParseLevel (string (logging .DefaultHertzBeatLoggingLevel (level )))
133- core := zapcore .NewCore (zapcore .NewConsoleEncoder (zap .NewDevelopmentEncoderConfig ()), zapcore .AddSync (w ), zap .NewAtomicLevelAt (parseLevel ))
133+ parseLevel , _ := zapcore .ParseLevel (string (logging .DefaultHertzBeatLoggingLevel (level )))
134+ core := zapcore .NewCore (zapcore .NewConsoleEncoder (zap .NewDevelopmentEncoderConfig ()), zapcore .AddSync (w ), zap .NewAtomicLevelAt (parseLevel ))
134135
135- return zap .New (core , zap .AddCaller ())
136+ return zap .New (core , zap .AddCaller ())
136137}
0 commit comments