Question

我正在实现一个UITableView，它将包含多个部分。部分的数量是动态的。这是一些属性：

public struct User { 
    let userRole: UserRole
}

public enum UserRole: Int, CaseIterable { 
    case superUser, admin, recruiter
}

我的UITableView拥有一个users的数组。部分的数量是userRoles数组中不同的users的数量。稍后，我需要使用此功能来确定每个节中有多少行：

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

我想知道最干净/最快的方法是返回该节中的行数。假设user数组包含2个users，一个数组的userRole为superUser，而另一个recruiter。这些部分为0和1，尽管枚举原始值为0和2。

我不知道如何以干净的方式返回该部分中正确的行数。这是我在写的，但是由于必须有更好的方法而停止了，因为它是：

extension Array where Element: User {
    func contains(userRole: UserRole) -> Bool {
        return contains(where: { $0.userRole == userRole })
    }
}

private func determineUserRoleForSection(section: Int) -> UserRole {
    let containsSuperUser = users.contains(userRole: .superUser)
    let containsAdmin = users.contains(userRole: .admin)
    let containsRecruiter = users.contains(userRole: .recruiter)

    switch section {
    case 0:
        if containsSuperUser {
            return .superUser
        }

        if containsAdmin {
            return .admin
        }

        if containsRecruiter {
            return .recruiter
        }
    case 1:
        // Repeat without superUser case
    case 2:
    default: fatalError()
    }
}

Answer 1

按照建立模型的建议，您可以执行以下操作：

public struct UsersViewModel {
    var users: [User]
    private var sections: [UserRole: [User]] {
        return [
            .admin : users.filter { $0.userRole == .admin },
            .superUser : users.filter { $0.userRole == .superUser },
            .recruiter : users.filter { $0.userRole == .recruiter }
        ]
    }
    var superUserSections: Int { return sections[.superUser]!.count}
    var adminSections: Int { return sections[.admin]!.count }
    var recruiterSections: Int { return sections[.recruiter]!.count }
}

然后只需使用：

public func tableView(_ tableView: UITableView, numberOfSections section: Int) -> Int {
    return 3 // section 0 is superUser, section 1 is admin, section 2 is recruiter.
}

public func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
    switch section {
    case 0:
        return viewModel.superUserSections
    case 1:
        return viewModel.adminSections
    case 2:
        return viewModel.recruiterSections
    }   
}

用部分确定枚举大小写

1 个答案: