188 lines
9.5 KiB
Java
188 lines
9.5 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 jakarta.servlet.http.HttpServletResponse;
|
|
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 void exportContestants(HttpServletResponse response) throws Exception {
|
|
byte[] data = globalAwardService.exportContestants();
|
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
|
response.setHeader("Content-Disposition", "attachment; filename=\"contestants.xlsx\"");
|
|
response.setContentLength(data.length);
|
|
response.getOutputStream().write(data);
|
|
response.getOutputStream().flush();
|
|
}
|
|
|
|
@PostMapping("/contestants/export/files")
|
|
@ApiOperation(value = "导出参赛者文件到本地", notes = "根据参赛者编号范围导出PDF和视频文件到本地temp/uploads/contestants目录")
|
|
public Response<Integer> exportContestantFiles(@ApiParam(value = "参赛者文件导出请求", required = true) @RequestBody ContestantExportRequest request) throws Exception {
|
|
int exportedCount = globalAwardService.exportContestantFiles(request.getMinContestantNumber(), request.getMaxContestantNumber());
|
|
return Response.success(exportedCount);
|
|
}
|
|
|
|
}
|
|
|
|
|