Examples with PrestoThriftNullableTableMetadata io.prestosql.plugin.thrift.api.PrestoThriftNullableTableMetadata used on opensource projects

Example 1 with PrestoThriftNullableTableMetadata

use of io.prestosql.plugin.thrift.api.PrestoThriftNullableTableMetadata in project hetu-core by openlookeng.

the class ThriftTpchService method getTableMetadata.

@Override
public final PrestoThriftNullableTableMetadata getTableMetadata(PrestoThriftSchemaTableName schemaTableName) {
    String schemaName = schemaTableName.getSchemaName();
    String tableName = schemaTableName.getTableName();
    if (!SCHEMAS.contains(schemaName) || TpchTable.getTables().stream().noneMatch(table -> table.getTableName().equals(tableName))) {
        return new PrestoThriftNullableTableMetadata(null);
    }
    TpchTable<?> tpchTable = TpchTable.getTable(schemaTableName.getTableName());
    List<PrestoThriftColumnMetadata> columns = new ArrayList<>();
    for (TpchColumn<? extends TpchEntity> column : tpchTable.getColumns()) {
        columns.add(new PrestoThriftColumnMetadata(column.getSimplifiedColumnName(), getTypeString(column), null, false));
    }
    List<Set<String>> indexableKeys = getIndexableKeys(schemaName, tableName);
    return new PrestoThriftNullableTableMetadata(new PrestoThriftTableMetadata(schemaTableName, columns, null, !indexableKeys.isEmpty() ? indexableKeys : null));
}

Example 2 with PrestoThriftNullableTableMetadata

use of io.prestosql.plugin.thrift.api.PrestoThriftNullableTableMetadata in project hetu-core by openlookeng.

the class ThriftMetadata method getTableMetadataInternal.

// this method makes actual thrift request and should be called only by cache load method
private Optional<ThriftTableMetadata> getTableMetadataInternal(SchemaTableName schemaTableName) {
    requireNonNull(schemaTableName, "schemaTableName is null");
    PrestoThriftNullableTableMetadata thriftTableMetadata = getTableMetadata(schemaTableName);
    if (thriftTableMetadata.getTableMetadata() == null) {
        return Optional.empty();
    }
    ThriftTableMetadata tableMetadata = new ThriftTableMetadata(thriftTableMetadata.getTableMetadata(), typeManager);
    if (!Objects.equals(schemaTableName, tableMetadata.getSchemaTableName())) {
        throw new PrestoException(THRIFT_SERVICE_INVALID_RESPONSE, "Requested and actual table names are different");
    }
    return Optional.of(tableMetadata);
}