代码结构做了一些调整

[ ]

作者你好。
简单的调整了下代码结构，看着比较清晰；
已经自测过了。没问题。刚刚提交了pr。