Examples with ClassLoaderSafeConnectorSplitSource io.trino.plugin.base.classloader.ClassLoaderSafeConnectorSplitSource used on opensource projects

Example 1 with ClassLoaderSafeConnectorSplitSource

use of io.trino.plugin.base.classloader.ClassLoaderSafeConnectorSplitSource in project trino by trinodb.

the class IcebergSplitManager method getSplits.

@Override
public ConnectorSplitSource getSplits(ConnectorTransactionHandle transaction, ConnectorSession session, ConnectorTableHandle handle, SplitSchedulingStrategy splitSchedulingStrategy, DynamicFilter dynamicFilter, Constraint constraint) {
    IcebergTableHandle table = (IcebergTableHandle) handle;
    if (table.getSnapshotId().isEmpty()) {
        if (table.isRecordScannedFiles()) {
            return new FixedSplitSource(ImmutableList.of(), ImmutableList.of());
        }
        return new FixedSplitSource(ImmutableList.of());
    }
    Table icebergTable = transactionManager.get(transaction, session.getIdentity()).getIcebergTable(session, table.getSchemaTableName());
    Duration dynamicFilteringWaitTimeout = getDynamicFilteringWaitTimeout(session);
    TableScan tableScan = icebergTable.newScan().useSnapshot(table.getSnapshotId().get());
    IcebergSplitSource splitSource = new IcebergSplitSource(table, tableScan, table.getMaxScannedFileSize(), dynamicFilter, dynamicFilteringWaitTimeout, constraint, typeManager, table.isRecordScannedFiles());
    return new ClassLoaderSafeConnectorSplitSource(splitSource, Thread.currentThread().getContextClassLoader());
}