gu.dtalk.activemq.DefaultLocalActivemqConfigProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dtalk-common-activemq Show documentation

common module for activemq

There is a newer version: 0.8.2

Show newest version

package gu.dtalk.activemq;

import java.net.URI;
import java.net.URISyntaxException;
import java.net.UnknownHostException;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;

import gu.simplemq.MQProperties;
import net.gdface.utils.JcifsUtil;

import static com.google.common.base.Preconditions.*;
import static gu.dtalk.activemq.ActivemqContext.HELPER;
import static gu.dtalk.activemq.ActivemqContext.CONST_PROVIDER;

/**
 * 局域网配置
 * @author guyadong
 *
 */
public class DefaultLocalActivemqConfigProvider extends BaseActivemqConfigProvider{
	private static final String REG_IPV4 = "^((?:(?:25[0-5]|2[0-4]\\d|[01]?\\d?\\d)\\.){3}(?:25[0-5]|2[0-4]\\d|[01]?\\d?\\d))$";
	public static final MQProperties INIT_PROPERTIES;
	static{
		INIT_PROPERTIES = HELPER.initParameters(null).initURI(makeLantalkURL("landtalkhost"));
		initMqttURI(INIT_PROPERTIES,DEFAULT_MQTT_PORT);
	}
	public DefaultLocalActivemqConfigProvider(){
	}
	
	@Override
	public final ActivemqConfigType type() {
		return ActivemqConfigType.LAN;
	}
	@Override
	protected MQProperties selfProp() {
		MQProperties props = INIT_PROPERTIES;
		URI uri = HELPER.getLocation(props);
		if(!uri.getHost().matches(REG_IPV4)){
			// 如果host不是IP地址格式，则替换主机名为对应的IP地址
			String host = IP_CACHE.getUnchecked(uri.getHost());
			URI u2 = changeHost(uri,host);
			props.initURI(u2);
		}
		return props;
	}

	private static final LoadingCache IP_CACHE = CacheBuilder.newBuilder().build(new CacheLoader(){

		@Override
		public String load(String landtalkhost) throws Exception {
			try {
				return JcifsUtil.hostAddressOf(landtalkhost);
			} catch (UnknownHostException e) {
				return landtalkhost;
			}	
		}});

	private static URI changeHost(URI uri,String host){
		// 替换主机名
		try {
			return new URI(uri.getScheme(),uri.getUserInfo(),host,uri.getPort(),uri.getPath(),uri.getQuery(),uri.getFragment());
		} catch (URISyntaxException e) {
			throw new RuntimeException(e);
		}
	}
	private static String makeLantalkURL(String landtalkhost){
		try {
			String defaultLocation = CONST_PROVIDER.getDefaultMQLocation();
			URI uri = new URI(defaultLocation);
			return changeHost(uri,landtalkhost).toString();
		} catch (URISyntaxException e) {
			throw new RuntimeException(e);
		}
	}
	/**
	 * 初始化局域网 ActiveMQ 主机名，默认值为'landtalkhost'
	 * @param landtalkhost 要设置的 landtalkhost
	 */
	public static void initLandtalkhost(String landtalkhost) {
		INIT_PROPERTIES.initURI(makeLantalkURL(checkNotNull(landtalkhost,"landtalkhost is null")));
	}

}