package com.alibaba.csp.sentinel.dashboard.rule.nacos.authority;
|
|
|
import com.alibaba.csp.sentinel.dashboard.constants.SentinelConStants;
|
import com.alibaba.csp.sentinel.dashboard.datasource.entity.rule.AuthorityRuleEntity;
|
import com.alibaba.csp.sentinel.dashboard.rule.DynamicRuleProvider;
|
import com.alibaba.csp.sentinel.dashboard.rule.nacos.entity.AuthorityRuleCorrectEntity;
|
import com.alibaba.csp.sentinel.datasource.Converter;
|
import com.alibaba.csp.sentinel.slots.block.authority.AuthorityRule;
|
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("authorityRuleNacosProvider")
|
public class AuthorityRuleNacosProvider implements DynamicRuleProvider<List<AuthorityRuleEntity>> {
|
@Autowired
|
private ConfigService configService;
|
@Autowired
|
private Converter<String, List<AuthorityRuleCorrectEntity>> converter;
|
|
@Override
|
public List<AuthorityRuleEntity> getRules(String appName) throws Exception {
|
String rules = configService.getConfig(appName + SentinelConStants.AUTHORITY_DATA_ID_POSTFIX,
|
SentinelConStants.GROUP_ID, 3000);
|
if (StringUtil.isEmpty(rules)) {
|
return new ArrayList<>();
|
}
|
List<AuthorityRuleCorrectEntity> entityList = converter.convert(rules);
|
return entityList.stream().map(rule -> {
|
AuthorityRule authorityRule = new AuthorityRule();
|
BeanUtils.copyProperties(rule, authorityRule);
|
AuthorityRuleEntity entity = AuthorityRuleEntity.fromAuthorityRule(rule.getApp(), rule.getIp(), rule.getPort(), authorityRule);
|
entity.setId(rule.getId());
|
entity.setGmtCreate(rule.getGmtCreate());
|
return entity;
|
}).collect(Collectors.toList());
|
}
|
}
|
