#3768

Committed 28 Mar 2025 10:15AM UTC coverage: 33.726% (-0.3%) from 33.993%

Build # #3768

Build Type

push

travis-ci

Committed by

happyguoxy

Commit Message

test:alter lcov result

Run Details

144891 of 592084 branches covered (24.47%)

Branch coverage included in aggregate %.

218795 of 486283 relevant lines covered (44.99%)

765715.29 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

47.37

/source/libs/stream/src/streamBackendRocksdb.c

/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "streamBackendRocksdb.h"
#include "lz4.h"
#include "streamInt.h"
#include "tcommon.h"
#include "tref.h"

typedef struct SCompactFilteFactory {
  void* status;
} SCompactFilteFactory;

typedef struct {
  rocksdb_t*                       db;
  rocksdb_column_family_handle_t** pHandle;
  rocksdb_writeoptions_t*          wOpt;
  rocksdb_readoptions_t*           rOpt;
  rocksdb_options_t**              cfOpt;
  rocksdb_options_t*               dbOpt;
  RocksdbCfParam*                  param;
  void*                            pBackend;
  SListNode*                       pCompareNode;
  rocksdb_comparator_t**           pCompares;
} RocksdbCfInst;

int32_t streamStateOpenBackendCf(void* backend, char* name, char** cfs, int32_t nCf);

void            destroyRocksdbCfInst(RocksdbCfInst* inst);
int32_t         getCfIdx(const char* cfName);
STaskDbWrapper* taskDbOpenImpl(const char* key, char* statePath, char* dbPath);

static int32_t backendCopyFiles(const char* src, const char* dst);

void        destroyCompactFilteFactory(void* arg);
void        destroyCompactFilte(void* arg);
const char* compactFilteFactoryName(void* arg);
const char* compactFilteFactoryNameSess(void* arg);
const char* compactFilteFactoryNameState(void* arg);
const char* compactFilteFactoryNameFunc(void* arg);
const char* compactFilteFactoryNameFill(void* arg);

const char* compactFilteName(void* arg);
const char* compactFilteNameSess(void* arg);
const char* compactFilteNameState(void* arg);
const char* compactFilteNameFill(void* arg);
const char* compactFilteNameFunc(void* arg);

unsigned char compactFilte(void* arg, int level, const char* key, size_t klen, const char* val, size_t vlen,
                           char** newval, size_t* newvlen, unsigned char* value_changed);
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilter(void* arg, rocksdb_compactionfiltercontext_t* ctx);
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterSess(void* arg, rocksdb_compactionfiltercontext_t* ctx);
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterState(void* arg, rocksdb_compactionfiltercontext_t* ctx);
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterFunc(void* arg, rocksdb_compactionfiltercontext_t* ctx);
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterFill(void* arg, rocksdb_compactionfiltercontext_t* ctx);

typedef int (*__db_key_encode_fn_t)(void* key, char* buf);
typedef int (*__db_key_decode_fn_t)(void* key, char* buf);
typedef int (*__db_key_tostr_fn_t)(void* key, char* buf);
typedef const char* (*__db_key_cmpname_fn_t)(void* statue);
typedef int (*__db_key_cmp_fn_t)(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
typedef void (*__db_key_cmp_destroy_fn_t)(void* state);
typedef int32_t (*__db_value_encode_fn_t)(void* value, int32_t vlen, int64_t ttl, char** dest);
typedef int32_t (*__db_value_decode_fn_t)(void* value, int32_t vlen, int64_t* ttl, char** dest);

typedef rocksdb_compactionfilter_t* (*__db_factory_create_fn_t)(void* arg, rocksdb_compactionfiltercontext_t* ctx);
typedef const char* (*__db_factory_name_fn_t)(void* arg);
typedef void (*__db_factory_destroy_fn_t)(void* arg);
typedef struct {
  const char*               key;
  int32_t                   len;
  int                       idx;
  __db_key_cmp_fn_t         cmpKey;
  __db_key_encode_fn_t      enFunc;
  __db_key_decode_fn_t      deFunc;
  __db_key_tostr_fn_t       toStrFunc;
  __db_key_cmpname_fn_t     cmpName;
  __db_key_cmp_destroy_fn_t destroyCmp;
  __db_value_encode_fn_t    enValueFunc;
  __db_value_decode_fn_t    deValueFunc;

  __db_factory_create_fn_t  createFilter;
  __db_factory_destroy_fn_t destroyFilter;
  __db_factory_name_fn_t    funcName;

} SCfInit;

const char* compareDefaultName(void* name);
const char* compareStateName(void* name);
const char* compareWinKeyName(void* name);
const char* compareSessionKeyName(void* name);
const char* compareFuncKeyName(void* name);
const char* compareParKeyName(void* name);
const char* comparePartagKeyName(void* name);

int defaultKeyComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
int defaultKeyEncode(void* k, char* buf);
int defaultKeyDecode(void* k, char* buf);
int defaultKeyToString(void* k, char* buf);

int stateKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
int stateKeyEncode(void* k, char* buf);
int stateKeyDecode(void* k, char* buf);
int stateKeyToString(void* k, char* buf);

int stateSessionKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
int stateSessionKeyEncode(void* ses, char* buf);
int stateSessionKeyDecode(void* ses, char* buf);
int stateSessionKeyToString(void* k, char* buf);

int winKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
int winKeyEncode(void* k, char* buf);
int winKeyDecode(void* k, char* buf);
int winKeyToString(void* k, char* buf);

int tupleKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
int tupleKeyEncode(void* k, char* buf);
int tupleKeyDecode(void* k, char* buf);
int tupleKeyToString(void* k, char* buf);

int parKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen);
int parKeyEncode(void* k, char* buf);
int parKeyDecode(void* k, char* buf);
int parKeyToString(void* k, char* buf);

int32_t valueEncode(void* value, int32_t vlen, int64_t ttl, char** dest);
int32_t valueDecode(void* value, int32_t vlen, int64_t* ttl, char** dest);
int32_t valueToString(void* k, char* buf);
int32_t valueIsStale(void* k, int64_t ts);

void        destroyCompare(void* arg);
static void cleanDir(const char* pPath, const char* id);

static bool                streamStateIterSeekAndValid(rocksdb_iterator_t* iter, char* buf, size_t len);
static rocksdb_iterator_t* streamStateIterCreate(SStreamState* pState, const char* cfName,
                                                 rocksdb_snapshot_t** snapshot, rocksdb_readoptions_t** readOpt);

void taskDbRefChkp(STaskDbWrapper* pTaskDb, int64_t chkp);
void taskDbUnRefChkp(STaskDbWrapper* pTaskDb, int64_t chkp);

int32_t chkpAddExtraInfo(char* pChkpIdDir, int64_t chkpId, int64_t processId);
int32_t chkpLoadExtraInfo(char* pChkpIdDir, int64_t* chkpId, int64_t* processId);

int32_t  copyFiles(const char* src, const char* dst);
uint32_t nextPow2(uint32_t x);

SCfInit ginitDict[] = {
    {"default", 7, 0, defaultKeyComp, defaultKeyEncode, defaultKeyDecode, defaultKeyToString, compareDefaultName,
     destroyCompare, valueEncode, valueDecode, compactFilteFactoryCreateFilter, destroyCompactFilteFactory,
     compactFilteFactoryName},

    {"state", 5, 1, stateKeyDBComp, stateKeyEncode, stateKeyDecode, stateKeyToString, compareStateName, destroyCompare,
     valueEncode, valueDecode, compactFilteFactoryCreateFilterState, destroyCompactFilteFactory,
     compactFilteFactoryNameState},

    {"fill", 4, 2, winKeyDBComp, winKeyEncode, winKeyDecode, winKeyToString, compareWinKeyName, destroyCompare,
     valueEncode, valueDecode, compactFilteFactoryCreateFilterFill, destroyCompactFilteFactory,
     compactFilteFactoryNameFill},

    {"sess", 4, 3, stateSessionKeyDBComp, stateSessionKeyEncode, stateSessionKeyDecode, stateSessionKeyToString,
     compareSessionKeyName, destroyCompare, valueEncode, valueDecode, compactFilteFactoryCreateFilterSess,
     destroyCompactFilteFactory, compactFilteFactoryNameSess},

    {"func", 4, 4, tupleKeyDBComp, tupleKeyEncode, tupleKeyDecode, tupleKeyToString, compareFuncKeyName, destroyCompare,
     valueEncode, valueDecode, compactFilteFactoryCreateFilterFunc, destroyCompactFilteFactory,
     compactFilteFactoryNameFunc},

    {"parname", 7, 5, parKeyDBComp, parKeyEncode, parKeyDecode, parKeyToString, compareParKeyName, destroyCompare,
     valueEncode, valueDecode, compactFilteFactoryCreateFilter, destroyCompactFilteFactory, compactFilteFactoryName},

    {"partag", 6, 6, parKeyDBComp, parKeyEncode, parKeyDecode, parKeyToString, comparePartagKeyName, destroyCompare,
     valueEncode, valueDecode, compactFilteFactoryCreateFilter, destroyCompactFilteFactory, compactFilteFactoryName},
};

int32_t getCfIdx(const char* cfName) {
  int    idx = -1;
  size_t len = strlen(cfName);
  for (int i = 0; i < sizeof(ginitDict) / sizeof(ginitDict[0]); i++) {
    if (len == ginitDict[i].len && strncmp(cfName, ginitDict[i].key, strlen(cfName)) == 0) {
      idx = i;
      break;
    }
  }
  return idx;
}

bool isValidCheckpoint(const char* dir) {
  // not implement yet
  return true;
}

/*
 *copy pChkpIdDir's file to state dir
 */
int32_t rebuildDirFromCheckpoint(const char* path, int64_t chkpId, char** dst) {
  // impl later
  int32_t code = 0;
  int32_t cap = strlen(path) + 64;
  int32_t nBytes = 0;

  char* state = taosMemoryCalloc(1, cap);
  if (state == NULL) {
    return terrno;
  }

  nBytes = snprintf(state, cap, "%s%s%s", path, TD_DIRSEP, "state");
  if (nBytes <= 0 || nBytes >= cap) {
    taosMemoryFree(state);
    return TSDB_CODE_OUT_OF_RANGE;
  }

  if (chkpId != 0) {
    char* chkp = taosMemoryCalloc(1, cap);
    if (chkp == NULL) {
      taosMemoryFree(state);
      return terrno;
    }

    nBytes = snprintf(chkp, cap, "%s%s%s%scheckpoint%" PRId64, path, TD_DIRSEP, "checkpoints", TD_DIRSEP, chkpId);
    if (nBytes <= 0 || nBytes >= cap) {
      taosMemoryFree(state);
      taosMemoryFree(chkp);
      return TSDB_CODE_OUT_OF_RANGE;
    }

    if (taosIsDir(chkp) && isValidCheckpoint(chkp)) {
      cleanDir(state, "");
      code = backendCopyFiles(chkp, state);
      if (code != 0) {
        stError("failed to restart stream backend from %s, reason: %s", chkp, tstrerror(TAOS_SYSTEM_ERROR(code)));
      } else {
        stInfo("start to restart stream backend at checkpoint path: %s", chkp);
      }

    } else {
      stError("failed to start stream backend at %s, reason: %s, restart from default state dir:%s", chkp,
              tstrerror(terrno), state);
      code = taosMkDir(state);
      if (code != 0) {
        code = TAOS_SYSTEM_ERROR(ERRNO);
      }
    }

    taosMemoryFree(chkp);
  }

  *dst = state;
  return code;
}

typedef struct {
  char    pCurrName[24];
  int64_t currChkptId;

  char    pManifestName[24];
  int64_t manifestChkptId;

  char    processName[24];
  int64_t processId;
} SSChkpMetaOnS3;

int32_t remoteChkp_readMetaData(char* path, SSChkpMetaOnS3** pMeta) {
  int32_t   code = 0;
  int32_t   cap = strlen(path) + 32;
  TdFilePtr pFile = NULL;

  char* metaPath = taosMemoryCalloc(1, cap);
  if (metaPath == NULL) {
    return terrno;
  }

  int32_t n = snprintf(metaPath, cap, "%s%s%s", path, TD_DIRSEP, "META");
  if (n <= 0 || n >= cap) {
    taosMemoryFree(metaPath);
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  pFile = taosOpenFile(path, TD_FILE_READ);
  if (pFile == NULL) {
    code = terrno;
    goto _EXIT;
  }

  char buf[256] = {0};
  if (taosReadFile(pFile, buf, sizeof(buf)) <= 0) {
    code = terrno;
    goto _EXIT;
  }

  SSChkpMetaOnS3* p = taosMemoryCalloc(1, sizeof(SSChkpMetaOnS3));
  if (p == NULL) {
    code = terrno;
    goto _EXIT;
  }
  n = sscanf(buf, META_ON_S3_FORMATE, p->pCurrName, &p->currChkptId, p->pManifestName, &p->manifestChkptId,
             p->processName, &p->processId);
  if (n != 6) {
    code = TSDB_CODE_INVALID_MSG;
    taosMemoryFree(p);
    goto _EXIT;
  }

  if (p->currChkptId != p->manifestChkptId) {
    code = TSDB_CODE_INVALID_MSG;
    taosMemoryFree(p);
    goto _EXIT;
  }
  *pMeta = p;
  code = 0;
_EXIT:
  taosCloseFile(&pFile);
  taosMemoryFree(metaPath);
  return code;
}

int32_t remoteChkp_validAndCvtMeta(char* path, SSChkpMetaOnS3* pMeta, int64_t chkpId) {
  int32_t code = 0;
  int32_t nBytes = 0;

  int32_t cap = strlen(path) + 64;
  char*   src = taosMemoryCalloc(1, cap);
  char*   dst = taosMemoryCalloc(1, cap);
  if (src == NULL || dst == NULL) {
    code = terrno;
    goto _EXIT;
  }

  if (pMeta->currChkptId != chkpId || pMeta->manifestChkptId != chkpId) {
    code = TSDB_CODE_INVALID_CFG;
    goto _EXIT;
  }
  // rename current_chkp/mainfest to current
  for (int i = 0; i < 2; i++) {
    char* key = (i == 0 ? pMeta->pCurrName : pMeta->pManifestName);
    if (strlen(key) <= 0) {
      code = TSDB_CODE_INVALID_PARA;
      goto _EXIT;
    }

    nBytes = snprintf(src, cap, "%s%s%s_%" PRId64, path, TD_DIRSEP, key, pMeta->currChkptId);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _EXIT;
    }

    if (taosStatFile(src, NULL, NULL, NULL) != 0) {
      code = terrno;
      goto _EXIT;
    }

    nBytes = snprintf(dst, cap, "%s%s%s", path, TD_DIRSEP, key);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _EXIT;
    }

    code = taosRenameFile(src, dst);
    if (code != 0) {
      goto _EXIT;
    }

    memset(src, 0, cap);
    memset(dst, 0, cap);
  }
  code = 0;

// rename manifest_chkp to manifest
_EXIT:
  taosMemoryFree(src);
  taosMemoryFree(dst);
  return code;
}
int32_t remoteChkpGetDelFile(char* path, SArray* toDel) {
  int32_t code = 0;
  int32_t nBytes = 0;

  SSChkpMetaOnS3* pMeta = NULL;
  code = remoteChkp_readMetaData(path, &pMeta);
  if (code != 0) {
    return code;
  }

  for (int i = 0; i < 2; i++) {
    char* key = (i == 0 ? pMeta->pCurrName : pMeta->pManifestName);

    int32_t cap = strlen(key) + 32;
    char*   p = taosMemoryCalloc(1, cap);
    if (p == NULL) {
      taosMemoryFree(pMeta);
      return terrno;
    }

    nBytes = snprintf(p, cap, "%s_%" PRId64, key, pMeta->currChkptId);
    if (nBytes <= 0 || nBytes >= cap) {
      taosMemoryFree(pMeta);
      taosMemoryFree(p);
      return TSDB_CODE_OUT_OF_RANGE;
    }
    if (taosArrayPush(toDel, &p) == NULL) {
      taosMemoryFree(pMeta);
      taosMemoryFree(p);
      return terrno;
    }
  }

  return 0;
}

void cleanDir(const char* pPath, const char* id) {
  if (pPath == NULL) {
    stError("%s try to clean dir, but path is NULL", id);
    return;
  }

  if (taosIsDir(pPath)) {
    taosRemoveDir(pPath);
    if (taosMkDir(pPath) != 0) {
      stError("%s failed to create dir:%s", id, pPath);
    }
    stInfo("%s clear dir:%s, succ", id, pPath);
  }
}

int32_t createDirIfNotExist(const char* pPath) {
  if (!taosIsDir(pPath)) {
    return taosMulMkDir(pPath);
  } else {
    return 0;
  }
}

int32_t rebuildFromRemoteChkp_rsync(const char* key, char* checkpointPath, int64_t checkpointId, char* defaultPath) {
  int32_t code = 0;
  if (taosIsDir(checkpointPath)) {
    taosRemoveDir(checkpointPath);
    stDebug("remove local checkpoint data dir:%s succ", checkpointPath);
  }

  cleanDir(defaultPath, key);
  stDebug("clear local default dir before downloading checkpoint data:%s succ", defaultPath);

  code = streamTaskDownloadCheckpointData(key, checkpointPath, checkpointId);
  if (code != 0) {
    stError("failed to download checkpoint data:%s", key);
    return code;
  }

  stDebug("download remote checkpoint data for checkpointId:%" PRId64 ", %s", checkpointId, key);
  return backendCopyFiles(checkpointPath, defaultPath);
}

int32_t rebuildDataFromS3(char* chkpPath, int64_t chkpId) {
  SSChkpMetaOnS3* pMeta = NULL;

  int32_t code = remoteChkp_readMetaData(chkpPath, &pMeta);
  if (code != 0) {
    return code;
  }

  if (pMeta->currChkptId != chkpId || pMeta->manifestChkptId != chkpId) {
    taosMemoryFree(pMeta);
    return TSDB_CODE_INVALID_PARA;
  }

  code = remoteChkp_validAndCvtMeta(chkpPath, pMeta, chkpId);
  if (code != 0) {
    taosMemoryFree(pMeta);
    return code;
  }
  taosMemoryFree(pMeta);

  return chkpAddExtraInfo(chkpPath, chkpId, pMeta->processId);
}

int32_t rebuildFromRemoteChkp_s3(const char* key, char* chkpPath, int64_t chkpId, char* defaultPath) {
  int8_t  rename = 0;
  int32_t code = streamTaskDownloadCheckpointData(key, chkpPath, chkpId);
  if (code != 0) {
    return code;
  }

  int32_t cap = strlen(defaultPath) + 32;

  char* defaultTmp = taosMemoryCalloc(1, cap);
  if (defaultTmp == NULL) {
    return terrno;
  }

  int32_t nBytes = snprintf(defaultPath, cap, "%s%s", defaultPath, "_tmp");
  if (nBytes <= 0 || nBytes >= cap) {
    taosMemoryFree(defaultPath);
    return TSDB_CODE_OUT_OF_RANGE;
  }

  if (taosIsDir(defaultTmp)) taosRemoveDir(defaultTmp);
  if (taosIsDir(defaultPath)) {
    code = taosRenameFile(defaultPath, defaultTmp);
    if (code != 0) {
      goto _EXIT;
    } else {
      rename = 1;
    }
  } else {
    code = taosMkDir(defaultPath);
    if (code != 0) {
      code = TAOS_SYSTEM_ERROR(ERRNO);
      goto _EXIT;
    }
  }

  code = rebuildDataFromS3(chkpPath, chkpId);
  if (code != 0) {
    goto _EXIT;
  }

  code = backendCopyFiles(chkpPath, defaultPath);
  if (code != 0) {
    goto _EXIT;
  }
  code = 0;

_EXIT:
  if (code != 0) {
    if (rename) {
      TAOS_UNUSED(taosRenameFile(defaultTmp, defaultPath));
    }
  }

  if (taosIsDir(defaultPath)) {
    taosRemoveDir(defaultPath);
  }

  taosMemoryFree(defaultTmp);
  return code;
}

int32_t rebuildFromRemoteCheckpoint(const char* key, char* checkpointPath, int64_t checkpointId, char* defaultPath) {
  ECHECKPOINT_BACKUP_TYPE type = streamGetCheckpointBackupType();
  if (type == DATA_UPLOAD_S3) {
    return rebuildFromRemoteChkp_s3(key, checkpointPath, checkpointId, defaultPath);
  } else if (type == DATA_UPLOAD_RSYNC) {
    return rebuildFromRemoteChkp_rsync(key, checkpointPath, checkpointId, defaultPath);
  } else {
    stError("%s no remote backup checkpoint data for:%" PRId64, key, checkpointId);
  }

  return -1;
}

int32_t copyFiles_create(char* src, char* dst, int8_t type) {
  // create and copy file
  int32_t err = taosCopyFile(src, dst);

  if (ERRNO == EXDEV || ERRNO == ENOTSUP) {
    SET_ERRNO(0);
    return 0;
  }
  return 0;
}
int32_t copyFiles_hardlink(char* src, char* dst, int8_t type) {
  // same fs and hard link
  return taosLinkFile(src, dst);
}

int32_t backendFileCopyFilesImpl(const char* src, const char* dst) {
  const char* current = "CURRENT";
  size_t      currLen = strlen(current);

  const char* info = "info";
  size_t      infoLen = strlen(info);

  int32_t code = 0;
  int32_t sLen = strlen(src);
  int32_t dLen = strlen(dst);
  int32_t cap = TMAX(sLen, dLen) + 64;
  int32_t nBytes = 0;

  char* srcName = taosMemoryCalloc(1, cap);
  char* dstName = taosMemoryCalloc(1, cap);
  if (srcName == NULL || dstName == NULL) {
    taosMemoryFree(srcName);
    taosMemoryFree(dstName);
    return terrno;
  }

  // copy file to dst
  TdDirPtr pDir = taosOpenDir(src);
  if (pDir == NULL) {
    code = terrno;
    goto _ERROR;
  }

  SET_ERRNO(0);
  TdDirEntryPtr de = NULL;
  while ((de = taosReadDir(pDir)) != NULL) {
    char* name = taosGetDirEntryName(de);
    if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) {
      continue;
    }

    nBytes = snprintf(srcName, cap, "%s%s%s", src, TD_DIRSEP, name);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _ERROR;
    }

    nBytes = snprintf(dstName, cap, "%s%s%s", dst, TD_DIRSEP, name);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _ERROR;
    }

    if (strncmp(name, current, strlen(name) <= currLen ? strlen(name) : currLen) == 0) {
      code = copyFiles_create(srcName, dstName, 0);
      if (code != 0) {
        code = TAOS_SYSTEM_ERROR(ERRNO);
        stError("failed to copy file, detail: %s to %s reason:%s", srcName, dstName, tstrerror(code));
        goto _ERROR;
      }
    } else if (strncmp(name, info, strlen(name) <= infoLen ? strlen(name) : infoLen) == 0) {
      code = copyFiles_create(srcName, dstName, 0);
      if (code != 0) {
        code = TAOS_SYSTEM_ERROR(ERRNO);
        stError("failed to copy file, detail: %s to %s reason:%s", srcName, dstName, tstrerror(code));
        goto _ERROR;
      }

    } else {
      code = copyFiles_hardlink(srcName, dstName, 0);
      if (code != 0) {
        code = TAOS_SYSTEM_ERROR(ERRNO);
        stError("failed to hard link file, detail:%s to %s, reason:%s", srcName, dstName, tstrerror(code));
        goto _ERROR;
      } else {
        stDebug("succ hard link file:%s to %s", srcName, dstName);
      }
    }

    memset(srcName, 0, cap);
    memset(dstName, 0, cap);
  }

  taosMemoryFreeClear(srcName);
  taosMemoryFreeClear(dstName);
  TAOS_UNUSED(taosCloseDir(&pDir));
  return code;

_ERROR:
  taosMemoryFreeClear(srcName);
  taosMemoryFreeClear(dstName);
  TAOS_UNUSED(taosCloseDir(&pDir));
  return code;
}

int32_t backendCopyFiles(const char* src, const char* dst) { return backendFileCopyFilesImpl(src, dst); }

