• Home
• com.scudata.esproc
• esproc
• 20240823
• source code
• Cursor.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.scudata.dw.Cursor Maven / Gradle / Ivy

package com.scudata.dw;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import com.scudata.array.IArray;
import com.scudata.array.ObjectArray;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.BaseRecord;
import com.scudata.dm.Context;
import com.scudata.dm.DataStruct;
import com.scudata.dm.ObjectReader;
import com.scudata.dm.Record;
import com.scudata.dm.Sequence;
import com.scudata.dm.Table;
import com.scudata.dm.cursor.ICursor;
import com.scudata.dm.op.New;
import com.scudata.dm.op.Operation;
import com.scudata.dm.op.Select;
import com.scudata.dm.op.Switch;
import com.scudata.expression.Constant;
import com.scudata.expression.CurrentElement;
import com.scudata.expression.CurrentSeq;
import com.scudata.expression.ElementRef;
import com.scudata.expression.Expression;
import com.scudata.expression.FieldRef;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.expression.Moves;
import com.scudata.expression.Node;
import com.scudata.expression.UnknownSymbol;
import com.scudata.expression.fn.Between;
import com.scudata.expression.fn.gather.Top;
import com.scudata.expression.mfn.sequence.Contain;
import com.scudata.expression.mfn.sequence.Find;
import com.scudata.expression.mfn.sequence.PFind;
import com.scudata.expression.mfn.serial.Sbs;
import com.scudata.expression.operator.And;
import com.scudata.expression.operator.Assign;
import com.scudata.expression.operator.Comma;
import com.scudata.expression.operator.DotOperator;
import com.scudata.expression.operator.Equals;
import com.scudata.expression.operator.Greater;
import com.scudata.expression.operator.Not;
import com.scudata.expression.operator.NotEquals;
import com.scudata.expression.operator.NotGreater;
import com.scudata.expression.operator.NotSmaller;
import com.scudata.expression.operator.Or;
import com.scudata.expression.operator.Smaller;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;

/**
 * ?д???????α???
 * @author runqian
 *
 */
public class Cursor extends IDWCursor {
	private ColPhyTable table;//ԭ??
	private String []fields;//ȡ???ֶ?
	protected DataStruct ds;//???ݽṹ
	private String []sortedFields;//?????ֶ?
	private Expression filter;//???˱???ʽ
	
	// K:T	??K:T?????ϻ?????????K=T.find(K)??K??????ѡ???ֶ?
	private String []fkNames;
	private Sequence []codes;
	private String []opts;
	
	private IFilter []filters;//??????
	protected FindFilter []findFilters; // ??????switch???ֶ?
	private int []seqs; // colReaders??Ӧ???ֶκţ??????ֶο??ܲ?ѡ??
	
	private ColumnMetaData []columns;//?õ?????
	private BlockLinkReader rowCountReader;
	private BlockLinkReader []colReaders;
	private ObjectReader []segmentReaders;
	
	private int startBlock; // ????
	private int endBlock = -1; // ??????
	private int curBlock = 0;
	protected Sequence cache;
	
	private long prevRecordSeq = 0; // ǰһ????¼?????
	private int []findex; // ѡ???ֶζ?Ӧ???ֶκ?
	private ArrayList modifyRecords;
	private int mindex = 0;
	private int mcount = 0;
	
	// ͬ???ֶ?ʱ??Ҫ??????һ?ε?һ???????ڱ??εIJ???
	private Sequence appendData;
	private int appendIndex = 0;
	
	private boolean isClosed = false;
	private boolean isFirstSkip = true;
	private boolean isSegment = false;
	private Expression []exps;//????ʽ?ֶ?
	private Expression []expsBakup;
	private String []names;//ȡ????
	private TableGather []gathers;//??????????ȡ
	private boolean isField[];//true,???ֶΣ?false???DZ???ʽ
	private DataStruct tempDs;
	
	public Cursor() {
	}
	
	public Cursor(ColPhyTable table) {
		this(table, null);
	}
	
	public Cursor(ColPhyTable table, String []fields) {
		this(table, fields, null, null);
	}
	
	public Cursor(ColPhyTable table, String []fields, Expression filter, Context ctx) {
		this.table = table;
		this.fields = fields;
		this.filter = filter;
		this.ctx = ctx;
		
		init();
	}
	
	public Cursor(ColPhyTable table, String []fields, Expression filter, String []fkNames, 
			Sequence []codes, String []opts, Context ctx) {
		this.table = table;
		this.fields = fields;
		this.filter = filter;
		this.fkNames = fkNames;
		this.codes = codes;
		this.opts = opts;
		this.ctx = ctx;
		
		init();
	}
	
	public Cursor(ColPhyTable table, Expression []exps, String []names, Expression filter, Context ctx) {
		this.table = table;
		this.filter = filter;
		this.names = names;
		this.ctx = ctx;
		
		if (exps == null && names != null) {
			this.fields = names;
			this.names = null;
		}
		
		initExps(exps);
		init();
	}
	
	public Cursor(ColPhyTable table, Expression []exps, String []names, Expression filter, 
			String []fkNames, Sequence []codes, String []opts, Context ctx) {
		this.table = table;
		this.filter = filter;
		this.fkNames = fkNames;
		this.codes = codes;
		this.opts = opts;
		this.names = names;
		this.ctx = ctx;
		
		if (exps == null && names != null) {
			this.fields = names;
			this.names = null;
		}
		
		initExps(exps);		
		init();
	}
	
