schemacrawler.utility.JavaSqlTypes Maven / Gradle / Ivy

Go to download
/*
========================================================================
SchemaCrawler
http://www.schemacrawler.com
Copyright (c) 2000-2020, Sualeh Fatehi .
All rights reserved.
------------------------------------------------------------------------

SchemaCrawler 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.

SchemaCrawler and the accompanying materials are made available under
the terms of the Eclipse Public License v1.0, GNU General Public License
v3 or GNU Lesser General Public License v3.

You may elect to redistribute this code under any of these licenses.

The Eclipse Public License is available at:
http://www.eclipse.org/legal/epl-v10.html

The GNU General Public License v3 and the GNU Lesser General Public
License v3 are available at:
http://www.gnu.org/licenses/

========================================================================
*/

package schemacrawler.utility;


import static us.fatehi.utility.Utility.isBlank;

import java.sql.JDBCType;
import java.sql.SQLType;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import schemacrawler.schema.JavaSqlType;
import schemacrawler.schema.JavaSqlTypeGroup;

/**
 * Utility to work with java.sql.Types.
 *
 * @author Sualeh Fatehi
 */
public final class JavaSqlTypes
  implements Iterable
{

  private final Map javaSqlTypeMap;

  public JavaSqlTypes()
  {
    // Load default type mappings
    final TypeMap typeMap = new TypeMap();
    javaSqlTypeMap = new HashMap<>();
    for (final SQLType sqlType : JDBCType.values())
    {
      final Integer sqlTypeInt = sqlType.getVendorTypeNumber();
      final JavaSqlTypeGroup sqlTypeGroup =
        JavaSqlTypeGroup.valueOf(sqlTypeInt);
      final Class mappedClass = typeMap.get(sqlType.getName());
      final JavaSqlType javaSqlType =
        new JavaSqlType(sqlType, mappedClass, sqlTypeGroup);
      javaSqlTypeMap.put(sqlTypeInt, javaSqlType);
    }
  }

  /**
   * Lookup java.sql.Types type, and return more detailed information, including
   * the mapped Java class.
   *
   * @param typeName
   *   java.sql.Types type name
   * @return JavaSqlType type
   */
  public JavaSqlType getFromJavaSqlTypeName(final String typeName)
  {
    JavaSqlType sqlDataType = JavaSqlType.UNKNOWN;
    if (isBlank(typeName))
    {
      return sqlDataType;
    }

    for (final JavaSqlType javaSqlType : javaSqlTypeMap.values())
    {
      if (typeName.equals(javaSqlType.getName()))
      {
        sqlDataType = javaSqlType;
        break;
      }
    }
    return sqlDataType;
  }

  @Override
  public Iterator iterator()
  {
    return javaSqlTypeMap
      .values()
      .iterator();
  }

  public int size()
  {
    return javaSqlTypeMap.size();
  }

  @Override
  public String toString()
  {
    return javaSqlTypeMap.toString();
  }

  public JavaSqlType valueOf(final int key)
  {
    if (javaSqlTypeMap.containsKey(key))
    {
      return javaSqlTypeMap.get(key);
    }
    else
    {
      return JavaSqlType.UNKNOWN;
    }
  }

}