Fix IE Edge Object doesn't support property or method 'removeAttribute'

close ant-design/ant-design-pro#2423
ardok · Nov 8, 2018 · 73e4519 · 73e4519
1 parent 59a42ab
commit 73e4519
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/components/_util/wave.tsx b/components/_util/wave.tsx
@@ -21,7 +21,7 @@ export default class Wave extends React.Component<{insertExtraNode?: boolean}> {
   }
 
   onClick = (node: HTMLElement, waveColor: string) => {
-    if (node.className.indexOf('-leave') >= 0) {
+    if (!node || node.className.indexOf('-leave') >= 0) {
       return;
     }
     const { insertExtraNode } = this.props;
@@ -113,7 +113,11 @@ export default class Wave extends React.Component<{insertExtraNode?: boolean}> {
   }
 
   componentDidMount() {
-    this.instance = this.bindAnimationEvent(findDOMNode(this) as HTMLElement);
+    const node = findDOMNode(this) as HTMLElement;
+    if (node.nodeType !== 1) {
+      return;
+    }
+    this.instance = this.bindAnimationEvent(node);
   }
 
   componentWillUnmount() {