整理代码，统一处理逻辑