static int32_t rebuildFromLocalCheckpoint(const char* pTaskIdStr, const char* checkpointPath, int64_t checkpointId,
                                          const char* defaultPath, int64_t* processVer) {
  int32_t code = 0;
  cleanDir(defaultPath, pTaskIdStr);

  if (taosIsDir(checkpointPath) && isValidCheckpoint(checkpointPath)) {
    stDebug("%s local checkpoint data existed, checkpointId:%" PRId64 " copy to backend dir", pTaskIdStr, checkpointId);

    code = backendCopyFiles(checkpointPath, defaultPath);
    if (code != TSDB_CODE_SUCCESS) {
      cleanDir(defaultPath, pTaskIdStr);
      stError("%s failed to start stream backend from local %s, reason:%s, try download checkpoint from remote",
              pTaskIdStr, checkpointPath, tstrerror(TAOS_SYSTEM_ERROR(code)));
      code = TSDB_CODE_SUCCESS;
    } else {
      stInfo("%s copy checkpoint data from:%s to:%s succ, try to start stream backend", pTaskIdStr, checkpointPath,
             defaultPath);
    }
  } else {
    code = terrno;
    stError("%s no valid data for checkpointId:%" PRId64 " in %s", pTaskIdStr, checkpointId, checkpointPath);
  }

  return code;
}

int32_t restoreCheckpointData(const char* path, const char* key, int64_t chkptId, char** dbPrefixPath, char** dbPath,
                              int64_t* processVer) {
  int32_t code = 0;

  char* prefixPath = NULL;
  char* defaultPath = NULL;
  char* checkpointPath = NULL;
  char* checkpointRoot = NULL;

  int32_t cap = strlen(path) + 128;
  int32_t nBytes;

  // alloc buf
  prefixPath = taosMemoryCalloc(1, cap);
  defaultPath = taosMemoryCalloc(1, cap);
  checkpointPath = taosMemoryCalloc(1, cap);
  checkpointRoot = taosMemoryCalloc(1, cap);
  if (prefixPath == NULL || defaultPath == NULL || checkpointPath == NULL || checkpointRoot == NULL) {
    code = terrno;
    goto _EXIT;
  }

  nBytes = snprintf(prefixPath, cap, "%s%s%s", path, TD_DIRSEP, key);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _EXIT;
  }

  code = createDirIfNotExist(prefixPath);
  if (code != 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
    goto _EXIT;
  }

  nBytes = snprintf(defaultPath, cap, "%s%s%s", prefixPath, TD_DIRSEP, "state");
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _EXIT;
  }

  code = createDirIfNotExist(defaultPath);
  if (code != 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
    goto _EXIT;
  }

  nBytes = snprintf(checkpointRoot, cap, "%s%s%s", prefixPath, TD_DIRSEP, "checkpoints");
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _EXIT;
  }

  code = createDirIfNotExist(checkpointRoot);
  if (code != 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
    goto _EXIT;
  }

  stDebug("%s check local backend dir:%s, checkpointId:%" PRId64 " succ", key, defaultPath, chkptId);
  if (chkptId > 0) {
    nBytes = snprintf(checkpointPath, cap, "%s%s%s%s%s%" PRId64, prefixPath, TD_DIRSEP, "checkpoints", TD_DIRSEP,
                      "checkpoint", chkptId);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _EXIT;
    }

    code = rebuildFromLocalCheckpoint(key, checkpointPath, chkptId, defaultPath, processVer);
    if (code != 0) {
      code = rebuildFromRemoteCheckpoint(key, checkpointPath, chkptId, defaultPath);
    }

    if (code != 0) {
      stError("failed to start stream backend at %s, restart from defaultPath:%s, reason:%s", checkpointPath,
              defaultPath, tstrerror(code));
      code = 0;  // reset the error code
    }
  } else {  // no valid checkpoint id
    stInfo("%s no valid checkpoint ever generated, no need to copy checkpoint data, clean defaultPath:%s", key,
           defaultPath);
    cleanDir(defaultPath, key);
  }

  *dbPath = defaultPath;
  *dbPrefixPath = prefixPath;
  defaultPath = NULL;
  prefixPath = NULL;

  code = 0;

_EXIT:
  taosMemoryFree(defaultPath);
  taosMemoryFree(prefixPath);
  taosMemoryFree(checkpointPath);
  taosMemoryFree(checkpointRoot);
  return code;
}
bool streamBackendDataIsExist(const char* path, int64_t chkpId) {
  bool    exist = true;
  int32_t cap = strlen(path) + 32;

  char* state = taosMemoryCalloc(1, cap);
  if (state == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return false;
  }

  int16_t nBytes = snprintf(state, cap, "%s%s%s", path, TD_DIRSEP, "state");
  if (nBytes <= 0 || nBytes >= cap) {
    terrno = TSDB_CODE_OUT_OF_RANGE;
    exist = false;
  } else {
    if (!taosDirExist(state)) {
      exist = false;
    }
  }

  taosMemoryFree(state);
  return exist;
}

int32_t streamBackendInit(const char* streamPath, int64_t chkpId, int32_t vgId, SBackendWrapper** pBackend) {
  char*   backendPath = NULL;
  int32_t code = 0;
  int32_t lino = 0;
  char*   err = NULL;
  size_t  nCf = 0;

  *pBackend = NULL;

  code = rebuildDirFromCheckpoint(streamPath, chkpId, &backendPath);
  TSDB_CHECK_CODE(code, lino, _EXIT);

  stDebug("start to init stream backend:%s, checkpointId:%" PRId64 " vgId:%d", backendPath, chkpId, vgId);

  uint32_t         dbMemLimit = nextPow2(tsMaxStreamBackendCache) << 20;
  SBackendWrapper* pHandle = taosMemoryCalloc(1, sizeof(SBackendWrapper));
  TSDB_CHECK_NULL(pHandle, code, lino, _EXIT, terrno);

  pHandle->list = tdListNew(sizeof(SCfComparator));
  TSDB_CHECK_NULL(pHandle->list, code, lino, _EXIT, terrno);

  code = taosThreadMutexInit(&pHandle->mutex, NULL);
  TSDB_CHECK_CODE(code, lino, _EXIT);

  code = taosThreadMutexInit(&pHandle->cfMutex, NULL);
  TSDB_CHECK_CODE(code, lino, _EXIT);

  pHandle->cfInst = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
  TSDB_CHECK_NULL(pHandle->cfInst, code, lino, _EXIT, terrno);

  pHandle->vgId = vgId;

  rocksdb_env_t* env = rocksdb_create_default_env();  // rocksdb_envoptions_create();

  int32_t nBGThread = tsNumOfSnodeStreamThreads <= 2 ? 1 : tsNumOfSnodeStreamThreads / 2;
  rocksdb_env_set_low_priority_background_threads(env, nBGThread);
  rocksdb_env_set_high_priority_background_threads(env, nBGThread);

  rocksdb_cache_t* cache = rocksdb_cache_create_lru(dbMemLimit / 2);

  rocksdb_options_t* opts = rocksdb_options_create();
  rocksdb_options_set_env(opts, env);
  rocksdb_options_set_create_if_missing(opts, 1);
  rocksdb_options_set_create_missing_column_families(opts, 1);
  rocksdb_options_set_max_total_wal_size(opts, dbMemLimit);
  rocksdb_options_set_recycle_log_file_num(opts, 6);
  rocksdb_options_set_max_write_buffer_number(opts, 3);
  rocksdb_options_set_info_log_level(opts, 1);
  rocksdb_options_set_db_write_buffer_size(opts, dbMemLimit);
  rocksdb_options_set_write_buffer_size(opts, dbMemLimit / 2);
  rocksdb_options_set_atomic_flush(opts, 1);

  pHandle->env = env;
  pHandle->dbOpt = opts;
  pHandle->cache = cache;
  pHandle->filterFactory = rocksdb_compactionfilterfactory_create(
      NULL, destroyCompactFilteFactory, compactFilteFactoryCreateFilter, compactFilteFactoryName);
  rocksdb_options_set_compaction_filter_factory(pHandle->dbOpt, pHandle->filterFactory);

  char** cfs = rocksdb_list_column_families(opts, backendPath, &nCf, &err);
  if (nCf == 0 || nCf == 1 || err != NULL) {
    taosMemoryFreeClear(err);
    pHandle->db = rocksdb_open(opts, backendPath, &err);
    if (err != NULL) {
      stError("failed to open rocksdb, path:%s, reason:%s", backendPath, err);
      taosMemoryFreeClear(err);
      rocksdb_list_column_families_destroy(cfs, nCf);
      goto _EXIT;
    }
  } else {
    /*
      list all cf and get prefix
    */
    code = streamStateOpenBackendCf(pHandle, (char*)backendPath, cfs, nCf);
    if (code != 0) {
      rocksdb_list_column_families_destroy(cfs, nCf);
      goto _EXIT;
    }
  }

  if (cfs != NULL) {
    rocksdb_list_column_families_destroy(cfs, nCf);
  }

  stDebug("init stream backend at %s, backend:%p, vgId:%d", backendPath, pHandle, vgId);
  taosMemoryFreeClear(backendPath);

  *pBackend = pHandle;
  return code;

_EXIT:
  rocksdb_options_destroy(opts);
  rocksdb_cache_destroy(cache);
  rocksdb_env_destroy(env);
  streamMutexDestroy(&pHandle->mutex);
  streamMutexDestroy(&pHandle->cfMutex);
  taosHashCleanup(pHandle->cfInst);
  pHandle->list = tdListFree(pHandle->list);
  taosMemoryFree(pHandle);
  stDebug("failed to init stream backend at %s, vgId:%d line:%d code:%s", backendPath, vgId, lino, tstrerror(code));
  taosMemoryFree(backendPath);
  return code;
}

void streamBackendCleanup(void* arg) {
  SBackendWrapper* pHandle = (SBackendWrapper*)arg;

  void* pIter = taosHashIterate(pHandle->cfInst, NULL);
  while (pIter != NULL) {
    RocksdbCfInst* inst = *(RocksdbCfInst**)pIter;
    destroyRocksdbCfInst(inst);
    pIter = taosHashIterate(pHandle->cfInst, pIter);
  }

  taosHashCleanup(pHandle->cfInst);

  if (pHandle->db) {
    rocksdb_close(pHandle->db);
    pHandle->db = NULL;
  }

  rocksdb_options_destroy(pHandle->dbOpt);
  rocksdb_env_destroy(pHandle->env);
  rocksdb_cache_destroy(pHandle->cache);

  SListNode* head = tdListPopHead(pHandle->list);
  while (head != NULL) {
    streamStateDestroyCompar(head->data);
    taosMemoryFree(head);
    head = tdListPopHead(pHandle->list);
  }

  pHandle->list = tdListFree(pHandle->list);
  streamMutexDestroy(&pHandle->mutex);

  streamMutexDestroy(&pHandle->cfMutex);
  stDebug("vgId:%d destroy stream backend:%p", (int32_t)pHandle->vgId, pHandle);
  taosMemoryFree(pHandle);
}

void streamBackendHandleCleanup(void* arg) {
  SBackendCfWrapper* wrapper = arg;
  bool               remove = wrapper->remove;
  TAOS_UNUSED(taosThreadRwlockWrlock(&wrapper->rwLock));

  stDebug("start to do-close backendWrapper %p, %s", wrapper, wrapper->idstr);
  if (wrapper->rocksdb == NULL) {
    TAOS_UNUSED(taosThreadRwlockUnlock(&wrapper->rwLock));
    return;
  }

  int cfLen = sizeof(ginitDict) / sizeof(ginitDict[0]);

  char* err = NULL;
  if (remove) {
    for (int i = 0; i < cfLen; i++) {
      if (wrapper->pHandle[i] != NULL) rocksdb_drop_column_family(wrapper->rocksdb, wrapper->pHandle[i], &err);
      if (err != NULL) {
        stError("failed to drop cf:%s_%s, reason:%s", wrapper->idstr, ginitDict[i].key, err);
        taosMemoryFreeClear(err);
      }
    }
  } else {
    rocksdb_flushoptions_t* flushOpt = rocksdb_flushoptions_create();
    rocksdb_flushoptions_set_wait(flushOpt, 1);

    for (int i = 0; i < cfLen; i++) {
      if (wrapper->pHandle[i] != NULL) rocksdb_flush_cf(wrapper->rocksdb, flushOpt, wrapper->pHandle[i], &err);
      if (err != NULL) {
        stError("failed to flush cf:%s_%s, reason:%s", wrapper->idstr, ginitDict[i].key, err);
        taosMemoryFreeClear(err);
      }
    }
    rocksdb_flushoptions_destroy(flushOpt);
  }

  for (int i = 0; i < cfLen; i++) {
    if (wrapper->pHandle[i] != NULL) {
      rocksdb_column_family_handle_destroy(wrapper->pHandle[i]);
    }
  }
  taosMemoryFreeClear(wrapper->pHandle);

  for (int i = 0; i < cfLen; i++) {
    rocksdb_options_destroy(wrapper->cfOpts[i]);
    rocksdb_block_based_options_destroy(((RocksdbCfParam*)wrapper->param)[i].tableOpt);
  }

  if (remove) {
    streamBackendDelCompare(wrapper->pBackend, wrapper->pComparNode);
  }
  rocksdb_writeoptions_destroy(wrapper->writeOpts);
  wrapper->writeOpts = NULL;

  rocksdb_readoptions_destroy(wrapper->readOpts);
  wrapper->readOpts = NULL;
  taosMemoryFreeClear(wrapper->cfOpts);
  taosMemoryFreeClear(wrapper->param);
  TAOS_UNUSED(taosThreadRwlockUnlock(&wrapper->rwLock));

  TAOS_UNUSED(taosThreadRwlockDestroy(&wrapper->rwLock));
  wrapper->rocksdb = NULL;
  // taosReleaseRef(streamBackendId, wrapper->backendId);

  stDebug("end to do-close backendwrapper %p, %s", wrapper, wrapper->idstr);
  taosMemoryFree(wrapper);
  return;
}

#ifdef BUILD_NO_CALL
int32_t getLatestCheckpoint(void* arg, int64_t* checkpoint) {
  SStreamMeta* pMeta = arg;
  taosWLockLatch(&pMeta->chkpDirLock);
  int64_t tc = 0;
  int32_t sz = taosArrayGetSize(pMeta->chkpSaved);
  if (sz <= 0) {
    taosWUnLockLatch(&pMeta->chkpDirLock);
    return -1;
  } else {
    tc = *(int64_t*)taosArrayGetLast(pMeta->chkpSaved);
  }

  taosArrayPush(pMeta->chkpInUse, &tc);

  *checkpoint = tc;
  taosWUnLockLatch(&pMeta->chkpDirLock);
  return 0;
}
/*
 *  checkpointSave |--cp1--|--cp2--|--cp3--|--cp4--|--cp5--|
 *  chkpInUse: |--cp2--|--cp4--|
 *  chkpInUse is doing translation, cannot del until
 *  replication is finished
 */
int32_t delObsoleteCheckpoint(void* arg, const char* path) {
  SStreamMeta* pMeta = arg;

  taosWLockLatch(&pMeta->chkpDirLock);

  SArray* chkpDel = taosArrayInit(10, sizeof(int64_t));
  SArray* chkpDup = taosArrayInit(10, sizeof(int64_t));

  int64_t firsId = 0;
  if (taosArrayGetSize(pMeta->chkpInUse) >= 1) {
    firsId = *(int64_t*)taosArrayGet(pMeta->chkpInUse, 0);

    for (int i = 0; i < taosArrayGetSize(pMeta->chkpSaved); i++) {
      int64_t id = *(int64_t*)taosArrayGet(pMeta->chkpSaved, i);
      if (id >= firsId) {
        taosArrayPush(chkpDup, &id);
      } else {
        taosArrayPush(chkpDel, &id);
      }
    }
  } else {
    int32_t sz = taosArrayGetSize(pMeta->chkpSaved);
    int32_t dsz = sz - pMeta->chkpCap;  // del size

    for (int i = 0; i < dsz; i++) {
      int64_t id = *(int64_t*)taosArrayGet(pMeta->chkpSaved, i);
      taosArrayPush(chkpDel, &id);
    }
    for (int i = dsz < 0 ? 0 : dsz; i < sz; i++) {
      int64_t id = *(int64_t*)taosArrayGet(pMeta->chkpSaved, i);
      taosArrayPush(chkpDup, &id);
    }
  }
  taosArrayDestroy(pMeta->chkpSaved);
  pMeta->chkpSaved = chkpDup;

  taosWUnLockLatch(&pMeta->chkpDirLock);

  for (int i = 0; i < taosArrayGetSize(chkpDel); i++) {
    int64_t id = *(int64_t*)taosArrayGet(chkpDel, i);
    char    tbuf[256] = {0};
    sprintf(tbuf, "%s%scheckpoint%" PRId64, path, TD_DIRSEP, id);
    if (taosIsDir(tbuf)) {
      taosRemoveDir(tbuf);
    }
  }
  taosArrayDestroy(chkpDel);
  return 0;
}
#endif
/*
 *  checkpointSave |--cp1--|--cp2--|--cp3--|--cp4--|--cp5--|
 *  chkpInUse: |--cp2--|--cp4--|
 *  chkpInUse is doing translation, cannot del until
 *  replication is finished
 */
int32_t chkpMayDelObsolete(void* arg, int64_t chkpId, char* path) {
  int32_t         code = 0;
  STaskDbWrapper* pBackend = arg;
  SArray *        chkpDel = NULL, *chkpDup = NULL;
  TAOS_UNUSED(taosThreadRwlockWrlock(&pBackend->chkpDirLock));

  if (taosArrayPush(pBackend->chkpSaved, &chkpId) == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _exception);
  }

  chkpDel = taosArrayInit(8, sizeof(int64_t));
  if (chkpDel == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _exception);
  }

  chkpDup = taosArrayInit(8, sizeof(int64_t));
  if (chkpDup == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _exception);
  }

  int64_t firsId = 0;
  if (taosArrayGetSize(pBackend->chkpInUse) >= 1) {
    firsId = *(int64_t*)taosArrayGet(pBackend->chkpInUse, 0);

    for (int i = 0; i < taosArrayGetSize(pBackend->chkpSaved); i++) {
      int64_t id = *(int64_t*)taosArrayGet(pBackend->chkpSaved, i);
      if (id >= firsId) {
        if (taosArrayPush(chkpDup, &id) == NULL) {
          TAOS_CHECK_GOTO(terrno, NULL, _exception);
        }
      } else {
        if (taosArrayPush(chkpDel, &id) == NULL) {
          TAOS_CHECK_GOTO(terrno, NULL, _exception);
        }
      }
    }
  } else {
    int32_t sz = taosArrayGetSize(pBackend->chkpSaved);
    int32_t dsz = sz - pBackend->chkpCap;  // del size

    for (int i = 0; i < dsz; i++) {
      int64_t id = *(int64_t*)taosArrayGet(pBackend->chkpSaved, i);
      if (taosArrayPush(chkpDel, &id) == NULL) {
        TAOS_CHECK_GOTO(terrno, NULL, _exception);
      }
    }
    for (int i = dsz < 0 ? 0 : dsz; i < sz; i++) {
      int64_t id = *(int64_t*)taosArrayGet(pBackend->chkpSaved, i);
      if (taosArrayPush(chkpDup, &id) == NULL) {
        TAOS_CHECK_GOTO(terrno, NULL, _exception);
      }
    }
  }

  taosArrayDestroy(pBackend->chkpSaved);
  pBackend->chkpSaved = chkpDup;
  chkpDup = NULL;

  TAOS_UNUSED(taosThreadRwlockUnlock(&pBackend->chkpDirLock));

  for (int i = 0; i < taosArrayGetSize(chkpDel); i++) {
    int64_t id = *(int64_t*)taosArrayGet(chkpDel, i);
    char    tbuf[256] = {0};
    if (snprintf(tbuf, sizeof(tbuf), "%s%scheckpoint%" PRId64, path, TD_DIRSEP, id) >= sizeof(tbuf)) {
      code = TSDB_CODE_OUT_OF_RANGE;
      TAOS_CHECK_GOTO(code, NULL, _exception);
    }

    stInfo("backend remove obsolete checkpoint: %s", tbuf);
    if (taosIsDir(tbuf)) {
      taosRemoveDir(tbuf);
    }
  }
  taosArrayDestroy(chkpDel);
  return 0;
_exception:
  taosArrayDestroy(chkpDup);
  taosArrayDestroy(chkpDel);
  TAOS_UNUSED(taosThreadRwlockUnlock(&pBackend->chkpDirLock));
  return code;
}

int chkpIdComp(const void* a, const void* b) {
  int64_t x = *(int64_t*)a;
  int64_t y = *(int64_t*)b;
  if (x == y) return 0;

  return x < y ? -1 : 1;
}
int32_t taskDbLoadChkpInfo(STaskDbWrapper* pBackend) {
  int32_t code = 0;
  int32_t nBytes = 0;
  int32_t cap = 256;
  char*   pChkpDir = taosMemoryCalloc(1, cap);
  if (pChkpDir == NULL) {
    return terrno;
  }

  nBytes = snprintf(pChkpDir, cap, "%s%s%s", pBackend->path, TD_DIRSEP, "checkpoints");
  if (nBytes >= cap) {
    return TSDB_CODE_OUT_OF_RANGE;
  }
  if (!taosIsDir(pChkpDir)) {
    taosMemoryFree(pChkpDir);
    return 0;
  }
  TdDirPtr pDir = taosOpenDir(pChkpDir);
  if (pDir == NULL) {
    taosMemoryFree(pChkpDir);
    return 0;
  }
  TdDirEntryPtr de = NULL;
  while ((de = taosReadDir(pDir)) != NULL) {
    if (strcmp(taosGetDirEntryName(de), ".") == 0 || strcmp(taosGetDirEntryName(de), "..") == 0) continue;

    if (taosDirEntryIsDir(de)) {
      char    checkpointPrefix[32] = {0};
      int64_t checkpointId = 0;

      int ret = sscanf(taosGetDirEntryName(de), "checkpoint%" PRId64, &checkpointId);
      if (ret == 1) {
        if (taosArrayPush(pBackend->chkpSaved, &checkpointId) == NULL) {
          TAOS_CHECK_GOTO(terrno, NULL, _exception);
        }
      }
    } else {
      continue;
    }
  }
  taosArraySort(pBackend->chkpSaved, chkpIdComp);

  taosMemoryFree(pChkpDir);
  TAOS_UNUSED(taosCloseDir(&pDir));

  return 0;
_exception:
  taosMemoryFree(pChkpDir);
  TAOS_UNUSED(taosCloseDir(&pDir));
  return code;
}
int32_t chkpGetAllDbCfHandle2(STaskDbWrapper* pBackend, rocksdb_column_family_handle_t*** ppHandle) {
  int32_t code = 0;
  SArray* pHandle = taosArrayInit(8, POINTER_BYTES);
  for (int i = 0; i < sizeof(ginitDict) / sizeof(ginitDict[0]); i++) {
    if (pBackend->pCf[i]) {
      rocksdb_column_family_handle_t* p = pBackend->pCf[i];
      if (taosArrayPush(pHandle, &p) == NULL) {
        code = terrno;
        goto _exception;
      }
    }
  }
  int32_t nCf = taosArrayGetSize(pHandle);
  if (nCf == 0) {
    taosArrayDestroy(pHandle);
    return nCf;
  }

  rocksdb_column_family_handle_t** ppCf = taosMemoryCalloc(nCf, sizeof(rocksdb_column_family_handle_t*));
  if (ppCf == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _exception);
  }
  for (int i = 0; i < nCf; i++) {
    ppCf[i] = taosArrayGetP(pHandle, i);
  }

  taosArrayDestroy(pHandle);

  *ppHandle = ppCf;
  return nCf;
_exception:
  taosArrayDestroy(pHandle);
  return code;
}

int32_t chkpDoDbCheckpoint(rocksdb_t* db, char* path) {
  int32_t               code = -1;
  char*                 err = NULL;
  rocksdb_checkpoint_t* cp = rocksdb_checkpoint_object_create(db, &err);
  if (cp == NULL || err != NULL) {
    stError("failed to do checkpoint at:%s, reason:%s", path, err);
    taosMemoryFreeClear(err);
    code = TSDB_CODE_THIRDPARTY_ERROR;
    goto _ERROR;
  }
  rocksdb_checkpoint_create(cp, path, UINT64_MAX, &err);
  if (err != NULL) {
    stError("failed to do checkpoint at:%s, reason:%s", path, err);
    taosMemoryFreeClear(err);
    code = TSDB_CODE_THIRDPARTY_ERROR;
  } else {
    code = 0;
  }
_ERROR:
  rocksdb_checkpoint_object_destroy(cp);
  return code;
}

