From d86238f0716933f38138d67a02afae8654971b7d Mon Sep 17 00:00:00 2001
From: chenjiahan <chenjiahan@youzan.com>
Date: Sun, 23 Aug 2020 11:42:12 +0800
Subject: [PATCH] refactor(Badge): use setup

---
 src/badge/index.js | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/badge/index.js b/src/badge/index.js
index 35620c38d..bea6bbeaf 100644
--- a/src/badge/index.js
+++ b/src/badge/index.js
@@ -8,14 +8,14 @@ export default createComponent({
     badge: [Number, String],
   },
 
-  render() {
-    const { dot, badge } = this;
-    const showBadge = isDef(badge) && badge !== '';
+  setup(props) {
+    return () => {
+      const { dot, badge } = props;
+      const hasBadge = isDef(badge) && badge !== '';
 
-    if (!dot && !showBadge) {
-      return;
-    }
-
-    return <div class={bem({ dot })}>{dot ? '' : badge}</div>;
+      if (dot || hasBadge) {
+        return <div class={bem({ dot })}>{dot ? '' : badge}</div>;
+      }
+    };
   },
 });