mndDb.c 42.2 KB
Newer Older
H
refact  
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

S
Shengliang Guan 已提交
16
#define _DEFAULT_SOURCE
S
Shengliang Guan 已提交
17
#include "mndDb.h"
S
Shengliang Guan 已提交
18
#include "mndDnode.h"
S
Shengliang Guan 已提交
19 20
#include "mndShow.h"
#include "mndTrans.h"
S
Shengliang Guan 已提交
21
#include "mndUser.h"
22
#include "mndVgroup.h"
S
Shengliang Guan 已提交
23

S
Shengliang Guan 已提交
24
#define TSDB_DB_VER_NUMBER 1
S
Shengliang Guan 已提交
25
#define TSDB_DB_RESERVE_SIZE 64
S
Shengliang Guan 已提交
26 27 28 29 30

static SSdbRaw *mndDbActionEncode(SDbObj *pDb);
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw);
static int32_t  mndDbActionInsert(SSdb *pSdb, SDbObj *pDb);
static int32_t  mndDbActionDelete(SSdb *pSdb, SDbObj *pDb);
S
Shengliang Guan 已提交
31
static int32_t  mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew);
S
Shengliang Guan 已提交
32 33 34 35 36 37
static int32_t  mndProcessCreateDbReq(SMnodeMsg *pReq);
static int32_t  mndProcessAlterDbReq(SMnodeMsg *pReq);
static int32_t  mndProcessDropDbReq(SMnodeMsg *pReq);
static int32_t  mndProcessUseDbReq(SMnodeMsg *pReq);
static int32_t  mndProcessSyncDbReq(SMnodeMsg *pReq);
static int32_t  mndProcessCompactDbReq(SMnodeMsg *pReq);
S
Shengliang Guan 已提交
38
static int32_t  mndGetDbMeta(SMnodeMsg *pReq, SShowObj *pShow, STableMetaRsp *pMeta);
S
Shengliang Guan 已提交
39
static int32_t  mndRetrieveDbs(SMnodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows);
S
Shengliang Guan 已提交
40
static void     mndCancelGetNextDb(SMnode *pMnode, void *pIter);
S
Shengliang Guan 已提交
41

S
Shengliang Guan 已提交
42
int32_t mndInitDb(SMnode *pMnode) {
S
Shengliang Guan 已提交
43
  SSdbTable table = {.sdbType = SDB_DB,
S
Shengliang Guan 已提交
44 45 46 47 48 49 50
                     .keyType = SDB_KEY_BINARY,
                     .encodeFp = (SdbEncodeFp)mndDbActionEncode,
                     .decodeFp = (SdbDecodeFp)mndDbActionDecode,
                     .insertFp = (SdbInsertFp)mndDbActionInsert,
                     .updateFp = (SdbUpdateFp)mndDbActionUpdate,
                     .deleteFp = (SdbDeleteFp)mndDbActionDelete};

S
Shengliang Guan 已提交
51 52 53 54 55 56
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_DB, mndProcessCreateDbReq);
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
  mndSetMsgHandle(pMnode, TDMT_MND_SYNC_DB, mndProcessSyncDbReq);
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
S
Shengliang Guan 已提交
57 58 59 60 61

  mndAddShowMetaHandle(pMnode, TSDB_MGMT_TABLE_DB, mndGetDbMeta);
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);

S
Shengliang Guan 已提交
62
  return sdbSetTable(pMnode->pSdb, table);
S
Shengliang Guan 已提交
63
}
S
Shengliang Guan 已提交
64

S
Shengliang Guan 已提交
65
void mndCleanupDb(SMnode *pMnode) {}
S
Shengliang Guan 已提交
66

S
Shengliang Guan 已提交
67
static SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
68 69
  terrno = TSDB_CODE_OUT_OF_MEMORY;

S
Shengliang Guan 已提交
70
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, TSDB_DB_VER_NUMBER, sizeof(SDbObj) + TSDB_DB_RESERVE_SIZE);
71
  if (pRaw == NULL) goto DB_ENCODE_OVER;
S
Shengliang Guan 已提交
72 73

  int32_t dataPos = 0;
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
  SDB_SET_BINARY(pRaw, dataPos, pDb->name, TSDB_DB_FNAME_LEN, DB_ENCODE_OVER)
  SDB_SET_BINARY(pRaw, dataPos, pDb->acct, TSDB_USER_LEN, DB_ENCODE_OVER)
  SDB_SET_INT64(pRaw, dataPos, pDb->createdTime, DB_ENCODE_OVER)
  SDB_SET_INT64(pRaw, dataPos, pDb->updateTime, DB_ENCODE_OVER)
  SDB_SET_INT64(pRaw, dataPos, pDb->uid, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfgVersion, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->vgVersion, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->hashMethod, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.numOfVgroups, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.cacheBlockSize, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.totalBlocks, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysPerFile, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysToKeep0, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysToKeep1, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysToKeep2, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.minRows, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.maxRows, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.commitTime, DB_ENCODE_OVER)
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.fsyncPeriod, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.walLevel, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.precision, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.compression, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.replications, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.quorum, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.update, DB_ENCODE_OVER)
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.cacheLastRow, DB_ENCODE_OVER)
  SDB_SET_RESERVE(pRaw, dataPos, TSDB_DB_RESERVE_SIZE, DB_ENCODE_OVER)
  SDB_SET_DATALEN(pRaw, dataPos, DB_ENCODE_OVER)

  terrno = 0;

DB_ENCODE_OVER:
  if (terrno != 0) {
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
    sdbFreeRaw(pRaw);
    return NULL;
  }
S
Shengliang Guan 已提交
111

112
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
S
Shengliang Guan 已提交
113 114 115 116
  return pRaw;
}

static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
117 118
  terrno = TSDB_CODE_OUT_OF_MEMORY;

S
Shengliang Guan 已提交
119
  int8_t sver = 0;
120
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto DB_DECODE_OVER;
S
Shengliang Guan 已提交
121

S
Shengliang Guan 已提交
122
  if (sver != TSDB_DB_VER_NUMBER) {
S
Shengliang Guan 已提交
123
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
124
    goto DB_DECODE_OVER;
S
Shengliang Guan 已提交
125 126 127
  }

  SSdbRow *pRow = sdbAllocRow(sizeof(SDbObj));
128 129 130 131
  if (pRow == NULL) goto DB_DECODE_OVER;

  SDbObj *pDb = sdbGetRowObj(pRow);
  if (pDb == NULL) goto DB_DECODE_OVER;
S
Shengliang Guan 已提交
132 133

  int32_t dataPos = 0;
134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
  SDB_GET_BINARY(pRaw, dataPos, pDb->name, TSDB_DB_FNAME_LEN, DB_DECODE_OVER)
  SDB_GET_BINARY(pRaw, dataPos, pDb->acct, TSDB_USER_LEN, DB_DECODE_OVER)
  SDB_GET_INT64(pRaw, dataPos, &pDb->createdTime, DB_DECODE_OVER)
  SDB_GET_INT64(pRaw, dataPos, &pDb->updateTime, DB_DECODE_OVER)
  SDB_GET_INT64(pRaw, dataPos, &pDb->uid, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfgVersion, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->vgVersion, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->hashMethod, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfVgroups, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.cacheBlockSize, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.totalBlocks, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysPerFile, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep0, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep1, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep2, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.minRows, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.maxRows, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.commitTime, DB_DECODE_OVER)
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.fsyncPeriod, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.walLevel, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.precision, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.compression, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.replications, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.quorum, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.update, DB_DECODE_OVER)
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.cacheLastRow, DB_DECODE_OVER)
  SDB_GET_RESERVE(pRaw, dataPos, TSDB_DB_RESERVE_SIZE, DB_DECODE_OVER)

  terrno = 0;

