From c40d07e760598c0aae4353f7a1e2d6b01747e83f Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 12 六月 2024 11:54:54 +0800
Subject: [PATCH] 文件下载&状态筛选
---
controllers/audio.go | 45 +++++++++++++++++++++++++++++++++++++++++----
1 files changed, 41 insertions(+), 4 deletions(-)
diff --git a/controllers/audio.go b/controllers/audio.go
index 5d035d8..7c58dd9 100644
--- a/controllers/audio.go
+++ b/controllers/audio.go
@@ -2,8 +2,11 @@
import (
"errors"
+ "fmt"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
+ "io"
+ "os"
"path"
"speechAnalysis/constvar"
"speechAnalysis/extend/code"
@@ -181,6 +184,7 @@
SetBeginTime(params.BeginTime).
SetEndTime(params.EndTime).
SetIsFollowed(params.IsFollowed).
+ SetAudioStatusList(params.StatusList).
Find()
if err != nil {
@@ -266,10 +270,27 @@
return
}
- c.Header("Content-Description", "File Transfer")
- c.Header("Content-Disposition", "attachment; filename="+audio.Name)
- c.Header("Content-Type", "application/octet-stream")
- c.File(audio.FilePath)
+ file, err := os.Open(audio.FilePath)
+ if err != nil {
+ util.ResponseFormat(c, code.InternalError, "鏂囦欢鎵撳紑澶辫触")
+ return
+ }
+ defer file.Close()
+
+ fileInfo, err := file.Stat()
+ if err != nil {
+ util.ResponseFormat(c, code.InternalError, "鑾峰彇鏂囦欢淇℃伅澶辫触")
+ return
+ }
+
+ c.Header("Content-Disposition", "inline; filename="+audio.Name) // 鍦ㄦ祻瑙堝櫒涓洿鎺ユ墦寮�
+ c.Header("Content-Length", fmt.Sprint(fileInfo.Size()))
+ c.Header("Content-Type", "audio/mpeg") // 璁剧疆闊抽鏂囦欢绫诲瀷
+
+ if _, err := io.Copy(c.Writer, file); err != nil {
+ util.ResponseFormat(c, code.InternalError, "鏂囦欢浼犺緭澶辫触")
+ return
+ }
}
// BatchProcess
@@ -339,6 +360,13 @@
return
}
+ go func() {
+ err = os.Remove(audio.FilePath)
+ if err != nil {
+ logx.Warnf("remove file err:%v, file:%v", err, audio.FilePath)
+ }
+ }()
+
util.ResponseFormat(c, code.DeleteSuccess, "鎴愬姛")
}
@@ -375,6 +403,15 @@
return
}
+ go func() {
+ for _, audio := range audioList {
+ err = os.Remove(audio.FilePath)
+ if err != nil {
+ logx.Warnf("remove file err:%v, file:%v", err, audio.FilePath)
+ }
+ }
+ }()
+
util.ResponseFormat(c, code.DeleteSuccess, "鎴愬姛")
}
--
Gitblit v1.8.0