重写原代码生成器部分逻辑