refactor(material/table): remove use of zone onStable for style scheduling

Author: mmalerba

src/cdk/table/coalesced-style-scheduler.ts

@@ -6,7 +6,15 @@
  * found in the LICENSE file at https://angular.io/license
  */
 
-import {Injectable, NgZone, OnDestroy, InjectionToken} from '@angular/core';
+import {
+  Injectable,
+  NgZone,
+  OnDestroy,
+  InjectionToken,
+  afterNextRender,
+  inject,
+  Injector,
+} from '@angular/core';
 import {from, Subject} from 'rxjs';
 import {take, takeUntil} from 'rxjs/operators';
 
@@ -35,6 +43,8 @@ export class _CoalescedStyleScheduler implements OnDestroy {
   private _currentSchedule: _Schedule | null = null;
   private readonly _destroyed = new Subject<void>();
 
+  private _injector = inject(Injector);
+
   constructor(private readonly _ngZone: NgZone) {}
 
   /**
@@ -69,33 +79,30 @@ export class _CoalescedStyleScheduler implements OnDestroy {
 
     this._currentSchedule = new _Schedule();
 
-    this._getScheduleObservable()
-      .pipe(takeUntil(this._destroyed))
-      .subscribe(() => {
-        while (this._currentSchedule!.tasks.length || this._currentSchedule!.endTasks.length) {
-          const schedule = this._currentSchedule!;
-
-          // Capture new tasks scheduled by the current set of tasks.
-          this._currentSchedule = new _Schedule();
-
-          for (const task of schedule.tasks) {
-            task();
-          }
-
-          for (const task of schedule.endTasks) {
-            task();
-          }
-        }
-
-        this._currentSchedule = null;
-      });
-  }
-
-  private _getScheduleObservable() {
-    // Use onStable when in the context of an ongoing change detection cycle so that we
-    // do not accidentally trigger additional cycles.
-    return this._ngZone.isStable
-      ? from(Promise.resolve(undefined))
-      : this._ngZone.onStable.pipe(take(1));
+    this._ngZone.run(() =>
+      queueMicrotask(() =>
+        afterNextRender(
+          () => {
+            while (this._currentSchedule!.tasks.length || this._currentSchedule!.endTasks.length) {
+              const schedule = this._currentSchedule!;
+
+              // Capture new tasks scheduled by the current set of tasks.
+              this._currentSchedule = new _Schedule();
+
+              for (const task of schedule.tasks) {
+                task();
+              }
+
+              for (const task of schedule.endTasks) {
+                task();
+              }
+            }
+
+            this._currentSchedule = null;
+          },
+          {injector: this._injector},
+        ),
+      ),
+    );
   }
 }