From 7d61aba7b69fe1462efa6fb2bc7a8012f35c2838 Mon Sep 17 00:00:00 2001 From: wangwh <2397901735@qq.com> Date: Wed, 11 Sep 2024 14:00:07 +0800 Subject: [PATCH] update --- dyh-service/dyh-mediate/src/main/java/cn/huge/module/ai/controller/AiController.java | 21 +++++++++++++++++++-- 1 files changed, 19 insertions(+), 2 deletions(-) diff --git a/dyh-service/dyh-mediate/src/main/java/cn/huge/module/ai/controller/AiController.java b/dyh-service/dyh-mediate/src/main/java/cn/huge/module/ai/controller/AiController.java index c15986c..2348397 100644 --- a/dyh-service/dyh-mediate/src/main/java/cn/huge/module/ai/controller/AiController.java +++ b/dyh-service/dyh-mediate/src/main/java/cn/huge/module/ai/controller/AiController.java @@ -1,6 +1,8 @@ package cn.huge.module.ai.controller; +import cn.huge.base.common.utils.GuavaCacheUtils; import cn.huge.base.common.utils.HttpClientUtils; +import cn.huge.base.common.utils.ObjectUtils; import cn.huge.base.common.utils.ReturnSucUtils; import cn.huge.module.ai.controller.domain.dto.CaseInfoDetailDto; import cn.huge.module.ai.controller.domain.dto.CaseInfoDto; @@ -32,12 +34,16 @@ @PostMapping("/get-case") public Object getCase(@RequestBody AiRequestVo aiRequestVo) { + Object cacheAi = GuavaCacheUtils.getCacheAi("get-case"+aiRequestVo.getCaseId()); + if(ObjectUtils.isNotEmpty(cacheAi)){ + return ReturnSucUtils.getRepInfo("请求成功", cacheAi); + } List<CaseInfoDto> caseInfoDtoList = new ArrayList<>(); Map<String, String> params = new HashMap<>(); params.put("caseDes", aiRequestVo.getCaseDes());//案件描述 params.put("caseClaim", aiRequestVo.getCaseClaim());//案件诉求 params.put("caseId", aiRequestVo.getCaseId());//案件id - params.put("topK", "3"); + params.put("topK", "2"); params.put("metricType", "IP"); String s = HttpClientUtils.httpPostForm(aiUrl + "/queryConciliationCase", params, new HashMap<>(), "utf-8"); JSONObject object = JSONObject.parseObject(s); @@ -53,7 +59,7 @@ caseInfoDtoList.add(caseInfoDto); } } - if (caseInfoDtoList.size() < 6) { + if (caseInfoDtoList.size() < 4) { s = HttpClientUtils.httpPostForm(aiUrl + "/queryJudgmentCase", params, new HashMap<>(), "utf-8"); object = JSONObject.parseObject(s); code = object.getIntValue("code"); @@ -71,12 +77,17 @@ } } } + GuavaCacheUtils.putCacheAi("get-case"+aiRequestVo.getCaseId(), caseInfoDtoList); return ReturnSucUtils.getRepInfo("请求成功", caseInfoDtoList); } @PostMapping("/get-law") public Object getLaw(@RequestBody AiRequestVo aiRequestVo) { + Object cacheAi = GuavaCacheUtils.getCacheAi("get-law"+aiRequestVo.getCaseId()); + if(ObjectUtils.isNotEmpty(cacheAi)){ + return ReturnSucUtils.getRepInfo("请求成功", cacheAi); + } List<LawInfoDto> lawInfoDtos = new ArrayList<>(); Map<String, String> params = new HashMap<>(); params.put("caseDes", aiRequestVo.getCaseDes());//案件描述 @@ -97,6 +108,7 @@ lawInfoDtos.add(lawInfoDto); } } + GuavaCacheUtils.putCacheAi("get-law"+aiRequestVo.getCaseId(), lawInfoDtos); return ReturnSucUtils.getRepInfo("请求成功", lawInfoDtos); } @@ -134,6 +146,10 @@ @PostMapping("/get-clain-rise-case") public Object getClaimRiskByCase(@RequestBody AiRequestVo aiRequestVo) { + Object cacheAi = GuavaCacheUtils.getCacheAi("get-clain-rise-case"+aiRequestVo.getCaseId()); + if(ObjectUtils.isNotEmpty(cacheAi)){ + return ReturnSucUtils.getRepInfo("请求成功", cacheAi); + } CaseRiskDto caseRiskDto = new CaseRiskDto(); Map<String, String> params = new HashMap<>(); params.put("caseDes", aiRequestVo.getCaseDes());//案件描述 @@ -147,6 +163,7 @@ caseRiskDto.setIsClaimRisk(data.getIntValue("is_claim_risk")); caseRiskDto.setAnalysisProcess(data.getString("analysis_process")); } + GuavaCacheUtils.putCacheAi("get-clain-rise-case"+aiRequestVo.getCaseId(), caseRiskDto); return ReturnSucUtils.getRepInfo("请求成功", caseRiskDto); } -- Gitblit v1.8.0