add other session providers (#5963)

vendor/github.com/lunny/log/filewriter.go

@@ -0,0 +1,112 @@
+package log
+
+import (
+	"io"
+	"os"
+	"path/filepath"
+	"sync"
+	"time"
+)
+
+var _ io.Writer = &Files{}
+
+type ByType int
+
+const (
+	ByDay ByType = iota
+	ByHour
+	ByMonth
+)
+
+var (
+	formats = map[ByType]string{
+		ByDay:   "2006-01-02",
+		ByHour:  "2006-01-02-15",
+		ByMonth: "2006-01",
+	}
+)
+
+func SetFileFormat(t ByType, format string) {
+	formats[t] = format
+}
+
+func (b ByType) Format() string {
+	return formats[b]
+}
+
+type Files struct {
+	FileOptions
+	f          *os.File
+	lastFormat string
+	lock       sync.Mutex
+}
+
+type FileOptions struct {
+	Dir    string
+	ByType ByType
+	Loc    *time.Location
+}
+
+func prepareFileOption(opts []FileOptions) FileOptions {
+	var opt FileOptions
+	if len(opts) > 0 {
+		opt = opts[0]
+	}
+	if opt.Dir == "" {
+		opt.Dir = "./"
+	}
+	err := os.MkdirAll(opt.Dir, os.ModePerm)
+	if err != nil {
+		panic(err.Error())
+	}
+
+	if opt.Loc == nil {
+		opt.Loc = time.Local
+	}
+	return opt
+}
+
+func NewFileWriter(opts ...FileOptions) *Files {
+	opt := prepareFileOption(opts)
+	return &Files{
+		FileOptions: opt,
+	}
+}
+
+func (f *Files) getFile() (*os.File, error) {
+	var err error
+	t := time.Now().In(f.Loc)
+	if f.f == nil {
+		f.lastFormat = t.Format(f.ByType.Format())
+		f.f, err = os.OpenFile(filepath.Join(f.Dir, f.lastFormat+".log"),
+			os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0600)
+		return f.f, err
+	}
+	if f.lastFormat != t.Format(f.ByType.Format()) {
+		f.f.Close()
+		f.lastFormat = t.Format(f.ByType.Format())
+		f.f, err = os.OpenFile(filepath.Join(f.Dir, f.lastFormat+".log"),
+			os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0600)
+		return f.f, err
+	}
+	return f.f, nil
+}
+
+func (f *Files) Write(bs []byte) (int, error) {
+	f.lock.Lock()
+	defer f.lock.Unlock()
+
+	w, err := f.getFile()
+	if err != nil {
+		return 0, err
+	}
+	return w.Write(bs)
+}
+
+func (f *Files) Close() {
+	if f.f != nil {
+		f.f.Close()
+		f.f = nil
+	}
+	f.lastFormat = ""
+}