improved: 重构核心代码