package com.alibaba.csp.sentinel.dashboard.rule.nacos.paramflow;
|
|
|
import com.alibaba.csp.sentinel.dashboard.constants.SentinelConStants;
|
import com.alibaba.csp.sentinel.dashboard.datasource.entity.rule.ParamFlowRuleEntity;
|
import com.alibaba.csp.sentinel.dashboard.rule.DynamicRuleProvider;
|
import com.alibaba.csp.sentinel.dashboard.rule.nacos.SentinelConfig;
|
import com.alibaba.csp.sentinel.dashboard.rule.nacos.entity.ParamFlowRuleCorrectEntity;
|
import com.alibaba.csp.sentinel.datasource.Converter;
|
import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowRule;
|
import com.alibaba.csp.sentinel.util.StringUtil;
|
import com.alibaba.nacos.api.config.ConfigService;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 加载热点参数规则
|
*
|
* @author zyf
|
* @date 2022-04-13
|
*/
|
@Component("paramFlowRuleNacosProvider")
|
public class ParamFlowRuleNacosProvider implements DynamicRuleProvider<List<ParamFlowRuleEntity>> {
|
|
@Autowired
|
private ConfigService configService;
|
@Autowired
|
private Converter<String, List<ParamFlowRuleCorrectEntity>> converter;
|
|
@Override
|
public List<ParamFlowRuleEntity> getRules(String appName) throws Exception {
|
String rules = configService.getConfig(appName + SentinelConStants.PARAM_FLOW_DATA_ID_POSTFIX,
|
SentinelConStants.GROUP_ID, 3000);
|
if (StringUtil.isEmpty(rules)) {
|
return new ArrayList<>();
|
}
|
List<ParamFlowRuleCorrectEntity> entityList = converter.convert(rules);
|
return entityList.stream().map(rule -> {
|
ParamFlowRule paramFlowRule = new ParamFlowRule();
|
BeanUtils.copyProperties(rule, paramFlowRule);
|
ParamFlowRuleEntity entity = ParamFlowRuleEntity.fromParamFlowRule(rule.getApp(), rule.getIp(), rule.getPort(), paramFlowRule);
|
entity.setId(rule.getId());
|
entity.setGmtCreate(rule.getGmtCreate());
|
return entity;
|
}).collect(Collectors.toList());
|
}
|
}
|
