mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-03-30 04:03:18 +00:00
23 lines
514 B
Go
23 lines
514 B
Go
package middleware
|
||
|
||
import (
|
||
"github.com/QuantumNous/new-api/common"
|
||
"github.com/QuantumNous/new-api/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// BodyStorageCleanup 请求体存储清理中间件
|
||
// 在请求处理完成后自动清理磁盘/内存缓存
|
||
func BodyStorageCleanup() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
// 处理请求
|
||
c.Next()
|
||
|
||
// 请求结束后清理存储
|
||
common.CleanupBodyStorage(c)
|
||
|
||
// 清理文件缓存（URL 下载的文件等）
|
||
service.CleanupFileSources(c)
|
||
}
|
||
}
|