refactor(Badge): use setup

2020-08-23 11:42:12 +08:00
parent b67d0f10f8
commit d86238f071
1 changed files with 8 additions and 8 deletions
@@ -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>;
+      }
+    };
  },
 });