com.scudata.expression.mfn.db.Update Maven / Gradle / Ivy

Go to download
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: 20241126
package com.scudata.expression.mfn.db;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.DataStruct;
import com.scudata.dm.Sequence;
import com.scudata.dm.cursor.ICursor;
import com.scudata.expression.DBFunction;
import com.scudata.expression.Expression;
import com.scudata.expression.IParam;
import com.scudata.resources.EngineMessage;

/**
 * ִ?????ݿ???????
 * db.update(A:A',tbl,F:x,??;P,??)
 * @author RunQian
 *
 */
public class Update extends DBFunction {
	public Object calculate(Context ctx) {
		String []pks = null;
		IParam param = this.param;
		if (param != null && param.getType() == IParam.Semicolon) {
			if (param.getSubSize() != 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update" + mm.getMessage("function.invalidParam"));
			}

			IParam p = param.getSub(1);
			if (p == null) {
			} else if (p.isLeaf()) {
				String name = p.getLeafExpression().getIdentifierName();
				pks = new String[] {name};
			} else {
				int size = p.getSubSize();
				pks = new String[size];
				for (int i = 0; i < size; ++i) {
					IParam sub = p.getSub(i);
					if (sub == null) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("update" + mm.getMessage("function.invalidParam"));
					}

					pks[i] = sub.getLeafExpression().getIdentifierName();
				}
			}

			param = param.getSub(0);
		}

		if (param == null || param.getType() != IParam.Comma) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("update" + mm.getMessage("function.invalidParam"));
		}

		Sequence seq1 = null;
		ICursor cursor = null;
		Sequence seq2 = null;
		
		IParam srcParam = param.getSub(0);
		if (srcParam == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("update" + mm.getMessage("function.invalidParam"));
		} else if (srcParam.isLeaf()) {
			Object srcObj = srcParam.getLeafExpression().calculate(ctx);
			if (srcObj instanceof Sequence) {
				seq1 = (Sequence)srcObj;
			} else if (srcObj instanceof ICursor) {
				cursor = (ICursor)srcObj;
			} else if (srcObj != null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update" + mm.getMessage("function.paramTypeError"));
			}
		} else {
			if (srcParam.getSubSize() != 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update" + mm.getMessage("function.invalidParam"));
			}
			
			IParam sub0 = srcParam.getSub(0);
			IParam sub1 = srcParam.getSub(1);
			if (sub0 == null || sub1 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update" + mm.getMessage("function.invalidParam"));
			}
			
			Object srcObj = sub0.getLeafExpression().calculate(ctx);
			if (!(srcObj instanceof Sequence) && srcObj != null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update" + mm.getMessage("function.paramTypeError"));
			}
			
			seq1 = (Sequence)srcObj;
			srcObj = sub1.getLeafExpression().calculate(ctx);
			if (!(srcObj instanceof Sequence) && srcObj != null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update" + mm.getMessage("function.paramTypeError"));
			}
			
			seq2 = (Sequence)srcObj;
		}

		Sequence sequence = null;
		if (seq1 != null) {
			sequence = seq1;
		} else if (seq2 != null) {
			sequence = seq2;
		} else if (cursor != null) {
			sequence = cursor.peek(1);
		}
		
		if (sequence == null) {
			return new Integer(0);
		}

		IParam sub1 = param.getSub(1);
		if (sub1 == null || !sub1.isLeaf()) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("update" + mm.getMessage("function.invalidParam"));
		}

		String tableName = sub1.getLeafExpression().getIdentifierName();
		String []names;
		Expression []valExps;
		int fcount;

		int size = param.getSubSize();
		if (size > 2) {
			fcount = size - 2;
			names = new String[fcount];
			valExps = new Expression[fcount];
			for (int i = 2, seq = 0; i < size; ++i, ++seq) {
				IParam sub = param.getSub(i);
				if (sub == null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("update" + mm.getMessage("function.invalidParam"));
				} else if (sub.isLeaf()) {
					names[seq] = sub.getLeafExpression().getIdentifierName();
					valExps[seq] = new Expression(ctx, names[seq]);
				} else {
					if (sub.getSubSize() != 2) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("update" + mm.getMessage("function.invalidParam"));
					}

					IParam nameParam = sub.getSub(0);
					IParam valParam = sub.getSub(1);
					if (nameParam == null || valParam == null) {
						MessageManager mm = EngineMessage.get();
						throw new RQException("update" + mm.getMessage("function.invalidParam"));
					}

					names[seq] = nameParam.getLeafExpression().getIdentifierName();
					valExps[seq] = valParam.getLeafExpression();
				}
			}
		} else {
			DataStruct ds = sequence.dataStruct();
			if (ds == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("update: " + mm.getMessage("engine.needPurePmt"));
			}

			names = ds.getFieldNames();
			fcount = names.length;
			valExps = new Expression[fcount];
			for (int f = 0; f < fcount; ++f) {
				valExps[f] = new Expression(ctx, "#" + (f + 1));
			}
		}

		String []opts = null;
		if (pks != null) {
			opts = new String[fcount];

			Next:
			for (int p = 0; p < pks.length; ++p) {
				String pkName = pks[p];
				for (int f = 0; f < fcount; ++f) {
					if (pkName.equalsIgnoreCase(names[f])) {
						opts[f] = "p";
						continue Next;
					}
				}

				MessageManager mm = EngineMessage.get();
				throw new RQException(mm.getMessage("function.paramError", "update", mm.getMessage("mfn.dbUpdateKey")));
			}

			if (option != null && option.indexOf('1') != -1) {
				if (opts[0] == null) {
					opts[0] = "a";
				} else {
					opts[0] += "a";
				}
			}
		} else if (option != null && option.indexOf('1') != -1) {
			opts = new String[fcount];
			opts[0] = "a";
		}

		if (cursor != null) {
			return db.update(cursor, tableName, names, opts, valExps, option, ctx);
		} else if (seq2 != null) {
			return db.update(seq1, seq2, tableName, names, opts, valExps, option, ctx);
		} else {
			return db.update(seq1, tableName, names, opts, valExps, option, ctx);
		}
	}
}