com.alogic.bundle.Bootstrap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of alogic-bundle Show documentation
an easy way to use alogic framework.
The newest version!
package com.alogic.bundle;

import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.SocketException;
import java.util.Enumeration;

import org.apache.commons.lang3.StringUtils;

import com.anysoft.util.PropertiesConstants;
import com.anysoft.util.Settings;
import com.logicbus.backend.server.LogicBusApp;

/**
 * 引导类
 * 
 * @author duanyy
 * @version 1.6.13.2 [20200312 duanyy] 

 * - 增加host.ip和host.port环境变量;
 */
public class Bootstrap extends LogicBusApp {
	public void onInit(Settings settings){		
		String app = settings.GetValue("app", "${server.app}");
		String niName = PropertiesConstants.getString(settings, "ketty.ip.ni", "");
		String ip = getHostIp(niName);
		
		long mem = 0;
		{
			mem = PropertiesConstants.getLong(settings, "mem", 0);
			if (mem <= 0){
				mem = Runtime.getRuntime().maxMemory() / 1000 / 1000;
			}
		}
		long cores = 0;
		{
			cores = PropertiesConstants.getLong(settings, "vcores", 0);
			if (cores <= 0){
				cores = Runtime.getRuntime().availableProcessors();
			}
		}	
		
		String port = settings.GetValue("port", "${server.port}");
		
		System.setProperty("server.ip", ip);
		System.setProperty("server.port", port);
		System.setProperty("server.mem", String.valueOf(mem));
		System.setProperty("server.app", app);
		System.setProperty("server.cores", String.valueOf(cores));

		String hostIp = System.getenv("KETTY_HOST");
		if (StringUtils.isNotEmpty(hostIp) && !hostIp.equals(ip)){
			//可能运行在容器中
			System.setProperty("host.ip",hostIp);
			System.setProperty("host.port",ip);
		}else {
			System.setProperty("host.ip",ip);
			System.setProperty("host.port",port);
		}
		
		super.onInit(settings);
	}
	
	
	public void onDestroy(Settings settings){
		super.onDestroy(settings);
	}
	
	public static String getHostIp(String name){
		// get ip from env : KETTY_IP
		String ip = System.getenv("KETTY_IP");
		if (StringUtils.isNotEmpty(ip)){
			return ip;
		}
		
		try {
			InetAddress addr = InetAddress.getLocalHost();
			return addr.getHostAddress();
		}catch (Exception ex){
			logger.error("Can not get ip from Local Host");
		}
		
		if (StringUtils.isNotEmpty(name)){
			ip = getIpByInterface(name);
			if (StringUtils.isNotEmpty(ip)){
				return ip;
			}
		}
		
		logger.info("Try to get ip from network interface.");
		Enumeration interfaces = null;
		try {
			interfaces = NetworkInterface.getNetworkInterfaces();  
		}catch (Exception ex){
			return "127.0.0.1";
		}
        while (interfaces.hasMoreElements()) {  
            NetworkInterface ni = interfaces.nextElement();  
            Enumeration addrs = ni.getInetAddresses();
            while (addrs.hasMoreElements()){
            	InetAddress inetAddr = addrs.nextElement();
            	if (!inetAddr.isLoopbackAddress()){
            		return inetAddr.getHostAddress();
            	}
            }
        }
        return "127.0.0.1";
	}
	
	public static String getIpByInterface(String name){
		NetworkInterface ni;
		try {
			ni = NetworkInterface.getByName(name);
	        Enumeration addrs = ni.getInetAddresses();
	        while (addrs.hasMoreElements()){
	        	InetAddress inetAddr = addrs.nextElement();
				if (!inetAddr.isLoopbackAddress()
						&& !inetAddr.isAnyLocalAddress()
						&& !inetAddr.isLinkLocalAddress()
						&& !inetAddr.isMulticastAddress()) {
					return inetAddr.getHostAddress();
				}
	        }
		} catch (SocketException e) {
			logger.error("Can not get ip by interface name.");
		}
        return null;
	}
}