代码优化，主要优化状态机