优化代码,解决bug