大模型重构实战:拆分巨类与长函数的系统化方法
一、遗留代码的"脂肪堆积"
一个类写了三年,方法堆到四十个。
一个函数三百行,改一行心惊胆战。
新人接手,光搞清调用关系就花一周。
这类"胖"代码是技术债的重灾区。
它不报错,却让每次改动都变慢、变险。
重构它,是提升可维护性的高回报动作。
大模型擅长读长代码、提拆分方案。
但它不能盲改,否则引入新 bug。
本文探讨如何用 LLM 系统性重构巨类长函数。
二、重构的执行机制
安全重构分三步:理解、提议、验证。
理解段让模型梳理职责与依赖,画出边界。
提议段给出拆分方案,而非直接改。
验证段用测试守住行为不变。
关键在"行为等价"原则。
重构只改结构,不改对外表现。
任何破坏兼容的改动,都要单独评审。
重构的闭环流程具体而言:首先针对巨类或长函数,由 LLM 梳理职责边界并生成拆分方案。随后进入人工确认环节,若方案未通过则调整,若通过则应用重构。接着跑测试守住行为,若测试通过则合入,若失败则回退并定位问题。
关键在"先方案后改动"。
模型先输出设计,人确认再执行。
避免直接大改带来的不可控。
三、生产级重构实现
下面用代码描述基于依赖聚类的拆分建议。
python
from dataclasses import dataclass
from collections import defaultdict
@dataclass
class MethodInfo:
name: str
calls: set[str]
def cluster_methods(methods: list[MethodInfo]) -> list[set[str]]:
"""按方法间调用关系聚类,识别内聚子群,指导拆分"""
graph: defaultdict[str, set[str]] = defaultdict(set)
for m in methods:
for callee in m.calls:
graph[m.name].add(callee)
# 简化:把互相调用紧密的方法归为一组
groups: list[set[str]] = []
for m in methods:
placed = False
for g in groups:
if graph[m.name] & g:
g.add(m.name)
placed = True
break
if not placed:
groups.append({m.name})
return groups
def propose_split(methods: list[MethodInfo]) -> str:
groups = cluster_methods(methods)
if len(groups) <= 1:
return "内聚度高,暂不需拆分"
return f"建议拆分为 {len(groups)} 个职责类: {groups}"
if __name__ == "__main__":
methods = [
MethodInfo("load", {"parse"}),
MethodInfo("parse", set()),
MethodInfo("render", {"format"}),
MethodInfo("format", set()),
]
print(propose_split(methods))
真实重构前会先补测试。
用测试锁住当前行为,重构后对比结果。
无测试的老代码,先补再改,顺序不能反。
四、边界分析与架构权衡
LLM 重构高效,但有禁区。
行为等价难保证 。模型可能"顺手"改了语义。
必须靠测试与人工 review 兜底。
对外接口变更要单独评估,不可随重构溜进去。
大改的风险集中 。一次拆太大,回归面太广。
建议小步多批:先抽方法,再提类,逐步合入。
每步可独立验证,出问题易定位。
领域知识依赖 。哪些该归一类,模型不一定懂业务。
拆分方案要先经人确认,尤其是跨域边界。
模型给结构,人给判断。
测试缺失的硬伤 。没测试,重构等于裸奔。
老代码重构前必须先补关键路径测试。
这是不可跳过的先决条件。
重构的"测试策略"决定成败。没有测试的老代码,重构等于裸奔,任何"行为等价"的声称都无依据。建议先补关键路径的 characterization test( characterization 测试),锁住当前对外行为,再动手重构,改完对比结果一致才算过关。另一个被忽视的点是"重构与功能开发分离":不要在重构的同时加新功能,否则分不清行为变化来自重构还是新需求,回滚也难。最后,重构要给 reviewer 清晰的 diff:拆成"只移动、只改名、只改结构"的小 PR,而非一个大杂烩,便于快速 review 与 bisect。
五、总结
用大模型重构巨类长函数,本质是"模型提结构,人把方向"。
机制上靠理解---提议---验证闭环,以行为等价为红线。
工程上先补测试再改,小步合入。
落地路线:先梳理职责边界生成方案;人确认后小步应用;每步跑测试守行为;接口变更单独评审。重构的价值,是让后来的每一次改动都更轻。