iOS app内埋点切面的分析思路（基于swift3）

前言

埋点统计在项目中还是比较常见的，可以用来分析用户的习惯，从而有针对性的去优化app。传统的做法就是在每个具体的事件触发的地方进行埋点，这种方法比较机械，更多的是一项体力活，而且等项目越来越大，埋的点遍布于真个项目中，可能你自己都找不到，非常不利于后期的维护。当然最好的办法就是利用OC的runtime黑魔法，swift也是可以用的，只不过有些方法是不可用的，但是功能是可以满足的，接下来就带大家看一下统计进行埋点的思路分析。

runtime和Method Swizzling

用到的方法大致是：

public func class_getInstanceMethod(_ cls: Swift.AnyClass!, _ name: Selector!) -> Method!

public func class_addMethod(_ cls: Swift.AnyClass!, _ name: Selector!, _ imp: IMP!, _ types: UnsafePointer!) -> Bool

public func method_exchangeImplementations(_ m1: Method!, _ m2: Method!)

大致的思路就是：找到一个底层的方法（就是事件下发都会经过的一个方法），然后利用runtime替换调两个方法的实现。

唯一字符串——identifier生成的规则

为什么要生成identifier？这是为了保证确定某个控件触发的事件是唯一的，事件唯一就能够映射上埋点的事件。那么它的生成规则是什么？
基本上生成规则是：当前某个视图名+某控件+action名称+target名称。
注意swift是具有命名控件规则的，获取的class名一般是这样的：projectname.classname这种格式的，用的时候稍微注意下。

具体的埋点实例

首先定义一个埋点的管理对象：

struct AspectManager {
    static func swizzle(inClass `class`: AnyClass, swizzle : Selector, original: Selector){
        
        let originalMethod = class_getInstanceMethod(`class`, original)
        let swizzleMethod = class_getInstanceMethod(`class`, swizzle)
        
        let didAddMethod = class_addMethod(`class`, original, method_getImplementation(swizzleMethod), method_getTypeEncoding(swizzleMethod))
        
        if didAddMethod {
            class_replaceMethod(`class`, swizzle, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod))
        } else {
            method_exchangeImplementations(originalMethod, swizzleMethod)
        }
    } 
}

用来交换某个类的两个方法的实现。

UIControl

这里我们需要用到的方法是：

open func sendAction(_ action: Selector, to target: Any?, for event: UIEvent?)

只要是继承于UIControl的事件都会经过这个方法，接下来直接上代码：

extension UIControl{
    open override class func initialize() {
        super.initialize()
        
        //make sure this isn't a subclass
        if self !== UIControl.self {
            return
        }
        
        DispatchQueue.once(token: "com.moglo.niqq.UIControl") {
            swizzle()
        }
        
    }
    
    
    fileprivate class func swizzle(){
        let originalSelector = #selector(UIControl.sendAction(_:to:for:))
        let swizzleSelector = #selector(UIControl.nsh_sendAction(_:to:for:))
        AspectManager.swizzle(inClass: self, swizzle: swizzleSelector, original: originalSelector)
    }
    
    
    func nsh_sendAction(_ action: Selector, to target: Any?, for event: UIEvent?){
        //因为交换了两个方法的实现所以不用担心会死循环
        nsh_sendAction(action, to: target, for: event)
        
        //在这里做你想要处理的埋点事件，identifier可以自己配置一个文件，然后按照生成的id去取定义好的一些需要埋点的事件名
    }
}

这样一来所有的UIButton、UITextView等的事件都会被拦截，需要处理的事情可以统一在这里处理

UITableView与UICollectionView

这里我们需要替换的是代理方法的cell的点击事件：

optional public func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath)

那么首先我们需要替换的是代理的设置：

    fileprivate class func swizzle(){
        let originalSelector = #selector(setter: UITableView.delegate)
        let swizzleSelector = #selector(UITableView.nsh_set(delegate:))
        AspectManager.swizzle(inClass: self, swizzle: swizzleSelector, original: originalSelector)
    }
    
    
    func nsh_set(delegate: UITableViewDelegate?){
        nsh_set(delegate: delegate)
        
        guard let delegate =  delegate else {return}
        Logger.Debug(info: "UITableView set delegate:\(delegate)")
        
    }

其次是要给代理添加一个在自定义的方法

    func nsh_tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath){
         nsh_tableView(tableView, didSelectRowAt: indexPath)
         //这里添加你需要的埋点代码
    }

接下来就是直接替换两个类的实现，完整代码：

       func nsh_set(delegate: UITableViewDelegate?){
        nsh_set(delegate: delegate)
        
        guard let delegate =  delegate else {return}
        Logger.Debug(info: "UITableView set delegate:\(delegate)")
        //交换cell点击事件
        
        let originalSelector = #selector(delegate.tableView(_:didSelectRowAt:))
        let swizzleSelector = #selector(UITableView.nsh_tableView(_:didSelectRowAt:))
        let swizzleMethod = class_getInstanceMethod(UITableView.self, swizzleSelector)
        
        let didAddMethod = class_addMethod(type(of: delegate), swizzleSelector, method_getImplementation(swizzleMethod), method_getTypeEncoding(swizzleMethod))
        if didAddMethod{
            let didSelectOriginalMethod = class_getInstanceMethod(type(of: delegate), NSSelectorFromString("nsh_tableView:didSelectRowAt:"))
            let didSelectSwizzledMethod = class_getInstanceMethod(type(of: delegate), originalSelector)
            method_exchangeImplementations(didSelectOriginalMethod, didSelectSwizzledMethod)
        }
        
    }

UICollectionView同理，这里就不在赘述。

这里总结的可能还有不完善的地方，希望大家批评指正