疯狂的狮子li
2021-10-20 82f1f5d0cf1b51a5d81915e842e01760f404fa74
ruoyi-oss/src/main/java/com/ruoyi/oss/factory/OssFactory.java
@@ -1,18 +1,18 @@
package com.ruoyi.oss.factory;
import cn.hutool.core.convert.Convert;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.common.utils.JsonUtils;
import com.ruoyi.common.utils.RedisUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.reflect.ReflectUtils;
import com.ruoyi.common.utils.spring.SpringUtils;
import com.ruoyi.oss.constant.CloudConstant;
import com.ruoyi.oss.enumd.CloudServiceEnumd;
import com.ruoyi.oss.exception.OssException;
import com.ruoyi.oss.properties.CloudStorageProperties;
import com.ruoyi.oss.service.ICloudStorageStrategy;
import com.ruoyi.oss.service.abstractd.AbstractCloudStorageStrategy;
import lombok.extern.slf4j.Slf4j;
import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
@@ -21,12 +21,14 @@
 *
 * @author Lion Li
 */
@Slf4j
public class OssFactory {
   private static RedisCache redisCache;
   static {
      OssFactory.redisCache = SpringUtils.getBean(RedisCache.class);
      RedisUtils.subscribe(CloudConstant.CACHE_CONFIG_KEY, String.class, msg -> {
         refreshService(msg);
         log.info("订阅刷新OSS配置 => " + msg);
      });
   }
   /**
@@ -35,16 +37,11 @@
   private static final Map<String, ICloudStorageStrategy> SERVICES = new ConcurrentHashMap<>();
   /**
    * 服务配置更新时间缓存
    */
   private static final Map<String, Date> SERVICES_UPDATE_TIME = new ConcurrentHashMap<>();
   /**
    * 获取默认实例
    */
   public static ICloudStorageStrategy instance() {
      // 获取redis 默认类型
      String type = Convert.toStr(redisCache.getCacheObject(CloudConstant.CACHE_CONFIG_KEY));
      String type = Convert.toStr(RedisUtils.getCacheObject(CloudConstant.CACHE_CONFIG_KEY));
      if (StringUtils.isEmpty(type)) {
         throw new OssException("文件存储服务类型无法找到!");
      }
@@ -56,22 +53,23 @@
    */
   public static ICloudStorageStrategy instance(String type) {
      ICloudStorageStrategy service = SERVICES.get(type);
      Date oldDate = SERVICES_UPDATE_TIME.get(type);
      Object json = redisCache.getCacheObject(CloudConstant.SYS_OSS_KEY + type);
      if (service == null) {
         refreshService(type);
         service = SERVICES.get(type);
      }
      return service;
   }
   private static void refreshService(String type) {
      Object json = RedisUtils.getCacheObject(CloudConstant.SYS_OSS_KEY + type);
      CloudStorageProperties properties = JsonUtils.parseObject(json.toString(), CloudStorageProperties.class);
      if (properties == null) {
         throw new OssException("系统异常, '" + type + "'配置信息不存在!");
      }
      Date nowDate = properties.getUpdateTime();
      // 服务存在并更新时间相同则返回(使用更新时间确保配置最终一致性)
      if (service != null && oldDate.equals(nowDate)) {
         return service;
      }
      // 获取redis配置信息 创建对象 并缓存
      service = (ICloudStorageStrategy) ReflectUtils.newInstance(CloudServiceEnumd.getServiceClass(type), properties);
      ICloudStorageStrategy service = (ICloudStorageStrategy) ReflectUtils.newInstance(CloudServiceEnumd.getServiceClass(type));
      ((AbstractCloudStorageStrategy)service).init(properties);
      SERVICES.put(type, service);
      SERVICES_UPDATE_TIME.put(type, nowDate);
      return service;
   }
}