修复 bug

leejayID · Jun 2, 2017 · 28f1c70 · 28f1c70
1 parent 38ebe57
commit 28f1c70
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/Linkage/CollectionView/CollectionViewController.swift b/Linkage/CollectionView/CollectionViewController.swift
@@ -131,14 +131,11 @@ extension CollectionViewController {
 
     fileprivate func frameForHeader(section: Int) -> CGRect {
         let indexPath = IndexPath(item: 0, section: section)
-        let attributes = collectionView.layoutAttributesForSupplementaryElement(ofKind: UICollectionElementKindSectionHeader, at: indexPath)
-
+        let attributes = collectionView.collectionViewLayout.layoutAttributesForSupplementaryView(ofKind: UICollectionElementKindSectionHeader, at: indexPath)
         guard let frameForFirstCell = attributes?.frame else {
             return .zero
         }
-
-        let headerHeight = collectionView(collectionView, layout: flowlayout, referenceSizeForHeaderInSection: section).height
-        return frameForFirstCell.offsetBy(dx: 0, dy: -headerHeight);
+        return frameForFirstCell;
     }
 }