Examples with ColumnChange org.apache.spark.sql.connector.catalog.TableChange.ColumnChange used on opensource projects

Example 1 with ColumnChange

use of org.apache.spark.sql.connector.catalog.TableChange.ColumnChange in project iceberg by apache.

the class SparkCatalog method alterTable.

@Override
public SparkTable alterTable(Identifier ident, TableChange... changes) throws NoSuchTableException {
    SetProperty setLocation = null;
    SetProperty setSnapshotId = null;
    SetProperty pickSnapshotId = null;
    List<TableChange> propertyChanges = Lists.newArrayList();
    List<TableChange> schemaChanges = Lists.newArrayList();
    for (TableChange change : changes) {
        if (change instanceof SetProperty) {
            SetProperty set = (SetProperty) change;
            if (TableCatalog.PROP_LOCATION.equalsIgnoreCase(set.property())) {
                setLocation = set;
            } else if ("current-snapshot-id".equalsIgnoreCase(set.property())) {
                setSnapshotId = set;
            } else if ("cherry-pick-snapshot-id".equalsIgnoreCase(set.property())) {
                pickSnapshotId = set;
            } else if ("sort-order".equalsIgnoreCase(set.property())) {
                throw new UnsupportedOperationException("Cannot specify the 'sort-order' because it's a reserved table " + "property. Please use the command 'ALTER TABLE ... WRITE ORDERED BY' to specify write sort-orders.");
            } else {
                propertyChanges.add(set);
            }
        } else if (change instanceof RemoveProperty) {
            propertyChanges.add(change);
        } else if (change instanceof ColumnChange) {
            schemaChanges.add(change);
        } else {
            throw new UnsupportedOperationException("Cannot apply unknown table change: " + change);
        }
    }
    try {
        Table table = load(ident).first();
        commitChanges(table, setLocation, setSnapshotId, pickSnapshotId, propertyChanges, schemaChanges);
        return new SparkTable(table, true);
    } catch (org.apache.iceberg.exceptions.NoSuchTableException e) {
        throw new NoSuchTableException(ident);
    }
}