From 439f0a209b936e1f6f06dcd1694160070fdbef57 Mon Sep 17 00:00:00 2001
From: "yuqi.pyq" <yuqi.pyq@antgroup.com>
Date: Tue, 19 Mar 2024 10:08:23 +0800
Subject: [PATCH] fix: add draw frame callback

---
 packages/g-canvas/package.json  | 2 +-
 packages/g-canvas/src/canvas.ts | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/g-canvas/package.json b/packages/g-canvas/package.json
index b91263eaf..1bf90cdeb 100644
--- a/packages/g-canvas/package.json
+++ b/packages/g-canvas/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@antv/g-canvas",
-  "version": "0.5.16",
+  "version": "0.5.17",
   "description": "A canvas library which providing 2d",
   "main": "lib/index.js",
   "module": "esm/index.js",
diff --git a/packages/g-canvas/src/canvas.ts b/packages/g-canvas/src/canvas.ts
index 1c53f1dd4..cb792ac9c 100644
--- a/packages/g-canvas/src/canvas.ts
+++ b/packages/g-canvas/src/canvas.ts
@@ -214,6 +214,7 @@ class Canvas extends AbstractCanvas {
   // 触发绘制
   _startDraw() {
     let drawFrame = this.get('drawFrame');
+    let drawFrameCallback = this.get('drawFrameCallback');
     if (!drawFrame) {
       drawFrame = requestAnimationFrame(() => {
         if (this.get('localRefresh')) {
@@ -222,6 +223,9 @@ class Canvas extends AbstractCanvas {
           this._drawAll();
         }
         this.set('drawFrame', null);
+        if (drawFrameCallback) {
+          drawFrameCallback();
+        }
       });
       this.set('drawFrame', drawFrame);
     }