crp / crp-operation

Go to a project
Toggle navigation Toggle navigation pinning
wanghongbo
1b02e171 1 parent 2fee3911

垫片校验

Inline Side-by-side
Showing 1 changed file with 8 additions and 1 deletions
@@ -79,9 +79,16 @@ public class BackupMaterialServiceImpl extends ServiceImpl<BackupMaterialMapper, @@ -79,9 +79,16 @@ public class BackupMaterialServiceImpl extends ServiceImpl<BackupMaterialMapper,
79 @Override 79 @Override
80 public ResultBean upload(MultipartFile backupFile, BackupUploadVo backupUploadVo) { 80 public ResultBean upload(MultipartFile backupFile, BackupUploadVo backupUploadVo) {
81 //校验 81 //校验
82 - if(!"mp4".equalsIgnoreCase(backupUploadVo.getFileType())){ 82 +// if(!"mp4".equalsIgnoreCase(backupUploadVo.getFileType())){
  83 +// throw new ServiceException("仅支持上传mp4格式的垫片文件!");
  84 +// }
  85 + //校验backupFile后缀名
  86 + String realName = backupFile.getOriginalFilename();
  87 + String suffix = realName.substring(realName.lastIndexOf(".") + 1);
  88 + if (!suffix.equalsIgnoreCase("mp4")) {
83 throw new ServiceException("仅支持上传mp4格式的垫片文件!"); 89 throw new ServiceException("仅支持上传mp4格式的垫片文件!");
84 } 90 }
  91 +
85 //对应垫片任务 92 //对应垫片任务
86 LambdaQueryWrapper<StreamTask> backupWrapper = new LambdaQueryWrapper<>(); 93 LambdaQueryWrapper<StreamTask> backupWrapper = new LambdaQueryWrapper<>();
87 backupWrapper.eq(StreamTask::getRoomId,backupUploadVo.getRoomId()); 94 backupWrapper.eq(StreamTask::getRoomId,backupUploadVo.getRoomId());