refactor code

Signed-off-by: zjwzcnjsy <zjwzcnjsy@gmail.com>