SpringBoot读取/上传Nacos上的Sentinel流量规则

其中引入的JAR包有

<dependencies>

    <dependency>
      <groupId>com.alibaba.csp</groupId>
      <artifactId>sentinel-datasource-nacos</artifactId>
    </dependency>

    <dependency>
      <groupId>com.steven</groupId>
      <artifactId>solomon-base</artifactId>
      <version>1.0</version>
    </dependency>

    <dependency>
      <groupId>com.alibaba.csp</groupId>
      <artifactId>sentinel-spring-cloud-gateway-adapter</artifactId>
    </dependency>

    <dependency>
      <groupId>com.alibaba.cloud</groupId>
      <artifactId>spring-cloud-starter-alibaba-sentinel</artifactId>
    </dependency>

    <dependency>
      <groupId>org.springframework.cloud</groupId>
      <artifactId>spring-cloud-starter-loadbalancer</artifactId>
      <version>${spring.loadbalancer.version}</version>
    </dependency>

    <dependency>
      <groupId>org.springframework.cloud</groupId>
      <artifactId>spring-cloud-starter-gateway</artifactId>
      <version>${Spring.cloud.gateway.version}</version>
    </dependency>

    <dependency>
      <groupId>org.apache.dubbo</groupId>
      <artifactId>dubbo-spring-boot-starter</artifactId>
    </dependency>
  </dependencies>

第一步创建Nacos的ConfigService

private ConfigService getConfigService() throws NacosException {
    Properties properties = new Properties();
    properties.setProperty(PropertyKeyConst.SERVER_ADDR,url);
    if(ValidateUtils.isNotEmpty(userName)){
      properties.setProperty(PropertyKeyConst.USERNAME,userName);
    }
    if(ValidateUtils.isNotEmpty(password)){
      properties.setProperty(PropertyKeyConst.PASSWORD,password);
    }
    if(ValidateUtils.isNotEmpty(namespace)){
      properties.setProperty(PropertyKeyConst.NAMESPACE,namespace);
    }
    return ConfigFactory.createConfigService(properties);
  }

其中URL就是nacos地址，userName就是nacos账号，password就是密码，namespace就是命名空间

第二步就是整个方法的实现

public class NacosSentinelWritableUtils {
  
  private String url;
  
  private String userName;
  
  private String password;
  
  private String namespace;

  private ConfigService configService;

  private ConfigService getConfigService() throws NacosException {
    Properties properties = new Properties();
    properties.setProperty(PropertyKeyConst.SERVER_ADDR,url);
    if(ValidateUtils.isNotEmpty(userName)){
      properties.setProperty(PropertyKeyConst.USERNAME,userName);
    }
    if(ValidateUtils.isNotEmpty(password)){
      properties.setProperty(PropertyKeyConst.PASSWORD,password);
    }
    if(ValidateUtils.isNotEmpty(namespace)){
      properties.setProperty(PropertyKeyConst.NAMESPACE,namespace);
    }
    return ConfigFactory.createConfigService(properties);
  }
  
  public NacosSentinelWritableUtils(String url,String userName,String password,String namespace) throws NacosException {
    this.url = url;
    this.userName = userName;
    this.password = password;
    this.namespace = namespace;
    this.configService = getConfigService();
  }

  /**
   * 提交sentinel流量规则
   * @param dataId nacos中的dataId
   * @param groupId nacos中的groupId
   * @param obj 提交的数据（需要全量）
   * @param type nacos配置类型
   * @return
   * @throws Exception
   */
  public void publishSentinelFlowRule(String dataId,String groupId,List<FlowRule> obj, ConfigType type) throws Exception {

    configService.publishConfig(dataId,groupId, JackJsonUtils.formatJsonByFilter(obj),type.getType());

    SentinelProperty<List<FlowRule>> sentinelProperty = getSentinelFlowRuleDataSource(groupId,dataId).getProperty();
    FlowRuleManager.register2Property(sentinelProperty);
  }