int32_t chkpPreFlushDb(rocksdb_t* db, rocksdb_column_family_handle_t** cf, int32_t nCf) {
  int   code = 0;
  char* err = NULL;

  rocksdb_flushoptions_t* flushOpt = rocksdb_flushoptions_create();
  if (flushOpt == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  rocksdb_flushoptions_set_wait(flushOpt, 1);

  rocksdb_flush_cfs(db, flushOpt, cf, nCf, &err);
  if (err != NULL) {
    stError("failed to flush db before streamBackend clean up, reason:%s", err);
    taosMemoryFree(err);
    code = TSDB_CODE_THIRDPARTY_ERROR;
  }
  rocksdb_flushoptions_destroy(flushOpt);
  return code;
}

int32_t chkpPreBuildDir(char* path, int64_t chkpId, char** chkpDir, char** chkpIdDir) {
  int32_t code = 0;
  int32_t cap = strlen(path) + 256;
  int32_t nBytes = 0;

  char* pChkpDir = taosMemoryCalloc(1, cap);
  char* pChkpIdDir = taosMemoryCalloc(1, cap);
  if (pChkpDir == NULL || pChkpIdDir == NULL) {
    code = terrno;
    goto _EXIT;
  }

  nBytes = snprintf(pChkpDir, cap, "%s%s%s", path, TD_DIRSEP, "checkpoints");
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _EXIT;
  }

  nBytes = snprintf(pChkpIdDir, cap, "%s%s%s%" PRId64, pChkpDir, TD_DIRSEP, "checkpoint", chkpId);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _EXIT;
  }

  code = taosMulModeMkDir(pChkpDir, 0755, true);
  if (code != 0) {
    code = terrno;
    stError("failed to prepare checkpoint dir, path:%s, reason:%s", path, tstrerror(code));
    goto _EXIT;
  }

  if (taosIsDir(pChkpIdDir)) {
    stInfo("stream rm exist checkpoint%s", pChkpIdDir);
    taosRemoveDir(pChkpIdDir);
  }

  *chkpDir = pChkpDir;
  *chkpIdDir = pChkpIdDir;
  return 0;
_EXIT:
  taosMemoryFree(pChkpDir);
  taosMemoryFree(pChkpIdDir);
  return code;
}

int32_t taskDbBuildSnap(void* arg, SArray* pSnap) {
  // vnode task->db
  SStreamMeta* pMeta = arg;

  streamMutexLock(&pMeta->backendMutex);
  void*   pIter = taosHashIterate(pMeta->pTaskDbUnique, NULL);
  int32_t code = 0;

  while (pIter) {
    STaskDbWrapper* pTaskDb = *(STaskDbWrapper**)pIter;

    void* p = taskDbAddRef(pTaskDb);
    if (p == NULL) {
      terrno = 0;
      pIter = taosHashIterate(pMeta->pTaskDbUnique, pIter);
      continue;
    }

    // add chkpId to in-use-ckpkIdSet
    taskDbRefChkp(pTaskDb, pTaskDb->chkpId);

    code = taskDbDoCheckpoint(pTaskDb, pTaskDb->chkpId, ((SStreamTask*)pTaskDb->pTask)->chkInfo.processedVer);
    if (code != 0) {
      // remove chkpId from in-use-ckpkIdSet
      taskDbUnRefChkp(pTaskDb, pTaskDb->chkpId);
      taskDbRemoveRef(pTaskDb);
      break;
    }

    SStreamTask*    pTask = pTaskDb->pTask;
    SStreamTaskSnap snap = {.streamId = pTask->id.streamId,
                            .taskId = pTask->id.taskId,
                            .chkpId = pTaskDb->chkpId,
                            .dbPrefixPath = taosStrdup(pTaskDb->path)};
    if (snap.dbPrefixPath == NULL) {
      // remove chkpid from chkp-in-use set
      taskDbUnRefChkp(pTaskDb, pTaskDb->chkpId);
      taskDbRemoveRef(pTaskDb);
      code = terrno;
      break;
    }
    if (taosArrayPush(pSnap, &snap) == NULL) {
      taskDbUnRefChkp(pTaskDb, pTaskDb->chkpId);
      taskDbRemoveRef(pTaskDb);
      code = terrno;
      break;
    }

    taskDbRemoveRef(pTaskDb);
    pIter = taosHashIterate(pMeta->pTaskDbUnique, pIter);
  }
  streamMutexUnlock(&pMeta->backendMutex);
  return code;
}
int32_t taskDbDestroySnap(void* arg, SArray* pSnapInfo) {
  if (pSnapInfo == NULL) return 0;
  SStreamMeta* pMeta = arg;
  int32_t      code = 0;
  int32_t      cap = 256;
  int32_t      nBytes = 0;
  streamMutexLock(&pMeta->backendMutex);

  char buf[256] = {0};
  for (int i = 0; i < taosArrayGetSize(pSnapInfo); i++) {
    SStreamTaskSnap* pSnap = taosArrayGet(pSnapInfo, i);
    nBytes = snprintf(buf, cap, "0x%" PRIx64 "-0x%x", pSnap->streamId, (int32_t)pSnap->taskId);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      break;
    }
    STaskDbWrapper** pTaskDb = taosHashGet(pMeta->pTaskDbUnique, buf, strlen(buf));
    if (pTaskDb == NULL || *pTaskDb == NULL) {
      stWarn("stream backend:%p failed to find task db, streamId:% " PRId64, pMeta, pSnap->streamId);
      memset(buf, 0, sizeof(buf));
      continue;
    }
    memset(buf, 0, sizeof(buf));

    taskDbUnRefChkp(*pTaskDb, pSnap->chkpId);
  }
  streamMutexUnlock(&pMeta->backendMutex);
  return code;
}
#ifdef BUILD_NO_CALL
int32_t streamBackendAddInUseChkp(void* arg, int64_t chkpId) {
  // if (arg == NULL) return 0;

  // SStreamMeta* pMeta = arg;
  // taosWLockLatch(&pMeta->chkpDirLock);
  // taosArrayPush(pMeta->chkpInUse, &chkpId);
  // taosWUnLockLatch(&pMeta->chkpDirLock);
  return 0;
}
int32_t streamBackendDelInUseChkp(void* arg, int64_t chkpId) {
  return 0;
  // if (arg == NULL) return 0;

  // SStreamMeta* pMeta = arg;
  // taosWLockLatch(&pMeta->chkpDirLock);
  // if (taosArrayGetSize(pMeta->chkpInUse) > 0) {
  //   int64_t id = *(int64_t*)taosArrayGet(pMeta->chkpInUse, 0);
  //   if (id == chkpId) {
  //     taosArrayPopFrontBatch(pMeta->chkpInUse, 1);
  //   }
  // }
  // taosWUnLockLatch(&pMeta->chkpDirLock);
}
#endif

/*
   0
*/

void* taskAcquireDb(int64_t refId) {
  // acquire
  void* p = taosAcquireRef(taskDbWrapperId, refId);
  return p;
}
void taskReleaseDb(int64_t refId) {
  // release
  TAOS_UNUSED(taosReleaseRef(taskDbWrapperId, refId));
}

int64_t taskGetDBRef(void* arg) {
  if (arg == NULL) return -1;
  STaskDbWrapper* pDb = arg;
  return pDb->refId;
}

int32_t chkpLoadExtraInfo(char* pChkpIdDir, int64_t* chkpId, int64_t* processId) {
  TdFilePtr pFile = NULL;
  int32_t   code = 0;

  char    buf[256] = {0};
  int32_t nBytes = 0;

  int32_t len = strlen(pChkpIdDir);
  if (len == 0) {
    code = TSDB_CODE_INVALID_PARA;
    stError("failed to load extra info, dir:%s, reason:%s", pChkpIdDir, tstrerror(code));
    return code;
  }

  int32_t cap = len + 64;
  char*   pDst = taosMemoryCalloc(1, cap);
  if (pDst == NULL) {
    code = terrno;
    stError("failed to alloc memory to load extra info, dir:%s", pChkpIdDir);
    goto _EXIT;
  }

  nBytes = snprintf(pDst, cap, "%s%sinfo", pChkpIdDir, TD_DIRSEP);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    stError("failed to build dst to load extra info, dir:%s", pChkpIdDir);
    goto _EXIT;
  }

  pFile = taosOpenFile(pDst, TD_FILE_READ);
  if (pFile == NULL) {
    // compatible with previous version
    *processId = -1;
    code = 0;
    stWarn("failed to open file to load extra info, file:%s, reason:%s", pDst, tstrerror(terrno));
    goto _EXIT;
  }

  if (taosReadFile(pFile, buf, sizeof(buf)) <= 0) {
    code = terrno;
    stError("failed to read file to load extra info, file:%s, reason:%s", pDst, tstrerror(code));
    goto _EXIT;
  }

  if (sscanf(buf, "%" PRId64 " %" PRId64, chkpId, processId) < 2) {
    code = TSDB_CODE_INVALID_PARA;
    stError("failed to read file content to load extra info, file:%s, reason:%s", pDst, tstrerror(code));
    goto _EXIT;
  }
  code = 0;
_EXIT:
  taosMemoryFree(pDst);
  TAOS_UNUSED(taosCloseFile(&pFile));
  return code;
}
int32_t chkpAddExtraInfo(char* pChkpIdDir, int64_t chkpId, int64_t processId) {
  int32_t code = 0;

  TdFilePtr pFile = NULL;

  char    buf[256] = {0};
  int32_t nBytes = 0;

  int32_t len = strlen(pChkpIdDir);
  if (len == 0) {
    code = TSDB_CODE_INVALID_PARA;
    stError("failed to add extra info, dir:%s, reason:%s", pChkpIdDir, tstrerror(code));
    return code;
  }

  int32_t cap = len + 64;
  char*   pDst = taosMemoryCalloc(1, cap);
  if (pDst == NULL) {
    code = terrno;
    stError("failed to alloc memory to add extra info, dir:%s", pChkpIdDir);
    goto _EXIT;
  }

  nBytes = snprintf(pDst, cap, "%s%sinfo", pChkpIdDir, TD_DIRSEP);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    stError("failed to build dst to add extra info, dir:%s, reason:%s", pChkpIdDir, tstrerror(code));
    goto _EXIT;
  }

  pFile = taosOpenFile(pDst, TD_FILE_CREATE | TD_FILE_WRITE | TD_FILE_TRUNC);
  if (pFile == NULL) {
    code = terrno;
    stError("failed to open file to add extra info, file:%s, reason:%s", pDst, tstrerror(code));
    goto _EXIT;
  }

  nBytes = snprintf(buf, sizeof(buf), "%" PRId64 " %" PRId64, chkpId, processId);
  if (nBytes <= 0 || nBytes >= sizeof(buf)) {
    code = TSDB_CODE_OUT_OF_RANGE;
    stError("failed to build content to add extra info, dir:%s,reason:%s", pChkpIdDir, tstrerror(code));
    goto _EXIT;
  }

  if (nBytes != taosWriteFile(pFile, buf, nBytes)) {
    code = terrno;
    stError("failed to write file to add extra info, file:%s, reason:%s", pDst, tstrerror(code));
    goto _EXIT;
  }
  code = 0;

_EXIT:
  TAOS_UNUSED(taosCloseFile(&pFile));
  taosMemoryFree(pDst);
  return code;
}
int32_t taskDbDoCheckpoint(void* arg, int64_t chkpId, int64_t processId) {
  STaskDbWrapper* pTaskDb = arg;
  int64_t         st = taosGetTimestampMs();
  int32_t         code = 0;
  int64_t         refId = pTaskDb->refId;

  if (taosAcquireRef(taskDbWrapperId, refId) == NULL) {
    code = terrno;
    return code;
  }

  char* pChkpDir = NULL;
  char* pChkpIdDir = NULL;
  if ((code = chkpPreBuildDir(pTaskDb->path, chkpId, &pChkpDir, &pChkpIdDir)) < 0) {
    goto _EXIT;
  }
  // Get all cf and acquire cfWrappter
  rocksdb_column_family_handle_t** ppCf = NULL;

  int32_t nCf = chkpGetAllDbCfHandle2(pTaskDb, &ppCf);
  stDebug("stream backend:%p start to do checkpoint at:%s, cf num: %d ", pTaskDb, pChkpIdDir, nCf);

  int64_t written = atomic_load_64(&pTaskDb->dataWritten);

  // flush db
  if (written > 0) {
    stDebug("stream backend:%p start to flush db at:%s, data written:%" PRId64, pTaskDb, pChkpIdDir, written);
    code = chkpPreFlushDb(pTaskDb->db, ppCf, nCf);
    if (code != 0) goto _EXIT;
  } else {
    stDebug("stream backend:%p not need flush db at:%s, data written:%" PRId64, pTaskDb, pChkpIdDir, written);
  }

  // do checkpoint
  if ((code = chkpDoDbCheckpoint(pTaskDb->db, pChkpIdDir)) != 0) {
    stError("stream backend:%p failed to do checkpoint at:%s", pTaskDb, pChkpIdDir);
    goto _EXIT;
  } else {
    stDebug("stream backend:%p end to do checkpoint at:%s, time cost:%" PRId64 "ms", pTaskDb, pChkpIdDir,
            taosGetTimestampMs() - st);
  }

  // add extra info to checkpoint
  if ((code = chkpAddExtraInfo(pChkpIdDir, chkpId, processId)) != 0) {
    stError("stream backend:%p failed to add extra info to checkpoint at:%s", pTaskDb, pChkpIdDir);
    goto _EXIT;
  }

  // delete ttl checkpoint
  code = chkpMayDelObsolete(pTaskDb, chkpId, pChkpDir);
  if (code < 0) {
    goto _EXIT;
  }

  TAOS_UNUSED(atomic_store_64(&pTaskDb->dataWritten, 0));
  pTaskDb->chkpId = chkpId;

_EXIT:

  // clear checkpoint dir if failed
  if (code != 0 && pChkpDir != NULL) {
    if (taosDirExist(pChkpIdDir)) {
      taosRemoveDir(pChkpIdDir);
    }
  }
  taosMemoryFree(pChkpIdDir);
  taosMemoryFree(pChkpDir);

  TAOS_UNUSED(taosReleaseRef(taskDbWrapperId, refId));
  taosMemoryFree(ppCf);
  return code;
}

int32_t streamBackendDoCheckpoint(void* arg, int64_t chkpId, int64_t processVer) {
  return taskDbDoCheckpoint(arg, chkpId, processVer);
}

SListNode* streamBackendAddCompare(void* backend, void* arg) {
  SBackendWrapper* pHandle = (SBackendWrapper*)backend;
  SListNode*       node = NULL;
  streamMutexLock(&pHandle->mutex);
  node = tdListAdd(pHandle->list, arg);
  streamMutexUnlock(&pHandle->mutex);
  return node;
}
void streamBackendDelCompare(void* backend, void* arg) {
  SBackendWrapper* pHandle = (SBackendWrapper*)backend;
  SListNode*       node = NULL;
  streamMutexLock(&pHandle->mutex);
  node = tdListPopNode(pHandle->list, arg);
  streamMutexUnlock(&pHandle->mutex);
  if (node) {
    streamStateDestroyCompar(node->data);
    taosMemoryFree(node);
  }
}
void destroyRocksdbCfInst(RocksdbCfInst* inst) {
  int cfLen = sizeof(ginitDict) / sizeof(ginitDict[0]);
  if (inst->pHandle) {
    for (int i = 0; i < cfLen; i++) {
      if (inst->pHandle[i]) rocksdb_column_family_handle_destroy((inst->pHandle)[i]);
    }
    taosMemoryFree(inst->pHandle);
  }

  if (inst->cfOpt) {
    for (int i = 0; i < cfLen; i++) {
      rocksdb_options_destroy(inst->cfOpt[i]);
      rocksdb_block_based_options_destroy(((RocksdbCfParam*)inst->param)[i].tableOpt);
    }
    taosMemoryFreeClear(inst->cfOpt);
    taosMemoryFreeClear(inst->param);
  }
  if (inst->wOpt) rocksdb_writeoptions_destroy(inst->wOpt);
  if (inst->rOpt) rocksdb_readoptions_destroy(inst->rOpt);

  taosMemoryFree(inst);
}

// |key|-----value------|
// |key|ttl|len|userData

int defaultKeyComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen) {
  int len = aLen < bLen ? aLen : bLen;
  int ret = memcmp(aBuf, bBuf, len);
  if (ret == 0) {
    if (aLen < bLen)
      return -1;
    else if (aLen > bLen)
      return 1;
    else
      return 0;
  } else {
    return ret;
  }
}
int streamStateValueIsStale(char* v) {
  int64_t ts = 0;
  TAOS_UNUSED(taosDecodeFixedI64(v, &ts));
  return (ts != 0 && ts < taosGetTimestampMs()) ? 1 : 0;
}
int iterValueIsStale(rocksdb_iterator_t* iter) {
  size_t len;
  char*  v = (char*)rocksdb_iter_value(iter, &len);
  return streamStateValueIsStale(v);
}
int defaultKeyEncode(void* k, char* buf) {
  int len = strlen((char*)k);
  memcpy(buf, (char*)k, len);
  return len;
}
int defaultKeyDecode(void* k, char* buf) {
  int len = strlen(buf);
  memcpy(k, buf, len);
  return len;
}
int defaultKeyToString(void* k, char* buf) {
  // just to debug
  return sprintf(buf, "key: %s", (char*)k);
}
//
//  SStateKey
//  |--groupid--|---ts------|--opNum----|
//  |--uint64_t-|-uint64_t--|--int64_t--|
//
//
//
int stateKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen) {
  SStateKey key1, key2;
  memset(&key1, 0, sizeof(key1));
  memset(&key2, 0, sizeof(key2));

  char* p1 = (char*)aBuf;
  char* p2 = (char*)bBuf;

  p1 = taosDecodeFixedU64(p1, &key1.key.groupId);
  p2 = taosDecodeFixedU64(p2, &key2.key.groupId);

  p1 = taosDecodeFixedI64(p1, &key1.key.ts);
  p2 = taosDecodeFixedI64(p2, &key2.key.ts);

  TAOS_UNUSED(taosDecodeFixedI64(p1, &key1.opNum));
  TAOS_UNUSED(taosDecodeFixedI64(p2, &key2.opNum));

  return stateKeyCmpr(&key1, sizeof(key1), &key2, sizeof(key2));
}

int stateKeyEncode(void* k, char* buf) {
  SStateKey* key = k;
  int        len = 0;
  len += taosEncodeFixedU64((void**)&buf, key->key.groupId);
  len += taosEncodeFixedI64((void**)&buf, key->key.ts);
  len += taosEncodeFixedI64((void**)&buf, key->opNum);
  return len;
}
int stateKeyDecode(void* k, char* buf) {
  SStateKey* key = k;
  int        len = 0;
  char*      p = buf;
  p = taosDecodeFixedU64(p, &key->key.groupId);
  p = taosDecodeFixedI64(p, &key->key.ts);
  p = taosDecodeFixedI64(p, &key->opNum);
  return p - buf;
}

int32_t stateKeyToString(void* k, char* buf) {
  SStateKey* key = k;
  int        n = 0;
  n += sprintf(buf + n, "[groupId:%" PRIu64 ",", key->key.groupId);
  n += sprintf(buf + n, "ts:%" PRIi64 ",", key->key.ts);
  n += sprintf(buf + n, "opNum:%" PRIi64 "]", key->opNum);
  return n;
}

//
// SStateSessionKey
//  |-----------SSessionKey----------|
//  |-----STimeWindow-----|
//  |---skey--|---ekey----|--groupId-|--opNum--|
//  |---int64-|--int64_t--|--uint64--|--int64_t|
// |
//
int stateSessionKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen) {
  SStateSessionKey w1, w2;
  memset(&w1, 0, sizeof(w1));
  memset(&w2, 0, sizeof(w2));

  char* p1 = (char*)aBuf;
  char* p2 = (char*)bBuf;

  p1 = taosDecodeFixedI64(p1, &w1.key.win.skey);
  p2 = taosDecodeFixedI64(p2, &w2.key.win.skey);

  p1 = taosDecodeFixedI64(p1, &w1.key.win.ekey);
  p2 = taosDecodeFixedI64(p2, &w2.key.win.ekey);

  p1 = taosDecodeFixedU64(p1, &w1.key.groupId);
  p2 = taosDecodeFixedU64(p2, &w2.key.groupId);

  p1 = taosDecodeFixedI64(p1, &w1.opNum);
  p2 = taosDecodeFixedI64(p2, &w2.opNum);

  return stateSessionKeyCmpr(&w1, sizeof(w1), &w2, sizeof(w2));
}
int stateSessionKeyEncode(void* k, char* buf) {
  SStateSessionKey* sess = k;
  int               len = 0;
  len += taosEncodeFixedI64((void**)&buf, sess->key.win.skey);
  len += taosEncodeFixedI64((void**)&buf, sess->key.win.ekey);
  len += taosEncodeFixedU64((void**)&buf, sess->key.groupId);
  len += taosEncodeFixedI64((void**)&buf, sess->opNum);
  return len;
}
int stateSessionKeyDecode(void* k, char* buf) {
  SStateSessionKey* sess = k;
  int               len = 0;

  char* p = buf;
  p = taosDecodeFixedI64(p, &sess->key.win.skey);
  p = taosDecodeFixedI64(p, &sess->key.win.ekey);
  p = taosDecodeFixedU64(p, &sess->key.groupId);
  p = taosDecodeFixedI64(p, &sess->opNum);
  return p - buf;
}
int stateSessionKeyToString(void* k, char* buf) {
  SStateSessionKey* key = k;
  int               n = 0;
  n += sprintf(buf + n, "[skey:%" PRIi64 ",", key->key.win.skey);
  n += sprintf(buf + n, "ekey:%" PRIi64 ",", key->key.win.ekey);
  n += sprintf(buf + n, "groupId:%" PRIu64 ",", key->key.groupId);
  n += sprintf(buf + n, "opNum:%" PRIi64 "]", key->opNum);
  return n;
}

/**
 *  SWinKey
 *  |------groupId------|-----ts------|
 *  |------uint64-------|----int64----|
 */
int winKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen) {
  SWinKey w1, w2;
  memset(&w1, 0, sizeof(w1));
  memset(&w2, 0, sizeof(w2));

  char* p1 = (char*)aBuf;
  char* p2 = (char*)bBuf;

  p1 = taosDecodeFixedU64(p1, &w1.groupId);
  p2 = taosDecodeFixedU64(p2, &w2.groupId);

  p1 = taosDecodeFixedI64(p1, &w1.ts);
  p2 = taosDecodeFixedI64(p2, &w2.ts);

  return winKeyCmpr(&w1, sizeof(w1), &w2, sizeof(w2));
}

int winKeyEncode(void* k, char* buf) {
  SWinKey* key = k;
  int      len = 0;
  len += taosEncodeFixedU64((void**)&buf, key->groupId);
  len += taosEncodeFixedI64((void**)&buf, key->ts);
  return len;
}

int winKeyDecode(void* k, char* buf) {
  SWinKey* key = k;
  int      len = 0;
  char*    p = buf;
  p = taosDecodeFixedU64(p, &key->groupId);
  p = taosDecodeFixedI64(p, &key->ts);
  return len;
}

int winKeyToString(void* k, char* buf) {
  SWinKey* key = k;
  int      n = 0;
  n += sprintf(buf + n, "[groupId:%" PRIu64 ",", key->groupId);
  n += sprintf(buf + n, "ts:%" PRIi64 "]", key->ts);
  return n;
}
/*
 * STupleKey
 * |---groupId---|---ts---|---exprIdx---|
 * |---uint64--|---int64--|---int32-----|
 */
int tupleKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen) {
  STupleKey w1, w2;
  memset(&w1, 0, sizeof(w1));
  memset(&w2, 0, sizeof(w2));

  char* p1 = (char*)aBuf;
  char* p2 = (char*)bBuf;

  p1 = taosDecodeFixedU64(p1, &w1.groupId);
  p2 = taosDecodeFixedU64(p2, &w2.groupId);

  p1 = taosDecodeFixedI64(p1, &w1.ts);
  p2 = taosDecodeFixedI64(p2, &w2.ts);

  p1 = taosDecodeFixedI32(p1, &w1.exprIdx);
  p2 = taosDecodeFixedI32(p2, &w2.exprIdx);

  return STupleKeyCmpr(&w1, sizeof(w1), &w2, sizeof(w2));
}

int tupleKeyEncode(void* k, char* buf) {
  STupleKey* key = k;
  int        len = 0;
  len += taosEncodeFixedU64((void**)&buf, key->groupId);
  len += taosEncodeFixedI64((void**)&buf, key->ts);
  len += taosEncodeFixedI32((void**)&buf, key->exprIdx);
  return len;
}
int tupleKeyDecode(void* k, char* buf) {
  STupleKey* key = k;
  int        len = 0;
  char*      p = buf;
  p = taosDecodeFixedU64(p, &key->groupId);
  p = taosDecodeFixedI64(p, &key->ts);
  p = taosDecodeFixedI32(p, &key->exprIdx);
  return len;
}
int tupleKeyToString(void* k, char* buf) {
  int        n = 0;
  STupleKey* key = k;
  n += sprintf(buf + n, "[groupId:%" PRIu64 ",", key->groupId);
  n += sprintf(buf + n, "ts:%" PRIi64 ",", key->ts);
  n += sprintf(buf + n, "exprIdx:%d]", key->exprIdx);
  return n;
}

