All Downloads are FREE. Search and download functionalities are using the official Maven repository.

schemacrawler.tools.analysis.associations.CatalogWithAssociations Maven / Gradle / Ivy

/*
 * SchemaCrawler
 * http://sourceforge.net/projects/schemacrawler
 * Copyright (c) 2000-2015, Sualeh Fatehi.
 * This library is free software; you can redistribute it and/or modify it under
 * the terms
 * of the GNU Lesser General Public License as published by the Free Software
 * Foundation;
 * either version 2.1 of the License, or (at your option) any later version.
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License
 * along with this
 * library; if not, write to the Free Software Foundation, Inc., 59 Temple
 * Place, Suite 330,
 * Boston, MA 02111-1307, USA.
 */
package schemacrawler.tools.analysis.associations;


import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;

import schemacrawler.schema.Catalog;
import schemacrawler.schema.ColumnReference;
import schemacrawler.schema.Table;
import schemacrawler.schema.TableReference;
import schemacrawler.schemacrawler.BaseCatalogDecorator;

public final class CatalogWithAssociations
  extends BaseCatalogDecorator
{

  public static final Collection getWeakAssociations(final Table table)
  {
    if (table == null)
    {
      return null;
    }

    final SortedSet weakAssociations = table
      .getAttribute(WEAK_ASSOCIATIONS_KEY, new TreeSet());
    final List weakAssociationsList = new ArrayList<>(weakAssociations);
    Collections.sort(weakAssociationsList);
    return weakAssociationsList;
  }

  static void addWeakAssociationToTable(final TableReference table,
                                        final ColumnReference weakAssociation)
  {
    if (table != null && weakAssociation != null)
    {
      final SortedSet tableWeakAssociations = table
        .getAttribute(WEAK_ASSOCIATIONS_KEY, new TreeSet());
      tableWeakAssociations.add(weakAssociation);
      table.setAttribute(WEAK_ASSOCIATIONS_KEY, tableWeakAssociations);
    }
  }

  private static final long serialVersionUID = -3953296149824921463L;

  private static final String WEAK_ASSOCIATIONS_KEY = "schemacrawler.weak_associations";

  private final Collection weakAssociations;

  public CatalogWithAssociations(final Catalog catalog)
  {
    super(catalog);

    final List allTables = new ArrayList<>(catalog.getTables());
    final WeakAssociationsAnalyzer weakAssociationsAnalyzer = new WeakAssociationsAnalyzer(allTables);
    weakAssociations = weakAssociationsAnalyzer.analyzeTables();
  }

  public Collection getWeakAssociations()
  {
    return weakAssociations;
  }

}