[jbosstools-commits] JBoss Tools SVN: r36551 - branches/hibernatetools-multiversion2/plugins/org.jboss.tools.hibernate4_0/src/org/jboss/tools/hibernate4_0/console.

Author: dgeraskov Date: 2011-11-22 19:11:35 -0500 (Tue, 22 Nov 2011) New Revision: 36551 Added: branches/hibernatetools-multiversion2/plugins/org.jboss.tools.hibernate4_0/src/org/jboss/tools/hibernate4_0/console/ConsoleDatabaseCollectorImpl.java Log: Remove strategy usage Added: branches/hibernatetools-multiversion2/plugins/org.jboss.tools.hibernate4_0/src/org/jboss/tools/hibernate4_0/console/ConsoleDatabaseCollectorImpl.java =================================================================== --- branches/hibernatetools-multiversion2/plugins/org.jboss.tools.hibernate4_0/src/org/jboss/tools/hibernate4_0/console/ConsoleDatabaseCollectorImpl.java (rev 0) +++ branches/hibernatetools-multiversion2/plugins/org.jboss.tools.hibernate4_0/src/org/jboss/tools/hibernate4_0/console/ConsoleDatabaseCollectorImpl.java 2011-11-23 00:11:35 UTC (rev 36551) @@ -0,0 +1,59 @@ +/******************************************************************************* + * Copyright (c) 2011 Red Hat, Inc. + * Distributed under license by Red Hat, Inc. All rights reserved. + * This program is 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 + * + * Contributor: + * Red Hat, Inc. - initial API and implementation + ******************************************************************************/ +package org.jboss.tools.hibernate4_0.console; + +import java.util.HashMap; +import java.util.Iterator; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; +import java.util.Map.Entry; + +import org.hibernate.cfg.reveng.DefaultDatabaseCollector; +import org.hibernate.console.ext.api.ConsoleDatabaseCollector; +import org.hibernate.console.ext.api.ITable; +import org.hibernate.mapping.Table; +import org.jboss.tools.hibernate4_0.mapping.TableImpl; + +/** + * @author Dmitry Geraskov {geraskov(a)gmail.com} + * + */ +public class ConsoleDatabaseCollectorImpl implements ConsoleDatabaseCollector { + + private DefaultDatabaseCollector collector; + + private Map<String, List<ITable>> qualifiers = new HashMap<String, List<ITable>>(); + + public ConsoleDatabaseCollectorImpl(DefaultDatabaseCollector databaseCollector){ + this.collector = databaseCollector; + this.initializeTableMap(); + } + + @SuppressWarnings({ "unchecked", "rawtypes" }) + protected void initializeTableMap(){ + Iterator qualifierEntries = collector.getQualifierEntries(); + while (qualifierEntries.hasNext()) { + Entry<String, List<Table>> entry = (Entry<String, List<Table>>) qualifierEntries.next(); + List<ITable> list = new LinkedList<ITable>(); + for (Table table : entry.getValue()) { + list.add(new TableImpl(table)); + } + qualifiers.put(entry.getKey(), list); + } + } + + @Override + public Iterator<Entry<String, List<ITable>>> getQualifierEntries() { + return qualifiers.entrySet().iterator(); + } + +}