DB_DECODE_OVER:
  if (terrno != 0) {
    mError("db:%s, failed to decode from raw:%p since %s", pDb->name, pRaw, terrstr());
    tfree(pRow);
    return NULL;
  }
S
Shengliang Guan 已提交
170

171
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
S
Shengliang Guan 已提交
172 173 174 175
  return pRow;
}

static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
176
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
S
Shengliang Guan 已提交
177 178 179 180
  return 0;
}

static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
181
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
S
Shengliang Guan 已提交
182 183 184
  return 0;
}

S
Shengliang Guan 已提交
185 186 187 188 189 190
static int32_t mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew) {
  mTrace("db:%s, perform update action, old row:%p new row:%p", pOld->name, pOld, pNew);
  pOld->updateTime = pNew->updateTime;
  pOld->cfgVersion = pNew->cfgVersion;
  pOld->vgVersion = pNew->vgVersion;
  memcpy(&pOld->cfg, &pNew->cfg, sizeof(SDbCfg));
S
Shengliang Guan 已提交
191 192 193
  return 0;
}

S
Shengliang Guan 已提交
194
SDbObj *mndAcquireDb(SMnode *pMnode, char *db) {
S
Shengliang Guan 已提交
195 196
  SSdb   *pSdb = pMnode->pSdb;
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
S
Shengliang Guan 已提交
197
  if (pDb == NULL && terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
S
Shengliang Guan 已提交
198 199 200
    terrno = TSDB_CODE_MND_DB_NOT_EXIST;
  }
  return pDb;
S
Shengliang Guan 已提交
201
}
S
Shengliang Guan 已提交
202

S
Shengliang Guan 已提交
203 204 205 206 207
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
  SSdb *pSdb = pMnode->pSdb;
  sdbRelease(pSdb, pDb);
}

S
Shengliang Guan 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
static int32_t mndCheckDbName(char *dbName, SUserObj *pUser) {
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
  if (pos == NULL) {
    terrno = TSDB_CODE_MND_INVALID_DB;
    return -1;
  }

  int32_t acctId = atoi(dbName);
  if (acctId != pUser->acctId) {
    terrno = TSDB_CODE_MND_INVALID_DB_ACCT;
    return -1;
  }

  return 0;
}

S
Shengliang Guan 已提交
224
static int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
225
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return -1;
S
Shengliang Guan 已提交
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248
  if (pCfg->cacheBlockSize < TSDB_MIN_CACHE_BLOCK_SIZE || pCfg->cacheBlockSize > TSDB_MAX_CACHE_BLOCK_SIZE) return -1;
  if (pCfg->totalBlocks < TSDB_MIN_TOTAL_BLOCKS || pCfg->totalBlocks > TSDB_MAX_TOTAL_BLOCKS) return -1;
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return -1;
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return -1;
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return -1;
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return -1;
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return -1;
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return -1;
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return -1;
  if (pCfg->minRows < TSDB_MIN_MIN_ROW_FBLOCK || pCfg->minRows > TSDB_MAX_MIN_ROW_FBLOCK) return -1;
  if (pCfg->maxRows < TSDB_MIN_MAX_ROW_FBLOCK || pCfg->maxRows > TSDB_MAX_MAX_ROW_FBLOCK) return -1;
  if (pCfg->minRows > pCfg->maxRows) return -1;
  if (pCfg->commitTime < TSDB_MIN_COMMIT_TIME || pCfg->commitTime > TSDB_MAX_COMMIT_TIME) return -1;
  if (pCfg->fsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->fsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return -1;
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return -1;
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return -1;
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return -1;
  if (pCfg->replications < TSDB_MIN_DB_REPLICA_OPTION || pCfg->replications > TSDB_MAX_DB_REPLICA_OPTION) return -1;
  if (pCfg->replications > mndGetDnodeSize(pMnode)) return -1;
  if (pCfg->quorum < TSDB_MIN_DB_QUORUM_OPTION || pCfg->quorum > TSDB_MAX_DB_QUORUM_OPTION) return -1;
  if (pCfg->quorum > pCfg->replications) return -1;
  if (pCfg->update < TSDB_MIN_DB_UPDATE || pCfg->update > TSDB_MAX_DB_UPDATE) return -1;
  if (pCfg->cacheLastRow < TSDB_MIN_DB_CACHE_LAST_ROW || pCfg->cacheLastRow > TSDB_MAX_DB_CACHE_LAST_ROW) return -1;
S
Shengliang Guan 已提交
249 250 251 252
  return TSDB_CODE_SUCCESS;
}

static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
253
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
S
Shengliang Guan 已提交
254 255 256 257 258 259
  if (pCfg->cacheBlockSize < 0) pCfg->cacheBlockSize = TSDB_DEFAULT_CACHE_BLOCK_SIZE;
  if (pCfg->totalBlocks < 0) pCfg->totalBlocks = TSDB_DEFAULT_TOTAL_BLOCKS;
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DAYS_PER_FILE;
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = TSDB_DEFAULT_KEEP;
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = TSDB_DEFAULT_KEEP;
S
Shengliang Guan 已提交
260 261
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MIN_ROW_FBLOCK;
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAX_ROW_FBLOCK;
S
Shengliang Guan 已提交
262 263 264 265 266 267 268 269 270 271
  if (pCfg->commitTime < 0) pCfg->commitTime = TSDB_DEFAULT_COMMIT_TIME;
  if (pCfg->fsyncPeriod < 0) pCfg->fsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA_OPTION;
  if (pCfg->quorum < 0) pCfg->quorum = TSDB_DEFAULT_DB_QUORUM_OPTION;
  if (pCfg->update < 0) pCfg->update = TSDB_DEFAULT_DB_UPDATE_OPTION;
  if (pCfg->cacheLastRow < 0) pCfg->cacheLastRow = TSDB_DEFAULT_CACHE_LAST_ROW;
}
S
Shengliang Guan 已提交
272

S
Shengliang Guan 已提交
273
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
S
Shengliang Guan 已提交
274
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
S
Shengliang Guan 已提交
275 276 277
  if (pDbRaw == NULL) return -1;
  if (mndTransAppendRedolog(pTrans, pDbRaw) != 0) return -1;
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
S
Shengliang Guan 已提交
278

S
Shengliang Guan 已提交
279
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
S
Shengliang Guan 已提交
280
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
S
Shengliang Guan 已提交
281 282 283
    if (pVgRaw == NULL) return -1;
    if (mndTransAppendRedolog(pTrans, pVgRaw) != 0) return -1;
    if (sdbSetRawStatus(pVgRaw, SDB_STATUS_CREATING) != 0) return -1;
S
Shengliang Guan 已提交
284 285 286 287 288
  }

  return 0;
}

S
Shengliang Guan 已提交
289
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
S
Shengliang Guan 已提交
290
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
S
Shengliang Guan 已提交
291 292 293
  if (pDbRaw == NULL) return -1;
  if (mndTransAppendUndolog(pTrans, pDbRaw) != 0) return -1;
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED) != 0) return -1;
S
Shengliang Guan 已提交
294

