Feat: use sync.pool to obtain new FileSystem instance

pkg/filesystem/filesystem.go

@@ -11,8 +11,16 @@ import (
 	"github.com/gin-gonic/gin"
 	"io"
 	"net/url"
+	"sync"
 )
 
+// FSPool 文件系统资源池
+var FSPool = sync.Pool{
+	New: func() interface{} {
+		return &FileSystem{}
+	},
+}
+
 // FileHeader 上传来的文件数据处理器
 type FileHeader interface {
 	io.Reader
@@ -77,12 +85,35 @@ type FileSystem struct {
 	Handler Handler
 }
 
+// getEmptyFS 从pool中获取新的FileSystem
+func getEmptyFS() *FileSystem {
+	fs := FSPool.Get().(*FileSystem)
+	return fs
+}
+
+// Recycle 回收FileSystem资源
+func (fs *FileSystem) Recycle() {
+	fs.reset()
+	FSPool.Put(fs)
+}
+
+// reset 重设文件系统，以便回收使用
+func (fs *FileSystem) reset() {
+	fs.User = nil
+	fs.CleanTargets()
+	fs.Policy = nil
+	fs.BeforeUpload = fs.BeforeUpload[:0]
+	fs.AfterUpload = fs.AfterUpload[:0]
+	fs.AfterValidateFailed = fs.AfterValidateFailed[:0]
+	fs.AfterUploadCanceled = fs.AfterUploadCanceled[:0]
+	fs.BeforeFileDownload = fs.BeforeFileDownload[:0]
+	fs.Handler = nil
+}
+
 // NewFileSystem 初始化一个文件系统
 func NewFileSystem(user *model.User) (*FileSystem, error) {
-	fs := &FileSystem{
-		User: user,
-	}
-
+	fs := getEmptyFS()
+	fs.User = user
 	// 分配存储策略适配器
 	err := fs.dispatchHandler()
 
@@ -92,9 +123,8 @@ func NewFileSystem(user *model.User) (*FileSystem, error) {
 
 // NewAnonymousFileSystem 初始化匿名文件系统
 func NewAnonymousFileSystem() (*FileSystem, error) {
-	fs := &FileSystem{
-		User: &model.User{},
-	}
+	fs := getEmptyFS()
+	fs.User = &model.User{}
 
 	// 如果是主机模式下，则为匿名文件系统分配游客用户组
 	if conf.SystemConfig.Mode == "master" {
@@ -182,6 +212,6 @@ func (fs *FileSystem) SetTargetFileByIDs(ids []uint) error {
 
 // CleanTargets 清空目标
 func (fs *FileSystem) CleanTargets() {
-	fs.FileTarget = []model.File{}
-	fs.DirTarget = []model.Folder{}
+	fs.FileTarget = fs.FileTarget[:0]
+	fs.DirTarget = fs.DirTarget[:0]
 }

pkg/filesystem/filesystem_test.go

@@ -133,3 +133,18 @@ func TestNewAnonymousFileSystem(t *testing.T) {
 		asserts.Nil(fs)
 	}
 }
+
+func TestFileSystem_Recycle(t *testing.T) {
+	fs := &FileSystem{
+		User:        &model.User{},
+		Policy:      &model.Policy{},
+		FileTarget:  []model.File{model.File{}},
+		DirTarget:   []model.Folder{model.Folder{}},
+		AfterUpload: []Hook{GenericAfterUpdate},
+	}
+	fs.Recycle()
+	newFS := getEmptyFS()
+	if fs != newFS {
+		t.Error("指针不一致")
+	}
+}

pkg/filesystem/remote/handler.go

@@ -38,7 +38,7 @@ func (handler Handler) Delete(ctx context.Context, files []string) ([]string, er
 
 // Thumb 获取文件缩略图
 func (handler Handler) Thumb(ctx context.Context, path string) (*response.ContentResponse, error) {
-	return nil, nil
+	return nil, errors.New("未实现")
 }
 
 // Source 获取外链URL