more effective lock utilization

Azure · Jul 29, 2024 · c3af4b5 · c3af4b5
1 parent 1e7a071
commit c3af4b5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/tools/test-proxy/Azure.Sdk.Tools.TestProxy/Common/RecordSession.cs b/tools/test-proxy/Azure.Sdk.Tools.TestProxy/Common/RecordSession.cs
@@ -118,5 +118,15 @@ public void Sanitize(RecordedTestSanitizer sanitizer)
                 sanitizer.Sanitize(this);
             }
         }
+        public void Sanitize(IEnumerable<RecordedTestSanitizer> sanitizers)
+        {
+            lock (Entries)
+            {
+                foreach(var sanitizer in sanitizers)
+                {
+                    sanitizer.Sanitize(this);
+                }
+            }
+        }
     }
 }
diff --git a/tools/test-proxy/Azure.Sdk.Tools.TestProxy/RecordingHandler.cs b/tools/test-proxy/Azure.Sdk.Tools.TestProxy/RecordingHandler.cs
@@ -114,11 +114,7 @@ public void StopRecording(string sessionId, IDictionary<string, string> variable
             }
 
             var sanitizers = SanitizerRegistry.GetSanitizers(recordingSession);
-
-            foreach (RecordedTestSanitizer sanitizer in sanitizers)
-            {
-                recordingSession.Session.Sanitize(sanitizer);
-            }
+            recordingSession.Session.Sanitize(sanitizers);
 
             if (variables != null)
             {
@@ -464,11 +460,7 @@ public async Task HandlePlaybackRequest(string recordingId, HttpRequest incoming
                 {
                     if (!session.IsSanitized)
                     {
-                        foreach (RecordedTestSanitizer sanitizer in sanitizers)
-                        {
-                            session.Session.Sanitize(sanitizer);
-                        }
-
+                        session.Session.Sanitize(sanitizers);
                         session.IsSanitized = true;
                     }
                 }