Class TrinoMetadata

src/main/java/org/apache/paimon/trino/TrinoMetadata.java:101–755

Trino ConnectorMetadata.

/** Trino {@link ConnectorMetadata}. */
public class TrinoMetadata implements ConnectorMetadata {
    private static final String TAG_PREFIX = "tag-";

    protected final TrinoCatalog catalog;

    public TrinoMetadata(TrinoCatalog catalog) {
        this.catalog = catalog;
    }

    public TrinoCatalog catalog() {
        return catalog;
    }

    @Override
    public Optional<ConnectorTableLayout> getInsertLayout(
            ConnectorSession session, ConnectorTableHandle tableHandle) {
        TrinoTableHandle trinoTableHandle = (TrinoTableHandle) tableHandle;
        Table table = trinoTableHandle.table(catalog);
        if (!(table instanceof FileStoreTable)) {
            throw new IllegalArgumentException(table.getClass() + " is not supported");
        }
        FileStoreTable storeTable = (FileStoreTable) table;
        BucketMode bucketMode = storeTable.bucketMode();
        switch (bucketMode) {
            case HASH_FIXED:
                try {
                    return Optional.of(
                            new ConnectorTableLayout(
                                    new TrinoPartitioningHandle(
                                            InstantiationUtil.serializeObject(storeTable.schema())),
                                    storeTable.schema().bucketKeys(),
                                    false));
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            case BUCKET_UNAWARE:
                return Optional.empty();
            default:
                throw new IllegalArgumentException("Unsupported table bucket mode: " + bucketMode);
        }
    }

    @Override
    public ConnectorOutputTableHandle beginCreateTable(
            ConnectorSession session,
            ConnectorTableMetadata tableMetadata,
            Optional<ConnectorTableLayout> layout,
            RetryMode retryMode,
            boolean replace) {
        if (replace) {
            throw new UnsupportedOperationException("Create or replace table is not supported.");
        }
        createTable(session, tableMetadata, SaveMode.FAIL);
        return getTableHandle(session, tableMetadata.getTable(), Collections.emptyMap());
    }

    @Override
    public Optional<ConnectorOutputMetadata> finishCreateTable(