com.scudata.dw.compress.BigDecimalColumn 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.dw.compress;

import java.io.IOException;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.ArrayList;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dw.BufferReader;
import com.scudata.resources.EngineMessage;

/**
 * ???????????
 * @author runqian
 *
 */
public class BigDecimalColumn extends Column {
	private static final int NULL = -1;
	
	private int scale = Integer.MIN_VALUE; // ?Ƿ????Ҫ??һ?£?
	
	// ???ݰ???洢??ÿ????Column.BLOCK_RECORD_COUNT????¼
	private ArrayList blockList = new ArrayList(1024);
	
	// ??????blockList???Ӧ???λ??, -1??ʾnull
	private ArrayList posList = new ArrayList(1024);
	private int lastRecordCount = Column.BLOCK_RECORD_COUNT; // ???һ??ļ?¼??
	private int nextPos = -1; // ??һ??д?뵽buffer?е?λ??
	
	// ?????¿?ʱʹ??buffer??buffer????ʱ????????1.5?????ȵ???buffer?????¼??д????????ʵ?ʵĴ?С???浽blockList??
	private byte []buffer = new byte[65536];

	public void addData(Object data) {
		if (lastRecordCount == Column.BLOCK_RECORD_COUNT) {
			if (blockList.size() > 0) {
				// ???¼??д??????????СΪʵ?ʴ?С
				byte []block = new byte[nextPos];
				System.arraycopy(buffer, 0, block, 0, nextPos);
				blockList.set(blockList.size() - 1, block);
			}
			
			int []posBlock = new int[Column.BLOCK_RECORD_COUNT];
			blockList.add(buffer);
			posList.add(posBlock);
			lastRecordCount = 0;
			nextPos = 0;
		}
		
		if (data instanceof BigDecimal) {
			BigDecimal decimal = (BigDecimal)data;
			int scale = decimal.scale();
			if (this.scale != scale) {
				if (scale != Integer.MIN_VALUE) {
					MessageManager mm = EngineMessage.get();
					throw new RQException(mm.getMessage("ds.colTypeDif"));
				} else {
					this.scale = scale;
				}
			}
			
			byte []bts = decimal.unscaledValue().toByteArray();
			int len = bts.length;
			int free = buffer.length - nextPos;
			
			// ??????һ??Ļ??????ռ䲻????????????һ???Ļ?????
			if (free < len) {
				byte []tmp = new byte[buffer.length * 3 / 2];
				System.arraycopy(buffer, 0, tmp, 0, buffer.length);
				buffer = tmp;
				blockList.set(blockList.size() - 1, tmp);
			}
			
			System.arraycopy(bts, 0, buffer, nextPos, len);
			int []posBlock = posList.get(posList.size() - 1);
			posBlock[lastRecordCount++] = nextPos;
			nextPos += len;
		} else if (data == null) {
			int []posBlock = posList.get(posList.size() - 1);
			posBlock[lastRecordCount++] = NULL;
		} else {
			// ???쳣
			MessageManager mm = EngineMessage.get();
			throw new RQException(mm.getMessage("ds.colTypeDif"));
		}
	}
	
	// ȡ??row?е?????
	public Object getData(int row) {
		// row?кţ???1??ʼ????
		row--;
		int b = row / Column.BLOCK_RECORD_COUNT;
		int index = row % Column.BLOCK_RECORD_COUNT;
		
		int []posBlock = posList.get(b);
		int startPos = posBlock[index];
		if (startPos == NULL) {
			return null;
		}
		
		byte []block = blockList.get(b);
		int endPos;
		
		// ?Ƿ????һ??
		if (b == blockList.size() - 1) {
			endPos = nextPos;
			for (int i = index + 1, end = lastRecordCount; i < end; ++i) {
				if (posBlock[i] != NULL) {
					endPos = posBlock[i];
					break;
				}
			}
		} else {
			endPos = block.length;
			for (int i = index + 1; i < Column.BLOCK_RECORD_COUNT; ++i) {
				if (posBlock[i] != NULL) {
					endPos = posBlock[i];
					break;
				}
			}
		}
		
		int len = endPos - startPos;
		byte []bts = new byte[len];
		System.arraycopy(block, startPos, bts, 0, len);
		BigInteger bigInt = new BigInteger(bts);
		return new BigDecimal(bigInt, scale);
	}
	
	public Column clone() {
		return new BigDecimalColumn();
	}

	public void appendData(BufferReader br) throws IOException {
		addData(br.readObject());
	}
}