网络请求框架设计

网络请求框架设计

前言

---

我们产品线主要生产nas设备，nas设备有个核心功能是远程访问。为了优化用户体验，前期尝试了许多第三方连接方案，导致网络层架构设计的十分复杂。其网络请求流程图如下：

图片的副本.png

我们可以看到，之前的方案逻辑很简单，都是简单的使用if判断连接起来，但是代码都是写在一块，看起来非常混乱，是否配置都是采用block回调，block里面套用block，可读性较差，不方便维护。并且后续如果决策要去除A、B连接方式，那代码层面改动非常大，还有可能会造成逻辑遗漏。综合以上原因，我打算对连接模块进行一次重构。

重构方案

---

前期连接方式多是因为产品希望通过尝试不同厂商的方案寻找最适合的远程连接方式，目前是通过灰度开关进行调节。当后续功能稳定后会把不适合的连接方式进行去除。考虑到这层，我打算采取比较灵活的编程方式AOP编程。经过调研后，打算采用Aspects框架。

思路：我们的应用接口请求都是通过HTTP和HTTPS协议来传输，主要方式是GET、POST和GET长连接。目前的连接模块已经将这3个方法封装，所有请求都会经过这3个核心方法。如下图：

屏幕快照 2019-08-30 下午3.32.57.png

连接方式A、B、C实际上只是把请求的URL和Port端口后替换成不同的服务器，从而达到走不同连接方式的目的，因此只要使用Aspects将网络核心类中的方法替换，根据不同的连接情况将请求导到不同的连接模块就能实现连接切换问题。优化后结构如下：

AOP优化.png

这样实现有如下优点

• 没有block"地狱嵌套",逻辑简单，方法明了

• 辅助连接方式添加方便，可以快速删除，不影响主连接方式的功能，整体实现更加灵活，满足产品频繁变更连接方式的业务需求

• 各个连接模块之间无逻辑耦合

使用Aspects框架

---

Aspects是一个开源的的库,面向切面编程,它能允许你在每一个类和每一个实例中存在的方法里面加入任何代码。可以在方法执行之前或者之后执行,也可以替换掉原有的方法。通过Runtime消息转发实现Hook。Aspects会自动处理超类,比常规方法调用更容易使用,github上Star已经超过6000多,目前已经比较稳定。这里采用的就是它的替换方法，将原先的GET、POST、active方法替换成AOP_GET、AOP_POST、AOP_active。下面以hook GET请求为例：

GET_HooK.png

首先利用Aspects的方法将GET方法替换成AOP_GET,将连接切换的代码逻辑写在AOP_GET方法中。Aspects hook后返回的是NSInvocation，这里需要利用反射的知识对其处理。为了完整将方法和参数"完全"拷贝到AOP_GET中，我这里写了一个通用的模板，如上图所示。不过需要注意的是，我这里只处理了参数为nil，NSNumber,BOOL，long long，NSObject类型，因为我们GET方法的参数只有这些，如果要写通用模板还需要补充其他参数类型的处理。

经过不断踩坑，我这里总结下以下几点：

• [invocation setArgument:&result atIndex:i + 2];这里需要+2，因为第一位和第二位分别被self和方法签名占用，参数是从第3位开始

• 参数为nil可以通过以下代码传参数

参数为nil的处理.png

• 如果方法有返回值，比如我这是返回NSURLSessionDataTask，就需要先调用invoke 然后获取return的值的地址，再返给原先的originalInvocation

返回参数处理.png

总结

---

从实战中可以看出，通过Aspects进行AOP变成可以很好的优化代码结构，对原代码的侵入性低，是另一种比较巧妙的编程方式。