微调代码结构,抽出部分head