【Swift】いいねボタンの表示切り替えがうまくいかない

Question

前提・実現したいこと

tertailのAPIを叩き、質問一覧をTableViewに表示させています（Alamofire SwiftyJSONを使用）。
アーキテクチャはMVPです。
Likeボタンを設置しており、ボタンをタップされた質問のIDを取得し、Cloud Firestoreに質問のIDを保存しています。
既にLike済かどうかわかるように、ボタンの色を変化させたいです（Twitterのいいねのように）
Like済かどうかはisLikedをフラグとして使用しています。

発生している問題・エラーメッセージ

QuestionEntityのイニシャライザでFirestoreのデータに質問のIDがあるかどうかを判別し、IDがある（isLiked == true）場合はQuestionListTableViewCellにてボタンの表示を変化させています。

しかし、viewDidLoad、viewWillAppear、UIRefershControl実行時にはデフォルトの状態で表示されてしまいます。
つまり、func updateQuestions()実行時に問題が発生します。
ただ、その後、一度スクロールしてセルを再表示させると、意図した通りにボタンの表示が変更されます。

デバッグエリアには、
QuestionListTableViewCellの関数func setQuestion()内にあるprint("star"), print("starfill")が表示された後に、
QuestionEntity内のprint("TRUE"), print("FALSE")が表示されています。
そのため、恐らくQuestionEntityでFiresotoreのデータを検索しisLikedの値を変更しているタイミングに問題があるのでは？と考えています。

ただ、そこでどのように解決すべきかいい案が思い付かず、解決策をご教示いただけないでしょうか？
そもそもの仮説が間違っている等の場合も、ご指摘いただけると助かります。

該当のソースコード

class QuestionEntity {
    var id: Int
    var title: String
    var tags: [String]
    var displayName: String
    var photo: String
    var created: String
    var isAccepted: Bool
    var likes: [Int] = []
    var isLiked: Bool = false
    
    init(id: Int, title: String, tags: [String], displayName: String, photo: String, created: String, isAccepted: Bool) {
        self.id = id
        self.title = title
        self.tags = tags
        self.displayName = displayName
        self.photo = photo
        self.created = created
        self.isAccepted = isAccepted

        if let myId = Auth.auth().currentUser?.uid {
            let userRef = Firestore.firestore().collection("users").document(myId)
            userRef.getDocument { (document, error) in
                if let document = document {
                    if let likes = document.data()?["likes"] as? [Int] {
                        self.likes = likes
                    }
                    if self.likes.firstIndex(of: id) != nil {
                        self.isLiked = true
                        print("TRUE")
                    } else {
                        self.isLiked = false
                        print("FALSE")
                    }
                }
            }
        }
    }
}

class QuestionModel {
    
    var questions: [QuestionEntity] = []
    var notificationName: Notification.Name {
        return Notification.Name(rawValue: "questions")
    }
    
    func fetchNewArivalQuestions() {
        print("fetch")
        Alamofire.request("https://teratail.com/api/v1/questions").responseJSON { response in
            self.questions = []
            guard let object = response.result.value else { return }
            let json = JSON(object)
            json["questions"].forEach { (_, json) in
                let id = json["id"].intValue
                let title = json["title"].stringValue
                let tags = json["tags"].arrayObject
                let displayName = json["user"]["display_name"].stringValue
                let photo = json["user"]["photo"].stringValue
                let created = json["created"].stringValue
                let isAccepted = json["is_accepted"].boolValue
                self.questions.append(QuestionEntity(id: id, title: title, tags: tags as! [String], displayName: displayName, photo: photo, created: created, isAccepted: isAccepted))
            }
            self.notify()
            print("通知")
        }
    }
    
    func addObserver(_ observer: Any, selector: Selector) {
        NotificationCenter.default.addObserver(observer, selector: selector, name: notificationName, object: nil)
    }
    
    func removeObserver(_ observer: Any) {
        NotificationCenter.default.removeObserver(observer)
    }
    
    func notify() {
        NotificationCenter.default.post(name: notificationName, object: nil)
    }
     
}

class QuestionListTableViewCell: UITableViewCell {
        
    @IBOutlet weak var photoImageView: UIImageView!
    @IBOutlet weak var displayNameLabel: UILabel!
    @IBOutlet weak var createdLabel: UILabel!
    @IBOutlet weak var titleLabel: UILabel!
    @IBOutlet weak var likeButton: UIButton!
    @IBOutlet weak var tagsLabel: UILabel!
    @IBOutlet weak var isAcceptedLabel: UILabel!
 
    func setQuestion(entity: QuestionEntity) {
        self.displayNameLabel.text = entity.displayName
        
//        Likeボタンの表示
        if entity.isLiked == true {
            likeButton.setImage(UIImage(systemName: "star.fill"), for: .normal)
            print("starfill")
        } else {
            likeButton.setImage(UIImage(systemName: "star"), for: .normal)
            print("star")
        }
    }
}

protocol ListViewInterface: class {
    func reloadData()
    func navigateDetail(entity: QuestionEntity)
}

protocol LikeInterface: class {
    var questionId: Int { get }
}

class NewArrivalQuestionListViewController: UIViewController, ListViewInterface {
    
    @IBOutlet weak var newArrivalQuestionListTableView: UITableView!
    
    var presenter: QuestionListViewPresenter!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        initializePresenter()
        initializeTableView()
        newArrivalQuestionListTableView.refreshControl = UIRefreshControl()
        newArrivalQuestionListTableView.refreshControl?.addTarget(self, action: #selector(self.updateQuestions), for: .valueChanged)
        print("didload")
    }
    
    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(true)
        updateQuestions()
        print("willappear")
        
    }

    func initializeTableView() {
        newArrivalQuestionListTableView.delegate = self
        newArrivalQuestionListTableView.dataSource = self
        newArrivalQuestionListTableView.register(UINib(nibName: "QuestionListTableViewCell", bundle: nil), forCellReuseIdentifier: "QuestionListTableViewCell")
    }
    
    func initializePresenter() {
        presenter = QuestionListViewPresenter(view: self)
    }
    
    func reloadData() {
        newArrivalQuestionListTableView.refreshControl?.endRefreshing()
        newArrivalQuestionListTableView.reloadData()
        print("リロード")
    }
    
    func navigateDetail(entity: QuestionEntity) {
        let questoinDetailViewController = QuestionDetailViewController(entity: entity)
        self.navigationController?.pushViewController(questoinDetailViewController, animated: true)
       }

    @objc func updateQuestions() {
        presenter.updateQuestions()
        newArrivalQuestionListTableView.refreshControl?.beginRefreshing()
    }
    
    @objc func likeButtonTapped(_ sender: UIButton, forEvent event: UIEvent) {
        let touch = event.allTouches?.first
        let point = touch!.location(in: self.newArrivalQuestionListTableView)
        let indexPath = newArrivalQuestionListTableView.indexPathForRow(at: point)!
        presenter.likeButtonTapped(at: indexPath)
    }
}

Answer 1

fetchNewArivalQuestions() が何をしているのか提示されていないので想像の域を出ませんが、completionHandler等で完了を待つ必要があるのではないでしょうか。

Comments

[duck015]

申し訳ございません。func fetchNewArivalQuestions()が記載されているQuestionModelを追記しましたので、ご確認お願いいたします。