Skip to content

Commit 1e372a3

Browse files
committed
polish code
1 parent 3ff71a5 commit 1e372a3

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/SampleOperator.java

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -27,38 +27,37 @@
2727

2828
public class SampleOperator implements Operator {
2929

30-
private boolean finished;
30+
public record Factory(double probability, int seed) implements OperatorFactory {
31+
32+
@Override
33+
public SampleOperator get(DriverContext driverContext) {
34+
return new SampleOperator(probability, seed);
35+
}
36+
37+
@Override
38+
public String describe() {
39+
return "SampleOperator[probability = " + probability + ", seed = " + seed + "]";
40+
}
41+
}
42+
3143
private final Deque<Page> outputPages;
3244
private final RandomSamplingQuery.RandomSamplingIterator randomSamplingIterator;
45+
private boolean finished;
3346

3447
private int pagesProcessed = 0;
3548
private int rowsReceived = 0;
3649
private int rowsEmitted = 0;
37-
3850
private long collectNanos;
3951
private long emitNanos;
4052

41-
public SampleOperator(double probability, int seed) {
53+
private SampleOperator(double probability, int seed) {
4254
finished = false;
4355
outputPages = new LinkedList<>();
4456
SplittableRandom random = new SplittableRandom(seed);
4557
randomSamplingIterator = new RandomSamplingQuery.RandomSamplingIterator(Integer.MAX_VALUE, probability, random::nextInt);
4658
randomSamplingIterator.nextDoc();
4759
}
4860

49-
public record Factory(double probability, int seed) implements OperatorFactory {
50-
51-
@Override
52-
public SampleOperator get(DriverContext driverContext) {
53-
return new SampleOperator(probability, seed);
54-
}
55-
56-
@Override
57-
public String describe() {
58-
return "SampleOperator[probability = " + probability + ", seed = " + seed + "]";
59-
}
60-
}
61-
6261
/**
6362
* whether the given operator can accept more input pages
6463
*/

0 commit comments

Comments
 (0)