int parKeyDBComp(void* state, const char* aBuf, size_t aLen, const char* bBuf, size_t bLen) {
  int64_t w1, w2;
  memset(&w1, 0, sizeof(w1));
  memset(&w2, 0, sizeof(w2));
  char* p1 = (char*)aBuf;
  char* p2 = (char*)bBuf;

  TAOS_UNUSED(taosDecodeFixedI64(p1, &w1));
  TAOS_UNUSED(taosDecodeFixedI64(p2, &w2));
  if (w1 == w2) {
    return 0;
  } else {
    return w1 < w2 ? -1 : 1;
  }
}
int parKeyEncode(void* k, char* buf) {
  int64_t* groupid = k;
  int      len = taosEncodeFixedI64((void**)&buf, *groupid);
  return len;
}
int parKeyDecode(void* k, char* buf) {
  char*    p = buf;
  int64_t* groupid = k;

  p = taosDecodeFixedI64(p, groupid);
  return p - buf;
}
int parKeyToString(void* k, char* buf) {
  int64_t* key = k;
  int      n = 0;
  n = sprintf(buf + n, "[groupId:%" PRIi64 "]", *key);
  return n;
}
int32_t valueToString(void* k, char* buf) {
  SStreamValue* key = k;
  int           n = 0;
  n += sprintf(buf + n, "[unixTimestamp:%" PRIi64 ",", key->unixTimestamp);
  n += sprintf(buf + n, "len:%d,", key->len);
  n += sprintf(buf + n, "data:%s]", key->data);
  return n;
}

/*1: stale, 0: no stale*/
int32_t valueIsStale(void* k, int64_t ts) {
  SStreamValue* key = k;
  if (key->unixTimestamp < ts) {
    return 1;
  }
  return 0;
}

void destroyCompare(void* arg) {
  TAOS_UNUSED(arg);
  return;
}

int32_t valueEncode(void* value, int32_t vlen, int64_t ttl, char** dest) {
  int32_t      code = 0;
  SStreamValue key = {.unixTimestamp = ttl, .len = vlen, .rawLen = vlen, .compress = 0, .data = (char*)(value)};
  int32_t      len = 0;
  char*        dst = NULL;
  if (vlen > 512) {
    dst = taosMemoryCalloc(1, vlen + 128);
    if (dst == NULL) {
      return terrno;
    }
    int32_t dstCap = vlen + 128;
    int32_t compressedSize = LZ4_compress_default((char*)value, dst, vlen, dstCap);
    if (compressedSize < vlen) {
      key.compress = 1;
      key.len = compressedSize;
      value = dst;
    }
  }

  if (*dest == NULL) {
    size_t size = sizeof(key.unixTimestamp) + sizeof(key.len) + sizeof(key.rawLen) + sizeof(key.compress) + key.len;
    char*  p = taosMemoryCalloc(1, size);
    if (p == NULL) {
      code = terrno;
      goto _exception;
    }
    char* buf = p;
    len += taosEncodeFixedI64((void**)&buf, key.unixTimestamp);
    len += taosEncodeFixedI32((void**)&buf, key.len);
    len += taosEncodeFixedI32((void**)&buf, key.rawLen);
    len += taosEncodeFixedI8((void**)&buf, key.compress);
    if (value != NULL && key.len != 0) {
      len += taosEncodeBinary((void**)&buf, (char*)value, key.len);
    }
    *dest = p;
  } else {
    char* buf = *dest;
    len += taosEncodeFixedI64((void**)&buf, key.unixTimestamp);
    len += taosEncodeFixedI32((void**)&buf, key.len);
    len += taosEncodeFixedI32((void**)&buf, key.rawLen);
    len += taosEncodeFixedI8((void**)&buf, key.compress);
    if (value != NULL && key.len != 0) {
      len += taosEncodeBinary((void**)&buf, (char*)value, key.len);
    }
  }

  taosMemoryFree(dst);
  return len;
_exception:
  taosMemoryFree(dst);
  return code;
}

/*
 *  ret >= 0 : found valid value
 *  ret < 0 : error or timeout
 */
int32_t valueDecode(void* value, int32_t vlen, int64_t* ttl, char** dest) {
  int32_t      code = -1;
  SStreamValue key = {0};
  char*        p = value;

  char* pCompressData = NULL;
  char* pOutput = NULL;
  if (streamStateValueIsStale(p)) {
    code = TSDB_CODE_INVALID_DATA_FMT;
    goto _EXCEPT;
  }

  p = taosDecodeFixedI64(p, &key.unixTimestamp);
  p = taosDecodeFixedI32(p, &key.len);
  if (key.len == 0) {
    code = 0;
    goto _EXCEPT;
  }
  if (vlen == (sizeof(key.unixTimestamp) + sizeof(key.len) + key.len)) {
    // compatiable with previous data
    p = taosDecodeBinary(p, (void**)&pOutput, key.len);
    if (p == NULL) {
      code = terrno;
      goto _EXCEPT;
    }

  } else {
    p = taosDecodeFixedI32(p, &key.rawLen);
    p = taosDecodeFixedI8(p, &key.compress);
    if (vlen != (sizeof(key.unixTimestamp) + sizeof(key.len) + sizeof(key.rawLen) + sizeof(key.compress) + key.len)) {
      stError("vlen: %d, read len: %d", vlen, key.len);
      code = TSDB_CODE_INVALID_DATA_FMT;
      goto _EXCEPT;
    }
    if (key.compress == 1) {
      p = taosDecodeBinary(p, (void**)&pCompressData, key.len);
      if (p == NULL) {
        code = terrno;
        goto _EXCEPT;
      }
      pOutput = taosMemoryCalloc(1, key.rawLen);
      if (pOutput == NULL) {
        code = terrno;
        goto _EXCEPT;
      }

      int32_t rawLen = LZ4_decompress_safe(pCompressData, pOutput, key.len, key.rawLen);
      if (rawLen != key.rawLen) {
        stError("read invalid read, rawlen: %d, currlen: %d", key.rawLen, key.len);
        code = TSDB_CODE_INVALID_DATA_FMT;
        goto _EXCEPT;
      }
      key.len = rawLen;
    } else {
      p = taosDecodeBinary(p, (void**)&pOutput, key.len);
      if (p == NULL) {
        code = terrno;
        goto _EXCEPT;
      }
    }
  }

  if (ttl != NULL) *ttl = key.unixTimestamp == 0 ? 0 : key.unixTimestamp - taosGetTimestampMs();

  code = 0;
  if (dest) {
    *dest = pOutput;
    pOutput = NULL;
  }
  taosMemoryFree(pCompressData);
  taosMemoryFree(pOutput);
  return key.len;

_EXCEPT:
  if (dest != NULL) *dest = NULL;
  if (ttl != NULL) *ttl = 0;

  taosMemoryFree(pOutput);
  taosMemoryFree(pCompressData);
  return code;
}

const char* compareDefaultName(void* arg) {
  TAOS_UNUSED(arg);
  return ginitDict[0].key;
}
const char* compareStateName(void* arg) {
  TAOS_UNUSED(arg);
  return ginitDict[1].key;
}
const char* compareWinKeyName(void* arg) { return ginitDict[2].key; }
const char* compareSessionKeyName(void* arg) {
  TAOS_UNUSED(arg);
  return ginitDict[3].key;
}
const char* compareFuncKeyName(void* arg) {
  TAOS_UNUSED(arg);
  return ginitDict[4].key;
}
const char* compareParKeyName(void* arg) {
  TAOS_UNUSED(arg);
  return ginitDict[5].key;
}
const char* comparePartagKeyName(void* arg) {
  TAOS_UNUSED(arg);
  return ginitDict[6].key;
}

void destroyCompactFilteFactory(void* arg) {
  if (arg == NULL) return;
}
const char* compactFilteFactoryName(void* arg) {
  SCompactFilteFactory* state = arg;
  return "stream_compact_factory_filter_default";
}
const char* compactFilteFactoryNameSess(void* arg) {
  SCompactFilteFactory* state = arg;
  return "stream_compact_factory_filter_sess";
}
const char* compactFilteFactoryNameState(void* arg) {
  SCompactFilteFactory* state = arg;
  return "stream_compact_factory_filter_state";
}
const char* compactFilteFactoryNameFill(void* arg) {
  SCompactFilteFactory* state = arg;
  return "stream_compact_factory_filter_fill";
}
const char* compactFilteFactoryNameFunc(void* arg) {
  SCompactFilteFactory* state = arg;
  return "stream_compact_factory_filter_func";
}

void          destroyCompactFilte(void* arg) { TAOS_UNUSED(arg); }
unsigned char compactFilte(void* arg, int level, const char* key, size_t klen, const char* val, size_t vlen,
                           char** newval, size_t* newvlen, unsigned char* value_changed) {
  return streamStateValueIsStale((char*)val) ? 1 : 0;
}
const char* compactFilteName(void* arg) { return "stream_filte_default"; }
const char* compactFilteNameSess(void* arg) { return "stream_filte_sess"; }
const char* compactFilteNameState(void* arg) { return "stream_filte_state"; }
const char* compactFilteNameFill(void* arg) { return "stream_filte_fill"; }
const char* compactFilteNameFunc(void* arg) { return "stream_filte_func"; }

unsigned char compactFilteSess(void* arg, int level, const char* key, size_t klen, const char* val, size_t vlen,
                               char** newval, size_t* newvlen, unsigned char* value_changed) {
  // not impl yet
  return 0;
}

unsigned char compactFilteState(void* arg, int level, const char* key, size_t klen, const char* val, size_t vlen,
                                char** newval, size_t* newvlen, unsigned char* value_changed) {
  // not impl yet
  return 0;
}

unsigned char compactFilteFill(void* arg, int level, const char* key, size_t klen, const char* val, size_t vlen,
                               char** newval, size_t* newvlen, unsigned char* value_changed) {
  // not impl yet
  return 0;
}

unsigned char compactFilteFunc(void* arg, int level, const char* key, size_t klen, const char* val, size_t vlen,
                               char** newval, size_t* newvlen, unsigned char* value_changed) {
  // not impl yet
  return 0;
  // return streamStateValueIsStale((char*)val) ? 1 : 0;
}

rocksdb_compactionfilter_t* compactFilteFactoryCreateFilter(void* arg, rocksdb_compactionfiltercontext_t* ctx) {
  SCompactFilteFactory*       state = arg;
  rocksdb_compactionfilter_t* filter =
      rocksdb_compactionfilter_create(state, destroyCompactFilte, compactFilte, compactFilteName);
  return filter;
}
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterSess(void* arg, rocksdb_compactionfiltercontext_t* ctx) {
  SCompactFilteFactory*       state = arg;
  rocksdb_compactionfilter_t* filter =
      rocksdb_compactionfilter_create(state, destroyCompactFilte, compactFilteSess, compactFilteNameSess);
  return filter;
}
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterState(void* arg, rocksdb_compactionfiltercontext_t* ctx) {
  SCompactFilteFactory*       state = arg;
  rocksdb_compactionfilter_t* filter =
      rocksdb_compactionfilter_create(state, destroyCompactFilte, compactFilteState, compactFilteNameState);
  return filter;
}
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterFill(void* arg, rocksdb_compactionfiltercontext_t* ctx) {
  SCompactFilteFactory*       state = arg;
  rocksdb_compactionfilter_t* filter =
      rocksdb_compactionfilter_create(state, destroyCompactFilte, compactFilteFill, compactFilteNameFill);
  return filter;
}
rocksdb_compactionfilter_t* compactFilteFactoryCreateFilterFunc(void* arg, rocksdb_compactionfiltercontext_t* ctx) {
  SCompactFilteFactory*       state = arg;
  rocksdb_compactionfilter_t* filter =
      rocksdb_compactionfilter_create(state, destroyCompactFilte, compactFilteFunc, compactFilteNameFunc);
  return filter;
}

int32_t taskDbOpenCfs(STaskDbWrapper* pTask, char* path, char** pCfNames, int32_t nCf) {
  int32_t code = -1;
  char*   err = NULL;

  rocksdb_options_t** cfOpts = taosMemoryCalloc(nCf, sizeof(rocksdb_options_t*));
  if (cfOpts == NULL) {
    return terrno;
  }
  rocksdb_column_family_handle_t** cfHandle = taosMemoryCalloc(nCf, sizeof(rocksdb_column_family_handle_t*));
  if (cfHandle == NULL) {
    return terrno;
  }

  for (int i = 0; i < nCf; i++) {
    int32_t idx = getCfIdx(pCfNames[i]);
    cfOpts[i] = pTask->pCfOpts[idx];
  }

  rocksdb_t* db = rocksdb_open_column_families(pTask->dbOpt, path, nCf, (const char* const*)pCfNames,
                                               (const rocksdb_options_t* const*)cfOpts, cfHandle, &err);

  if (err != NULL) {
    stError("failed to open cf path: %s", err);
    taosMemoryFree(err);
    goto _EXIT;
  }

  for (int i = 0; i < nCf; i++) {
    int32_t idx = getCfIdx(pCfNames[i]);
    pTask->pCf[idx] = cfHandle[i];
  }

  pTask->db = db;
  code = 0;

_EXIT:
  taosMemoryFree(cfOpts);
  taosMemoryFree(cfHandle);
  return code;
}

void* taskDbAddRef(void* pTaskDb) {
  STaskDbWrapper* pBackend = pTaskDb;
  return taosAcquireRef(taskDbWrapperId, pBackend->refId);
}

void taskDbRemoveRef(void* pTaskDb) {
  if (pTaskDb == NULL) {
    return;
  }

  STaskDbWrapper* pBackend = pTaskDb;
  TAOS_UNUSED(taosReleaseRef(taskDbWrapperId, pBackend->refId));
}

void taskDbSetClearFileFlag(void* pTaskDb) {
  if (pTaskDb == NULL) {
    return;
  }

  STaskDbWrapper* pBackend = pTaskDb;
  atomic_store_8(&pBackend->removeAllFiles, 1);
}

void taskDbInitOpt(STaskDbWrapper* pTaskDb) {
  rocksdb_env_t* env = rocksdb_create_default_env();

  rocksdb_cache_t*   cache = rocksdb_cache_create_lru(256);
  rocksdb_options_t* opts = rocksdb_options_create();
  rocksdb_options_set_env(opts, env);
  rocksdb_options_set_create_if_missing(opts, 1);
  rocksdb_options_set_create_missing_column_families(opts, 1);
  // rocksdb_options_set_max_total_wal_size(opts, dbMemLimit);
  // rocksdb_options_set_ecycle_log_file_num(opts, 6);
  rocksdb_options_set_max_write_buffer_number(opts, 3);
  rocksdb_options_set_info_log_level(opts, 1);
  rocksdb_options_set_db_write_buffer_size(opts, 256 << 20);
  rocksdb_options_set_write_buffer_size(opts, 128 << 20);
  rocksdb_options_set_atomic_flush(opts, 1);

  pTaskDb->dbOpt = opts;
  pTaskDb->env = env;
  pTaskDb->cache = cache;
  pTaskDb->filterFactory = rocksdb_compactionfilterfactory_create(
      NULL, destroyCompactFilteFactory, compactFilteFactoryCreateFilter, compactFilteFactoryName);
  rocksdb_options_set_compaction_filter_factory(pTaskDb->dbOpt, pTaskDb->filterFactory);
  pTaskDb->readOpt = rocksdb_readoptions_create();
  pTaskDb->writeOpt = rocksdb_writeoptions_create();
  rocksdb_writeoptions_disable_WAL(pTaskDb->writeOpt, 1);

  size_t nCf = sizeof(ginitDict) / sizeof(ginitDict[0]);
  pTaskDb->pCf = taosMemoryCalloc(nCf, sizeof(rocksdb_column_family_handle_t*));
  pTaskDb->pCfParams = taosMemoryCalloc(nCf, sizeof(RocksdbCfParam));
  pTaskDb->pCfOpts = taosMemoryCalloc(nCf, sizeof(rocksdb_options_t*));
  pTaskDb->pCompares = taosMemoryCalloc(nCf, sizeof(rocksdb_comparator_t*));
  if (pTaskDb->pCf == NULL || pTaskDb->pCfParams == NULL || pTaskDb->pCfOpts == NULL || pTaskDb->pCompares == NULL) {
    stError("failed to alloc memory for cf");
    taosMemoryFreeClear(pTaskDb->pCf);
    taosMemoryFreeClear(pTaskDb->pCfParams);
    taosMemoryFreeClear(pTaskDb->pCfOpts);
    taosMemoryFreeClear(pTaskDb->pCompares);
    return;
  }

  for (int i = 0; i < nCf; i++) {
    rocksdb_options_t*                   opt = rocksdb_options_create_copy(pTaskDb->dbOpt);
    rocksdb_block_based_table_options_t* tableOpt = rocksdb_block_based_options_create();
    rocksdb_block_based_options_set_block_cache(tableOpt, pTaskDb->cache);
    rocksdb_block_based_options_set_partition_filters(tableOpt, 1);

    rocksdb_filterpolicy_t* filter = rocksdb_filterpolicy_create_bloom(15);
    rocksdb_block_based_options_set_filter_policy(tableOpt, filter);

    rocksdb_options_set_block_based_table_factory((rocksdb_options_t*)opt, tableOpt);

    SCfInit* cfPara = &ginitDict[i];

    rocksdb_comparator_t* compare =
        rocksdb_comparator_create(NULL, cfPara->destroyCmp, cfPara->cmpKey, cfPara->cmpName);
    rocksdb_options_set_comparator((rocksdb_options_t*)opt, compare);

    rocksdb_compactionfilterfactory_t* filterFactory =
        rocksdb_compactionfilterfactory_create(NULL, cfPara->destroyFilter, cfPara->createFilter, cfPara->funcName);
    rocksdb_options_set_compaction_filter_factory(opt, filterFactory);

    pTaskDb->pCompares[i] = compare;
    pTaskDb->pCfOpts[i] = opt;
    pTaskDb->pCfParams[i].tableOpt = tableOpt;
  }
  return;
}
void taskDbInitChkpOpt(STaskDbWrapper* pTaskDb) {
  pTaskDb->chkpId = -1;
  pTaskDb->chkpCap = 4;
  pTaskDb->chkpSaved = taosArrayInit(4, sizeof(int64_t));
  TAOS_UNUSED(taskDbLoadChkpInfo(pTaskDb));

  pTaskDb->chkpInUse = taosArrayInit(4, sizeof(int64_t));

  TAOS_UNUSED(taosThreadRwlockInit(&pTaskDb->chkpDirLock, NULL));
}

void taskDbRefChkp(STaskDbWrapper* pTaskDb, int64_t chkp) {
  TAOS_UNUSED(taosThreadRwlockWrlock(&pTaskDb->chkpDirLock));
  if (taosArrayPush(pTaskDb->chkpInUse, &chkp) == NULL) {
    stError("failed to push chkp: %" PRIi64 " into inuse", chkp);
  }
  taosArraySort(pTaskDb->chkpInUse, chkpIdComp);
  TAOS_UNUSED(taosThreadRwlockUnlock(&pTaskDb->chkpDirLock));
}

void taskDbUnRefChkp(STaskDbWrapper* pTaskDb, int64_t chkp) {
  TAOS_UNUSED(taosThreadRwlockWrlock(&pTaskDb->chkpDirLock));
  int32_t size = taosArrayGetSize(pTaskDb->chkpInUse);
  for (int i = 0; i < size; i++) {
    int64_t* p = taosArrayGet(pTaskDb->chkpInUse, i);
    if (*p == chkp) {
      taosArrayRemove(pTaskDb->chkpInUse, i);
      break;
    }
  }
  TAOS_UNUSED(taosThreadRwlockUnlock(&pTaskDb->chkpDirLock));
}

void taskDbDestroyChkpOpt(STaskDbWrapper* pTaskDb) {
  taosArrayDestroy(pTaskDb->chkpSaved);
  taosArrayDestroy(pTaskDb->chkpInUse);
  TAOS_UNUSED(taosThreadRwlockDestroy(&pTaskDb->chkpDirLock));
}

int32_t taskDbBuildFullPath(char* path, char* key, char** dbFullPath, char** stateFullPath) {
  int32_t code = 0;
  int32_t cap = strlen(path) + 128, nBytes = 0;
  char*   statePath = NULL;
  char*   dbPath = NULL;

  statePath = taosMemoryCalloc(1, cap);
  if (statePath == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _err);
  }

  nBytes = snprintf(statePath, cap, "%s%s%s", path, TD_DIRSEP, key);
  if (nBytes < 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    TAOS_CHECK_GOTO(code, NULL, _err);
  }

  if (!taosDirExist(statePath)) {
    code = taosMulMkDir(statePath);
    TAOS_CHECK_GOTO(code, NULL, _err);
  }

  dbPath = taosMemoryCalloc(1, cap);
  if (dbPath == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _err);
  }
  nBytes = snprintf(dbPath, cap, "%s%s%s", statePath, TD_DIRSEP, "state");
  if (nBytes < 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    TAOS_CHECK_GOTO(code, NULL, _err);
  }

  if (!taosDirExist(dbPath)) {
    code = taosMulMkDir(dbPath);
    TAOS_CHECK_GOTO(code, NULL, _err);
  }

  *dbFullPath = dbPath;
  *stateFullPath = statePath;
  return 0;
_err:
  stError("failed to create dir: %s, reason:%s", dbPath, tstrerror(code));

  taosMemoryFree(statePath);
  taosMemoryFree(dbPath);
  return code;
}

void taskDbUpdateChkpId(void* pTaskDb, int64_t chkpId) {
  STaskDbWrapper* p = pTaskDb;
  TAOS_UNUSED(streamMutexLock(&p->mutex));
  p->chkpId = chkpId;
  TAOS_UNUSED(streamMutexUnlock(&p->mutex));
}

STaskDbWrapper* taskDbOpenImpl(const char* key, char* statePath, char* dbPath) {
  char*   err = NULL;
  char**  cfNames = NULL;
  size_t  nCf = 0;
  int32_t code = 0;
  int32_t lino = 0;

  STaskDbWrapper* pTaskDb = taosMemoryCalloc(1, sizeof(STaskDbWrapper));
  TSDB_CHECK_NULL(pTaskDb, code, lino, _EXIT, terrno);

  pTaskDb->idstr = key ? taosStrdup(key) : NULL;
  pTaskDb->path = statePath ? taosStrdup(statePath) : NULL;

  code = taosThreadMutexInit(&pTaskDb->mutex, NULL);
  TSDB_CHECK_CODE(code, lino, _EXIT);

  taskDbInitChkpOpt(pTaskDb);
  taskDbInitOpt(pTaskDb);

  cfNames = rocksdb_list_column_families(pTaskDb->dbOpt, dbPath, &nCf, &err);
  if (nCf == 0) {
    stInfo("%s newly create db in state-backend", key);
    // pre create db
    pTaskDb->db = rocksdb_open(pTaskDb->pCfOpts[0], dbPath, &err);
    if (pTaskDb->db == NULL) {
      stError("%s open state-backend failed, reason:%s", key, err);
      code = TSDB_CODE_STREAM_INTERNAL_ERROR;
      goto _EXIT;
    }

    rocksdb_close(pTaskDb->db);
    pTaskDb->db = NULL;

    if (cfNames != NULL) {
      rocksdb_list_column_families_destroy(cfNames, nCf);
    }

    taosMemoryFree(err);
    err = NULL;

    cfNames = rocksdb_list_column_families(pTaskDb->dbOpt, dbPath, &nCf, &err);
    if (err != NULL) {
      stError("%s failed to create column-family, %s, %" PRIzu ", reason:%s", key, dbPath, nCf, err);
      code = TSDB_CODE_STREAM_INTERNAL_ERROR;
      goto _EXIT;
    }
  }

  if ((code = taskDbOpenCfs(pTaskDb, dbPath, cfNames, nCf)) != 0) {
    goto _EXIT;
  }

  if (cfNames != NULL) {
    rocksdb_list_column_families_destroy(cfNames, nCf);
    cfNames = NULL;
  }

  stDebug("init s-task backend in:%s, backend:%p, %s", dbPath, pTaskDb, key);
  return pTaskDb;

_EXIT:
  stError("%s taskDb open failed, %s at line:%d code:%s", key, __func__, lino, tstrerror(code));

  taskDbDestroy(pTaskDb, false);
  if (err) taosMemoryFree(err);
  if (cfNames) rocksdb_list_column_families_destroy(cfNames, nCf);
  return NULL;
}

