彻底重构代码，优化逻辑