【Swift】配列へのフィルターのかけかた

Question

●やりたいこと
チャットアプリで、まだ開いていないメッセージの数を表示したい。

●試みた方法
配列(複数のkeyとvalueをもつ独自クラス)のうち、filter機能を使って、特定のkeyとvalueの組み合わせの数(＝未読のメッセージの数)を数える。

●問題
ちゃんとカウントできていない

以下コードです。
お助け頂けますと幸いです。よろしくお願いいたします！！

class New2: NSObject {

    var sender_id: String?
   
    var timeStamp: NSNumber?
   
    var toID: String?
    
    var listened: Bool?　//Firebaseのデータベースから取得するこのlistenedフラグで、既読か未読かを区別
  
}

var news = [New2]()
var newsDictionary = [String: New2]()

以下viewDidload()内
Database.database().reference().child("Media_Messages").observe(.childAdded, with: { (snapshot) in
            if let dictionary = snapshot.value as? [String: AnyObject]{
                let new2 = New2()
                new2.sender_id = dictionary["sender_id"] as? String
                new2.toID = dictionary["toID"] as? String
                new2.listened = dictionary["listened"] as? Bool
              
  //送り主が自分なら、表示する名前とアイコンは、受信者。送り主が自分じゃないなら、表示するのは送信者。
                if new2.sender_id == Auth.auth().currentUser?.uid {
                    self.newsDictionary[new2.toID!] = new2
                    self.news = Array(self.newsDictionary.values)
                    self.NewTalkCollectionView.reloadData()
            
                }
                //送る主が自分以外で、かつ、送り先が自分であるとき
                else if new2.toID == Auth.auth().currentUser?.uid {
                    self.newsDictionary[new2.sender_id!] = new2
                    self.news = Array(self.newsDictionary.values)
                    self.NewTalkCollectionView.reloadData()
                }
              
                
            }
        }, withCancel: nil)

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "NewTalkCollectionViewCell", for: indexPath) as! NewTalkCollectionViewCell
        
        let user = news[indexPath.row]
        let numberOfMessages: Int = news.filter{ _ in user.listened == false}.count
       
        cell.numberOfMessages.text = "(numberOfMessages)"

上記コードですと、ちゃんとメッセージ数が１と表示され、開いたら、メッセージ数は０になります。
ただし、未読のメッセージが３つの状態でも、メッセージ数は１と表示され、三つのうち１つでも聞いたら、まだ未読が２つ残っているものの、メッセージ数が０と表示されてしまいます。

Filterのところを以下コードに変えてみたりしましたが、下記のエラーが。

 let numberOfMessages: Int = news.filter{ (key: String, value: Bool) -> Bool in return key == "listened" || value  == false}.count

Contextual closure type '(New2) -> Bool' expects 1 argument, but 2 were used in closure body

---

以下10/6追加

 var numberOfMessages: Int = news.filter {$0.listened == false}.count
 news.enumerated().forEach {
            print($0, $1.listened ?? "nil")
            if let listened = $1.listened {
                if listened {
                    numberOfMessages += 1
                }
            }
        }
        print("numberOfMessages =", numberOfMessages)

こちらの出力が、
0 false
numberOfMessages = 1

//var numberOfMessages: Int = news.filter {$0.listened == false}.count
var numberOfMessages = 0
 news.enumerated().forEach {
            print($0, $1.listened ?? "nil")
            if let listened = $1.listened {
                if listened {
                    numberOfMessages += 1
                }
            }
        }
        print("numberOfMessages =", numberOfMessages)

こちらの出力が、
0 false
numberOfMessages = 0

です。両者とも、開いていないメッセージ（listened = false）が複数ある状態です。

Answer 1

user.listenedってnews[indexPath.row].listenedってことだから、newsの中をループしているのではなく、ずっと同じものを見てます。

news.filter({ (new: New2) -> Bool in new.listened == false}).count

短くすると、

news.filter {$0.listened == false}.count

test

var numberOfMessages = 0
news.enumerated().forEach {
    print($0, $1.listened ?? "nil")
    if let listened = $1.listened {
        if listened {
            numberOfMessages += 1
        }
    }
}
print("numberOfMessages =", numberOfMessages)

test（修正版）

var countRead = 0, countUnread = 0, countNil = 0
news.enumerated().forEach {
    print($0, $1.listened ?? "nil")
    if let listened = $1.listened {
        if listened {
            countRead += 1
        } else {
            countUnread += 1
        }
    } else {
        countNil += 1
    }
}
print("count:(news.count) read:(countRead) unread:(countUnread) nil:(countNil)")

Comments

[ishiishiyay]

fuzzballさん
ご回答いただきありがとうございます！
教えていただきました両パターン試しましたが、メッセージ数は１のままです。。。
引き続き、アドバイス頂けますと大変幸いです。

[fuzzball]

テストコードを追記しましたので、出力を教えて下さい。

[ishiishiyay]

fuzzballさん
遅れましてすみません。
textcode、わざわざありがとうございます！

0 false
numberOfMessages = 0
0 false
numberOfMessages = 0
0 false
numberOfMessages = 0
・・・・

と出力されます。ちなみに放っておくと、時間経過とともに、0 falseとnumberOfMessages=0が繰り返しずーっと増えていくのですが、これは正常でしょうか？
Firebaseで更新がないかを都度都度チェックしに行ってるからということでしょうか。

お手数ですが、引き続きよろしくお願いいたします。。。！

[fuzzball]

おかしな状況のときのログを書いて下さい。

実行タイミングは私には分かりません。あなたが想定しているたびに表示されていれば正常で、そうでなければ何かがおかしいことになります。（「都度チェックしに行ってる」というのが、あなたが想定している状況なのであれば「正常」ということです）

[ishiishiyay]

fuzzballさん
申し訳ございません、おかしな状況とは何を指しますでしょうか？
本文の一番下に、テストコード２パターンの出力結果を追記しました。
firebase上で、listenedフラグがfalseになっている未読メールが複数ある状態で、実行しています。
お手数ですが、引き続きアドバイス頂けますと幸いです、よろしくお願いいたします。

[fuzzball]

まず、私のテストプログラムは判定が逆になっていて既読をカウントするようになっていました。
なので、今回のテスト結果だと、
・データは1個
・既読は0通（未読が1通ということ）
です。

>>listenedフラグがfalseになっている未読メールが複数ある状態で、実行しています。
ということは、フィルタリングが原因ではなく、データが1個しか取れていないことが原因です。
その原因を調べて下さい。

テストプログラムは少し修正しておきます。