我们平时写代码,最怕的不是逻辑复杂,而是复杂逻辑全堆在一个类里,变成一座谁都不敢碰的屎山。最近我重构了一下手头模拟项目X里的TestChannel类,这个类光是源码就有四千多行,硬件交互、数据采集、状态管理、事件通知全部搅在一起,加个新功能要同时改七八个方法,跑一次测试要等半天。重构完成之后,代码量虽然没减少太多,但结构清晰了不止一个量级,排查问题的时间从小时级降到了分钟级。这篇东西就是这次重构的完整记录,包括拆分思路、异步改造方案、踩过的坑和验证方法,适合那些手头也有巨型类不知道从哪下刀的开发者参考。
TestChannel这个类的历史可以追溯到项目早期,当时为了快速出Demo,把串口通信、数据解析、状态机、消息分发全塞进了一个类。等到业务量上来之后,这个类就变成了典型的上帝类,几乎每个模块都要跟它打交道,改任何一行代码都要小心翼翼。我这边的重构目标很明确:不改变对外行为,只调整内部结构,把硬件交互、数据采集、状态管理、事件通知拆成独立模块,同时把阻塞式的硬件读写改成异步模型,让整个数据链路不再互相卡脖子。
1. TestChannel为什么会长成一座屎山:先解剖再动手
动手重构之前,我先花了一整天时间把这个类从头到尾读了一遍,把每个方法的作用、调用关系、依赖方向全部画出来。这里分享一下我当时记录的核心问题清单。
1.1 上帝类症状的典型表现:一个类干了四个人的活
TestChannel类的代码大致可以分成四个职责域:底层硬件交互(串口打开、关闭、读写、参数配置)、数据采集与解析(轮询传感器状态、解析二进制协议包、CRC校验)、业务状态管理(连接状态、采集状态、错误状态的状态机流转)、事件通知(向上层模块广播数据更新、错误告警、状态变更)。
这四个职责域之间确实有天然的数据依赖。硬件层读到原始字节,解析层需要把它变成结构化数据,状态管理层要根据数据内容切换状态,事件通知层要把结果推送给上层。但问题是,这些依赖全部通过类的私有字段和方法直接耦合在一起,没有通过接口隔离。比如解析方法直接读写串口对象的内部缓冲,状态管理方法直接修改硬件配置参数,导致任何一个职责域的改动都会牵动其他所有代码。
更麻烦的是线程模型。原来的代码里,硬件读取是通过一个while循环加Thread.sleep实现的,读取线程同时承担了数据解析和状态更新的工作。采集线程、UI刷新线程、命令发送线程共享同一个socket对象,没有任何同步保护。我随便一搜就发现,同一个硬件句柄被三个线程同时读写,这种问题不爆雷纯属运气好。
1.2 不可测试性是最大的隐性成本
做重构决策的时候,我评估了一下当时的测试情况,结果吓人。TestChannel类没有单元测试,因为所有依赖都是直接new出来的,串口对象、配置对象、日志对象全部硬编码在方法内部,根本没法mock。想要验证一个数据解析逻辑,必须插上真实硬件,跑完整数据链路,一旦出错只能靠日志全文检索。
这一点其实是促使我下决心重构的核心原因。功能还能用,但已经失去了演进能力。任何新需求进来,不管是加一个传感器类型还是改一种协议版本,都要在四千行代码里找到对应的分支,然后祈祷不要影响其他路径。项目组里已经连续三个迭代不敢碰这个类了,这种状态必须结束。
1.3 重构策略选择:绞杀者模式而不是推倒重来
有人可能会问,既然这么烂,为什么不直接重写一个类?我的选择是绞杀者模式,保留原有TestChannel的对外接口,一点点把内部实现替换掉。原因有两条。一是这个类被几十处调用,接口签名改动的成本太高,不如让旧接口内部委托给新模块,逐步迁移。二是硬件交互行为的细节非常多,比如某些传感器在异常情况下返回的报文格式不标准,这些兼容逻辑都在旧代码里,直接重写很容易丢掉这些隐式经验。
所以我定下的策略是:先定义模块边界,再逐个抽取,每抽出一个模块就同步补充测试,最后整体切换线程模型。整个过程大概花了三周,前两周是拆分和补测试,最后一周是异步化改造和全量回归。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 按边界切开:六个模块各自管好自己的一亩三分地
这一节说一下拆分出来的模块划分和每个模块的职责边界。我的分法不一定是唯一正确的方案,但在我这个场景下验证下来是靠谱的。
2.1 模块划分的逻辑:从数据流向找切分线
我根据前面画出来的调用关系图,找到三条天然的数据流边界。第一条边界是硬件字节流和结构化数据之间,对应底层通信层和协议解析层。第二条边界是数据生产和数据消费之间,对应采集层和业务处理层。第三条边界是状态变化和外部感知之间,对应状态机层和事件通知层。
沿着这三条边界,我把原类切成了六个模块。
| 模块名 | 职责范围 | 原类中的对应内容 |
|---|---|---|
| ChannelConnector | 负责物理连接的建立、断开、参数配置、重连策略 | open、close、setConfig、心跳检测 |
| ProtocolCodec | 负责二进制协议的编码、解码、CRC校验、粘包拆包 | parsePacket、buildCommand、checksum校验 |
| DataAcquirer | 负责按策略轮询数据、触发采集、向上游请求数据 | startCapture、stopCapture、pollData |
| ChannelStateMachine | 负责连接状态、采集状态、错误状态的状态管理与流转校验 | state字段、transitionTo、checkState |
| StateManager | 负责业务级快照管理、数据缓存、断线续采管理 | snapshot、cache、resume |
| ChannelEventBus | 负责事件发布订阅、生命周期管理、异步分发 | listeners、emitEvent、addListener |
每个模块都是独立的类,有自己的构造函数参数,不直接感知其他模块的存在。它们之间的协作通过接口和事件完成。比如DataAcquirer不关心数据最终去了哪里,采集到结构化数据之后只管往ChannelEventBus投递事件;ChannelStateMachine只知道状态流转规则,不知道状态变化之后业务方要做什么。
2.2 接口设计的原则:模块之间只传递必要信息
模块划分好了,接口设计就是下一步的重头戏。我在设计接口的时候给自己定了三条纪律。
第一条纪律是模块之间只传数据对象,不传内部实现。比如DataAcquirer拿到的是ProtocolCodec解析好的SensorData对象,而不是原始字节流。第二条纪律是任何模块不直接依赖其他模块的具体类。ChannelStateMachine只定义一个StateListener接口,StateManager可以自己实现这个接口来感知状态变化,但ChannelStateMachine不关心实现者是谁。第三条纪律是同步调用和异步回调要分开。查询类操作走同步返回,例如获取当前状态直接拉取;通知类操作走事件异步分发,例如数据到达和状态切换。
这几条纪律在后期改造中帮了很大的忙。因为模块之间没有交叉依赖,我后面把其中一个模块替换成新实现的时候,其他模块完全不用改代码。
2.3 用组合替代继承:把职责外置而不是下沉
原类里其实也做过一些职责划分的尝试,但都是通过继承实现的。有一个基类负责所有硬件协议,子类扩展具体传感器类型,结果子类越来越多,基类越来越臃肿,很多方法子类根本用不上但被迫实现。
这次我一律改成组合模式。TestChannel类现在还保留着,但它变成了一个门面类,内部持有上述六个模块的实例,把所有外部调用委托出去。原本四千行类现在只剩一百多行装配逻辑,核心业务全部下沉到了各模块内部。这样做的另一个好处是,新加一个传感器类型只需要新增一个ProtocolCodec的实现类,不需要改动任何现有模块代码。
3. 异步化改造:从阻塞等待到事件驱动的事件管道
模块拆分完成之后,测试可以跑了,但性能问题依然存在。原来的软件启动流程是:初始化硬件连接,同步等待握手响应,然后进入轮询循环,每次读取都阻塞在IO上。我用一个干脆的异步模型替换了这套流程,效果是吞吐量提升接近三倍,CPU占用率下降了不少。
3.1 为什么要改异步:三个无法回避的性能痛点
原来的阻塞模型有三个很明显的痛点。第一个是握手和重连期间线程完全空转,一个线程阻塞在等待硬件响应的Socket读操作上,什么活都不干,但占据了整个线程池的一个席位。第二个是采集线程和事件分发线程共用同一个线程,一旦事件回调里有耗时操作,比如写数据库或刷新UI,采集周期就会直接拉长。第三个是多个硬件通道需要同时采集时,只能用一个线程循环轮询不同通道,一个通道慢,全部通道跟着慢。
异步化改造的核心思路,是把所有的IO等待都改成非阻塞模式。连接握手不再同步等待响应,而是注册一个完成回调,等到硬件返回数据时再触发后续逻辑。采集循环不再用Thread.sleep控制节奏,而是通过定时调度器定期触发采集任务,任务本身不等待IO结果,而是发出命令后直接返回,数据到达后通过事件管道回调处理。
3.2 异步方案选型:基于CompletableFuture还是自研事件循环
异步方案上,我对比了CompletableFuture方案和自研事件循环方案,最终选择了自研一个轻量级事件循环,配合原有线程池使用。原因在于硬件交互场景有一些特殊要求,CompletableFuture在超时控制和背压处理方面不够灵活。
CompletableFuture应对一次性异步任务很顺手,比如发起一个命令然后等待响应。但硬件数据是持续不断的流式输入,需要长期维护一个事件注册中心。而且硬件连接断开时,所有正在等待的CompletableFuture需要统一取消,逐个处理非常麻烦。
自研的事件循环结构其实很简单:一个专门的事件线程负责调度,事件队列里放着待分发的数据对象和回调,模块通过ChannelEventBus注册自己感兴趣的事件类型。这个自研方案的核心代码只有两百多行,但比通用异步框架更贴合硬件场景,因为我可以精确控制每个事件的生命周期和处理优先级。
3.3 异步改造的关键一步:把IO线程和业务线程彻底分离
改造过程中最关键的一步,是把IO读写线程和业务处理线程彻底分离。原来的问题本质上是线程职责混杂,IO线程既要读数据又要处理业务逻辑。我现在的结构是这样的:底层有一个专用的IO线程池,只负责读写硬件和收发原始字节;ProtocolCodec在IO线程内完成拆包和解码,解码完成后把结构化数据封装成事件,投递到ChannelEventBus;ChannelEventBus内部有一个独立的派遣线程,负责把事件分发到业务回调。
这样做的好处是两层的延迟互相隔离。如果某个业务回调执行了耗时的操作,比如触发了一次数据库写入,受影响的是业务回调线程池,IO线程不会阻塞,后续硬件数据依然可以正常接入和缓存。这一点在硬件采集场景里太重要了,因为硬件缓冲区很小,如果数据不能及时读走,设备端会直接丢弃数据。
4. 拆分与异步叠加后的大坑:并发、事件顺序与状态一致性
结构拆分和异步改造同步进行,必然会踩一些组合性质的坑。我在这里列出三个比较有代表性的问题,每个都花了不少时间排查,写出来给大家避避雷。
4.1 状态判断出现竞态条件:读取和切换不是原子的
拆分之前,状态字段是在单个线程内被读写,不存在并发问题。拆分后,状态写入发生在IO线程,状态读取发生在业务线程,于是经典的竞态条件出现了。表现是,上层模块拿到状态为CONNECTED之后,准备发送命令,但实际上连接刚刚断开,底层连接已经不存在了。
这个问题的修复方案是给状态变更加版本号。ChannelStateMachine内部维护一个递增的revision,每次状态切换时revision加一。外部模块在读取状态的时候同时读取revision,并保存下来。当外部模块准备基于某个状态执行操作时,先检查当前revision是否还是之前拿到的值,如果不是,说明状态已经变了,操作需要丢弃或者重新评估。这种方式不依赖锁,也不会阻塞高频读取,非常适合状态读取频繁、状态切换相对低频的场景。
4.2 事件顺序错乱:同一条数据链路的不同类型事件被并行分发
事件异步化之后,出现过一个很隐蔽的问题。某一次采集流程中,业务方先收到数据处理完成的事件,然后才收到数据采集开始的事件。原因也很简单,两个事件被投递到了不同的队列,执行线程不同,顺序自然不保证。
解决这个问题的方案是事件分组加顺序队列。每个事件在创建时就指定一个sequenceId,这个序列号由核心的事件生成器统一分配,保证同一数据流上的事件序列号单调递增。ChannelEventBus内部为每个sequenceId维护一个顺序队列,同一链路的事件进入同一个队列,按序号依次分发。不属于同一链路的事件可以并行处理,互不影响。
这个方案比全局串行处理效率高很多,又比完全并行更安全,是目前比较推荐的思路。
4.3 硬件对象生命周期管理:关闭时的悬空引用
还有一个问题是硬件关闭时的对象生命周期管理。原来的代码里,关闭连接就是简单调用socket.close(),但在异步模型下,关闭动作发出时可能还有事件正在投递途中。如果这些事件到达时引用了已经关闭的连接对象,轻则抛异常,重则导致内存泄漏。
我引入了一个统一的关闭协调器。所有模块的关闭方法都接受一个回调参数,关闭协调器会等所有在途事件处理完成之后,再调用实际的关闭动作。同时每个模块在被关闭后会将内部持有的连接对象和回调引用全部置为null,让GC能够正常回收。经历这次改造之后,我连续跑了三天的长时间压测,内存曲线非常平稳,没有出现任何泄露迹象。
5. 验证与迁移路径:让重构平稳落地,不被同事打死
重构这种东西,如果只改不验,改了跟没改一样,甚至更危险。因为读者和下游模块还按照老系统的行为来理解新系统,一旦行为不一致,排查成本会非常高。所以我的验证思路是:先在老系统内部并行运行一套新逻辑,做差分对比,确认无差异后,再逐步把流量从老系统切换到新系统。
5.1 分层测试策略:从解析器开始逐层往上验
我把测试分成了四层。第一层是协议解析层测试,输入手工构造的二进制报文,验证解析结果。这一层最好测,因为输入输出都是纯函数,不需要依赖真实硬件,用模拟数据就能覆盖各种边界情况,包括CRC错误、半包、粘包、超长包等。
第二层是状态机层测试,定义好状态流转规则表,验证合法流转和非法流转的行为。第三层是事件管道测试,验证事件投递顺序、重复订阅、取消订阅、异常回调隔离这些内容。第四层是整体联调测试,接上模拟硬件设备,跑完整的数据采集和事件分发链路,验证端到端的行为和性能。
5.2 回归对比工具:通过影子模式证明行为一致
前面我提到绞杀者模式的迁移策略,实操上我用了影子模式来验证新旧实现的一致性。具体做法是,让老代码和新代码同时接收同一份硬件数据,但新代码只做处理和记录,不对外产生实际影响。然后比较两边处理结果是否一致。
这个对比工具在协议解析阶段最有效。新旧解析器输入同一份原始字节流,输出应该是相同的结构化数据对象。我把两边输出做字段级对比,发现三处不一致,全部是旧代码处理半包情况时的边界条件和新代码不同,最后以新代码为准并补充了测试。这个阶段如果没有影子模式,只靠人工观察日志对比,几乎不可能发现这些细节差异。
5.3 灰度迁移:先替换读链路,再替换写链路
影子模式验证通过后,我开始正式迁移。顺序是先易后难。先切换只影响系统内部行为的部分,比如事件总线内部的实现替换;再切换涉及外部交互的读写链路,因为这部分回滚影响大。
我为切换设计了开关控制逻辑。开关可以动态调整,开启新链路后观察一段时间,如果指标正常就继续保留,如果出现异常就切回旧链路。第一次切换我保留了三天观察期,确认稳定后逐步清理旧代码,最终删除了原来TestChannel内部大约六十个遗留方法,整个类瘦身成包装壳。
6. 拆分之后的维护体验:几周之后回过头看值不值
重构完成后,我又在这个项目里迭代了一个新功能需求,需要增加一种新的采集命令类型。放到以前,这个需求意味着改采集逻辑、改解析逻辑、改状态判断、改事件定义,牵一发动全身。现在只需要新增一个命令构建方法重载,加一个解析分支处理新的响应类型,然后注册一个对应的事件类型,改动范围清晰可控,单测也容易补。
6.1 最值得的投资:测试覆盖率和可调试性
如果问这次重构最值的地方是什么,我会说是测试覆盖率和可调试性。以前想验证一个解析问题,要接硬件,打日志,或者写一堆临时脚本。现在可以直接用模拟报文跑测试,几十毫秒出结果。某次硬件设备返回了一个不符合协议文档的响应包,如果是在旧结构下,我可能要在日志里翻半天才能定位到解析逻辑;现在直接在解析测试里构造了同样的畸形报文,很快就定位到是长度字段解析方式与文档不一致,整个排查过程不到半小时。
6.2 性能数据的实测对比
性能提升的量化数据也补充一下。改造前的采集链路,从硬件响应到业务方收到完整事件,平均耗时大约在480毫秒,其中IO等待占了大半。改造后平均耗时降到不到170毫秒,吞吐量从每秒处理不到50个采集事件提升到每秒接近140个。CPU占用率因为IO等待减少而有小幅下降,线程阻塞时间几乎降到了零。
当然这个数据受硬件本身速度限制,但就这个项目而言,异步模型的收益是实打实的。
6.3 一个值得固化的习惯:定期检查是否存在巨人类
这次重构经历让我养成一个习惯:每个迭代结束后,检查一下代码里最大的几个类,看它们是否还能用一句话说清楚职责。如果存在某一个类,你介绍它的时候需要加好几个“同时”,那基本就是上帝类的前兆。与其等到膨胀到无法收拾再重构,不如在膨胀到两三百行的时候就考虑拆分。
最后分享一个小技巧:重构的时候,保留一份原始类代码的压缩包,不要马上删除,而是放进项目的一个archive目录里。遇到行为不一致的问题,就能直接对照旧实现查找原因。等新代码跑稳了一个月之后再删掉,这样心里踏实很多。
