fix: handleMenuClick trigger multiple times

1. handleMenuClick listerner not removed on menu hide
antvis · Aug 12, 2020 · c3b16cc · c3b16cc
1 parent 55ab1fe
commit c3b16cc
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/src/plugins/menu/index.ts b/src/plugins/menu/index.ts
@@ -118,7 +118,11 @@ export default class Menu extends Base {
 
     const handleMenuClick = this.get('handleMenuClick')
     if (handleMenuClick) {
-      menuDom.addEventListener('click', handleMenuClick);
+      const handleMenuClickWrapper = (evt) => {
+        handleMenuClick(evt.target, e.item)
+      }
+      this.set('handleMenuClickWrapper', handleMenuClickWrapper)
+      menuDom.addEventListener('click', handleMenuClickWrapper);
     }
 
     const graph: Graph = this.get('graph');
@@ -166,19 +170,19 @@ export default class Menu extends Base {
     // 隐藏菜单后需要移除事件监听
     document.body.removeEventListener('click', this.get('handler'));
 
-    const handleMenuClick = this.get('handleMenuClick')
-    if (handleMenuClick) {
-      menuDom.removeEventListener('click', handleMenuClick)
+    const handleMenuClickWrapper = this.get('handleMenuClickWrapper');
+    if (handleMenuClickWrapper) {
+      menuDom.removeEventListener('click', handleMenuClickWrapper);
     }
   }
 
   public destroy() {
     const menu = this.get('menu')
     const handler = this.get('handler');
 
-    const handleMenuClick = this.get('handleMenuClick')
-    if (handleMenuClick) {
-      menu.removeEventListener('click', handleMenuClick)
+    const handleMenuClickWrapper = this.get('handleMenuClickWrapper');
+    if (handleMenuClickWrapper) {
+      menu.removeEventListener('click', handleMenuClickWrapper);
     }
 
     if (menu) {