From 20dc892fd72e5e9c19a5a5188f7daf9bcd87e0f0 Mon Sep 17 00:00:00 2001 From: houzhongjian <houzhongyi@126.com> Date: 星期五, 22 十一月 2024 09:57:41 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java | 24 ++++++++++++++++++++++++ 1 files changed, 24 insertions(+), 0 deletions(-) diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java index 2922b37..892bae3 100644 --- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java +++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java @@ -12,9 +12,11 @@ import org.springframework.util.CollectionUtils; import java.math.BigDecimal; +import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.UUID; +import java.util.concurrent.ConcurrentHashMap; /** * @author PanZhibao @@ -25,15 +27,36 @@ @Autowired private MmPredictModelDao mmPredictModelDao; + + private static Map<String, MmPredictModelEntity> modelEntityMap = new ConcurrentHashMap<>(); + @Override public void savePredictModel(MmPredictModelEntity predictModel) { predictModel.setId(UUID.randomUUID().toString()); mmPredictModelDao.insert(predictModel); + clearCache(); } @Override public void update(MmPredictModelEntity predictModel) { this.updateById(predictModel); + clearCache(); + } + + @Override + public MmPredictModelEntity getInfoFromCatch(String id) { + if (!modelEntityMap.containsKey(id)) { + MmPredictModelEntity modelEntity = getInfo(id); + if (modelEntity != null) { + modelEntityMap.put(id, modelEntity); + } + } + return modelEntityMap.get(id); + } + + @Override + public void clearCache() { + modelEntityMap.clear(); } @Override @@ -57,6 +80,7 @@ QueryWrapper<MmPredictModelEntity> queryWrapper = new QueryWrapper<>(); queryWrapper.in("itemid", itemIds); mmPredictModelDao.delete(queryWrapper); + clearCache(); } public MmPredictModelEntity getByItemid(String itemid) { -- Gitblit v1.9.3