处理 首页的消息刷新

galaxy/galaxy/Classes/Modules/Home(首页)/C/YHHomeHoldViewPageViewController.swift

@@ -54,6 +54,8 @@ class YHHomeHoldViewPageViewController: YHBaseViewController {
     
     var tapTimestamp : CFAbsoluteTime = -10.0
     
+    var needReloadDataFlag : Bool = false
+    
     override func viewDidLoad() {
         super.viewDidLoad()
         setupUI()
@@ -73,6 +75,12 @@ class YHHomeHoldViewPageViewController: YHBaseViewController {
         updateDefaultSearchKeyWord()
         
         requestAppNewVersion()
+        
+        
+        if needReloadDataFlag {
+            needReloadDataFlag = false
+            homePageVC.requestOrder()
+        }
     }
 
     override func viewWillDisappear(_ animated: Bool) {
@@ -113,6 +121,10 @@ class YHHomeHoldViewPageViewController: YHBaseViewController {
 
 //MARK: - 内部函数
 extension YHHomeHoldViewPageViewController {
+    @objc func clickMsgitem() {
+        needReloadDataFlag = true
+    }
+    
     @objc func didRefreshMessageCount() {
         getTotalUnreadMsgCount()
     }
@@ -207,10 +219,12 @@ extension YHHomeHoldViewPageViewController {
         
         NotificationCenter.default.addObserver(self, selector: #selector(clearUnreadBadge), name: YhConstant.YhNotification.didMarkAllMessagesReadedNotifiction, object: nil)
         
-                
         NotificationCenter.default.addObserver(self, selector: #selector(didRefreshMessageCount), name: YhConstant.YhNotification.didRequestUnreadMsgTotalCountNotification, object: nil)
         
         
+        NotificationCenter.default.addObserver(self, selector: #selector(clickMsgitem), name: YhConstant.YhNotification.clickHomeMsgNotifiction, object: nil)
+        
+        
         for i in 0..<arrItemTitles.count {
             if i == 0 {
                 //首页

galaxy/galaxy/Classes/Modules/Home(首页)/C/YHHomePageViewController.swift

@@ -25,7 +25,6 @@ class YHHomePageViewController: YHBaseViewController {
     
     var homeHeaderViewHeight = YHHomeHeadView.fixViewH
     
-    var needReloadDataFlag : Bool = false
     
     lazy var homeCollectView = {
         // 设置布局方向
@@ -56,14 +55,6 @@ class YHHomePageViewController: YHBaseViewController {
         getData()
     }
     
-    override func viewWillAppear(_ animated: Bool) {
-        super.viewWillAppear(animated)
-        
-        if needReloadDataFlag {
-            needReloadDataFlag = false
-            requestOrder()
-        }
-    }
     
     lazy var homeHeaderView: YHHomeHeadView = {
         let view = YHHomeHeadView()
@@ -113,10 +104,6 @@ private extension YHHomePageViewController {
         getCacheDataForHomeNewsData()
     }
     
-    @objc func clickMsgitem() {
-        needReloadDataFlag = true
-    }
-    
     //获取缓存的 banner数据
     func getCacheDataForBanner() -> [YHBannerModel] {
         let arrString2 = UserDefaults.standard.value(forKey: "homeBannerData")
@@ -229,7 +216,7 @@ private extension YHHomePageViewController {
     
     func setupUI() {
         gk_navigationBar.isHidden = true
-        NotificationCenter.default.addObserver(self, selector: #selector(clickMsgitem), name: YhConstant.YhNotification.clickHomeMsgNotifiction, object: nil)
+        
         
         view.addSubview(homeCollectView)
         homeCollectView.snp.makeConstraints { make in

galaxy/galaxy/Classes/Modules/Home(首页)/V/YHMessageBellView.swift

@@ -91,6 +91,8 @@ class YHMessageBellView: UIView {
             YHOneKeyLoginManager.shared.oneKeyLogin()
             return
         }
+        
+        NotificationCenter.default.post(name: YhConstant.YhNotification.clickHomeMsgNotifiction, object: nil)
         let vc = YHMessageListVC()
         UIViewController.current?.navigationController?.pushViewController(vc)
     }