🐎 加入缓存

server/controllers/get.go

@@ -0,0 +1,70 @@
+package controllers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/alidrive"
+	"github.com/Xhofe/alist/conf"
+	"github.com/gin-gonic/gin"
+	"github.com/patrickmn/go-cache"
+	log "github.com/sirupsen/logrus"
+	"strings"
+)
+
+func Get(c *gin.Context) {
+	var get alidrive.GetReq
+	if err := c.ShouldBindJSON(&get); err != nil {
+		c.JSON(200,metaResponse(400,"Bad Request"))
+		return
+	}
+	log.Debugf("get:%v",get)
+	// cache
+	cacheKey:=fmt.Sprintf("%s-%s","g",get.FileId)
+	if conf.Conf.Cache.Enable {
+		file,exist:=conf.Cache.Get(cacheKey)
+		if exist {
+			log.Debugf("使用了缓存:%s",cacheKey)
+			c.JSON(200,dataResponse(file))
+			return
+		}
+	}
+	file,err:=alidrive.GetFile(get.FileId)
+	if err !=nil {
+		c.JSON(200,metaResponse(500,err.Error()))
+		return
+	}
+	paths,err:=alidrive.GetPaths(get.FileId)
+	if err!=nil {
+		c.JSON(200,metaResponse(500,err.Error()))
+		return
+	}
+	file.Paths=*paths
+	if conf.Conf.Cache.Enable {
+		conf.Cache.Set(cacheKey,file,cache.DefaultExpiration)
+	}
+	c.JSON(200,dataResponse(file))
+}
+
+func Down(c *gin.Context) {
+	fileIdParam:=c.Param("file_id")
+	log.Debugf("down:%s",fileIdParam)
+	fileId:=strings.Split(fileIdParam,"/")[1]
+	cacheKey:=fmt.Sprintf("%s-%s","d",fileId)
+	if conf.Conf.Cache.Enable {
+		downloadUrl,exist:=conf.Cache.Get(cacheKey)
+		if exist {
+			log.Debugf("使用了缓存:%s",cacheKey)
+			c.Redirect(301,downloadUrl.(string))
+			return
+		}
+	}
+	file,err:=alidrive.GetFile(fileId)
+	if err != nil {
+		c.JSON(200, metaResponse(500,err.Error()))
+		return
+	}
+	if conf.Conf.Cache.Enable {
+		conf.Cache.Set(cacheKey,file.DownloadUrl,cache.DefaultExpiration)
+	}
+	c.Redirect(301,file.DownloadUrl)
+	return
+}