	public PhyTable getTableMetaData() {
		return table;
	}

	/**
	 * ???÷ֶ?startBlock??????endBlock??????
	 */
	public void setSegment(int startBlock, int endBlock) {
		isSegment = true;
		this.startBlock = startBlock;
		this.curBlock = startBlock;
		this.endBlock = endBlock;
		
		if (startBlock == 0 || startBlock >= endBlock) {
			return;
		}
		
		ColumnMetaData []columns = this.columns;
		BlockLinkReader rowCountReader = this.rowCountReader;
		int colCount = columns.length;
		long prevRecordSeq = 0;
		
		try {
			if (filters == null) {
				BlockLinkReader []colReaders = this.colReaders;
				ObjectReader []segmentReaders = new ObjectReader[colCount];
				for (int i = 0; i < colCount; ++i) {
					if (columns[i] != null) {
						segmentReaders[i] = columns[i].getSegmentReader();
					}
				}
				
				for (int i = 0; i < startBlock; ++i) {
					prevRecordSeq += rowCountReader.readInt32();
					for (int f = 0; f < colCount; ++f) {
						if (segmentReaders[f] != null) {
							segmentReaders[f].readLong40();
							if (columns[f].hasMaxMinValues()) {
								segmentReaders[f].skipObject();
								segmentReaders[f].skipObject();
								segmentReaders[f].skipObject();
							}
						}
					}
				}
				
				for (int f = 0; f < colCount; ++f) {
					if (segmentReaders[f] != null) {
						long pos = segmentReaders[f].readLong40();
						colReaders[f].seek(pos);
					}
				}
			} else {
				ObjectReader []segmentReaders = this.segmentReaders;
				for (int i = 0; i < startBlock; ++i) {
					prevRecordSeq += rowCountReader.readInt32();
					for (int f = 0; f < colCount; ++f) {
						if (segmentReaders[f] != null) {
							segmentReaders[f].readLong40();
							if (columns[f].hasMaxMinValues()) {
								segmentReaders[f].skipObject();
								segmentReaders[f].skipObject();
								segmentReaders[f].skipObject();
							}
						}
					}
				}
			}
			
			this.prevRecordSeq = prevRecordSeq;
			if (prevRecordSeq > 0 && mcount > 0) {
				// ????ҲҪ??Ӧ?????ֶ?
				ArrayList modifyRecords = this.modifyRecords;
				int mindex = 0;
				for (ModifyRecord r : modifyRecords) {
					if (r.getRecordSeq() <= prevRecordSeq) {
						mindex++;
					} else {
						break;
					}
				}
				
				this.mindex = mindex;
			}
			
			if (gathers != null) {
				for (TableGather gather : gathers) {
					if (gather != null) {
						gather.setSegment(startBlock, endBlock);
					}
				}
			}
		} catch (IOException e) {
			throw new RQException(e.getMessage(), e);
		}
	}
	
	/**
	 *  ͬ???ֶ?ʱ??Ҫ??????һ?ε?һ???????ڱ??εIJ???
	 */
	public void setAppendData(Sequence seq) {
		this.appendData = (Sequence) seq;
		if (seq != null && seq.length() > 0) {
			appendIndex = 1;
		}
	}
	
	public Sequence getAppendData() {
		return appendData;
	}
	
	/**
	 * ?ӱ???ʽ????ȡ?漰??????
	 * @param table
	 * @param node
	 * @return
	 */
	private static ColumnMetaData getColumn(ColPhyTable table, Node node) {
		if (node instanceof UnknownSymbol) {
			/**
			 * ֱ?Ӿ????ֶ?
			 */
			String keyName = ((UnknownSymbol)node).getName();
			return table.getColumn(keyName);
		} else if (node instanceof DotOperator && node.getLeft() instanceof CurrentElement && 
				node.getRight() instanceof FieldRef) {
			/**
			 *  ~.key??ʽ??
			 */
			FieldRef fieldNode = (FieldRef)node.getRight();
			String keyName = fieldNode.getName();
			return table.getColumn(keyName);
		} else if (node instanceof DotOperator && node.getRight() instanceof Sbs) {
			/**
			 * ?ź?k.sbs()
			 */
			Node left = node.getLeft();
			return getColumn(table, left);
		} else if (node instanceof com.scudata.expression.fn.math.And) {
			ColumnMetaData col = getColumn(table, ((Function) node).getParam().getSub(0).getLeafExpression().getHome());
			if (col == null) {
				col = getColumn(table, ((Function) node).getParam().getSub(1).getLeafExpression().getHome());
			}
			return col;
		} else {
			return null;
		}
	}
	
	private static Object combineAnd(Node node, Object left, Object right) {
		/*if (left instanceof ColumnsOr && right instanceof ColumnsOr) {
			return node;
		}
		if (left instanceof ColumnsOr) {
			left = ((ColumnsOr) left).getNode();
		}
		if (right instanceof ColumnsOr) {
			right = ((ColumnsOr) right).getNode();
		}*/
		
		if (left instanceof IFilter) {
			if (right instanceof IFilter) {
				IFilter f1 = (IFilter)left;
				IFilter f2 = (IFilter)right;
				if (f1.isSameColumn(f2)) {
					return new LogicAnd(f1,f2);
				} else {
					ArrayList