S
Shengliang Guan 已提交
295
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
S
Shengliang Guan 已提交
296
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
S
Shengliang Guan 已提交
297 298 299
    if (pVgRaw == NULL) return -1;
    if (mndTransAppendUndolog(pTrans, pVgRaw) != 0) return -1;
    if (sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED) != 0) return -1;
S
Shengliang Guan 已提交
300 301 302 303 304
  }

  return 0;
}

S
Shengliang Guan 已提交
305
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
S
Shengliang Guan 已提交
306
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
S
Shengliang Guan 已提交
307 308 309
  if (pDbRaw == NULL) return -1;
  if (mndTransAppendCommitlog(pTrans, pDbRaw) != 0) return -1;
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_READY) != 0) return -1;
S
Shengliang Guan 已提交
310

S
Shengliang Guan 已提交
311
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
S
Shengliang Guan 已提交
312
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
S
Shengliang Guan 已提交
313 314 315
    if (pVgRaw == NULL) return -1;
    if (mndTransAppendCommitlog(pTrans, pVgRaw) != 0) return -1;
    if (sdbSetRawStatus(pVgRaw, SDB_STATUS_READY) != 0) return -1;
S
Shengliang Guan 已提交
316 317 318 319 320
  }

  return 0;
}

S
Shengliang Guan 已提交
321 322
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
S
Shengliang Guan 已提交
323
    SVgObj *pVgroup = pVgroups + vg;
S
Shengliang Guan 已提交
324 325

    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
S
Shengliang Guan 已提交
326 327
      STransAction action = {0};
      SVnodeGid   *pVgid = pVgroup->vnodeGid + vn;
S
Shengliang Guan 已提交
328

S
Shengliang Guan 已提交
329 330 331
      SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
      if (pDnode == NULL) return -1;
      action.epSet = mndGetDnodeEpset(pDnode);
S
Shengliang Guan 已提交
332 333
      mndReleaseDnode(pMnode, pDnode);

S
Shengliang Guan 已提交
334 335
      SCreateVnodeReq *pReq = mndBuildCreateVnodeReq(pMnode, pDnode, pDb, pVgroup);
      if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
336

S
Shengliang Guan 已提交
337
      action.pCont = pReq;
S
Shengliang Guan 已提交
338
      action.contLen = sizeof(SCreateVnodeReq);
H
Hongze Cheng 已提交
339
      action.msgType = TDMT_DND_CREATE_VNODE;
S
Shengliang Guan 已提交
340
      action.acceptableCode = TSDB_CODE_DND_VNODE_ALREADY_DEPLOYED;
S
Shengliang Guan 已提交
341
      if (mndTransAppendRedoAction(pTrans, &action) != 0) {
S
Shengliang Guan 已提交
342
        free(pReq);
S
Shengliang Guan 已提交
343 344 345 346 347
        return -1;
      }
    }
  }

S
Shengliang Guan 已提交
348 349 350
  return 0;
}

S
Shengliang Guan 已提交
351 352
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
S
Shengliang Guan 已提交
353
    SVgObj *pVgroup = pVgroups + vg;
S
Shengliang Guan 已提交
354 355

    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
S
Shengliang Guan 已提交
356 357
      STransAction action = {0};
      SVnodeGid   *pVgid = pVgroup->vnodeGid + vn;
S
Shengliang Guan 已提交
358

S
Shengliang Guan 已提交
359 360 361
      SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
      if (pDnode == NULL) return -1;
      action.epSet = mndGetDnodeEpset(pDnode);
S
Shengliang Guan 已提交
362 363
      mndReleaseDnode(pMnode, pDnode);

S
Shengliang Guan 已提交
364 365
      SDropVnodeReq *pReq = mndBuildDropVnodeReq(pMnode, pDnode, pDb, pVgroup);
      if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
366

S
Shengliang Guan 已提交
367
      action.pCont = pReq;
S
Shengliang Guan 已提交
368
      action.contLen = sizeof(SDropVnodeReq);
H
Hongze Cheng 已提交
369
      action.msgType = TDMT_DND_DROP_VNODE;
S
Shengliang Guan 已提交
370
       action.acceptableCode = TSDB_CODE_DND_VNODE_NOT_DEPLOYED;
S
Shengliang Guan 已提交
371
      if (mndTransAppendUndoAction(pTrans, &action) != 0) {
S
Shengliang Guan 已提交
372
        free(pReq);
S
Shengliang Guan 已提交
373 374 375 376 377
        return -1;
      }
    }
  }

S
Shengliang Guan 已提交
378 379 380
  return 0;
}

S
Shengliang Guan 已提交
381
static int32_t mndCreateDb(SMnode *pMnode, SMnodeMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser) {
S
Shengliang Guan 已提交
382
  SDbObj dbObj = {0};
383
  memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
S
Shengliang Guan 已提交
384
  memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
S
Shengliang Guan 已提交
385 386
  dbObj.createdTime = taosGetTimestampMs();
  dbObj.updateTime = dbObj.createdTime;
387
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
S
Shengliang Guan 已提交
388 389
  dbObj.cfgVersion = 1;
  dbObj.vgVersion = 1;
S
Shengliang Guan 已提交
390
  dbObj.hashMethod = 1;
391 392
  dbObj.cfg = (SDbCfg){.numOfVgroups = pCreate->numOfVgroups,
                       .cacheBlockSize = pCreate->cacheBlockSize,
S
Shengliang Guan 已提交
393 394 395 396 397
                       .totalBlocks = pCreate->totalBlocks,
                       .daysPerFile = pCreate->daysPerFile,
                       .daysToKeep0 = pCreate->daysToKeep0,
                       .daysToKeep1 = pCreate->daysToKeep1,
                       .daysToKeep2 = pCreate->daysToKeep2,
S
Shengliang Guan 已提交
398 399
                       .minRows = pCreate->minRows,
                       .maxRows = pCreate->maxRows,
S
Shengliang Guan 已提交
400 401 402 403 404 405 406 407 408 409 410 411
                       .fsyncPeriod = pCreate->fsyncPeriod,
                       .commitTime = pCreate->commitTime,
                       .precision = pCreate->precision,
                       .compression = pCreate->compression,
                       .walLevel = pCreate->walLevel,
                       .replications = pCreate->replications,
                       .quorum = pCreate->quorum,
                       .update = pCreate->update,
                       .cacheLastRow = pCreate->cacheLastRow};

  mndSetDefaultDbCfg(&dbObj.cfg);

S
Shengliang Guan 已提交
412 413 414 415 416
  if (mndCheckDbName(dbObj.name, pUser) != 0) {
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }

S
Shengliang Guan 已提交
417 418
  if (mndCheckDbCfg(pMnode, &dbObj.cfg) != 0) {
    terrno = TSDB_CODE_MND_INVALID_DB_OPTION;
S
Shengliang Guan 已提交
419 420 421
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }
S
Shengliang Guan 已提交
422

S
Shengliang Guan 已提交
423 424
  SVgObj *pVgroups = NULL;
  if (mndAllocVgroup(pMnode, &dbObj, &pVgroups) != 0) {
425 426 427 428
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }

S
Shengliang Guan 已提交
429
  int32_t code = -1;
S
Shengliang Guan 已提交
430 431
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, &pReq->rpcMsg);
  if (pTrans == NULL) goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
432

S
Shengliang Guan 已提交
433 434
  mDebug("trans:%d, used to create db:%s", pTrans->id, pCreate->db);

