com.scudata.expression.mfn.dw.Update 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 java.io.IOException;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.Sequence;
import com.scudata.dm.cursor.ICursor;
import com.scudata.dm.cursor.MemoryCursor;
import com.scudata.dw.ColPhyTable;
import com.scudata.expression.IParam;
import com.scudata.expression.PhyTableFunction;
import com.scudata.resources.EngineMessage;

/**
 * ??????????ݣ??¼?ֵ????룬???ּ????????????????
 * T.update(P:D)
 * @author RunQian
 *
 */
public class Update extends PhyTableFunction {
	public Object calculate(Context ctx) {
		//?ɸ?ʽ???????֧?ָ???
		if (table instanceof ColPhyTable) {
			ColPhyTable colTable = (ColPhyTable) table;
			if (!colTable.getGroupTable().isPureFormat()) {
				MessageManager mm = EngineMessage.get();
				throw new RQException(mm.getMessage("dw.oldVersion2"));
			}
		}
		
		String opt = option;
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("update" + mm.getMessage("function.missingParam"));
		}
		
		boolean hasN = opt != null && opt.indexOf('n') != -1;
		boolean hasW = opt != null && opt.indexOf('w') != -1 && table instanceof ColPhyTable;
		Object result = null;
		Object obj, obj2 = null;
		
		if (param.getType() == IParam.Colon) {
			IParam sub0 = param.getSub(0);
			IParam sub1 = param.getSub(1);
			obj = sub0 == null ? null : sub0.getLeafExpression().calculate(ctx);
			obj2 = sub1 == null ? null : sub1.getLeafExpression().calculate(ctx);
			
		} else {
			obj = param.getLeafExpression().calculate(ctx);
		}
		
		//????????
		if (obj == null) {
			if (!hasN) {
				result = table;
			} else {
				result = null;
			}
		} else if (obj instanceof Sequence) {
			Sequence seq = (Sequence) obj;
			if (seq.length() == 0) {
				if (!hasN) {
					result = table;
				} else {
					result = null;
				}
			} else {
				if (hasW) {
					ICursor cs = new MemoryCursor(seq);
					updateColumn(cs, opt);
					result = table;
				} else {
					try {
						if (!hasN) {
							table.update(seq, opt);
							result = table;
						} else {
							result = table.update(seq, opt);
						}
					} catch (IOException e) {
						throw new RQException(e);
					}
				}
			}
			

		} else if (hasW && obj instanceof ICursor ) {
			ICursor cs = (ICursor)obj;
			updateColumn(cs, opt);
			result = table;
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("update" + mm.getMessage("function.paramTypeError"));
		}
		
		//????ɾ??
		if (obj2 != null) {
			try {
				Sequence result2 = table.delete((Sequence)obj2, opt);
				if (hasN && result != null) {
					((Sequence)result).addAll(result2);
				}
			} catch (IOException e) {
				throw new RQException(e);
			}
		}
		return result;
	}
	
	private void updateColumn(ICursor cs, String opt) {
		try {
			/**
			 * ???α????ʱ???????Ӽ?¼,
			 * ???Բ??ܳ???@i
			 */
			if (opt != null && opt.indexOf('i') != -1) {
				opt = opt.replace("i", "");
			}
			((ColPhyTable)table).update(cs, opt);
		} catch (IOException e) {
			throw new RQException(e);
		}
	}
}