diff --git a/my_logger.go b/my_logger.go index 2fc1007..5580917 100644 --- a/my_logger.go +++ b/my_logger.go @@ -44,9 +44,12 @@ func funcName() string { return "" } -func NewLogger() Logger { +func NewLogger(w io.Writer) Logger { + if w == nil { + w = os.Stdout + } return myLogger{ - Writer: os.Stdout, + Writer: w, begin: time.Now(), level: 63, }