From 1416da19670c50f9c42f62f6aa3fda547e4d8c0a Mon Sep 17 00:00:00 2001 From: yihaoDeng Date: Mon, 10 Apr 2023 04:51:00 +0000 Subject: [PATCH] opt write snapshot --- source/libs/stream/src/tstreamFileState.c | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/source/libs/stream/src/tstreamFileState.c b/source/libs/stream/src/tstreamFileState.c index 4b634754c1..e60a789e7b 100644 --- a/source/libs/stream/src/tstreamFileState.c +++ b/source/libs/stream/src/tstreamFileState.c @@ -17,6 +17,7 @@ #include "streamBackendRocksdb.h" #include "taos.h" +#include "tcommon.h" #include "thash.h" #include "tsimplehash.h" @@ -334,8 +335,9 @@ int32_t flushSnapshot(SStreamFileState* pFileState, SStreamSnapshot* pSnapshot, code = streamStatePutBatch_rocksdb(pFileState->pFileStore, batch); streamStateClearBatch(batch); } - code = - streamStatePutBatch(pFileState->pFileStore, "default", batch, pPos->pKey, pPos->pRowBuff, pFileState->rowSize); + + SStateKey sKey = {.key = *((SWinKey*)pPos->pKey), .opNum = ((SStreamState*)pFileState->pFileStore)->number}; + code = streamStatePutBatch(pFileState->pFileStore, "default", batch, &sKey, pPos->pRowBuff, pFileState->rowSize); } if (streamStateGetBatchSize(batch) > 0) { code = streamStatePutBatch_rocksdb(pFileState->pFileStore, batch);