DefaultTableCollector.java

/*******************************************************************************
 * Copyright (c) 2012 OpenLegacy Inc.
 * All rights reserved. This program and the accompanying materials 
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     OpenLegacy Inc. - initial API and implementation
 *******************************************************************************/
package org.openlegacy.terminal.modules.table;

import org.openlegacy.terminal.ScreenPojoFieldAccessor;
import org.openlegacy.terminal.TerminalSession;
import org.openlegacy.terminal.definitions.ScreenEntityDefinition;
import org.openlegacy.terminal.definitions.ScreenTableDefinition;
import org.openlegacy.terminal.services.ScreenEntitiesRegistry;
import org.openlegacy.terminal.table.ScreenTableCollector;
import org.openlegacy.terminal.utils.SimpleScreenPojoFieldAccessor;

import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import javax.inject.Inject;

public class DefaultTableCollector<T> implements ScreenTableCollector<T> {

	@Inject
	private ScreenEntitiesRegistry screenEntitiesRegistry;

	public List<T> collectAll(TerminalSession terminalSession, Class<?> screenEntityClass, Class<T> rowClass) {
		return collectAllInner(terminalSession, screenEntityClass, rowClass, false);
	}

	public List<T> collectOne(TerminalSession terminalSession, Class<?> screenEntityClass, Class<T> rowClass) {
		return collectAllInner(terminalSession, screenEntityClass, rowClass, true);
	}

	@SuppressWarnings("unchecked")
	private List<T> collectAllInner(TerminalSession terminalSession, Class<?> screenEntityClass, Class<T> rowClass,
			boolean firstOnly) {

		ScreenEntityDefinition screenEntityDefintion = screenEntitiesRegistry.get(screenEntityClass);
		Map<String, ScreenTableDefinition> tableDefinitionsMap = screenEntityDefintion.getTableDefinitions();
		Set<Entry<String, ScreenTableDefinition>> tableDefinitionEntries = tableDefinitionsMap.entrySet();

		Entry<String, ScreenTableDefinition> matchingDefinitionEntry = null;

		for (Entry<String, ScreenTableDefinition> tableDefinitionEntry : tableDefinitionEntries) {
			if (tableDefinitionEntry.getValue().getTableClass() == rowClass) {
				matchingDefinitionEntry = tableDefinitionEntry;
				break;
			}
		}

		if (matchingDefinitionEntry == null) {
			throw (new IllegalArgumentException(MessageFormat.format("Could not find table of type {0} in {1}", rowClass,
					screenEntityDefintion)));
		}

		ScreenTableDefinition matchingTableDefinition = matchingDefinitionEntry.getValue();
		int rowsCount = matchingTableDefinition.getEndRow() - matchingTableDefinition.getStartRow() + 1;

		List<T> allRows = new ArrayList<T>();

		List<T> rows = null;
		boolean cont = true;

		Object screenEntity = terminalSession.getEntity(screenEntityClass);
		ScreenPojoFieldAccessor fieldAccessor = new SimpleScreenPojoFieldAccessor(screenEntity);

		while (cont) {

			rows = (List<T>)fieldAccessor.getFieldValue(matchingDefinitionEntry.getKey());
			allRows.addAll(rows);

			// more rows exists
			if (rows.size() < rowsCount) {
				cont = false;
			} else {
				if (firstOnly) {
					break;
				}
				screenEntity = terminalSession.doAction(matchingTableDefinition.getNextScreenAction());
				fieldAccessor = new SimpleScreenPojoFieldAccessor(screenEntity);
			}
		}

		return allRows;
	}

}