int32_t taskDbOpen(const char* path, const char* key, int64_t chkptId, int64_t* processVer, STaskDbWrapper** ppTaskDb) {
  char* statePath = NULL;
  char* dbPath = NULL;
  int   code = 0;
  if ((code = restoreCheckpointData(path, key, chkptId, &statePath, &dbPath, processVer)) < 0) {
    stError("failed to restore checkpoint data, path:%s, key:%s, checkpointId: %" PRId64 "reason:%s", path, key,
            chkptId, tstrerror(code));
    return code;
  }

  STaskDbWrapper* pTaskDb = taskDbOpenImpl(key, statePath, dbPath);
  if (pTaskDb != NULL) {
    int64_t chkpId = -1, ver = -1;
    if ((code = chkpLoadExtraInfo(dbPath, &chkpId, &ver)) == 0) {
      *processVer = ver;
    } else {
      stError("failed to load extra info, path:%s, key:%s, checkpointId: %" PRId64 "reason:%s", path, key, chkptId,
              tstrerror(code));
      taskDbDestroy(pTaskDb, false);
      return code;
    }
  } else {
    code = TSDB_CODE_INVALID_PARA;
  }

  taosMemoryFree(dbPath);
  taosMemoryFree(statePath);
  *ppTaskDb = pTaskDb;
  return code;
}

void taskDbDestroy(void* pDb, bool flush) {
  STaskDbWrapper* wrapper = pDb;
  if (wrapper == NULL) {
    return;
  }

  int64_t st = taosGetTimestampMs();
  streamMetaRemoveDB(wrapper->pMeta, wrapper->idstr);

  stDebug("%s succ to destroy stream backend:%p", wrapper->idstr, wrapper);

  int8_t nCf = tListLen(ginitDict);
  if (flush && wrapper->removeAllFiles == 0) {
    if (wrapper->db && wrapper->pCf) {
      rocksdb_flushoptions_t* flushOpt = rocksdb_flushoptions_create();
      rocksdb_flushoptions_set_wait(flushOpt, 1);

      char*                            err = NULL;
      rocksdb_column_family_handle_t** cfs = taosMemoryCalloc(1, sizeof(rocksdb_column_family_handle_t*) * nCf);
      int                              numOfFlushCf = 0;
      for (int i = 0; i < nCf; i++) {
        if (wrapper->pCf[i] != NULL) {
          cfs[numOfFlushCf++] = wrapper->pCf[i];
        }
      }
      if (numOfFlushCf != 0) {
        rocksdb_flush_cfs(wrapper->db, flushOpt, cfs, numOfFlushCf, &err);
        if (err != NULL) {
          stError("failed to flush all cfs, reason:%s", err);
          taosMemoryFreeClear(err);
        }
      }
      taosMemoryFree(cfs);
      rocksdb_flushoptions_destroy(flushOpt);
    }
  }

  if (wrapper->pCf != NULL) {
    for (int i = 0; i < nCf; i++) {
      if (wrapper->pCf[i] != NULL) {
        rocksdb_column_family_handle_destroy(wrapper->pCf[i]);
      }
    }
  }

  if (wrapper->db) {
    rocksdb_close(wrapper->db);
    wrapper->db = NULL;
  }

  rocksdb_options_destroy(wrapper->dbOpt);
  rocksdb_readoptions_destroy(wrapper->readOpt);
  rocksdb_writeoptions_destroy(wrapper->writeOpt);
  rocksdb_env_destroy(wrapper->env);
  rocksdb_cache_destroy(wrapper->cache);

  taosMemoryFree(wrapper->pCf);
  for (int i = 0; i < nCf; i++) {
    rocksdb_options_t*                   opt = wrapper->pCfOpts[i];
    rocksdb_comparator_t*                compare = wrapper->pCompares[i];
    rocksdb_block_based_table_options_t* tblOpt = wrapper->pCfParams[i].tableOpt;

    rocksdb_options_destroy(opt);
    rocksdb_comparator_destroy(compare);
    rocksdb_block_based_options_destroy(tblOpt);
  }

  taosMemoryFree(wrapper->pCompares);
  taosMemoryFree(wrapper->pCfOpts);
  taosMemoryFree(wrapper->pCfParams);

  streamMutexDestroy(&wrapper->mutex);
  taskDbDestroyChkpOpt(wrapper);

  if (wrapper->removeAllFiles) {
    char* err = NULL;
    stInfo("drop task remove backend data:%s", wrapper->path);
    taosRemoveDir(wrapper->path);
  }

  int64_t et = taosGetTimestampMs();
  stDebug("%s destroy stream backend:%p completed, elapsed time:%.2fs", wrapper->idstr, wrapper, (et - st)/1000.0);

  taosMemoryFree(wrapper->idstr);
  taosMemoryFree(wrapper->path);
  taosMemoryFree(wrapper);
}

void taskDbDestroy2(void* pDb) { taskDbDestroy(pDb, true); }

int32_t taskDbGenChkpUploadData__rsync(STaskDbWrapper* pDb, int64_t chkpId, char** path) {
  int32_t code = 0;
  int64_t refId = pDb->refId;
  int32_t nBytes = 0;

  if (taosAcquireRef(taskDbWrapperId, refId) == NULL) {
    code = terrno;
    return code;
  }

  int32_t cap = strlen(pDb->path) + 128;

  char* buf = taosMemoryCalloc(1, cap);
  if (buf == NULL) {
    TAOS_UNUSED(taosReleaseRef(taskDbWrapperId, refId));
    return terrno;
  }

  nBytes =
      snprintf(buf, cap, "%s%s%s%s%s%" PRId64, pDb->path, TD_DIRSEP, "checkpoints", TD_DIRSEP, "checkpoint", chkpId);
  if (nBytes <= 0 || nBytes >= cap) {
    taosMemoryFree(buf);
    TAOS_UNUSED(taosReleaseRef(taskDbWrapperId, refId));
    return TSDB_CODE_OUT_OF_RANGE;
  }

  if (taosIsDir(buf)) {
    code = 0;
    *path = buf;
  } else {
    taosMemoryFree(buf);
  }

  TAOS_UNUSED(taosReleaseRef(taskDbWrapperId, refId));
  return code;
}

int32_t taskDbGenChkpUploadData__s3(STaskDbWrapper* pDb, void* bkdChkpMgt, int64_t chkpId, char** path, SArray* list,
                                    const char* idStr) {
  int32_t  code = 0;
  int32_t  cap = strlen(pDb->path) + 32;
  SBkdMgt* p = (SBkdMgt*)bkdChkpMgt;

  char* temp = taosMemoryCalloc(1, cap);
  if (temp == NULL) {
    return terrno;
  }

  int32_t nBytes = snprintf(temp, cap, "%s%s%s%" PRId64, pDb->path, TD_DIRSEP, "tmp", chkpId);
  if (nBytes <= 0 || nBytes >= cap) {
    taosMemoryFree(temp);
    return TSDB_CODE_OUT_OF_RANGE;
  }

  if (taosDirExist(temp)) {
    cleanDir(temp, idStr);
  } else {
    code = taosMkDir(temp);
    if (code != 0) {
      taosMemoryFree(temp);
      return TAOS_SYSTEM_ERROR(ERRNO);
    }
  }

  code = bkdMgtGetDelta(p, pDb->idstr, chkpId, list, temp);
  *path = temp;

  return code;
}

int32_t taskDbGenChkpUploadData(void* arg, void* mgt, int64_t chkpId, int8_t type, char** path, SArray* list,
                                const char* idStr) {
  int32_t                 code = -1;
  STaskDbWrapper*         pDb = arg;
  ECHECKPOINT_BACKUP_TYPE utype = type;

  taskDbRefChkp(pDb, chkpId);
  if (utype == DATA_UPLOAD_RSYNC) {
    code = taskDbGenChkpUploadData__rsync(pDb, chkpId, path);
  } else if (utype == DATA_UPLOAD_S3) {
    code = taskDbGenChkpUploadData__s3(pDb, mgt, chkpId, path, list, idStr);
  }
  taskDbUnRefChkp(pDb, chkpId);
  return code;
}

int32_t taskDbOpenCfByKey(STaskDbWrapper* pDb, const char* key) {
  int32_t code = 0;
  char*   err = NULL;
  int8_t  idx = getCfIdx(key);

  if (idx == -1) return -1;

  if (pDb->pCf[idx] != NULL) return code;

  rocksdb_column_family_handle_t* cf =
      rocksdb_create_column_family(pDb->db, pDb->pCfOpts[idx], ginitDict[idx].key, &err);
  if (err != NULL) {
    stError("failed to open cf, key:%s, reason: %s", key, err);
    taosMemoryFree(err);
    code = -1;
    return code;
  }

  pDb->pCf[idx] = cf;
  return code;
}
int32_t copyDataAt(RocksdbCfInst* pSrc, STaskDbWrapper* pDst, int8_t i) {
  int32_t WRITE_BATCH = 1024;
  char*   err = NULL;
  int     code = 0;

  rocksdb_readoptions_t* pRdOpt = rocksdb_readoptions_create();

  rocksdb_writebatch_t* wb = rocksdb_writebatch_create();
  rocksdb_iterator_t*   pIter = rocksdb_create_iterator_cf(pSrc->db, pRdOpt, pSrc->pHandle[i]);
  rocksdb_iter_seek_to_first(pIter);
  while (rocksdb_iter_valid(pIter)) {
    if (rocksdb_writebatch_count(wb) >= WRITE_BATCH) {
      rocksdb_write(pDst->db, pDst->writeOpt, wb, &err);
      if (err != NULL) {
        code = -1;
        goto _EXIT;
      }
      rocksdb_writebatch_clear(wb);
    }

    size_t klen = 0, vlen = 0;
    char*  key = (char*)rocksdb_iter_key(pIter, &klen);
    char*  val = (char*)rocksdb_iter_value(pIter, &vlen);

    rocksdb_writebatch_put_cf(wb, pDst->pCf[i], key, klen, val, vlen);
    rocksdb_iter_next(pIter);
  }

  if (rocksdb_writebatch_count(wb) > 0) {
    rocksdb_write(pDst->db, pDst->writeOpt, wb, &err);
    if (err != NULL) {
      code = -1;
      goto _EXIT;
    }
  }

_EXIT:
  rocksdb_writebatch_destroy(wb);
  rocksdb_iter_destroy(pIter);
  rocksdb_readoptions_destroy(pRdOpt);
  taosMemoryFree(err);

  return code;
}

int32_t streamStateCvtDataFormat(char* path, char* key, void* pCfInst) {
  int nCf = sizeof(ginitDict) / sizeof(ginitDict[0]);

  int32_t code = 0;

  int64_t         processVer = -1;
  STaskDbWrapper* pTaskDb = NULL;

  code = taskDbOpen(path, key, 0, &processVer, &pTaskDb);
  RocksdbCfInst* pSrcBackend = pCfInst;

  for (int i = 0; i < nCf; i++) {
    rocksdb_column_family_handle_t* pSrcCf = pSrcBackend->pHandle[i];
    if (pSrcCf == NULL) continue;

    code = taskDbOpenCfByKey(pTaskDb, ginitDict[i].key);
    if (code != 0) goto _EXIT;

    code = copyDataAt(pSrcBackend, pTaskDb, i);
    if (code != 0) goto _EXIT;
  }

_EXIT:
  taskDbDestroy(pTaskDb, true);

  return code;
}
int32_t streamStateOpenBackendCf(void* backend, char* name, char** cfs, int32_t nCf) {
  SBackendWrapper* handle = backend;
  char*            err = NULL;
  int64_t          streamId;
  int32_t          taskId, dummy = 0;
  char             suffix[64] = {0};
  int32_t          code = 0;

  rocksdb_options_t**              cfOpts = taosMemoryCalloc(nCf, sizeof(rocksdb_options_t*));
  RocksdbCfParam*                  params = taosMemoryCalloc(nCf, sizeof(RocksdbCfParam));
  rocksdb_comparator_t**           pCompare = taosMemoryCalloc(nCf, sizeof(rocksdb_comparator_t*));
  rocksdb_column_family_handle_t** cfHandle = taosMemoryCalloc(nCf, sizeof(rocksdb_column_family_handle_t*));

  for (int i = 0; i < nCf; i++) {
    char* cf = cfs[i];
    char  funcname[64] = {0};

    cfOpts[i] = rocksdb_options_create_copy(handle->dbOpt);
    if (i == 0) continue;
    if (3 == sscanf(cf, "0x%" PRIx64 "-%d_%s", &streamId, &taskId, funcname)) {
      rocksdb_block_based_table_options_t* tableOpt = rocksdb_block_based_options_create();
      rocksdb_block_based_options_set_block_cache(tableOpt, handle->cache);
      rocksdb_block_based_options_set_partition_filters(tableOpt, 1);

      rocksdb_filterpolicy_t* filter = rocksdb_filterpolicy_create_bloom(15);
      rocksdb_block_based_options_set_filter_policy(tableOpt, filter);

      rocksdb_options_set_block_based_table_factory((rocksdb_options_t*)cfOpts[i], tableOpt);
      params[i].tableOpt = tableOpt;

      int idx = streamStateGetCfIdx(NULL, funcname);
      if (idx < 0 || idx >= sizeof(ginitDict) / sizeof(ginitDict[0])) {
        stError("failed to open cf");
        return -1;
      }
      SCfInit* cfPara = &ginitDict[idx];

      rocksdb_comparator_t* compare =
          rocksdb_comparator_create(NULL, cfPara->destroyCmp, cfPara->cmpKey, cfPara->cmpName);
      rocksdb_options_set_comparator((rocksdb_options_t*)cfOpts[i], compare);
      pCompare[i] = compare;
    }
  }
  rocksdb_t* db = rocksdb_open_column_families(handle->dbOpt, name, nCf, (const char* const*)cfs,
                                               (const rocksdb_options_t* const*)cfOpts, cfHandle, &err);
  if (err != NULL) {
    stError("failed to open rocksdb cf, reason:%s", err);
    taosMemoryFree(err);
    taosMemoryFree(cfHandle);
    taosMemoryFree(pCompare);
    taosMemoryFree(params);
    taosMemoryFree(cfOpts);
    // fix other leak
    return TSDB_CODE_THIRDPARTY_ERROR;
  } else {
    stDebug("succ to open rocksdb cf");
  }
  // close default cf
  if (((rocksdb_column_family_handle_t**)cfHandle)[0] != 0) {
    rocksdb_column_family_handle_destroy(cfHandle[0]);
    cfHandle[0] = NULL;
  }
  rocksdb_options_destroy(cfOpts[0]);

  handle->db = db;

  static int32_t cfLen = sizeof(ginitDict) / sizeof(ginitDict[0]);
  for (int i = 0; i < nCf; i++) {
    char* cf = cfs[i];
    if (i == 0) continue;  // skip default column family, not set opt

    char funcname[64] = {0};
    if (3 == sscanf(cf, "0x%" PRIx64 "-%d_%s", &streamId, &taskId, funcname)) {
      char    idstr[128] = {0};
      int32_t nBytes = snprintf(idstr, sizeof(idstr), "0x%" PRIx64 "-%d", streamId, taskId);
      if (nBytes <= 0 || nBytes >= sizeof(idstr)) {
        code = TSDB_CODE_OUT_OF_RANGE;
        stError("failed to open cf since %s", tstrerror(code));
        return code;
      }

      int idx = streamStateGetCfIdx(NULL, funcname);

      RocksdbCfInst*  inst = NULL;
      RocksdbCfInst** pInst = taosHashGet(handle->cfInst, idstr, strlen(idstr) + 1);
      if (pInst == NULL || *pInst == NULL) {
        inst = taosMemoryCalloc(1, sizeof(RocksdbCfInst));
        inst->pHandle = taosMemoryCalloc(cfLen, sizeof(rocksdb_column_family_handle_t*));
        inst->cfOpt = taosMemoryCalloc(cfLen, sizeof(rocksdb_options_t*));
        inst->wOpt = rocksdb_writeoptions_create();
        inst->rOpt = rocksdb_readoptions_create();
        inst->param = taosMemoryCalloc(cfLen, sizeof(RocksdbCfParam));
        inst->pBackend = handle;
        inst->db = db;
        inst->pCompares = taosMemoryCalloc(cfLen, sizeof(rocksdb_comparator_t*));

        inst->dbOpt = handle->dbOpt;
        rocksdb_writeoptions_disable_WAL(inst->wOpt, 1);
        TAOS_UNUSED(taosHashPut(handle->cfInst, idstr, strlen(idstr) + 1, &inst, sizeof(void*)));
      } else {
        inst = *pInst;
      }
      inst->cfOpt[idx] = cfOpts[i];
      inst->pCompares[idx] = pCompare[i];
      memcpy(&(inst->param[idx]), &(params[i]), sizeof(RocksdbCfParam));
      inst->pHandle[idx] = cfHandle[i];
    }
  }
  void* pIter = taosHashIterate(handle->cfInst, NULL);
  while (pIter) {
    RocksdbCfInst* inst = *(RocksdbCfInst**)pIter;

    for (int i = 0; i < cfLen; i++) {
      if (inst->cfOpt[i] == NULL) {
        rocksdb_options_t*                   opt = rocksdb_options_create_copy(handle->dbOpt);
        rocksdb_block_based_table_options_t* tableOpt = rocksdb_block_based_options_create();
        rocksdb_block_based_options_set_block_cache(tableOpt, handle->cache);
        rocksdb_block_based_options_set_partition_filters(tableOpt, 1);

        rocksdb_filterpolicy_t* filter = rocksdb_filterpolicy_create_bloom(15);
        rocksdb_block_based_options_set_filter_policy(tableOpt, filter);

        rocksdb_options_set_block_based_table_factory((rocksdb_options_t*)opt, tableOpt);

        SCfInit* cfPara = &ginitDict[i];

        rocksdb_comparator_t* compare =
            rocksdb_comparator_create(NULL, cfPara->destroyCmp, cfPara->cmpKey, cfPara->cmpName);
        rocksdb_options_set_comparator((rocksdb_options_t*)opt, compare);

        inst->pCompares[i] = compare;
        inst->cfOpt[i] = opt;
        inst->param[i].tableOpt = tableOpt;
      }
    }
    SCfComparator compare = {.comp = inst->pCompares, .numOfComp = cfLen};
    inst->pCompareNode = streamBackendAddCompare(handle, &compare);
    pIter = taosHashIterate(handle->cfInst, pIter);
  }

  taosMemoryFree(cfHandle);
  taosMemoryFree(pCompare);
  taosMemoryFree(params);
  taosMemoryFree(cfOpts);
  return 0;
}

void streamStateDestroyCompar(void* arg) {
  SCfComparator* comp = (SCfComparator*)arg;
  for (int i = 0; i < comp->numOfComp; i++) {
    if (comp->comp[i]) rocksdb_comparator_destroy(comp->comp[i]);
  }
  taosMemoryFree(comp->comp);
}

int streamStateGetCfIdx(SStreamState* pState, const char* funcName) {
  int    idx = -1;
  size_t len = strlen(funcName);
  for (int i = 0; i < sizeof(ginitDict) / sizeof(ginitDict[0]); i++) {
    if (len == ginitDict[i].len && strncmp(funcName, ginitDict[i].key, strlen(funcName)) == 0) {
      idx = i;
      break;
    }
  }
  if (pState != NULL && idx != -1) {
    STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
    if (pState->pTdbState->recalc) {
      wrapper = pState->pTdbState->pOwner->pRecalBackend;
    }
    if (wrapper == NULL) {
      return -1;
    }

    streamMutexLock(&wrapper->mutex);

    rocksdb_column_family_handle_t* cf = wrapper->pCf[idx];
    if (cf == NULL) {
      char* err = NULL;
      cf = rocksdb_create_column_family(wrapper->db, wrapper->pCfOpts[idx], ginitDict[idx].key, &err);
      if (err != NULL) {
        idx = -1;
        stError("failed to open cf, %p %s_%s, reason:%s", pState, wrapper->idstr, funcName, err);
        rocksdb_column_family_handle_destroy(cf);
        taosMemoryFree(err);
      } else {
        stDebug("succ to open cf, %p %s_%s", pState, wrapper->idstr, funcName);
        wrapper->pCf[idx] = cf;
      }
    }
    streamMutexUnlock(&wrapper->mutex);
  }

  return idx;
}
bool streamStateIterSeekAndValid(rocksdb_iterator_t* iter, char* buf, size_t len) {
  rocksdb_iter_seek(iter, buf, len);
  if (!rocksdb_iter_valid(iter)) {
    rocksdb_iter_seek_for_prev(iter, buf, len);
    if (!rocksdb_iter_valid(iter)) {
      return false;
    }
  }
  return true;
}
rocksdb_iterator_t* streamStateIterCreate(SStreamState* pState, const char* cfKeyName, rocksdb_snapshot_t** snapshot,
                                          rocksdb_readoptions_t** readOpt) {
  int idx = streamStateGetCfIdx(pState, cfKeyName);

  *readOpt = rocksdb_readoptions_create();

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  if (snapshot != NULL) {
    *snapshot = (rocksdb_snapshot_t*)rocksdb_create_snapshot(wrapper->db);
    rocksdb_readoptions_set_snapshot(*readOpt, *snapshot);
    rocksdb_readoptions_set_fill_cache(*readOpt, 0);
  }

  return rocksdb_create_iterator_cf(wrapper->db, *readOpt, ((rocksdb_column_family_handle_t**)wrapper->pCf)[idx]);
}

#define STREAM_STATE_PUT_ROCKSDB(pState, funcname, key, value, vLen)                                                 \
  do {                                                                                                               \
    code = 0;                                                                                                        \
    char  buf[128] = {0};                                                                                            \
    char* err = NULL;                                                                                                \
    int   i = streamStateGetCfIdx(pState, funcname);                                                                 \
    if (i < 0) {                                                                                                     \
      stWarn("streamState failed to get cf name: %s", funcname);                                                     \
      code = TSDB_CODE_THIRDPARTY_ERROR;                                                                             \
      break;                                                                                                         \
    }                                                                                                                \
    STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;                                                   \
    TAOS_UNUSED(atomic_add_fetch_64(&wrapper->dataWritten, 1));                                                      \
    char toString[128] = {0};                                                                                        \
    TAOS_UNUSED((ginitDict[i].toStrFunc((void*)key, toString)));                      \
    int32_t                         klen = ginitDict[i].enFunc((void*)key, buf);                                     \
    rocksdb_column_family_handle_t* pHandle = ((rocksdb_column_family_handle_t**)wrapper->pCf)[ginitDict[i].idx];    \
    rocksdb_writeoptions_t*         opts = wrapper->writeOpt;                                                        \
    rocksdb_t*                      db = wrapper->db;                                                                \
    char*                           ttlV = NULL;                                                                     \
    int32_t                         ttlVLen = ginitDict[i].enValueFunc((char*)value, vLen, 0, &ttlV);                \
    rocksdb_put_cf(db, opts, pHandle, (const char*)buf, klen, (const char*)ttlV, (size_t)ttlVLen, &err);             \
    if (err != NULL) {                                                                                               \
      stError("streamState str: %s failed to write to %s, err: %s", toString, funcname, err);                        \
      taosMemoryFree(err);                                                                                           \
      code = TSDB_CODE_THIRDPARTY_ERROR;                                                                             \
    } else {                                                                                                         \
      stDebug("[InternalERR] write streamState str:%s succ to write to %s, rowValLen:%d, ttlValLen:%d, %p", toString, \
             funcname, vLen, ttlVLen, wrapper);                                                                      \
    }                                                                                                                \
    taosMemoryFree(ttlV);                                                                                            \
  } while (0);

