This commit is contained in:
2026-08-27 11:04:56 +08:00
parent 0ca8f4c3ab
commit bb7bac3dee
30 changed files with 171 additions and 260 deletions
+2 -17
View File
@@ -2,8 +2,6 @@ package controller
import (
"context"
"fmt"
"net/url"
"github.com/gogf/gf/v2/net/ghttp"
@@ -119,13 +117,14 @@ func (c *cAdmin) UploadCover(ctx context.Context, req *dto.AdminDatasetCoverUplo
return service.Dataset.AdminUploadCover(ctx, req)
}
// Cover 封面访问(直写响应体)
// Cover 封面访问(直写响应体;URL 不随封面更换变化,必须 no-store 防浏览器缓存旧图
func (c *cAdmin) Cover(ctx context.Context, req *dto.AdminDatasetCoverReq) (*dto.AdminDatasetCoverRes, error) {
path, err := service.Dataset.CoverFile(ctx, req.DatasetId)
if err != nil {
return nil, err
}
r := ghttp.RequestFromCtx(ctx)
r.Response.Header().Set("Cache-Control", "no-store")
r.Response.ServeFile(path)
return &dto.AdminDatasetCoverRes{}, nil
}
@@ -135,20 +134,6 @@ func (c *cAdmin) DeleteCover(ctx context.Context, req *dto.AdminDatasetCoverDele
return service.Dataset.AdminDeleteCover(ctx, req)
}
// ExportDataset 导出数据集 zip(直写响应体下载)
func (c *cAdmin) ExportDataset(ctx context.Context, req *dto.AdminDatasetExportReq) (*dto.AdminDatasetExportRes, error) {
data, filename, err := service.Dataset.ExportZip(ctx, req.DatasetId)
if err != nil {
return nil, err
}
r := ghttp.RequestFromCtx(ctx)
r.Response.Header().Set("Content-Disposition",
fmt.Sprintf("attachment; filename*=UTF-8''%s", url.PathEscape(filename)))
r.Response.Header().Set("Content-Type", "application/zip")
r.Response.Write(data)
return &dto.AdminDatasetExportRes{}, nil
}
// ---------- 训练编排 ----------
// ListTrainings 训练任务列表