#

vincentlu

1 天以前 f7543a1cc6c89ea03c0d6a0591526baf10d7934b

 zy-acs-manager/src/main/java/com/zy/acs/manager/system/service/impl/ConfigServiceImpl.java

@@ -1,11 +1,23 @@
package com.zy.acs.manager.system.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.acs.common.utils.GsonUtils;
import com.zy.acs.framework.common.Cools;
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 com.zy.acs.framework.exception.CoolException;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Created by vincent on 8/30/2024
@@ -13,9 +25,137 @@
@Service("configService")
public class ConfigServiceImpl extends ServiceImpl<ConfigMapper, Config> implements ConfigService {

    @Override
    public String getVal(String key) {
        throw new CoolException("Config还没做");
    public static final Map<String, Config> CONFIG_CACHE = new ConcurrentHashMap<>();

    @PostConstruct
    public void init() {
        CONFIG_CACHE.clear();

        List<Config> list = this.list(new LambdaQueryWrapper<Config>().eq(Config::getStatus, StatusType.ENABLE.val));
        for (Config config : list) {
            if (null != config && !Cools.isEmpty(config.getFlag())) {
                CONFIG_CACHE.put(config.getFlag(), config);
            }
        }
    }

}
    @Override
    @SuppressWarnings("unchecked")
    public <T> T getVal(String key, Class<T> clazz) {
        if (Cools.isEmpty(key.trim(), clazz)) {
            return null;
        }

        Config config = CONFIG_CACHE.get(key);
        if (config == null) {
            config = getOne(new LambdaQueryWrapper<Config>()
                    .eq(Config::getFlag, key)
                    .eq(Config::getStatus, StatusType.ENABLE.val), false);

            if (null == config) {
                return null;
            }

            if (!Cools.isEmpty(config.getFlag())) {
                CONFIG_CACHE.put(config.getFlag(), config);
            }
        }

        String raw = config.getVal();
        String val = Cools.isEmpty(raw) ? "" : raw.trim();
        ConfigType type = ConfigType.query(config.getType());
        if (null == type) {
            return null;
        }
        switch (type) {
            case BOOLEAN: {
                if ("1".equals(val) || "true".equalsIgnoreCase(val) || "yes".equalsIgnoreCase(val)) {
                    return (T) Boolean.TRUE;
                }
                return (T) Boolean.FALSE;
            }
            case NUMBER: {
                if (clazz == Integer.class || clazz == int.class) {
                    return (T) Integer.valueOf(val);
                } else if (clazz == Short.class || clazz == short.class) {
                    return (T) Short.valueOf(val);
                } else if (clazz == Long.class || clazz == long.class) {
                    return (T) Long.valueOf(val);
                } else if (clazz == Double.class || clazz == double.class) {
                    return (T) Double.valueOf(val);
                } else if (clazz == Float.class || clazz == float.class) {
                    return (T) Float.valueOf(val);
                } else if (clazz == BigDecimal.class) {
                    return (T) new BigDecimal(val);
                }
                throw new UnsupportedOperationException("Unsupported number type: " + clazz.getName());
            }
            case STRING:
                return (T) raw;
            case JSON:
                return GsonUtils.fromJson(raw, clazz);
            case DATE:
                return (T) DateUtils.convert(val);
            default:
                return null;
        }
    }

    @Override
    public <T> boolean setVal(String key, T val) {
        if (key == null || key.trim().isEmpty()) {
            return false;
        }

        Config config = CONFIG_CACHE.get(key);
        if (config == null) {
            config = getOne(new LambdaQueryWrapper<Config>()
                    .eq(Config::getFlag, key)
                    .eq(Config::getStatus, StatusType.ENABLE.val), false);
            if (config == null) {
                return false;
            }
        } else {
            if (config.getStatus() == null || config.getStatus() != StatusType.ENABLE.val) {
                return false;
            }
        }

        ConfigType configType = ConfigType.query(config.getType());
        switch (configType) {
            case BOOLEAN:
                if (!(val instanceof Boolean)) {
                    throw new IllegalArgumentException("Expected Boolean value for key: " + key);
                }
                config.setVal(((Boolean) val) ? "TRUE" : "FALSE");
                break;
            case NUMBER:
                if (val instanceof Integer || val instanceof Short || val instanceof Long
                        || val instanceof Double || val instanceof Float || val instanceof BigDecimal) {
                    config.setVal(String.valueOf(val));
                } else {
                    throw new IllegalArgumentException("Expected a numeric value for key: " + key);
                }
                break;
            case STRING:
                if (!(val instanceof String)) {
                    throw new IllegalArgumentException("Expected a String value for key: " + key);
                }
                config.setVal((String) val);
                break;
            case JSON:
                config.setVal(GsonUtils.toJson(val));
                break;
            case DATE:
                if (!(val instanceof Date)) {
                    throw new IllegalArgumentException("Expected a Date value for key: " + key);
                }
                config.setVal(DateUtils.convert((Date) val));
                break;
            default:
                throw new UnsupportedOperationException("Unsupported ConfigType: " + configType);
        }

        return this.updateById(config);
    }
}