diff --git a/core/client.go b/core/client.go index d22b4ac2f..4013b047a 100644 --- a/core/client.go +++ b/core/client.go @@ -280,11 +280,6 @@ func (c *Client) Close() (err error) { return err } -// EnableDebug enables the development model for logging. -// func (c *Client) EnableDebug() { -// logger.EnableDebug() -// } - // WriteFrame writes a frame to the connection, gurantee threadsafe. func (c *Client) WriteFrame(frm frame.Frame) error { // write on QUIC stream diff --git a/core/log/logger.go b/core/log/logger.go index 03fc45150..79134c383 100644 --- a/core/log/logger.go +++ b/core/log/logger.go @@ -24,7 +24,7 @@ type Logger interface { SetLevel(Level) // SetEncoding sets the logger's encoding SetEncoding(encoding string) - // Printf logs a message wihout level + // Printf logs a message without level Printf(template string, args ...interface{}) // Debugf logs a message at DebugLevel Debugf(template string, args ...interface{}) diff --git a/pkg/logger/logger.go b/pkg/logger/logger.go index 7b0699d92..3f4ad1de8 100644 --- a/pkg/logger/logger.go +++ b/pkg/logger/logger.go @@ -7,9 +7,19 @@ import ( "github.com/yomorun/yomo/core/log" ) -var logger = Default(isEnableDebug()) +var logger log.Logger + +func init() { + logger = Default(isEnableDebug()) +} + +// SetLogger allows developers to customize the logger instance. +func SetLogger(l log.Logger) { + logger = l +} // EnableDebug enables the development model for logging. +// Deprecated func EnableDebug() { logger = Default(true) }