S
Shengliang Guan 已提交
435 436 437 438 439 440
  if (mndSetCreateDbRedoLogs(pMnode, pTrans, &dbObj, pVgroups) != 0) goto CREATE_DB_OVER;
  if (mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups) != 0) goto CREATE_DB_OVER;
  if (mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups) != 0) goto CREATE_DB_OVER;
  if (mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups) != 0) goto CREATE_DB_OVER;
  if (mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups) != 0) goto CREATE_DB_OVER;
  if (mndTransPrepare(pMnode, pTrans) != 0) goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
441

S
Shengliang Guan 已提交
442 443 444 445
  code = 0;

CREATE_DB_OVER:
  free(pVgroups);
S
Shengliang Guan 已提交
446
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
447
  return code;
S
Shengliang Guan 已提交
448 449
}

S
Shengliang Guan 已提交
450 451
static int32_t mndProcessCreateDbReq(SMnodeMsg *pReq) {
  SMnode       *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
452
  SCreateDbReq *pCreate = pReq->rpcMsg.pCont;
S
Shengliang Guan 已提交
453

454
  pCreate->numOfVgroups = htonl(pCreate->numOfVgroups);
S
Shengliang Guan 已提交
455 456 457 458 459 460
  pCreate->cacheBlockSize = htonl(pCreate->cacheBlockSize);
  pCreate->totalBlocks = htonl(pCreate->totalBlocks);
  pCreate->daysPerFile = htonl(pCreate->daysPerFile);
  pCreate->daysToKeep0 = htonl(pCreate->daysToKeep0);
  pCreate->daysToKeep1 = htonl(pCreate->daysToKeep1);
  pCreate->daysToKeep2 = htonl(pCreate->daysToKeep2);
S
Shengliang Guan 已提交
461 462
  pCreate->minRows = htonl(pCreate->minRows);
  pCreate->maxRows = htonl(pCreate->maxRows);
S
Shengliang Guan 已提交
463 464 465
  pCreate->commitTime = htonl(pCreate->commitTime);
  pCreate->fsyncPeriod = htonl(pCreate->fsyncPeriod);

S
Shengliang Guan 已提交
466
  mDebug("db:%s, start to create, vgroups:%d", pCreate->db, pCreate->numOfVgroups);
S
Shengliang Guan 已提交
467 468 469

  SDbObj *pDb = mndAcquireDb(pMnode, pCreate->db);
  if (pDb != NULL) {
S
Shengliang Guan 已提交
470
    mndReleaseDb(pMnode, pDb);
S
Shengliang Guan 已提交
471 472 473 474 475 476 477 478
    if (pCreate->ignoreExist) {
      mDebug("db:%s, already exist, ignore exist is set", pCreate->db);
      return 0;
    } else {
      terrno = TSDB_CODE_MND_DB_ALREADY_EXIST;
      mError("db:%s, failed to create since %s", pCreate->db, terrstr());
      return -1;
    }
S
Shengliang Guan 已提交
479 480 481
  } else if (terrno != TSDB_CODE_MND_DB_NOT_EXIST) {
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
S
Shengliang Guan 已提交
482 483
  }

S
Shengliang Guan 已提交
484
  SUserObj *pOperUser = mndAcquireUser(pMnode, pReq->user);
S
Shengliang Guan 已提交
485 486 487 488 489
  if (pOperUser == NULL) {
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }

S
Shengliang Guan 已提交
490
  int32_t code = mndCreateDb(pMnode, pReq, pCreate, pOperUser);
S
Shengliang Guan 已提交
491 492 493 494 495 496 497 498 499 500
  mndReleaseUser(pMnode, pOperUser);

  if (code != 0) {
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }

  return TSDB_CODE_MND_ACTION_IN_PROGRESS;
}

S
Shengliang Guan 已提交
501
static int32_t mndSetDbCfgFromAlterDbMsg(SDbObj *pDb, SAlterDbReq *pAlter) {
S
Shengliang Guan 已提交
502
  terrno = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
S
Shengliang Guan 已提交
503 504 505

  if (pAlter->totalBlocks >= 0 && pAlter->totalBlocks != pDb->cfg.totalBlocks) {
    pDb->cfg.totalBlocks = pAlter->totalBlocks;
S
Shengliang Guan 已提交
506
    terrno = 0;
S
Shengliang Guan 已提交
507 508 509 510
  }

  if (pAlter->daysToKeep0 >= 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
S
Shengliang Guan 已提交
511
    terrno = 0;
S
Shengliang Guan 已提交
512 513 514 515
  }

  if (pAlter->daysToKeep1 >= 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
S
Shengliang Guan 已提交
516
    terrno = 0;
S
Shengliang Guan 已提交
517 518 519 520
  }

  if (pAlter->daysToKeep2 >= 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
S
Shengliang Guan 已提交
521
    terrno = 0;
S
Shengliang Guan 已提交
522 523 524 525
  }

  if (pAlter->fsyncPeriod >= 0 && pAlter->fsyncPeriod != pDb->cfg.fsyncPeriod) {
    pDb->cfg.fsyncPeriod = pAlter->fsyncPeriod;
S
Shengliang Guan 已提交
526
    terrno = 0;
S
Shengliang Guan 已提交
527 528 529 530
  }

  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
    pDb->cfg.walLevel = pAlter->walLevel;
S
Shengliang Guan 已提交
531
    terrno = 0;
S
Shengliang Guan 已提交
532 533 534 535
  }

  if (pAlter->quorum >= 0 && pAlter->quorum != pDb->cfg.quorum) {
    pDb->cfg.quorum = pAlter->quorum;
S
Shengliang Guan 已提交
536
    terrno = 0;
S
Shengliang Guan 已提交
537 538 539 540
  }

  if (pAlter->cacheLastRow >= 0 && pAlter->cacheLastRow != pDb->cfg.cacheLastRow) {
    pDb->cfg.cacheLastRow = pAlter->cacheLastRow;
S
Shengliang Guan 已提交
541
    terrno = 0;
S
Shengliang Guan 已提交
542 543
  }

S
Shengliang Guan 已提交
544 545 546
  return terrno;
}

S
Shengliang Guan 已提交
547 548
static int32_t mndSetUpdateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
S
Shengliang Guan 已提交
549 550 551 552 553 554 555
  if (pRedoRaw == NULL) return -1;
  if (mndTransAppendRedolog(pTrans, pRedoRaw) != 0) return -1;
  if (sdbSetRawStatus(pRedoRaw, SDB_STATUS_UPDATING) != 0) return -1;

  return 0;
}

S
Shengliang Guan 已提交
556 557
static int32_t mndSetUpdateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
S
Shengliang Guan 已提交
558 559 560
  if (pCommitRaw == NULL) return -1;
  if (mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) return -1;
  if (sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY) != 0) return -1;
S
Shengliang Guan 已提交
561 562 563 564

  return 0;
}

S
Shengliang Guan 已提交
565 566 567 568 569 570 571 572 573 574
static int32_t mndBuildUpdateVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
    STransAction action = {0};
    SVnodeGid   *pVgid = pVgroup->vnodeGid + vn;

    SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
    if (pDnode == NULL) return -1;
    action.epSet = mndGetDnodeEpset(pDnode);
    mndReleaseDnode(pMnode, pDnode);

S
Shengliang Guan 已提交
575 576
    SAlterVnodeReq *pReq = (SAlterVnodeReq *)mndBuildCreateVnodeReq(pMnode, pDnode, pDb, pVgroup);
    if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
577

S
Shengliang Guan 已提交
578
    action.pCont = pReq;
S
Shengliang Guan 已提交
579
    action.contLen = sizeof(SAlterVnodeReq);
