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);
	}
}