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

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.DataStruct;
import com.scudata.dm.FileObject;
import com.scudata.dm.Sequence;
import com.scudata.dm.cursor.ICursor;
import com.scudata.dm.cursor.MemoryCursor;
import com.scudata.dw.Cuboid;
import com.scudata.dw.IPhyTable;
import com.scudata.dw.PhyTable;
import com.scudata.dw.PhyTableGroup;
import com.scudata.expression.Expression;
import com.scudata.expression.IParam;
import com.scudata.expression.ParamInfo2;
import com.scudata.expression.PhyTableFunction;
import com.scudata.resources.EngineMessage;

/**
 * Ϊ???????Ԥ??????????
 * T.cuboid(C,Fi,??;y:Gi,??)
 * @author RunQian
 *
 */
public class CreateCuboid extends PhyTableFunction {
	public Object calculate(Context ctx) {
		if (table instanceof PhyTableGroup) {
			IPhyTable[] tables = ((PhyTableGroup)table).getTables();
			for (IPhyTable t : tables) {
				createCuboid((PhyTable) t, param, ctx);
			}
			return table;
		} else {
			PhyTable srcTable = (PhyTable)table;
			/*TableMetaData tmd = srcTable.getSupplementTable(false);
			
			if (tmd != null) {
				createCuboid(tmd, param, ctx);
			}*/
			return createCuboid(srcTable, param, ctx);
		}
	
	}
	
	private static Object createCuboid(PhyTable srcTable, IParam param,  Context ctx) {
		FileObject C;
		if (param == null) {
			try {
				return srcTable.deleteCuboid(null);
			} catch (IOException e) {
				throw new RQException(e.getMessage(), e);
			}
		} else if (param.isLeaf()) {
//			C = (String) param.getLeafExpression().getIdentifierName();
//			//delete
//			try {
//				return srcTable.deleteCuboid(C);
//			} catch (IOException e) {
//				throw new RQException(e.getMessage(), e);
//			}
			MessageManager mm = EngineMessage.get();
			throw new RQException("cuboid" + mm.getMessage("function.invalidParam"));
		}
		
		IParam sub0;
		IParam sub1 = null;
		if (param.getType() == IParam.Semicolon) {
			int size = param.getSubSize();
			if (size > 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("cuboid" + mm.getMessage("function.invalidParam"));
			}
			
			sub0 = param.getSub(0);
			if (sub0 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("cuboid" + mm.getMessage("function.invalidParam"));
			}
			
			sub1 = param.getSub(1);
		} else {
			sub0 = param;
		}
		
		//C = (String) sub0.getSub(0).getLeafExpression().getIdentifierName();
		C = (FileObject) sub0.getSub(0).getLeafExpression().calculate(ctx);
		sub0 = sub0.create(1, sub0.getSubSize());
		
		Expression []exps;
		String []names = null;
		Expression []newExps = null;
		String []newNames = null;
		ParamInfo2 pi0 = ParamInfo2.parse(sub0, "cuboid", true, false);
		exps = pi0.getExpressions1();
		names = new String[exps.length];
		int i = 0;
		for(Expression e : exps) {
			String s = e.getIdentifierName();
			names[i++] = s;
		}
		
		ParamInfo2 pi1 = null;
		if (sub1 != null) {
			pi1 = ParamInfo2.parse(sub1, "cuboid", true, false);
			newExps = pi1.getExpressions1();
			newNames = new String[newExps.length];
			i = 0;
			for(Expression e : newExps) {
				String s = e.getIdentifierName();
				newNames[i++] = s;
			}
		}
		
		//Ϊcuboid????
		//String dir = srcTable.getGroupTable().getFile().getAbsolutePath() + "_";
		//FileObject fo = new FileObject(dir + srcTable.getTableName() + Cuboid.CUBE_PREFIX + C);
		FileObject fo = C;
		if (fo.isExists())
		{
			//fo.delete();
			MessageManager mm = EngineMessage.get();
			throw new RQException(fo.getFileName() + mm.getMessage("dw.cuboidAlreadyExist"));
		}
		
		String cuboids[] = srcTable.getCuboids();
		if (cuboids != null) {
			for (String c : cuboids) {
				if(C.equals(c)) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("cuboid" + mm.getMessage("dw.cuboidAlreadyExist"));
				}
			}
		}
		
		int fcount = exps == null ? 0 : exps.length;
		if (newExps != null) fcount += newExps.length;
		ICursor cursor = new MemoryCursor(Cuboid.cgroups(sub0, sub1, srcTable, null, false, 0, null, ctx));
		
		//???浽??һ?????
		File file = fo.getLocalFile().file();
		String colNames[] = new String[fcount];
		int sbytes[] = new int[fcount];
		i = 0;
		for(String n : names) {
			colNames[i++] = "#" + n;
		}
		
		for(String s : newNames) {
			colNames[i++] = s;
		}
		Cuboid table = null;
		try {
			table = new Cuboid(file, colNames, sbytes, ctx, "cuboid", "cuboid",
					pi0.getExpressionStrs1(), pi1.getExpressionStrs1());
			table.save();
			table.close();
			table = new Cuboid(file, ctx);//???´?
			table.checkPassword("cuboid");
			
			Sequence data = cursor.peek(1);		
			if (data == null || data.length() <= 0) {
				return srcTable;
			}
			DataStruct ds = data.dataStruct();
			String fnames[] = ds.getFieldNames();
			System.arraycopy(colNames, 0, fnames, 0,  fcount);
			System.arraycopy(names, 0, fnames, 0,  names.length);
			
			table.append(cursor);
			table.setSrcCount(srcTable.getActualRecordCount());
			table.writeHeader();
			table.close();
			//srcTable.addCuboid(C);
		} catch (Exception e) {
			if (table != null) table.close();
			file.delete();
			throw new RQException(e.getMessage(), e);
		}
		return srcTable;
	}
}