com.scudata.expression.fn.parallel.Callx Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
The newest version!
package com.scudata.expression.fn.parallel;

import java.util.ArrayList;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.JobSpace;
import com.scudata.dm.Machines;
import com.scudata.dm.ParallelCaller;
import com.scudata.dm.Sequence;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.expression.Node;
import com.scudata.resources.EngineMessage;


// callx(dfx,??;;x)callx@la(dfx,??;hs;x)
public class Callx extends Function {
	public Node optimize(Context ctx) {
		if (param != null)
			param.optimize(ctx);
		return this;
	}

	
	public Object calculate(Context ctx) {
		IParam leftParam = null;
		String[] hosts = null;
		int[] ports = null;
		String reduce = null;

		if (param == null) {
		} else if (param.getType() == IParam.Semicolon) {
			int size = param.getSubSize();
			if (size > 3) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("callx" + mm.getMessage("function.invalidParam"));
			}

			leftParam = param.getSub(0);

			IParam rightParam = param.getSub(1);
			IParam hubParam = null;
			if (rightParam != null && rightParam.getType() == IParam.Comma) {
				hubParam = rightParam.getSub(0);
				IParam sParam = rightParam.getSub(1);

				if(sParam.getSubSize()==0){//s or m
					Object indexObj = sParam.getLeafExpression().calculate(
							ctx);
					if (!(indexObj instanceof Sequence)) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("callx" + mm.getMessage("function.invalidParam"));
					} else {
//						taskIndexes = (Sequence) indexObj;
					}
				}else{
					MessageManager mm = EngineMessage.get();
					throw new RQException("callx" + mm.getMessage("function.invalidParam"));
				}
				
			} else {
				hubParam = rightParam;
			}
			
			if (hubParam != null) {
				Object hostObj = hubParam.getLeafExpression().calculate(ctx);
				Machines mc = new Machines();
				if (!mc.set(hostObj)) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("callx"
							+ mm.getMessage("function.invalidParam"));
				}

				hosts = mc.getHosts();
				ports = mc.getPorts();
			}

			if (size > 2) {
				IParam reduceParam = param.getSub(2);
				if (reduceParam != null) {
//					reduce = reduceParam.getLeafExpression().toString();
					reduce = (String)reduceParam.getLeafExpression().calculate(ctx);
				}
			}
		} else {
			leftParam = param;
		}

		String dfx;
		IParam dfxParam;
		int mcount = -1; // ??????
		Object[] args = null; // ????

		if (leftParam != null && leftParam.getType() == IParam.Comma) {
			dfxParam = leftParam.getSub(0);
			int pcount = leftParam.getSubSize() - 1;
			args = new Object[pcount];
			for (int p = 0; p < pcount; ++p) {
				IParam sub = leftParam.getSub(p + 1);
				if (sub != null) {
					args[p] = sub.getLeafExpression().calculate(ctx);
					if (args[p] instanceof Sequence) {
						int len = ((Sequence) args[p]).length();
						if (len == 0) {
							MessageManager mm = EngineMessage.get();
							throw new RQException("callx"
									+ mm.getMessage("function.invalidParam"));
						}

						if (mcount == -1) {
							mcount = len;
						} else if (mcount != len) {
							MessageManager mm = EngineMessage.get();
							throw new RQException(
									"callx" + mm.getMessage("function.paramCountNotMatch"));
						}
					}
				}
			}
		} else {
			dfxParam = leftParam;
		}

		if (dfxParam == null) {
			dfx = null;
		} else if (dfxParam.isLeaf()) {
			Object obj = dfxParam.getLeafExpression().calculate(ctx);
			if (!(obj instanceof String)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("callx" + mm.getMessage("function.paramTypeError"));
			}

			dfx = (String) obj;
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("callx" + mm.getMessage("function.invalidParam"));
		}

		ParallelCaller caller = new ParallelCaller(dfx, hosts, ports);
		caller.setContext(ctx);
		caller.setOptions(option);
		caller.setReduce(reduce);

		if (args != null) {
			if (mcount == -1) {
				mcount = 1;
			}

			int pcount = args.length;
			for (int i = 1; i <= mcount; ++i) {
				ArrayList