Sentinel规则持久化

在这里插入图片描述

首先 Sentinel 控制台通过 API 将规则推送至客户端并更新到内存中,接着注册的写数据源会将新的规则保存到本地的文件中。

示例代码:

1.编写处理类

//规则持久化
public class FilePersistence implements InitFunc {
	@Value("spring.application:name")
	private String appcationName;
	
	@Override
	public void init() throws Exception {
	String ruleDir = System.getProperty("user.home") + "/sentinelrules/"+appcationName;
	String flowRulePath = ruleDir + "/flow-rule.json";
	String degradeRulePath = ruleDir + "/degrade-rule.json";
	String systemRulePath = ruleDir + "/system-rule.json";
	String authorityRulePath = ruleDir + "/authority-rule.json";
	String paramFlowRulePath = ruleDir + "/param-flow-rule.json";
	this.mkdirIfNotExits(ruleDir);
	this.createFileIfNotExits(flowRulePath);
	this.createFileIfNotExits(degradeRulePath);
	this.createFileIfNotExits(systemRulePath);
	this.createFileIfNotExits(authorityRulePath);
	this.createFileIfNotExits(paramFlowRulePath);
	
	// 流控规则
	ReadableDataSource<String, List<FlowRule>> flowRuleRDS = new FileRefreshableDataSource<>(
		flowRulePath,
		flowRuleListParser
	);
	FlowRuleManager.register2Property(flowRuleRDS.getProperty());
	WritableDataSource<List<FlowRule>> flowRuleWDS = new FileWritableDataSource<>(
		flowRulePath,
		this::encodeJson
	);
	WritableDataSourceRegistry.registerFlowDataSource(flowRuleWDS);
	
	// 降级规则
	ReadableDataSource<String, List<DegradeRule>> degradeRuleRDS = new FileRefreshableDataSource<>(
		degradeRulePath,
		degradeRuleListParser
	);
	DegradeRuleManager.register2Property(degradeRuleRDS.getProperty());
	WritableDataSource<List<DegradeRule>> degradeRuleWDS = new FileWritableDataSource<>(
		degradeRulePath,
		this::encodeJson
	);
	WritableDataSourceRegistry.registerDegradeDataSource(degradeRuleWDS);
	
	// 系统规则
	ReadableDataSource<String, List<SystemRule>> systemRuleRDS = new FileRefreshableDataSource<>(
		systemRulePath,
		systemRuleListParser
	);
	SystemRuleManager.register2Property(systemRuleRDS.getProperty());
	WritableDataSource<List<SystemRule>> systemRuleWDS = new FileWritableDataSource<>(
		systemRulePath,
		this::encodeJson
	);
	WritableDataSourceRegistry.registerSystemDataSource(systemRuleWDS);
	
	// 授权规则
	ReadableDataSource<String, List<AuthorityRule>> authorityRuleRDS = new FileRefreshableDataSource<>(
		authorityRulePath,
		authorityRuleListParser
	);
	AuthorityRuleManager.register2Property(authorityRuleRDS.getProperty());
	WritableDataSource<List<AuthorityRule>> authorityRuleWDS = new FileWritableDataSource<>(
		authorityRulePath,
		this::encodeJson
	);
	WritableDataSourceRegistry.registerAuthorityDataSource(authorityRuleWDS);
	
	// 热点参数规则
	ReadableDataSource<String, List<ParamFlowRule>> paramFlowRuleRDS = new FileRefreshableDataSource<>(
		paramFlowRulePath,
		paramFlowRuleListParser
	);
	ParamFlowRuleManager.register2Property(paramFlowRuleRDS.getProperty());
	WritableDataSource<List<ParamFlowRule>> paramFlowRuleWDS = new FileWritableDataSource<>(
		paramFlowRulePath,
		this::encodeJson
	);
	ModifyParamFlowRulesCommandHandler.setWritableDataSource(paramFlowRuleWDS);
	}
	private Converter<String, List<FlowRule>> flowRuleListParser = source -> JSON.parseObject(
		source,
		new TypeReference<List<FlowRule>>() {
		}
	);
	private Converter<String, List<DegradeRule>> degradeRuleListParser = source -> JSON.parseObject(
		source,
		new TypeReference<List<DegradeRule>>() {
		}
	);
	private Converter<String, List<SystemRule>> systemRuleListParser = source -> JSON.parseObject(
		source,
		new TypeReference<List<SystemRule>>() {
		}
	);
	private Converter<String, List<AuthorityRule>> authorityRuleListParser = source -> JSON.parseObject(
		source,
		new TypeReference<List<AuthorityRule>>() {
		}
	);
	private Converter<String, List<ParamFlowRule>> paramFlowRuleListParser = source -> JSON.parseObject(
		source,
		new TypeReference<List<ParamFlowRule>>() {
		}
	);
	private void mkdirIfNotExits(String filePath) throws IOException {
		File file = new File(filePath);
		if (!file.exists()) {
			file.mkdirs();
		}
	}
	private void createFileIfNotExits(String filePath) throws IOException {
		File file = new File(filePath);
		if (!file.exists()) {
			file.createNewFile();
		}
	}
	private <T> String encodeJson(T t) {
		return JSON.toJSONString(t);
	}
}

2.添加配置

在resources下创建配置目录 META-INF/services ,然后添加文件 com.alibaba.csp.sentinel.init.InitFunc 在文件中添加配置类的全路径

com.zxt.config.FilePersistence
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

新手且笨蛋37

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值