如何在CollectionView中处理含用户对象与"Yes"标记的Any类型数组

嘿，这个需求其实挺好实现的，我给你拆解成几个核心步骤，一步步来就搞定了：

原来的[Any]数组混着用户对象和字符串，处理起来容易踩类型判断的坑，不如先转成枚举类型，把两种元素明确区分开：

// 先定义你的User模型（假设你已经有了）
struct User {
    var name: String
    // 其他用户属性...
}

// 定义CollectionView的item类型枚举
enum CollectionViewItem {
    case user(User)       // 用户单元格对应的类型
    case dividerImage     // 对应"Yes"的图片单元格类型
}

// 把原数组转换成枚举数组（一次性转换，或者动态扩展时同步转换）
var processedData: [CollectionViewItem] = []
var userCounter = 0

for element in originalAnyArray {
    if let user = element as? User {
        processedData.append(.user(user))
        userCounter += 1
    } else if element as? String == "Yes" {
        processedData.append(.dividerImage)
        userCounter = 0 // 遇到分隔符后重置用户计数，方便后续扩展
    }
}

你需要两个自定义Cell：一个用来展示用户信息，一个用来显示图片。先在viewDidLoad里注册它们：

// 假设UserCell是展示用户的自定义Cell，DividerCell是仅含图片的Cell
collectionView.register(UserCell.self, forCellWithReuseIdentifier: "UserCell")
collectionView.register(DividerCell.self, forCellWithReuseIdentifier: "DividerCell")

接下来就是告诉CollectionView怎么展示数据：

3.1 返回总数量 #

直接用转换后的枚举数组的长度就行：

func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
    return processedData.count
}

3.2 复用并配置对应Cell #

根据枚举类型判断该用哪种Cell，然后配置内容：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let item = processedData[indexPath.item]
    
    switch item {
    case .user(let user):
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "UserCell", for: indexPath) as! UserCell
        // 配置用户Cell的内容，比如展示用户名、头像等
        cell.usernameLabel.text = user.name
        // cell.avatarImageView.image = ...
        return cell
        
    case .dividerImage:
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "DividerCell", for: indexPath) as! DividerCell
        // 配置图片Cell，比如设置你需要展示的图片
        cell.dividerImageView.image = UIImage(named: "your-divider-image")
        return cell
    }
}

3.3 （可选）设置不同Cell的尺寸 #

如果两种Cell大小不一样，在sizeForItemAt里分别设置：

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    let item = processedData[indexPath.item]
    switch item {
    case .user:
        return CGSize(width: collectionView.bounds.width, height: 80) // 用户Cell的尺寸
    case .dividerImage:
        return CGSize(width: collectionView.bounds.width, height: 50) // 图片Cell的尺寸
    }
}

要让CollectionView随数组动态扩展，关键是在加载新数据时，按照「每60个用户加一个图片分隔符」的规则更新数据源，然后刷新CollectionView：

// 假设你通过网络请求获取了新的用户数组newUsers
func loadMoreUsers() {
    // 模拟获取新用户，实际替换成你的网络请求逻辑
    let newUsers: [User] = fetchNewUsers()
    
    var currentUserCount = processedData.filter {
        if case .user = $0 { return true }
        return false
    }.count
    
    // 临时存储要插入的indexPath，用来做批量动画
    var insertIndexPaths: [IndexPath] = []
    let currentDataCount = processedData.count
    
    for user in newUsers {
        processedData.append(.user(user))
        currentUserCount += 1
        insertIndexPaths.append(IndexPath(item: currentDataCount + insertIndexPaths.count, section: 0))
        
        // 每满60个用户就添加一个图片分隔符
        if currentUserCount % 60 == 0 {
            processedData.append(.dividerImage)
            insertIndexPaths.append(IndexPath(item: currentDataCount + insertIndexPaths.count, section: 0))
        }
    }
    
    // 用批量插入动画刷新，比直接reloadData更流畅
    collectionView.performBatchUpdates({
        collectionView.insertItems(at: insertIndexPaths)
    }, completion: nil)
}

如果要实现「滚动到底部自动加载更多」，可以在UIScrollViewDelegate的方法里判断：

func scrollViewDidScroll(_ scrollView: UIScrollView) {
    let offsetY = scrollView.contentOffset.y
    let contentHeight = scrollView.contentSize.height
    let frameHeight = scrollView.bounds.height
    
    // 当滚动到距离底部200pt时触发加载
    if offsetY > contentHeight - frameHeight - 200 {
        // 这里要加一个加载锁，避免重复触发请求
        guard !isLoading else { return }
        isLoading = true
        loadMoreUsers()
        // 请求完成后记得把isLoading设回false
    }
}

内容的提问来源于stack exchange，提问作者kamenr