(re)store previous m_removeAsyncAnywhere state

Brightspace · Aug 9, 2023 · 9501d2f · 9501d2f
1 parent a5ca8cd
commit 9501d2f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/D2L.CodeStyle.Analyzers/Async/Generator/AsyncToSyncMethodTransformer.cs b/src/D2L.CodeStyle.Analyzers/Async/Generator/AsyncToSyncMethodTransformer.cs
@@ -352,6 +352,8 @@ private StatementSyntax Transform( ExpressionStatementSyntax exprStmt ) {
 
 	private ExpressionSyntax Transform( InvocationExpressionSyntax invocationExpr) {
 		ITypeSymbol? returnTypeInfo = Model.GetTypeInfo( invocationExpr ).Type;
+
+		bool prevRemoveAsyncAnywhereState = m_removeAsyncAnywhere;
 		if( returnTypeInfo?.MetadataName == "IAsyncEnumerable`1" && returnTypeInfo.ContainingNamespace.ToString() == "System.Collections.Generic" ) {
 			m_removeAsyncAnywhere = true;
 		}
@@ -369,9 +371,13 @@ private ExpressionSyntax Transform( InvocationExpressionSyntax invocationExpr) {
 			return SyntaxFactory.ParseExpression( $"Task.Run(() => {invocationExpr}).Result" );
 		}
 
-		return invocationExpr
+		invocationExpr = invocationExpr
 			.WithExpression( Transform( invocationExpr.Expression ) )
 			.WithArgumentList( TransformAll( invocationExpr.ArgumentList, Transform ) );
+
+		m_removeAsyncAnywhere = prevRemoveAsyncAnywhereState;
+
+		return invocationExpr;
 	}
 
 	bool ShouldRemoveReturnedMemberAccess( MemberAccessExpressionSyntax memberAccessExpr ) {