MutableSchema.java

/* 
 *
 * SchemaCrawler
 * http://sourceforge.net/projects/schemacrawler
 * Copyright (c) 2000-2010, 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.build;


import java.util.ArrayList;
import java.util.List;

import schemacrawler.schema.ColumnDataType;
import schemacrawler.schema.Procedure;
import schemacrawler.schema.Schema;
import schemacrawler.schema.Table;

/**
 * Represents the database schema.
 * 
 * @author Sualeh Fatehi
 */
public class MutableSchema
  extends BaseNamedObjectMetadata
  implements Schema
{

  private static final long serialVersionUID = 3258128063743931187L;

  private final SchemaReference schemaRef;
  private final ColumnDataType[] columnDataTypes;
  private final Table[] tables;
  private final Procedure[] procedures;

  private MutableSchema(final MutableSchemaBuilder builder)
  {
    super(builder);
    this.schemaRef = builder.schemaRef;
    columnDataTypes = builder.getColumnDataTypes();
    tables = builder.getTables();
    procedures = builder.getProcedures();
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getCatalogName()
   */
  public String getCatalogName()
  {
    return schemaRef.getCatalogName();
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getColumnDataType(java.lang.String)
   */
  public ColumnDataType getColumnDataType(final String name)
  {
    for (ColumnDataType columnDataType: columnDataTypes)
    {
      if (columnDataType.getName().equals(name))
      {
        return columnDataType;
      }
    }
    return null;
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Database#getSystemColumnDataTypes()
   */
  public ColumnDataType[] getColumnDataTypes()
  {
    return columnDataTypes;
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getFullName()
   */
  @Override
  public String getFullName()
  {
    return schemaRef.getFullName();
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getProcedure(java.lang.String)
   */
  public Procedure getProcedure(final String name)
  {
    for (Procedure procedure: procedures)
    {
      if (procedure.getName().equals(name))
      {
        return procedure;
      }
    }
    return null;
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getProcedures()
   */
  public Procedure[] getProcedures()
  {
    return procedures;
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getSchemaName()
   */
  public String getSchemaName()
  {
    return schemaRef.getSchemaName();
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getTable(java.lang.String)
   */
  public Table getTable(final String name)
  {
    for (Table table: tables)
    {
      if (table.getName().equals(name))
      {
        return table;
      }
    }
    return null;
  }

  /**
   * {@inheritDoc}
   * 
   * @see schemacrawler.schema.Schema#getTables()
   */
  public Table[] getTables()
  {
    return tables;
  }

  public static class MutableSchemaBuilder
    extends BaseNamedObjectBuilder
  {

    private static final long serialVersionUID = 3258128063743931187L;

    private final SchemaReference schemaRef;
    private ColumnDataTypes columnDataTypes = new ColumnDataTypes();
    private List<Table> tables = new ArrayList<Table>();
    private List<Procedure> procedures = new ArrayList<Procedure>();

    public MutableSchemaBuilder()
    {
      this(new SchemaReference(null, null));
    }

    public MutableSchemaBuilder(final SchemaReference schemaRef)
    {
      super(schemaRef.getFullName());
      this.schemaRef = schemaRef;
    }

    public String getCatalogName()
    {
      return schemaRef.getCatalogName();
    }

    public ColumnDataType[] getColumnDataTypes()
    {
      return columnDataTypes.values()
        .toArray(new ColumnDataType[columnDataTypes.size()]);
    }

    /**
     * {@inheritDoc}
     * 
     * @see schemacrawler.schema.Schema#getProcedures()
     */
    public Procedure[] getProcedures()
    {
      return procedures.toArray(new Procedure[procedures.size()]);
    }

    public String getSchemaName()
    {
      return schemaRef.getSchemaName();
    }

    public Table[] getTables()
    {
      return tables.toArray(new Table[tables.size()]);
    }

    void addColumnDataType(final ColumnDataTypeMetadata columnDataType)
    {
      if (columnDataType != null)
      {
        columnDataTypes.add(columnDataType);
      }
    }

    void addProcedure(final MutableProcedure procedure)
    {
      procedures.add(procedure);
    }

    void addTable(final MutableTable table)
    {
      tables.add(table);
    }

    ColumnDataTypeMetadata lookupColumnDataTypeByType(final int type)
    {
      return columnDataTypes.lookupColumnDataTypeByType(type);
    }

    void removeProcedure(final Procedure procedure)
    {
      procedures.remove(procedure);
    }

    void removeTable(final Table table)
    {
      tables.remove(table);
    }

  }

}