如何在Swift的扩展代码之前运行函数?

时间:2019-03-22 22:07:08

标签: ios swift xcode uitableview swift4

我正在一个项目上,正在使用tableView加载数据。问题是我需要由特定功能确定的单元数。我的tableView设置了我添加的扩展中的单元格数量,因此无论我在哪里调用该函数,它都仍将运行第二个。任何帮助将不胜感激,这是我的代码(函数和扩展名):

func setNumCells() {
    let uid = Auth.auth().currentUser?.uid
    var ref: DatabaseReference!
    ref = Database.database().reference()

    let applicationReference = ref.child("applications")

    ref.child("applications").child(uid!).observeSingleEvent(of: .value, with: { (snapshot) in
        if let dictionary = snapshot.value as? [String: AnyObject] {
            print("So far")
            let array = Array(dictionary.keys)
            print(array)
            for i in 0..<array.count {
                ref.child("applications").child(uid!).child(String(array[i])).observeSingleEvent(of: .value, with: { (snapshot) in
                    if let dictionary = snapshot.value as? [String: AnyObject] {
                        let array = Array(dictionary.keys)
                        self.numApplications += array.count - 1
                    }
                })
            }
        }
    })
}

... 

extension ApplicationViewController: UITableViewDataSource {

    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return numApplications
    }

    func tableView(_ tableView: UITableView, cellForRowAt inde xPath: IndexPath) -> UITableViewCell {
        let cell = UITableViewCell()
        cell.backgroundColor = UIColor.red
        tableView.rowHeight = 85
        cell.textLabel?.text = "\(indexPath.row)"

        return cell
    }
}

1 个答案:

答案 0 :(得分:1)

收到所有数据后,必须在表视图和主线程上调用reloadData

推荐用于处理时间的API为DispatchGroup

func setNumCells() {
    let uid = Auth.auth().currentUser?.uid
    var ref: DatabaseReference!
    ref = Database.database().reference()

    let applicationReference = ref.child("applications")
    let group = DispatchGroup()

    ref.child("applications").child(uid!).observeSingleEvent(of: .value, with: { (snapshot) in
        if let dictionary = snapshot.value as? [String: AnyObject] {
            print("So far")
            let array = Array(dictionary.keys)
            print(array)
            for item in array {
                group.enter()
                ref.child("applications").child(uid!).child(String(item)).observeSingleEvent(of: .value, with: { (snapshot) in
                    if let dictionary = snapshot.value as? [String: AnyObject] {
                        let array = Array(dictionary.keys)
                        self.numApplications += array.count - 1
                    }
                    group.leave()
                })
            }
            group.notify(queue: DispatchQueue.main) {
               self.tableView.reloadData()
            }
        }
    })
}

注意:

  • for i in 0..<array.count很可怕,因为实际上不需要该索引。请参阅我改进的代码。
  • 从不使用默认的初始化程序创建表视图单元格。 重复使用它们。

    let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)