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.redis.RedisDataSource;
020import com.alibaba.csp.sentinel.datasource.redis.config.RedisConnectionConfig;
021import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
022import com.alibaba.fastjson.JSON;
023import com.alibaba.fastjson.TypeReference;
024import io.jboot.Jboot;
025import io.jboot.utils.StrUtil;
026
027import java.util.List;
028
029public class RedisDatasourceFactory implements SentinelDatasourceFactory {
030
031    @Override
032    public ReadableDataSource createDataSource() {
033        RedisDatasourceConfig rdc = Jboot.config(RedisDatasourceConfig.class);
034        rdc.assertConfigOk();
035
036        RedisConnectionConfig.Builder builder = RedisConnectionConfig.builder()
037                .withHost(rdc.getHost())
038                .withPort(rdc.getPort())
039                .withDatabase(rdc.getDatabase());
040
041        if (StrUtil.isNotBlank(rdc.getPassword())) {
042            builder.withPassword(rdc.getPassword());
043        }
044
045        return new RedisDataSource<>(builder.build(), rdc.getRuleKey(), rdc.getChannel(),
046                source -> JSON.parseObject(source, new TypeReference<List<FlowRule>>() {
047                }));
048
049    }
050}