垫片校验

@@ -79,9 +79,16 @@ public class BackupMaterialServiceImpl extends ServiceImpl<BackupMaterialMapper,
     @Override
     public ResultBean upload(MultipartFile backupFile, BackupUploadVo backupUploadVo) {
         //校验
-        if(!"mp4".equalsIgnoreCase(backupUploadVo.getFileType())){
+//        if(!"mp4".equalsIgnoreCase(backupUploadVo.getFileType())){
+//            throw new ServiceException("仅支持上传mp4格式的垫片文件!");
+//        }
+        //校验backupFile后缀名
+        String realName = backupFile.getOriginalFilename();
+        String suffix = realName.substring(realName.lastIndexOf(".") + 1);
+        if (!suffix.equalsIgnoreCase("mp4")) {
             throw new ServiceException("仅支持上传mp4格式的垫片文件!");
         }
+
         //对应垫片任务
         LambdaQueryWrapper<StreamTask> backupWrapper = new LambdaQueryWrapper<>();
         backupWrapper.eq(StreamTask::getRoomId,backupUploadVo.getRoomId());