修改，优化

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