package foundation.io.file.pull;

import java.util.List;

import foundation.data.entity.Entity;
import foundation.data.entity.EntitySet;
import foundation.data.object.DataObject;
import foundation.io.object.Titles;

public class DataBucket {

	private int batchCount;
	private Titles titles;
	private List<EntitySet> dataList;
	private int pos;
	private DataObject dataObject;
	private EntitySet currentSet;
	
	public DataBucket(int batchCount) {
		this.batchCount = batchCount;
		
		pos = -1;
	}
	
	public Titles getTitles() {
		return titles;
	}

	public EntitySet getNext() {
		pos++;
		
		if (pos >= dataList.size()) {
			return null;
		}
		
		return dataList.get(pos);
	}

	public Entity addOneLine(int rowNo) {
		if (currentSet == null || currentSet.isFull()) {
			try {
				currentSet = dataObject.createTableEntitySet(batchCount);
			}
			catch (Exception e) {
				e.printStackTrace();
			}
		}
		
		return currentSet.append();
	}

	public void setDataObject(DataObject dataObject) {
		this.dataObject = dataObject;
	}

}