com.scudata.expression.mfn.record.Modify 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.record;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.BaseRecord;
import com.scudata.dm.Context;
import com.scudata.expression.ParamInfo2;
import com.scudata.expression.RecordFunction;
import com.scudata.resources.EngineMessage;

/**
 * ?޸ļ?¼???ֶ?ֵ
 * r.modify(xi:Fi,??) r.modify(r')
 * @author RunQian
 *
 */
public class Modify extends RecordFunction {
	public Object calculate(Context ctx) {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("modify" + mm.getMessage("function.missingParam"));
		}
		
		BaseRecord r = this.srcRecord;
		if (option != null && option.indexOf('r') != -1) {
			if (!param.isLeaf()) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("modify" + mm.getMessage("function.invalidParam"));
			}
			
			Object obj = param.getLeafExpression().calculate(ctx);
			if (!(obj instanceof BaseRecord)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("modify" + mm.getMessage("function.paramTypeError"));
			}
			
			r.paste((BaseRecord)obj, false);
		} else if (option != null && option.indexOf('f') != -1) {
			if (!param.isLeaf()) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("modify" + mm.getMessage("function.invalidParam"));
			}
			
			Object obj = param.getLeafExpression().calculate(ctx);
			if (!(obj instanceof BaseRecord)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("modify" + mm.getMessage("function.paramTypeError"));
			}
			
			r.paste((BaseRecord)obj, true);
		} else {
			ParamInfo2 pi = ParamInfo2.parse(param, "modify", true, false);
			Object []vals = pi.getValues1(ctx);
			String []names = pi.getExpressionStrs2();

			int count = vals.length;
			int prevIndex = -1;
			for (int i = 0; i < count; ++i) {
				if (names[i] == null) {
					prevIndex++;
				} else {
					prevIndex = r.getFieldIndex(names[i]);
					if (prevIndex < 0) {
						MessageManager mm = EngineMessage.get();
						throw new RQException(names[i] + mm.getMessage("ds.fieldNotExist"));
					}
				}
				r.set(prevIndex, vals[i]);
			}
		}

		return r;
	}
}