Swift 如何实现类似kingfisher点语法imageView.kf.setImage

1.声明一个类RoundCorner

public final class RoundCorner<Base> {
    
    public let base : Base
    public init(_ base : Base){
        self.base = base
    }
}

2.声明一个空协议,协议名RoundCornerCompatible

public protocol RoundCornerCompatible {

}

3.在协议扩展中,添加一个只读属性dx,返回一个RoundCorner对象

public extension RoundCornerCompatible {
    public var dx : RoundCorner<Self> {
        get { return RoundCorner(self) }
    }
}

4.将所需要实现.dx语法的Base类遵守RoundCornerCompatible
凡是遵守这个协议RoundCornerCompatible的类,便拥有了.dx这个属性

extension UIView : RoundCornerCompatible {}

到现在为止,凡是UIView的子类,便可以使用.dx,而这个.dx实际上就是RoundCorner这个实例对象,下面给这个对象添加自定义的方法
5.在RoundCorner的扩展中,添加定义的方法

extension RoundCorner where Base: UIView{
    /// 设置一个四角圆角
    ///
    /// - Parameters:
    ///   - radius: 圆角半径
    ///   - cornerColor: 圆角背景色
    public func roundCorner(radius: CGFloat,cornerColor: UIColor)  {
        
        base.layer.dx_roundCorner(radius: radius, cornerColor: cornerColor)
    }
    
    /// 设置一个普通圆角
    ///
    /// - Parameters:
    ///   - radius: 圆角半径
    ///   - cornerColor: 圆角背景色
    ///   - corners: 圆角位置
    public func roundCorner(radius: CGFloat,cornerColor: UIColor,corners: UIRectCorner) {
        base.layer.dx_roundCorner(radius: radius, cornerColor: cornerColor, corners: corners)
    }
    
    ///  设置一个带边框的圆角
    ///
    /// - Parameters:
    ///   - radius: 圆角半径
    ///   - cornerColor: 圆角背景色
    ///   - corners: 圆角位置
    ///   - borderColor: 边框颜色
    ///   - borderWidth: 边框线宽
    func roundCorner(radii: CGSize,cornerColor: UIColor, corners: UIRectCorner, borderColor: UIColor, borderWidth: CGFloat)  {
        base.layer.dx_roundCorner(cornerRadii: radii, cornerColor: cornerColor, corners: corners, borderColor: borderColor, borderWidth: borderWidth)
    }
}

使用示例

let circle = UIView()
circle.dx.roundCorner(radius: 15/2, cornerColor: color)

label.dx.roundCorner(radius: 20/2, cornerColor: UIColor.white,corners: [.topLeft,.topRight, .bottomRight, .bottomLeft])

demo下载示例
参考文章:Kingfisher学习笔记