add file recorder library

internal/log/file_record.go

@@ -0,0 +1,74 @@
+package log
+
+import (
+	"encoding/csv"
+	"fmt"
+	"github.com/tickstep/aliyunpan/internal/utils"
+	"github.com/tickstep/library-go/converter"
+	"github.com/tickstep/library-go/logger"
+	"os"
+	"path/filepath"
+	"sync"
+)
+
+type (
+	FileRecordItem struct {
+		Status   string `json:"status"`
+		TimeStr  string `json:"timeStr"`
+		FileSize int64  `json:"fileSize"`
+		FilePath string `json:"filePath"`
+	}
+
+	FileRecorder struct {
+		Path   string `json:"path"`
+		locker *sync.Mutex
+	}
+)
+
+// NewFileRecorder 创建文件记录器
+func NewFileRecorder(filePath string) *FileRecorder {
+	return &FileRecorder{
+		Path:   filePath,
+		locker: &sync.Mutex{},
+	}
+}
+
+// Append 增加数据记录
+func (f *FileRecorder) Append(item *FileRecordItem) error {
+	savePath := f.Path
+	folder := filepath.Dir(savePath)
+	if b, err := utils.PathExists(folder); err == nil && !b {
+		os.MkdirAll(folder, os.ModePerm)
+	}
+
+	var fp *os.File
+	var write *csv.Writer
+	if b, err := utils.PathExists(savePath); err == nil && b {
+		file, err1 := os.OpenFile(savePath, os.O_APPEND, 0755)
+		if err1 != nil {
+			logger.Verbosef("打开文件["+savePath+"]失败,%v", err1)
+			return err1
+		}
+		fp = file
+		write = csv.NewWriter(fp) //创建一个新的写入文件流
+	} else {
+		file, err1 := os.OpenFile(savePath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755) // 创建文件句柄
+		if err1 != nil {
+			logger.Verbosef("创建文件["+savePath+"]失败,%v", err1)
+			return err1
+		}
+		fp = file
+		fp.WriteString("\xEF\xBB\xBF") // 写入UTF-8 BOM
+		write = csv.NewWriter(fp)      //创建一个新的写入文件流
+		write.Write([]string{"状态", "时间", "文件大小", "文件路径"})
+	}
+	if fp == nil || write == nil {
+		return fmt.Errorf("open recorder file error")
+	}
+	defer fp.Close()
+
+	data := []string{item.Status, item.TimeStr, converter.ConvertFileSize(item.FileSize, 2), item.FilePath}
+	write.Write(data)
+	write.Flush()
+	return nil
+}

internal/log/file_record_test.go

@@ -0,0 +1,16 @@
+package log
+
+import (
+	"testing"
+)
+
+func TestCsvFile(t *testing.T) {
+	savePath := "D:\\smb\\feny\\goprojects\\dev\\logs\\file_upload_records.csv"
+	recorder := NewFileRecorder(savePath)
+	recorder.Append(&FileRecordItem{
+		Status:   "成功",
+		TimeStr:  "2022-12-19 16:46:36",
+		FileSize: 453450,
+		FilePath: "D:\\smb\\feny\\goprojects\\dev\\myfile.mp4",
+	})
+}