禁用时更改UIButton边框

我想在禁用时隐藏按钮的边框.当它启用时,我设置边框如下:

_btnInit.layer.borderWidth = 1;
_btnInit.layer.borderColor = [[UIColor grayColor]CGColor];

当它被禁用时,我想删除边框.

就像在图片中一样.左侧处于禁用状态,右侧处于正常状态.

子类UIButton并覆盖isEnabled属性,如下所示,

override var isEnabled: Bool {
    didSet {
        if isEnabled {
            self.layer.borderColor = UIColor.black.cgColor
            self.layer.borderWidth = 1.0
        } else {
            self.layer.borderWidth = 0.0
        }
    }
}