Computate invocation range lazily

Co-authored-by: Bastian Müller <[email protected]>
onflow · Dec 8, 2023 · 437ea4b · 437ea4b
1 parent bc8b2ca
commit 437ea4b
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/runtime/sema/check_invocation_expression.go b/runtime/sema/check_invocation_expression.go
@@ -459,10 +459,19 @@ func (checker *Checker) checkInvocation(
 		}
 	}
 
-	invocationRange := ast.NewRangeFromPositioned(
-		checker.memoryGauge,
-		invocationExpression,
-	)
+	// Compute the invocation range, once, if needed
+	getInvocationRange := func() func() ast.Range {
+		var invcocationRange ast.Range
+		return func() ast.Range {
+			if invocationRange == ast.EmptyRange {
+				invocationRange = ast.NewRangeFromPositioned(
+					checker.memoryGauge,
+					invocationExpression,
+				)
+			}
+			return invocationRange
+		}
+	}()
 
 	// The invokable type might have special checks for the arguments