重构代码

Signed-off-by: ci_knight <ci_knight@msn.cn>