FengChaoYu 3 месяцев назад
Родитель
Сommit
daa9df6d7e

+ 20 - 0
mall-miniapp-service/src/main/java/com/gree/mall/miniapp/logic/policy/WorkerLogic.java

@@ -3,6 +3,7 @@ package com.gree.mall.miniapp.logic.policy;
 import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.io.FileUtil;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.toolkit.IdWorker;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@@ -27,11 +28,13 @@ import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
+import org.springframework.util.FileCopyUtils;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletRequest;
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
+import java.io.File;
 import java.io.InputStream;
 import java.math.BigDecimal;
 import java.util.ArrayList;
@@ -630,6 +633,23 @@ public class WorkerLogic {
         }
         inputStream.close();
         baos.close();
+        String path = System.getProperty("user.dir");
+
+        // 生成的文件的存放路径
+        String fileSuffix = signFile.getName().substring(signFile.getName().lastIndexOf("."));
+
+        String mkdir = path + "/tmp/ocr";
+        if (!FileUtil.exist(path + "/tmp/ocr")) {
+            FileUtil.mkdir(path + "/tmp/ocr");
+        }
+
+        // 创建文件
+        File uploadFile = new File(mkdir + IdWorker.getIdStr() + fileSuffix);
+        uploadFile.createNewFile();
+
+        // 将文件写入磁盘
+        FileCopyUtils.copy(signFile.getBytes(), uploadFile);
+
         final String name = ocrUtil.handwritingORC(new ByteArrayInputStream(baos.toByteArray()));
         log.info("OCR识别名称:" + name + ", 师傅名称:" + user.getNickName());
         if (!StrUtil.equals(name, user.getNickName().replace(" ", ""))) {