sviolet.slate.common.helper.sentinel.EzSentinelConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of slate-helper Show documentation
Comprehensive Java library for spring framework (Java8+)
The newest version!
/*
 * Copyright (C) 2015-2019 S.Violet
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * Project GitHub: https://github.com/shepherdviolet/slate
 * Email: [email protected]
 */

package sviolet.slate.common.helper.sentinel;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * 注解方式启用JSON格式的Sentinel规则设置器. 说明文档见: https://github.com/shepherdviolet/slate/blob/master/docs/ezsentinel/guide.md
 *
 * 依赖: compile "com.google.code.gson:gson:$version_gson"
 *
 * @author S.Violet
 */
@Configuration
public class EzSentinelConfiguration {

    private final Logger logger = LoggerFactory.getLogger(getClass());

    /**
     * 规则设置器
     */
    @Bean("slate.common.ezSentinelRuleConfigurer")
    @ConditionalOnProperty(name = "spring.cloud.sentinel.enabled", matchIfMissing = true)
    public EzSentinelRuleConfigurer ezSentinelRuleConfigurer(){
        return new JsonEzSentinelRuleConfigurer();
    }

    /**
     * 规则设置器绑定参数slate.common.ez-sentinel.rule-data
     */
    @Bean("slate.common.ezSentinelRuleConfigurerPropertySetter")
    @ConditionalOnProperty(name = "spring.cloud.sentinel.enabled", matchIfMissing = true)
    public Object ezSentinelRuleConfigurerPropertySetter(AbstractEzSentinelRuleConfigurer ezSentinelRuleConfigurer){
        return new Object() {

            /**
             * 监听本地配置和Apollo配置, 更新规则
             */
            @Value("${slate.common.ez-sentinel.rule-data:}")
            public void setRuleData(String ruleData) {
                ezSentinelRuleConfigurer.update(ruleData);
            }

        };
    }

    /**
     * Sentinel开关
     */
    @Value("${spring.cloud.sentinel.enabled:true}")
    public void setEnabled(boolean enabled) {
        //这个只对核心有效, 即对本地统计限流熔断有效
        com.alibaba.csp.sentinel.Constants.ON = enabled;
        logger.info("EzSentinel | Sentinel " + (enabled ? "Enabled" : "Disabled"));
    }

}