#define STREAM_STATE_GET_ROCKSDB(pState, funcname, key, pVal, vLen)                                                   \
  do {                                                                                                                \
    code = 0;                                                                                                         \
    char  buf[128] = {0};                                                                                             \
    char* err = NULL;                                                                                                 \
    int   i = streamStateGetCfIdx(pState, funcname);                                                                  \
    if (i < 0) {                                                                                                      \
      stWarn("streamState failed to get cf name: %s", funcname);                                                      \
      code = -1;                                                                                                      \
      break;                                                                                                          \
    }                                                                                                                 \
    STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;                                                    \
    if (pState->pTdbState->recalc) {                                                                                  \
      wrapper = pState->pTdbState->pOwner->pRecalBackend;                                                             \
    }                                                                                                                 \
    char toString[128] = {0};                                                                                         \
    if (stDebugFlag & DEBUG_TRACE) TAOS_UNUSED((ginitDict[i].toStrFunc((void*)key, toString)));                       \
    int32_t                         klen = ginitDict[i].enFunc((void*)key, buf);                                      \
    rocksdb_column_family_handle_t* pHandle = ((rocksdb_column_family_handle_t**)wrapper->pCf)[ginitDict[i].idx];     \
    rocksdb_t*                      db = wrapper->db;                                                                 \
    rocksdb_readoptions_t*          opts = wrapper->readOpt;                                                          \
    size_t                          len = 0;                                                                          \
    char* val = rocksdb_get_cf(db, opts, pHandle, (const char*)buf, klen, (size_t*)&len, &err);                       \
    if (val == NULL || len == 0) {                                                                                    \
      if (err == NULL) {                                                                                              \
        stTrace("streamState str: %s failed to read from %s_%s, err: not exist", toString, wrapper->idstr, funcname); \
      } else {                                                                                                        \
        stError("streamState str: %s failed to read from %s_%s, err: %s", toString, wrapper->idstr, funcname, err);   \
        taosMemoryFreeClear(err);                                                                                     \
      }                                                                                                               \
      code = -1;                                                                                                      \
    } else {                                                                                                          \
      char*   p = NULL;                                                                                               \
      int32_t tlen = ginitDict[i].deValueFunc(val, len, NULL, (char**)pVal);                                          \
      if (tlen <= 0) {                                                                                                \
        stError("streamState str: %s failed to read from %s_%s, err: already ttl ", toString, wrapper->idstr,         \
                funcname);                                                                                            \
        code = -1;                                                                                                    \
      } else {                                                                                                        \
        stTrace("streamState str: %s succ to read from %s_%s, valLen:%d, %p", toString, wrapper->idstr, funcname,     \
                tlen, wrapper);                                                                                       \
      }                                                                                                               \
      taosMemoryFree(val);                                                                                            \
      if (vLen != NULL) *vLen = tlen;                                                                                 \
    }                                                                                                                 \
  } while (0);

#define STREAM_STATE_DEL_ROCKSDB(pState, funcname, key)                                                           \
  do {                                                                                                            \
    code = 0;                                                                                                     \
    char  buf[128] = {0};                                                                                         \
    char* err = NULL;                                                                                             \
    int   i = streamStateGetCfIdx(pState, funcname);                                                              \
    if (i < 0) {                                                                                                  \
      stWarn("streamState failed to get cf name: %s_%s", pState->pTdbState->idstr, funcname);                     \
      code = TSDB_CODE_THIRDPARTY_ERROR;                                                                          \
      break;                                                                                                      \
    }                                                                                                             \
    STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;                                                \
    if (pState->pTdbState->recalc) {                                                                              \
      wrapper = pState->pTdbState->pOwner->pRecalBackend;                                                         \
    }                                                                                                             \
    TAOS_UNUSED(atomic_add_fetch_64(&wrapper->dataWritten, 1));                                                   \
    char toString[128] = {0};                                                                                     \
    if (stDebugFlag & DEBUG_TRACE) TAOS_UNUSED(ginitDict[i].toStrFunc((void*)key, toString));                     \
    int32_t                         klen = ginitDict[i].enFunc((void*)key, buf);                                  \
    rocksdb_column_family_handle_t* pHandle = ((rocksdb_column_family_handle_t**)wrapper->pCf)[ginitDict[i].idx]; \
    rocksdb_t*                      db = wrapper->db;                                                             \
    rocksdb_writeoptions_t*         opts = wrapper->writeOpt;                                                     \
    rocksdb_delete_cf(db, opts, pHandle, (const char*)buf, klen, &err);                                           \
    if (err != NULL) {                                                                                            \
      stError("streamState str: %s failed to del from %s_%s, err: %s", toString, wrapper->idstr, funcname, err);  \
      taosMemoryFree(err);                                                                                        \
      code = TSDB_CODE_THIRDPARTY_ERROR;                                                                          \
    } else {                                                                                                      \
      stTrace("streamState str: %s succ to del from %s_%s", toString, wrapper->idstr, funcname);                  \
    }                                                                                                             \
  } while (0);

// state cf
int32_t streamStatePut_rocksdb(SStreamState* pState, const SWinKey* key, const void* value, int32_t vLen) {
  int code = 0;

  SStateKey sKey = {.key = *key, .opNum = pState->number};
  char*     dst = NULL;
  size_t    size = 0;
  if (pState->pResultRowStore.resultRowPut == NULL || pState->pExprSupp == NULL) {
    STREAM_STATE_PUT_ROCKSDB(pState, "state", &sKey, (void*)value, (int32_t)vLen);
  } else {
    code = (pState->pResultRowStore.resultRowPut)(pState->pExprSupp, value, vLen, &dst, &size);
    if (code != 0) {
      return code;
    }
    STREAM_STATE_PUT_ROCKSDB(pState, "state", &sKey, (void*)dst, (int32_t)size);
    taosMemoryFree(dst);
  }
  return code;
}
int32_t streamStateGet_rocksdb(SStreamState* pState, const SWinKey* key, void** pVal, int32_t* pVLen) {
  int       code = 0;
  SStateKey sKey = {.key = *key, .opNum = pState->number};

  char*  tVal = NULL;
  size_t tValLen = 0;
  STREAM_STATE_GET_ROCKSDB(pState, "state", &sKey, &tVal, &tValLen);
  if (code != 0) {
    taosMemoryFree(tVal);
    return code;
  }
  if (pState->pResultRowStore.resultRowGet == NULL || pState->pExprSupp == NULL) {
    *pVal = tVal;
    *pVLen = tValLen;
    return code;
  }
  size_t pValLen = 0;
  code = (pState->pResultRowStore.resultRowGet)(pState->pExprSupp, tVal, tValLen, (char**)pVal, &pValLen);
  *pVLen = (int32_t)pValLen;
  taosMemoryFree(tVal);
  return code;
}
int32_t streamStateDel_rocksdb(SStreamState* pState, const SWinKey* key) {
  int       code = 0;
  SStateKey sKey = {.key = *key, .opNum = pState->number};
  STREAM_STATE_DEL_ROCKSDB(pState, "state", &sKey);
  return code;
}
int32_t streamStateClear_rocksdb(SStreamState* pState) {
  stDebug("streamStateClear_rocksdb");

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  TAOS_UNUSED(atomic_add_fetch_64(&wrapper->dataWritten, 1));

  char      sKeyStr[128] = {0};
  char      eKeyStr[128] = {0};
  SStateKey sKey = {.key = {.ts = 0, .groupId = 0}, .opNum = pState->number};
  SStateKey eKey = {.key = {.ts = INT64_MAX, .groupId = UINT64_MAX}, .opNum = pState->number};

  int sLen = stateKeyEncode(&sKey, sKeyStr);
  int eLen = stateKeyEncode(&eKey, eKeyStr);

  if (wrapper->pCf[1] != NULL) {
    char* err = NULL;
    rocksdb_delete_range_cf(wrapper->db, wrapper->writeOpt, wrapper->pCf[1], sKeyStr, sLen, eKeyStr, eLen, &err);
    if (err != NULL) {
      char toStringStart[128] = {0};
      char toStringEnd[128] = {0};
      TAOS_UNUSED(stateKeyToString(&sKey, toStringStart));
      TAOS_UNUSED(stateKeyToString(&eKey, toStringEnd));

      stWarn("failed to delete range cf(state) start: %s, end:%s, reason:%s", toStringStart, toStringEnd, err);
      taosMemoryFree(err);
    } else {
      rocksdb_compact_range_cf(wrapper->db, wrapper->pCf[1], sKeyStr, sLen, eKeyStr, eLen);
    }
  }

  return 0;
}
void streamStateCurNext_rocksdb(SStreamStateCur* pCur) {
  if (pCur && pCur->iter && rocksdb_iter_valid(pCur->iter)) {
    rocksdb_iter_next(pCur->iter);
  }
}
int32_t streamStateGetFirst_rocksdb(SStreamState* pState, SWinKey* key) {
  int code = 0;
  stDebug("streamStateGetFirst_rocksdb");
  SWinKey tmp = {.ts = 0, .groupId = 0};
  code = streamStatePut_rocksdb(pState, &tmp, NULL, 0);
  if (code != 0) {
    return code;
  }

  SStreamStateCur* pCur = streamStateSeekKeyNext_rocksdb(pState, &tmp);
  code = streamStateGetKVByCur_rocksdb(pState, pCur, key, NULL, 0);
  if (code != 0) {
    return code;
  }
  streamStateFreeCur(pCur);
  return streamStateDel_rocksdb(pState, &tmp);
}

int32_t streamStateFillGetGroupKVByCur_rocksdb(SStreamStateCur* pCur, SWinKey* pKey, const void** pVal,
                                               int32_t* pVLen) {
  if (!pCur) {
    return -1;
  }
  uint64_t groupId = pKey->groupId;

  int32_t code = streamStateFillGetKVByCur_rocksdb(pCur, pKey, pVal, pVLen);
  if (code == 0) {
    if (pKey->groupId == groupId) {
      return 0;
    }
    if (pVal != NULL) {
      taosMemoryFree((void*)*pVal);
      *pVal = NULL;
    }
  }
  return -1;
}
int32_t streamStateAddIfNotExist_rocksdb(SStreamState* pState, const SWinKey* key, void** pVal, int32_t* pVLen) {
  stDebug("streamStateAddIfNotExist_rocksdb");
  int32_t size = *pVLen;
  if (streamStateGet_rocksdb(pState, key, pVal, pVLen) == 0) {
    return 0;
  }
  *pVal = taosMemoryMalloc(size);
  if (*pVal == NULL) {
    return terrno;
  }
  memset(*pVal, 0, size);
  return 0;
}
void streamStateCurPrev_rocksdb(SStreamStateCur* pCur) {
  if (pCur) {
    rocksdb_iter_prev(pCur->iter);
  }
}
int32_t streamStateGetKVByCur_rocksdb(SStreamState* pState, SStreamStateCur* pCur, SWinKey* pKey, const void** pVal,
                                      int32_t* pVLen) {
  if (!pCur) return -1;
  SStateKey  tkey;
  SStateKey* pKtmp = &tkey;

  if (rocksdb_iter_valid(pCur->iter) && !iterValueIsStale(pCur->iter)) {
    size_t tlen;
    char*  keyStr = (char*)rocksdb_iter_key(pCur->iter, &tlen);
    TAOS_UNUSED(stateKeyDecode((void*)pKtmp, keyStr));
    if (pKtmp->opNum != pCur->number) {
      return -1;
    }

    if (pVLen != NULL) {
      size_t      vlen = 0;
      const char* valStr = rocksdb_iter_value(pCur->iter, &vlen);
      char*       val = NULL;
      int32_t     len = valueDecode((void*)valStr, vlen, NULL, (char**)val);
      if (len <= 0) {
        taosMemoryFree(val);
        return -1;
      }

      char*  tVal = val;
      size_t tVlen = len;

      if (pVal != NULL) {
        if (pState != NULL && pState->pResultRowStore.resultRowGet != NULL && pState->pExprSupp != NULL) {
          int code =
              (pState->pResultRowStore.resultRowGet)(pState->pExprSupp, val, len, (char**)&tVal, (size_t*)&tVlen);
          if (code != 0) {
            taosMemoryFree(val);
            return code;
          }
          taosMemoryFree(val);
          *pVal = (char*)tVal;
        } else {
          stInfo("streamStateGetKVByCur_rocksdb, pState = %p, pResultRowStore = %p, pExprSupp = %p", pState,
                 pState->pResultRowStore.resultRowGet, pState->pExprSupp);
          *pVal = (char*)tVal;
        }
      } else {
        taosMemoryFree(val);
      }
      *pVLen = (int32_t)tVlen;
    }

    *pKey = pKtmp->key;
    return 0;
  }
  return -1;
}
SStreamStateCur* streamStateGetAndCheckCur_rocksdb(SStreamState* pState, SWinKey* key) {
  SStreamStateCur* pCur = streamStateFillGetCur_rocksdb(pState, key);
  if (pCur) {
    int32_t code = streamStateFillGetGroupKVByCur_rocksdb(pCur, key, NULL, 0);
    if (code == 0) return pCur;
    streamStateFreeCur(pCur);
  }
  return NULL;
}

SStreamStateCur* streamStateSeekKeyNext_rocksdb(SStreamState* pState, const SWinKey* key) {
  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  pCur->number = pState->number;
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "state", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);

  SStateKey sKey = {.key = *key, .opNum = pState->number};
  char      buf[128] = {0};
  int       len = stateKeyEncode((void*)&sKey, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  // skip ttl expired data
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_next(pCur->iter);
  }

  if (rocksdb_iter_valid(pCur->iter)) {
    SStateKey curKey;
    size_t    kLen;
    char*     keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(stateKeyDecode((void*)&curKey, keyStr));
    if (stateKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) > 0) {
      return pCur;
    }
    rocksdb_iter_next(pCur->iter);
    return pCur;
  }
  streamStateFreeCur(pCur);
  return NULL;
}

SStreamStateCur* streamStateSeekToLast_rocksdb(SStreamState* pState) {
  int32_t code = 0;

  const SStateKey maxStateKey = {.key = {.groupId = UINT64_MAX, .ts = INT64_MAX}, .opNum = INT64_MAX};
  STREAM_STATE_PUT_ROCKSDB(pState, "state", &maxStateKey, "", 0);
  if (code != 0) {
    return NULL;
  }

  {
    char tbuf[256] = {0};
    TAOS_UNUSED(stateKeyToString((void*)&maxStateKey, tbuf));
    stDebug("seek to last:%s", tbuf);
  }

  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) return NULL;

  pCur->number = pState->number;

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  pCur->db = ((STaskDbWrapper*)wrapper)->db;

  pCur->iter = streamStateIterCreate(pState, "state", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);

  char    buf[128] = {0};
  int32_t klen = stateKeyEncode((void*)&maxStateKey, buf);
  rocksdb_iter_seek(pCur->iter, buf, (size_t)klen);
  rocksdb_iter_prev(pCur->iter);
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_prev(pCur->iter);
  }

  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    pCur = NULL;
  }

  STREAM_STATE_DEL_ROCKSDB(pState, "state", &maxStateKey);
  return pCur;
}

SStreamStateCur* streamStateGetCur_rocksdb(SStreamState* pState, const SWinKey* key) {
  stDebug("streamStateGetCur_rocksdb");

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) return NULL;

  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "state", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  SStateKey sKey = {.key = *key, .opNum = pState->number};
  char      buf[128] = {0};
  int       len = stateKeyEncode((void*)&sKey, buf);

  rocksdb_iter_seek(pCur->iter, buf, len);

  if (rocksdb_iter_valid(pCur->iter) && !iterValueIsStale(pCur->iter)) {
    SStateKey curKey;
    size_t    kLen = 0;
    char*     keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(stateKeyDecode((void*)&curKey, keyStr));

    if (stateKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) == 0) {
      pCur->number = pState->number;
      return pCur;
    }
  }
  streamStateFreeCur(pCur);
  return NULL;
}

// func cf
int32_t streamStateFuncPut_rocksdb(SStreamState* pState, const STupleKey* key, const void* value, int32_t vLen) {
  int    code = 0;
  char*  dst = NULL;
  size_t size = 0;
  if (pState->pResultRowStore.resultRowPut == NULL || pState->pExprSupp == NULL) {
    STREAM_STATE_PUT_ROCKSDB(pState, "func", key, (void*)value, (int32_t)vLen);
    return code;
  }
  code = (pState->pResultRowStore.resultRowPut)(pState->pExprSupp, value, vLen, &dst, &size);
  if (code != 0) {
    return code;
  }
  STREAM_STATE_PUT_ROCKSDB(pState, "func", key, (void*)dst, (int32_t)size);
  taosMemoryFree(dst);

  return code;
}
int32_t streamStateFuncGet_rocksdb(SStreamState* pState, const STupleKey* key, void** pVal, int32_t* pVLen) {
  int    code = 0;
  char*  tVal = NULL;
  size_t tValLen = 0;
  STREAM_STATE_GET_ROCKSDB(pState, "func", key, tVal, &tValLen);
  if (code != 0) {
    taosMemoryFree(tVal);
    return code;
  }

  if (pState->pResultRowStore.resultRowGet == NULL || pState->pExprSupp == NULL) {
    *pVal = tVal;
    *pVLen = tValLen;
    return code;
  }

  size_t pValLen = 0;
  code = (pState->pResultRowStore.resultRowGet)(pState->pExprSupp, tVal, tValLen, (char**)pVal, &pValLen);
  *pVLen = (int32_t)pValLen;

  taosMemoryFree(tVal);
  return code;
}
int32_t streamStateFuncDel_rocksdb(SStreamState* pState, const STupleKey* key) {
  int code = 0;
  STREAM_STATE_DEL_ROCKSDB(pState, "func", key);
  return 0;
}

// session cf
int32_t streamStateSessionPut_rocksdb(SStreamState* pState, const SSessionKey* key, const void* value, int32_t vLen) {
  int              code = 0;
  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};
  if (value == NULL || vLen == 0) {
    stError("streamStateSessionPut_rocksdb val: %p, len: %d", value, vLen);
  }
  char*  dst = NULL;
  size_t size = 0;
  if (pState->pResultRowStore.resultRowPut == NULL || pState->pExprSupp == NULL) {
    STREAM_STATE_PUT_ROCKSDB(pState, "sess", &sKey, (void*)value, (int32_t)vLen);
    return code;
  }

  code = (pState->pResultRowStore.resultRowPut)(pState->pExprSupp, value, vLen, &dst, &size);
  if (code != 0) {
    return code;
  }
  STREAM_STATE_PUT_ROCKSDB(pState, "sess", &sKey, dst, (int32_t)size);
  taosMemoryFree(dst);

  return code;
}
int32_t streamStateSessionGet_rocksdb(SStreamState* pState, SSessionKey* key, void** pVal, int32_t* pVLen) {
  stDebug("streamStateSessionGet_rocksdb");
  int              code = 0;
  SStreamStateCur* pCur = streamStateSessionSeekKeyCurrentNext_rocksdb(pState, key);
  SSessionKey      resKey = *key;
  void*            tmp = NULL;
  int32_t          vLen = 0;

  code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, &resKey, &tmp, &vLen);
  if (code == 0 && key->win.skey == resKey.win.skey) {
    *key = resKey;

    if (pVal) {
      *pVal = tmp;
      tmp = NULL;
    };
    if (pVLen) *pVLen = vLen;
  } else {
    code = -1;
  }

  taosMemoryFree(tmp);
  streamStateFreeCur(pCur);
  return code;
}

int32_t streamStateSessionDel_rocksdb(SStreamState* pState, const SSessionKey* key) {
  int              code = 0;
  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};
  STREAM_STATE_DEL_ROCKSDB(pState, "sess", &sKey);
  return code;
}

SStreamStateCur* streamStateSessionSeekToLast_rocksdb(SStreamState* pState, int64_t groupId) {
  stDebug("streamStateSessionSeekToLast_rocksdb");

  int32_t code = 0;

  SSessionKey      maxSessionKey = {.groupId = groupId, .win = {.skey = INT64_MAX, .ekey = INT64_MAX}};
  SStateSessionKey maxKey = {.key = maxSessionKey, .opNum = pState->number};

  STREAM_STATE_PUT_ROCKSDB(pState, "sess", &maxKey, "", 0);
  if (code != 0) {
    return NULL;
  }
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  SStreamStateCur* pCur = createStreamStateCursor();
  pCur->number = pState->number;
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "sess", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);

  char    buf[128] = {0};
  int32_t klen = stateSessionKeyEncode((void*)&maxKey, buf);
  rocksdb_iter_seek(pCur->iter, buf, (size_t)klen);
  rocksdb_iter_prev(pCur->iter);
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_prev(pCur->iter);
  }

  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    pCur = NULL;
  }

  STREAM_STATE_DEL_ROCKSDB(pState, "sess", &maxKey);
  return pCur;
}

int32_t streamStateSessionCurPrev_rocksdb(SStreamStateCur* pCur) {
  stDebug("streamStateCurPrev_rocksdb");
  if (!pCur) return -1;

  rocksdb_iter_prev(pCur->iter);
  return 0;
}

SStreamStateCur* streamStateSessionSeekKeyCurrentPrev_rocksdb(SStreamState* pState, const SSessionKey* key) {
  stDebug("streamStateSessionSeekKeyCurrentPrev_rocksdb");

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }

  pCur->number = pState->number;
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "sess", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  if (pCur->iter == NULL) {
    streamStateFreeCur(pCur);
    return NULL;
  }

  char             buf[128] = {0};
  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};
  int              len = stateSessionKeyEncode(&sKey, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) rocksdb_iter_prev(pCur->iter);

  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }

  int32_t          c = 0;
  size_t           klen;
  const char*      iKey = rocksdb_iter_key(pCur->iter, &klen);
  SStateSessionKey curKey = {0};
  TAOS_UNUSED(stateSessionKeyDecode(&curKey, (char*)iKey));
  if (stateSessionKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) >= 0) return pCur;

  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }

  rocksdb_iter_prev(pCur->iter);
  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  return pCur;
}
SStreamStateCur* streamStateSessionSeekKeyCurrentNext_rocksdb(SStreamState* pState, const SSessionKey* key) {
  stDebug("streamStateSessionSeekKeyCurrentNext_rocksdb");

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "sess", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  char             buf[128] = {0};
  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};
  int              len = stateSessionKeyEncode(&sKey, buf);

  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  if (iterValueIsStale(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  size_t           klen;
  const char*      iKey = rocksdb_iter_key(pCur->iter, &klen);
  SStateSessionKey curKey = {0};
  TAOS_UNUSED(stateSessionKeyDecode(&curKey, (char*)iKey));
  if (stateSessionKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) <= 0) return pCur;

  rocksdb_iter_next(pCur->iter);
  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  return pCur;
}

SStreamStateCur* streamStateSessionSeekKeyNext_rocksdb(SStreamState* pState, const SSessionKey* key) {
  stDebug("streamStateSessionSeekKeyNext_rocksdb");
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "sess", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};

  char buf[128] = {0};
  int  len = stateSessionKeyEncode(&sKey, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) rocksdb_iter_next(pCur->iter);
  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }

  size_t           klen;
  const char*      iKey = rocksdb_iter_key(pCur->iter, &klen);
  SStateSessionKey curKey = {0};
  TAOS_UNUSED(stateSessionKeyDecode(&curKey, (char*)iKey));
  if (stateSessionKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) < 0) return pCur;

  rocksdb_iter_next(pCur->iter);
  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  return pCur;
}

