整理代码

Signed-off-by: yinlei <yinlei_nb@126.com>