diff --git a/source/dnode/vnode/src/tsdb/tsdbCache.c b/source/dnode/vnode/src/tsdb/tsdbCache.c
index b7e3bb6a07..1216f0da81 100644
--- a/source/dnode/vnode/src/tsdb/tsdbCache.c
+++ b/source/dnode/vnode/src/tsdb/tsdbCache.c
@@ -1703,8 +1703,14 @@ int32_t tsdbCacheGetBatch(STsdb *pTsdb, tb_uid_t uid, SArray *pLastArray, SCache
 
       if (!remainCols) {
         remainCols = taosArrayInit(num_keys, sizeof(SIdxKey));
+        if (!remainCols) {
+          TAOS_RETURN(TSDB_CODE_OUT_OF_MEMORY);
+        }
       }
-      (void)taosArrayPush(remainCols, &(SIdxKey){i, key});
+      if (NULL == taosArrayPush(remainCols, &(SIdxKey){i, key})) {
+        taosArrayDestroy(remainCols);
+        TAOS_RETURN(TSDB_CODE_OUT_OF_MEMORY);
+      };
     }
   }