在collectionView swift中以编程方式无限滚动

import UIKit

class ViewController: UIViewController, UICollectionViewDataSource, UICollectionViewDelegate {
    
    @IBOutlet weak var collectionView: UICollectionView!
    
    var data = [1, 2, 3, 4, 5] // 数据源数组
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 设置collectionView的dataSource和delegate
        collectionView.dataSource = self
        collectionView.delegate = self
        
        // 注册collectionView的cell
        collectionView.register(UICollectionViewCell.self, forCellWithReuseIdentifier: "Cell")
        
        // 设置collectionView的布局
        let layout = UICollectionViewFlowLayout()
        layout.itemSize = CGSize(width: 100, height: 100)
        collectionView.collectionViewLayout = layout
        
        // 设置collectionView的初始偏移量，使其显示第一个item
        collectionView.contentOffset = CGPoint(x: 0, y: 0)
    }
    
    // MARK: - UICollectionViewDataSource
    
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        // 返回数据源数组的count
        return data.count
    }
    
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "Cell", for: indexPath)
        
        // 配置每个item的内容
        cell.backgroundColor = UIColor.blue
        
        return cell
    }
    
    // MARK: - UICollectionViewDelegate
    
    func scrollViewDidEndDecelerating(_ scrollView: UIScrollView) {
        let offsetX = scrollView.contentOffset.x
        let itemWidth = collectionView.frame.size.width / 3 // 假设每行显示3个item
        
        // 当滚动到最后一个item时，重新设置contentOffset为初始位置
        if offsetX >= itemWidth * CGFloat(data.count - 1) {
            collectionView.contentOffset = CGPoint(x: 0, y: 0)
        }
    }
}