稍微重构了代码
Signed-off-by: caiz <fjcaiz007@sina.com>