随便重构了下代码

Signed-off-by: steven weng <bee0060@126.com>