feat: 输出日志到文件

yearnfar · Jun 29, 2020 · bae5f66 · bae5f66
1 parent a6b045c
commit bae5f66
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/internal/gexrender/gexrender.go b/internal/gexrender/gexrender.go
@@ -3,6 +3,7 @@ package gexrender
 import (
 	"errors"
 	"io/ioutil"
+	"os"
 	"path/filepath"
 
 	"github.com/go-playground/locales/zh"
@@ -34,6 +35,16 @@ func Render(setting *Setting) (err error) {
 		log.SetLevel(log.DebugLevel)
 	}
 
+	if setting.LogFile != "" {
+		logFile, err := os.OpenFile(setting.LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766)
+		if err != nil {
+			log.Error(err)
+			return err
+		}
+
+		log.SetOutput(logFile)
+	}
+
 	if setting.WorkPath == "" {
 		setting.WorkPath, err = ioutil.TempDir("", "gexrender")
 		if err != nil {

diff --git a/internal/gexrender/setting.go b/internal/gexrender/setting.go
@@ -25,6 +25,7 @@ type Setting struct {
 	SkipCleanup           bool   // 不清理缓存目录
 	ForceCommandLinePatch bool   // 补丁
 	Debug                 bool   // 调试模式
+	LogFile               string // 日志输出文件
 	MultiFrames           bool
 	Reuse                 bool
 	StopOnError           bool

diff --git a/pkg/gexrender/option.go b/pkg/gexrender/option.go
@@ -14,6 +14,13 @@ func WithDebug() Option {
 	}
 }
 
+// WithLogFile 日志输出到文件
+func WithLogFile(file string) Option {
+	return func(s *gexrender.Setting) {
+		s.LogFile = file
+	}
+}
+
 // WithConfigFile 提供一个包含作业的json文件的相对或绝对路径代替从参数中使用json。
 func WithConfigFile(filename string) Option {
 	return func(s *gexrender.Setting) {