From 4b60f4bcbbd6593ae27ec328d1a283fdb36bb9c6 Mon Sep 17 00:00:00 2001
From: rjbasitali <rjbasitali@gmail.com>
Date: Sun, 18 Jul 2021 15:59:33 +0500
Subject: [PATCH] support for custom writer

---
 my_logger.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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,
 	}