package com.dobbinsoft.fw.support.component.dynamic;

import com.alibaba.fastjson.JSONObject;
import com.dobbinsoft.fw.support.component.CacheComponent;
import com.dobbinsoft.fw.support.component.LockComponent;
import com.dobbinsoft.fw.support.constant.CacheConst;
import com.dobbinsoft.fw.support.constant.LockConst;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.function.Function;

/**
 * Description: 动态配置组件
 * User: rize
 * Date: 2020/8/5
 * Time: 11:31
 */
public class DynamicConfigComponent implements InitializingBean {

    @Autowired
    private DynamicStorageStrategy dynamicStorageStrategy;

    @Autowired
    private CacheComponent cacheComponent;

    @Autowired
    private LockComponent lockComponent;

    @Value("${com.dobbinsoft.fw.support.dynamic.clear-on-startup:F}")
    private String clearCacheOnStartup;

    private static final Logger logger = LoggerFactory.getLogger(DynamicConfigComponent.class);

    @Override
    public void afterPropertiesSet() throws Exception {
        if ("T".equals(clearCacheOnStartup)) {
            // 启动时清理缓存
            cacheComponent.delPrefixKey(CacheConst.DYNAMIC_CACHE);
            logger.info("[动态配置] 缓存清理成功！");
        }
    }

    /**
     * 写动态配置
     *
     * @param key
     * @param value
     */
    public void write(String key, String value) {
        // 防止重复提交
        if (lockComponent.tryLock(LockConst.DYNAMIC_CONFIG_LOCK + key, 15)) {
            try {
                dynamicStorageStrategy.write(key, value);
                // 由于是无事务单条写SQL，此处已经完成持久化
                cacheComponent.del(CacheConst.DYNAMIC_CACHE + key);
            } catch (Exception e) {
                logger.error("[写动态配置] 异常", e);
            } finally {
                lockComponent.release(LockConst.DYNAMIC_CONFIG_LOCK + key);
            }
        }
    }

    public Integer readInt(String key, Integer defaultValue) {
        return catchNumberFormatException(this.readAction(key, defaultValue, Integer::parseInt), defaultValue);
    }

    public Long readLong(String key, Long defaultValue) {
        return catchNumberFormatException(this.readAction(key, defaultValue, Long::parseLong), defaultValue);
    }

    public String readString(String key, String defaultValue) {
        return this.readAction(key, defaultValue, item->item);
    }

    public Boolean readBoolean(String key, Boolean defaultValue) {
        return this.readAction(key, defaultValue, Boolean::parseBoolean);
    }

    public <T> T readObj(String key, Class<T> clazz) {
        return this.readAction(key, null, item-> JSONObject.parseObject(item, clazz));
    }

    /**
     * 读取行为的统一封装
     * @param key
     * @param defaultValue
     * @param function 相当于是个返序列化的方法
     * @param <T>
     * @return
     */
    public <T> T readAction(String key, T defaultValue, Function<String, T> function) {
        String raw = cacheComponent.getRaw(CacheConst.DYNAMIC_CACHE + key);
        if (!StringUtils.isEmpty(raw)) {
            return function.apply(raw);
        }
        String value = dynamicStorageStrategy.read(key);
        if (value == null) {
            return defaultValue;
        }
        // 放入缓存
        cacheComponent.putRaw(CacheConst.DYNAMIC_CACHE + key, value);
        return function.apply(value);
    }

    /**
     * 捕获调数字格式化
     * @param value
     * @param defaultValue
     * @param <T>
     * @return
     */
    private <T extends Number> T catchNumberFormatException(T value, T defaultValue) {
        try {
            return value;
        } catch (NumberFormatException e) {
            return defaultValue;
        }
    }

}