add: 添加输出记录到文件的能力2

XinRoom · Feb 21, 2023 · e33a54e · e33a54e
1 parent 9871f5c
commit e33a54e
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/util/log.go b/util/log.go
@@ -0,0 +1,19 @@
+package util
+
+import (
+	"io"
+	"log"
+	"os"
+)
+
+func NewLogger(filename string, std bool) *log.Logger {
+	var out io.Writer
+	outFile, _ := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
+	if std {
+		out = io.MultiWriter(os.Stdout, outFile)
+	} else {
+		out = outFile
+	}
+	logger := log.New(out, "", 0)
+	return logger
+}