Java HystrixPlugins类代码示例

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
/**
 * registers the {@link ExecutionContextAwareHystrixStrategy}
 */
public static void init() {
    // keeps references of existing Hystrix plugins.
    HystrixConcurrencyStrategy existingConcurrencyStrategy = HystrixPlugins.getInstance().getConcurrencyStrategy();
    HystrixEventNotifier eventNotifier = HystrixPlugins.getInstance().getEventNotifier();
    HystrixMetricsPublisher metricsPublisher = HystrixPlugins.getInstance().getMetricsPublisher();
    HystrixPropertiesStrategy propertiesStrategy = HystrixPlugins.getInstance().getPropertiesStrategy();
    HystrixCommandExecutionHook commandExecutionHook = HystrixPlugins.getInstance().getCommandExecutionHook();
    // reset the Hystrix plugin
    HystrixPlugins.reset();
    // configure the  plugin
    HystrixPlugins.getInstance().registerConcurrencyStrategy(new ExecutionContextAwareHystrixStrategy(existingConcurrencyStrategy));
    HystrixPlugins.getInstance().registerEventNotifier(eventNotifier);
    HystrixPlugins.getInstance().registerMetricsPublisher(metricsPublisher);
    HystrixPlugins.getInstance().registerPropertiesStrategy(propertiesStrategy);
    HystrixPlugins.getInstance().registerCommandExecutionHook(commandExecutionHook);
    log.info("Context propagation enabled for Hystrix.");
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Override
public void bindTo(MeterRegistry registry) {
    // Keeps references of existing Hystrix plugins.
    HystrixEventNotifier eventNotifier = HystrixPlugins.getInstance().getEventNotifier();
    HystrixPropertiesStrategy propertiesStrategy = HystrixPlugins.getInstance().getPropertiesStrategy();
    HystrixCommandExecutionHook commandExecutionHook = HystrixPlugins.getInstance().getCommandExecutionHook();
    HystrixConcurrencyStrategy concurrencyStrategy = HystrixPlugins.getInstance().getConcurrencyStrategy();

    HystrixPlugins.reset();

    // Registers existing plugins except the new MicroMeter Strategy plugin.
    HystrixPlugins.getInstance().registerMetricsPublisher(new MicrometerMetricsPublisher(registry));
    HystrixPlugins.getInstance().registerConcurrencyStrategy(concurrencyStrategy);
    HystrixPlugins.getInstance().registerEventNotifier(eventNotifier);
    HystrixPlugins.getInstance().registerPropertiesStrategy(propertiesStrategy);
    HystrixPlugins.getInstance().registerCommandExecutionHook(commandExecutionHook);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Override
    public void onApplicationEvent(ApplicationReadyEvent event) {
        if (event.getApplicationContext().getParent() == null) {
//        	HystrixPlugins.getInstance().registerCommandExecutionHook(impl);
//        	HystrixPlugins.getInstance().getCommandExecutionHook();
//        	if(!(HystrixPlugins.getInstance().getCommandExecutionHook() instanceof SecurityContextRegistratorCommandHook)){
            try {
                HystrixPlugins.getInstance().registerCommandExecutionHook(new SecurityContextRegistratorCommandHook());
            } catch (Exception e) {
                // TODO: handle exception
            }

//        	}

            logger.debug("初始化HystrixPlugins完成");
        }
    }
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void checkNotifierIsCalledOnceForEveryCommand() {
    mocks.forEach(m -> doAnswer(a -> {
        HystrixCommandMetrics metrics = a.getArgument(0);
        log.info("initialize for command {} and group {}", metrics.getCommandKey().name(), metrics.getCommandGroup().name());
                return null;
            }).when(m).initialize(any())
    );

    HystrixPlugins.getInstance().registerMetricsPublisher(notifier);

    for (int i = 0; i < 3; i++) {
        Command.from(GROUP_KEY, COMMAND_1, SUCCESS, 50).observe();
    }

    for (int i = 0; i < 5; i++) {
        Command.from(GROUP_KEY, COMMAND_2, FAILURE, 60).observe();
    }

    mocks.forEach(m -> {
        verify(m, times(1)).initialize(withKey(COMMAND_1));
        verify(m, times(1)).initialize(withKey(COMMAND_2));
        verifyNoMoreInteractions(m);
    });
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void concurrentTest() throws Throwable {
    HystrixMetricsInitializationNotifier notifier = new HystrixMetricsInitializationNotifier();
    HystrixPlugins.getInstance().registerMetricsPublisher(notifier);
    runMultiThreaded(10, () -> {
        try {
            HystrixCommand<Integer> cmd = Command.from(GROUP_KEY, COMMAND_1, SUCCESS, 50);
            cmd.observe();

            TimeUnit.MILLISECONDS.sleep(new Random().nextInt(100));
            notifier.addListener(mock(HystrixMetricsInitializationListener.class));
            TimeUnit.MILLISECONDS.sleep(new Random().nextInt(100));

            HystrixCommand<Integer> cmd2 = Command.from(GROUP_KEY, COMMAND_2, SUCCESS, 50);
            cmd2.observe();
        } catch (InterruptedException e) {
            // do nothing
        }
    });

}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void should_not_override_existing_custom_strategies() {
	HystrixPlugins.getInstance().registerCommandExecutionHook(new MyHystrixCommandExecutionHook());
	HystrixPlugins.getInstance().registerEventNotifier(new MyHystrixEventNotifier());
	HystrixPlugins.getInstance().registerMetricsPublisher(new MyHystrixMetricsPublisher());
	HystrixPlugins.getInstance().registerPropertiesStrategy(new MyHystrixPropertiesStrategy());

	new SleuthHystrixConcurrencyStrategy(this.tracer, this.traceKeys);

	then(HystrixPlugins
			.getInstance().getCommandExecutionHook()).isExactlyInstanceOf(MyHystrixCommandExecutionHook.class);
	then(HystrixPlugins.getInstance()
			.getEventNotifier()).isExactlyInstanceOf(MyHystrixEventNotifier.class);
	then(HystrixPlugins.getInstance()
			.getMetricsPublisher()).isExactlyInstanceOf(MyHystrixMetricsPublisher.class);
	then(HystrixPlugins.getInstance()
			.getPropertiesStrategy()).isExactlyInstanceOf(MyHystrixPropertiesStrategy.class);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void shouldRegisterDespitePreviouslyRegisteredHystrixMetricsPlugins() {
    // given
    // ... a provider is already registered
    HystrixPlugins.getInstance().registerMetricsPublisher(HystrixMetricsPublisherDefault.getInstance());

    // when
    // ... we register (without throwing a "Another strategy was already registered" exception)
    HystrixPrometheusMetricsPublisher.register("exampleapp");

    // then
    // ... we'll be able to collect metrics for commands
    TestHystrixCommand command = new TestHystrixCommand("any");
    command.execute();

    assertThat(CollectorRegistry.defaultRegistry
            .getSampleValue("exampleapp_hystrix_command_event_total",
                    new String[]{"command_group", "command_name", "event"},
                    new String[]{"group_any",
                            "command_any", "success"}))
            .describedAs("counter is present")
            .isGreaterThan(0);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void shouldReRegisterCustomHystrixPlugins() {
    // given
    // ... a plugin is already registered
    HystrixCommandExecutionHook plugin = new HystrixCommandExecutionHook() {
    };
    HystrixPlugins.getInstance().registerCommandExecutionHook(plugin);

    // when
    // ... we register
    HystrixPrometheusMetricsPublisher.builder().shouldRegisterDefaultPlugins(false).buildAndRegister();

    // then
    // ... the other plugin is still registered.
    assertThat(HystrixPlugins.getInstance().getCommandExecutionHook()).isEqualTo(plugin);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@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.
	HystrixPlugins.getInstance().registerConcurrencyStrategy(
			new SecurityContextConcurrencyStrategy(existingConcurrencyStrategy));
	HystrixPlugins.getInstance().registerEventNotifier(eventNotifier);
	HystrixPlugins.getInstance().registerMetricsPublisher(metricsPublisher);
	HystrixPlugins.getInstance().registerPropertiesStrategy(propertiesStrategy);
	HystrixPlugins.getInstance().registerCommandExecutionHook(commandExecutionHook);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void shouldNotLogWhenShortCircuited() {
    final DefaultExceptionLogger defaultExceptionLogger = spy(new DefaultExceptionLogger());
    HystrixPlugins.getInstance().registerCommandExecutionHook(new ExceptionLoggingCommandHook(defaultExceptionLogger));

    try {
        new AlwaysShortCircuit().execute();
    } catch (HystrixRuntimeException err) {
        assertThat(Throwables.getCausalChain(err)
                .stream()
                .filter(AuthenticationException.class::isInstance)
                .findAny())
                .isNotEmpty();
    }

    verifyZeroInteractions(defaultExceptionLogger);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void shouldLogWhenExceptionIsThrown() throws AuthenticationException {
    final DefaultExceptionLogger defaultExceptionLogger = spy(new DefaultExceptionLogger());
    HystrixPlugins.getInstance().registerCommandExecutionHook(new ExceptionLoggingCommandHook(defaultExceptionLogger));
    when(mockAuthenticator.authenticate(any(BasicCredentials.class))).thenThrow(new AuthenticationException("test"));
    doCallRealMethod().when(defaultExceptionLogger).log(any(Exception.class), any(HystrixCommand.class));

    try {
        tenacityAuthenticator.authenticate(new BasicCredentials("foo", "foo"));
    } catch (HystrixRuntimeException err) {
        assertThat(Throwables.getCausalChain(err)
                .stream()
                .filter(AuthenticationException.class::isInstance)
                .findAny())
        .isNotEmpty();
    }

    verify(mockAuthenticator, times(1)).authenticate(any(BasicCredentials.class));
    verify(defaultExceptionLogger, times(1)).log(any(Exception.class), any(HystrixCommand.class));
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void testCreateBizkeeperCommand() {
  HystrixPlugins.reset();
  ConsumerBizkeeperHandler consumerBizkeeperHandler = new ConsumerBizkeeperHandler();

  Invocation invocation = Mockito.mock(Invocation.class);
  Mockito.when(invocation.getOperationMeta()).thenReturn(Mockito.mock(OperationMeta.class));
  Mockito.when(invocation.getOperationMeta().getMicroserviceQualifiedName()).thenReturn("test1");

  CommandKey.toHystrixCommandGroupKey("groupname", invocation);
  CommandKey.toHystrixCommandKey("groupname", invocation);
  BizkeeperCommand command = consumerBizkeeperHandler.createBizkeeperCommand(invocation);
  Assert.assertNotNull(command);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void testCreateBizkeeperCommand() {
  HystrixPlugins.reset();
  ProviderBizkeeperHanlder providerBizkeeperHanlder = new ProviderBizkeeperHanlder();

  Invocation invocation = Mockito.mock(Invocation.class);
  Mockito.when(invocation.getOperationMeta()).thenReturn(Mockito.mock(OperationMeta.class));
  Mockito.when(invocation.getOperationMeta().getMicroserviceQualifiedName()).thenReturn("test1");

  CommandKey.toHystrixCommandGroupKey("groupname", invocation);
  CommandKey.toHystrixCommandKey("groupname", invocation);
  BizkeeperCommand command = providerBizkeeperHanlder.createBizkeeperCommand(invocation);
  Assert.assertNotNull(command);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void testHandleWithException() {
  boolean validAssert;
  HystrixPlugins.reset();
  try {
    validAssert = true;
    bizkeeperHandler.handle(invocation, asyncResp);
  } catch (Exception e) {
    validAssert = false;
  }
  Assert.assertFalse(validAssert);
}
 

import com.netflix.hystrix.strategy.HystrixPlugins; //导入依赖的package包/类
@Test
public void testSetCommonProperties() {
  boolean validAssert;
  try {
    validAssert = true;
    HystrixPlugins.reset();
    HystrixCommandProperties.Setter setter = Mockito.mock(HystrixCommandProperties.Setter.class);
    bizkeeperHandler.setCommonProperties(invocation, setter);
  } catch (Exception e) {
    validAssert = false;
  }
  Assert.assertTrue(validAssert);
}