Merge upstream/master into tyrielv/hydration-status-1-itracer

Resolve conflicts in EnlistmentHydrationSummary.cs and GitStatusCache.cs
by combining ITracer + Stopwatch timing from this branch with
CancellationToken support from PR microsoft#1914.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: tyrielv

GVFS/GVFS.Common/GitStatusCache.cs

@@ -47,6 +47,8 @@ public class GitStatusCache : IDisposable
         private bool isStopping;
         private bool isInitialized;
         private StatusStatistics statistics;
+        private CancellationTokenSource shutdownTokenSource;
+        private Task activeHydrationTask;
 
         private volatile CacheState cacheState = CacheState.Dirty;
 
@@ -65,6 +67,7 @@ public GitStatusCache(GVFSContext context, TimeSpan backoffTime)
             this.backoffTime = backoffTime;
             this.serializedGitStatusFilePath = this.context.Enlistment.GitStatusCachePath;
             this.statistics = new StatusStatistics();
+            this.shutdownTokenSource = new CancellationTokenSource();
 
             this.wakeUpThread = new AutoResetEvent(false);
         }
@@ -79,6 +82,7 @@ public virtual void Initialize()
         public virtual void Shutdown()
         {
             this.isStopping = true;
+            this.shutdownTokenSource.Cancel();
 
             if (this.isInitialized && this.updateStatusCacheThread != null)
             {
@@ -177,6 +181,27 @@ public virtual void Dispose()
         {
             this.Shutdown();
 
+            // Wait for the hydration task to complete before disposing the
+            // token source it may still be using.
+            if (this.activeHydrationTask != null)
+            {
+                try
+                {
+                    this.activeHydrationTask.Wait();
+                }
+                catch (AggregateException)
+                {
+                }
+
+                this.activeHydrationTask = null;
+            }
+
+            if (this.shutdownTokenSource != null)
+            {
+                this.shutdownTokenSource.Dispose();
+                this.shutdownTokenSource = null;
+            }
+
             if (this.wakeUpThread != null)
             {
                 this.wakeUpThread.Dispose();
@@ -317,10 +342,29 @@ private void RebuildStatusCacheIfNeeded(bool ignoreBackoff)
             if (needToRebuild)
             {
                 this.statistics.RecordBackgroundStatusScanRun();
-                this.UpdateHydrationSummary();
+
+                // Run hydration summary in parallel with git status — they are independent
+                // operations and neither should delay the other.
+                Task hydrationTask = Task.Run(() => this.UpdateHydrationSummary());
+                this.activeHydrationTask = hydrationTask;
 
                 bool rebuildStatusCacheSucceeded = this.TryRebuildStatusCache();
 
+                // Wait for hydration to complete before logging final stats.
+                try
+                {
+                    hydrationTask.Wait();
+                }
+                catch (AggregateException ex)
+                {
+                    EventMetadata errorMetadata = new EventMetadata();
+                    errorMetadata.Add("Area", EtwArea);
+                    errorMetadata.Add("Exception", ex.InnerException?.ToString());
+                    this.context.Tracer.RelatedError(
+                        errorMetadata,
+                        $"{nameof(GitStatusCache)}.{nameof(RebuildStatusCacheIfNeeded)}: Unhandled exception in hydration summary task.");
+                }
+
                 TimeSpan delayedTime = startTime - this.initialDelayTime;
                 TimeSpan statusRunTime = DateTime.UtcNow - startTime;
 
@@ -356,7 +400,7 @@ private void UpdateHydrationSummary()
                  * and this is also a convenient place to log telemetry for it.
                  */
                 EnlistmentHydrationSummary hydrationSummary =
-                    EnlistmentHydrationSummary.CreateSummary(this.context.Enlistment, this.context.FileSystem, this.context.Tracer);
+                    EnlistmentHydrationSummary.CreateSummary(this.context.Enlistment, this.context.FileSystem, this.context.Tracer, cancellationToken: this.shutdownTokenSource.Token);
                 EventMetadata metadata = new EventMetadata();
                 metadata.Add("Area", EtwArea);
                 if (hydrationSummary.IsValid)
@@ -372,14 +416,20 @@ private void UpdateHydrationSummary()
                         metadata,
                         Keywords.Telemetry);
                 }
-                else
+                else if (hydrationSummary.Error != null)
                 {
-                    metadata["Exception"] = hydrationSummary.Error?.ToString();
+                    metadata["Exception"] = hydrationSummary.Error.ToString();
                     this.context.Tracer.RelatedWarning(
                         metadata,
                         $"{nameof(GitStatusCache)}{nameof(RebuildStatusCacheIfNeeded)}: hydration summary could not be calculated.",
                         Keywords.Telemetry);
                 }
+                else
+                {
+                    // Invalid summary with no error — likely cancelled during shutdown
+                    this.context.Tracer.RelatedInfo(
+                        $"{nameof(GitStatusCache)}{nameof(RebuildStatusCacheIfNeeded)}: hydration summary was cancelled.");
+                }
             }
             catch (Exception ex)
             {

GVFS/GVFS.Common/HealthCalculator/EnlistmentHydrationSummary.cs

@@ -4,7 +4,8 @@
 using System;
 using System.Diagnostics;
 using System.IO;
-using System.Linq;
+using System.Linq;
+using System.Threading;
 
 namespace GVFS.Common
 {
@@ -43,7 +44,8 @@ public string ToMessage()
         public static EnlistmentHydrationSummary CreateSummary(
             GVFSEnlistment enlistment,
             PhysicalFileSystem fileSystem,
-            ITracer tracer)
+            ITracer tracer,
+            CancellationToken cancellationToken = default)
         {
             Stopwatch totalStopwatch = Stopwatch.StartNew();
             Stopwatch phaseStopwatch = new Stopwatch();
@@ -56,17 +58,23 @@ public static EnlistmentHydrationSummary CreateSummary(
                 int totalFileCount = GetIndexFileCount(enlistment, fileSystem);
                 long indexReadMs = phaseStopwatch.ElapsedMilliseconds;
 
+                cancellationToken.ThrowIfCancellationRequested();
+
                 EnlistmentPathData pathData = new EnlistmentPathData();
 
                 /* FUTURE: These could be optimized to only deal with counts instead of full path lists */
                 phaseStopwatch.Restart();
                 pathData.LoadPlaceholdersFromDatabase(enlistment);
                 long placeholderLoadMs = phaseStopwatch.ElapsedMilliseconds;
 
+                cancellationToken.ThrowIfCancellationRequested();
+
                 phaseStopwatch.Restart();
                 pathData.LoadModifiedPaths(enlistment, tracer);
                 long modifiedPathsLoadMs = phaseStopwatch.ElapsedMilliseconds;
 
+                cancellationToken.ThrowIfCancellationRequested();
+
                 int hydratedFileCount = pathData.ModifiedFilePaths.Count + pathData.PlaceholderFilePaths.Count;
                 int hydratedFolderCount = pathData.ModifiedFolderPaths.Count + pathData.PlaceholderFolderPaths.Count;
 
@@ -108,6 +116,16 @@ public static EnlistmentHydrationSummary CreateSummary(
                     TotalFileCount = totalFileCount,
                     TotalFolderCount = totalFolderCount,
                 };
+            }
+            catch (OperationCanceledException)
+            {
+                return new EnlistmentHydrationSummary()
+                {
+                    HydratedFileCount = -1,
+                    HydratedFolderCount = -1,
+                    TotalFileCount = -1,
+                    TotalFolderCount = -1,
+                };
             }
             catch (Exception e)
             {