代码优化.

Signed-off-by: ryan <836045106@qq.com>