Skip to content

Commit 232cad4

Browse files
committed
Fix IT
Signed-off-by: Lantao Jin <ltjin@amazon.com>
1 parent 3238f5f commit 232cad4

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

core/src/main/java/org/opensearch/sql/calcite/ExtendedRexBuilder.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,11 @@ public static int getNextId() {
4040
return threadCounter.get().getAndIncrement();
4141
}
4242

43+
// For test only. Reset id to avoid unstable explain output of query.
44+
public static void resetId() {
45+
threadCounter.get().set(0);
46+
}
47+
4348
public ExtendedRexBuilder(RexBuilder rexBuilder) {
4449
super(rexBuilder.getTypeFactory());
4550
}

integ-test/src/test/java/org/opensearch/sql/calcite/remote/CalciteExplainIT.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import org.junit.Assume;
1212
import org.junit.Ignore;
1313
import org.junit.Test;
14+
import org.opensearch.sql.calcite.ExtendedRexBuilder;
1415
import org.opensearch.sql.ppl.ExplainIT;
1516

1617
public class CalciteExplainIT extends ExplainIT {
@@ -19,6 +20,7 @@ public void init() throws Exception {
1920
super.init();
2021
enableCalcite();
2122
disallowCalciteFallback();
23+
ExtendedRexBuilder.resetId();
2224
}
2325

2426
@Override

0 commit comments

Comments
 (0)