使用节确定枚举大小写

Question

我正在实现一个UITableView，它将有多个部分。截面的数量是动态的。以下是一些属性：

public struct User { 
let userRole: UserRole
}
public enum UserRole: Int, CaseIterable { 
case superUser, admin, recruiter
}

我的UITableView包含一个users数组。部分的数量是users的阵列中不同的userRoles的数量。稍后，我需要使用以下函数来确定每个部分中有多少行：

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
}   
}