重构main中代码