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.apollo.ApolloDataSource;
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 ApolloDatasourceFactory implements SentinelDatasourceFactory {
028
029
030    @Override
031    public ReadableDataSource createDataSource() {
032        ApolloDatasourceConfig adc = Jboot.config(ApolloDatasourceConfig.class);
033        adc.assertConfigOk();
034
035        String appId = System.getProperty("app.id", adc.getAppId());
036        String serverAddr = System.getProperty("apollo.meta", adc.getServerAddress());
037
038        System.setProperty("app.id", appId);
039        System.setProperty("apollo.meta", serverAddr);
040
041        return new ApolloDataSource<>(adc.getNamespaceName(), adc.getRuleKey(), adc.getDefaultFlowRules(),
042                source -> JSON.parseObject(source, new TypeReference<List<FlowRule>>() {
043                }));
044    }
045}