Skip to content

Commit 5442131

Browse files
committed
try fixing trinodb#8404
1 parent fcd6b8e commit 5442131

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

plugin/trino-druid/src/main/java/io/trino/plugin/druid/DruidJdbcClient.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import io.trino.spi.connector.ConnectorSession;
3434
import io.trino.spi.connector.ConnectorTableMetadata;
3535
import io.trino.spi.connector.SchemaTableName;
36+
import io.trino.spi.type.TimestampType;
3637
import io.trino.spi.type.Type;
3738

3839
import javax.inject.Inject;
@@ -56,9 +57,11 @@
5657
import static com.google.common.collect.Iterables.getOnlyElement;
5758
import static io.trino.plugin.jdbc.JdbcErrorCode.JDBC_ERROR;
5859
import static io.trino.plugin.jdbc.StandardColumnMappings.defaultVarcharColumnMapping;
60+
import static io.trino.plugin.jdbc.StandardColumnMappings.timestampColumnMappingUsingSqlTimestampWithRounding;
5961
import static io.trino.plugin.jdbc.StandardColumnMappings.varcharColumnMapping;
6062
import static io.trino.spi.StandardErrorCode.NOT_SUPPORTED;
6163
import static io.trino.spi.type.VarcharType.createUnboundedVarcharType;
64+
import static io.trino.spi.type.TimestampType.createTimestampType;
6265

6366
public class DruidJdbcClient
6467
extends BaseJdbcClient
@@ -148,6 +151,10 @@ public Optional<ColumnMapping> toColumnMapping(ConnectorSession session, Connect
148151
return Optional.of(varcharColumnMapping(createUnboundedVarcharType(), true));
149152
}
150153
return Optional.of(defaultVarcharColumnMapping(columnSize, true));
154+
case Types.TIMESTAMP:
155+
TimestampType timestampType = createTimestampType(3);
156+
// TODO: use {@link #timestampColumnMapping} when https://issues.apache.org/jira/browse/CALCITE-1630 gets resolved
157+
return Optional.of(timestampColumnMappingUsingSqlTimestampWithRounding(timestampType));
151158
}
152159
// TODO implement proper type mapping
153160
return legacyColumnMapping(session, connection, typeHandle);

0 commit comments

Comments
 (0)