1
This commit is contained in:
@@ -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 训练任务列表
|
||||
|
||||
Reference in New Issue
Block a user