SStreamStateCur* streamStateSessionSeekKeyPrev_rocksdb(SStreamState* pState, const SSessionKey* key) {
  stDebug("streamStateSessionSeekKeyPrev_rocksdb");
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "sess", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};

  char buf[128] = {0};
  int  len = stateSessionKeyEncode(&sKey, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) rocksdb_iter_prev(pCur->iter);
  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }

  size_t           klen;
  const char*      iKey = rocksdb_iter_key(pCur->iter, &klen);
  SStateSessionKey curKey = {0};
  TAOS_UNUSED(stateSessionKeyDecode(&curKey, (char*)iKey));
  if (stateSessionKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) > 0) return pCur;

  rocksdb_iter_prev(pCur->iter);
  if (!rocksdb_iter_valid(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  return pCur;
}

int32_t streamStateSessionGetKVByCur_rocksdb(SStreamState* pState, SStreamStateCur* pCur, SSessionKey* pKey,
                                             void** pVal, int32_t* pVLen) {
  if (!pCur) {
    return -1;
  }
  SStateSessionKey ktmp = {0};
  size_t           kLen = 0, vLen = 0;

  if (!rocksdb_iter_valid(pCur->iter) || iterValueIsStale(pCur->iter)) {
    return -1;
  }
  const char* curKey = rocksdb_iter_key(pCur->iter, (size_t*)&kLen);
  TAOS_UNUSED(stateSessionKeyDecode((void*)&ktmp, (char*)curKey));

  if (pVal != NULL) *pVal = NULL;
  if (pVLen != NULL) *pVLen = 0;

  SStateSessionKey* pKTmp = &ktmp;
  const char*       vval = rocksdb_iter_value(pCur->iter, (size_t*)&vLen);
  char*             val = NULL;
  int32_t           len = valueDecode((void*)vval, vLen, NULL, &val);
  if (len < 0) {
    taosMemoryFree(val);
    return -1;
  }

  if (pKTmp->opNum != pCur->number) {
    taosMemoryFree(val);
    return -1;
  }
  if (pKey->groupId != 0 && pKey->groupId != pKTmp->key.groupId) {
    taosMemoryFree(val);
    return -1;
  }

  char*  tVal = val;
  size_t tVlen = len;

  if (pVal != NULL) {
    if (pState != NULL && pState->pResultRowStore.resultRowGet != NULL && pState->pExprSupp != NULL) {
      int code = (pState->pResultRowStore.resultRowGet)(pState->pExprSupp, val, len, (char**)&tVal, (size_t*)&tVlen);
      if (code != 0) {
        taosMemoryFree(val);
        return code;
      }
      taosMemoryFree(val);
      *pVal = (char*)tVal;
    } else {
      *pVal = (char*)tVal;
    }
  } else {
    taosMemoryFree(val);
  }

  if (pVLen != NULL) *pVLen = (int32_t)tVlen;

  *pKey = pKTmp->key;
  return 0;
}
// fill cf
int32_t streamStateFillPut_rocksdb(SStreamState* pState, const SWinKey* key, const void* value, int32_t vLen) {
  int code = 0;

  STREAM_STATE_PUT_ROCKSDB(pState, "fill", key, value, vLen);
  return code;
}

int32_t streamStateFillGet_rocksdb(SStreamState* pState, const SWinKey* key, void** pVal, int32_t* pVLen) {
  int code = 0;

  STREAM_STATE_GET_ROCKSDB(pState, "fill", key, pVal, pVLen);
  return code;
}
int32_t streamStateFillDel_rocksdb(SStreamState* pState, const SWinKey* key) {
  int code = 0;
  STREAM_STATE_DEL_ROCKSDB(pState, "fill", key);
  return code;
}

SStreamStateCur* streamStateFillGetCur_rocksdb(SStreamState* pState, const SWinKey* key) {
  stDebug("streamStateFillGetCur_rocksdb");
  SStreamStateCur* pCur = createStreamStateCursor();
  STaskDbWrapper*  wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  if (pCur == NULL) return NULL;

  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "fill", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  char buf[128] = {0};
  int  len = winKeyEncode((void*)key, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  if (iterValueIsStale(pCur->iter)) {
    streamStateFreeCur(pCur);
    return NULL;
  }

  if (rocksdb_iter_valid(pCur->iter)) {
    size_t  kLen;
    SWinKey curKey;
    char*   keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(winKeyDecode((void*)&curKey, keyStr));
    if (winKeyCmpr(key, sizeof(*key), &curKey, sizeof(curKey)) == 0) {
      return pCur;
    }
  }

  streamStateFreeCur(pCur);
  return NULL;
}
int32_t streamStateFillGetKVByCur_rocksdb(SStreamStateCur* pCur, SWinKey* pKey, const void** pVal, int32_t* pVLen) {
  if (!pCur) {
    return -1;
  }
  SWinKey winKey;
  if (!rocksdb_iter_valid(pCur->iter) || iterValueIsStale(pCur->iter)) {
    return -1;
  }
  size_t klen, vlen;
  char*  keyStr = (char*)rocksdb_iter_key(pCur->iter, &klen);
  TAOS_UNUSED(winKeyDecode(&winKey, keyStr));

  const char* valStr = rocksdb_iter_value(pCur->iter, &vlen);
  int32_t     len = valueDecode((void*)valStr, vlen, NULL, (char**)pVal);
  if (len < 0) {
    return -1;
  }
  if (pVLen != NULL) *pVLen = len;

  *pKey = winKey;
  return 0;
}

SStreamStateCur* streamStateFillSeekKeyNext_rocksdb(SStreamState* pState, const SWinKey* key) {
  stDebug("streamStateFillSeekKeyNext_rocksdb");
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  SStreamStateCur* pCur = createStreamStateCursor();
  if (!pCur) {
    return NULL;
  }

  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "fill", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  char buf[128] = {0};
  int  len = winKeyEncode((void*)key, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  // skip stale data
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_next(pCur->iter);
  }

  if (rocksdb_iter_valid(pCur->iter)) {
    SWinKey curKey;
    size_t  kLen = 0;
    char*   keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(winKeyDecode((void*)&curKey, keyStr));
    if (winKeyCmpr(key, sizeof(*key), &curKey, sizeof(curKey)) < 0) {
      return pCur;
    }
    rocksdb_iter_next(pCur->iter);
    return pCur;
  }
  streamStateFreeCur(pCur);
  return NULL;
}
SStreamStateCur* streamStateFillSeekKeyPrev_rocksdb(SStreamState* pState, const SWinKey* key) {
  stDebug("streamStateFillSeekKeyPrev_rocksdb");
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }

  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "fill", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  char buf[128] = {0};
  int  len = winKeyEncode((void*)key, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_prev(pCur->iter);
  }

  if (rocksdb_iter_valid(pCur->iter)) {
    SWinKey curKey;
    size_t  kLen = 0;
    char*   keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(winKeyDecode((void*)&curKey, keyStr));
    if (winKeyCmpr(key, sizeof(*key), &curKey, sizeof(curKey)) > 0) {
      return pCur;
    }
    rocksdb_iter_prev(pCur->iter);
    return pCur;
  }

  streamStateFreeCur(pCur);
  return NULL;
}

SStreamStateCur* streamStateFillSeekToLast_rocksdb(SStreamState* pState) {
  SWinKey key = {.groupId = UINT64_MAX, .ts = INT64_MAX};
  return streamStateFillSeekKeyPrev_rocksdb(pState, &key);
}

#ifdef BUILD_NO_CALL
int32_t streamStateSessionGetKeyByRange_rocksdb(SStreamState* pState, const SSessionKey* key, SSessionKey* curKey) {
  stDebug("streamStateSessionGetKeyByRange_rocksdb");
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return -1;
  }
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "sess", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  SStateSessionKey sKey = {.key = *key, .opNum = pState->number};
  int32_t          c = 0;
  char             buf[128] = {0};
  int              len = stateSessionKeyEncode(&sKey, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return -1;
  }

  size_t           kLen;
  const char*      iKeyStr = rocksdb_iter_key(pCur->iter, (size_t*)&kLen);
  SStateSessionKey iKey = {0};
  stateSessionKeyDecode(&iKey, (char*)iKeyStr);

  c = stateSessionKeyCmpr(&sKey, sizeof(sKey), &iKey, sizeof(iKey));

  SSessionKey resKey = *key;
  int32_t     code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, &resKey, NULL, NULL);
  if (code == 0 && sessionRangeKeyCmpr(key, &resKey) == 0) {
    *curKey = resKey;
    streamStateFreeCur(pCur);
    return code;
  }

  if (c > 0) {
    streamStateCurNext_rocksdb(pCur);
    code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, &resKey, NULL, NULL);
    if (code == 0 && sessionRangeKeyCmpr(key, &resKey) == 0) {
      *curKey = resKey;
      streamStateFreeCur(pCur);
      return code;
    }
  } else if (c < 0) {
    streamStateCurPrev(pState, pCur);
    code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, &resKey, NULL, NULL);
    if (code == 0 && sessionRangeKeyCmpr(key, &resKey) == 0) {
      *curKey = resKey;
      streamStateFreeCur(pCur);
      return code;
    }
  }

  streamStateFreeCur(pCur);
  return -1;
}
#endif

int32_t streamStateSessionAddIfNotExist_rocksdb(SStreamState* pState, SSessionKey* key, TSKEY gap, void** pVal,
                                                int32_t* pVLen) {
  stDebug("streamStateSessionAddIfNotExist_rocksdb");
  // todo refactor
  int32_t     res = 0;
  SSessionKey originKey = *key;
  SSessionKey searchKey = *key;
  searchKey.win.skey = key->win.skey - gap;
  searchKey.win.ekey = key->win.ekey + gap;
  int32_t valSize = *pVLen;

  void* tmp = taosMemoryMalloc(valSize);
  if (tmp == NULL) {
    return terrno;
  }

  SStreamStateCur* pCur = streamStateSessionSeekKeyCurrentPrev_rocksdb(pState, key);
  int32_t          code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, key, pVal, pVLen);

  if (code == 0) {
    if (sessionRangeKeyCmpr(&searchKey, key) == 0) {
      memcpy(tmp, *pVal, *pVLen);
      taosMemoryFreeClear(*pVal);
      goto _end;
    }
    taosMemoryFreeClear(*pVal);
    streamStateCurNext_rocksdb(pCur);
  } else {
    *key = originKey;
    streamStateFreeCur(pCur);
    taosMemoryFreeClear(*pVal);
    pCur = streamStateSessionSeekKeyNext_rocksdb(pState, key);
  }

  code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, key, pVal, pVLen);
  if (code == 0) {
    if (sessionRangeKeyCmpr(&searchKey, key) == 0) {
      memcpy(tmp, *pVal, *pVLen);
      goto _end;
    }
  }

  *key = originKey;
  res = 1;
  memset(tmp, 0, valSize);

_end:
  taosMemoryFree(*pVal);
  *pVal = tmp;
  streamStateFreeCur(pCur);
  return res;
}
void streamStateSessionClear_rocksdb(SStreamState* pState) {
  stDebug("streamStateSessionClear_rocksdb");
  SSessionKey      key = {.win.skey = 0, .win.ekey = 0, .groupId = 0};
  SStreamStateCur* pCur = streamStateSessionSeekKeyCurrentNext_rocksdb(pState, &key);

  while (1) {
    SSessionKey delKey = {0};
    void*       buf = NULL;
    int32_t     size = 0;
    int32_t     code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, &delKey, &buf, &size);
    if (code == 0 && size > 0) {
      memset(buf, 0, size);
      // refactor later
      TAOS_UNUSED(streamStateSessionPut_rocksdb(pState, &delKey, buf, size));
    } else {
      taosMemoryFreeClear(buf);
      break;
    }
    taosMemoryFreeClear(buf);

    streamStateCurNext_rocksdb(pCur);
  }
  streamStateFreeCur(pCur);
}
int32_t streamStateStateAddIfNotExist_rocksdb(SStreamState* pState, SSessionKey* key, char* pKeyData,
                                              int32_t keyDataLen, state_key_cmpr_fn fn, void** pVal, int32_t* pVLen) {
  stDebug("streamStateStateAddIfNotExist_rocksdb");
  // todo refactor
  int32_t     res = 0;
  SSessionKey tmpKey = *key;
  int32_t     valSize = *pVLen;

  SStreamStateCur* pCur = streamStateSessionSeekKeyCurrentPrev_rocksdb(pState, key);
  int32_t          code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, key, pVal, pVLen);
  if (code == 0) {
    if (key->win.skey <= tmpKey.win.skey && tmpKey.win.ekey <= key->win.ekey) {
      goto _end;
    }

    void* stateKey = (char*)(*pVal) + (valSize - keyDataLen);
    if (fn(pKeyData, stateKey) == true) {
      goto _end;
    }

    streamStateCurNext_rocksdb(pCur);
  } else {
    *key = tmpKey;
    streamStateFreeCur(pCur);
    pCur = streamStateSessionSeekKeyNext_rocksdb(pState, key);
  }
  taosMemoryFreeClear(*pVal);
  code = streamStateSessionGetKVByCur_rocksdb(pState, pCur, key, pVal, pVLen);
  if (code == 0) {
    void* stateKey = (char*)(*pVal) + (valSize - keyDataLen);
    if (fn(pKeyData, stateKey) == true) {
      goto _end;
    }
  }
  taosMemoryFreeClear(*pVal);

  *key = tmpKey;
  res = 1;

_end:
  if (res == 0 && valSize > *pVLen){
    stError("[InternalERR] [skey:%"PRId64 ",ekey:%"PRId64 ",groupId:%"PRIu64 "],valSize:%d bigger than get rocksdb len:%d", key->win.skey, key->win.ekey, key->groupId, valSize, *pVLen);
  }
  streamStateFreeCur(pCur);
  return res;
}

//  partag cf
int32_t streamStatePutParTag_rocksdb(SStreamState* pState, int64_t groupId, const void* tag, int32_t tagLen) {
  int    code = 0;
  char*  dst = NULL;
  size_t size = 0;
  if (pState->pResultRowStore.resultRowPut == NULL || pState->pExprSupp == NULL || tag == NULL) {
    STREAM_STATE_PUT_ROCKSDB(pState, "partag", &groupId, tag, tagLen);
    return code;
  }
  code = (pState->pResultRowStore.resultRowPut)(pState->pExprSupp, tag, tagLen, &dst, &size);
  if (code != 0) {
    return code;
  }
  STREAM_STATE_PUT_ROCKSDB(pState, "partag", &groupId, dst, (int32_t)size);
  taosMemoryFree(dst);
  return code;
}

void streamStateParTagSeekKeyNext_rocksdb(SStreamState* pState, const int64_t groupId, SStreamStateCur* pCur) {
  if (pCur == NULL) {
    return;
  }
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  pCur->number = pState->number;
  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "partag", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  int i = streamStateGetCfIdx(pState, "partag");
  if (i < 0) {
    stError("streamState failed to put to cf name:%s", "partag");
    return;
  }

  char    buf[128] = {0};
  int32_t klen = ginitDict[i].enFunc((void*)&groupId, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, klen)) {
    return;
  }
  // skip ttl expired data
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_next(pCur->iter);
  }

  if (rocksdb_iter_valid(pCur->iter)) {
    int64_t curGroupId;
    size_t  kLen = 0;
    char*   keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(parKeyDecode((void*)&curGroupId, keyStr));
    if (curGroupId > groupId) return;

    rocksdb_iter_next(pCur->iter);
  }
}

int32_t streamStateParTagGetKVByCur_rocksdb(SStreamStateCur* pCur, int64_t* pGroupId, void** pVal, int32_t* pVLen) {
  stDebug("streamStateFillGetKVByCur_rocksdb");
  if (!pCur) {
    return -1;
  }
  SWinKey winKey;
  if (!rocksdb_iter_valid(pCur->iter) || iterValueIsStale(pCur->iter)) {
    return -1;
  }

  size_t klen, vlen;
  char*  keyStr = (char*)rocksdb_iter_key(pCur->iter, &klen);
  (void)parKeyDecode(pGroupId, keyStr);

  if (pVal) {
    const char* valStr = rocksdb_iter_value(pCur->iter, &vlen);
    int32_t     len = valueDecode((void*)valStr, vlen, NULL, (char**)pVal);
    if (len < 0) {
      return -1;
    }
    if (pVLen != NULL) *pVLen = len;
  }

  return 0;
}

int32_t streamStateGetParTag_rocksdb(SStreamState* pState, int64_t groupId, void** tagVal, int32_t* tagLen) {
  int    code = 0;
  char*  tVal;
  size_t tValLen = 0;
  STREAM_STATE_GET_ROCKSDB(pState, "partag", &groupId, &tVal, &tValLen);
  if (code != 0) {
    taosMemoryFree(tVal);
    return code;
  }
  code = (pState->pResultRowStore.resultRowGet)(pState->pExprSupp, tVal, tValLen, (char**)tagVal, (size_t*)tagLen);
  taosMemoryFree(tVal);

  return code;
}

// parname cfg
int32_t streamStatePutParName_rocksdb(SStreamState* pState, int64_t groupId, const char tbname[TSDB_TABLE_NAME_LEN]) {
  int code = 0;
  STREAM_STATE_PUT_ROCKSDB(pState, "parname", &groupId, (char*)tbname, TSDB_TABLE_NAME_LEN);
  return code;
}
int32_t streamStateGetParName_rocksdb(SStreamState* pState, int64_t groupId, void** pVal) {
  int    code = 0;
  size_t tagLen;
  STREAM_STATE_GET_ROCKSDB(pState, "parname", &groupId, pVal, &tagLen);
  return code;
}

int32_t streamStateDeleteParName_rocksdb(SStreamState* pState, int64_t groupId) {
  int code = 0;
  STREAM_STATE_DEL_ROCKSDB(pState, "parname", &groupId);
  return code;
}

int32_t streamDefaultPut_rocksdb(SStreamState* pState, const void* key, void* pVal, int32_t pVLen) {
  int code = 0;
  STREAM_STATE_PUT_ROCKSDB(pState, "default", key, pVal, pVLen);
  return code;
}
int32_t streamDefaultGet_rocksdb(SStreamState* pState, const void* key, void** pVal, int32_t* pVLen) {
  int code = 0;
  STREAM_STATE_GET_ROCKSDB(pState, "default", key, pVal, pVLen);
  return code;
}
int32_t streamDefaultDel_rocksdb(SStreamState* pState, const void* key) {
  int code = 0;
  STREAM_STATE_DEL_ROCKSDB(pState, "default", key);
  return code;
}

int32_t streamDefaultIterGet_rocksdb(SStreamState* pState, const void* start, const void* end, SArray* result) {
  int   code = 0;
  char* err = NULL;

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  rocksdb_snapshot_t*    snapshot = NULL;
  rocksdb_readoptions_t* readopts = NULL;
  rocksdb_iterator_t*    pIter = streamStateIterCreate(pState, "default", &snapshot, &readopts);
  if (pIter == NULL) {
    return -1;
  }
  size_t klen = 0;
  rocksdb_iter_seek(pIter, start, strlen(start));
  while (rocksdb_iter_valid(pIter)) {
    const char* key = rocksdb_iter_key(pIter, &klen);
    int32_t     vlen = 0;
    const char* vval = rocksdb_iter_value(pIter, (size_t*)&vlen);
    char*       val = NULL;
    int32_t     len = valueDecode((void*)vval, vlen, NULL, NULL);
    if (len < 0) {
      rocksdb_iter_next(pIter);
      continue;
    }

    if (end != NULL && strcmp(key, end) > 0) {
      break;
    }
    if (strncmp(key, start, strlen(start)) == 0 && strlen(key) >= strlen(start) + 1) {
      int64_t checkPoint = 0;
      if (sscanf(key + strlen(key), ":%" PRId64, &checkPoint) == 1) {
        if (taosArrayPush(result, &checkPoint) == NULL) {
          code = terrno;
          break;
        }
      }
    } else {
      break;
    }
    rocksdb_iter_next(pIter);
  }
  rocksdb_release_snapshot(wrapper->db, snapshot);
  rocksdb_readoptions_destroy(readopts);
  rocksdb_iter_destroy(pIter);
  return code;
}
#ifdef BUILD_NO_CALL
void* streamDefaultIterCreate_rocksdb(SStreamState* pState) {
  SStreamStateCur* pCur = createStreamStateCursor();
  STaskDbWrapper*  wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "default", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;
  return pCur;
}
bool streamDefaultIterValid_rocksdb(void* iter) {
  if (iter) {
    return false;
  }
  SStreamStateCur* pCur = iter;
  return (rocksdb_iter_valid(pCur->iter) && !iterValueIsStale(pCur->iter)) ? true : false;
}
void streamDefaultIterSeek_rocksdb(void* iter, const char* key) {
  SStreamStateCur* pCur = iter;
  rocksdb_iter_seek(pCur->iter, key, strlen(key));
}
void streamDefaultIterNext_rocksdb(void* iter) {
  SStreamStateCur* pCur = iter;
  rocksdb_iter_next(pCur->iter);
}
char* streamDefaultIterKey_rocksdb(void* iter, int32_t* len) {
  SStreamStateCur* pCur = iter;
  return (char*)rocksdb_iter_key(pCur->iter, (size_t*)len);
}
char* streamDefaultIterVal_rocksdb(void* iter, int32_t* len) {
  SStreamStateCur* pCur = iter;
  char*            ret = NULL;

  int32_t     vlen = 0;
  const char* val = rocksdb_iter_value(pCur->iter, (size_t*)&vlen);
  *len = valueDecode((void*)val, vlen, NULL, &ret);
  if (*len < 0) {
    taosMemoryFree(ret);
    return NULL;
  }

  return ret;
}
#endif
// batch func
void* streamStateCreateBatch() {
  rocksdb_writebatch_t* pBatch = rocksdb_writebatch_create();
  return pBatch;
}
int32_t streamStateGetBatchSize(void* pBatch) {
  if (pBatch == NULL) return 0;
  return rocksdb_writebatch_count(pBatch);
}

void    streamStateClearBatch(void* pBatch) { rocksdb_writebatch_clear((rocksdb_writebatch_t*)pBatch); }
void    streamStateDestroyBatch(void* pBatch) { rocksdb_writebatch_destroy((rocksdb_writebatch_t*)pBatch); }
int32_t streamStatePutBatch(SStreamState* pState, const char* cfKeyName, rocksdb_writebatch_t* pBatch, void* key,
                            void* val, int32_t vlen, int64_t ttl) {
  int32_t         code = 0;
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  TAOS_UNUSED(atomic_add_fetch_64(&wrapper->dataWritten, 1));

  int i = streamStateGetCfIdx(pState, cfKeyName);
  if (i < 0) {
    stError("streamState failed to put to cf name:%s", cfKeyName);
    return -1;
  }

  char    buf[128] = {0};
  int32_t klen = ginitDict[i].enFunc((void*)key, buf);

  char*   ttlV = NULL;
  int32_t ttlVLen = ginitDict[i].enValueFunc(val, vlen, ttl, &ttlV);

  rocksdb_column_family_handle_t* pCf = wrapper->pCf[ginitDict[i].idx];
  rocksdb_writebatch_put_cf((rocksdb_writebatch_t*)pBatch, pCf, buf, (size_t)klen, ttlV, (size_t)ttlVLen);
  taosMemoryFree(ttlV);

  {
    char tbuf[256] = {0};
    TAOS_UNUSED(ginitDict[i].toStrFunc((void*)key, tbuf));
    stTrace("streamState str: %s succ to write to %s_%s, len: %d", tbuf, wrapper->idstr, ginitDict[i].key, vlen);
  }
  return 0;
}

int32_t streamStatePutBatchOptimize(SStreamState* pState, int32_t cfIdx, rocksdb_writebatch_t* pBatch, void* key,
                                    void* val, int32_t vlen, int64_t ttl, void* tmpBuf) {
  int32_t code = 0;
  char    buf[128] = {0};
  char    toString[128] = {0};

  char*  dst = NULL;
  size_t size = 0;
  if (pState->pResultRowStore.resultRowPut == NULL || pState->pExprSupp == NULL) {
    dst = val;
    size = vlen;
  } else {
    code = (pState->pResultRowStore.resultRowPut)(pState->pExprSupp, val, vlen, &dst, &size);
    if (code != 0) {
      return code;
    }
  }
  int32_t klen = ginitDict[cfIdx].enFunc((void*)key, buf);

  ginitDict[cfIdx].toStrFunc((void*)key, toString);
  stDebug("[InternalERR] write cfIdx:%d key:%s vlen:%d", cfIdx, toString, vlen);

  char*   ttlV = tmpBuf;
  int32_t ttlVLen = ginitDict[cfIdx].enValueFunc(dst, size, ttl, &ttlV);

  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  TAOS_UNUSED(atomic_add_fetch_64(&wrapper->dataWritten, 1));

  rocksdb_column_family_handle_t* pCf = wrapper->pCf[ginitDict[cfIdx].idx];
  rocksdb_writebatch_put_cf((rocksdb_writebatch_t*)pBatch, pCf, buf, (size_t)klen, ttlV, (size_t)ttlVLen);

  if (pState->pResultRowStore.resultRowPut != NULL && pState->pExprSupp != NULL) {
    taosMemoryFree(dst);
  }

  if (tmpBuf == NULL) {
    taosMemoryFree(ttlV);
  }

  {
    char tbuf[256] = {0};
    TAOS_UNUSED(ginitDict[cfIdx].toStrFunc((void*)key, tbuf));
    stTrace("streamState str: %s succ to write to %s_%s", tbuf, wrapper->idstr, ginitDict[cfIdx].key);
  }
  return 0;
}
int32_t streamStatePutBatch_rocksdb(SStreamState* pState, void* pBatch) {
  char*           err = NULL;
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }
  TAOS_UNUSED(atomic_add_fetch_64(&wrapper->dataWritten, 1));
  rocksdb_write(wrapper->db, wrapper->writeOpt, (rocksdb_writebatch_t*)pBatch, &err);
  if (err != NULL) {
    stError("streamState failed to write batch, err:%s", err);
    taosMemoryFree(err);
    return -1;
  } else {
    stDebug("write batch to backend:%p", wrapper->db);
  }
  return 0;
}
uint32_t nextPow2(uint32_t x) {
  if (x <= 1) return 2;
  x = x - 1;
  x = x | (x >> 1);
  x = x | (x >> 2);
  x = x | (x >> 4);
  x = x | (x >> 8);
  x = x | (x >> 16);
  return x + 1;
}

#ifdef BUILD_NO_CALL
int32_t copyFiles(const char* src, const char* dst) {
  int32_t code = 0;
  // opt later, just hard link
  int32_t sLen = strlen(src);
  int32_t dLen = strlen(dst);
  char*   srcName = taosMemoryCalloc(1, sLen + 64);
  char*   dstName = taosMemoryCalloc(1, dLen + 64);

  TdDirPtr pDir = taosOpenDir(src);
  if (pDir == NULL) {
    taosMemoryFree(srcName);
    taosMemoryFree(dstName);
    return -1;
  }

  TdDirEntryPtr de = NULL;
  while ((de = taosReadDir(pDir)) != NULL) {
    char* name = taosGetDirEntryName(de);
    if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) continue;

    sprintf(srcName, "%s%s%s", src, TD_DIRSEP, name);
    sprintf(dstName, "%s%s%s", dst, TD_DIRSEP, name);
    if (!taosDirEntryIsDir(de)) {
      code = taosCopyFile(srcName, dstName);
      if (code == -1) {
        goto _err;
      }
    }

    memset(srcName, 0, sLen + 64);
    memset(dstName, 0, dLen + 64);
  }

