调整代码结构