// 我的好友

galaxy/galaxy/Classes/Modules/Community(社区)/C/YHCommunityViewController.swift

@@ -78,17 +78,22 @@ class YHCommunityViewController: YHBaseViewController {
     
     var unreadCount:Int = 0 {
         didSet {
-            self.unreadLabel.isHidden = (self.unreadCount <= 0)
-            if 0 < self.unreadCount && self.unreadCount < 100 {
-                self.unreadLabel.text = "\(self.unreadCount)"
-            } else if self.unreadCount >= 100 {
+            self.updateUnreadCount(self.unreadCount)
+        }
+    }
+    
+    func updateUnreadCount(_ count : Int) {
+        self.unreadLabel.isHidden = (count <= 0)
+        if 0 < count && count < 100 {
+            self.unreadLabel.text = "\(count)"
+        } else if count >= 100 {
             self.unreadLabel.text = "99+"
         }
         var width = unreadHeight
         var leftMargin = -8
-            if 10 <= self.unreadCount && self.unreadCount < 100 {
+        if 10 <= count && count < 100 {
             width = 22.0
-            } else if self.unreadCount >= 100 {
+        } else if count >= 100 {
             width = 30.0
             leftMargin = -14
         }
@@ -97,7 +102,6 @@ class YHCommunityViewController: YHBaseViewController {
             make.left.equalTo(self.friendsBtn.snp.right).offset(leftMargin)
         }
     }
-    }
 
     @objc func didBackBtnClicked() {
         self.navigationController?.popViewController(animated: true)
@@ -206,8 +210,7 @@ extension YHCommunityViewController: JXSegmentedViewDelegate {
         if segmentedView.selectedIndex == 0 {
             bgImgV.isHidden = false
             friendsBtn.isHidden = false
-            let count = self.unreadCount
-            self.unreadCount = count
+            self.updateUnreadCount(self.unreadCount)
             segmentedDataSource.titleNormalColor = UIColor(hex: 0xFFFFFF, alpha: 0.6)
             segmentedDataSource.titleSelectedColor = .white
             let indicator = JXSegmentedIndicatorLineView()
@@ -217,7 +220,7 @@ extension YHCommunityViewController: JXSegmentedViewDelegate {
         } else {
             bgImgV.isHidden = true
             friendsBtn.isHidden = true
-            self.unreadCount = 0
+            self.updateUnreadCount(0)
             segmentedDataSource.titleNormalColor = UIColor(hex: 0x94a3b8)
             segmentedDataSource.titleSelectedColor = UIColor.mainTextColor
             let indicator = JXSegmentedIndicatorLineView()