From 4cec01609a40b04dc03fcdff881636966c309afe Mon Sep 17 00:00:00 2001 From: mountain Date: Tue, 13 Jun 2023 23:49:37 +0900 Subject: [PATCH] =?UTF-8?q?=EB=A1=9C=EA=B1=B0=20=EC=A0=95=EB=A6=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- shared/logger.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/shared/logger.go b/shared/logger.go index 2e35f10..1b7ea1b 100644 --- a/shared/logger.go +++ b/shared/logger.go @@ -1,19 +1,24 @@ package shared import ( + "io" "log" "os" "path" ) var defLogger *log.Logger +var defCloser func() func InitLogger() { name, _ := os.Executable() base := path.Base(name) logfile, _ := os.OpenFile(base+".log", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0666) - defLogger = log.New(logfile, "", log.LstdFlags) + defCloser = func() { + logfile.Close() + } + defLogger = log.New(io.MultiWriter(logfile, os.Stdout), "", log.LstdFlags) } func Logger() *log.Logger { @@ -21,8 +26,7 @@ func Logger() *log.Logger { } func CloseLogger() { - outfile := defLogger.Writer().(*os.File) - if outfile != nil { - outfile.Close() + if defCloser != nil { + defCloser() } }