@@ -2,55 +2,54 @@ package gcloud
22
33import (
44 gologger "github.com/ralvarezdev/go-logger"
5- gologgerstatus "github.com/ralvarezdev/go-logger/status"
5+ gologgermode "github.com/ralvarezdev/go-logger/mode"
6+ gologgermodenamed "github.com/ralvarezdev/go-logger/mode/named"
67 "google.golang.org/grpc/credentials/oauth"
78)
89
910// Logger is the logger for Google Cloud
1011type Logger struct {
11- logger gologger .Logger
12+ logger gologgermodenamed .Logger
1213}
1314
1415// NewLogger is the logger for Google Cloud
15- func NewLogger (logger gologger .Logger ) (* Logger , error ) {
16+ func NewLogger (header string , modeLogger gologgermode .Logger ) (* Logger , error ) {
1617 // Check if the logger is nil
17- if logger == nil {
18+ if modeLogger == nil {
1819 return nil , gologger .ErrNilLogger
1920 }
2021
21- return & Logger {logger : logger }, nil
22+ // Initialize the mode named logger
23+ namedLogger , _ := gologgermodenamed .NewDefaultLogger (header , modeLogger )
24+
25+ return & Logger {logger : namedLogger }, nil
26+ }
27+
28+ // FailedToLoadTokenSource logs the failed to load token source
29+ func (l * Logger ) FailedToLoadTokenSource (err error ) {
30+ l .logger .Error (
31+ "failed to load token source" ,
32+ err ,
33+ )
2234}
2335
2436// LoadedTokenSource logs the loaded token source
2537func (l * Logger ) LoadedTokenSource (tokenSource * oauth.TokenSource ) {
2638 // Check if the token source is nil
2739 if tokenSource == nil {
28- l .logger .LogError (
29- gologger .NewLogError (
30- "failed to load token source" ,
31- ErrNilTokenSource ,
32- ),
33- )
40+ l .FailedToLoadTokenSource (ErrNilTokenSource )
3441 return
3542 }
3643
3744 // Get the access token from the token source
3845 token , err := tokenSource .Token ()
3946 if err != nil {
40- l .logger .LogError (
41- gologger .NewLogError (
42- "failed to load token source" ,
43- err ,
44- ),
45- )
47+ l .FailedToLoadTokenSource (err )
4648 return
4749 }
4850
49- l .logger .LogMessage (
50- gologger .NewLogMessage (
51- "loaded token source" ,
52- gologgerstatus .Debug ,
53- token .AccessToken ,
54- ),
51+ l .logger .Debug (
52+ "loaded token source" ,
53+ token .AccessToken ,
5554 )
5655}
0 commit comments