* add switch
Signed-off-by: yisaer <disxiaofei@163.com>

* address the comment
Signed-off-by: yisaer <disxiaofei@163.com>