IGListKit - добавление нескольких разделов с дополнительным представлением заголовка для каждого

Question

Я пытаюсь реализовать список заказов на акции и принадлежащих акций в UICollectionView. Я использую IGListKit для реализации представления коллекции, и я могу заставить ячейки отображаться правильно. Проблема возникает, когда я пытаюсь добавить Дополнительное представление к OrderStockFeedListSectionController и OwnedStockFeedListSectionController для создания представления заголовка.

---

Если я просто передам модель для заказа, я получу ожидаемый заголовок:

---

Но когда я прохожу обе модели ( Заказ акций и собственных акций), я получаю несколько дублирующих заголовков и даже нижний колонтитул по какой-то странной причине.

TSLA и WF - это заказы на акции | AAPL и MSFT являются собственностью

---

Я уже просматривал документацию IGListKit и реализовал представления заголовков, как они, но я не могу для добавления нескольких заголовков.

Источник данных ListAdapter

    func objects(for listAdapter: ListAdapter) -> [ListDiffable] {
        var orderList: [OrderStockDataBucket] = []
        orderList.append(OrderStockDataBucket(Symbol: "TSLA", Name: "Tesla Motors", Price: 231.32, PercentChange: 2.32))
        orderList.append(OrderStockDataBucket(Symbol: "WF", Name: "Wells Fargo", Price: 231.32, PercentChange: 2.32))
        let orderStockDataBatch = OrderStockDataBatch(List: orderList)

        var ownedList: [OwnedStockDataBucket] = []
        ownedList.append(OwnedStockDataBucket(Symbol: "AAPL", Name: "Apple Inc.", Price: 123.21, PercentChange: 2.32))
        ownedList.append(OwnedStockDataBucket(Symbol: "MSFT", Name: "Microsoft Corp.", Price: 231.23, PercentChange: 22.32))
        let ownedStockDataBatch = OwnedStockDataBatch(List: ownedList)

        return [orderStockDataBatch, ownedStockDataBatch] as [ListDiffable]
    }

    func listAdapter(_ listAdapter: ListAdapter, sectionControllerFor object: Any) -> ListSectionController {
        if let _ = object as? OrderStockDataBatch {
            return OrderStockFeedListSectionController()
        }
        else if let _ = object as? OwnedStockDataBatch{
            return OwnedStockFeedListSectionController()
        }
        print("List Adapter object did not match any type!")
        return ListSectionController()
    }

    func emptyView(for listAdapter: ListAdapter) -> UIView? {
        nil
    }

---

OrderStockFeedListSectionController ( OwnedStockFeedListSectionController точно такой же, я просто изменил метку представления заголовка, чтобы сказать "Ваши акции")

import UIKit
import IGListKit

class OrderStockFeedListSectionController: ListSectionController, ListSupplementaryViewSource {

    private var orderStockDataBatch: OrderStockDataBatch?

    override init() {
        super.init()
        self.supplementaryViewSource = self
    }

    override func numberOfItems() -> Int {
        return orderStockDataBatch?.orderStockDataList.count ?? 0
    }

    override func sizeForItem(at index: Int) -> CGSize {
        let containerWidth = self.collectionContext?.containerSize.width ?? 50
        let containerHeight = self.collectionContext?.containerSize.height ?? 50
        let width: CGFloat!
        if (containerWidth < containerHeight) {
            width = containerWidth
        } else {
            width = containerHeight
        }
        let height = CGFloat(OwnedStockFeedCollectionViewCell.CELL_HEIGHT)
        return CGSize(width: width, height: height)
    }

    override func cellForItem(at index: Int) -> UICollectionViewCell {
        let cell = collectionContext!.dequeueReusableCell(of: OwnedStockFeedCollectionViewCell.self, for: self, at: index) as! OwnedStockFeedCollectionViewCell

        cell.symbolLabel.text = orderStockDataBatch?.orderStockDataList[index].stockSymbol
        cell.nameLabel.text = orderStockDataBatch?.orderStockDataList[index].stockName
        cell.priceLabel.text = "\(orderStockDataBatch?.orderStockDataList[index].stockPrice ?? 0)"
        cell.percentChangeLabel.text = "\(orderStockDataBatch?.orderStockDataList[index].stockPercentChange ?? 0)"
        //cell.percentChangeLabel.backgroundColor = UIColor(named: "MarketGreen")

        return cell
    }

    override func didUpdate(to object: Any) {
        orderStockDataBatch = object as? OrderStockDataBatch
    }



    func supportedElementKinds() -> [String] {
        return [UICollectionView.elementKindSectionHeader]
    }

    func viewForSupplementaryElement(ofKind elementKind: String, at index: Int) -> UICollectionReusableView {
        return userHeaderView(atIndex: index)
    }

    func sizeForSupplementaryView(ofKind elementKind: String, at index: Int) -> CGSize {
        return CGSize(width: collectionContext!.containerSize.width, height: 25)
    }

    private func userHeaderView(atIndex index: Int) -> UICollectionReusableView {
        let view = collectionContext!.dequeueReusableSupplementaryView(ofKind: UICollectionView.elementKindSectionHeader, for: self, class: UICollectionReusableView.self, at: index)

        let label = UIInsetLabel(frame: view.frame)
        label.text = "Your Orders"
        label.textColor = UIColor(named: "DarkLightGray")
        label.font = UIFont(name: "HelveticaNeue-Bold", size: 16)
        label.backgroundColor = UIColor(named: "LightDarkGray")
        label.contentInsets = UIEdgeInsets(top: 0, left: 15, bottom: 0, right: 0)
        view.addSubview(label)

        return view
    }

}

---

I искал ответ уже несколько дней, но я никуда не попал. Пожалуйста, помогите.

Answer 1

Моя проблема была в том, что угур указал. Мне просто нужно было расширить UICollectionReusableView на моем собственном представлении. Тогда я бы просто получил этот тип представления при удалении из представления. Спасибо угур еще раз.