[Swift] カスタムクラスから未読数をカウントする方法

Question

LINEのようなアプリを作っています。
トーク一覧画面にて、まだ開いていないメッセージの数を表示したいのですが、未読数の取得方法が分からず困っています...。お力添え頂けますと幸いです。

以下が作りたい画面のイメージです。名前、プロフ画像、送信時間などの表示はすでに出来ています。

Table Viewに各種情報を表示させるためのカスタムクラスは以下のような構図になっています。

class New {
    private var _sender_id = "";
    private var _sender_name = "";
    private var _url = "";
    private var _timeStamp: NSNumber = 1
    private var _toID = "";
    private var _childkey = "";
    private var _listened = false
    private var _toName = "";
   
 init(sender_id: String, sender_name: String, url: String, toID: String, timeStamp: NSNumber , childkey: String, listened: Bool, toName: String) {
        _sender_id = sender_id;
        _url = url;
        _toID = toID
        _timeStamp = timeStamp
        _childkey = childkey
        _listened = listened
        _sender_name = sender_name
        _toName = toName
    }
    
    var sender_id: String? {
            return _sender_id;
    }
    
    var sender_name: String? {
        return _sender_name;
    }
    
    var toID: String? {
        return _toID;
    }
    
    var toName: String? {
        return _toName;
    }
    
    var url: String? {
        return _url;
    }
    
    var timeStamp: NSNumber? {
        return _timeStamp;
    }
  
    var childkey: String? {
        return _childkey;
    }
    
    var listened: Bool? {
        return _listened;
    }
}

未読数表示は難易度が高そうだったので、cell.label.text = user.sender_name.countなどと書いてみてまずは合計メッセージ数の表示にトライしましたが、適切な数字を取得できませんでした。

三つのハードルがあるのではないかと考えています
⑴各友達とやり取りしたメッセージの合計数の取得（⬅︎実際にはこの数値は使いませんが。）
⑵各友達とのメッセージのやり取りうち、未読数の取得（自分が送信したメールで、相手がまだ開けていないものを含む）
⑶各友達が自分に送ったメッセージのうち未読数の取得

⑵ができれば、⑶はif文などを駆使すればできる気もするので、ひとまず⑵をクリアしたいです。

class NewTalkVC: UIViewController, UICollectionViewDataSource, UICollectionViewDelegate/*, FetchData*/ {
   var news = [New]()
   var newsDictionary = [String: New]()
   var friendName: String = ""
   var friendName2: String = ""
   var toID: String = ""
   var chatPartnerId: String = ""
   var chatPartnerId2: String = ""
   var chatPartnerName: String = ""
   var messages = [Message]()
    
    @IBOutlet weak var NewTalkCollectionView: UICollectionView!
    override func viewDidLoad() {
        super.viewDidLoad()
        getNewTalk()
    }

    func getNewTalk() {
        let uid = Auth.auth().currentUser?.uid
        Database.database().reference().child("user-messages").child(uid!).observe(.childAdded, with: { (snapshot) in
            let messageId = snapshot.key
             DBProvider.Instance.mediaMessagesRef.child(messageId).observe(.value, with: { (snapshot) in
                 if let d = snapshot.value as? [String:AnyObject]{
                    if let sender_id    = d["sender_id"]   as? String {
                        let sender_name = d["sender_name"] as? String
                        let timeStamp   = d["timeStamp"]   as? NSNumber
                        let toID        = d["toID"]        as? String
                        let toName      = d["toName"]      as? String
                        let listened    = d["listened"]    as? Bool
                        let childkey    = d["childkey"]    as? String
                        let url         = d["url"]         as? String
                      
                        let newMessage = New(sender_id: sender_id, sender_name: sender_name!, url:url!, toID: toID!, timeStamp: timeStamp!, childkey: childkey!, listened: listened!, toName: toName!)
                       
                        if sender_id == Auth.auth().currentUser?.uid {
                            self.chatPartnerId = toID!
                            self.newsDictionary[self.chatPartnerId] = newMessage                             self.news = Array(self.newsDictionary.values) 
                            self.news.sort(by: {Double($0.timeStamp!) < Double($1.timeStamp!)})
                        }
                             //送る主が自分以外で、かつ、送り先が自分であるとき
                        else if toID == Auth.auth().currentUser?.uid {
                            self.chatPartnerId = sender_id
                             self.newsDictionary[self.chatPartnerId] = newMessage
                            self.news = Array(self.newsDictionary.values)
                            self.news.sort(by: {Double($0.timeStamp!) < Double($1.timeStamp!)})
                             }
                    }
                    }
                    }, withCancel: nil)
                    
                }, withCancel: nil)
     } //getNewTalk()終わり
 
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return news.count
    }
    
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
     let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "NewTalkCollectionViewCell", for: indexPath) as! NewTalkCollectionViewCell
        let user = news[indexPath.row] 
       
        if user.sender_id == Auth.auth().currentUser?.uid {
            self.chatPartnerId2 = user.toID! 
        } else  {
            self.chatPartnerId2 = user.sender_id!
        }
        
      
        if user.sender_id == Auth.auth().currentUser?.uid {
            cell.label.text = user.toName
        }
           
        else if user.toID == Auth.auth().currentUser?.uid {
            cell.label.text = user.sender_name
        }
 
       //以下試したこと
       // var test1: Int = user.filter {$0.listened == false}.count
       // var test2: Int = user.sender_name?.count
       //  cell.numberOfMessages.text = "(test1!)"
       
        return cell
    }
 
}//class

