代码结构做了一些调整

[nightzjp]

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