Draw corner round at layoutSubviews ,but not work for all screen size

Question

I want to add top left and top right corner round at first cell and bottom left,right corner round at last cell

I set param corners for cell

and draw corner round use UIBezierPath and CAShapeLayer at layoutSubviews

It works fine at iphone6 screen size

But i did not work at other Screen Size seem cell did not update it's width at 6plus size

here iphone5 screen size miss top left and bottom left corner

Please help me , tried many way didn't find why

Answer 1

I've faced similar problem and applying mask inside

DispatchQueue.main.async {
    //draw
}

block helped.