当前位置：首页 > ops >正文

附件上传唯一性校验

ops 2025/7/17 18:51:07

    @Overridepublic String uploadFile(MultipartFile file, String id, String funNo, String ctType) {//TODO 附件重复判断// 计算文件哈希值// 将MultipartFile转换为临时File对象String fileHash = "";try {File tempFile = convertMultipartFileToFile(file);// 计算文件哈希值fileHash = calculateFileHash(tempFile);} catch (Exception e) {e.printStackTrace();}TbFile checkFileHash = new TbFile();checkFileHash.setFileHash(fileHash);List<TbFile> checkFileHashList = tbFileMapper.selectTbFileList(checkFileHash);if(checkFileHashList.size()>=1){throw new com.ruoyi.common.exception.CustomException("文件已存在，请勿重复上传！");}// 生成新的唯一文件名并保存文件String uniqueFileName = generateUniqueFileName(file.getName());SysUser user = SecurityUtils.getLoginUser().getUser();// 判断附件类型String extension = "." + FileUploadUtils.getExtension(file);if (!ALLOW_EXCEL_FORM.contains(extension.toLowerCase())) {throw new CustomException("请上传正确的文件类型");}UploadConfig uploadConfig = ApplicationContextUtils.getBean(UploadConfig.class);String shortPath = uploadConfig.getRootUploadPath() + "image/";String[] split = file.getOriginalFilename().split("\\.");String primaryKey = split[0];String wjtype = split[1];String filePath = shortPath + primaryKey + extension;// 临时文件File localTempFile = new File(shortPath + primaryKey + extension);if (!localTempFile.getParentFile().exists()) {localTempFile.getParentFile().mkdir();}if (!localTempFile.exists()) {Try.run(localTempFile::createNewFile).onFailure(e -> log.error("新增文件失败"));}Try.run(() -> file.transferTo(localTempFile)).onFailure(e -> log.error("上传失败"));if (!localTempFile.exists()) {throw new CustomException("上传失败，服务器环境创建文件服务存储的文件夹失败，请联系管理员处理！");}//单附件  先删除，在上传if (funNo.equals("yth_1000") || funNo.equals("yth_1001") || funNo.equals("yth_1003") || funNo.equals("yth_1005") || funNo.equals("yth yth_1006")) {TbFile checkFile = new TbFile();checkFile.setCtIdP(id);checkFile.setCtFunNo(funNo);checkFile.setCtType(ctType);List<TbFile> tbFilesList = tbFileMapper.selectTbFileList(checkFile);for (TbFile tbFile : tbFilesList) {int i = tbFileMapper.deleteTbFileByCtId(tbFile.getCtId());if (i < 1) {throw new CustomException("附件替换失败");}}}TbFile tbFile = new TbFile();tbFile.setCtFileName(primaryKey + "." + wjtype);tbFile.setCtIdP(id);tbFile.setCtFileType(primaryKey);tbFile.setCtFunNo(funNo);tbFile.setCtFileUel(filePath);tbFile.setCtType(ctType);tbFile.setBy5(DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD_HH_MM_SS, DateUtils.getNowDate()));//附件上传时间tbFile.setBy4(String.valueOf(user.getUserId()));//上传附件的用户idtodo  tbFile.setFileHash(fileHash);int i = tbFileMapper.insertTbFile(tbFile);if (i != 1) {throw new CustomException(FunNoConstants.AJK_AJQQ);}return "文件上传成功";}

    /*** 计算文件的MD5哈希值*/private String calculateFileHash(File file) throws Exception {MessageDigest digest = MessageDigest.getInstance("MD5");try (FileInputStream fis = new FileInputStream(file)) {byte[] buffer = new byte[8192];int read;while ((read = fis.read(buffer)) != -1) {digest.update(buffer, 0, read);}}byte[] hashBytes = digest.digest();StringBuilder sb = new StringBuilder();for (byte b : hashBytes) {sb.append(String.format("%02x", b));}return sb.toString();}/*** 生成唯一文件名*/private String generateUniqueFileName(String originalFileName) {// 实际应用中可能需要更复杂的逻辑// 这里简单地使用时间戳String extension = "";int dotIndex = originalFileName.lastIndexOf('.');if (dotIndex > 0) {extension = originalFileName.substring(dotIndex);}return System.currentTimeMillis() + extension;}/*** 将MultipartFile转换为File对象*/private File convertMultipartFileToFile(MultipartFile multipartFile) throws IOException {File tempFile = File.createTempFile("temp-", "-" + multipartFile.getOriginalFilename());try (FileOutputStream fos = new FileOutputStream(tempFile)) {fos.write(multipartFile.getBytes());}// 设置JVM退出时删除临时文件tempFile.deleteOnExit();return tempFile;}

3.数据库