Only clear GL context after changing the thread configuration

fml/raster_thread_merger.cc

@@ -21,6 +21,10 @@ RasterThreadMerger::RasterThreadMerger(fml::TaskQueueId platform_queue_id,
   FML_CHECK(!task_queues_->Owns(platform_queue_id_, gpu_queue_id_));
 }
 
+void RasterThreadMerger::SetMergeUnmergeCallback(const fml::closure& callback) {
+  merge_unmerge_callback_ = callback;
+}
+
 void RasterThreadMerger::MergeWithLease(size_t lease_term) {
   std::scoped_lock lock(lease_term_mutex_);
   if (TaskQueuesAreSame()) {
@@ -30,11 +34,19 @@ void RasterThreadMerger::MergeWithLease(size_t lease_term) {
     return;
   }
   FML_DCHECK(lease_term > 0) << "lease_term should be positive.";
-  if (!IsMergedUnSafe()) {
-    bool success = task_queues_->Merge(platform_queue_id_, gpu_queue_id_);
-    FML_CHECK(success) << "Unable to merge the raster and platform threads.";
-    lease_term_ = lease_term;
+
+  if (IsMergedUnSafe()) {
+    merged_condition_.notify_one();
+    return;
+  }
+
+  bool success = task_queues_->Merge(platform_queue_id_, gpu_queue_id_);
+  if (success && merge_unmerge_callback_ != nullptr) {
+    merge_unmerge_callback_();
   }
+  FML_CHECK(success) << "Unable to merge the raster and platform threads.";
+  lease_term_ = lease_term;
+
   merged_condition_.notify_one();
 }
 
@@ -48,6 +60,9 @@ void RasterThreadMerger::UnMergeNow() {
   }
   lease_term_ = 0;
   bool success = task_queues_->Unmerge(platform_queue_id_);
+  if (success && merge_unmerge_callback_ != nullptr) {
+    merge_unmerge_callback_();
+  }
   FML_CHECK(success) << "Unable to un-merge the raster and platform threads.";
 }
 

fml/raster_thread_merger.h

@@ -85,6 +85,13 @@ class RasterThreadMerger
   // noop.
   bool IsEnabled();
 
+  // Registers a callback that can be used to clean up global state right after
+  // the thread configuration has changed.
+  //
+  // For example, it can be used to clear the GL context so it can be used in
+  // the next task from a different thread.
+  void SetMergeUnmergeCallback(const fml::closure& callback);
+
  private:
   static const int kLeaseNotSet;
   fml::TaskQueueId platform_queue_id_;
@@ -93,6 +100,7 @@ class RasterThreadMerger
   std::atomic_int lease_term_;
   std::condition_variable merged_condition_;
   std::mutex lease_term_mutex_;
+  fml::closure merge_unmerge_callback_;
   bool enabled_;
 
   bool IsMergedUnSafe() const;

shell/common/rasterizer.cc

@@ -89,6 +89,14 @@ void Rasterizer::Setup(std::unique_ptr<Surface> surface) {
         delegate_.GetTaskRunners().GetRasterTaskRunner()->GetTaskQueueId();
     raster_thread_merger_ =
         fml::MakeRefCounted<fml::RasterThreadMerger>(platform_id, gpu_id);
+
+    raster_thread_merger_->SetMergeUnmergeCallback(
+        [weak_this = weak_factory_.GetWeakPtr()]() {
+          // Clear the GL context after the thread configuration has changed.
+          if (weak_this && weak_this->surface_) {
+            weak_this->surface_->ClearRenderContext();
+          }
+        });
   }
 #endif
 }
@@ -467,14 +475,6 @@ RasterStatus Rasterizer::DrawToSurface(flutter::LayerTree& layer_tree) {
 
     FireNextFrameCallbackIfPresent();
 
-    // Clear the render context after submitting the frame.
-    // This ensures that the GL context is released after drawing to the
-    // surface.
-    //
-    // The GL context must be clear before performing Gr context deferred
-    // cleanup.
-    surface_->ClearRenderContext();
-
     if (surface_->GetContext()) {
       TRACE_EVENT0("flutter", "PerformDeferredSkiaCleanup");
       surface_->GetContext()->performDeferredCleanup(kSkiaCleanupExpiration);