com.scudata.expression.mfn.dw.Icursor 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.mfn.dw;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.Env;
import com.scudata.dm.FileObject;
import com.scudata.dm.cursor.ICursor;
import com.scudata.dw.IndexCursor;
import com.scudata.expression.Expression;
import com.scudata.expression.IParam;
import com.scudata.expression.Node;
import com.scudata.expression.PhyTableFunction;
import com.scudata.expression.fn.Between;
import com.scudata.resources.EngineMessage;

/**
 * ?????????????????????й???
 * T.icursor(C,??;w,I)
 * @author RunQian
 *
 */
public class Icursor extends PhyTableFunction {
	public Object calculate(Context ctx) {		
		IParam param = this.param;
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("icursor" + mm.getMessage("function.missingParam"));
		}
		
		boolean isMultiThread = option != null && option.indexOf('m') != -1;
		String []fields = null;
		Expression w = null;
		Object I = null;
		
		int segCount = 0;
		if (isMultiThread) {
			segCount = Env.getParallelNum();
		}
		
		if (param.getType() == IParam.Semicolon) {
			if (param.getSubSize() != 2 && param.getSubSize() != 3) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("icursor" + mm.getMessage("function.invalidParam"));
			}

			IParam sub = param.getSub(0);
			if (sub == null) {
				fields = null;
			} else {
				if (sub.isLeaf()) {
					fields = new String[]{sub.getLeafExpression().getIdentifierName()};
				} else {
					int fcount = sub.getSubSize();
					fields = new String[fcount];
					for (int f = 0; f < fcount; ++f) {
						IParam p = sub.getSub(f);
						if (p == null) {
							MessageManager mm = EngineMessage.get();
							throw new RQException("icursor" + mm.getMessage("function.invalidParam"));
						}
						
						fields[f] = p.getLeafExpression().getIdentifierName();
					}
				}
			}
			
			if (param.getSubSize() == 3) {
				sub = param.getSub(2);
				//?????ǶԶ?·?Ĵ??????Ա???ע??
//				if (sub != null) {
//					Object obj = sub.getLeafExpression().calculate(ctx);
//					if (!(obj instanceof Number)) {
//						MessageManager mm = EngineMessage.get();
//						throw new RQException("cursor" + mm.getMessage("function.paramTypeError"));
//					}
//
//					segCount = ((Number)obj).intValue();
//				}
				int size = sub.getSubSize();
				if (size == 0) {
					FileObject f = (FileObject) sub.getLeafExpression().calculate(ctx);
					I = new FileObject[] {f};	
				} else {
					FileObject[] files = new FileObject[size];
					for (int i = 0; i < size; i++) {
						files[i] = (FileObject) sub.getSub(i).getLeafExpression().calculate(ctx);
					}
					I = files;
				}
			}
			
			param = param.getSub(1);
			if (param == null) {
				return table.cursor(fields);//û?й???ʱ
			}

		}

		IParam sub0;
		if (param.isLeaf()) {
			sub0 = param;
		} else {
			sub0 = param.getSub(0);
			I= param.getSub(1).getLeafExpression().getIdentifierName();
		}
		
		if (sub0 != null) {
			w = sub0.getLeafExpression();
		}
		
		if (w == null) {
			w = new Expression("true");
		}
		
		//????between????ʽ
		Node home = w.getHome();
		if (home instanceof Between) {
			Between bt = (Between) home;
			IParam f = bt.getParam().getSub(0);
			IParam l = bt.getParam().getSub(1).getSub(0);
			IParam r = bt.getParam().getSub(1).getSub(1);
			String field = f.getLeafExpression().getIdentifierName();
			String left = l.getLeafExpression().getIdentifierName();
			String right = r.getLeafExpression().getIdentifierName();
			w = new Expression(field + ">=" + left + "&&" + field + "<=" + right);
		}
		
		ICursor cursor = table.icursor(fields, w, I, option, ctx);
		if (segCount != 0 && cursor instanceof IndexCursor) {
			return ((IndexCursor)cursor).toMultiCursor(segCount);
		} else {
			return cursor;
		}
	}
}