72 lines
3.1 KiB
Java
72 lines
3.1 KiB
Java
/**
|
|
* Copyright (c) 2018-2028, DreamLu 卢春梦 (qq596392912@gmail.com).
|
|
* <p>
|
|
* Licensed under the GNU LESSER GENERAL PUBLIC LICENSE 3.0;
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
* <p>
|
|
* http://www.gnu.org/licenses/lgpl.html
|
|
* <p>
|
|
* 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.
|
|
*/
|
|
package org.springblade.core.cloud.hystrix;
|
|
|
|
import com.netflix.hystrix.Hystrix;
|
|
import com.netflix.hystrix.strategy.HystrixPlugins;
|
|
import com.netflix.hystrix.strategy.concurrency.HystrixConcurrencyStrategy;
|
|
import com.netflix.hystrix.strategy.eventnotifier.HystrixEventNotifier;
|
|
import com.netflix.hystrix.strategy.executionhook.HystrixCommandExecutionHook;
|
|
import com.netflix.hystrix.strategy.metrics.HystrixMetricsPublisher;
|
|
import com.netflix.hystrix.strategy.properties.HystrixPropertiesStrategy;
|
|
import org.springblade.core.cloud.props.BladeHystrixHeadersProperties;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.lang.Nullable;
|
|
|
|
import javax.annotation.PostConstruct;
|
|
|
|
/**
|
|
* Hystrix 配置
|
|
*
|
|
* @author L.cm
|
|
*/
|
|
@Configuration
|
|
@ConditionalOnClass(Hystrix.class)
|
|
@EnableConfigurationProperties(BladeHystrixHeadersProperties.class)
|
|
public class BladeHystrixAutoConfiguration {
|
|
@Nullable
|
|
@Autowired(required = false)
|
|
private HystrixConcurrencyStrategy existingConcurrencyStrategy;
|
|
@Nullable
|
|
@Autowired(required = false)
|
|
private BladeHystrixAccountGetter accountGetter;
|
|
@Autowired
|
|
private BladeHystrixHeadersProperties properties;
|
|
|
|
@PostConstruct
|
|
public void init() {
|
|
// Keeps references of existing Hystrix plugins.
|
|
HystrixEventNotifier eventNotifier = HystrixPlugins.getInstance().getEventNotifier();
|
|
HystrixMetricsPublisher metricsPublisher = HystrixPlugins.getInstance().getMetricsPublisher();
|
|
HystrixPropertiesStrategy propertiesStrategy = HystrixPlugins.getInstance().getPropertiesStrategy();
|
|
HystrixCommandExecutionHook commandExecutionHook = HystrixPlugins.getInstance().getCommandExecutionHook();
|
|
|
|
HystrixPlugins.reset();
|
|
|
|
// Registers existing plugins excepts the Concurrent Strategy plugin.
|
|
HystrixConcurrencyStrategy strategy = new BladeHystrixConcurrencyStrategy(existingConcurrencyStrategy, accountGetter, properties);
|
|
HystrixPlugins.getInstance().registerConcurrencyStrategy(strategy);
|
|
HystrixPlugins.getInstance().registerEventNotifier(eventNotifier);
|
|
HystrixPlugins.getInstance().registerMetricsPublisher(metricsPublisher);
|
|
HystrixPlugins.getInstance().registerPropertiesStrategy(propertiesStrategy);
|
|
HystrixPlugins.getInstance().registerCommandExecutionHook(commandExecutionHook);
|
|
}
|
|
|
|
}
|