Examples with ArrayMapping io.trino.plugin.postgresql.PostgreSqlConfig.ArrayMapping used on opensource projects

Example 1 with ArrayMapping

use of io.trino.plugin.postgresql.PostgreSqlConfig.ArrayMapping in project trino by trinodb.

the class PostgreSqlClient method arrayToTrinoType.

private Optional<ColumnMapping> arrayToTrinoType(ConnectorSession session, Connection connection, JdbcTypeHandle typeHandle) {
    checkArgument(typeHandle.getJdbcType() == Types.ARRAY, "Not array type");
    ArrayMapping arrayMapping = getArrayMapping(session);
    if (arrayMapping == DISABLED) {
        return Optional.empty();
    }
    // resolve and map base array element type
    JdbcTypeHandle baseElementTypeHandle = getArrayElementTypeHandle(connection, typeHandle);
    String baseElementTypeName = baseElementTypeHandle.getJdbcTypeName().orElseThrow(() -> new TrinoException(JDBC_ERROR, "Element type name is missing: " + baseElementTypeHandle));
    if (baseElementTypeHandle.getJdbcType() == Types.BINARY) {
        // https://github.com/pgjdbc/pgjdbc/pull/1184
        return Optional.empty();
    }
    Optional<ColumnMapping> baseElementMapping = toColumnMapping(session, connection, baseElementTypeHandle);
    if (arrayMapping == AS_ARRAY) {
        if (typeHandle.getArrayDimensions().isEmpty()) {
            return Optional.empty();
        }
        return baseElementMapping.map(elementMapping -> {
            ArrayType prestoArrayType = new ArrayType(elementMapping.getType());
            ColumnMapping arrayColumnMapping = arrayColumnMapping(session, prestoArrayType, elementMapping, baseElementTypeName);
            int arrayDimensions = typeHandle.getArrayDimensions().get();
            for (int i = 1; i < arrayDimensions; i++) {
                prestoArrayType = new ArrayType(prestoArrayType);
                arrayColumnMapping = arrayColumnMapping(session, prestoArrayType, arrayColumnMapping, baseElementTypeName);
            }
            return arrayColumnMapping;
        });
    }
    if (arrayMapping == AS_JSON) {
        return baseElementMapping.map(elementMapping -> arrayAsJsonColumnMapping(session, elementMapping));
    }
    throw new IllegalStateException("Unsupported array mapping type: " + arrayMapping);
}