package foundation.data.getter;

import java.sql.ResultSet;

import foundation.data.entity.EntityNode;
import foundation.data.entity.EntityTree;
import foundation.data.meta.field.FieldsRuntime;

public class EntityTreeGetter extends DataGetter {

	private EntityTree entityTree;
	
	public EntityTreeGetter(FieldsRuntime tableMeta) {
		this.fieldsMeta = tableMeta;
	}
	
	protected void loadData(ResultSet rslt) throws Exception {
		if (rslt == null) {
			return;
		}

		entityTree = new EntityTree(fieldsMeta);
		
		int cnt = entityTree.getFieldCount();
		EntityNode node; Object obj;
		
		while (rslt.next()) {
			node = new EntityNode(fieldsMeta);
			
			for (int i = 0; i < cnt; i++) {
				obj = rslt.getObject(i + 1);
				node.set(i, obj);
			}
			
			entityTree.append(node);
		}
		
		entityTree.initRelation();
	}

	public EntityTree getEntityTree() {
		return entityTree;
	}

}