@Override public ResultSet map(ResultSet resultSet) { ResultSetSchema schema = map(resultSet.getSchema()); MetricColumn column = schema.getColumn(ROW_NUM_COLUMN_NAME, MetricColumn.class).get(); int resultSetSize = resultSet.size(); List<Result> newResults = new ArrayList<>(resultSetSize); for (int i = 0; i < resultSetSize; i++) { newResults.add(rowNumMap(resultSet.get(i), column, i)); } ResultSet newResultSet = new ResultSet(schema, newResults); LOG.trace("Mapped resultSet: {} to new resultSet {}", resultSet, newResultSet); return newResultSet; }