なお、以前似たような質問をしたことがあり、当時は回答にもあるFilterを駆使すれば可能と思いましたが、
この方法だと、全友達とのやり取りの未読数の合計値しか取得できないため、やりたいことが実現できず...
https://teratail.com/questions/149399
（こちらはカスタムクラスに友達別に情報を分けずに、全友達との全メッセージを入れています。）

20190727以下追記

各メッセージはListened変数を有します。

Collection View内で該当のセルをタップしたら、Firebase上の変数が未読から既読に変わるようにしています。

    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
       
       
        let newValuesForProfile = ["listened": true]
       
//タップしたセルが自分に送られたメールであれば、既読に変更
     if messages[indexPath.row].senderId == self.toID {
        DBProvider.Instance.dbRef.child("Media_Messages").child(messages[indexPath.row].childkey!).updateChildValues(newValuesForProfile)
        }

Answer 1

まずはDBの構造がわからないと何とも言えないかと思います。
各ユーザ同士のメッセージを見たかどうかをどこに保存しているか、適切なタイミングで更新しているのか。
1対1のメッセージのやり取りだけを考えた場合には各メッセージに未読判定をつけておいてその件数を数えるような構造にすれば良いかなと思います。

変数名を見たところだと_listenedが未読、既読を管理しているものでしょうか？
だとすればそれを各ユーザごとにカウントしたら良いような気もしますが。
（構造を自分で考えたのであればわかりそうな気もするのでそう言う問題ではない？）

===2019/07/28 追記===
現在の作りでいくと一覧表示をさせるためにself.newsには送信ユーザごとの最新メッセージしか入っていないと思うのでfilterではカウントできないと思われます。
全メッセージが入っている配列に対してであれば、filterで複数条件を指定すれば良いので以下のようにすれば未読カウントがとれるかと思います。

var allMessages = [New]() // その人の全メッセージが入っている配列と仮定する

// 未読かつ送信IDが同じ（その人の全メッセージではなくて、本当に全メッセージが入ってしまうのであればtoIDが自分宛の条件を追加）
var test1: Int = allMessages.filter {$0.listened == false && $0.sender_id == user.sender_id}.count

または表示データを作成するgetNewTalk内で未読カウントなども考えられるかと思います。
Newクラスはメッセージ内容を保持するクラスだと思うので、メッセージ一覧の内容を保持するような新たなクラスを作ってgetNewTalk内で値を入れていく方がもしかしたら良いかもしれません。
（現在表示させてる内容だけであれば、名前、最終時間、未読数があれば良い？）
とりあえず、IDごとに未読数を保持するようなものを作って書いた場合

var unreadCount = [String:New]()
// 送る主が自分以外で、かつ、送り先が自分であるとき
// ここの中で未読数をカウントしていく
else if toID == Auth.auth().currentUser?.uid {
    self.chatPartnerId = toID!
    ~ 省略
    // 未読がfalseのときにカウントしていく
    if !listened {
        if let count = unreadCount[self.chatPartnerId] {
            unreadCount[self.chatPartnerId] = count + 1
        } else {
            unreadCount[self.chatPartnerId] = 1
        }
    }
}
// 上記でunreadCountにIDごとの未読数がカウントされてると思うので正しいかをログなどで確認してみる
~ 省略 
// 以下cellForItemAt内
    let user = news[indexPath.row]
    var test1 = unreadCount[user.sender_id] // これで未読数が取れるはず

実装するための考え方はこんな感じかと思います。
そもそものデータの持ち方をどうするか、APIがあるのであればそちら側に寄せるなどの考え方もあるかと思います。
長くなってしまったところと分かりにくい部分などもあるかもしれませんので何かあればコメントをお願いいたします。

Comments

[ishiishiyay]

razumaさん

ご回答いただきありがとうございます。
以下インラインにてコメントさせていただいたので、ご確認頂けますと幸いです。

>>まずはDBの構造がわからないと何とも言えないかと思います。
本文の下の方に構造を追記しました！

>>変数名を見たところだと_listenedが未読、既読を管理しているものでしょうか？
はい、ご認識のとおりです。

>>だとすればそれを各ユーザごとにカウントしたら良いような気もしますが。
はい、ユーザーごとにカウントしたいのですが、そのカウント方法が分からず...　

お力添え頂けますと幸いです...!

[ishiishiyay]

Razumaさん、
詳細にありがとうございます！試してみます！

[ishiishiyay]

Razumaさん、
前者の方法で無事に未読数を反映できました！！
自分では到底思いつかない方法でした、本当にありがとうございます...