package com.ruoyi.common.oss.factory; import com.ruoyi.common.core.constant.CacheNames; import com.ruoyi.common.core.utils.JsonUtils; import com.ruoyi.common.core.utils.StringUtils; import com.ruoyi.common.oss.constant.OssConstant; import com.ruoyi.common.oss.core.OssClient; import com.ruoyi.common.oss.exception.OssException; import com.ruoyi.common.oss.properties.OssProperties; import com.ruoyi.common.redis.utils.CacheUtils; import com.ruoyi.common.redis.utils.RedisUtils; import lombok.extern.slf4j.Slf4j; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; /** * 文件上传Factory * * @author Lion Li */ @Slf4j public class OssFactory { private static final Map CLIENT_CACHE = new ConcurrentHashMap<>(); /** * 初始化工厂 */ public static void init() { log.info("初始化OSS工厂"); RedisUtils.subscribe(OssConstant.DEFAULT_CONFIG_KEY, String.class, configKey -> { OssClient client = getClient(configKey); // 未初始化不处理 if (client != null) { refresh(configKey); log.info("订阅刷新OSS配置 => " + configKey); } }); } /** * 获取默认实例 */ public static OssClient instance() { // 获取redis 默认类型 String configKey = RedisUtils.getCacheObject(OssConstant.DEFAULT_CONFIG_KEY); if (StringUtils.isEmpty(configKey)) { throw new OssException("文件存储服务类型无法找到!"); } return instance(configKey); } /** * 根据类型获取实例 */ public static OssClient instance(String configKey) { OssClient client = getClient(configKey); if (client == null) { refresh(configKey); return getClient(configKey); } return client; } private static void refresh(String configKey) { String json = CacheUtils.get(CacheNames.SYS_OSS_CONFIG, configKey); if (json == null) { throw new OssException("系统异常, '" + configKey + "'配置信息不存在!"); } OssProperties properties = JsonUtils.parseObject(json, OssProperties.class); CLIENT_CACHE.put(configKey, new OssClient(configKey, properties)); } private static OssClient getClient(String configKey) { return CLIENT_CACHE.get(configKey); } }