From 6cbe7ba4a186fa3f4e38d383c106a253d1839dad Mon Sep 17 00:00:00 2001
From: vincentlu <t1341870251@gmail.com>
Date: 星期一, 13 一月 2025 14:48:59 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/system/service/impl/ConfigServiceImpl.java |   39 ++++++++++++++-----
 zy-acs-manager/src/main/java/com/zy/acs/manager/system/controller/ConfigController.java    |   24 ++++++++++++
 2 files changed, 53 insertions(+), 10 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/system/controller/ConfigController.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/system/controller/ConfigController.java
index b0eb20d..436f283 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/system/controller/ConfigController.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/system/controller/ConfigController.java
@@ -12,7 +12,9 @@
 import com.zy.acs.manager.common.domain.PageParam;
 import com.zy.acs.manager.common.utils.ExcelUtil;
 import com.zy.acs.manager.system.entity.Config;
+import com.zy.acs.manager.system.entity.User;
 import com.zy.acs.manager.system.service.ConfigService;
+import com.zy.acs.manager.system.service.impl.ConfigServiceImpl;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.*;
@@ -59,6 +61,10 @@
     @OperationLog("Create Config")
     @PostMapping("/config/save")
     public R save(@RequestBody Config config) {
+        if (!Cools.isEmpty(config.getFlag())
+                && 0 < configService.count(new LambdaQueryWrapper<Config>().eq(Config::getFlag, config.getFlag()))) {
+            return R.error("failed to save config, because the " + config.getFlag() + " flag already exist");
+        }
         config.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
         config.setCreateBy(getLoginUserId());
         config.setCreateTime(new Date());
@@ -66,6 +72,8 @@
         config.setUpdateTime(new Date());
         if (!configService.save(config)) {
             return R.error("Save Fail");
+        } else {
+            ConfigServiceImpl.CONFIG_CACHE.put(config.getFlag(), config);
         }
         return R.ok("Save Success").add(config);
     }
@@ -74,10 +82,17 @@
     @OperationLog("Update Config")
     @PostMapping("/config/update")
     public R update(@RequestBody Config config) {
+        if (!Cools.isEmpty(config.getFlag()) && configService.count(new LambdaQueryWrapper<Config>()
+                .eq(Config::getFlag, config.getFlag())
+                .ne(Config::getId, config.getId())) > 0) {
+            return R.error("the flag already exist");
+        }
         config.setUpdateBy(getLoginUserId());
         config.setUpdateTime(new Date());
         if (!configService.updateById(config)) {
             return R.error("Update Fail");
+        } else {
+            ConfigServiceImpl.CONFIG_CACHE.put(config.getFlag(), config);
         }
         return R.ok("Update Success").add(config);
     }
@@ -86,8 +101,17 @@
     @OperationLog("Delete Config")
     @PostMapping("/config/remove/{ids}")
     public R remove(@PathVariable Long[] ids) {
+        List<String> flagList = new ArrayList<>();
+        for (Long id : ids) {
+            Config config = configService.getById(id);
+            flagList.add(config.getFlag());
+        }
         if (!configService.removeByIds(Arrays.asList(ids))) {
             return R.error("Delete Fail");
+        } else {
+            for (String flag : flagList) {
+                ConfigServiceImpl.CONFIG_CACHE.remove(flag);
+            }
         }
         return R.ok("Delete Success").add(ids);
     }
diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/system/service/impl/ConfigServiceImpl.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/system/service/impl/ConfigServiceImpl.java
index 2f65707..d20cd4c 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/system/service/impl/ConfigServiceImpl.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/system/service/impl/ConfigServiceImpl.java
@@ -4,14 +4,16 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.zy.acs.common.utils.GsonUtils;
 import com.zy.acs.framework.common.DateUtils;
+import com.zy.acs.manager.manager.enums.StatusType;
 import com.zy.acs.manager.system.entity.Config;
 import com.zy.acs.manager.system.enums.ConfigType;
 import com.zy.acs.manager.system.mapper.ConfigMapper;
 import com.zy.acs.manager.system.service.ConfigService;
 import org.springframework.stereotype.Service;
 
-import java.util.Date;
-import java.util.List;
+import javax.annotation.PostConstruct;
+import java.util.*;
+import java.util.concurrent.ConcurrentHashMap;
 
 /**
  * Created by vincent on 8/30/2024
@@ -19,18 +21,31 @@
 @Service("configService")
 public class ConfigServiceImpl extends ServiceImpl<ConfigMapper, Config> implements ConfigService {
 
+    public static final Map<String, Config> CONFIG_CACHE = new ConcurrentHashMap<>();
+
+    @PostConstruct
+    public void init() {
+        List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getStatus, StatusType.ENABLE.val));
+        for (Config config : list) {
+            CONFIG_CACHE.put(config.getFlag(), config);
+        }
+    }
+
     @Override
     @SuppressWarnings("unchecked")
     public <T> T getVal(String key, Class<T> clazz) {
-        List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getFlag, key));
-        Config config = list.stream().findFirst().orElse(null);
-        if (null == config) {
-            return null;
+        Config config = CONFIG_CACHE.get(key);
+        if (config == null) {
+            List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getFlag, key));
+            config = list.stream().findFirst().orElse(null);
+            if (null == config) {
+                return null;
+            }
         }
         String val = config.getVal();
         switch (ConfigType.query(config.getType())) {
             case BOOLEAN:
-                if (val.equals("1") || val.trim().toUpperCase().equals("TRUE")) {
+                if (val.equals("1") || val.trim().equalsIgnoreCase("TRUE")) {
                     return (T) Boolean.TRUE;
                 }
                 return (T) Boolean.FALSE;
@@ -58,11 +73,15 @@
 
     @Override
     public <T> boolean setVal(String key, T val) {
-        List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getFlag, key));
-        Config config = list.stream().findFirst().orElse(null);
+        Config config = CONFIG_CACHE.get(key);
         if (config == null) {
-            return false;
+            List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getFlag, key));
+            config = list.stream().findFirst().orElse(null);
+            if (null == config) {
+                return false;
+            }
         }
+
         ConfigType configType = ConfigType.query(config.getType());
         switch (configType) {
             case BOOLEAN:

--
Gitblit v1.9.1