SCViewer/com.minres.scviewer.database.sqlite/src/com/minres/scviewer/database/sqlite/Stream.java

package com.minres.scviewer.database.sqlite;

import java.beans.IntrospectionException;
import java.lang.reflect.InvocationTargetException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;

import com.minres.scviewer.database.ITrDb;
import com.minres.scviewer.database.ITrGenerator;
import com.minres.scviewer.database.ITrStream;
import com.minres.scviewer.database.ITransaction;
import com.minres.scviewer.database.sqlite.db.SQLiteDatabaseSelectHandler;
import com.minres.scviewer.database.sqlite.tables.ScvGenerator;
import com.minres.scviewer.database.sqlite.tables.ScvStream;
import com.minres.scviewer.database.sqlite.tables.ScvTx;

public class Stream extends HierNode implements ITrStream {

	private String fullName;
	private SQLiteDb db;
	private ScvStream scvStream;
	private HashMap<Integer, Generator> generators;
	
	private List<ITransaction> transactions;
	
	public Stream(SQLiteDb trSQLiteDb, ScvStream scvStream) {
		super(scvStream.getName());
		fullName=scvStream.getName();
		this.scvStream=scvStream;
		db=trSQLiteDb;
	}

	@Override
	public String getFullName() {
		return fullName;
	}

	@Override
	public Long getId() {
		return (long) scvStream.getId();
	}

	@Override
	public String getKind() {
		return scvStream.getKind();
	}

	@Override
	public SQLiteDb getDb() {
		return db;
	}

	@Override
	public List<ITrGenerator> getGenerators() {
		if(generators==null){
			SQLiteDatabaseSelectHandler<ScvGenerator> handler = new SQLiteDatabaseSelectHandler<ScvGenerator>(
					ScvGenerator.class, db.getDb(), "stream="+scvStream.getId());
			generators=new HashMap<Integer, Generator>();
			try {
				for(ScvGenerator scvGenerator:handler.selectObjects()){
					generators.put(scvGenerator.getId(), new Generator(this, scvGenerator));
				}
			} catch (SecurityException | IllegalArgumentException | InstantiationException | IllegalAccessException
					| InvocationTargetException | SQLException | IntrospectionException e) {
				e.printStackTrace();
			}
		}
		return new ArrayList<ITrGenerator>(generators.values());
	}

	@Override
	public List<ITransaction> getTransactions() {
		checkTransactions();
		return transactions;
	}

	@Override
	public ITransaction getTransactionById(long id) {
		checkTransactions();
		for(ITransaction trans:transactions){
			if(trans.getId()==id)
				return trans;
		}
		return null;
	}

	protected void checkTransactions() {
		if(transactions==null){
			if(generators==null) getGenerators();
			SQLiteDatabaseSelectHandler<ScvTx> handler = new SQLiteDatabaseSelectHandler<ScvTx>(ScvTx.class, db.getDb(),
					"stream="+scvStream.getId());
			transactions=new ArrayList<ITransaction>();
			try {
				for(ScvTx scvTx:handler.selectObjects()){
					transactions.add(new Transaction(this, generators.get(scvTx.getGenerator()), scvTx));
				}
			} catch (SecurityException | IllegalArgumentException | InstantiationException | IllegalAccessException
					| InvocationTargetException | SQLException | IntrospectionException e) {
				e.printStackTrace();
			}
		}
	}

}