|
17 | 17 | import org.elasticsearch.compute.ann.Fixed; |
18 | 18 | import org.elasticsearch.compute.ann.Position; |
19 | 19 | import org.elasticsearch.compute.data.BytesRefBlock; |
20 | | -import org.elasticsearch.compute.operator.BreakingBytesRefBuilder; |
21 | 20 | import org.elasticsearch.compute.operator.EvalOperator; |
22 | 21 | import org.elasticsearch.xpack.esql.core.expression.Expression; |
23 | 22 | import org.elasticsearch.xpack.esql.core.tree.NodeInfo; |
|
31 | 30 |
|
32 | 31 | import java.io.IOException; |
33 | 32 | import java.net.InetAddress; |
34 | | -import java.util.ArrayList; |
35 | 33 | import java.util.Arrays; |
36 | 34 | import java.util.List; |
37 | | -import java.util.stream.Collectors; |
38 | 35 |
|
39 | 36 | import static org.elasticsearch.compute.ann.Fixed.Scope.THREAD_LOCAL; |
40 | 37 |
|
@@ -121,15 +118,15 @@ public EvalOperator.ExpressionEvaluator.Factory toEvaluator(ToEvaluator toEvalua |
121 | 118 | var internalNetworksEvaluatorSupplier = toEvaluator.apply(internalNetworks); |
122 | 119 | return new NetworkDirectionEvaluator.Factory( |
123 | 120 | source(), |
| 121 | + context -> new BytesRef(), |
124 | 122 | sourceIpEvaluatorSupplier, |
125 | 123 | destinationIpEvaluatorSupplier, |
126 | | - internalNetworksEvaluatorSupplier, |
127 | | - context -> new BreakingBytesRefBuilder(context.breaker(), "network_direction") |
| 124 | + internalNetworksEvaluatorSupplier |
128 | 125 | ); |
129 | 126 | } |
130 | 127 |
|
131 | 128 | @Evaluator |
132 | | - static boolean process(BytesRef sourceIp, BytesRef destinationIp, @Position int position, BytesRefBlock internalNetworks, @Fixed(includeInToString=false, scope=THREAD_LOCAL) BreakingBytesRefBuilder scratch) { |
| 129 | + static boolean process(@Fixed(includeInToString=false, scope=THREAD_LOCAL) BytesRef scratch, BytesRef sourceIp, BytesRef destinationIp, @Position int position, BytesRefBlock internalNetworks) { |
133 | 130 | // Pulling the bytes out directly using InetAddress.getByAddress() requires error handling TODO |
134 | 131 | InetAddress sourceIpAddress = InetAddresses.forString(sourceIp.utf8ToString()); |
135 | 132 | InetAddress destinationIpAddress = InetAddresses.forString(destinationIp.utf8ToString()); |
|
0 commit comments