feat: 实现多部分上传功能,支持初始化、上传部分、完成和中止上传,添加媒体资产删除功能

This commit is contained in:
2026-01-04 15:20:06 +08:00
parent 2ab1238ef7
commit 2438d363f5
9 changed files with 454 additions and 21 deletions

View File

@@ -66,3 +66,77 @@ func (c *Common) GetOptions(ctx fiber.Ctx) (*dto.OptionsResponse, error) {
func (c *Common) CheckHash(ctx fiber.Ctx, user *models.User, hash string) (*dto.UploadResult, error) {
return services.Common.CheckHash(ctx, user.ID, hash)
}
// @Router /v1/upload/init [post]
// @Summary Init multipart upload
// @Description Initialize multipart upload
// @Tags Common
// @Accept json
// @Produce json
// @Param form body dto.UploadInitForm true "Init form"
// @Success 200 {object} dto.UploadInitResponse
// @Bind user local key(__ctx_user)
// @Bind form body
func (c *Common) InitUpload(ctx fiber.Ctx, user *models.User, form *dto.UploadInitForm) (*dto.UploadInitResponse, error) {
return services.Common.InitUpload(ctx.Context(), user.ID, form)
}
// @Router /v1/upload/part [post]
// @Summary Upload part
// @Description Upload a part
// @Tags Common
// @Accept multipart/form-data
// @Produce json
// @Param file formData file true "Part File"
// @Param form formData dto.UploadPartForm true "Part form"
// @Success 200 {string} string "OK"
// @Bind user local key(__ctx_user)
// @Bind file file
// @Bind form body
func (c *Common) UploadPart(ctx fiber.Ctx, user *models.User, file *multipart.FileHeader, form *dto.UploadPartForm) error {
return services.Common.UploadPart(ctx.Context(), user.ID, file, form)
}
// @Router /v1/upload/complete [post]
// @Summary Complete upload
// @Description Complete multipart upload
// @Tags Common
// @Accept json
// @Produce json
// @Param form body dto.UploadCompleteForm true "Complete form"
// @Success 200 {object} dto.UploadResult
// @Bind user local key(__ctx_user)
// @Bind form body
func (c *Common) CompleteUpload(ctx fiber.Ctx, user *models.User, form *dto.UploadCompleteForm) (*dto.UploadResult, error) {
return services.Common.CompleteUpload(ctx.Context(), user.ID, form)
}
// @Router /v1/upload/:uploadId [delete]
// @Summary Abort upload
// @Description Abort multipart upload
// @Tags Common
// @Accept json
// @Produce json
// @Param uploadId path string true "Upload ID"
// @Success 200 {string} string "OK"
// @Bind user local key(__ctx_user)
// @Bind uploadId path
func (c *Common) AbortUpload(ctx fiber.Ctx, user *models.User, uploadId string) error {
return services.Common.AbortUpload(ctx.Context(), user.ID, uploadId)
}
// @Router /v1/media-assets/:id [delete]
// @Summary Delete media asset
// @Description Delete media asset
// @Tags Common
// @Accept json
// @Produce json
// @Param id path string true "Asset ID"
// @Success 200 {string} string "OK"
// @Bind user local key(__ctx_user)
// @Bind id path
func (c *Common) DeleteMediaAsset(ctx fiber.Ctx, user *models.User, id string) error {
return services.Common.DeleteMediaAsset(ctx.Context(), user.ID, id)
}
// Upload file

View File

@@ -0,0 +1,24 @@
package dto
type UploadInitForm struct {
Hash string `json:"hash"`
Size int64 `json:"size"`
Filename string `json:"filename"`
MimeType string `json:"mime_type"`
Type string `json:"type"`
}
type UploadInitResponse struct {
UploadID string `json:"upload_id"`
Key string `json:"key"` // For S3 direct
ChunkSize int64 `json:"chunk_size"`
}
type UploadPartForm struct {
UploadID string `form:"upload_id"`
PartNumber int `form:"part_number"`
}
type UploadCompleteForm struct {
UploadID string `json:"upload_id"`
}

View File

@@ -60,6 +60,37 @@ func (r *Routes) Register(router fiber.Router) {
Local[*models.User]("__ctx_user"),
QueryParam[string]("hash"),
))
r.log.Debugf("Registering route: Post /v1/upload/init -> common.InitUpload")
router.Post("/v1/upload/init"[len(r.Path()):], DataFunc2(
r.common.InitUpload,
Local[*models.User]("__ctx_user"),
Body[dto.UploadInitForm]("form"),
))
r.log.Debugf("Registering route: Post /v1/upload/part -> common.UploadPart")
router.Post("/v1/upload/part"[len(r.Path()):], Func3(
r.common.UploadPart,
Local[*models.User]("__ctx_user"),
File[multipart.FileHeader]("file"),
Body[dto.UploadPartForm]("form"),
))
r.log.Debugf("Registering route: Post /v1/upload/complete -> common.CompleteUpload")
router.Post("/v1/upload/complete"[len(r.Path()):], DataFunc2(
r.common.CompleteUpload,
Local[*models.User]("__ctx_user"),
Body[dto.UploadCompleteForm]("form"),
))
r.log.Debugf("Registering route: Delete /v1/upload/:uploadId -> common.AbortUpload")
router.Delete("/v1/upload/:uploadId"[len(r.Path()):], Func2(
r.common.AbortUpload,
Local[*models.User]("__ctx_user"),
PathParam[string]("uploadId"),
))
r.log.Debugf("Registering route: Delete /v1/media-assets/:id -> common.DeleteMediaAsset")
router.Delete("/v1/media-assets/:id"[len(r.Path()):], Func2(
r.common.DeleteMediaAsset,
Local[*models.User]("__ctx_user"),
PathParam[string]("id"),
))
r.log.Debugf("Registering route: Post /v1/upload -> common.Upload")
router.Post("/v1/upload"[len(r.Path()):], DataFunc3(
r.common.Upload,