_err:
  taosMemoryFreeClear(srcName);
  taosMemoryFreeClear(dstName);
  taosCloseDir(&pDir);
  return code >= 0 ? 0 : -1;
}
#endif

int32_t isBkdDataMeta(char* name, int32_t len) {
  const char* pCurrent = "CURRENT";
  int32_t     currLen = strlen(pCurrent);

  const char* pManifest = "MANIFEST-";
  int32_t     maniLen = strlen(pManifest);

  if (len >= maniLen && strncmp(name, pManifest, maniLen) == 0) {
    return 1;
  } else if (len == currLen && strcmp(name, pCurrent) == 0) {
    return 1;
  }
  return 0;
}
int32_t compareHashTableImpl(SHashObj* p1, SHashObj* p2, SArray* diff) {
  int32_t code = 0;
  size_t  len = 0;
  void*   pIter = taosHashIterate(p2, NULL);
  while (pIter) {
    char* name = taosHashGetKey(pIter, &len);
    if (!isBkdDataMeta(name, len) && !taosHashGet(p1, name, len)) {
      int32_t cap = len + 1;
      char*   fname = taosMemoryCalloc(1, cap);
      if (fname == NULL) {
        return terrno;
      }
      tstrncpy(fname, name, cap);
      if (taosArrayPush(diff, &fname) == NULL) {
        taosMemoryFree(fname);
        return terrno;
      }
    }
    pIter = taosHashIterate(p2, pIter);
  }
  return code;
}
int32_t compareHashTable(SHashObj* p1, SHashObj* p2, SArray* add, SArray* del) {
  int32_t code = 0;

  code = compareHashTableImpl(p1, p2, add);
  if (code != 0) {
    code = compareHashTableImpl(p2, p1, del);
  }

  return code;
}

void hashTableToDebug(SHashObj* pTbl, char** buf) {
  size_t  sz = taosHashGetSize(pTbl);
  int32_t total = 0;
  int32_t cap = sz * 16 + 4;

  char* p = taosMemoryCalloc(1, cap);
  if (p == NULL) {
    stError("failed to alloc memory for stream snapshot debug info");
    return;
  }

  void* pIter = taosHashIterate(pTbl, NULL);
  while (pIter) {
    size_t len = 0;
    char*  name = taosHashGetKey(pIter, &len);
    if (name == NULL || len <= 0) {
      pIter = taosHashIterate(pTbl, pIter);
      continue;
    }
    int32_t left = cap - strlen(p);
    int32_t nBytes = snprintf(p + total, left, "%s,", name);
    if (nBytes <= 0 || nBytes >= left) {
      stError("failed to debug snapshot info since %s", tstrerror(TSDB_CODE_OUT_OF_RANGE));
      taosMemoryFree(p);
      return;
    }

    pIter = taosHashIterate(pTbl, pIter);
    total += nBytes;
  }
  if (total > 0) {
    p[total - 1] = 0;
  }
  *buf = p;
}
void strArrayDebugInfo(SArray* pArr, char** buf) {
  int32_t sz = taosArrayGetSize(pArr);
  if (sz <= 0) return;

  int32_t code = 0;
  int32_t total = 0, nBytes = 0;
  int32_t cap = 64 + sz * 64;

  char* p = (char*)taosMemoryCalloc(1, cap);
  if (p == NULL) {
    stError("failed to alloc memory for stream snapshot debug info");
    return;
  }

  for (int i = 0; i < sz; i++) {
    char*   name = taosArrayGetP(pArr, i);
    int32_t left = cap - strlen(p);
    nBytes = snprintf(p + total, left, "%s,", name);
    if (nBytes <= 0 || nBytes >= left) {
      code = TSDB_CODE_OUT_OF_RANGE;
      stError("failed to debug snapshot info since %s", tstrerror(code));
      taosMemoryFree(p);
      return;
    }

    total += nBytes;
  }

  p[total - 1] = 0;

  *buf = p;
}
void dbChkpDebugInfo(SDbChkp* pDb) {
  if (stDebugFlag & DEBUG_INFO) {
    char* p[4] = {NULL};

    hashTableToDebug(pDb->pSstTbl[pDb->idx], &p[0]);
    if (p[0]) stTrace("chkp previous file: [%s]", p[0]);

    hashTableToDebug(pDb->pSstTbl[1 - pDb->idx], &p[1]);
    if (p[1]) stTrace("chkp curr file: [%s]", p[1]);

    strArrayDebugInfo(pDb->pAdd, &p[2]);
    if (p[2]) stTrace("chkp newly addded file: [%s]", p[2]);

    strArrayDebugInfo(pDb->pDel, &p[3]);
    if (p[3]) stTrace("chkp newly deleted file: [%s]", p[3]);

    for (int i = 0; i < 4; i++) {
      taosMemoryFree(p[i]);
    }
  }
}
int32_t dbChkpGetDelta(SDbChkp* p, int64_t chkpId, SArray* list) {
  int32_t code = 0;
  int32_t nBytes;
  TAOS_UNUSED(taosThreadRwlockWrlock(&p->rwLock));

  p->preCkptId = p->curChkpId;
  p->curChkpId = chkpId;
  const char* pCurrent = "CURRENT";
  int32_t     currLen = strlen(pCurrent);

  const char* pManifest = "MANIFEST-";
  int32_t     maniLen = strlen(pManifest);

  const char* pSST = ".sst";
  int32_t     sstLen = strlen(pSST);

  memset(p->buf, 0, p->len);

  nBytes =
      snprintf(p->buf, p->len, "%s%s%s%scheckpoint%" PRId64, p->path, TD_DIRSEP, "checkpoints", TD_DIRSEP, chkpId);
  if (nBytes <= 0 || nBytes >= p->len) {
    TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));
    return TSDB_CODE_OUT_OF_RANGE;
  }

  taosArrayClearP(p->pAdd, NULL);
  taosArrayClearP(p->pDel, NULL);
  taosHashClear(p->pSstTbl[1 - p->idx]);

  TdDirPtr pDir = taosOpenDir(p->buf);
  if (pDir == NULL) {
    TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));
    return terrno;
  }

  TdDirEntryPtr de = NULL;
  int8_t        dummy = 0;
  while ((de = taosReadDir(pDir)) != NULL) {
    char* name = taosGetDirEntryName(de);
    if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) continue;
    if (strlen(name) == currLen && strcmp(name, pCurrent) == 0) {
      taosMemoryFreeClear(p->pCurrent);

      p->pCurrent = taosStrdup(name);
      if (p->pCurrent == NULL) {
        code = terrno;
        break;
      }
      continue;
    }

    if (strlen(name) >= maniLen && strncmp(name, pManifest, maniLen) == 0) {
      taosMemoryFreeClear(p->pManifest);
      p->pManifest = taosStrdup(name);
      if (p->pManifest == NULL) {
        code = terrno;
        break;
      }
      continue;
    }
    if (strlen(name) >= sstLen && strncmp(name + strlen(name) - 4, pSST, sstLen) == 0) {
      if (taosHashPut(p->pSstTbl[1 - p->idx], name, strlen(name), &dummy, sizeof(dummy)) != 0) {
        break;
      }
      continue;
    }
  }
  TAOS_UNUSED(taosCloseDir(&pDir));
  if (code != 0) {
    TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));
    return code;
  }

  if (p->init == 0) {
    void* pIter = taosHashIterate(p->pSstTbl[1 - p->idx], NULL);
    while (pIter) {
      size_t len = 0;
      char*  name = taosHashGetKey(pIter, &len);
      if (name != NULL && !isBkdDataMeta(name, len)) {
        int32_t cap = len + 1;
        char*   fname = taosMemoryCalloc(1, cap);
        if (fname == NULL) {
          TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));
          return terrno;
        }

        tstrncpy(fname, name, cap);
        if (taosArrayPush(p->pAdd, &fname) == NULL) {
          taosMemoryFree(fname);
          TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));
          return terrno;
        }
      }
      pIter = taosHashIterate(p->pSstTbl[1 - p->idx], pIter);
    }
    if (taosArrayGetSize(p->pAdd) > 0) p->update = 1;

    p->init = 1;
    p->preCkptId = -1;
    p->curChkpId = chkpId;
  } else {
    int32_t code = compareHashTable(p->pSstTbl[p->idx], p->pSstTbl[1 - p->idx], p->pAdd, p->pDel);
    if (code != 0) {
      // dead code
      taosArrayClearP(p->pAdd, NULL);
      taosArrayClearP(p->pDel, NULL);
      taosHashClear(p->pSstTbl[1 - p->idx]);
      p->update = 0;
      return code;
    }

    if (taosArrayGetSize(p->pAdd) == 0 && taosArrayGetSize(p->pDel) == 0) {
      p->update = 0;
    }

    p->preCkptId = p->curChkpId;
    p->curChkpId = chkpId;
  }

  dbChkpDebugInfo(p);

  p->idx = 1 - p->idx;

  TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));

  return code;
}

void dbChkpDestroy(SDbChkp* pChkp);

int32_t dbChkpCreate(char* path, int64_t initChkpId, SDbChkp** ppChkp) {
  int32_t  code = 0;
  SDbChkp* p = taosMemoryCalloc(1, sizeof(SDbChkp));
  if (p == NULL) {
    code = terrno;
    goto _EXIT;
  }

  p->curChkpId = initChkpId;
  p->preCkptId = -1;
  p->pSST = taosArrayInit(64, sizeof(void*));
  if (p->pSST == NULL) {
    code = terrno;
    dbChkpDestroy(p);
    return code;
  }

  p->path = path;
  p->len = strlen(path) + 128;
  p->buf = taosMemoryCalloc(1, p->len);
  if (p->buf == NULL) {
    code = terrno;
    goto _EXIT;
  }

  p->idx = 0;
  p->pSstTbl[0] = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
  if (p->pSstTbl[0] == NULL) {
    code = terrno;
    goto _EXIT;
  }

  p->pSstTbl[1] = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
  if (p->pSstTbl[1] == NULL) {
    code = terrno;
    goto _EXIT;
  }

  p->pAdd = taosArrayInit(64, sizeof(void*));
  if (p->pAdd == NULL) {
    code = terrno;
    goto _EXIT;
  }

  p->pDel = taosArrayInit(64, sizeof(void*));
  if (p->pDel == NULL) {
    code = terrno;
    goto _EXIT;
  }

  p->update = 0;
  TAOS_UNUSED(taosThreadRwlockInit(&p->rwLock, NULL));

  SArray* list = NULL;
  code = dbChkpGetDelta(p, initChkpId, list);
  if (code != 0) {
    goto _EXIT;
  }
  *ppChkp = p;
  return code;
_EXIT:
  dbChkpDestroy(p);
  return code;
}

void dbChkpDestroy(SDbChkp* pChkp) {
  if (pChkp == NULL) return;

  taosMemoryFree(pChkp->buf);
  taosMemoryFree(pChkp->path);

  taosArrayDestroyP(pChkp->pSST, NULL);
  taosArrayDestroyP(pChkp->pAdd, NULL);
  taosArrayDestroyP(pChkp->pDel, NULL);

  taosHashCleanup(pChkp->pSstTbl[0]);
  taosHashCleanup(pChkp->pSstTbl[1]);

  taosMemoryFree(pChkp->pCurrent);
  taosMemoryFree(pChkp->pManifest);
  taosMemoryFree(pChkp);
}
#ifdef BUILD_NO_CALL
int32_t dbChkpInit(SDbChkp* p) {
  if (p == NULL) return 0;
  return 0;
}
#endif
int32_t dbChkpDumpTo(SDbChkp* p, char* dname, SArray* list) {
  static char* chkpMeta = "META";
  int32_t      code = 0;

  TAOS_UNUSED(taosThreadRwlockRdlock(&p->rwLock));

  int32_t cap = p->len + 128;

  char* buffer = taosMemoryCalloc(4, cap);
  if (buffer == NULL) {
    code = terrno;
    goto _ERROR;
  }

  char* srcBuf = buffer;
  char* dstBuf = &srcBuf[cap];
  char* srcDir = &dstBuf[cap];
  char* dstDir = &srcDir[cap];

  int nBytes = snprintf(srcDir, cap, "%s%s%s%s%s%" PRId64, p->path, TD_DIRSEP, "checkpoints", TD_DIRSEP,
                        "checkpoint", p->curChkpId);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  nBytes = snprintf(dstDir, cap, "%s", dname);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  if (!taosDirExist(srcDir)) {
    stError("failed to dump srcDir %s, reason: not exist such dir", srcDir);
    code = TSDB_CODE_INVALID_PARA;
    goto _ERROR;
  }
  int64_t chkpId = 0, processId = -1;
  code = chkpLoadExtraInfo(srcDir, &chkpId, &processId);
  if (code < 0) {
    stError("failed to load extra info from %s, reason:%s", srcDir, code != 0 ? "unkown" : tstrerror(code));

    goto _ERROR;
  }

  // add file to $name dir
  for (int i = 0; i < taosArrayGetSize(p->pAdd); i++) {
    memset(srcBuf, 0, cap);
    memset(dstBuf, 0, cap);

    char* filename = taosArrayGetP(p->pAdd, i);
    nBytes = snprintf(srcBuf, cap, "%s%s%s", srcDir, TD_DIRSEP, filename);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _ERROR;
    }

    nBytes = snprintf(dstBuf, cap, "%s%s%s", dstDir, TD_DIRSEP, filename);
    if (nBytes <= 0 || nBytes >= cap) {
      code = TSDB_CODE_OUT_OF_RANGE;
      goto _ERROR;
    }

    if (taosCopyFile(srcBuf, dstBuf) < 0) {
      code = TAOS_SYSTEM_ERROR(ERRNO);
      stError("failed to copy file from %s to %s, reason:%s", srcBuf, dstBuf, tstrerror(code));
      goto _ERROR;
    }
  }
  // del file in $name
  for (int i = 0; i < taosArrayGetSize(p->pDel); i++) {
    char* filename = taosArrayGetP(p->pDel, i);
    char* p = taosStrdup(filename);
    if (p == NULL) {
      code = terrno;
      goto _ERROR;
    }
    if (taosArrayPush(list, &p) == NULL) {
      taosMemoryFree(p);
      code = terrno;
      goto _ERROR;
    }
  }

  // copy current file to dst dir
  memset(srcBuf, 0, cap);
  memset(dstBuf, 0, cap);

  nBytes = snprintf(srcBuf, cap, "%s%s%s", srcDir, TD_DIRSEP, p->pCurrent);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  nBytes = snprintf(dstBuf, cap, "%s%s%s_%" PRId64, dstDir, TD_DIRSEP, p->pCurrent, p->curChkpId);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  if (taosCopyFile(srcBuf, dstBuf) < 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
    stError("failed to copy file from %s to %s, reason:%s", srcBuf, dstBuf, tstrerror(code));
    goto _ERROR;
  }

  // copy manifest file to dst dir
  memset(srcBuf, 0, cap);
  memset(dstBuf, 0, cap);

  nBytes = snprintf(srcBuf, cap, "%s%s%s", srcDir, TD_DIRSEP, p->pManifest);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  nBytes = snprintf(dstBuf, cap, "%s%s%s_%" PRId64, dstDir, TD_DIRSEP, p->pManifest, p->curChkpId);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  if (taosCopyFile(srcBuf, dstBuf) < 0) {
    code = terrno;
    stError("failed to copy file from %s to %s, reason:%s", srcBuf, dstBuf, tstrerror(code));
    goto _ERROR;
  }
  memset(dstBuf, 0, cap);
  nBytes = snprintf(dstBuf, cap, "%s%s%s", dstDir, TD_DIRSEP, chkpMeta);
  if (nBytes <= 0 || nBytes >= cap) {
    code = TSDB_CODE_OUT_OF_RANGE;
    goto _ERROR;
  }

  TdFilePtr pFile = taosOpenFile(dstBuf, TD_FILE_CREATE | TD_FILE_WRITE | TD_FILE_TRUNC);
  if (pFile == NULL) {
    code = terrno;
    stError("chkp failed to create meta file: %s, reason:%s", dstDir, tstrerror(code));
    goto _ERROR;
  }

  char content[256] = {0};
  nBytes = tsnprintf(content, sizeof(content), META_ON_S3_FORMATE, p->pCurrent, p->curChkpId, p->pManifest,
                     p->curChkpId, "processVer", processId);
  if (nBytes <= 0 || nBytes >= sizeof(content)) {
    code = TSDB_CODE_OUT_OF_RANGE;
    stError("chkp failed to format meta file: %s, reason: invalid msg", dstDir);
    TAOS_UNUSED(taosCloseFile(&pFile));
    goto _ERROR;
  }

  nBytes = taosWriteFile(pFile, content, strlen(content));
  if (nBytes != strlen(content)) {
    code = terrno;
    stError("chkp failed to write meta file: %s,reason:%s", dstDir, tstrerror(code));
    TAOS_UNUSED(taosCloseFile(&pFile));
    goto _ERROR;
  }
  TAOS_UNUSED(taosCloseFile(&pFile));

  // clear delta data buf
  taosArrayClearP(p->pAdd, NULL);
  taosArrayClearP(p->pDel, NULL);
  code = 0;

_ERROR:
  taosMemoryFree(buffer);
  TAOS_UNUSED(taosThreadRwlockUnlock(&p->rwLock));
  return code;
}

int32_t bkdMgtCreate(char* path, SBkdMgt** mgt) {
  int32_t  code = 0;
  SBkdMgt* p = taosMemoryCalloc(1, sizeof(SBkdMgt));
  if (p == NULL) {
    return terrno;
  }

  p->pDbChkpTbl = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
  if (p->pDbChkpTbl == NULL) {
    code = terrno;
    bkdMgtDestroy(p);
    return code;
  }

  p->path = taosStrdup(path);
  if (p->path == NULL) {
    code = terrno;
    bkdMgtDestroy(p);
    return code;
  }

  if (taosThreadRwlockInit(&p->rwLock, NULL) != 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
    bkdMgtDestroy(p);
    return code;
  }
  *mgt = p;

  return code;
}

void bkdMgtDestroy(SBkdMgt* bm) {
  if (bm == NULL) return;
  void* pIter = taosHashIterate(bm->pDbChkpTbl, NULL);
  while (pIter) {
    SDbChkp* pChkp = *(SDbChkp**)(pIter);
    dbChkpDestroy(pChkp);

    pIter = taosHashIterate(bm->pDbChkpTbl, pIter);
  }

  TAOS_UNUSED(taosThreadRwlockDestroy(&bm->rwLock));
  taosMemoryFree(bm->path);
  taosHashCleanup(bm->pDbChkpTbl);

  taosMemoryFree(bm);
}
int32_t bkdMgtGetDelta(SBkdMgt* bm, char* taskId, int64_t chkpId, SArray* list, char* dname) {
  int32_t code = 0;
  TAOS_UNUSED(taosThreadRwlockWrlock(&bm->rwLock));
  SDbChkp** ppChkp = taosHashGet(bm->pDbChkpTbl, taskId, strlen(taskId));
  SDbChkp*  pChkp = ppChkp != NULL ? *ppChkp : NULL;

  if (pChkp == NULL) {
    int32_t cap = strlen(bm->path) + 64;
    char*   path = taosMemoryCalloc(1, cap);
    if (path == NULL) {
      TAOS_UNUSED(taosThreadRwlockUnlock(&bm->rwLock));
      return terrno;
    }

    int32_t nBytes = snprintf(path, cap, "%s%s%s", bm->path, TD_DIRSEP, taskId);
    if (nBytes <= 0 || nBytes >= cap) {
      taosMemoryFree(path);
      TAOS_UNUSED(taosThreadRwlockUnlock(&bm->rwLock));
      code = TSDB_CODE_OUT_OF_RANGE;
      return code;
    }

    SDbChkp* p = NULL;
    code = dbChkpCreate(path, chkpId, &p);
    if (code != 0) {
      taosMemoryFree(path);
      TAOS_UNUSED(taosThreadRwlockUnlock(&bm->rwLock));
      return code;
    }

    if (taosHashPut(bm->pDbChkpTbl, taskId, strlen(taskId), &p, sizeof(void*)) != 0) {
      dbChkpDestroy(p);
      TAOS_UNUSED(taosThreadRwlockUnlock(&bm->rwLock));
      code = terrno;
      return code;
    }

    pChkp = p;
    code = dbChkpDumpTo(pChkp, dname, list);
    TAOS_UNUSED(taosThreadRwlockUnlock(&bm->rwLock));
    return code;
  } else {
    code = dbChkpGetDelta(pChkp, chkpId, NULL);
    if (code == 0) {
      code = dbChkpDumpTo(pChkp, dname, list);
    }
  }

  TAOS_UNUSED(taosThreadRwlockUnlock(&bm->rwLock));
  return code;
}

#ifdef BUILD_NO_CALL
int32_t bkdMgtAddChkp(SBkdMgt* bm, char* task, char* path) {
  int32_t code = -1;

  taosThreadRwlockWrlock(&bm->rwLock);
  SDbChkp** pp = taosHashGet(bm->pDbChkpTbl, task, strlen(task));
  if (pp == NULL) {
    SDbChkp* p = NULL;
    code = dbChkpCreate(path, 0, &p);
    if (code != 0) {
      taosHashPut(bm->pDbChkpTbl, task, strlen(task), &p, sizeof(void*));
      code = 0;
    }
  } else {
    stError("task chkp already exists");
  }

  taosThreadRwlockUnlock(&bm->rwLock);

  return code;
}

int32_t bkdMgtDumpTo(SBkdMgt* bm, char* taskId, char* dname) {
  int32_t code = 0;
  taosThreadRwlockRdlock(&bm->rwLock);

  SDbChkp* p = taosHashGet(bm->pDbChkpTbl, taskId, strlen(taskId));
  code = dbChkpDumpTo(p, dname, NULL);

  taosThreadRwlockUnlock(&bm->rwLock);
  return code;
}
#endif

SStreamStateCur* streamStateSeekKeyPrev_rocksdb(SStreamState* pState, const SWinKey* key) {
  stDebug("streamStateSeekKeyPrev_rocksdb");
  STaskDbWrapper* wrapper = pState->pTdbState->pOwner->pBackend;
  if (pState->pTdbState->recalc) {
    wrapper = pState->pTdbState->pOwner->pRecalBackend;
  }

  SStreamStateCur* pCur = createStreamStateCursor();
  if (pCur == NULL) {
    return NULL;
  }

  pCur->db = wrapper->db;
  pCur->iter = streamStateIterCreate(pState, "state", (rocksdb_snapshot_t**)&pCur->snapshot,
                                     (rocksdb_readoptions_t**)&pCur->readOpt);
  pCur->number = pState->number;

  SStateKey sKey = {.key = *key, .opNum = pState->number};
  char      buf[128] = {0};
  int       len = stateKeyEncode((void*)&sKey, buf);
  if (!streamStateIterSeekAndValid(pCur->iter, buf, len)) {
    streamStateFreeCur(pCur);
    return NULL;
  }
  while (rocksdb_iter_valid(pCur->iter) && iterValueIsStale(pCur->iter)) {
    rocksdb_iter_prev(pCur->iter);
  }

  if (rocksdb_iter_valid(pCur->iter)) {
    SWinKey curKey;
    size_t  kLen = 0;
    char*   keyStr = (char*)rocksdb_iter_key(pCur->iter, &kLen);
    TAOS_UNUSED(winKeyDecode((void*)&curKey, keyStr));
    if (winKeyCmpr(&sKey, sizeof(sKey), &curKey, sizeof(curKey)) > 0) {
      return pCur;
    }
    rocksdb_iter_prev(pCur->iter);
    return pCur;
  }

  streamStateFreeCur(pCur);
  return NULL;
}

int32_t streamStateGetGroupKVByCur_rocksdb(SStreamState* pState, SStreamStateCur* pCur, SWinKey* pKey,
                                           const void** pVal, int32_t* pVLen) {
  if (!pCur) {
    return -1;
  }
  uint64_t groupId = pKey->groupId;

  int32_t code = streamStateGetKVByCur_rocksdb(pState, pCur, pKey, pVal, pVLen);
  if (code == 0) {
    if (pKey->groupId == groupId) {
      return 0;
    }
    if (pVal != NULL) {
      taosMemoryFree((void*)*pVal);
      *pVal = NULL;
    }
  }
  return -1;
}

taosdata / TDengine / #3768

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous