ocr识别-同步至文档库

jero-boot/jero-boot-modules/src/main/java/com/jero/modules/ocr/controller/OcrRecordEOController.java

@@ -5,12 +5,19 @@ import java.util.List;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import cn.hutool.core.collection.CollectionUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.jero.common.api.vo.Result;
+import com.jero.modules.document.entity.BussDocumentLibraryEO;
+import com.jero.modules.document.service.IBussDocumentLibraryEOService;
 import com.jero.modules.ocr.entity.OcrRecordEO;
+import com.jero.modules.ocr.enums.FileTypeEnum;
+import com.jero.modules.ocr.enums.ResultContentEnum;
 import com.jero.modules.ocr.service.IOcrRecordEOService;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.jero.modules.oss.entity.OSSFile;
+import com.jero.modules.oss.service.IOSSFileService;
 import lombok.extern.slf4j.Slf4j;
 import com.jero.common.system.base.controller.JeroController;
 import org.apache.commons.lang.StringUtils;
@@ -37,6 +44,10 @@ import com.jero.common.aspect.annotation.AutoLog;
 public class OcrRecordEOController extends JeroController<OcrRecordEO, IOcrRecordEOService> {
 	@Autowired
 	private IOcrRecordEOService ocrRecordService;
+	@Autowired
+	private IBussDocumentLibraryEOService bussDocumentLibraryEOService;
+	@Autowired
+	private IOSSFileService ossFileService;
 
 	@Value("${OCR.ocrDownPath}")
 	private String ocrDownPath;
@@ -76,6 +87,35 @@ public class OcrRecordEOController extends JeroController<OcrRecordEO, IOcrRecor
 		return Result.OK(pageList);
 	}
 
+	 @AutoLog(value = "OCR识别转换记录表-同步至文档库")
+	 @ApiOperation(value="OCR识别转换记录表-同步至文档库", notes="OCR识别转换记录表-同步至文档库")
+	 @GetMapping(value = "/syncToDocument")
+	 public Result<?> syncToDocument(@RequestBody OcrRecordEO ocrRecordEO) {
+		 //通过编号和标题查询对应文档
+		 LambdaQueryWrapper<BussDocumentLibraryEO> queryWrapper = new LambdaQueryWrapper<>();
+		 queryWrapper.eq(BussDocumentLibraryEO::getTitle, ocrRecordEO.getStandName())
+				 .eq(BussDocumentLibraryEO::getSerialNumber, ocrRecordEO.getStandNumber());
+		 List<BussDocumentLibraryEO> bussDocumentLibraryEOList = bussDocumentLibraryEOService.list(queryWrapper);
+		 if(CollectionUtil.isNotEmpty(bussDocumentLibraryEOList)){
+			 OSSFile ossFile = new OSSFile();
+			 //设置doc文件的关联id
+			 if (FileTypeEnum.RELEASE_DRAFT.getValue().equals(ocrRecordEO.getFileType())){
+				 ossFile.setConnectId(bussDocumentLibraryEOList.get(0).getReleaseDraft());
+			 }else if (FileTypeEnum.SUPPLEMENT.getValue().equals(ocrRecordEO.getFileType())){
+				 ossFile.setConnectId(bussDocumentLibraryEOList.get(0).getSupplement());
+			 }else if (FileTypeEnum.APPROVAL_DRAFT.getValue().equals(ocrRecordEO.getFileType())){
+				 ossFile.setConnectId(bussDocumentLibraryEOList.get(0).getApprovalDraft());
+			 }else if (FileTypeEnum.EXPOSURE_DRAFT.getValue().equals(ocrRecordEO.getFileType())){
+				 ossFile.setConnectId(bussDocumentLibraryEOList.get(0).getExposureDraft());
+			 }
+			 ossFile.setId(ocrRecordEO.getAttId());
+			 ossFileService.updateById(ossFile);
+			 return Result.OK();
+		 }else{
+		 	return Result.error("没有可同步的文档");
+		 }
+	 }
+
     /**
 	 * 列表查询
 	 *

jero-boot/jero-boot-modules/src/main/java/com/jero/modules/ocr/enums/FileTypeEnum.java

@@ -6,9 +6,10 @@ package com.jero.modules.ocr.enums;
  * @Date: Created in 14:49 2022/2/16
  */
 public enum FileTypeEnum {
-    OCR_CONVERTING("转换中","转换中"),
-    OCR_CONVERT_SUCCESS("转换成功","转换成功"),
-    OCR_CONVERT_FAIL("转换失败","转换失败");
+    RELEASE_DRAFT("发布稿","1"),
+    SUPPLEMENT("增补件","2"),
+    APPROVAL_DRAFT("报批稿","3"),
+    EXPOSURE_DRAFT("征求意见稿","4");
 
     String name;
     String value;

jero-boot/jero-boot-modules/src/main/java/com/jero/modules/ocr/enums/ResultContentEnum.java

@@ -0,0 +1,36 @@
+package com.jero.modules.ocr.enums;
+
+/**
+ * @Author: liyawei
+ * @Description:
+ * @Date: Created in 14:49 2022/2/16
+ */
+public enum ResultContentEnum {
+    OCR_CONVERTING("转换中","转换中"),
+    OCR_CONVERT_SUCCESS("转换成功","转换成功"),
+    OCR_CONVERT_FAIL("转换失败","转换失败");
+
+    String name;
+    String value;
+
+    private ResultContentEnum(String name, String value) {
+        this.name = name;
+        this.value = value;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public String getValue() {
+        return value;
+    }
+
+    public void setValue(String value) {
+        this.value = value;
+    }
+}