修改架构，简化逻辑，去掉多余的类和函数