walRef.c 2.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
/*
 * 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 "cJSON.h"
#include "os.h"
#include "taoserror.h"
#include "tutil.h"
#include "walInt.h"

SWalRef *walOpenRef(SWal *pWal) {
  SWalRef *pRef = taosMemoryCalloc(1, sizeof(SWalRef));
  if (pRef == NULL) {
    return NULL;
  }
  pRef->refId = tGenIdPI64();
  pRef->refVer = -1;
  pRef->refFile = -1;
  pRef->pWal = pWal;
  taosHashPut(pWal->pRefHash, &pRef->refId, sizeof(int64_t), &pRef, sizeof(void *));
  return pRef;
}

void walCloseRef(SWal *pWal, int64_t refId) {
L
Liu Jicong 已提交
36 37 38
  SWalRef **ppRef = taosHashGet(pWal->pRefHash, &refId, sizeof(int64_t));
  if (ppRef == NULL) return;
  SWalRef *pRef = *ppRef;
L
Liu Jicong 已提交
39 40 41 42 43
  if (pRef) {
    wDebug("vgId:%d, wal close ref %" PRId64 ", refId %" PRId64, pWal->cfg.vgId, pRef->refVer, pRef->refId);
  } else {
    wDebug("vgId:%d, wal close ref null, refId %" PRId64, pWal->cfg.vgId, refId);
  }
44 45 46 47 48 49
  taosHashRemove(pWal->pRefHash, &refId, sizeof(int64_t));
  taosMemoryFree(pRef);
}

int32_t walRefVer(SWalRef *pRef, int64_t ver) {
  SWal *pWal = pRef->pWal;
L
Liu Jicong 已提交
50
  wDebug("vgId:%d, wal ref version %" PRId64 ", refId %" PRId64, pWal->cfg.vgId, ver, pRef->refId);
51 52 53 54 55 56 57 58 59 60 61 62 63
  if (pRef->refVer != ver) {
    taosThreadMutexLock(&pWal->mutex);
    if (ver < pWal->vers.firstVer || ver > pWal->vers.lastVer) {
      taosThreadMutexUnlock(&pWal->mutex);
      terrno = TSDB_CODE_WAL_INVALID_VER;
      return -1;
    }

    pRef->refVer = ver;
    // bsearch in fileSet
    SWalFileInfo tmpInfo;
    tmpInfo.firstVer = ver;
    SWalFileInfo *pRet = taosArraySearch(pWal->fileInfoSet, &tmpInfo, compareWalFileInfo, TD_LE);
64
    ASSERT(pRet != NULL);
65 66 67 68 69 70 71 72
    pRef->refFile = pRet->firstVer;

    taosThreadMutexUnlock(&pWal->mutex);
  }

  return 0;
}

L
Liu Jicong 已提交
73
#if 1
74 75 76 77
void walUnrefVer(SWalRef *pRef) {
  pRef->refId = -1;
  pRef->refFile = -1;
}
L
Liu Jicong 已提交
78
#endif
79 80 81 82 83 84 85 86 87 88 89 90 91 92 93

SWalRef *walRefCommittedVer(SWal *pWal) {
  SWalRef *pRef = walOpenRef(pWal);
  if (pRef == NULL) {
    return NULL;
  }
  taosThreadMutexLock(&pWal->mutex);

  int64_t ver = walGetCommittedVer(pWal);

  pRef->refVer = ver;
  // bsearch in fileSet
  SWalFileInfo tmpInfo;
  tmpInfo.firstVer = ver;
  SWalFileInfo *pRet = taosArraySearch(pWal->fileInfoSet, &tmpInfo, compareWalFileInfo, TD_LE);
94
  ASSERT(pRet != NULL);
95 96 97 98 99
  pRef->refFile = pRet->firstVer;

  taosThreadMutexUnlock(&pWal->mutex);
  return pRef;
}