• Home
• com.scudata.esproc
• esproc
• 20240823
• source code
• UnitClient.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.scudata.parallel.UnitClient Maven / Gradle / Ivy

package com.scudata.parallel;

import java.io.Serializable;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Map;

import com.scudata.common.Logger;
import com.scudata.common.RQException;
import com.scudata.dm.Env;
import com.scudata.dm.JobSpace;
import com.scudata.dm.JobSpaceManager;
import com.scudata.dm.Param;
import com.scudata.dm.Sequence;
import com.scudata.dm.Table;

/**
 * ?ֻ??ķ??ʿͻ???
 * @author Joancy
 *
 */
public class UnitClient implements Serializable {
	private static final long serialVersionUID = 1L;

	String host = null;
	int port = 0;

	boolean isDispatchUC = false;
	// ȡ????
	transient SocketData socketData = null;
	/**
	 * ???ӳ?ʱʱ??????λms
	 */
	private int connectTimeout = 5000;

	/**
	 * ???ݵ?ַ?Ͷ˿ڹ???һ???ֻ??ͻ???
	 * @param host ????IP
	 * @param port ?˿ں?
	 */
	public UnitClient(String host, int port) {
		this.host = host;
		this.port = port;
	}

	/**
	 * ??????????????(???磺  host:ip)????ֻ??ͻ???
	 * @param add ??????????
	 */
	public UnitClient(String add) {
		if (add != null) {
			int index = add.lastIndexOf(':');
			if (index > 0) {
				host = add.substring(0, index).trim();
				port = Integer.parseInt(add.substring(index + 1).trim());
			}
		}
	}

	/**
	 * ???÷ֻ?Ϊ???ط???Ŀͻ???
	 * ע?⣺ֻ??ͨ??getClient????getRandomClient??????ȡ??UC??????Ϊtrue??
	 * Ϊtrue??ʾ??uc???ڶ???????ά????????????ģ?ͣ?????ʱ???ٶ??У???????????ͷŵ????С?
	 */
	public void setDispatchable() {
		isDispatchUC = true;
	}

	/**
	 * ?жϷֻ??Ƿ???ط?????ҵ
	 * @return ????????true?????򷵻?false
	 */
	public boolean isDispatchable() {
		return isDispatchUC;
	}

	/**
	 * ??ȡ????IP
	 * @return IP??
	 */
	public String getHost() {
		return host;
	}

	/**
	 * ??ȡ?????˿ں?
	 * @return ?˿ں?
	 */
	public int getPort() {
		return port;
	}

	/**
	 * ??¡һ???ֻ??ͻ???
	 * @return ??¡??ķֻ??ͻ???
	 */
	public UnitClient clone() {
		UnitClient uc = new UnitClient(host, port);
		return uc;
	}

	/**
	 * ?ж??????ֻ??ͻ????Ƿ????
	 * @param nodeHost ??һ???ֻ?IP
	 * @param nodePort ??һ???ֻ??˿ں?
	 * @return ??ͬʱ????true?????򷵻?false
	 */
	public boolean equals(String nodeHost, int nodePort) {
		return host.equalsIgnoreCase(nodeHost) && port == nodePort;
	}

	/**
	 * ?жϵ?ǰ?ֻ??Ƿ??Ѿ????????ɷ???
	 * @return ?ɷ??ʷ???true?????򷵻?false
	 */
	public boolean isReachable() {
		try {
			InetAddress address = InetAddress.getByName(host);
			boolean isArrived = address.isReachable(2000);
			return isArrived;
		} catch (Exception x) {
			Logger.debug("Ping " + host + " failure.", x);
			return false;
		}
	}

	/**
	 * ??鵱ǰ?ֻ??Ƿ???
	 * @return ???ʱ????true?????򷵻?false
	 */
	public boolean isAlive() {
		return isAlive(null);
	}

	/**
	 * ?жϵ?ǰ?ֻ??Ƿ?λ?ڱ???
	 * @return ?DZ??ػ?????true?????򷵻?false
	 */
	public boolean isEqualToLocal() {
		// callx֧?? :, ""д??????ʹ?ñ????߳?
		if (host == null)
			return true;

		HostManager hostManager = HostManager.instance();
		return host.equals(hostManager.getHost())
				&& port == hostManager.getPort();
	}

	/**
	 * ?жϷֻ??Ƿ??????쳣?????쳣ԭ??д??ԭ?򻺳?reason
	 * @param reason ԭ?򻺳?
	 * @return ???ʱ????true?????򷵻?false
	 */
	public boolean isAlive(StringBuffer reason) {
		if (isEqualToLocal()) {
			return true;
		}
		SocketData sd = null;

		try {
			sd = newSocketData();
		} catch (Exception x) {
			if (reason != null) {
				if (!isReachable()) {
					reason.append(this + " is not exist.\n");
				} else {
					reason.append("UnitServer or UnitServerConsole is not started on "
							+ this + "\n");
				}
			}
			return false;
		} finally {
			if (sd != null) {
				try {
					sd.clientClose();
				} catch (Exception x) {
				}
				sd = null;
			}
		}
		return true;
	}

