Refactor code

Signed-off-by: kim.shen <sjw_job@126.com>