增加判断

Signed-off-by: taurus <xxh0755@gmail.com>