前言

其实这篇文章的重点并不在于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下。