Skip to content

Commit

Permalink
Change extend to PrestoDialect
Browse files Browse the repository at this point in the history
  • Loading branch information
xuyu committed Feb 8, 2025
1 parent a18c5d2 commit 23c29b1
Showing 1 changed file with 1 addition and 130 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,32 +18,12 @@

import org.apache.calcite.avatica.util.Casing;
import org.apache.calcite.config.NullCollation;
import org.apache.calcite.rel.type.RelDataType;
import org.apache.calcite.rel.type.RelDataTypeSystem;
import org.apache.calcite.rex.RexCall;
import org.apache.calcite.rex.RexNode;
import org.apache.calcite.rex.RexUtil;
import org.apache.calcite.sql.SqlCall;
import org.apache.calcite.sql.SqlDialect;
import org.apache.calcite.sql.SqlIntervalQualifier;
import org.apache.calcite.sql.SqlKind;
import org.apache.calcite.sql.SqlNode;
import org.apache.calcite.sql.SqlWriter;
import org.apache.calcite.sql.fun.SqlLibraryOperators;
import org.apache.calcite.sql.fun.SqlStdOperatorTable;
import org.apache.calcite.sql.parser.SqlParserPos;
import org.apache.calcite.sql.type.SqlTypeName;
import org.apache.calcite.sql.type.SqlTypeUtil;
import org.apache.calcite.util.RelToSqlConverterUtil;

import org.checkerframework.checker.nullness.qual.Nullable;

import static com.google.common.base.Preconditions.checkArgument;

/**
* A <code>SqlDialect</code> implementation for the Trino database.
*/
public class TrinoSqlDialect extends SqlDialect {
public class TrinoSqlDialect extends PrestoSqlDialect {
public static final Context DEFAULT_CONTEXT = SqlDialect.EMPTY_CONTEXT
.withDatabaseProduct(DatabaseProduct.TRINO)
.withIdentifierQuoteString("\"")
Expand All @@ -59,113 +39,4 @@ public TrinoSqlDialect(Context context) {
super(context);
}

@Override public boolean supportsApproxCountDistinct() {
return true;
}

@Override public boolean supportsCharSet() {
return false;
}

@Override public boolean requiresAliasForFromItems() {
return true;
}

@Override public boolean supportsTimestampPrecision() {
return false;
}

@Override public void unparseOffsetFetch(SqlWriter writer, @Nullable SqlNode offset,
@Nullable SqlNode fetch) {
unparseUsingLimit(writer, offset, fetch);
}

@Override public boolean supportsImplicitTypeCoercion(RexCall call) {
RexNode rexNode = call.getOperands().get(0);
return super.supportsImplicitTypeCoercion(call)
&& RexUtil.isLiteral(rexNode, false)
&& (rexNode.getType().getSqlTypeName() == SqlTypeName.VARCHAR
|| rexNode.getType().getSqlTypeName() == SqlTypeName.CHAR)
&& !SqlTypeUtil.isNumeric(call.type)
&& !SqlTypeUtil.isDate(call.type)
&& !SqlTypeUtil.isTimestamp(call.type);
}

/** Unparses offset/fetch using "OFFSET offset LIMIT fetch " syntax. */
private static void unparseUsingLimit(SqlWriter writer, @Nullable SqlNode offset,
@Nullable SqlNode fetch) {
checkArgument(fetch != null || offset != null);
unparseOffset(writer, offset);
unparseLimit(writer, fetch);
}

@Override public @Nullable SqlNode emulateNullDirection(SqlNode node,
boolean nullsFirst, boolean desc) {
return emulateNullDirectionWithIsNull(node, nullsFirst, desc);
}

@Override public boolean supportsAggregateFunction(SqlKind kind) {
switch (kind) {
case AVG:
case COUNT:
case CUBE:
case SUM:
case MIN:
case MAX:
case ROLLUP:
return true;
default:
break;
}
return false;
}

@Override public boolean supportsGroupByWithCube() {
return true;
}

@Override public boolean supportsNestedAggregations() {
return false;
}

@Override public boolean supportsGroupByWithRollup() {
return true;
}

@Override public CalendarPolicy getCalendarPolicy() {
return CalendarPolicy.SHIFT;
}

@Override public @Nullable SqlNode getCastSpec(RelDataType type) {
return super.getCastSpec(type);
}

@Override public void unparseCall(SqlWriter writer, SqlCall call,
int leftPrec, int rightPrec) {
if (call.getOperator() == SqlStdOperatorTable.SUBSTRING) {
RelToSqlConverterUtil.specialOperatorByName("SUBSTR")
.unparse(writer, call, 0, 0);
} else if (call.getOperator() == SqlStdOperatorTable.APPROX_COUNT_DISTINCT) {
RelToSqlConverterUtil.specialOperatorByName("APPROX_DISTINCT")
.unparse(writer, call, 0, 0);
} else {
switch (call.getKind()) {
case CHAR_LENGTH:
SqlCall lengthCall = SqlLibraryOperators.LENGTH
.createCall(SqlParserPos.ZERO, call.getOperandList());
super.unparseCall(writer, lengthCall, leftPrec, rightPrec);
break;
default:
// Current impl is same with Postgresql.
PostgresqlSqlDialect.DEFAULT.unparseCall(writer, call, leftPrec, rightPrec);
}
}
}

@Override public void unparseSqlIntervalQualifier(SqlWriter writer,
SqlIntervalQualifier qualifier, RelDataTypeSystem typeSystem) {
// Current impl is same with MySQL.
MysqlSqlDialect.DEFAULT.unparseSqlIntervalQualifier(writer, qualifier, typeSystem);
}

}

0 comments on commit 23c29b1

Please sign in to comment.