H
Hongze Cheng 已提交
580
    action.msgType = TDMT_DND_ALTER_VNODE;
S
Shengliang Guan 已提交
581
    if (mndTransAppendRedoAction(pTrans, &action) != 0) {
S
Shengliang Guan 已提交
582
      free(pReq);
S
Shengliang Guan 已提交
583 584 585
      return -1;
    }
  }
S
Shengliang Guan 已提交
586 587 588 589

  return 0;
}

S
Shengliang Guan 已提交
590
static int32_t mndSetUpdateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
S
Shengliang Guan 已提交
591 592
  SSdb *pSdb = pMnode->pSdb;
  void *pIter = NULL;
S
Shengliang Guan 已提交
593

S
Shengliang Guan 已提交
594 595 596 597
  while (1) {
    SVgObj *pVgroup = NULL;
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
    if (pIter == NULL) break;
S
Shengliang Guan 已提交
598

S
Shengliang Guan 已提交
599 600
    if (pVgroup->dbUid == pNew->uid) {
      if (mndBuildUpdateVgroupAction(pMnode, pTrans, pNew, pVgroup) != 0) {
S
Shengliang Guan 已提交
601 602 603 604 605 606 607 608 609 610 611
        sdbCancelFetch(pSdb, pIter);
        sdbRelease(pSdb, pVgroup);
        return -1;
      }
    }

    sdbRelease(pSdb, pVgroup);
  }

  return 0;
}
S
Shengliang Guan 已提交
612

S
Shengliang Guan 已提交
613
static int32_t mndUpdateDb(SMnode *pMnode, SMnodeMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
S
Shengliang Guan 已提交
614
  int32_t code = -1;
S
Shengliang Guan 已提交
615
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, &pReq->rpcMsg);
S
Shengliang Guan 已提交
616
  if (pTrans == NULL) goto UPDATE_DB_OVER;
S
Shengliang Guan 已提交
617

S
Shengliang Guan 已提交
618
  mDebug("trans:%d, used to update db:%s", pTrans->id, pOld->name);
S
Shengliang Guan 已提交
619

S
Shengliang Guan 已提交
620 621 622 623
  if (mndSetUpdateDbRedoLogs(pMnode, pTrans, pOld, pNew) != 0) goto UPDATE_DB_OVER;
  if (mndSetUpdateDbCommitLogs(pMnode, pTrans, pOld, pNew) != 0) goto UPDATE_DB_OVER;
  if (mndSetUpdateDbRedoActions(pMnode, pTrans, pOld, pNew) != 0) goto UPDATE_DB_OVER;
  if (mndTransPrepare(pMnode, pTrans) != 0) goto UPDATE_DB_OVER;
S
Shengliang Guan 已提交
624

S
Shengliang Guan 已提交
625 626 627
  code = 0;

UPDATE_DB_OVER:
S
Shengliang Guan 已提交
628
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
629
  return code;
S
Shengliang Guan 已提交
630 631
}

S
Shengliang Guan 已提交
632 633
static int32_t mndProcessAlterDbReq(SMnodeMsg *pReq) {
  SMnode      *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
634
  SAlterDbReq *pAlter = pReq->rpcMsg.pCont;
S
Shengliang Guan 已提交
635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651
  pAlter->totalBlocks = htonl(pAlter->totalBlocks);
  pAlter->daysToKeep0 = htonl(pAlter->daysToKeep0);
  pAlter->daysToKeep1 = htonl(pAlter->daysToKeep1);
  pAlter->daysToKeep2 = htonl(pAlter->daysToKeep2);
  pAlter->fsyncPeriod = htonl(pAlter->fsyncPeriod);

  mDebug("db:%s, start to alter", pAlter->db);

  SDbObj *pDb = mndAcquireDb(pMnode, pAlter->db);
  if (pDb == NULL) {
    mError("db:%s, failed to alter since %s", pAlter->db, terrstr());
    return TSDB_CODE_MND_DB_NOT_EXIST;
  }

  SDbObj dbObj = {0};
  memcpy(&dbObj, pDb, sizeof(SDbObj));

S
Shengliang Guan 已提交
652
  int32_t code = mndSetDbCfgFromAlterDbMsg(&dbObj, pAlter);
S
Shengliang Guan 已提交
653 654 655 656 657 658
  if (code != 0) {
    mndReleaseDb(pMnode, pDb);
    mError("db:%s, failed to alter since %s", pAlter->db, tstrerror(code));
    return code;
  }

S
Shengliang Guan 已提交
659 660
  dbObj.cfgVersion++;
  dbObj.updateTime = taosGetTimestampMs();
S
Shengliang Guan 已提交
661
  code = mndUpdateDb(pMnode, pReq, pDb, &dbObj);
S
Shengliang Guan 已提交
662 663 664 665 666 667 668 669 670 671
  mndReleaseDb(pMnode, pDb);

  if (code != 0) {
    mError("db:%s, failed to alter since %s", pAlter->db, tstrerror(code));
    return code;
  }

  return TSDB_CODE_MND_ACTION_IN_PROGRESS;
}

S
Shengliang Guan 已提交
672 673 674 675 676 677 678 679 680 681 682 683 684 685 686
static int32_t mndSetDropDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
  if (pRedoRaw == NULL) return -1;
  if (mndTransAppendRedolog(pTrans, pRedoRaw) != 0) return -1;
  if (sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING) != 0) return -1;

  return 0;
}

static int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
  if (pCommitRaw == NULL) return -1;
  if (mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) return -1;
  if (sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED) != 0) return -1;

S
Shengliang Guan 已提交
687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713
  SSdb *pSdb = pMnode->pSdb;
  void *pIter = NULL;

  while (1) {
    SVgObj *pVgroup = NULL;
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
    if (pIter == NULL) break;

    if (pVgroup->dbUid == pDb->uid) {
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
      if (pVgRaw == NULL || mndTransAppendCommitlog(pTrans, pVgRaw) != 0) {
        sdbCancelFetch(pSdb, pIter);
        sdbRelease(pSdb, pVgroup);
        return -1;
      }
      sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED);
    }

    sdbRelease(pSdb, pVgroup);
  }

  return 0;
}

static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
    STransAction action = {0};
S
Shengliang Guan 已提交
714
    SVnodeGid   *pVgid = pVgroup->vnodeGid + vn;
S
Shengliang Guan 已提交
715 716 717 718 719 720

    SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
    if (pDnode == NULL) return -1;
    action.epSet = mndGetDnodeEpset(pDnode);
    mndReleaseDnode(pMnode, pDnode);

S
Shengliang Guan 已提交
721 722
    SDropVnodeReq *pReq = mndBuildDropVnodeReq(pMnode, pDnode, pDb, pVgroup);
    if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
723

S
Shengliang Guan 已提交
724
    action.pCont = pReq;
S
Shengliang Guan 已提交
725
    action.contLen = sizeof(SDropVnodeReq);
H
Hongze Cheng 已提交
726
    action.msgType = TDMT_DND_DROP_VNODE;
S
Shengliang Guan 已提交
727
    action.acceptableCode = TSDB_CODE_DND_VNODE_NOT_DEPLOYED;
S
Shengliang Guan 已提交
728
    if (mndTransAppendRedoAction(pTrans, &action) != 0) {
S
Shengliang Guan 已提交
729
      free(pReq);
S
Shengliang Guan 已提交
730 731 732 733
      return -1;
    }
  }

S
Shengliang Guan 已提交
734 735 736
  return 0;
}

