optimize performance

Reduce memory allocation times

Signed-off-by: Jianhui Zhao <[email protected]>