Skip to content

Commit 6390d86

Browse files
l46kokcopybara-github
l46kok
authored and
copybara-github
committed
Do not mangle comprehension identifiers if AST was not optimized
PiperOrigin-RevId: 797458253
1 parent fe46a63 commit 6390d86

9 files changed

Lines changed: 171 additions & 170 deletions

‎optimizer/src/main/java/dev/cel/optimizer/optimizers/SubexpressionOptimizer.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ private OptimizationResult optimizeUsingCelBlock(CelAbstractSyntaxTree ast, Cel
184184

185185
if (iterCount == 0) {
186186
// No modification has been made.
187-
return OptimizationResult.create(astToModify.toParsedAst());
187+
return OptimizationResult.create(ast);
188188
}
189189

190190
ImmutableList.Builder<CelVarDecl> newVarDecls = ImmutableList.builder();
@@ -395,7 +395,7 @@ private OptimizationResult optimizeUsingCelBind(CelAbstractSyntaxTree ast) {
395395

396396
if (iterCount == 0) {
397397
// No modification has been made.
398-
return OptimizationResult.create(astToModify.toParsedAst());
398+
return OptimizationResult.create(ast);
399399
}
400400

401401
astToModify = astMutator.renumberIdsConsecutively(astToModify);

‎optimizer/src/test/java/dev/cel/optimizer/optimizers/SubexpressionOptimizerTest.java‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,8 @@ private enum CseNoOpTestCase {
160160
NESTED_FUNCTION("int(timestamp(int(timestamp(1000000000))))"),
161161
// This cannot be optimized. Extracting the common subexpression would presence test
162162
// the bound identifier (e.g: has(@r0)), which is not valid.
163-
UNOPTIMIZABLE_TERNARY("has(msg.single_any) ? msg.single_any : 10");
163+
UNOPTIMIZABLE_TERNARY("has(msg.single_any) ? msg.single_any : 10"),
164+
MACRO("[1, 2, 3].exists(x, x > 0)");
164165

165166
private final String source;
166167

‎optimizer/src/test/resources/subexpression_ast_block_common_subexpr_only.baseline‎

Lines changed: 28 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -1781,7 +1781,7 @@ CALL [31] {
17811781
function: _==_
17821782
args: {
17831783
COMPREHENSION [30] {
1784-
iter_var: @it:0:0
1784+
iter_var: y
17851785
iter_range: {
17861786
LIST [1] {
17871787
elements: {
@@ -1790,7 +1790,7 @@ CALL [31] {
17901790
}
17911791
}
17921792
}
1793-
accu_var: @ac:0:0
1793+
accu_var: @result
17941794
accu_init: {
17951795
LIST [24] {
17961796
elements: {
@@ -1805,12 +1805,12 @@ CALL [31] {
18051805
function: _+_
18061806
args: {
18071807
IDENT [26] {
1808-
name: @ac:0:0
1808+
name: @result
18091809
}
18101810
LIST [27] {
18111811
elements: {
18121812
COMPREHENSION [23] {
1813-
iter_var: @it:1:0
1813+
iter_var: x
18141814
iter_range: {
18151815
LIST [6] {
18161816
elements: {
@@ -1820,7 +1820,7 @@ CALL [31] {
18201820
}
18211821
}
18221822
}
1823-
accu_var: @ac:1:0
1823+
accu_var: @result
18241824
accu_init: {
18251825
LIST [15] {
18261826
elements: {
@@ -1838,37 +1838,37 @@ CALL [31] {
18381838
function: _==_
18391839
args: {
18401840
IDENT [12] {
1841-
name: @it:1:0
1841+
name: x
18421842
}
18431843
IDENT [14] {
1844-
name: @it:0:0
1844+
name: y
18451845
}
18461846
}
18471847
}
18481848
CALL [19] {
18491849
function: _+_
18501850
args: {
18511851
IDENT [17] {
1852-
name: @ac:1:0
1852+
name: @result
18531853
}
18541854
LIST [18] {
18551855
elements: {
18561856
IDENT [11] {
1857-
name: @it:1:0
1857+
name: x
18581858
}
18591859
}
18601860
}
18611861
}
18621862
}
18631863
IDENT [20] {
1864-
name: @ac:1:0
1864+
name: @result
18651865
}
18661866
}
18671867
}
18681868
}
18691869
result: {
18701870
IDENT [22] {
1871-
name: @ac:1:0
1871+
name: @result
18721872
}
18731873
}
18741874
}
@@ -1879,7 +1879,7 @@ CALL [31] {
18791879
}
18801880
result: {
18811881
IDENT [29] {
1882-
name: @ac:0:0
1882+
name: @result
18831883
}
18841884
}
18851885
}
@@ -2380,10 +2380,10 @@ Test case: MACRO_SHADOWED_VARIABLE_2
23802380
Source: ["foo", "bar"].map(x, [x + x, x + x]).map(x, [x + x, x + x])
23812381
=====>
23822382
COMPREHENSION [35] {
2383-
iter_var: @it:0:0
2383+
iter_var: x
23842384
iter_range: {
23852385
COMPREHENSION [19] {
2386-
iter_var: @it:1:0
2386+
iter_var: x
23872387
iter_range: {
23882388
LIST [1] {
23892389
elements: {
@@ -2392,7 +2392,7 @@ COMPREHENSION [35] {
23922392
}
23932393
}
23942394
}
2395-
accu_var: @ac:1:0
2395+
accu_var: @result
23962396
accu_init: {
23972397
LIST [13] {
23982398
elements: {
@@ -2407,7 +2407,7 @@ COMPREHENSION [35] {
24072407
function: _+_
24082408
args: {
24092409
IDENT [15] {
2410-
name: @ac:1:0
2410+
name: @result
24112411
}
24122412
LIST [16] {
24132413
elements: {
@@ -2417,21 +2417,21 @@ COMPREHENSION [35] {
24172417
function: _+_
24182418
args: {
24192419
IDENT [7] {
2420-
name: @it:1:0
2420+
name: x
24212421
}
24222422
IDENT [9] {
2423-
name: @it:1:0
2423+
name: x
24242424
}
24252425
}
24262426
}
24272427
CALL [11] {
24282428
function: _+_
24292429
args: {
24302430
IDENT [10] {
2431-
name: @it:1:0
2431+
name: x
24322432
}
24332433
IDENT [12] {
2434-
name: @it:1:0
2434+
name: x
24352435
}
24362436
}
24372437
}
@@ -2444,12 +2444,12 @@ COMPREHENSION [35] {
24442444
}
24452445
result: {
24462446
IDENT [18] {
2447-
name: @ac:1:0
2447+
name: @result
24482448
}
24492449
}
24502450
}
24512451
}
2452-
accu_var: @ac:0:0
2452+
accu_var: @result
24532453
accu_init: {
24542454
LIST [29] {
24552455
elements: {
@@ -2464,7 +2464,7 @@ COMPREHENSION [35] {
24642464
function: _+_
24652465
args: {
24662466
IDENT [31] {
2467-
name: @ac:0:0
2467+
name: @result
24682468
}
24692469
LIST [32] {
24702470
elements: {
@@ -2474,21 +2474,21 @@ COMPREHENSION [35] {
24742474
function: _+_
24752475
args: {
24762476
IDENT [23] {
2477-
name: @it:0:0
2477+
name: x
24782478
}
24792479
IDENT [25] {
2480-
name: @it:0:0
2480+
name: x
24812481
}
24822482
}
24832483
}
24842484
CALL [27] {
24852485
function: _+_
24862486
args: {
24872487
IDENT [26] {
2488-
name: @it:0:0
2488+
name: x
24892489
}
24902490
IDENT [28] {
2491-
name: @it:0:0
2491+
name: x
24922492
}
24932493
}
24942494
}
@@ -2501,7 +2501,7 @@ COMPREHENSION [35] {
25012501
}
25022502
result: {
25032503
IDENT [34] {
2504-
name: @ac:0:0
2504+
name: @result
25052505
}
25062506
}
25072507
}

‎optimizer/src/test/resources/subexpression_ast_block_recursion_depth_6.baseline‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -2441,10 +2441,10 @@ Test case: MACRO_SHADOWED_VARIABLE_2
24412441
Source: ["foo", "bar"].map(x, [x + x, x + x]).map(x, [x + x, x + x])
24422442
=====>
24432443
COMPREHENSION [35] {
2444-
iter_var: @it:0:0
2444+
iter_var: x
24452445
iter_range: {
24462446
COMPREHENSION [19] {
2447-
iter_var: @it:1:0
2447+
iter_var: x
24482448
iter_range: {
24492449
LIST [1] {
24502450
elements: {
@@ -2453,7 +2453,7 @@ COMPREHENSION [35] {
24532453
}
24542454
}
24552455
}
2456-
accu_var: @ac:1:0
2456+
accu_var: @result
24572457
accu_init: {
24582458
LIST [13] {
24592459
elements: {
@@ -2468,7 +2468,7 @@ COMPREHENSION [35] {
24682468
function: _+_
24692469
args: {
24702470
IDENT [15] {
2471-
name: @ac:1:0
2471+
name: @result
24722472
}
24732473
LIST [16] {
24742474
elements: {
@@ -2478,21 +2478,21 @@ COMPREHENSION [35] {
24782478
function: _+_
24792479
args: {
24802480
IDENT [7] {
2481-
name: @it:1:0
2481+
name: x
24822482
}
24832483
IDENT [9] {
2484-
name: @it:1:0
2484+
name: x
24852485
}
24862486
}
24872487
}
24882488
CALL [11] {
24892489
function: _+_
24902490
args: {
24912491
IDENT [10] {
2492-
name: @it:1:0
2492+
name: x
24932493
}
24942494
IDENT [12] {
2495-
name: @it:1:0
2495+
name: x
24962496
}
24972497
}
24982498
}
@@ -2505,12 +2505,12 @@ COMPREHENSION [35] {
25052505
}
25062506
result: {
25072507
IDENT [18] {
2508-
name: @ac:1:0
2508+
name: @result
25092509
}
25102510
}
25112511
}
25122512
}
2513-
accu_var: @ac:0:0
2513+
accu_var: @result
25142514
accu_init: {
25152515
LIST [29] {
25162516
elements: {
@@ -2525,7 +2525,7 @@ COMPREHENSION [35] {
25252525
function: _+_
25262526
args: {
25272527
IDENT [31] {
2528-
name: @ac:0:0
2528+
name: @result
25292529
}
25302530
LIST [32] {
25312531
elements: {
@@ -2535,21 +2535,21 @@ COMPREHENSION [35] {
25352535
function: _+_
25362536
args: {
25372537
IDENT [23] {
2538-
name: @it:0:0
2538+
name: x
25392539
}
25402540
IDENT [25] {
2541-
name: @it:0:0
2541+
name: x
25422542
}
25432543
}
25442544
}
25452545
CALL [27] {
25462546
function: _+_
25472547
args: {
25482548
IDENT [26] {
2549-
name: @it:0:0
2549+
name: x
25502550
}
25512551
IDENT [28] {
2552-
name: @it:0:0
2552+
name: x
25532553
}
25542554
}
25552555
}
@@ -2562,7 +2562,7 @@ COMPREHENSION [35] {
25622562
}
25632563
result: {
25642564
IDENT [34] {
2565-
name: @ac:0:0
2565+
name: @result
25662566
}
25672567
}
25682568
}

0 commit comments

Comments
 (0)