com.scudata.expression.fn.Call 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.
There is a newer version: 20240823
package com.scudata.expression.fn;

import java.io.File;

import com.scudata.cellset.datamodel.PgmCellSet;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.common.UUID;
import com.scudata.dm.Context;
import com.scudata.dm.DfxManager;
import com.scudata.dm.FileObject;
import com.scudata.dm.JobSpace;
import com.scudata.dm.JobSpaceManager;
import com.scudata.dm.Param;
import com.scudata.dm.ParamList;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.expression.Node;
import com.scudata.resources.EngineMessage;
import com.scudata.thread.CallJob;

/**
 * ????ָ?????񣬷???????ķ???ֵ???෵??ֵƴ??????
 * call(dfx,arg1,??) ???????arg1,???????????ļ?dfx?????????һ??returnֵ???رա?
 * @author RunQian
 *
 */
public class Call extends Function {
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("call" + mm.getMessage("function.missingParam"));
		}
	}

	public Node optimize(Context ctx) {
		param.optimize(ctx);
		return this;
	}

	public Object calculate(Context ctx) {
		PgmCellSet pcs = getCallPgmCellSet(ctx);
		if (option != null && option.indexOf('n') != -1) {
			// ???????߳?ִ?нű???ֱ?ӷ???
			String uuid = UUID.randomUUID().toString();
			JobSpace jobSpace = JobSpaceManager.getSpace(uuid);
			pcs.getContext().setJobSpace(jobSpace);
			
			CallJob job = new CallJob(pcs, option);
			Thread thread = new Thread(job);
			thread.start();
			return null;
		}
		
		Object val = pcs.execute();
		if (option == null || option.indexOf('r') == -1) {
			pcs.reset();
			DfxManager.getInstance().putDfx(pcs);
		}
		
		return val;
	}
	
	// ?ڱ?????dfx???Ҳ???????null
	public String getDfxPathName(Context ctx) {
		IParam param = this.param;
		if (!param.isLeaf()) {
			param = param.getSub(0);
			if (param == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("call" + mm.getMessage("function.invalidParam"));
			}
		}
		
		FileObject fo;
		Object obj = param.getLeafExpression().calculate(ctx);
		if (obj instanceof String) {
			fo = new FileObject((String)obj, null, "s", ctx);
		} else if (obj instanceof FileObject) {
			fo = (FileObject)obj;
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("call" + mm.getMessage("function.paramTypeError"));
		}
		
		File file = fo.getLocalFile().getFile();
		if (file != null) {
			return file.getAbsolutePath();
		} else {
			return null;
		}
	}
	
	// ide????ȡ????????????е???????
	public PgmCellSet getCallPgmCellSet(Context ctx) {
		IParam param = this.param;
		boolean useCache = option == null || option.indexOf('r') == -1;
		DfxManager dfxManager = DfxManager.getInstance();
		PgmCellSet pcs;
		if (param.isLeaf()) {
			Object obj = param.getLeafExpression().calculate(ctx);
			if (obj instanceof String) {
				if (useCache) {
					pcs = dfxManager.removeDfx((String)obj, ctx);
				} else {
					pcs = dfxManager.readDfx((String)obj, ctx);
				}
			} else if (obj instanceof FileObject) {
				if (useCache) {
					pcs = dfxManager.removeDfx((FileObject)obj, ctx);
				} else {
					pcs = dfxManager.readDfx((FileObject)obj, ctx);
				}
			} else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("call" + mm.getMessage("function.paramTypeError"));
			}
			
			pcs.setParamToContext();
		} else {
			IParam sub0 = param.getSub(0);
			if (sub0 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("call" + mm.getMessage("function.invalidParam"));
			}

			Object obj = sub0.getLeafExpression().calculate(ctx);
			if (obj instanceof String) {
				if (useCache) {
					pcs = dfxManager.removeDfx((String)obj, ctx);
				} else {
					pcs = dfxManager.readDfx((String)obj, ctx);
				}
			} else if (obj instanceof FileObject) {
				if (useCache) {
					pcs = dfxManager.removeDfx((FileObject)obj, ctx);
				} else {
					pcs = dfxManager.readDfx((FileObject)obj, ctx);
				}
			} else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("call" + mm.getMessage("function.paramTypeError"));
			}

			pcs.setParamToContext();
			ParamList list = pcs.getParamList();
			if (list != null) {
				int size = param.getSubSize();
				if (size - 1 > list.count()) size = list.count() + 1;

				Context curCtx = pcs.getContext();
				for (int i = 1; i < size; ++i) {
					IParam sub = param.getSub(i);
					Param p = list.get(i - 1);
					if (sub != null) {
						Object val = sub.getLeafExpression().calculate(ctx);
						curCtx.setParamValue(p.getName(), val);
					} else {
						curCtx.setParamValue(p.getName(), null);
					}
				}
			}
		}
		
		return pcs;
	}
	
	// ide???????ú???ôκ???
	public void finish(PgmCellSet pcs) {
		if (option == null || option.indexOf('r') == -1) {
			pcs.reset();
			DfxManager.getInstance().putDfx(pcs);
		}
	}
}