class PrometheusRecordSetProvider implements ConnectorRecordSetProvider {

    @Override
    public RecordSet getRecordSet(ConnectorTransactionHandle transaction, ConnectorSession session, ConnectorSplit split, ConnectorTableHandle table, List<? extends ColumnHandle> columns)
    {
        PrometheusSplit prometheusSplit = (PrometheusSplit) split;

        ImmutableList.Builder<PrometheusColumnHandle> handles = ImmutableList.builder();
        for (ColumnHandle handle : columns) {
            handles.add((PrometheusColumnHandle) handle);
        }

        return new PrometheusRecordSet(prometheusClient, prometheusSplit, handles.build());
    }

    @Inject public  PrometheusRecordSetProvider(PrometheusClient prometheusClient);

}