org.frameworkset.spi.remote.http.proxy.HttpHostDiscover Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of bboss-http Show documentation
Show all versions of bboss-http Show documentation
bboss http an http cluster lb component.
package org.frameworkset.spi.remote.http.proxy;
import org.frameworkset.spi.assemble.GetProperties;
import org.frameworkset.spi.remote.http.ClientConfiguration;
import org.frameworkset.spi.remote.http.HttpHost;
import org.frameworkset.util.shutdown.ShutdownUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.List;
/**
* 自动发现es 主机节点
*/
public abstract class HttpHostDiscover extends Thread{
private static Logger logger = LoggerFactory.getLogger(HttpHostDiscover.class);
private long discoverInterval = 10000l;
private HttpServiceHosts httpServiceHosts;
public HttpHostDiscover( ){
super("Http HostDiscover Thread");
ShutdownUtil.addShutdownHook(new Runnable() {
@Override
public void run() {
stopCheck();
}
});
this.setDaemon(true);
}
boolean stop = false;
public synchronized void stopCheck(){
if(stop)
return;
this.stop = true;
this.interrupt();
try {
this.join();
} catch (InterruptedException e) {
}
}
/**
* 主动定时服务发现和被动服务发现api
* @param _hosts
*/
public synchronized void handleDiscoverHosts(List _hosts){
handleDiscoverHosts( _hosts,(String)null);
}
/**
* routing必须在配置文件中指定，否则中间不允许切换
* 路由组发生变化时，切换路由组
* @param newCurrentRounte
*/
public synchronized void changeRouting(String newCurrentRounte){
httpServiceHosts.routingGroup(false,newCurrentRounte);
}
/**
* 主动定时服务发现和被动服务发现api
* @param _hosts
*/
public synchronized void handleDiscoverHosts(List _hosts,String newCurrentRounte){
List hosts = null;
if(_hosts != null && _hosts.size() > 0)
hosts = new ArrayList(_hosts);
else{
hosts = new ArrayList();
}
List newAddress = new ArrayList();
//恢复移除节点
boolean changed = httpServiceHosts.recoverRemovedNodes(hosts);
HttpHost httpHost = null;
String health = httpServiceHosts.getHttpServiceHostsConfig().getHealth();
//识别新增节点
for(int i = 0; hosts !=null && i < hosts.size();i ++){
httpHost = hosts.get(i);
HttpAddress address = new HttpAddress(httpHost.getOrigineAddress(),
httpHost.getHostAddress(),
httpHost.getRouting(),health);
// if(address.getRouting() == null || address.getRouting().equals(""))
// address.setRouting(httpHost.getRouting());
address.setAttributes(httpHost.getAttributes());
if(!httpServiceHosts.containAddress(address)){
newAddress.add(address);
}
}
//处理新增节点
if(newAddress.size() > 0) {
if(!changed )
changed = true;
if (logger.isInfoEnabled()) {
logger.info(new StringBuilder().append("Discovery new Http pool[")
.append(httpServiceHosts.getClientConfiguration().getBeanName()).append("] servers ").append(newAddress).append(".").toString());
}
httpServiceHosts.addAddresses(newAddress);
}
//处理删除节点
httpServiceHosts.handleRemoved( hosts);
//如果数据有变化，则根据routing规则对地址进行重新分组
if(changed) {
httpServiceHosts.routingGroup(true,newCurrentRounte);
}
else if(newCurrentRounte != null){
httpServiceHosts.routingGroup(false,newCurrentRounte);
}
}
protected abstract List discover(HttpServiceHostsConfig httpServiceHostsConfig, ClientConfiguration configuration, GetProperties context);
@Override
public void run() {
do {
if(this.stop)
break;
doDiscover();
try {
sleep(discoverInterval);
} catch (InterruptedException e) {
break;
}
}while(true);
}
private void doDiscover(){
try {
// clientInterface.discover("_nodes/http",ClientInterface.HTTP_GET, new ResponseHandler() {
//
// @Override
// public Void handleResponse(HttpResponse response) throws ClientProtocolException, IOException {
// int status = response.getStatusLine().getStatusCode();
// if (status >= 200 && status < 300) {
// List hosts = readHosts(response.getEntity());
// handleDiscoverHosts(hosts);
//
// } else {
//
// }
// return null;
// }
// });
if (logger.isDebugEnabled())
logger.debug(new StringBuilder().append("Discovery Http pool[")
.append(httpServiceHosts.getClientConfiguration().getBeanName()).append("] servers.").toString());
List httpHosts = discover(httpServiceHosts.getHttpServiceHostsConfig(), httpServiceHosts.getClientConfiguration(), httpServiceHosts.getClientConfiguration().getContextProperties());
if(httpHosts == null || httpHosts.size() == 0){
Boolean handleNullOrEmptyHostsByDiscovery = handleNullOrEmptyHostsByDiscovery();
if(handleNullOrEmptyHostsByDiscovery == null){
handleNullOrEmptyHostsByDiscovery = httpServiceHosts.getHandleNullOrEmptyHostsByDiscovery();
}
if( handleNullOrEmptyHostsByDiscovery == null || !handleNullOrEmptyHostsByDiscovery) {
if (logger.isInfoEnabled())
logger.info(new StringBuilder().append("Discovery ")
.append(httpServiceHosts.getClientConfiguration().getBeanName()).append(" servers : ignore with httpHosts == null || httpHosts.size() == 0").toString());
return;
}
}
handleDiscoverHosts( httpHosts);
} catch (Exception e) {
if (logger.isInfoEnabled())
logger.info(new StringBuilder().append("Discovery ")
.append(httpServiceHosts.getClientConfiguration().getBeanName()).append(" servers failed:").toString(),e);
}
}
public void setHttpServiceHosts(HttpServiceHosts httpServiceHosts) {
this.httpServiceHosts = httpServiceHosts;
if(this.httpServiceHosts.getDiscoverServiceInterval() > 0l){
this.discoverInterval = this.httpServiceHosts.getDiscoverServiceInterval();
}
String threadName = "Http pool["+httpServiceHosts.getClientConfiguration().getBeanName()+"] HostDiscover Thread";
this.setName(threadName);
if(logger.isInfoEnabled())
logger.info("HttpServiceHosts dicover thread:"+threadName);
}
public void start(){
//do server discover first，then start the thread.
if (logger.isInfoEnabled())
logger.info("First doDiscovery Http pool["+httpServiceHosts.getClientConfiguration().getBeanName()+"] servers at start time.");
this.doDiscover();
super.start();
}
/**
* 返回false，忽略对返回的null或者空的hosts进行处理；
* 返回true，要对null或者空的hosts进行处理，这样会导致所有的地址不可用
*
* @return 默认返回false
*/
protected Boolean handleNullOrEmptyHostsByDiscovery(){
return null;
}
}