S
Shengliang Guan 已提交
737 738 739
static int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
  SSdb *pSdb = pMnode->pSdb;
  void *pIter = NULL;
S
Shengliang Guan 已提交
740

S
Shengliang Guan 已提交
741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
  while (1) {
    SVgObj *pVgroup = NULL;
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
    if (pIter == NULL) break;

    if (pVgroup->dbUid == pDb->uid) {
      if (mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup) != 0) {
        sdbCancelFetch(pSdb, pIter);
        sdbRelease(pSdb, pVgroup);
        return -1;
      }
    }

    sdbRelease(pSdb, pVgroup);
  }

  return 0;
}
S
Shengliang Guan 已提交
759

S
Shengliang Guan 已提交
760
static int32_t mndDropDb(SMnode *pMnode, SMnodeMsg *pReq, SDbObj *pDb) {
S
Shengliang Guan 已提交
761
  int32_t code = -1;
S
Shengliang Guan 已提交
762 763
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, &pReq->rpcMsg);
  if (pTrans == NULL) goto DROP_DB_OVER;
S
Shengliang Guan 已提交
764

S
Shengliang Guan 已提交
765 766
  mDebug("trans:%d, used to drop db:%s", pTrans->id, pDb->name);

S
Shengliang Guan 已提交
767 768 769 770
  if (mndSetDropDbRedoLogs(pMnode, pTrans, pDb) != 0) goto DROP_DB_OVER;
  if (mndSetDropDbCommitLogs(pMnode, pTrans, pDb) != 0) goto DROP_DB_OVER;
  if (mndSetDropDbRedoActions(pMnode, pTrans, pDb) != 0) goto DROP_DB_OVER;
  if (mndTransPrepare(pMnode, pTrans) != 0) goto DROP_DB_OVER;
S
Shengliang Guan 已提交
771

S
Shengliang Guan 已提交
772 773 774
  code = 0;

DROP_DB_OVER:
S
Shengliang Guan 已提交
775
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
776
  return code;
S
Shengliang Guan 已提交
777 778
}

S
Shengliang Guan 已提交
779 780
static int32_t mndProcessDropDbReq(SMnodeMsg *pReq) {
  SMnode     *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
781
  SDropDbReq *pDrop = pReq->rpcMsg.pCont;
S
Shengliang Guan 已提交
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796

  mDebug("db:%s, start to drop", pDrop->db);

  SDbObj *pDb = mndAcquireDb(pMnode, pDrop->db);
  if (pDb == NULL) {
    if (pDrop->ignoreNotExists) {
      mDebug("db:%s, not exist, ignore not exist is set", pDrop->db);
      return TSDB_CODE_SUCCESS;
    } else {
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
      mError("db:%s, failed to drop since %s", pDrop->db, terrstr());
      return -1;
    }
  }

S
Shengliang Guan 已提交
797
  int32_t code = mndDropDb(pMnode, pReq, pDb);
S
Shengliang Guan 已提交
798 799 800 801 802 803 804 805 806
  mndReleaseDb(pMnode, pDb);

  if (code != 0) {
    mError("db:%s, failed to drop since %s", pDrop->db, terrstr());
    return code;
  }

  return TSDB_CODE_MND_ACTION_IN_PROGRESS;
}
S
Shengliang Guan 已提交
807

D
dapan1121 已提交
808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845
static void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SVgroupInfo *vgList, int32_t *vgNum) {
  int32_t vindex = 0;
  SSdb *pSdb = pMnode->pSdb;

  void *pIter = NULL;
  while (vindex < pDb->cfg.numOfVgroups) {
    SVgObj *pVgroup = NULL;
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
    if (pIter == NULL) break;

    if (pVgroup->dbUid == pDb->uid) {
      SVgroupInfo *pInfo = &vgList[vindex];
      pInfo->vgId = htonl(pVgroup->vgId);
      pInfo->hashBegin = htonl(pVgroup->hashBegin);
      pInfo->hashEnd = htonl(pVgroup->hashEnd);
      pInfo->numOfEps = pVgroup->replica;
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
        SVnodeGid  *pVgid = &pVgroup->vnodeGid[gid];
        SEpAddr *pEpArrr = &pInfo->epAddr[gid];
        SDnodeObj  *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
        if (pDnode != NULL) {
          memcpy(pEpArrr->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
          pEpArrr->port = htons(pDnode->port);
        }
        mndReleaseDnode(pMnode, pDnode);
        if (pVgid->role == TAOS_SYNC_STATE_LEADER) {
          pInfo->inUse = gid;
        }
      }
      vindex++;
    }

    sdbRelease(pSdb, pVgroup);
  }

  *vgNum = vindex;
}

S
Shengliang Guan 已提交
846 847
static int32_t mndProcessUseDbReq(SMnodeMsg *pReq) {
  SMnode    *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
848
  SSdb      *pSdb = pMnode->pSdb;
S
Shengliang Guan 已提交
849
  SUseDbReq *pUse = pReq->rpcMsg.pCont;
S
Shengliang Guan 已提交
850
  pUse->vgVersion = htonl(pUse->vgVersion);
S
Shengliang Guan 已提交
851

S
Shengliang Guan 已提交
852 853 854
  SDbObj *pDb = mndAcquireDb(pMnode, pUse->db);
  if (pDb == NULL) {
    terrno = TSDB_CODE_MND_DB_NOT_EXIST;
S
Shengliang Guan 已提交
855
    mError("db:%s, failed to process use db req since %s", pUse->db, terrstr());
S
Shengliang Guan 已提交
856 857
    return -1;
  }
S
Shengliang Guan 已提交
858

S
Shengliang Guan 已提交
859
  int32_t    contLen = sizeof(SUseDbRsp) + pDb->cfg.numOfVgroups * sizeof(SVgroupInfo);
S
Shengliang Guan 已提交
860 861
  SUseDbRsp *pRsp = rpcMallocCont(contLen);
  if (pRsp == NULL) {
S
Shengliang Guan 已提交
862
    mndReleaseDb(pMnode, pDb);
S
Shengliang Guan 已提交
863 864 865 866
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

D
dapan1121 已提交
867
  int32_t vgNum = 0;
S
Shengliang Guan 已提交
868 869

  if (pUse->vgVersion < pDb->vgVersion) {
D
dapan1121 已提交
870
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->vgroupInfo, &vgNum);
S
Shengliang Guan 已提交
871 872
  }

873
  memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
D
dapan1121 已提交
874
  pRsp->uid = htobe64(pDb->uid);
S
Shengliang Guan 已提交
875
  pRsp->vgVersion = htonl(pDb->vgVersion);
D
dapan1121 已提交
876
  pRsp->vgNum = htonl(vgNum);
S
Shengliang Guan 已提交
877 878
  pRsp->hashMethod = pDb->hashMethod;

S
Shengliang Guan 已提交
879 880
  pReq->pCont = pRsp;
  pReq->contLen = contLen;
S
Shengliang Guan 已提交
881 882 883
  mndReleaseDb(pMnode, pDb);

  return 0;
S
Shengliang Guan 已提交
884
}
S
Shengliang Guan 已提交
885