  /**
   * 提交sentinel熔断规则
   * @param dataId nacos中的dataId
   * @param groupId nacos中的groupId
   * @param obj 提交的数据（需要全量）
   * @param type nacos配置类型
   * @return
   * @throws Exception
   */
  public void publishSentinelDegradeRule(String dataId,String groupId,List<DegradeRule> obj, ConfigType type) throws Exception {

    configService.publishConfig(dataId,groupId, JackJsonUtils.formatJsonByFilter(obj),type.getType());

    SentinelProperty<List<DegradeRule>> sentinelProperty = getSentinelDegradeRuleDataSource(groupId,dataId).getProperty();
    DegradeRuleManager.register2Property(sentinelProperty);
  }

  /**
   * 提交sentinel热点规则
   * @param dataId nacos中的dataId
   * @param groupId nacos中的groupId
   * @param obj 提交的数据（需要全量）
   * @param type nacos配置类型
   * @return
   * @throws Exception
   */
  public void publishSentinelParamFlowRule(String dataId,String groupId,List<ParamFlowRule> obj, ConfigType type) throws Exception {

    configService.publishConfig(dataId,groupId, JackJsonUtils.formatJsonByFilter(obj),type.getType());

    SentinelProperty<List<ParamFlowRule>> sentinelProperty = getSentinelParamFlowRuleDataSource(groupId,dataId).getProperty();
    ParamFlowRuleManager.register2Property(sentinelProperty);
  }

  /**
   * 提交sentinel系统规则
   * @param dataId nacos中的dataId
   * @param groupId nacos中的groupId
   * @param obj 提交的数据（需要全量）
   * @param type nacos配置类型
   * @return
   * @throws Exception
   */
  public void publishSentinelSystemRule(String dataId,String groupId,List<SystemRule> obj, ConfigType type) throws Exception {

    configService.publishConfig(dataId,groupId, JackJsonUtils.formatJsonByFilter(obj),type.getType());

    SentinelProperty<List<SystemRule>> sentinelProperty = getSentinelSystemRuleDataSource(groupId,dataId).getProperty();
    SystemRuleManager.register2Property(sentinelProperty);
  }

  /**
   * 获取sentinel配置
   */
  public NacosDataSource getSentinelFlowRuleDataSource(String groupId,String dataId) throws Exception {
    return new NacosDataSource<>(url, groupId, dataId, source -> JSON
        .parseObject(source, new TypeReference<List<FlowRule>>() {}));
  }

  /**
   * 获取sentinel配置
   */
  public List<FlowRule> getSentinelFlowRule(String groupId,String dataId) throws Exception {
    return (List<FlowRule>) getSentinelFlowRuleDataSource(groupId,dataId).loadConfig();
  }

  /**
   * 获取sentinel配置
   */
  public NacosDataSource getSentinelDegradeRuleDataSource(String groupId,String dataId) throws Exception {
    return new NacosDataSource<>(url, groupId, dataId, source -> JSON
        .parseObject(source, new TypeReference<List<DegradeRule>>() {}));
  }

  /**
   * 获取sentinel配置
   */
  public List<DegradeRule> getSentinelDegradeRule(String groupId,String dataId) throws Exception {
    return (List<DegradeRule>) getSentinelDegradeRuleDataSource(groupId,dataId).loadConfig();
  }

  /**
   * 获取sentinel配置
   */
  public NacosDataSource getSentinelParamFlowRuleDataSource(String groupId,String dataId) throws Exception {
    return new NacosDataSource<>(url, groupId, dataId, source -> JSON
        .parseObject(source, new TypeReference<List<ParamFlowRule>>() {}));
  }

  /**
   * 获取sentinel配置
   */
  public List<ParamFlowRule> getSentinelParamFlowRule(String groupId,String dataId) throws Exception {
    return (List<ParamFlowRule>) getSentinelParamFlowRuleDataSource(groupId,dataId).loadConfig();
  }

  /**
   * 获取sentinel配置
   */
  public NacosDataSource getSentinelSystemRuleDataSource(String groupId,String dataId) throws Exception {
    return new NacosDataSource<>(url, groupId, dataId, source -> JSON
        .parseObject(source, new TypeReference<List<SystemRule>>() {}));
  }

  /**
   * 获取sentinel配置
   */
  public List<SystemRule> getSentinelSystemRule(String groupId,String dataId) throws Exception {
    return (List<SystemRule>) getSentinelSystemRuleDataSource(groupId,dataId).loadConfig();
  }
}

里面的JackJsonUtils其实就是自己封装的JSON工具类，每个人都可以按照自己项目风格使用不同的JSON封装工具