【iOS】collection获取contentsize

项目中有一块需要获取collectionView的ContentSize来设置宽高，但是collectionView在reloadData之后获取contentSize是0，0。
改成layout的contentsize也不行。
后来发现如果想立即布局需要调用layoutIfNeeded

//刷新网格  
 [self.mainCollection reloadData];   

 [self.mainCollection layoutIfNeeded];