177 lines
8.6 KiB
Java
177 lines
8.6 KiB
Java
package com.ai.da.controller;
|
||
|
||
import com.ai.da.common.response.Response;
|
||
import com.ai.da.model.dto.*;
|
||
import com.ai.da.model.dto.ContestantDTO;
|
||
import com.ai.da.model.vo.CheckOTPVO;
|
||
import com.ai.da.service.GlobalAwardService;
|
||
import com.ai.da.service.upload.UploadService;
|
||
import com.ai.da.service.upload.UploadTask;
|
||
import io.swagger.annotations.Api;
|
||
import io.swagger.annotations.ApiOperation;
|
||
import io.swagger.annotations.ApiParam;
|
||
import jakarta.annotation.Resource;
|
||
import org.springframework.web.bind.annotation.*;
|
||
import org.springframework.web.multipart.MultipartFile;
|
||
|
||
import java.util.Map;
|
||
|
||
@RestController
|
||
@RequestMapping("/api/global-award")
|
||
@Api(tags = "全球奖项API", description = "全球奖项大赛管理和文件上传")
|
||
public class GlobalAwardController {
|
||
|
||
@Resource
|
||
private GlobalAwardService globalAwardService;
|
||
|
||
@Resource
|
||
private UploadService uploadService;
|
||
|
||
// @PostMapping("/uploads/pdf")
|
||
// public Response<String> uploadPdf(@RequestParam("file") MultipartFile file,
|
||
// @RequestParam(value = "email", required = false) String email) throws Exception {
|
||
// return Response.success(globalAwardService.uploadPdf(file, email));
|
||
// }
|
||
//
|
||
// @PostMapping("/uploads/video")
|
||
// public Response<String> uploadVideo(@RequestParam("file") MultipartFile file,
|
||
// @RequestParam(value = "email", required = false) String email) throws Exception {
|
||
// return Response.success(globalAwardService.uploadVideo(file, email));
|
||
// }
|
||
|
||
// ===== 新增分片上传接口 =====
|
||
|
||
// ===== PDF分片上传接口 =====
|
||
|
||
/** 初始化PDF上传任务 */
|
||
@PostMapping("/uploads/pdf/init")
|
||
@ApiOperation(value = "初始化PDF上传", notes = "创建新的PDF上传任务并返回上传参数")
|
||
public Response<UploadInitResponse> initPdfUpload(@ApiParam(value = "PDF上传初始化请求", required = true) @RequestBody UploadInitRequest request) {
|
||
UploadTask uploadTask = uploadService.initPdfUpload(request);
|
||
return Response.success(UploadInitResponse.builder()
|
||
.uploadId(uploadTask.getUploadId())
|
||
.chunkSize(uploadTask.getChunkSize())
|
||
.totalChunks(uploadTask.getTotalChunks())
|
||
.expiresAt(uploadTask.getExpiresAt())
|
||
.build());
|
||
}
|
||
|
||
/** 上传PDF分片 */
|
||
@PostMapping("/uploads/pdf/chunk")
|
||
@ApiOperation(value = "上传PDF分片", notes = "上传PDF文件的单个分片")
|
||
public Response<UploadChunkResponse> uploadPdfChunk(
|
||
@ApiParam(value = "PDF文件分片", required = true) @RequestParam("chunk") MultipartFile chunk,
|
||
@ApiParam(value = "上传任务ID", required = true) @RequestParam("uploadId") String uploadId,
|
||
@ApiParam(value = "分片索引(从0开始)", required = true) @RequestParam("chunkIndex") int chunkIndex,
|
||
@ApiParam(value = "分片总数", required = true) @RequestParam("totalChunks") int totalChunks) {
|
||
|
||
UploadChunkResponse uploadChunkResponse = uploadService.uploadPdfChunk(uploadId, chunk, chunkIndex, totalChunks);
|
||
return Response.success(uploadChunkResponse);
|
||
}
|
||
|
||
/** 完成PDF上传 */
|
||
@PostMapping("/uploads/pdf/complete")
|
||
@ApiOperation(value = "完成PDF上传", notes = "完成PDF上传并合并所有分片")
|
||
public Response<UploadCompleteResponse> completePdfUpload(@ApiParam(value = "PDF上传完成请求", required = true) @RequestBody UploadCompleteRequest request) {
|
||
UploadCompleteResponse uploadCompleteResponse = uploadService.completePdfUpload(
|
||
request.getUploadId(),
|
||
request.getFileName(),
|
||
request.getTotalSize(),
|
||
request.getEmail(),
|
||
request.getSecureToken());
|
||
return Response.success(uploadCompleteResponse);
|
||
}
|
||
|
||
/** 查询PDF上传状态 */
|
||
@GetMapping("/uploads/pdf/status/{uploadId}")
|
||
@ApiOperation(value = "查询PDF上传状态", notes = "获取PDF上传任务的当前状态")
|
||
public Response<UploadStatusResponse> getPdfUploadStatus(@ApiParam(value = "上传任务ID", required = true) @PathVariable String uploadId) {
|
||
UploadStatusResponse pdfUploadStatus = uploadService.getPdfUploadStatus(uploadId);
|
||
return Response.success(pdfUploadStatus);
|
||
}
|
||
|
||
// ===== 视频分片上传接口 =====
|
||
|
||
/** 初始化视频上传任务 */
|
||
@PostMapping("/uploads/video/init")
|
||
@ApiOperation(value = "初始化视频上传", notes = "创建新的视频上传任务并返回上传参数")
|
||
public Response<UploadInitResponse> initVideoUpload(@ApiParam(value = "视频上传初始化请求", required = true) @RequestBody UploadInitRequest request) {
|
||
UploadTask uploadTask = uploadService.initVideoUpload(request);
|
||
return Response.success(UploadInitResponse.builder()
|
||
.uploadId(uploadTask.getUploadId())
|
||
.chunkSize(uploadTask.getChunkSize())
|
||
.totalChunks(uploadTask.getTotalChunks())
|
||
.expiresAt(uploadTask.getExpiresAt())
|
||
.build());
|
||
}
|
||
|
||
/** 上传视频分片 */
|
||
@PostMapping("/uploads/video/chunk")
|
||
@ApiOperation(value = "上传视频分片", notes = "上传视频文件的单个分片")
|
||
public Response<UploadChunkResponse> uploadVideoChunk(
|
||
@ApiParam(value = "视频文件分片", required = true) @RequestParam("chunk") MultipartFile chunk,
|
||
@ApiParam(value = "上传任务ID", required = true) @RequestParam("uploadId") String uploadId,
|
||
@ApiParam(value = "分片索引(从0开始)", required = true) @RequestParam("chunkIndex") int chunkIndex,
|
||
@ApiParam(value = "分片总数", required = true) @RequestParam("totalChunks") int totalChunks) {
|
||
|
||
UploadChunkResponse uploadChunkResponse = uploadService.uploadVideoChunk(uploadId, chunk, chunkIndex, totalChunks);
|
||
return Response.success(uploadChunkResponse);
|
||
}
|
||
|
||
/** 完成视频上传 */
|
||
@PostMapping("/uploads/video/complete")
|
||
@ApiOperation(value = "完成视频上传", notes = "完成视频上传并合并所有分片")
|
||
public Response<UploadCompleteResponse> completeVideoUpload(@ApiParam(value = "视频上传完成请求", required = true) @RequestBody UploadCompleteRequest request) {
|
||
UploadCompleteResponse uploadCompleteResponse = uploadService.completeVideoUpload(
|
||
request.getUploadId(),
|
||
request.getFileName(),
|
||
request.getTotalSize(),
|
||
request.getEmail(),
|
||
request.getSecureToken());
|
||
return Response.success(uploadCompleteResponse);
|
||
}
|
||
|
||
/** 查询视频上传状态 */
|
||
@GetMapping("/uploads/video/status/{uploadId}")
|
||
@ApiOperation(value = "查询视频上传状态", notes = "获取视频上传任务的当前状态")
|
||
public Response<UploadStatusResponse> getVideoUploadStatus(@ApiParam(value = "上传任务ID", required = true) @PathVariable String uploadId) {
|
||
UploadStatusResponse videoUploadStatus = uploadService.getVideoUploadStatus(uploadId);
|
||
return Response.success(videoUploadStatus);
|
||
}
|
||
|
||
@PostMapping("/contestants/save")
|
||
@ApiOperation(value = "保存参赛者信息", notes = "保存或更新参赛者信息及已上传的文件")
|
||
public Response<Map<String,Object>> submit(@ApiParam(value = "参赛者信息", required = true) @RequestBody ContestantDTO request) {
|
||
return Response.success(globalAwardService.saveContestant(request));
|
||
}
|
||
|
||
@GetMapping("/contestants/{id}")
|
||
@ApiOperation(value = "根据id查询参赛者", notes = "根据id获取参赛者信息")
|
||
public Response<ContestantDTO> getContestantByID(@ApiParam(value = "参赛者id", required = true) @PathVariable("id") String id) {
|
||
ContestantDTO dto = globalAwardService.getContestantByID(id);
|
||
return Response.success(dto);
|
||
}
|
||
|
||
@GetMapping("/checkEmail")
|
||
public Response<String> checkEmail(@RequestParam("email") String email) {
|
||
globalAwardService.checkEmail(email);
|
||
return Response.success();
|
||
}
|
||
|
||
@GetMapping("/checkCode")
|
||
public Response<CheckOTPVO> checkCode(@RequestParam("email") String email, @RequestParam("code") String code) {
|
||
return Response.success(globalAwardService.checkCode(email, code));
|
||
}
|
||
|
||
@GetMapping("/contestants/export")
|
||
@ApiOperation(value = "导出参赛者列表为Excel", notes = "导出所有参赛者信息为xlsx并触发下载")
|
||
public Response<Void> exportContestants() throws Exception {
|
||
// 将文件保存到服务端本地目录（uploadDir/exports），不返回文件内容给客户端
|
||
globalAwardService.saveContestantsToLocal();
|
||
return Response.success();
|
||
}
|
||
|
||
}
|
||
|
||
|