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); } 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) { Config config = CONFIG_CACHE.get(key); if (config == null) { List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getFlag, key)); Config config = list.stream().findFirst().orElse(null); 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) { 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: