優化代碼

Signed-off-by: wenweiwei <wenweiwei@boromax.com>