From 959aae945779b5e46d5d362e371a0c3b746e2b9f Mon Sep 17 00:00:00 2001
From: luodangjia <luodangjia>
Date: 星期二, 21 一月 2025 19:51:10 +0800
Subject: [PATCH] 1
---
ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/CommonController.java | 68 ++++++++++++++++++++++++++++-----
1 files changed, 57 insertions(+), 11 deletions(-)
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/CommonController.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/CommonController.java
index 6353cd7..335d461 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/CommonController.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/CommonController.java
@@ -1,9 +1,12 @@
package com.ruoyi.system.controller;
+import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.common.core.domain.R;
+import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.utils.UrlDownloader;
import com.ruoyi.common.core.utils.orc.OcrUtils;
+import com.ruoyi.system.api.model.BusinessLicense;
import com.ruoyi.system.api.model.IdCard;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
@@ -13,7 +16,6 @@
import org.springframework.web.bind.annotation.RestController;
import java.io.InputStream;
-import java.net.URLEncoder;
@RestController
@RequestMapping("/common")
@@ -28,19 +30,63 @@
public R<IdCard> idCard(@Parameter(name = "url", description = "图片地址") String url) {
InputStream inputStream;
try {
- // 找到最后一个斜杠的位置
- int lastSlashIndex = url.lastIndexOf("/");
- // 从斜杠后提取文件名
- String baseUrl = url.substring(0, lastSlashIndex);
- String fileName = url.substring(lastSlashIndex + 1);
- String encode = URLEncoder.encode(fileName, "UTF-8");
- url = baseUrl + "/" + encode;
+
inputStream = UrlDownloader.downloadAsStream(url);
- JSONObject jsonObject = OcrUtils.idCard(inputStream);
- IdCard idCard = JSONObject.parseObject(jsonObject.toJSONString(), IdCard.class);
- return R.ok(idCard);
+ String jsonStr = OcrUtils.idCard(inputStream,"IdCard");
+ if (StringUtils.isEmpty(jsonStr)){
+ return R.fail();
+ }
+
+ JSONObject jsonObject = JSONObject.parseObject(jsonStr);
+ Integer statusCode = jsonObject.getInteger("statusCode");
+ if (statusCode != 200) {
+ return R.fail("识别失败");
+ }
+ JSONObject body = jsonObject.getJSONObject("body");
+ JSONObject data = body.getJSONObject("Data");
+ JSONArray subImages = data.getJSONArray("SubImages");
+ if (subImages != null && !subImages.isEmpty()){
+ JSONObject subImage = subImages.getJSONObject(0);
+ JSONObject kvInfo = subImage.getJSONObject("KvInfo");
+ IdCard idCard = kvInfo.getObject("Data", IdCard.class);
+ return R.ok(idCard);
+ }
} catch (Exception e) {
throw new RuntimeException(e);
}
+ return R.fail();
+ }
+
+ /**
+ * ocr提取营业执照信息
+ */
+ @GetMapping("/businessLicense")
+ @Operation(summary = "ocr提取营业执照信息", description = "ocr提取营业执照信息")
+ public R<BusinessLicense> businessLicense(@Parameter(name = "url", description = "图片地址") String url) {
+ InputStream inputStream;
+ try {
+ inputStream = UrlDownloader.downloadAsStream(url);
+ String jsonStr = OcrUtils.idCard(inputStream,"BusinessLicense");
+ if (StringUtils.isEmpty(jsonStr)){
+ return R.fail();
+ }
+ JSONObject jsonObject = JSONObject.parseObject(jsonStr);
+ Integer statusCode = jsonObject.getInteger("statusCode");
+ if (statusCode != 200) {
+ return R.fail("识别失败");
+ }
+ JSONObject body = jsonObject.getJSONObject("body");
+ JSONObject data = body.getJSONObject("Data");
+ JSONArray subImages = data.getJSONArray("SubImages");
+ if (subImages != null && !subImages.isEmpty()){
+ JSONObject subImage = subImages.getJSONObject(0);
+ JSONObject kvInfo = subImage.getJSONObject("KvInfo");
+ BusinessLicense businessLicense = kvInfo.getObject("Data", BusinessLicense.class);
+ return R.ok(businessLicense);
+ }
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ return R.fail();
}
}
--
Gitblit v1.7.1