From d190f46ff2073213f668c2bd8a55debbaf3b7d25 Mon Sep 17 00:00:00 2001 From: tjxt <1325917080@qq.com> Date: Tue, 30 Jul 2024 11:26:49 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9--=E5=A2=9E=E5=8A=A0log?= =?UTF-8?q?=E7=9B=B8=E5=85=B3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/bonus/ai/controller/OcrRecogController.java | 4 +++- .../bonus/ai/service/impl/OcrRecogServiceImpl.java | 13 +++++++++++-- .../bonus-ai/src/main/resources/logback.xml | 2 +- 3 files changed, 15 insertions(+), 4 deletions(-) diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/OcrRecogController.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/OcrRecogController.java index 2bf34aa..20e2226 100644 --- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/OcrRecogController.java +++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/OcrRecogController.java @@ -3,10 +3,10 @@ package com.bonus.ai.controller; import com.bonus.ai.service.IOcrRecogService; import com.bonus.ai.utils.FileTypeUtils; import com.bonus.common.core.web.domain.AjaxResult; +import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; - import static com.bonus.common.core.web.domain.AjaxResult.error; /** @@ -16,6 +16,7 @@ import static com.bonus.common.core.web.domain.AjaxResult.error; */ @RestController @RequestMapping("/ocr") +@Slf4j public class OcrRecogController { @Autowired @@ -40,6 +41,7 @@ public class OcrRecogController } } catch (Exception e) { e.printStackTrace(); + log.error("数据上传失败:{}",e); return error("数据上传失败！！！"); } diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/impl/OcrRecogServiceImpl.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/impl/OcrRecogServiceImpl.java index e7a8dc8..b5da114 100644 --- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/impl/OcrRecogServiceImpl.java +++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/impl/OcrRecogServiceImpl.java @@ -20,6 +20,8 @@ import org.springframework.stereotype.Service; import org.springframework.web.client.RestClientException; import org.springframework.web.client.RestTemplate; import org.springframework.web.multipart.MultipartFile; + + import java.io.*; import java.nio.file.*; import java.time.LocalDateTime; @@ -59,8 +61,10 @@ public class OcrRecogServiceImpl implements IOcrRecogService { try { getfilePath = convertFilesToBase64(files); idCardVo = processPythonResponse(getfilePath, recogurl); + log.info("获得data:{}",idCardVo); ajax.put("data",idCardVo); } catch (Exception e) { + log.error("调用大模型失败:{}",e); ajax.put("601", "调用大模型服务出错"); e.printStackTrace(); } @@ -73,7 +77,6 @@ public class OcrRecogServiceImpl implements IOcrRecogService { String currentDate = System.currentTimeMillis()+""; String frontFileName = String.format(FRONT_FILE_NAME_TEMPLATE, idNumber, currentDate); String backFileName = String.format(BACK_FILE_NAME_TEMPLATE, idNumber, currentDate); - try { byte[] frontImageBytes = Base64.getDecoder().decode(getfilePath[0]); byte[] backImageBytes = Base64.getDecoder().decode(getfilePath[1]); @@ -82,12 +85,14 @@ public class OcrRecogServiceImpl implements IOcrRecogService { Files.write(frontFilePath, frontImageBytes); Files.write(backFilePath, backImageBytes); } catch (IOException e) { + log.error("获取身份证信息出错:{}",e); ajax.put("602", "获取身份证信息出错"); e.printStackTrace(); } saveRecognitionResult(idCardVo, ifComplete, frontFileName, backFileName,startTime); } + log.info("服务层返回给控制层的数据：{}",ajax); return ajax; } @@ -103,11 +108,13 @@ public class OcrRecogServiceImpl implements IOcrRecogService { e.printStackTrace(); } } + log.info("base64格式的图片:{}",base64Strings); return base64Strings; } private IdCardVo processPythonResponse(String[] strings, String recogurl) throws Exception { if (strings.length != 2) { + log.info("Python响应数据不完整。"); throw new IllegalArgumentException("Python响应数据不完整。"); } ImageUpload imageUpload = new ImageUpload(); @@ -116,6 +123,7 @@ public class OcrRecogServiceImpl implements IOcrRecogService { imageUpload.setType("1"); ResponseEntity response = restTemplate.postForEntity(recogurl, imageUpload, String.class); if (response.getStatusCodeValue() != 200) { + log.info("调用服务失败，状态码：{}",response.getStatusCodeValue()); throw new RestClientException("调用服务失败，状态码：" + response.getStatusCodeValue()); } String json = response.getBody(); @@ -125,6 +133,7 @@ public class OcrRecogServiceImpl implements IOcrRecogService { json = StringEscapeUtils.unescapeJava(json); Gson gson = new Gson(); IdCardVo idCardVo = gson.fromJson(json, IdCardVo.class);//此方法 如果里面为空 错误 解决 + log.info("获取python服务得到的数据：{}",idCardVo); return idCardVo; } private void saveRecognitionResult(IdCardVo idCardVo, char ifComplete, String frontFileName, String backFileName,long startTime) { @@ -152,7 +161,7 @@ public class OcrRecogServiceImpl implements IOcrRecogService { .recognizeTime(LocalDateTime.now()) .responseLong(responseTime) .invokeIp(IpUtils.getIpAddr(ServletUtils.getRequest())) - // .updateBy(loginUser.getUserid()) // 假设getUserid()已返回正确用户ID +// .updateBy(loginUser.getUserid()) // 假设getUserid()已返回正确用户ID .updateTime(LocalDateTime.now()) .build(); aiIdcardrecognizeMapper.insert(aiIdcardrecognizeResult); diff --git a/bonus-modules/bonus-ai/src/main/resources/logback.xml b/bonus-modules/bonus-ai/src/main/resources/logback.xml index 7b9c29d..7c7bf48 100644 --- a/bonus-modules/bonus-ai/src/main/resources/logback.xml +++ b/bonus-modules/bonus-ai/src/main/resources/logback.xml @@ -1,7 +1,7 @@ - +