001/**
002 * Copyright (c) 2015-2022, Michael Yang 杨福海 (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.apache.org/licenses/LICENSE-2.0
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package io.jboot.support.sentinel.datasource;
017
018import com.alibaba.csp.sentinel.datasource.ReadableDataSource;
019import com.alibaba.csp.sentinel.datasource.zookeeper.ZookeeperDataSource;
020import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
021import com.alibaba.fastjson.JSON;
022import com.alibaba.fastjson.TypeReference;
023import io.jboot.Jboot;
024
025import java.util.List;
026
027public class ZookeeperDatasourceFactory implements SentinelDatasourceFactory {
028
029    @Override
030    public ReadableDataSource createDataSource() {
031        ZookeeperDatasourceConfig zdc = Jboot.config(ZookeeperDatasourceConfig.class);
032        zdc.assertConfigOk();
033
034        return new ZookeeperDataSource<>(zdc.getServerAddress(), zdc.getPath(),
035                source -> JSON.parseObject(source, new TypeReference<List<FlowRule>>() {
036                }));
037    }
038}