前言
其实这篇文章的重点并不在于runtime，只是我选择了用runtime扩展，而不是继承一个类的方法来实现的。
在想项目开发，会遇到这么一个需求：点击一个按钮变亮，其他按钮变暗，再点击变亮的按钮，按钮又会朝上，再点击又会朝下。这个需求不算大，也不难，找第三方库又显得大才小用了。
图-1.png
如图-1所示，按钮有3个状态，默认状态，选中向上状态，选中向下状态。选中向下的状态，美工没有给，所以，这里要自己变换（刚好可以把业务变的复杂点）。
思路
给UIView扩展一个字典，这个字典里key存状态，value存闭包（现在swift的closure可以转成Any类型了）如下图-2所示
图-2
这里我决定把这个字典属性私有，通过下面图-3两个方法来使用字典（我觉得这样做更容易阅读。添加状态，执行状态）
图-3
最后我们还需要一个记录当前状态的属性，还是用runtime来做
图-4.png
修改executeState 的方法 如下图-5
图-5.png
先看效果图
图-6
关于这些button的点击事件如图-6
图-6.png
根据button的当前的状态，然后去执行应该切换到的状态的closure
看这些代码，是不是觉得代码清晰了很多。
我们只要去配置下 "normal" "selected_up" "selected_down" 的状态就使用这些state的闭包了。
func setupBtnState() {
/**
给每个button添加一个默认的状态
*/
Abtn.addState("normal") { [weak self] in
self?.Abtn.isSelected = false
self?.Abtn.imageView?.transform = .init(rotationAngle: CGFloat(0))
}
Bbtn.addState("normal") { [weak self] in
self?.Bbtn.isSelected = false
self?.Bbtn.imageView?.transform = .init(rotationAngle: CGFloat(0))
}
Cbtn.addState("normal") { [weak self] in
self?.Cbtn.isSelected = false
self?.Cbtn.imageView?.transform = .init(rotationAngle: CGFloat(0))
}
/**
给每个button添加选中向上的状态
*/
Abtn.addState("selected_up") { [weak self] in
self?.Abtn.isSelected = true
self?.Abtn.imageView?.transform = .init(rotationAngle: CGFloat(0))
//其他的两个按钮变成默认的状态
self?.Bbtn.executeState("normal")
self?.Cbtn.executeState("normal")
}
Bbtn.addState("selected_up") { [weak self] in
self?.Bbtn.isSelected = true
self?.Bbtn.imageView?.transform = .init(rotationAngle: CGFloat(0))
//其他的两个按钮变成默认的状态
self?.Abtn.executeState("normal")
self?.Cbtn.executeState("normal")
}
Cbtn.addState("selected_up") { [weak self] in
self?.Cbtn.isSelected = true
self?.Cbtn.imageView?.transform = .init(rotationAngle: CGFloat(0))
//其他的两个按钮变成默认的状态
self?.Bbtn.executeState("normal")
self?.Abtn.executeState("normal")
}
/**
给每个button添加选中向下的时状态。
*/
Abtn.addState("selected_down") { [weak self] in
self?.Abtn.imageView?.transform = .init(rotationAngle: CGFloat(M_PI))
/**
只有在选中向上的情况下才能变成选中向下。
根据需求来选择该状态时的样式，以及其他视图的样式
self?.Abtn.isSelected = true
self?.Bbtn.executeState("normal")
self?.Cbtn.executeState("normal")
*/
}
Bbtn.addState("selected_down") { [weak self] in
self?.Bbtn.imageView?.transform = .init(rotationAngle: CGFloat(M_PI))
}
Cbtn.addState("selected_down") { [weak self] in
self?.Cbtn.imageView?.transform = .init(rotationAngle: CGFloat(M_PI))
}
}
虽然配置代码有点多，但是方便修改，而且可以嵌套执行。比如点击Cbtn的时候还要请求一些数据，就可以把这些逻辑添加到状态selected_down下。