	/**
	 * ???ӵ??ֻ?
	 * @throws Exception ???ӳ???ʱ?׳??쳣
	 */
	public void connect() throws Exception {
		if (!isEqualToLocal()) {
			socketData = newSocketData();
		}
	}

	/**
	 * ?ж??Ƿ??Ѿ????ӵ??ֻ?
	 * @return ???Ӻú󷵻?true?????򷵻?false
	 */
	public boolean isConnected() {
		return !isClosed() || isEqualToLocal();// ?????߳?Ҳ??????״̬
	}

	/**
	 * ???ֻ?д??һ??????
	 * @param obj ???ݶ???
	 * @throws Exception д????ʱ?׳??쳣
	 */
	public void write(Object obj) throws Exception {
		socketData.write(obj);
	}

	/**
	 * ?ӷֻ?????һ??????
	 * @return ????????
	 * @throws Exception ??????ʱ?׳??쳣
	 */
	public Object read() throws Exception {
		return socketData.read();
	}
	
	/**
	 * ???????ӳ?ʱʱ??????λ????
	 * @param timeout
	 */
	public void setConnectTimeout(int timeout){
		this.connectTimeout = timeout;
	}

	/**
	 * ????һ???µ?ͨѶ?׽???
	 * @return ͨѶ?׽???
	 * @throws Exception ????ʱ?׳??쳣
	 */
	public SocketData newSocketData() throws Exception {
		Socket s = new Socket();
		SocketData sd = new SocketData(s);
		InetSocketAddress isa = new InetSocketAddress(host, port);
		sd.connect(isa, connectTimeout);

		return sd;
	}

	/**
	 * ȡ????ǰ????
	 * @param taskId ??????
	 * @param reason ȡ??ԭ??
	 */
	public void cancel(Integer taskId, String reason) {
		Request req = new Request(Request.DFX_CANCEL);
		req.setAttr(Request.CANCEL_TaskId, taskId);
		req.setAttr(Request.CANCEL_Reason, reason);
		try {
			sendByNewSocket(req);
		} catch (Exception x) {
			x.printStackTrace();
		}
	}

	/**
	 * ֹͣ??????
	 */
	public void shutDown() {
		Request req = new Request(Request.SERVER_SHUTDOWN);
		try {
			sendByNewSocket(req);
		} catch (Exception x) {
		}
	}

	/**
	 * ?г???ǰ?ֻ????????ҵ??
	 * @return ?????ҵ??
	 */
	public int getUnitMaxNum() {
		if (isEqualToLocal()) {
			return HostManager.maxTaskNum;
		}
		Request req = new Request(Request.SERVER_GETUNITS_MAXNUM);
		try {
			Response res = send(req);
			return (Integer) res.getResult();
		} catch (Exception x) {
			throw new RQException(x);
		}
	}

	/**
	 * ??ȡ?ֻ???????J???ڴ?????
	 * @param J ????????
	 * @return ?ڴ?????
	 */
	public Integer getAreaNo(String J) {
		Request req = new Request(Request.SERVER_GETAREANO);
		req.setAttr(Request.GETAREANO_TaskName, J);
		try {
			Response res = sendByNewSocket(req);
			return (Integer) res.getResult();
		} catch (Exception x) {
			throw new RQException(x);
		}
	}

	/**
	 * spaceId - Param[]
	 * 
	 * @return HashMap
	 */
	public Table getEnvParamList() {
		Request req = new Request(Request.SERVER_LISTPARAM);
		try {
			Response res = sendByNewSocket(req);
			if (res.getException() != null) {
				throw res.getException();
			}
			return (Table) res.getResult();
		} catch (Exception x) {
			x.printStackTrace();
		}
		return null;
	}

	public Table getTaskList() {
		Request req = new Request(Request.SERVER_LISTTASK);
		try {
			Response res = sendByNewSocket(req);
			if (res.getException() != null) {
				throw res.getException();
			}
			return (Table) res.getResult();
		} catch (Exception x) {
			x.printStackTrace();
		}
		return null;
	}

	public void closeSpace(String spaceId) {
		if (!isAlive()) {
			return;
		}
		Request req = new Request(Request.SERVER_CLOSESPACE);
		req.setAttr(Request.CLOSESPACE_SpaceId, spaceId);
		try {
			Response res = sendByNewSocket(req);
			if (res.getException() != null) {
				throw res.getException();
			}
		} catch (Exception x) {
			x.printStackTrace();
		}
	}

	public void initNode(int i, int N, String j) throws Exception {
		Logger.debug("Before init zone: " + i + " on " + this);

		Request req = new Request(Request.ZONE_INITDFX);
		// req.setAttr(Request.EXECDFX_DfxName, "init.dfx");
		ArrayList