D
dapan1121 已提交
886 887 888 889 890 891 892 893 894 895 896
int32_t mndValidateDBInfo(SMnode *pMnode, SDbVgVersion *dbs, int32_t num, void **rsp, int32_t *rspLen) {
  SSdb *pSdb = pMnode->pSdb;
  int32_t bufSize = num * (sizeof(SUseDbRsp) + TSDB_DEFAULT_VN_PER_DB * sizeof(SVgroupInfo));
  void *buf = malloc(bufSize);
  int32_t len = 0;
  int32_t contLen = 0;
  int32_t bufOffset = 0;
  SUseDbRsp *pRsp = NULL;

  for (int32_t i = 0; i < num; ++i) {
    SDbVgVersion *db = &dbs[i];
D
dapan1121 已提交
897 898
    db->dbId = be64toh(db->dbId);
    db->vgVersion = ntohl(db->vgVersion);
D
dapan1121 已提交
899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933

    len = 0;
    
    SDbObj *pDb = mndAcquireDb(pMnode, db->dbName);
    if (pDb == NULL) {
      mInfo("db %s not exist", db->dbName);
      
      len = sizeof(SUseDbRsp);
    } else if (pDb->uid != db->dbId || db->vgVersion < pDb->vgVersion) {
      len = sizeof(SUseDbRsp) + pDb->cfg.numOfVgroups * sizeof(SVgroupInfo);
    }

    if (0 == len) {
      mndReleaseDb(pMnode, pDb);
      
      continue;
    }
    
    contLen += len;
    
    if (contLen > bufSize) {
      buf = realloc(buf, contLen);
    }
    
    pRsp = (SUseDbRsp *)((char *)buf + bufOffset);
    memcpy(pRsp->db, db->dbName, TSDB_DB_FNAME_LEN);
    if (pDb) {
      int32_t vgNum = 0;
      mndBuildDBVgroupInfo(pDb, pMnode, pRsp->vgroupInfo, &vgNum);

      pRsp->uid = htobe64(pDb->uid);
      pRsp->vgVersion = htonl(pDb->vgVersion);
      pRsp->vgNum = htonl(vgNum);
      pRsp->hashMethod = pDb->hashMethod;
    } else {
D
dapan1121 已提交
934 935 936
      pRsp->uid = htobe64(db->dbId);
      pRsp->vgNum = htonl(0);
      pRsp->hashMethod = 0;
D
dapan1121 已提交
937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956
      pRsp->vgVersion = htonl(-1);
    }

    bufOffset += len;
    
    mndReleaseDb(pMnode, pDb);
  }

  if (contLen > 0) {
    *rsp = buf;
    *rspLen = contLen;
  } else {
    *rsp = NULL;
    tfree(buf);
    *rspLen = 0;
  }

  return 0;
}

S
Shengliang Guan 已提交
957 958
static int32_t mndProcessSyncDbReq(SMnodeMsg *pReq) {
  SMnode     *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
959
  SSyncDbReq *pSync = pReq->rpcMsg.pCont;
S
Shengliang Guan 已提交
960
  SDbObj     *pDb = mndAcquireDb(pMnode, pSync->db);
S
Shengliang Guan 已提交
961
  if (pDb == NULL) {
S
Shengliang Guan 已提交
962
    mError("db:%s, failed to process sync db req since %s", pSync->db, terrstr());
S
Shengliang Guan 已提交
963 964
    return -1;
  }
S
Shengliang Guan 已提交
965 966 967

  mndReleaseDb(pMnode, pDb);
  return 0;
S
Shengliang Guan 已提交
968 969
}

S
Shengliang Guan 已提交
970 971
static int32_t mndProcessCompactDbReq(SMnodeMsg *pReq) {
  SMnode        *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
972
  SCompactDbReq *pCompact = pReq->rpcMsg.pCont;
S
Shengliang Guan 已提交
973
  SDbObj        *pDb = mndAcquireDb(pMnode, pCompact->db);
S
Shengliang Guan 已提交
974
  if (pDb == NULL) {
S
Shengliang Guan 已提交
975
    mError("db:%s, failed to process compact db req since %s", pCompact->db, terrstr());
S
Shengliang Guan 已提交
976 977
    return -1;
  }
S
Shengliang Guan 已提交
978 979 980

  mndReleaseDb(pMnode, pDb);
  return 0;
S
Shengliang Guan 已提交
981
}
S
Shengliang Guan 已提交
982

S
Shengliang Guan 已提交
983
static int32_t mndGetDbMeta(SMnodeMsg *pReq, SShowObj *pShow, STableMetaRsp *pMeta) {
S
Shengliang Guan 已提交
984
  SMnode *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
985 986 987
  SSdb   *pSdb = pMnode->pSdb;

  int32_t  cols = 0;
S
Shengliang Guan 已提交
988
  SSchema *pSchema = pMeta->pSchema;
S
Shengliang Guan 已提交
989 990 991 992

  pShow->bytes[cols] = (TSDB_DB_NAME_LEN - 1) + VARSTR_HEADER_SIZE;
  pSchema[cols].type = TSDB_DATA_TYPE_BINARY;
  strcpy(pSchema[cols].name, "name");
H
Haojun Liao 已提交
993
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
994 995 996 997
  cols++;

  pShow->bytes[cols] = 8;
  pSchema[cols].type = TSDB_DATA_TYPE_TIMESTAMP;
H
Haojun Liao 已提交
998 999
  strcpy(pSchema[cols].name, "create_time");
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1000 1001
  cols++;

S
Shengliang Guan 已提交
1002 1003 1004
  pShow->bytes[cols] = 2;
  pSchema[cols].type = TSDB_DATA_TYPE_SMALLINT;
  strcpy(pSchema[cols].name, "vgroups");
1005
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1006 1007
  cols++;

S
Shengliang Guan 已提交
1008 1009 1010 1011 1012 1013
  pShow->bytes[cols] = 4;
  pSchema[cols].type = TSDB_DATA_TYPE_INT;
  strcpy(pSchema[cols].name, "ntables");
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
  cols++;

S
Shengliang Guan 已提交
1014 1015 1016
  pShow->bytes[cols] = 2;
  pSchema[cols].type = TSDB_DATA_TYPE_SMALLINT;
  strcpy(pSchema[cols].name, "replica");
H
Haojun Liao 已提交
1017
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1018 1019 1020 1021 1022
  cols++;

  pShow->bytes[cols] = 2;
  pSchema[cols].type = TSDB_DATA_TYPE_SMALLINT;
  strcpy(pSchema[cols].name, "quorum");
H
Haojun Liao 已提交
1023
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1024 1025 1026 1027 1028
  cols++;

  pShow->bytes[cols] = 2;
  pSchema[cols].type = TSDB_DATA_TYPE_SMALLINT;
  strcpy(pSchema[cols].name, "days");
H
Haojun Liao 已提交
1029
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1030 1031 1032 1033 1034
  cols++;

  pShow->bytes[cols] = 24 + VARSTR_HEADER_SIZE;
  pSchema[cols].type = TSDB_DATA_TYPE_BINARY;
  strcpy(pSchema[cols].name, "keep0,keep1,keep2");
H
Haojun Liao 已提交
1035
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1036 1037 1038 1039
  cols++;

  pShow->bytes[cols] = 4;
  pSchema[cols].type = TSDB_DATA_TYPE_INT;
H
Haojun Liao 已提交
1040 1041
  strcpy(pSchema[cols].name, "cache");
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1042 1043 1044 1045 1046
  cols++;

  pShow->bytes[cols] = 4;
  pSchema[cols].type = TSDB_DATA_TYPE_INT;
  strcpy(pSchema[cols].name, "blocks");
H
Haojun Liao 已提交
1047
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1048 1049 1050 1051 1052
  cols++;

  pShow->bytes[cols] = 4;
  pSchema[cols].type = TSDB_DATA_TYPE_INT;
  strcpy(pSchema[cols].name, "minrows");
H
Haojun Liao 已提交
1053
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1054 1055 1056 1057 1058
  cols++;

  pShow->bytes[cols] = 4;
  pSchema[cols].type = TSDB_DATA_TYPE_INT;
  strcpy(pSchema[cols].name, "maxrows");
H
Haojun Liao 已提交
1059
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1060 1061 1062 1063 1064
  cols++;

  pShow->bytes[cols] = 1;
  pSchema[cols].type = TSDB_DATA_TYPE_TINYINT;
  strcpy(pSchema[cols].name, "wallevel");
H
Haojun Liao 已提交
1065
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1066 1067 1068 1069 1070
  cols++;

  pShow->bytes[cols] = 4;
  pSchema[cols].type = TSDB_DATA_TYPE_INT;
  strcpy(pSchema[cols].name, "fsync");
H
Haojun Liao 已提交
1071
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1072 1073 1074 1075 1076
  cols++;

  pShow->bytes[cols] = 1;
  pSchema[cols].type = TSDB_DATA_TYPE_TINYINT;
  strcpy(pSchema[cols].name, "comp");
H
Haojun Liao 已提交
1077
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1078 1079 1080 1081 1082
  cols++;

  pShow->bytes[cols] = 1;
  pSchema[cols].type = TSDB_DATA_TYPE_TINYINT;
  strcpy(pSchema[cols].name, "cachelast");
H
Haojun Liao 已提交
1083
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1084 1085 1086 1087 1088
  cols++;

  pShow->bytes[cols] = 3 + VARSTR_HEADER_SIZE;
  pSchema[cols].type = TSDB_DATA_TYPE_BINARY;
  strcpy(pSchema[cols].name, "precision");
H
Haojun Liao 已提交
1089
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1090 1091 1092 1093 1094
  cols++;

  pShow->bytes[cols] = 1;
  pSchema[cols].type = TSDB_DATA_TYPE_TINYINT;
  strcpy(pSchema[cols].name, "update");
H
Haojun Liao 已提交
1095
  pSchema[cols].bytes = htonl(pShow->bytes[cols]);
S
Shengliang Guan 已提交
1096 1097
  cols++;

S
Shengliang Guan 已提交
1098
  pMeta->numOfColumns = htonl(cols);
S
Shengliang Guan 已提交
1099 1100 1101 1102 1103 1104 1105 1106 1107
  pShow->numOfColumns = cols;

  pShow->offset[0] = 0;
  for (int32_t i = 1; i < cols; ++i) {
    pShow->offset[i] = pShow->offset[i - 1] + pShow->bytes[i - 1];
  }

  pShow->numOfRows = sdbGetSize(pSdb, SDB_DB);
  pShow->rowSize = pShow->offset[cols - 1] + pShow->bytes[cols - 1];
1108
  strcpy(pMeta->tbFname, mndShowStr(pShow->type));
S
Shengliang Guan 已提交
1109 1110 1111 1112

  return 0;
}

