代码有点长，还能优化一下