com.hazelcast.org.apache.calcite.sql.SqlDialectFactoryImpl Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to you under the Apache License, Version 2.0
* (the "License"); you may not use this file except in com.hazelcast.com.liance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.com.hazelcast.org.licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.org.apache.calcite.sql;
import com.hazelcast.org.apache.calcite.avatica.util.Casing;
import com.hazelcast.org.apache.calcite.config.NullCollation;
import com.hazelcast.org.apache.calcite.sql.dialect.AccessSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.AnsiSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.BigQuerySqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.CalciteSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.ClickHouseSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.Db2SqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.DerbySqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.FirebirdSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.H2SqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.HiveSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.HsqldbSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.InfobrightSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.InformixSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.IngresSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.InterbaseSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.JethroDataSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.LucidDbSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.MssqlSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.MysqlSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.NeoviewSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.NetezzaSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.OracleSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.ParaccelSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.PhoenixSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.PostgresqlSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.RedshiftSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.SnowflakeSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.SparkSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.SybaseSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.TeradataSqlDialect;
import com.hazelcast.org.apache.calcite.sql.dialect.VerticaSqlDialect;
import com.hazelcast.org.slf4j.Logger;
import com.hazelcast.org.slf4j.LoggerFactory;
import java.sql.DatabaseMetaData;
import java.sql.SQLException;
import java.util.Locale;
/**
* The default implementation of a SqlDialectFactory.
*/
public class SqlDialectFactoryImpl implements SqlDialectFactory {
private static final Logger LOGGER = LoggerFactory.getLogger(SqlDialectFactoryImpl.class);
public static final SqlDialectFactoryImpl INSTANCE = new SqlDialectFactoryImpl();
private final JethroDataSqlDialect.JethroInfoCache jethroCache =
JethroDataSqlDialect.createCache();
public SqlDialect create(DatabaseMetaData databaseMetaData) {
String databaseProductName;
int databaseMajorVersion;
int databaseMinorVersion;
String databaseVersion;
try {
databaseProductName = databaseMetaData.getDatabaseProductName();
databaseMajorVersion = databaseMetaData.getDatabaseMajorVersion();
databaseMinorVersion = databaseMetaData.getDatabaseMinorVersion();
databaseVersion = databaseMetaData.getDatabaseProductVersion();
} catch (SQLException e) {
throw new RuntimeException("while detecting database product", e);
}
final String upperProductName =
databaseProductName.toUpperCase(Locale.ROOT).trim();
final String quoteString = getIdentifierQuoteString(databaseMetaData);
final NullCollation nullCollation = getNullCollation(databaseMetaData);
final Casing unquotedCasing = getCasing(databaseMetaData, false);
final Casing quotedCasing = getCasing(databaseMetaData, true);
final boolean caseSensitive = isCaseSensitive(databaseMetaData);
final SqlDialect.Context c = SqlDialect.EMPTY_CONTEXT
.withDatabaseProductName(databaseProductName)
.withDatabaseMajorVersion(databaseMajorVersion)
.withDatabaseMinorVersion(databaseMinorVersion)
.withDatabaseVersion(databaseVersion)
.withIdentifierQuoteString(quoteString)
.withUnquotedCasing(unquotedCasing)
.withQuotedCasing(quotedCasing)
.withCaseSensitive(caseSensitive)
.withNullCollation(nullCollation);
switch (upperProductName) {
case "ACCESS":
return new AccessSqlDialect(c);
case "APACHE DERBY":
return new DerbySqlDialect(c);
case "CLICKHOUSE":
return new ClickHouseSqlDialect(c);
case "DBMS:CLOUDSCAPE":
return new DerbySqlDialect(c);
case "HIVE":
return new HiveSqlDialect(c);
case "INGRES":
return new IngresSqlDialect(c);
case "INTERBASE":
return new InterbaseSqlDialect(c);
case "JETHRODATA":
return new JethroDataSqlDialect(
c.withJethroInfo(jethroCache.get(databaseMetaData)));
case "LUCIDDB":
return new LucidDbSqlDialect(c);
case "ORACLE":
return new OracleSqlDialect(c);
case "PHOENIX":
return new PhoenixSqlDialect(c);
case "MYSQL (INFOBRIGHT)":
return new InfobrightSqlDialect(c);
case "MYSQL":
return new MysqlSqlDialect(
c.withDataTypeSystem(MysqlSqlDialect.MYSQL_TYPE_SYSTEM));
case "REDSHIFT":
return new RedshiftSqlDialect(c);
case "SNOWFLAKE":
return new SnowflakeSqlDialect(c);
case "SPARK":
return new SparkSqlDialect(c);
}
// Now the fuzzy matches.
if (databaseProductName.startsWith("DB2")) {
return new Db2SqlDialect(c);
} else if (upperProductName.contains("FIREBIRD")) {
return new FirebirdSqlDialect(c);
} else if (databaseProductName.startsWith("Informix")) {
return new InformixSqlDialect(c);
} else if (upperProductName.contains("NETEZZA")) {
return new NetezzaSqlDialect(c);
} else if (upperProductName.contains("PARACCEL")) {
return new ParaccelSqlDialect(c);
} else if (databaseProductName.startsWith("HP Neoview")) {
return new NeoviewSqlDialect(c);
} else if (upperProductName.contains("POSTGRE")) {
return new PostgresqlSqlDialect(
c.withDataTypeSystem(PostgresqlSqlDialect.POSTGRESQL_TYPE_SYSTEM));
} else if (upperProductName.contains("SQL SERVER")) {
return new MssqlSqlDialect(c);
} else if (upperProductName.contains("SYBASE")) {
return new SybaseSqlDialect(c);
} else if (upperProductName.contains("TERADATA")) {
return new TeradataSqlDialect(c);
} else if (upperProductName.contains("HSQL")) {
return new HsqldbSqlDialect(c);
} else if (upperProductName.contains("H2")) {
return new H2SqlDialect(c);
} else if (upperProductName.contains("VERTICA")) {
return new VerticaSqlDialect(c);
} else if (upperProductName.contains("SNOWFLAKE")) {
return new SnowflakeSqlDialect(c);
} else if (upperProductName.contains("SPARK")) {
return new SparkSqlDialect(c);
} else {
return new AnsiSqlDialect(c);
}
}
private Casing getCasing(DatabaseMetaData databaseMetaData, boolean quoted) {
try {
if (quoted
? databaseMetaData.storesUpperCaseQuotedIdentifiers()
: databaseMetaData.storesUpperCaseIdentifiers()) {
return Casing.TO_UPPER;
} else if (quoted
? databaseMetaData.storesLowerCaseQuotedIdentifiers()
: databaseMetaData.storesLowerCaseIdentifiers()) {
return Casing.TO_LOWER;
} else if (quoted
? (databaseMetaData.storesMixedCaseQuotedIdentifiers()
|| databaseMetaData.supportsMixedCaseQuotedIdentifiers())
: (databaseMetaData.storesMixedCaseIdentifiers()
|| databaseMetaData.supportsMixedCaseIdentifiers())) {
return Casing.UNCHANGED;
} else {
return Casing.UNCHANGED;
}
} catch (SQLException e) {
throw new IllegalArgumentException("cannot deduce casing", e);
}
}
private boolean isCaseSensitive(DatabaseMetaData databaseMetaData) {
try {
return databaseMetaData.supportsMixedCaseIdentifiers()
|| databaseMetaData.supportsMixedCaseQuotedIdentifiers();
} catch (SQLException e) {
throw new IllegalArgumentException("cannot deduce case-sensitivity", e);
}
}
private NullCollation getNullCollation(DatabaseMetaData databaseMetaData) {
try {
if (databaseMetaData.nullsAreSortedAtEnd()) {
return NullCollation.LAST;
} else if (databaseMetaData.nullsAreSortedAtStart()) {
return NullCollation.FIRST;
} else if (databaseMetaData.nullsAreSortedLow()) {
return NullCollation.LOW;
} else if (databaseMetaData.nullsAreSortedHigh()) {
return NullCollation.HIGH;
} else if (isBigQuery(databaseMetaData)) {
return NullCollation.LOW;
} else {
throw new IllegalArgumentException("cannot deduce null collation");
}
} catch (SQLException e) {
throw new IllegalArgumentException("cannot deduce null collation", e);
}
}
private static boolean isBigQuery(DatabaseMetaData databaseMetaData)
throws SQLException {
return databaseMetaData.getDatabaseProductName()
.equals("Google Big Query");
}
private String getIdentifierQuoteString(DatabaseMetaData databaseMetaData) {
try {
return databaseMetaData.getIdentifierQuoteString();
} catch (SQLException e) {
throw new IllegalArgumentException("cannot deduce identifier quote string", e);
}
}
/** Returns a basic dialect for a given product, or null if none is known. */
static SqlDialect simple(SqlDialect.DatabaseProduct databaseProduct) {
switch (databaseProduct) {
case ACCESS:
return AccessSqlDialect.DEFAULT;
case BIG_QUERY:
return BigQuerySqlDialect.DEFAULT;
case CALCITE:
return CalciteSqlDialect.DEFAULT;
case CLICKHOUSE:
return ClickHouseSqlDialect.DEFAULT;
case DB2:
return Db2SqlDialect.DEFAULT;
case DERBY:
return DerbySqlDialect.DEFAULT;
case FIREBIRD:
return FirebirdSqlDialect.DEFAULT;
case H2:
return H2SqlDialect.DEFAULT;
case HIVE:
return HiveSqlDialect.DEFAULT;
case HSQLDB:
return HsqldbSqlDialect.DEFAULT;
case INFOBRIGHT:
return InfobrightSqlDialect.DEFAULT;
case INFORMIX:
return InformixSqlDialect.DEFAULT;
case INGRES:
return IngresSqlDialect.DEFAULT;
case INTERBASE:
return InterbaseSqlDialect.DEFAULT;
case JETHRO:
throw new RuntimeException("Jethro does not support simple creation");
case LUCIDDB:
return LucidDbSqlDialect.DEFAULT;
case MSSQL:
return MssqlSqlDialect.DEFAULT;
case MYSQL:
return MysqlSqlDialect.DEFAULT;
case NEOVIEW:
return NeoviewSqlDialect.DEFAULT;
case NETEZZA:
return NetezzaSqlDialect.DEFAULT;
case ORACLE:
return OracleSqlDialect.DEFAULT;
case PARACCEL:
return ParaccelSqlDialect.DEFAULT;
case PHOENIX:
return PhoenixSqlDialect.DEFAULT;
case POSTGRESQL:
return PostgresqlSqlDialect.DEFAULT;
case REDSHIFT:
return RedshiftSqlDialect.DEFAULT;
case SYBASE:
return SybaseSqlDialect.DEFAULT;
case TERADATA:
return TeradataSqlDialect.DEFAULT;
case VERTICA:
return VerticaSqlDialect.DEFAULT;
case SPARK:
return SparkSqlDialect.DEFAULT;
case SQLSTREAM:
case UNKNOWN:
default:
return null;
}
}
}