mirror of https://github.com/longtai-cn/hippo4j
HystrixThreadPoolAdapter supports hippo4j config (#1086)
yangsanity
2 years ago
committed by
GitHub
parent
2546f04a98
commit
ca8f5a9f06
@ -0,0 +1,163 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package cn.hippo4j.adapter.hystrix;
|
||||||
|
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapter;
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapterParameter;
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapterState;
|
||||||
|
import cn.hippo4j.common.toolkit.CollectionUtil;
|
||||||
|
import cn.hippo4j.common.toolkit.ThreadPoolExecutorUtil;
|
||||||
|
import com.netflix.hystrix.HystrixThreadPool;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.boot.context.event.ApplicationStartedEvent;
|
||||||
|
import org.springframework.context.ApplicationListener;
|
||||||
|
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
|
import java.util.concurrent.ThreadPoolExecutor;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
|
import static cn.hippo4j.common.constant.ChangeThreadPoolConstants.CHANGE_DELIMITER;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* abstract hystrix thread-pool adapter
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
public abstract class AbstractHystrixThreadPoolAdapter implements ThreadPoolAdapter, ApplicationListener<ApplicationStartedEvent> {
|
||||||
|
|
||||||
|
private static final String THREAD_POOL_FIELD = "threadPool";
|
||||||
|
|
||||||
|
private static final String THREAD_POOLS_FIELD = "threadPools";
|
||||||
|
|
||||||
|
private final Map<String, ThreadPoolExecutor> hystrixConsumeExecutor = new HashMap<>();
|
||||||
|
|
||||||
|
protected final ThreadPoolAdapterScheduler threadPoolAdapterScheduler;
|
||||||
|
|
||||||
|
protected AbstractHystrixThreadPoolAdapter(ThreadPoolAdapterScheduler threadPoolAdapterScheduler) {
|
||||||
|
this.threadPoolAdapterScheduler = threadPoolAdapterScheduler;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String mark() {
|
||||||
|
return "Hystrix";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ThreadPoolAdapterState getThreadPoolState(String identify) {
|
||||||
|
ThreadPoolAdapterState result = new ThreadPoolAdapterState();
|
||||||
|
ThreadPoolExecutor threadPoolExecutor = hystrixConsumeExecutor.get(identify);
|
||||||
|
if (threadPoolExecutor != null) {
|
||||||
|
result.setThreadPoolKey(identify);
|
||||||
|
result.setCoreSize(threadPoolExecutor.getCorePoolSize());
|
||||||
|
result.setMaximumSize(threadPoolExecutor.getMaximumPoolSize());
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
log.warn("[{}] Hystrix thread pool not found.", identify);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<ThreadPoolAdapterState> getThreadPoolStates() {
|
||||||
|
List<ThreadPoolAdapterState> threadPoolAdapterStates = new ArrayList<>();
|
||||||
|
hystrixConsumeExecutor.forEach((kel, val) -> threadPoolAdapterStates.add(getThreadPoolState(kel)));
|
||||||
|
return threadPoolAdapterStates;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean updateThreadPool(ThreadPoolAdapterParameter threadPoolAdapterParameter) {
|
||||||
|
String threadPoolKey = threadPoolAdapterParameter.getThreadPoolKey();
|
||||||
|
ThreadPoolExecutor threadPoolExecutor = hystrixConsumeExecutor.get(threadPoolKey);
|
||||||
|
if (threadPoolExecutor == null) {
|
||||||
|
log.warn("[{}] Hystrix thread pool not found.", threadPoolKey);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
int originalCoreSize = threadPoolExecutor.getCorePoolSize();
|
||||||
|
int originalMaximumPoolSize = threadPoolExecutor.getMaximumPoolSize();
|
||||||
|
ThreadPoolExecutorUtil.safeSetPoolSize(threadPoolExecutor, threadPoolAdapterParameter.getCorePoolSize(), threadPoolAdapterParameter.getMaximumPoolSize());
|
||||||
|
log.info("[{}] Hystrix thread pool parameter change. coreSize: {}, maximumSize: {}",
|
||||||
|
threadPoolKey,
|
||||||
|
String.format(CHANGE_DELIMITER, originalCoreSize, threadPoolExecutor.getCorePoolSize()),
|
||||||
|
String.format(CHANGE_DELIMITER, originalMaximumPoolSize, threadPoolExecutor.getMaximumPoolSize()));
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onApplicationEvent(ApplicationStartedEvent event) {
|
||||||
|
ScheduledExecutorService scheduler = threadPoolAdapterScheduler.getScheduler();
|
||||||
|
int taskIntervalSeconds = threadPoolAdapterScheduler.getTaskIntervalSeconds();
|
||||||
|
// Periodically update the Hystrix thread pool.
|
||||||
|
HystrixThreadPoolRefreshTask hystrixThreadPoolRefreshTask = new HystrixThreadPoolRefreshTask(scheduler, taskIntervalSeconds);
|
||||||
|
scheduler.schedule(hystrixThreadPoolRefreshTask, taskIntervalSeconds, TimeUnit.SECONDS);
|
||||||
|
}
|
||||||
|
|
||||||
|
class HystrixThreadPoolRefreshTask implements Runnable {
|
||||||
|
|
||||||
|
private final ScheduledExecutorService scheduler;
|
||||||
|
|
||||||
|
private final int taskIntervalSeconds;
|
||||||
|
|
||||||
|
HystrixThreadPoolRefreshTask(ScheduledExecutorService scheduler, int taskIntervalSeconds) {
|
||||||
|
this.scheduler = scheduler;
|
||||||
|
this.taskIntervalSeconds = taskIntervalSeconds;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
hystrixThreadPoolRefresh();
|
||||||
|
} finally {
|
||||||
|
if (!scheduler.isShutdown()) {
|
||||||
|
scheduler.schedule(this, taskIntervalSeconds, TimeUnit.MILLISECONDS);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void hystrixThreadPoolRefresh() {
|
||||||
|
try {
|
||||||
|
Class<HystrixThreadPool.Factory> factoryClass = HystrixThreadPool.Factory.class;
|
||||||
|
Field threadPoolsField = factoryClass.getDeclaredField(THREAD_POOLS_FIELD);
|
||||||
|
threadPoolsField.setAccessible(true);
|
||||||
|
ConcurrentHashMap<String, HystrixThreadPool> threadPools =
|
||||||
|
(ConcurrentHashMap<String, HystrixThreadPool>) threadPoolsField.get(factoryClass);
|
||||||
|
if (CollectionUtil.isNotEmpty(threadPools)) {
|
||||||
|
for (Map.Entry<String, HystrixThreadPool> stringHystrixThreadPoolEntry : threadPools.entrySet()) {
|
||||||
|
String key = stringHystrixThreadPoolEntry.getKey();
|
||||||
|
HystrixThreadPool value = stringHystrixThreadPoolEntry.getValue();
|
||||||
|
if (value instanceof HystrixThreadPool.HystrixThreadPoolDefault) {
|
||||||
|
HystrixThreadPool.HystrixThreadPoolDefault hystrixThreadPoolDefault =
|
||||||
|
(HystrixThreadPool.HystrixThreadPoolDefault) value;
|
||||||
|
Class<? extends HystrixThreadPool.HystrixThreadPoolDefault> hystrixThreadPoolDefaultClass = hystrixThreadPoolDefault.getClass();
|
||||||
|
Field threadPoolField = hystrixThreadPoolDefaultClass.getDeclaredField(THREAD_POOL_FIELD);
|
||||||
|
threadPoolField.setAccessible(true);
|
||||||
|
ThreadPoolExecutor threadPoolExecutor =
|
||||||
|
(ThreadPoolExecutor) threadPoolField.get(hystrixThreadPoolDefault);
|
||||||
|
hystrixConsumeExecutor.put(key, threadPoolExecutor);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (Exception ex) {
|
||||||
|
log.error("Failed to get Hystrix thread pool.", ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,245 +0,0 @@
|
|||||||
/*
|
|
||||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
||||||
* contributor license agreements. See the NOTICE file distributed with
|
|
||||||
* this work for additional information regarding copyright ownership.
|
|
||||||
* The ASF licenses this file to You 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package cn.hippo4j.adapter.hystrix;
|
|
||||||
|
|
||||||
import cn.hippo4j.adapter.base.ThreadPoolAdapter;
|
|
||||||
import cn.hippo4j.adapter.base.ThreadPoolAdapterCacheConfig;
|
|
||||||
import cn.hippo4j.adapter.base.ThreadPoolAdapterParameter;
|
|
||||||
import cn.hippo4j.adapter.base.ThreadPoolAdapterRegisterAction;
|
|
||||||
import cn.hippo4j.adapter.base.ThreadPoolAdapterState;
|
|
||||||
import cn.hippo4j.common.config.ApplicationContextHolder;
|
|
||||||
import cn.hippo4j.common.toolkit.CollectionUtil;
|
|
||||||
import cn.hippo4j.common.toolkit.ThreadPoolExecutorUtil;
|
|
||||||
import com.netflix.hystrix.HystrixThreadPool;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.springframework.boot.context.event.ApplicationStartedEvent;
|
|
||||||
import org.springframework.context.ApplicationListener;
|
|
||||||
|
|
||||||
import java.lang.reflect.Field;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
|
||||||
import java.util.concurrent.ScheduledExecutorService;
|
|
||||||
import java.util.concurrent.ThreadPoolExecutor;
|
|
||||||
import java.util.concurrent.TimeUnit;
|
|
||||||
import java.util.stream.Collectors;
|
|
||||||
|
|
||||||
import static cn.hippo4j.common.constant.ChangeThreadPoolConstants.CHANGE_DELIMITER;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* hystrix thread-pool adapter.
|
|
||||||
*/
|
|
||||||
@Slf4j
|
|
||||||
public class HystrixThreadPoolAdapter implements ThreadPoolAdapter, ApplicationListener<ApplicationStartedEvent> {
|
|
||||||
|
|
||||||
private static final String THREAD_POOL_FIELD = "threadPool";
|
|
||||||
|
|
||||||
private static final String THREAD_POOLS_FIELD = "threadPools";
|
|
||||||
|
|
||||||
private final Map<String, ThreadPoolExecutor> hystrixConsumeExecutor = new HashMap<>();
|
|
||||||
|
|
||||||
private final ThreadPoolAdapterScheduler threadPoolAdapterScheduler;
|
|
||||||
|
|
||||||
public HystrixThreadPoolAdapter(ThreadPoolAdapterScheduler threadPoolAdapterScheduler) {
|
|
||||||
this.threadPoolAdapterScheduler = threadPoolAdapterScheduler;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String mark() {
|
|
||||||
return "Hystrix";
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public ThreadPoolAdapterState getThreadPoolState(String identify) {
|
|
||||||
ThreadPoolAdapterState result = new ThreadPoolAdapterState();
|
|
||||||
ThreadPoolExecutor threadPoolExecutor = hystrixConsumeExecutor.get(identify);
|
|
||||||
if (threadPoolExecutor != null) {
|
|
||||||
result.setThreadPoolKey(identify);
|
|
||||||
result.setCoreSize(threadPoolExecutor.getCorePoolSize());
|
|
||||||
result.setMaximumSize(threadPoolExecutor.getMaximumPoolSize());
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
log.warn("[{}] Hystrix thread pool not found.", identify);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<ThreadPoolAdapterState> getThreadPoolStates() {
|
|
||||||
List<ThreadPoolAdapterState> threadPoolAdapterStates = new ArrayList<>();
|
|
||||||
hystrixConsumeExecutor.forEach((kel, val) -> threadPoolAdapterStates.add(getThreadPoolState(kel)));
|
|
||||||
return threadPoolAdapterStates;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean updateThreadPool(ThreadPoolAdapterParameter threadPoolAdapterParameter) {
|
|
||||||
String threadPoolKey = threadPoolAdapterParameter.getThreadPoolKey();
|
|
||||||
ThreadPoolExecutor threadPoolExecutor = hystrixConsumeExecutor.get(threadPoolKey);
|
|
||||||
if (threadPoolExecutor == null) {
|
|
||||||
log.warn("[{}] Hystrix thread pool not found.", threadPoolKey);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
int originalCoreSize = threadPoolExecutor.getCorePoolSize();
|
|
||||||
int originalMaximumPoolSize = threadPoolExecutor.getMaximumPoolSize();
|
|
||||||
ThreadPoolExecutorUtil.safeSetPoolSize(threadPoolExecutor, threadPoolAdapterParameter.getCorePoolSize(), threadPoolAdapterParameter.getMaximumPoolSize());
|
|
||||||
log.info("[{}] Hystrix thread pool parameter change. coreSize: {}, maximumSize: {}",
|
|
||||||
threadPoolKey,
|
|
||||||
String.format(CHANGE_DELIMITER, originalCoreSize, threadPoolExecutor.getCorePoolSize()),
|
|
||||||
String.format(CHANGE_DELIMITER, originalMaximumPoolSize, threadPoolExecutor.getMaximumPoolSize()));
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onApplicationEvent(ApplicationStartedEvent event) {
|
|
||||||
ScheduledExecutorService scheduler = threadPoolAdapterScheduler.getScheduler();
|
|
||||||
int taskIntervalSeconds = threadPoolAdapterScheduler.getTaskIntervalSeconds();
|
|
||||||
// Periodically update the Hystrix thread pool.
|
|
||||||
HystrixThreadPoolRefreshTask hystrixThreadPoolRefreshTask = new HystrixThreadPoolRefreshTask(scheduler, taskIntervalSeconds);
|
|
||||||
scheduler.schedule(hystrixThreadPoolRefreshTask, taskIntervalSeconds, TimeUnit.SECONDS);
|
|
||||||
// Periodically refresh registration.
|
|
||||||
ThreadPoolAdapterRegisterAction threadPoolAdapterRegisterAction = ApplicationContextHolder.getBean(ThreadPoolAdapterRegisterAction.class);
|
|
||||||
Map<String, ? extends HystrixThreadPoolAdapter> beansOfType = ApplicationContextHolder.getBeansOfType(this.getClass());
|
|
||||||
Map<String, ThreadPoolAdapter> map = new HashMap<>(beansOfType);
|
|
||||||
ThreadPoolAdapterRegisterTask threadPoolAdapterRegisterTask = new ThreadPoolAdapterRegisterTask(scheduler, taskIntervalSeconds, map, threadPoolAdapterRegisterAction);
|
|
||||||
scheduler.schedule(threadPoolAdapterRegisterTask, threadPoolAdapterScheduler.getTaskIntervalSeconds(), TimeUnit.SECONDS);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void hystrixThreadPoolRefresh() {
|
|
||||||
try {
|
|
||||||
Class<HystrixThreadPool.Factory> factoryClass = HystrixThreadPool.Factory.class;
|
|
||||||
Field threadPoolsField = factoryClass.getDeclaredField(THREAD_POOLS_FIELD);
|
|
||||||
threadPoolsField.setAccessible(true);
|
|
||||||
ConcurrentHashMap<String, HystrixThreadPool> threadPools =
|
|
||||||
(ConcurrentHashMap<String, HystrixThreadPool>) threadPoolsField.get(factoryClass);
|
|
||||||
if (CollectionUtil.isNotEmpty(threadPools)) {
|
|
||||||
for (Map.Entry<String, HystrixThreadPool> stringHystrixThreadPoolEntry : threadPools.entrySet()) {
|
|
||||||
String key = stringHystrixThreadPoolEntry.getKey();
|
|
||||||
HystrixThreadPool value = stringHystrixThreadPoolEntry.getValue();
|
|
||||||
if (value instanceof HystrixThreadPool.HystrixThreadPoolDefault) {
|
|
||||||
HystrixThreadPool.HystrixThreadPoolDefault hystrixThreadPoolDefault =
|
|
||||||
(HystrixThreadPool.HystrixThreadPoolDefault) value;
|
|
||||||
Class<? extends HystrixThreadPool.HystrixThreadPoolDefault> hystrixThreadPoolDefaultClass = hystrixThreadPoolDefault.getClass();
|
|
||||||
Field threadPoolField = hystrixThreadPoolDefaultClass.getDeclaredField(THREAD_POOL_FIELD);
|
|
||||||
threadPoolField.setAccessible(true);
|
|
||||||
ThreadPoolExecutor threadPoolExecutor =
|
|
||||||
(ThreadPoolExecutor) threadPoolField.get(hystrixThreadPoolDefault);
|
|
||||||
hystrixConsumeExecutor.put(key, threadPoolExecutor);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (Exception ex) {
|
|
||||||
log.error("Failed to get Hystrix thread pool.", ex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private boolean compareThreadPoolAdapterCacheConfigs(List<ThreadPoolAdapterCacheConfig> newThreadPoolAdapterCacheConfigs,
|
|
||||||
List<ThreadPoolAdapterCacheConfig> oldThreadPoolAdapterCacheConfigs) {
|
|
||||||
boolean registerFlag = false;
|
|
||||||
Map<String, List<ThreadPoolAdapterState>> newThreadPoolAdapterCacheConfigMap =
|
|
||||||
newThreadPoolAdapterCacheConfigs.stream().collect(Collectors.toMap(
|
|
||||||
ThreadPoolAdapterCacheConfig::getMark, ThreadPoolAdapterCacheConfig::getThreadPoolAdapterStates, (k1, k2) -> k2));
|
|
||||||
Map<String, List<ThreadPoolAdapterState>> oldThreadPoolAdapterCacheConfigMap =
|
|
||||||
oldThreadPoolAdapterCacheConfigs.stream().collect(Collectors.toMap(
|
|
||||||
ThreadPoolAdapterCacheConfig::getMark, ThreadPoolAdapterCacheConfig::getThreadPoolAdapterStates, (k1, k2) -> k2));
|
|
||||||
for (Map.Entry<String, List<ThreadPoolAdapterState>> entry : newThreadPoolAdapterCacheConfigMap.entrySet()) {
|
|
||||||
String key = entry.getKey();
|
|
||||||
List<ThreadPoolAdapterState> newValue = entry.getValue();
|
|
||||||
List<ThreadPoolAdapterState> oldValue = oldThreadPoolAdapterCacheConfigMap.get(key);
|
|
||||||
if (oldValue == null) {
|
|
||||||
registerFlag = true;
|
|
||||||
break;
|
|
||||||
} else {
|
|
||||||
if (newValue.size() != oldValue.size()) {
|
|
||||||
registerFlag = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return registerFlag;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Hystrix Thread Pool Refresh Task
|
|
||||||
*/
|
|
||||||
class HystrixThreadPoolRefreshTask implements Runnable {
|
|
||||||
|
|
||||||
private ScheduledExecutorService scheduler;
|
|
||||||
|
|
||||||
private int taskIntervalSeconds;
|
|
||||||
|
|
||||||
HystrixThreadPoolRefreshTask(ScheduledExecutorService scheduler, int taskIntervalSeconds) {
|
|
||||||
this.scheduler = scheduler;
|
|
||||||
this.taskIntervalSeconds = taskIntervalSeconds;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
try {
|
|
||||||
hystrixThreadPoolRefresh();
|
|
||||||
} finally {
|
|
||||||
if (!scheduler.isShutdown()) {
|
|
||||||
scheduler.schedule(this, taskIntervalSeconds, TimeUnit.MILLISECONDS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Thread Pool Adapter Register Task
|
|
||||||
*/
|
|
||||||
class ThreadPoolAdapterRegisterTask implements Runnable {
|
|
||||||
|
|
||||||
private ScheduledExecutorService scheduler;
|
|
||||||
|
|
||||||
private int taskIntervalSeconds;
|
|
||||||
|
|
||||||
Map<String, ThreadPoolAdapter> threadPoolAdapterMap;
|
|
||||||
|
|
||||||
ThreadPoolAdapterRegisterAction threadPoolAdapterRegisterAction;
|
|
||||||
|
|
||||||
private List<ThreadPoolAdapterCacheConfig> cacheConfigList = new ArrayList<>();
|
|
||||||
|
|
||||||
ThreadPoolAdapterRegisterTask(ScheduledExecutorService scheduler, int taskIntervalSeconds,
|
|
||||||
Map<String, ThreadPoolAdapter> threadPoolAdapterMap,
|
|
||||||
ThreadPoolAdapterRegisterAction threadPoolAdapterRegisterAction) {
|
|
||||||
this.scheduler = scheduler;
|
|
||||||
this.taskIntervalSeconds = taskIntervalSeconds;
|
|
||||||
this.threadPoolAdapterMap = threadPoolAdapterMap;
|
|
||||||
this.threadPoolAdapterRegisterAction = threadPoolAdapterRegisterAction;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
try {
|
|
||||||
List<ThreadPoolAdapterCacheConfig> newThreadPoolAdapterCacheConfigs = threadPoolAdapterRegisterAction.getThreadPoolAdapterCacheConfigs(threadPoolAdapterMap);
|
|
||||||
boolean registerFlag = compareThreadPoolAdapterCacheConfigs(newThreadPoolAdapterCacheConfigs, cacheConfigList);
|
|
||||||
cacheConfigList = newThreadPoolAdapterCacheConfigs;
|
|
||||||
if (registerFlag) {
|
|
||||||
threadPoolAdapterRegisterAction.doRegister(cacheConfigList);
|
|
||||||
}
|
|
||||||
} catch (Exception ex) {
|
|
||||||
log.error("Register task error.", ex);
|
|
||||||
} finally {
|
|
||||||
if (!scheduler.isShutdown()) {
|
|
||||||
scheduler.schedule(this, taskIntervalSeconds, TimeUnit.MILLISECONDS);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -0,0 +1,28 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package cn.hippo4j.adapter.hystrix;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hystrix thread-pool adapter for hippo4j config.
|
||||||
|
*/
|
||||||
|
public class HystrixThreadPoolAdapter4Config extends AbstractHystrixThreadPoolAdapter {
|
||||||
|
|
||||||
|
public HystrixThreadPoolAdapter4Config(ThreadPoolAdapterScheduler threadPoolAdapterScheduler) {
|
||||||
|
super(threadPoolAdapterScheduler);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,136 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package cn.hippo4j.adapter.hystrix;
|
||||||
|
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapter;
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapterCacheConfig;
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapterParameter;
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapterRegisterAction;
|
||||||
|
import cn.hippo4j.adapter.base.ThreadPoolAdapterState;
|
||||||
|
import cn.hippo4j.common.config.ApplicationContextHolder;
|
||||||
|
import cn.hippo4j.common.toolkit.CollectionUtil;
|
||||||
|
import com.netflix.hystrix.HystrixThreadPool;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.boot.context.event.ApplicationStartedEvent;
|
||||||
|
import org.springframework.context.ApplicationListener;
|
||||||
|
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
|
import java.util.concurrent.ThreadPoolExecutor;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
import static cn.hippo4j.common.constant.ChangeThreadPoolConstants.CHANGE_DELIMITER;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* hystrix thread-pool adapter for hippo4j server.
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
public class HystrixThreadPoolAdapter4Server extends AbstractHystrixThreadPoolAdapter {
|
||||||
|
|
||||||
|
public HystrixThreadPoolAdapter4Server(ThreadPoolAdapterScheduler threadPoolAdapterScheduler) {
|
||||||
|
super(threadPoolAdapterScheduler);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onApplicationEvent(ApplicationStartedEvent event) {
|
||||||
|
super.onApplicationEvent(event);
|
||||||
|
ScheduledExecutorService scheduler = threadPoolAdapterScheduler.getScheduler();
|
||||||
|
int taskIntervalSeconds = threadPoolAdapterScheduler.getTaskIntervalSeconds();
|
||||||
|
// Periodically refresh registration.
|
||||||
|
ThreadPoolAdapterRegisterAction threadPoolAdapterRegisterAction = ApplicationContextHolder.getBean(ThreadPoolAdapterRegisterAction.class);
|
||||||
|
Map<String, ? extends HystrixThreadPoolAdapter4Server> beansOfType = ApplicationContextHolder.getBeansOfType(this.getClass());
|
||||||
|
Map<String, ThreadPoolAdapter> map = new HashMap<>(beansOfType);
|
||||||
|
ThreadPoolAdapterRegisterTask threadPoolAdapterRegisterTask = new ThreadPoolAdapterRegisterTask(scheduler, taskIntervalSeconds, map, threadPoolAdapterRegisterAction);
|
||||||
|
scheduler.schedule(threadPoolAdapterRegisterTask, threadPoolAdapterScheduler.getTaskIntervalSeconds(), TimeUnit.SECONDS);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Thread Pool Adapter Register Task
|
||||||
|
*/
|
||||||
|
static class ThreadPoolAdapterRegisterTask implements Runnable {
|
||||||
|
|
||||||
|
private final ScheduledExecutorService scheduler;
|
||||||
|
|
||||||
|
private final int taskIntervalSeconds;
|
||||||
|
|
||||||
|
Map<String, ThreadPoolAdapter> threadPoolAdapterMap;
|
||||||
|
|
||||||
|
ThreadPoolAdapterRegisterAction threadPoolAdapterRegisterAction;
|
||||||
|
|
||||||
|
private List<ThreadPoolAdapterCacheConfig> cacheConfigList = new ArrayList<>();
|
||||||
|
|
||||||
|
ThreadPoolAdapterRegisterTask(ScheduledExecutorService scheduler, int taskIntervalSeconds,
|
||||||
|
Map<String, ThreadPoolAdapter> threadPoolAdapterMap,
|
||||||
|
ThreadPoolAdapterRegisterAction threadPoolAdapterRegisterAction) {
|
||||||
|
this.scheduler = scheduler;
|
||||||
|
this.taskIntervalSeconds = taskIntervalSeconds;
|
||||||
|
this.threadPoolAdapterMap = threadPoolAdapterMap;
|
||||||
|
this.threadPoolAdapterRegisterAction = threadPoolAdapterRegisterAction;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
List<ThreadPoolAdapterCacheConfig> newThreadPoolAdapterCacheConfigs = threadPoolAdapterRegisterAction.getThreadPoolAdapterCacheConfigs(threadPoolAdapterMap);
|
||||||
|
boolean registerFlag = compareThreadPoolAdapterCacheConfigs(newThreadPoolAdapterCacheConfigs, cacheConfigList);
|
||||||
|
cacheConfigList = newThreadPoolAdapterCacheConfigs;
|
||||||
|
if (registerFlag) {
|
||||||
|
threadPoolAdapterRegisterAction.doRegister(cacheConfigList);
|
||||||
|
}
|
||||||
|
} catch (Exception ex) {
|
||||||
|
log.error("Register task error.", ex);
|
||||||
|
} finally {
|
||||||
|
if (!scheduler.isShutdown()) {
|
||||||
|
scheduler.schedule(this, taskIntervalSeconds, TimeUnit.MILLISECONDS);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean compareThreadPoolAdapterCacheConfigs(List<ThreadPoolAdapterCacheConfig> newThreadPoolAdapterCacheConfigs,
|
||||||
|
List<ThreadPoolAdapterCacheConfig> oldThreadPoolAdapterCacheConfigs) {
|
||||||
|
boolean registerFlag = false;
|
||||||
|
Map<String, List<ThreadPoolAdapterState>> newThreadPoolAdapterCacheConfigMap =
|
||||||
|
newThreadPoolAdapterCacheConfigs.stream().collect(Collectors.toMap(
|
||||||
|
ThreadPoolAdapterCacheConfig::getMark, ThreadPoolAdapterCacheConfig::getThreadPoolAdapterStates, (k1, k2) -> k2));
|
||||||
|
Map<String, List<ThreadPoolAdapterState>> oldThreadPoolAdapterCacheConfigMap =
|
||||||
|
oldThreadPoolAdapterCacheConfigs.stream().collect(Collectors.toMap(
|
||||||
|
ThreadPoolAdapterCacheConfig::getMark, ThreadPoolAdapterCacheConfig::getThreadPoolAdapterStates, (k1, k2) -> k2));
|
||||||
|
for (Map.Entry<String, List<ThreadPoolAdapterState>> entry : newThreadPoolAdapterCacheConfigMap.entrySet()) {
|
||||||
|
String key = entry.getKey();
|
||||||
|
List<ThreadPoolAdapterState> newValue = entry.getValue();
|
||||||
|
List<ThreadPoolAdapterState> oldValue = oldThreadPoolAdapterCacheConfigMap.get(key);
|
||||||
|
if (oldValue == null) {
|
||||||
|
registerFlag = true;
|
||||||
|
break;
|
||||||
|
} else {
|
||||||
|
if (newValue.size() != oldValue.size()) {
|
||||||
|
registerFlag = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return registerFlag;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue