cdle
53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
package core
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
var statics sync.Map
|
|
|
|
func addStatic(uuid, path string) {
|
|
statics.Store(uuid, path)
|
|
}
|
|
|
|
func remStatic(uuid string) {
|
|
statics.Delete(uuid)
|
|
}
|
|
|
|
func FindFile(c *gin.Context) {
|
|
// 获取文件名
|
|
filename := c.Param("filename")
|
|
|
|
statics.Range(func(_, value any) bool {
|
|
path := value.(string)
|
|
// 拼接文件路径
|
|
filepath := filepath.Join(path, filename)
|
|
// 判断文件是否存在
|
|
_, err := os.Stat(filepath)
|
|
if err == nil {
|
|
// 文件存在，读取文件内容并返回
|
|
file, err := ioutil.ReadFile(filepath)
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, err)
|
|
return true
|
|
}
|
|
// 根据文件类型设置Content-Type
|
|
contentType := http.DetectContentType(file)
|
|
c.Header("Content-Type", contentType)
|
|
|
|
// 返回文件内容
|
|
c.Data(http.StatusOK, contentType, file)
|
|
return false
|
|
}
|
|
return true
|
|
})
|
|
// 如果文件不存在，返回404错误
|
|
c.AbortWithStatus(http.StatusNotFound)
|
|
}
|