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.DynamicRulePublisher;
|
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.util.AssertUtil;
|
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.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 持久化热点参数规则
|
*
|
* @author zyf
|
* @date 2022-04-13
|
*/
|
@Component("paramFlowRuleNacosPublisher")
|
public class ParamFlowRuleNacosPublisher implements DynamicRulePublisher<List<ParamFlowRuleEntity>> {
|
|
@Autowired
|
private ConfigService configService;
|
@Autowired
|
private Converter<List<ParamFlowRuleCorrectEntity>, String> converter;
|
|
@Override
|
public void publish(String app, List<ParamFlowRuleEntity> rules) throws Exception {
|
AssertUtil.notEmpty(app, "app name cannot be empty");
|
if (rules == null) {
|
return;
|
}
|
rules.forEach(e -> e.setApp(app));
|
|
// 转换
|
List<ParamFlowRuleCorrectEntity> list = rules.stream().map(rule -> {
|
ParamFlowRuleCorrectEntity entity = new ParamFlowRuleCorrectEntity();
|
BeanUtils.copyProperties(rule, entity);
|
return entity;
|
}).collect(Collectors.toList());
|
|
configService.publishConfig(app + SentinelConStants.PARAM_FLOW_DATA_ID_POSTFIX,
|
SentinelConStants.GROUP_ID, converter.convert(list));
|
|
}
|
}
|
