@Override public ConnectorPageSource createPageSource(ConnectorTransactionHandle transactionHandle, ConnectorSession session, ConnectorSplit split, List<ColumnHandle> columns) { return new RecordPageSource(recordSetProvider.getRecordSet(transactionHandle, session, split, columns)); } }
private static ConnectorTableHandle getTableHandle(ConnectorMetadata metadata, String name) { return metadata.getTableHandle(SESSION, new SchemaTableName("test", name)); } }
private static int totalRecords(RecordSet recordSet) { RecordCursor cursor = recordSet.cursor(); int result = 0; while (cursor.advanceNextPosition()) { result++; } return result; } }
@Override public ConnectorPageSource createPageSource(ConnectorTransactionHandle transactionHandle, ConnectorSession session, ConnectorSplit split, List<ColumnHandle> columns) { return new RecordPageSource(recordSetProvider.getRecordSet(transactionHandle, session, split, columns)); } }