S
Shengliang Guan 已提交
1113
char *mnGetDbStr(char *src) {
S
Shengliang Guan 已提交
1114 1115 1116
  char *pos = strstr(src, TS_PATH_DELIMITER);
  if (pos != NULL) ++pos;

S
Shengliang Guan 已提交
1117 1118 1119 1120
  if (pos == NULL) {
    return src;
  }

S
Shengliang Guan 已提交
1121 1122 1123
  return pos;
}

S
Shengliang Guan 已提交
1124 1125
static int32_t mndRetrieveDbs(SMnodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows) {
  SMnode *pMnode = pReq->pMnode;
S
Shengliang Guan 已提交
1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
  SSdb   *pSdb = pMnode->pSdb;
  int32_t numOfRows = 0;
  SDbObj *pDb = NULL;
  char   *pWrite;
  int32_t cols = 0;

  while (numOfRows < rows) {
    pShow->pIter = sdbFetch(pSdb, SDB_DB, pShow->pIter, (void **)&pDb);
    if (pShow->pIter == NULL) break;

    cols = 0;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
S
Shengliang Guan 已提交
1139
    char *name = mnGetDbStr(pDb->name);
S
Shengliang Guan 已提交
1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150
    if (name != NULL) {
      STR_WITH_MAXSIZE_TO_VARSTR(pWrite, name, pShow->bytes[cols]);
    } else {
      STR_TO_VARSTR(pWrite, "NULL");
    }
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int64_t *)pWrite = pDb->createdTime;
    cols++;

S
Shengliang Guan 已提交
1151
    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
1152
    *(int16_t *)pWrite = pDb->cfg.numOfVgroups;
S
Shengliang Guan 已提交
1153 1154
    cols++;

S
Shengliang Guan 已提交
1155 1156 1157 1158
    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int16_t *)pWrite = 0;  // todo
    cols++;

S
Shengliang Guan 已提交
1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189
    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int16_t *)pWrite = pDb->cfg.replications;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int16_t *)pWrite = pDb->cfg.quorum;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int16_t *)pWrite = pDb->cfg.daysPerFile;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    char tmp[128] = {0};
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
      sprintf(tmp, "%d,%d,%d", pDb->cfg.daysToKeep1, pDb->cfg.daysToKeep2, pDb->cfg.daysToKeep0);
    } else {
      sprintf(tmp, "%d,%d,%d", pDb->cfg.daysToKeep0, pDb->cfg.daysToKeep1, pDb->cfg.daysToKeep2);
    }
    STR_WITH_SIZE_TO_VARSTR(pWrite, tmp, strlen(tmp));
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int32_t *)pWrite = pDb->cfg.cacheBlockSize;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int32_t *)pWrite = pDb->cfg.totalBlocks;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
S
Shengliang Guan 已提交
1190
    *(int32_t *)pWrite = pDb->cfg.minRows;
S
Shengliang Guan 已提交
1191 1192 1193
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
S
Shengliang Guan 已提交
1194
    *(int32_t *)pWrite = pDb->cfg.maxRows;
S
Shengliang Guan 已提交
1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int8_t *)pWrite = pDb->cfg.walLevel;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int32_t *)pWrite = pDb->cfg.fsyncPeriod;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int8_t *)pWrite = pDb->cfg.compression;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int8_t *)pWrite = pDb->cfg.cacheLastRow;
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    char *prec = NULL;
    switch (pDb->cfg.precision) {
      case TSDB_TIME_PRECISION_MILLI:
        prec = TSDB_TIME_PRECISION_MILLI_STR;
        break;
      case TSDB_TIME_PRECISION_MICRO:
        prec = TSDB_TIME_PRECISION_MICRO_STR;
        break;
      case TSDB_TIME_PRECISION_NANO:
        prec = TSDB_TIME_PRECISION_NANO_STR;
        break;
      default:
        assert(false);
        break;
    }
    STR_WITH_SIZE_TO_VARSTR(pWrite, prec, 2);
    cols++;

    pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
    *(int8_t *)pWrite = pDb->cfg.update;
    cols++;

    numOfRows++;
    sdbRelease(pSdb, pDb);
  }

S
Shengliang Guan 已提交
1240
  mndVacuumResult(data, pShow->numOfColumns, numOfRows, rows, pShow);
S
Shengliang Guan 已提交
1241 1242 1243 1244 1245 1246 1247 1248
  pShow->numOfReads += numOfRows;

  return numOfRows;
}

static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
  SSdb *pSdb = pMnode->pSdb;
  sdbCancelFetch(pSdb, pIter);
D
dapan1121 已提交
1249
}