restructure code

Signed-off-by: charlown.cai <charlown@163.com>