How to add a gradient layer to bezierpath swift

addGradientToBezierPath.swift

func fillTrafficIndicatorBase(ref : CGContextRef, rect : CGRect, cornerRadius : CGFloat, colorLight : CGColorRef, colorDark: CGColorRef, fillWidth : CGFloat){
    CGContextSaveGState(ref)
    let gradient : CGGradientRef = CGGradientCreateWithColors(Colors.genericRGBSpace(), [colorLight, colorDark], nil)
    let path = UIBezierPath(roundedRect: rect, cornerRadius: cornerRadius)
    path.fill()
    path.addClip()
    CGContextDrawLinearGradient(ref, gradient, CGPointMake(CGRectGetMidX(rect), CGRectGetMinY(rect)), CGPointMake(CGRectGetMidX(rect), CGRectGetMaxY(rect)), 0)
    CGContextRestoreGState(ref)
}