mndDb.c 51.0 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 "mndAuth.h"
S
Shengliang Guan 已提交
19
#include "mndDnode.h"
S
Shengliang Guan 已提交
20
#include "mndShow.h"
21
#include "mndStb.h"
S
Shengliang Guan 已提交
22
#include "mndTrans.h"
S
Shengliang Guan 已提交
23
#include "mndUser.h"
24
#include "mndVgroup.h"
D
dapan1121 已提交
25
#include "mndSma.h"
S
Shengliang Guan 已提交
26

S
Shengliang Guan 已提交
27
#define TSDB_DB_VER_NUMBER   1
S
Shengliang Guan 已提交
28
#define TSDB_DB_RESERVE_SIZE 64
S
Shengliang Guan 已提交
29 30 31 32 33

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 已提交
34
static int32_t  mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew);
S
Shengliang Guan 已提交
35 36 37 38 39 40 41
static int32_t  mndProcessCreateDbReq(SNodeMsg *pReq);
static int32_t  mndProcessAlterDbReq(SNodeMsg *pReq);
static int32_t  mndProcessDropDbReq(SNodeMsg *pReq);
static int32_t  mndProcessUseDbReq(SNodeMsg *pReq);
static int32_t  mndProcessSyncDbReq(SNodeMsg *pReq);
static int32_t  mndProcessCompactDbReq(SNodeMsg *pReq);
static int32_t  mndRetrieveDbs(SNodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows);
S
Shengliang Guan 已提交
42
static void     mndCancelGetNextDb(SMnode *pMnode, void *pIter);
D
dapan1121 已提交
43 44
static int32_t  mndProcessGetDbCfgReq(SNodeMsg *pReq);
static int32_t  mndProcessGetIndexReq(SNodeMsg *pReq);
S
Shengliang Guan 已提交
45

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

S
Shengliang Guan 已提交
55 56 57 58 59 60
  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);
D
dapan1121 已提交
61
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
D
dapan1121 已提交
62
  mndSetMsgHandle(pMnode, TDMT_MND_GET_INDEX, mndProcessGetIndexReq);
S
Shengliang Guan 已提交
63 64 65 66

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

S
Shengliang Guan 已提交
67
  return sdbSetTable(pMnode->pSdb, table);
S
Shengliang Guan 已提交
68
}
S
Shengliang Guan 已提交
69

S
Shengliang Guan 已提交
70
void mndCleanupDb(SMnode *pMnode) {}
S
Shengliang Guan 已提交
71

S
Shengliang Guan 已提交
72
static SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
73 74
  terrno = TSDB_CODE_OUT_OF_MEMORY;

75 76
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, TSDB_DB_VER_NUMBER,
                              sizeof(SDbObj) + pDb->cfg.numOfRetensions * sizeof(SRetention) + TSDB_DB_RESERVE_SIZE);
77
  if (pRaw == NULL) goto DB_ENCODE_OVER;
S
Shengliang Guan 已提交
78 79

  int32_t dataPos = 0;
80 81
  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)
S
Shengliang Guan 已提交
82
  SDB_SET_BINARY(pRaw, dataPos, pDb->createUser, TSDB_USER_LEN, DB_ENCODE_OVER)
83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
  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)
S
sma  
Shengliang Guan 已提交
107 108 109 110 111 112 113 114 115
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.numOfRetensions, DB_ENCODE_OVER)
  for (int32_t i = 0; i < pDb->cfg.numOfRetensions; ++i) {
    SRetention *pRetension = taosArrayGet(pDb->cfg.pRetensions, i);
    SDB_SET_INT32(pRaw, dataPos, pRetension->freq, DB_ENCODE_OVER)
    SDB_SET_INT32(pRaw, dataPos, pRetension->keep, DB_ENCODE_OVER)
    SDB_SET_INT8(pRaw, dataPos, pRetension->freqUnit, DB_ENCODE_OVER)
    SDB_SET_INT8(pRaw, dataPos, pRetension->keepUnit, DB_ENCODE_OVER)
  }

116 117 118 119 120 121 122 123 124 125 126
  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 已提交
127

128
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
S
Shengliang Guan 已提交
129 130 131 132
  return pRaw;
}

static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
133 134
  terrno = TSDB_CODE_OUT_OF_MEMORY;

S
Shengliang Guan 已提交
135
  int8_t sver = 0;
136
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto DB_DECODE_OVER;
S
Shengliang Guan 已提交
137

S
Shengliang Guan 已提交
138
  if (sver != TSDB_DB_VER_NUMBER) {
S
Shengliang Guan 已提交
139
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
140
    goto DB_DECODE_OVER;
S
Shengliang Guan 已提交
141 142 143
  }

  SSdbRow *pRow = sdbAllocRow(sizeof(SDbObj));
144 145 146 147
  if (pRow == NULL) goto DB_DECODE_OVER;

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

  int32_t dataPos = 0;
150 151
  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)
S
Shengliang Guan 已提交
152
  SDB_GET_BINARY(pRaw, dataPos, pDb->createUser, TSDB_USER_LEN, DB_DECODE_OVER)
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
  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)
S
sma  
Shengliang Guan 已提交
177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfRetensions, DB_DECODE_OVER)
  if (pDb->cfg.numOfRetensions > 0) {
    pDb->cfg.pRetensions = taosArrayInit(pDb->cfg.numOfRetensions, sizeof(SRetention));
    if (pDb->cfg.pRetensions == NULL) goto DB_DECODE_OVER;
    for (int32_t i = 0; i < pDb->cfg.numOfRetensions; ++i) {
      SRetention retension = {0};
      SDB_GET_INT32(pRaw, dataPos, &retension.freq, DB_DECODE_OVER)
      SDB_GET_INT32(pRaw, dataPos, &retension.keep, DB_DECODE_OVER)
      SDB_GET_INT8(pRaw, dataPos, &retension.freqUnit, DB_DECODE_OVER)
      SDB_GET_INT8(pRaw, dataPos, &retension.keepUnit, DB_DECODE_OVER)
      if (taosArrayPush(pDb->cfg.pRetensions, &retension) == NULL) {
        goto DB_DECODE_OVER;
      }
    }
  }

193 194 195 196 197 198 199
  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());
wafwerar's avatar
wafwerar 已提交
200
    taosMemoryFreeClear(pRow);
201 202
    return NULL;
  }
S
Shengliang Guan 已提交
203

204
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
S
Shengliang Guan 已提交
205 206 207 208
  return pRow;
}

static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
209
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
S
Shengliang Guan 已提交
210 211 212 213
  return 0;
}

static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
214
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
S
sma  
Shengliang Guan 已提交
215
  taosArrayDestroy(pDb->cfg.pRetensions);
S
Shengliang Guan 已提交
216 217 218
  return 0;
}

S
Shengliang Guan 已提交
219 220 221 222 223 224
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 已提交
225 226 227
  return 0;
}

S
Shengliang Guan 已提交
228
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
S
Shengliang Guan 已提交
229 230
  SSdb   *pSdb = pMnode->pSdb;
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
S
Shengliang Guan 已提交
231
  if (pDb == NULL && terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
S
Shengliang Guan 已提交
232 233 234
    terrno = TSDB_CODE_MND_DB_NOT_EXIST;
  }
  return pDb;
S
Shengliang Guan 已提交
235
}
S
Shengliang Guan 已提交
236

S
Shengliang Guan 已提交
237 238 239 240 241
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
  SSdb *pSdb = pMnode->pSdb;
  sdbRelease(pSdb, pDb);
}

S
Shengliang Guan 已提交
242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257
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 已提交
258
static int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
259
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return -1;
S
Shengliang Guan 已提交
260 261 262 263 264 265
  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 < 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;
S
Shengliang Guan 已提交
266
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return -1;
S
Shengliang Guan 已提交
267 268 269 270 271 272 273
  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;
D
dapan1121 已提交
274
  if (pCfg->ttl < TSDB_MIN_DB_TTL_OPTION && pCfg->ttl != TSDB_DEFAULT_DB_TTL_OPTION) return -1;
S
Shengliang Guan 已提交
275 276 277 278 279 280 281 282 283
  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
fix ci  
Shengliang Guan 已提交
284
  if (pCfg->streamMode < TSDB_MIN_DB_STREAM_MODE || pCfg->streamMode > TSDB_MAX_DB_STREAM_MODE) return -1;
D
dapan1121 已提交
285
  if (pCfg->singleSTable < TSDB_MIN_DB_SINGLE_STABLE_OPTION || pCfg->streamMode > TSDB_MAX_DB_SINGLE_STABLE_OPTION) return -1;
S
Shengliang Guan 已提交
286 287 288 289
  return TSDB_CODE_SUCCESS;
}

static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
290
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
S
Shengliang Guan 已提交
291 292 293 294
  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;
295 296
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
S
Shengliang Guan 已提交
297 298
  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 已提交
299 300
  if (pCfg->commitTime < 0) pCfg->commitTime = TSDB_DEFAULT_COMMIT_TIME;
  if (pCfg->fsyncPeriod < 0) pCfg->fsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
D
dapan1121 已提交
301
  if (pCfg->ttl < 0) pCfg->ttl = TSDB_DEFAULT_DB_TTL_OPTION;
S
Shengliang Guan 已提交
302 303 304 305 306 307 308
  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;
309
  if (pCfg->streamMode < 0) pCfg->streamMode = TSDB_DEFAULT_DB_STREAM_MODE;
D
dapan1121 已提交
310
  if (pCfg->singleSTable < 0) pCfg->singleSTable = TSDB_DEFAULT_DB_SINGLE_STABLE_OPTION;
311
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
S
Shengliang Guan 已提交
312
}
S
Shengliang Guan 已提交
313

S
Shengliang Guan 已提交
314
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
S
Shengliang Guan 已提交
315
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
S
Shengliang Guan 已提交
316 317 318
  if (pDbRaw == NULL) return -1;
  if (mndTransAppendRedolog(pTrans, pDbRaw) != 0) return -1;
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
S
Shengliang Guan 已提交
319

S
Shengliang Guan 已提交
320
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
S
Shengliang Guan 已提交
321
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
S
Shengliang Guan 已提交
322 323 324
    if (pVgRaw == NULL) return -1;
    if (mndTransAppendRedolog(pTrans, pVgRaw) != 0) return -1;
    if (sdbSetRawStatus(pVgRaw, SDB_STATUS_CREATING) != 0) return -1;
S
Shengliang Guan 已提交
325 326 327 328 329
  }

  return 0;
}

S
Shengliang Guan 已提交
330
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
S
Shengliang Guan 已提交
331
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
S
Shengliang Guan 已提交
332 333 334
  if (pDbRaw == NULL) return -1;
  if (mndTransAppendUndolog(pTrans, pDbRaw) != 0) return -1;
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED) != 0) return -1;
S
Shengliang Guan 已提交
335

S
Shengliang Guan 已提交
336
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
S
Shengliang Guan 已提交
337
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
S
Shengliang Guan 已提交
338 339 340
    if (pVgRaw == NULL) return -1;
    if (mndTransAppendUndolog(pTrans, pVgRaw) != 0) return -1;
    if (sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED) != 0) return -1;
S
Shengliang Guan 已提交
341 342 343 344 345
  }

  return 0;
}

S
Shengliang Guan 已提交
346
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
S
Shengliang Guan 已提交
347
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
S
Shengliang Guan 已提交
348 349 350
  if (pDbRaw == NULL) return -1;
  if (mndTransAppendCommitlog(pTrans, pDbRaw) != 0) return -1;
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_READY) != 0) return -1;
S
Shengliang Guan 已提交
351

S
Shengliang Guan 已提交
352
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
S
Shengliang Guan 已提交
353
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
S
Shengliang Guan 已提交
354 355 356
    if (pVgRaw == NULL) return -1;
    if (mndTransAppendCommitlog(pTrans, pVgRaw) != 0) return -1;
    if (sdbSetRawStatus(pVgRaw, SDB_STATUS_READY) != 0) return -1;
S
Shengliang Guan 已提交
357 358 359 360 361
  }

  return 0;
}

S
Shengliang Guan 已提交
362 363
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 已提交
364
    SVgObj *pVgroup = pVgroups + vg;
S
Shengliang Guan 已提交
365 366

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

S
Shengliang Guan 已提交
370 371 372
      SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
      if (pDnode == NULL) return -1;
      action.epSet = mndGetDnodeEpset(pDnode);
S
Shengliang Guan 已提交
373 374
      mndReleaseDnode(pMnode, pDnode);

S
Shengliang Guan 已提交
375 376
      int32_t contLen = 0;
      void   *pReq = mndBuildCreateVnodeReq(pMnode, pDnode, pDb, pVgroup, &contLen);
S
Shengliang Guan 已提交
377
      if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
378

S
Shengliang Guan 已提交
379
      action.pCont = pReq;
S
Shengliang Guan 已提交
380
      action.contLen = contLen;
H
Hongze Cheng 已提交
381
      action.msgType = TDMT_DND_CREATE_VNODE;
382
      action.acceptableCode = TSDB_CODE_NODE_ALREADY_DEPLOYED;
S
Shengliang Guan 已提交
383
      if (mndTransAppendRedoAction(pTrans, &action) != 0) {
wafwerar's avatar
wafwerar 已提交
384
        taosMemoryFree(pReq);
S
Shengliang Guan 已提交
385 386 387 388 389
        return -1;
      }
    }
  }

S
Shengliang Guan 已提交
390 391 392
  return 0;
}

S
Shengliang Guan 已提交
393 394
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 已提交
395
    SVgObj *pVgroup = pVgroups + vg;
S
Shengliang Guan 已提交
396 397

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

S
Shengliang Guan 已提交
401 402 403
      SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
      if (pDnode == NULL) return -1;
      action.epSet = mndGetDnodeEpset(pDnode);
S
Shengliang Guan 已提交
404 405
      mndReleaseDnode(pMnode, pDnode);

S
Shengliang Guan 已提交
406 407
      int32_t contLen = 0;
      void   *pReq = mndBuildDropVnodeReq(pMnode, pDnode, pDb, pVgroup, &contLen);
S
Shengliang Guan 已提交
408
      if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
409

S
Shengliang Guan 已提交
410
      action.pCont = pReq;
S
Shengliang Guan 已提交
411
      action.contLen = contLen;
H
Hongze Cheng 已提交
412
      action.msgType = TDMT_DND_DROP_VNODE;
413
      action.acceptableCode = TSDB_CODE_NODE_NOT_DEPLOYED;
S
Shengliang Guan 已提交
414
      if (mndTransAppendUndoAction(pTrans, &action) != 0) {
wafwerar's avatar
wafwerar 已提交
415
        taosMemoryFree(pReq);
S
Shengliang Guan 已提交
416 417 418 419 420
        return -1;
      }
    }
  }

S
Shengliang Guan 已提交
421 422 423
  return 0;
}

S
Shengliang Guan 已提交
424
static int32_t mndCreateDb(SMnode *pMnode, SNodeMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser) {
S
Shengliang Guan 已提交
425
  SDbObj dbObj = {0};
426
  memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
S
Shengliang Guan 已提交
427
  memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
S
Shengliang Guan 已提交
428 429
  dbObj.createdTime = taosGetTimestampMs();
  dbObj.updateTime = dbObj.createdTime;
430
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
S
Shengliang Guan 已提交
431 432
  dbObj.cfgVersion = 1;
  dbObj.vgVersion = 1;
S
Shengliang Guan 已提交
433
  dbObj.hashMethod = 1;
S
Shengliang Guan 已提交
434
  memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
L
Liu Jicong 已提交
435 436 437 438 439 440 441 442 443 444 445
  dbObj.cfg = (SDbCfg){
      .numOfVgroups = pCreate->numOfVgroups,
      .cacheBlockSize = pCreate->cacheBlockSize,
      .totalBlocks = pCreate->totalBlocks,
      .daysPerFile = pCreate->daysPerFile,
      .daysToKeep0 = pCreate->daysToKeep0,
      .daysToKeep1 = pCreate->daysToKeep1,
      .daysToKeep2 = pCreate->daysToKeep2,
      .minRows = pCreate->minRows,
      .maxRows = pCreate->maxRows,
      .commitTime = pCreate->commitTime,
446
      .fsyncPeriod = pCreate->fsyncPeriod,
D
dapan1121 已提交
447
      .ttl = pCreate->ttl,
448
      .walLevel = pCreate->walLevel,
L
Liu Jicong 已提交
449 450 451 452 453 454 455
      .precision = pCreate->precision,
      .compression = pCreate->compression,
      .replications = pCreate->replications,
      .quorum = pCreate->quorum,
      .update = pCreate->update,
      .cacheLastRow = pCreate->cacheLastRow,
      .streamMode = pCreate->streamMode,
D
dapan1121 已提交
456
      .singleSTable = pCreate->singleSTable,
L
Liu Jicong 已提交
457
  };
S
Shengliang Guan 已提交
458

S
sma  
Shengliang Guan 已提交
459 460
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
  dbObj.cfg.pRetensions = pCreate->pRetensions;
461
  pCreate->pRetensions = NULL;
S
sma  
Shengliang Guan 已提交
462

S
Shengliang Guan 已提交
463 464
  mndSetDefaultDbCfg(&dbObj.cfg);

S
Shengliang Guan 已提交
465 466 467 468 469
  if (mndCheckDbName(dbObj.name, pUser) != 0) {
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }

S
Shengliang Guan 已提交
470 471
  if (mndCheckDbCfg(pMnode, &dbObj.cfg) != 0) {
    terrno = TSDB_CODE_MND_INVALID_DB_OPTION;
S
Shengliang Guan 已提交
472 473 474
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }
S
Shengliang Guan 已提交
475

S
Shengliang Guan 已提交
476 477
  SVgObj *pVgroups = NULL;
  if (mndAllocVgroup(pMnode, &dbObj, &pVgroups) != 0) {
478 479 480 481
    mError("db:%s, failed to create since %s", pCreate->db, terrstr());
    return -1;
  }

S
Shengliang Guan 已提交
482
  int32_t code = -1;
S
Shengliang Guan 已提交
483
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_TYPE_CREATE_DB, &pReq->rpcMsg);
S
Shengliang Guan 已提交
484
  if (pTrans == NULL) goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
485

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

S
Shengliang Guan 已提交
488
  mndTransSetDbInfo(pTrans, &dbObj);
S
Shengliang Guan 已提交
489 490 491 492 493 494
  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 已提交
495

S
Shengliang Guan 已提交
496 497 498
  code = 0;

CREATE_DB_OVER:
wafwerar's avatar
wafwerar 已提交
499
  taosMemoryFree(pVgroups);
S
Shengliang Guan 已提交
500
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
501
  return code;
S
Shengliang Guan 已提交
502 503
}

S
Shengliang Guan 已提交
504 505
static int32_t mndProcessCreateDbReq(SNodeMsg *pReq) {
  SMnode      *pMnode = pReq->pNode;
S
Shengliang Guan 已提交
506 507 508 509 510 511 512 513 514 515 516 517 518
  int32_t      code = -1;
  SDbObj      *pDb = NULL;
  SUserObj    *pUser = NULL;
  SCreateDbReq createReq = {0};

  if (tDeserializeSCreateDbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &createReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto CREATE_DB_OVER;
  }

  mDebug("db:%s, start to create, vgroups:%d", createReq.db, createReq.numOfVgroups);

  pDb = mndAcquireDb(pMnode, createReq.db);
S
Shengliang Guan 已提交
519
  if (pDb != NULL) {
S
Shengliang Guan 已提交
520 521 522 523
    if (createReq.ignoreExist) {
      mDebug("db:%s, already exist, ignore exist is set", createReq.db);
      code = 0;
      goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
524 525
    } else {
      terrno = TSDB_CODE_MND_DB_ALREADY_EXIST;
S
Shengliang Guan 已提交
526
      goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
527
    }
S
Shengliang Guan 已提交
528
  } else if (terrno != TSDB_CODE_MND_DB_NOT_EXIST) {
S
Shengliang Guan 已提交
529
    goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
530 531
  }

S
Shengliang Guan 已提交
532 533 534 535 536 537 538
  pUser = mndAcquireUser(pMnode, pReq->user);
  if (pUser == NULL) {
    goto CREATE_DB_OVER;
  }

  if (mndCheckCreateDbAuth(pUser) != 0) {
    goto CREATE_DB_OVER;
S
Shengliang Guan 已提交
539 540
  }

S
Shengliang Guan 已提交
541 542
  code = mndCreateDb(pMnode, pReq, &createReq, pUser);
  if (code == 0) code = TSDB_CODE_MND_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
543

S
Shengliang Guan 已提交
544 545 546
CREATE_DB_OVER:
  if (code != 0 && code != TSDB_CODE_MND_ACTION_IN_PROGRESS) {
    mError("db:%s, failed to create since %s", createReq.db, terrstr());
S
Shengliang Guan 已提交
547 548
  }

S
Shengliang Guan 已提交
549 550
  mndReleaseDb(pMnode, pDb);
  mndReleaseUser(pMnode, pUser);
S
sma  
Shengliang Guan 已提交
551
  tFreeSCreateDbReq(&createReq);
S
Shengliang Guan 已提交
552 553

  return code;
S
Shengliang Guan 已提交
554 555
}

S
Shengliang Guan 已提交
556
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
S
Shengliang Guan 已提交
557
  terrno = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
S
Shengliang Guan 已提交
558 559 560

  if (pAlter->totalBlocks >= 0 && pAlter->totalBlocks != pDb->cfg.totalBlocks) {
    pDb->cfg.totalBlocks = pAlter->totalBlocks;
S
Shengliang Guan 已提交
561
    terrno = 0;
S
Shengliang Guan 已提交
562 563 564 565
  }

  if (pAlter->daysToKeep0 >= 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
S
Shengliang Guan 已提交
566
    terrno = 0;
S
Shengliang Guan 已提交
567 568 569 570
  }

  if (pAlter->daysToKeep1 >= 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
S
Shengliang Guan 已提交
571
    terrno = 0;
S
Shengliang Guan 已提交
572 573 574 575
  }

  if (pAlter->daysToKeep2 >= 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
S
Shengliang Guan 已提交
576
    terrno = 0;
S
Shengliang Guan 已提交
577 578 579 580
  }

  if (pAlter->fsyncPeriod >= 0 && pAlter->fsyncPeriod != pDb->cfg.fsyncPeriod) {
    pDb->cfg.fsyncPeriod = pAlter->fsyncPeriod;
S
Shengliang Guan 已提交
581
    terrno = 0;
S
Shengliang Guan 已提交
582 583 584 585
  }

  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
    pDb->cfg.walLevel = pAlter->walLevel;
S
Shengliang Guan 已提交
586
    terrno = 0;
S
Shengliang Guan 已提交
587 588 589 590
  }

  if (pAlter->quorum >= 0 && pAlter->quorum != pDb->cfg.quorum) {
    pDb->cfg.quorum = pAlter->quorum;
S
Shengliang Guan 已提交
591
    terrno = 0;
S
Shengliang Guan 已提交
592 593 594 595
  }

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

S
Shengliang Guan 已提交
599 600 601
  return terrno;
}

S
Shengliang Guan 已提交
602 603
static int32_t mndSetUpdateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
S
Shengliang Guan 已提交
604 605 606 607 608 609 610
  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 已提交
611 612
static int32_t mndSetUpdateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
S
Shengliang Guan 已提交
613 614 615
  if (pCommitRaw == NULL) return -1;
  if (mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) return -1;
  if (sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY) != 0) return -1;
S
Shengliang Guan 已提交
616 617 618 619

  return 0;
}

S
Shengliang Guan 已提交
620 621 622 623 624 625 626 627 628 629
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 已提交
630 631
    int32_t contLen = 0;
    void   *pReq = mndBuildCreateVnodeReq(pMnode, pDnode, pDb, pVgroup, &contLen);
S
Shengliang Guan 已提交
632
    if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
633

S
Shengliang Guan 已提交
634
    action.pCont = pReq;
S
Shengliang Guan 已提交
635
    action.contLen = contLen;
H
Hongze Cheng 已提交
636
    action.msgType = TDMT_DND_ALTER_VNODE;
S
Shengliang Guan 已提交
637
    if (mndTransAppendRedoAction(pTrans, &action) != 0) {
wafwerar's avatar
wafwerar 已提交
638
      taosMemoryFree(pReq);
S
Shengliang Guan 已提交
639 640 641
      return -1;
    }
  }
S
Shengliang Guan 已提交
642 643 644 645

  return 0;
}

S
Shengliang Guan 已提交
646
static int32_t mndSetUpdateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
S
Shengliang Guan 已提交
647 648
  SSdb *pSdb = pMnode->pSdb;
  void *pIter = NULL;
S
Shengliang Guan 已提交
649

S
Shengliang Guan 已提交
650 651 652 653
  while (1) {
    SVgObj *pVgroup = NULL;
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
    if (pIter == NULL) break;
S
Shengliang Guan 已提交
654

S
Shengliang Guan 已提交
655 656
    if (pVgroup->dbUid == pNew->uid) {
      if (mndBuildUpdateVgroupAction(pMnode, pTrans, pNew, pVgroup) != 0) {
S
Shengliang Guan 已提交
657 658 659 660 661 662 663 664 665 666 667
        sdbCancelFetch(pSdb, pIter);
        sdbRelease(pSdb, pVgroup);
        return -1;
      }
    }

    sdbRelease(pSdb, pVgroup);
  }

  return 0;
}
S
Shengliang Guan 已提交
668

S
Shengliang Guan 已提交
669
static int32_t mndUpdateDb(SMnode *pMnode, SNodeMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
S
Shengliang Guan 已提交
670
  int32_t code = -1;
S
Shengliang Guan 已提交
671
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_TYPE_ALTER_DB, &pReq->rpcMsg);
S
Shengliang Guan 已提交
672
  if (pTrans == NULL) goto UPDATE_DB_OVER;
S
Shengliang Guan 已提交
673

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

S
Shengliang Guan 已提交
676
  mndTransSetDbInfo(pTrans, pOld);
S
Shengliang Guan 已提交
677 678 679 680
  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 已提交
681

S
Shengliang Guan 已提交
682 683 684
  code = 0;

UPDATE_DB_OVER:
S
Shengliang Guan 已提交
685
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
686
  return code;
S
Shengliang Guan 已提交
687 688
}

S
Shengliang Guan 已提交
689 690
static int32_t mndProcessAlterDbReq(SNodeMsg *pReq) {
  SMnode     *pMnode = pReq->pNode;
S
Shengliang Guan 已提交
691 692 693 694 695 696 697 698 699
  int32_t     code = -1;
  SDbObj     *pDb = NULL;
  SUserObj   *pUser = NULL;
  SAlterDbReq alterReq = {0};

  if (tDeserializeSAlterDbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &alterReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto ALTER_DB_OVER;
  }
S
Shengliang Guan 已提交
700

S
Shengliang Guan 已提交
701
  mDebug("db:%s, start to alter", alterReq.db);
S
Shengliang Guan 已提交
702

S
Shengliang Guan 已提交
703
  pDb = mndAcquireDb(pMnode, alterReq.db);
S
Shengliang Guan 已提交
704
  if (pDb == NULL) {
S
Shengliang Guan 已提交
705 706 707 708 709 710 711 712 713 714 715
    terrno = TSDB_CODE_MND_DB_NOT_EXIST;
    goto ALTER_DB_OVER;
  }

  pUser = mndAcquireUser(pMnode, pReq->user);
  if (pUser == NULL) {
    goto ALTER_DB_OVER;
  }

  if (mndCheckAlterDropCompactSyncDbAuth(pUser, pDb) != 0) {
    goto ALTER_DB_OVER;
S
Shengliang Guan 已提交
716 717 718 719 720
  }

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

S
Shengliang Guan 已提交
721
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
S
Shengliang Guan 已提交
722
  if (code != 0) {
S
Shengliang Guan 已提交
723
    goto ALTER_DB_OVER;
S
Shengliang Guan 已提交
724 725
  }

S
Shengliang Guan 已提交
726 727
  dbObj.cfgVersion++;
  dbObj.updateTime = taosGetTimestampMs();
S
Shengliang Guan 已提交
728
  code = mndUpdateDb(pMnode, pReq, pDb, &dbObj);
S
Shengliang Guan 已提交
729
  if (code == 0) code = TSDB_CODE_MND_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
730

S
Shengliang Guan 已提交
731 732 733
ALTER_DB_OVER:
  if (code != 0 && code != TSDB_CODE_MND_ACTION_IN_PROGRESS) {
    mError("db:%s, failed to alter since %s", alterReq.db, terrstr());
S
Shengliang Guan 已提交
734 735
  }

S
Shengliang Guan 已提交
736 737 738 739
  mndReleaseDb(pMnode, pDb);
  mndReleaseUser(pMnode, pUser);

  return code;
S
Shengliang Guan 已提交
740 741
}

D
dapan1121 已提交
742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806
static int32_t mndProcessGetDbCfgReq(SNodeMsg *pReq) {
  SMnode     *pMnode = pReq->pNode;
  int32_t     code = -1;
  SDbObj     *pDb = NULL;
  SDbCfgReq   cfgReq = {0};
  SDbCfgRsp   cfgRsp = {0};

  if (tDeserializeSDbCfgReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &cfgReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto GET_DB_CFG_OVER;
  }

  pDb = mndAcquireDb(pMnode, cfgReq.db);
  if (pDb == NULL) {
    terrno = TSDB_CODE_MND_DB_NOT_EXIST;
    goto GET_DB_CFG_OVER;
  }

  cfgRsp.numOfVgroups   = pDb->cfg.numOfVgroups;
  cfgRsp.cacheBlockSize = pDb->cfg.cacheBlockSize;
  cfgRsp.totalBlocks    = pDb->cfg.totalBlocks;
  cfgRsp.daysPerFile    = pDb->cfg.daysPerFile;
  cfgRsp.daysToKeep0    = pDb->cfg.daysToKeep0;
  cfgRsp.daysToKeep1    = pDb->cfg.daysToKeep1;
  cfgRsp.daysToKeep2    = pDb->cfg.daysToKeep2;
  cfgRsp.minRows        = pDb->cfg.minRows;
  cfgRsp.maxRows        = pDb->cfg.maxRows;
  cfgRsp.commitTime     = pDb->cfg.commitTime;
  cfgRsp.fsyncPeriod    = pDb->cfg.fsyncPeriod;
  cfgRsp.ttl            = pDb->cfg.ttl;
  cfgRsp.walLevel       = pDb->cfg.walLevel;
  cfgRsp.precision      = pDb->cfg.precision;
  cfgRsp.compression    = pDb->cfg.compression;
  cfgRsp.replications   = pDb->cfg.replications;
  cfgRsp.quorum         = pDb->cfg.quorum;
  cfgRsp.update         = pDb->cfg.update;
  cfgRsp.cacheLastRow   = pDb->cfg.cacheLastRow;
  cfgRsp.streamMode     = pDb->cfg.streamMode;
  cfgRsp.singleSTable   = pDb->cfg.singleSTable;

  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
  void   *pRsp = rpcMallocCont(contLen);
  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    code = -1;
    goto GET_DB_CFG_OVER;
  }

  tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp);

  pReq->pRsp = pRsp;
  pReq->rspLen = contLen;

GET_DB_CFG_OVER:

  if (code != 0 && code != TSDB_CODE_MND_ACTION_IN_PROGRESS) {
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
  }

  mndReleaseDb(pMnode, pDb);

  return code;
}


S
Shengliang Guan 已提交
807 808 809 810 811 812 813 814 815 816 817 818 819 820 821
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 已提交
822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842
  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);
  }

843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860
  while (1) {
    SStbObj *pStb = NULL;
    pIter = sdbFetch(pSdb, SDB_STB, pIter, (void **)&pStb);
    if (pIter == NULL) break;

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

    sdbRelease(pSdb, pStb);
  }

S
Shengliang Guan 已提交
861 862 863 864 865 866
  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 已提交
867
    SVnodeGid   *pVgid = pVgroup->vnodeGid + vn;
S
Shengliang Guan 已提交
868 869 870 871 872 873

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

S
Shengliang Guan 已提交
874 875
    int32_t contLen = 0;
    void   *pReq = mndBuildDropVnodeReq(pMnode, pDnode, pDb, pVgroup, &contLen);
S
Shengliang Guan 已提交
876
    if (pReq == NULL) return -1;
S
Shengliang Guan 已提交
877

S
Shengliang Guan 已提交
878
    action.pCont = pReq;
S
Shengliang Guan 已提交
879
    action.contLen = contLen;
H
Hongze Cheng 已提交
880
    action.msgType = TDMT_DND_DROP_VNODE;
881
    action.acceptableCode = TSDB_CODE_NODE_NOT_DEPLOYED;
S
Shengliang Guan 已提交
882
    if (mndTransAppendRedoAction(pTrans, &action) != 0) {
wafwerar's avatar
wafwerar 已提交
883
      taosMemoryFree(pReq);
S
Shengliang Guan 已提交
884 885 886 887
      return -1;
    }
  }

S
Shengliang Guan 已提交
888 889 890
  return 0;
}

S
Shengliang Guan 已提交
891 892 893
static int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
  SSdb *pSdb = pMnode->pSdb;
  void *pIter = NULL;
S
Shengliang Guan 已提交
894

S
Shengliang Guan 已提交
895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912
  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 已提交
913

914 915 916 917 918 919 920 921 922 923 924 925
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
  SDropDbRsp dropRsp = {0};
  if (pDb != NULL) {
    memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
    dropRsp.uid = pDb->uid;
  }

  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
  void   *pRsp = NULL;
  if (useRpcMalloc) {
    pRsp = rpcMallocCont(rspLen);
  } else {
wafwerar's avatar
wafwerar 已提交
926
    pRsp = taosMemoryMalloc(rspLen);
927 928 929 930 931 932 933 934 935 936 937 938 939
  }

  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

  tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp);
  *pRspLen = rspLen;
  *ppRsp = pRsp;
  return 0;
}

S
Shengliang Guan 已提交
940
static int32_t mndDropDb(SMnode *pMnode, SNodeMsg *pReq, SDbObj *pDb) {
S
Shengliang Guan 已提交
941
  int32_t code = -1;
S
Shengliang Guan 已提交
942
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_TYPE_DROP_DB, &pReq->rpcMsg);
S
Shengliang Guan 已提交
943
  if (pTrans == NULL) goto DROP_DB_OVER;
S
Shengliang Guan 已提交
944

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

S
Shengliang Guan 已提交
948 949 950
  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;
S
Shengliang Guan 已提交
951

952 953 954
  int32_t rspLen = 0;
  void   *pRsp = NULL;
  if (mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false) < 0) goto DROP_DB_OVER;
S
Shengliang Guan 已提交
955 956
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);

S
Shengliang Guan 已提交
957
  if (mndTransPrepare(pMnode, pTrans) != 0) goto DROP_DB_OVER;
S
Shengliang Guan 已提交
958

S
Shengliang Guan 已提交
959 960 961
  code = 0;

DROP_DB_OVER:
S
Shengliang Guan 已提交
962
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
963
  return code;
S
Shengliang Guan 已提交
964 965
}

S
Shengliang Guan 已提交
966 967
static int32_t mndProcessDropDbReq(SNodeMsg *pReq) {
  SMnode    *pMnode = pReq->pNode;
S
Shengliang Guan 已提交
968 969 970 971 972 973 974 975 976
  int32_t    code = -1;
  SDbObj    *pDb = NULL;
  SUserObj  *pUser = NULL;
  SDropDbReq dropReq = {0};

  if (tDeserializeSDropDbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &dropReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto DROP_DB_OVER;
  }
S
Shengliang Guan 已提交
977

S
Shengliang Guan 已提交
978
  mDebug("db:%s, start to drop", dropReq.db);
S
Shengliang Guan 已提交
979

S
Shengliang Guan 已提交
980
  pDb = mndAcquireDb(pMnode, dropReq.db);
S
Shengliang Guan 已提交
981
  if (pDb == NULL) {
S
Shengliang Guan 已提交
982
    if (dropReq.ignoreNotExists) {
983
      code = mndBuildDropDbRsp(pDb, &pReq->rspLen, &pReq->pRsp, true);
S
Shengliang Guan 已提交
984
      goto DROP_DB_OVER;
S
Shengliang Guan 已提交
985 986
    } else {
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
S
Shengliang Guan 已提交
987
      goto DROP_DB_OVER;
S
Shengliang Guan 已提交
988 989 990
    }
  }

S
Shengliang Guan 已提交
991 992 993 994
  pUser = mndAcquireUser(pMnode, pReq->user);
  if (pUser == NULL) {
    goto DROP_DB_OVER;
  }
S
Shengliang Guan 已提交
995

S
Shengliang Guan 已提交
996 997
  if (mndCheckAlterDropCompactSyncDbAuth(pUser, pDb) != 0) {
    goto DROP_DB_OVER;
S
Shengliang Guan 已提交
998 999
  }

S
Shengliang Guan 已提交
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
  code = mndDropDb(pMnode, pReq, pDb);
  if (code == 0) code = TSDB_CODE_MND_ACTION_IN_PROGRESS;

DROP_DB_OVER:
  if (code != 0 && code != TSDB_CODE_MND_ACTION_IN_PROGRESS) {
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
  }

  mndReleaseDb(pMnode, pDb);
  mndReleaseUser(pMnode, pUser);

  return code;
S
Shengliang Guan 已提交
1012
}
S
Shengliang Guan 已提交
1013

D
dapan1121 已提交
1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035
void mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode, int32_t *num) {
  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) {
      *num += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;

      vindex++;
    }

    sdbRelease(pSdb, pVgroup);
  }

  sdbCancelFetch(pSdb, pIter);
}

S
Shengliang Guan 已提交
1036
static void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
D
dapan1121 已提交
1037
  int32_t vindex = 0;
S
Shengliang Guan 已提交
1038
  SSdb   *pSdb = pMnode->pSdb;
D
dapan1121 已提交
1039 1040

  void *pIter = NULL;
1041
  while (1) {
D
dapan1121 已提交
1042 1043 1044 1045
    SVgObj *pVgroup = NULL;
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
    if (pIter == NULL) break;

D
dapan1121 已提交
1046
    if (NULL == pDb || pVgroup->dbUid == pDb->uid) {
S
Shengliang Guan 已提交
1047 1048 1049 1050
      SVgroupInfo vgInfo = {0};
      vgInfo.vgId = pVgroup->vgId;
      vgInfo.hashBegin = pVgroup->hashBegin;
      vgInfo.hashEnd = pVgroup->hashEnd;
D
dapan1121 已提交
1051
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
L
Liu Jicong 已提交
1052
      vgInfo.epSet.numOfEps = pVgroup->replica;
D
dapan1121 已提交
1053
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
S
Shengliang Guan 已提交
1054
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
L
Liu Jicong 已提交
1055
        SEp       *pEp = &vgInfo.epSet.eps[gid];
S
Shengliang Guan 已提交
1056
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
D
dapan1121 已提交
1057
        if (pDnode != NULL) {
1058
          memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
S
Shengliang Guan 已提交
1059
          pEp->port = pDnode->port;
D
dapan1121 已提交
1060 1061 1062
        }
        mndReleaseDnode(pMnode, pDnode);
        if (pVgid->role == TAOS_SYNC_STATE_LEADER) {
L
Liu Jicong 已提交
1063
          vgInfo.epSet.inUse = gid;
D
dapan1121 已提交
1064 1065 1066
        }
      }
      vindex++;
S
Shengliang Guan 已提交
1067
      taosArrayPush(pVgList, &vgInfo);
D
dapan1121 已提交
1068 1069 1070
    }

    sdbRelease(pSdb, pVgroup);
L
Liu Jicong 已提交
1071

D
dapan1121 已提交
1072 1073 1074
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
      break;
    }
D
dapan1121 已提交
1075
  }
S
Shengliang Guan 已提交
1076 1077

  sdbCancelFetch(pSdb, pIter);
D
dapan1121 已提交
1078 1079
}

L
Liu Jicong 已提交
1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
  if (pRsp->pVgroupInfos == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

  int32_t numOfTable = 0;
  mndGetDBTableNum(pDb, pMnode, &numOfTable);

  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable) {
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
  }

  memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
  pRsp->uid = pDb->uid;
  pRsp->vgVersion = pDb->vgVersion;
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
  pRsp->hashMethod = pDb->hashMethod;
  return 0;
}

S
Shengliang Guan 已提交
1102 1103
static int32_t mndProcessUseDbReq(SNodeMsg *pReq) {
  SMnode   *pMnode = pReq->pNode;
S
Shengliang Guan 已提交
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
  int32_t   code = -1;
  SDbObj   *pDb = NULL;
  SUserObj *pUser = NULL;
  SUseDbReq usedbReq = {0};
  SUseDbRsp usedbRsp = {0};

  if (tDeserializeSUseDbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &usedbReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto USE_DB_OVER;
  }
S
Shengliang Guan 已提交
1114

D
dapan1121 已提交
1115 1116
  char *p = strchr(usedbReq.db, '.');
  if (p && 0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB)) {
D
dapan1121 已提交
1117
    memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
D
dapan1121 已提交
1118
    static int32_t vgVersion = 1;
D
dapan1121 已提交
1119 1120 1121 1122 1123 1124
    if (usedbReq.vgVersion < vgVersion) {
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
      if (usedbRsp.pVgroupInfos == NULL) {
        terrno = TSDB_CODE_OUT_OF_MEMORY;
        goto USE_DB_OVER;
      }
L
Liu Jicong 已提交
1125

D
dapan1121 已提交
1126
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
D
dapan1121 已提交
1127
      usedbRsp.vgVersion = vgVersion++;
L
Liu Jicong 已提交
1128

D
dapan1121 已提交
1129 1130 1131
      if (taosArrayGetSize(usedbRsp.pVgroupInfos) <= 0) {
        terrno = TSDB_CODE_MND_DB_NOT_EXIST;
      }
D
dapan1121 已提交
1132 1133
    } else {
      usedbRsp.vgVersion = usedbReq.vgVersion;
D
dapan1121 已提交
1134
      code = 0;
D
dapan1121 已提交
1135
    }
L
Liu Jicong 已提交
1136
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
D
dapan1121 已提交
1137 1138

    // no jump, need to construct rsp
D
dapan1121 已提交
1139 1140 1141 1142
  } else {
    pDb = mndAcquireDb(pMnode, usedbReq.db);
    if (pDb == NULL) {
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
S
Shengliang Guan 已提交
1143

X
Xiaoyu Wang 已提交
1144 1145 1146
      memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
      usedbRsp.uid = usedbReq.dbId;
      usedbRsp.vgVersion = usedbReq.vgVersion;
S
Shengliang Guan 已提交
1147

X
Xiaoyu Wang 已提交
1148 1149 1150 1151 1152 1153
      mError("db:%s, failed to process use db req since %s", usedbReq.db, terrstr());
    } else {
      pUser = mndAcquireUser(pMnode, pReq->user);
      if (pUser == NULL) {
        goto USE_DB_OVER;
      }
S
Shengliang Guan 已提交
1154

X
Xiaoyu Wang 已提交
1155 1156 1157
      if (mndCheckUseDbAuth(pUser, pDb) != 0) {
        goto USE_DB_OVER;
      }
S
Shengliang Guan 已提交
1158

L
Liu Jicong 已提交
1159
      if (mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq) < 0) {
X
Xiaoyu Wang 已提交
1160 1161 1162 1163 1164
        goto USE_DB_OVER;
      }

      code = 0;
    }
D
dapan1121 已提交
1165
  }
S
Shengliang Guan 已提交
1166

S
Shengliang Guan 已提交
1167 1168 1169 1170
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
  void   *pRsp = rpcMallocCont(contLen);
  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
X
Xiaoyu Wang 已提交
1171
    code = -1;
S
Shengliang Guan 已提交
1172
    goto USE_DB_OVER;
S
Shengliang Guan 已提交
1173 1174
  }

S
Shengliang Guan 已提交
1175
  tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp);
S
Shengliang Guan 已提交
1176

S
Shengliang Guan 已提交
1177 1178
  pReq->pRsp = pRsp;
  pReq->rspLen = contLen;
S
Shengliang Guan 已提交
1179

S
Shengliang Guan 已提交
1180 1181 1182 1183
USE_DB_OVER:
  if (code != 0) {
    mError("db:%s, failed to process use db req since %s", usedbReq.db, terrstr());
  }
D
dapan1121 已提交
1184

S
Shengliang Guan 已提交
1185 1186 1187
  mndReleaseDb(pMnode, pDb);
  mndReleaseUser(pMnode, pUser);
  tFreeSUsedbRsp(&usedbRsp);
D
dapan1121 已提交
1188

S
Shengliang Guan 已提交
1189 1190
  return code;
}
S
Shengliang Guan 已提交
1191

S
Shengliang Guan 已提交
1192 1193 1194 1195 1196 1197 1198
int32_t mndValidateDbInfo(SMnode *pMnode, SDbVgVersion *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
  SUseDbBatchRsp batchUseRsp = {0};
  batchUseRsp.pArray = taosArrayInit(numOfDbs, sizeof(SUseDbRsp));
  if (batchUseRsp.pArray == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }
S
Shengliang Guan 已提交
1199

S
Shengliang Guan 已提交
1200 1201 1202 1203
  for (int32_t i = 0; i < numOfDbs; ++i) {
    SDbVgVersion *pDbVgVersion = &pDbs[i];
    pDbVgVersion->dbId = htobe64(pDbVgVersion->dbId);
    pDbVgVersion->vgVersion = htonl(pDbVgVersion->vgVersion);
D
dapan1121 已提交
1204
    pDbVgVersion->numOfTable = htonl(pDbVgVersion->numOfTable);
D
dapan1121 已提交
1205

S
Shengliang Guan 已提交
1206 1207
    SUseDbRsp usedbRsp = {0};

S
Shengliang Guan 已提交
1208
    SDbObj *pDb = mndAcquireDb(pMnode, pDbVgVersion->dbFName);
S
Shengliang Guan 已提交
1209 1210 1211 1212 1213 1214
    if (pDb == NULL) {
      mDebug("db:%s, no exist", pDbVgVersion->dbFName);
      memcpy(usedbRsp.db, pDbVgVersion->dbFName, TSDB_DB_FNAME_LEN);
      usedbRsp.uid = pDbVgVersion->dbId;
      usedbRsp.vgVersion = -1;
      taosArrayPush(batchUseRsp.pArray, &usedbRsp);
D
dapan1121 已提交
1215
      continue;
D
dapan1121 已提交
1216
    }
S
Shengliang Guan 已提交
1217

D
dapan1121 已提交
1218 1219
    int32_t numOfTable = 0;
    mndGetDBTableNum(pDb, pMnode, &numOfTable);
S
Shengliang Guan 已提交
1220

D
dapan1121 已提交
1221 1222
    if (pDbVgVersion->vgVersion >= pDb->vgVersion && numOfTable == pDbVgVersion->numOfTable) {
      mDebug("db:%s, version & numOfTable not changed", pDbVgVersion->dbFName);
S
Shengliang Guan 已提交
1223
      mndReleaseDb(pMnode, pDb);
D
dapan1121 已提交
1224
      continue;
D
dapan1121 已提交
1225
    }
L
Liu Jicong 已提交
1226

D
dapan1121 已提交
1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242
    usedbRsp.pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
    if (usedbRsp.pVgroupInfos == NULL) {
      mndReleaseDb(pMnode, pDb);
      mError("db:%s, failed to malloc usedb response", pDb->name);
      continue;
    }

    mndBuildDBVgroupInfo(pDb, pMnode, usedbRsp.pVgroupInfos);
    memcpy(usedbRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
    usedbRsp.uid = pDb->uid;
    usedbRsp.vgVersion = pDb->vgVersion;
    usedbRsp.vgNum = (int32_t)taosArrayGetSize(usedbRsp.pVgroupInfos);
    usedbRsp.hashMethod = pDb->hashMethod;

    taosArrayPush(batchUseRsp.pArray, &usedbRsp);
    mndReleaseDb(pMnode, pDb);
D
dapan1121 已提交
1243 1244
  }

S
Shengliang Guan 已提交
1245
  int32_t rspLen = tSerializeSUseDbBatchRsp(NULL, 0, &batchUseRsp);
wafwerar's avatar
wafwerar 已提交
1246
  void   *pRsp = taosMemoryMalloc(rspLen);
S
Shengliang Guan 已提交
1247 1248 1249 1250
  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    tFreeSUseDbBatchRsp(&batchUseRsp);
    return -1;
D
dapan1121 已提交
1251
  }
S
Shengliang Guan 已提交
1252
  tSerializeSUseDbBatchRsp(pRsp, rspLen, &batchUseRsp);
D
dapan1121 已提交
1253

S
Shengliang Guan 已提交
1254 1255 1256 1257
  *ppRsp = pRsp;
  *pRspLen = rspLen;

  tFreeSUseDbBatchRsp(&batchUseRsp);
D
dapan1121 已提交
1258 1259 1260
  return 0;
}

S
Shengliang Guan 已提交
1261 1262
static int32_t mndProcessSyncDbReq(SNodeMsg *pReq) {
  SMnode    *pMnode = pReq->pNode;
S
Shengliang Guan 已提交
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275
  int32_t    code = -1;
  SDbObj    *pDb = NULL;
  SUserObj  *pUser = NULL;
  SSyncDbReq syncReq = {0};

  if (tDeserializeSSyncDbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &syncReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto SYNC_DB_OVER;
  }

  mDebug("db:%s, start to sync", syncReq.db);

  pDb = mndAcquireDb(pMnode, syncReq.db);
S
Shengliang Guan 已提交
1276
  if (pDb == NULL) {
S
Shengliang Guan 已提交
1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293
    goto SYNC_DB_OVER;
  }

  pUser = mndAcquireUser(pMnode, pReq->user);
  if (pUser == NULL) {
    goto SYNC_DB_OVER;
  }

  if (mndCheckAlterDropCompactSyncDbAuth(pUser, pDb) != 0) {
    goto SYNC_DB_OVER;
  }

  // code = mndSyncDb();

SYNC_DB_OVER:
  if (code != 0) {
    mError("db:%s, failed to process sync db req since %s", syncReq.db, terrstr());
S
Shengliang Guan 已提交
1294
  }
S
Shengliang Guan 已提交
1295 1296

  mndReleaseDb(pMnode, pDb);
S
Shengliang Guan 已提交
1297 1298 1299
  mndReleaseUser(pMnode, pUser);

  return code;
S
Shengliang Guan 已提交
1300 1301
}

S
Shengliang Guan 已提交
1302 1303
static int32_t mndProcessCompactDbReq(SNodeMsg *pReq) {
  SMnode       *pMnode = pReq->pNode;
S
Shengliang Guan 已提交
1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316
  int32_t       code = -1;
  SDbObj       *pDb = NULL;
  SUserObj     *pUser = NULL;
  SCompactDbReq compactReq = {0};

  if (tDeserializeSSyncDbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &compactReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto SYNC_DB_OVER;
  }

  mDebug("db:%s, start to sync", compactReq.db);

  pDb = mndAcquireDb(pMnode, compactReq.db);
S
Shengliang Guan 已提交
1317
  if (pDb == NULL) {
S
Shengliang Guan 已提交
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334
    goto SYNC_DB_OVER;
  }

  pUser = mndAcquireUser(pMnode, pReq->user);
  if (pUser == NULL) {
    goto SYNC_DB_OVER;
  }

  if (mndCheckAlterDropCompactSyncDbAuth(pUser, pDb) != 0) {
    goto SYNC_DB_OVER;
  }

  // code = mndSyncDb();

SYNC_DB_OVER:
  if (code != 0) {
    mError("db:%s, failed to process compact db req since %s", compactReq.db, terrstr());
S
Shengliang Guan 已提交
1335
  }
S
Shengliang Guan 已提交
1336 1337

  mndReleaseDb(pMnode, pDb);
S
Shengliang Guan 已提交
1338 1339 1340
  mndReleaseUser(pMnode, pUser);

  return code;
S
Shengliang Guan 已提交
1341
}
S
Shengliang Guan 已提交
1342

S
Shengliang Guan 已提交
1343
char *mnGetDbStr(char *src) {
S
Shengliang Guan 已提交
1344 1345 1346
  char *pos = strstr(src, TS_PATH_DELIMITER);
  if (pos != NULL) ++pos;

S
Shengliang Guan 已提交
1347 1348 1349 1350
  if (pos == NULL) {
    return src;
  }

S
Shengliang Guan 已提交
1351 1352 1353
  return pos;
}

L
Liu Jicong 已提交
1354
static char *getDataPosition(char *pData, SShowObj *pShow, int32_t cols, int32_t rows, int32_t capacityOfRow) {
H
Haojun Liao 已提交
1355 1356 1357
  return pData + pShow->offset[cols] * capacityOfRow + pShow->bytes[cols] * rows;
}

L
Liu Jicong 已提交
1358 1359
static void dumpDbInfoToPayload(char *data, SDbObj *pDb, SShowObj *pShow, int32_t rows, int32_t rowCapacity,
                                int64_t numOfTables) {
S
Shengliang Guan 已提交
1360 1361
  int32_t cols = 0;

L
Liu Jicong 已提交
1362
  char *pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
H
Haojun Liao 已提交
1363 1364 1365 1366 1367 1368 1369
  char *name = mnGetDbStr(pDb->name);
  if (name != NULL) {
    STR_WITH_MAXSIZE_TO_VARSTR(pWrite, name, pShow->bytes[cols]);
  } else {
    STR_TO_VARSTR(pWrite, "NULL");
  }
  cols++;
S
Shengliang Guan 已提交
1370

H
Haojun Liao 已提交
1371 1372 1373
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int64_t *)pWrite = pDb->createdTime;
  cols++;
S
Shengliang Guan 已提交
1374

H
Haojun Liao 已提交
1375 1376 1377
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int16_t *)pWrite = pDb->cfg.numOfVgroups;
  cols++;
S
Shengliang Guan 已提交
1378

H
Haojun Liao 已提交
1379
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
H
Haojun Liao 已提交
1380
  *(int64_t *)pWrite = numOfTables;
H
Haojun Liao 已提交
1381
  cols++;
S
Shengliang Guan 已提交
1382

H
Haojun Liao 已提交
1383 1384 1385
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int16_t *)pWrite = pDb->cfg.replications;
  cols++;
S
Shengliang Guan 已提交
1386

H
Haojun Liao 已提交
1387 1388 1389
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int16_t *)pWrite = pDb->cfg.quorum;
  cols++;
S
Shengliang Guan 已提交
1390

H
Haojun Liao 已提交
1391
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
X
Xiaoyu Wang 已提交
1392
  *(int32_t *)pWrite = pDb->cfg.daysPerFile;
H
Haojun Liao 已提交
1393
  cols++;
S
Shengliang Guan 已提交
1394

H
Haojun Liao 已提交
1395 1396 1397 1398 1399 1400 1401 1402 1403
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  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++;
S
Shengliang Guan 已提交
1404

H
Haojun Liao 已提交
1405 1406 1407
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int32_t *)pWrite = pDb->cfg.cacheBlockSize;
  cols++;
S
Shengliang Guan 已提交
1408

H
Haojun Liao 已提交
1409 1410 1411
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int32_t *)pWrite = pDb->cfg.totalBlocks;
  cols++;
S
Shengliang Guan 已提交
1412

H
Haojun Liao 已提交
1413 1414 1415
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int32_t *)pWrite = pDb->cfg.minRows;
  cols++;
S
Shengliang Guan 已提交
1416

H
Haojun Liao 已提交
1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int32_t *)pWrite = pDb->cfg.maxRows;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int8_t *)pWrite = pDb->cfg.walLevel;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int32_t *)pWrite = pDb->cfg.fsyncPeriod;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int8_t *)pWrite = pDb->cfg.compression;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int8_t *)pWrite = pDb->cfg.cacheLastRow;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  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:
      prec = "none";
      break;
  }
  STR_WITH_SIZE_TO_VARSTR(pWrite, prec, 2);
  cols++;

D
dapan1121 已提交
1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472
  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int32_t *)pWrite = pDb->cfg.ttl;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int8_t *)pWrite = pDb->cfg.singleSTable;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  *(int8_t *)pWrite = pDb->cfg.streamMode;
  cols++;

  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  char *status = "ready";
  STR_WITH_SIZE_TO_VARSTR(pWrite, status, strlen(status));
  cols++;

L
Liu Jicong 已提交
1473 1474
  //  pWrite = getDataPosition(data, pShow, cols, rows, rowCapacity);
  //  *(int8_t *)pWrite = pDb->cfg.update;
H
Haojun Liao 已提交
1475 1476
}

L
Liu Jicong 已提交
1477
static void setInformationSchemaDbCfg(SDbObj *pDbObj) {
H
Haojun Liao 已提交
1478 1479 1480
  ASSERT(pDbObj != NULL);
  strncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));

L
Liu Jicong 已提交
1481
  pDbObj->createdTime = 0;
H
Haojun Liao 已提交
1482
  pDbObj->cfg.numOfVgroups = 0;
L
Liu Jicong 已提交
1483
  pDbObj->cfg.quorum = 1;
H
Haojun Liao 已提交
1484
  pDbObj->cfg.replications = 1;
L
Liu Jicong 已提交
1485 1486
  pDbObj->cfg.update = 1;
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
H
Haojun Liao 已提交
1487 1488
}

1489 1490 1491 1492 1493 1494 1495 1496
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
  SVgObj  *pVgroup = pObj;
  int32_t *numOfTables = p1;

  *numOfTables += pVgroup->numOfTables;
  return true;
}

X
Xiaoyu Wang 已提交
1497 1498
static int32_t mndRetrieveDbs(SNodeMsg *pReq, SShowObj *pShow, char *data, int32_t rowsCapacity) {
  SMnode *pMnode = pReq->pNode;
H
Haojun Liao 已提交
1499 1500 1501 1502 1503 1504 1505 1506
  SSdb   *pSdb = pMnode->pSdb;
  int32_t numOfRows = 0;
  SDbObj *pDb = NULL;

  while (numOfRows < rowsCapacity) {
    pShow->pIter = sdbFetch(pSdb, SDB_DB, pShow->pIter, (void **)&pDb);
    if (pShow->pIter == NULL) {
      break;
S
Shengliang Guan 已提交
1507 1508
    }

1509 1510 1511 1512
    int32_t numOfTables = 0;
    sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, NULL, NULL);

    dumpDbInfoToPayload(data, pDb, pShow, numOfRows, rowsCapacity, numOfTables);
S
Shengliang Guan 已提交
1513 1514 1515 1516
    numOfRows++;
    sdbRelease(pSdb, pDb);
  }

H
Haojun Liao 已提交
1517
  // Append the information_schema database into the result.
H
Haojun Liao 已提交
1518 1519 1520 1521 1522 1523
  if (numOfRows < rowsCapacity) {
    SDbObj dummyISDb = {0};
    setInformationSchemaDbCfg(&dummyISDb);
    dumpDbInfoToPayload(data, &dummyISDb, pShow, numOfRows, rowsCapacity, 14);
    numOfRows += 1;
  }
H
Haojun Liao 已提交
1524 1525

  mndVacuumResult(data, pShow->numOfColumns, numOfRows, rowsCapacity, pShow);
S
Shengliang Guan 已提交
1526 1527 1528 1529 1530 1531 1532
  pShow->numOfReads += numOfRows;

  return numOfRows;
}

static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
  SSdb *pSdb = pMnode->pSdb;
D
dapan1121 已提交
1533
  sdbCancelFetch(pSdb, pIter);
L
Liu Jicong 已提交
1534
}
D
dapan1121 已提交
1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554

static int32_t mndProcessGetIndexReq(SNodeMsg *pReq) {
  SUserIndexReq indexReq = {0};
  SMnode      *pMnode = pReq->pNode;
  int32_t      code = -1;
  SUserIndexRsp rsp = {0};
  bool exist = false;

  if (tDeserializeSUserIndexReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &indexReq) != 0) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto _OVER;
  }

  code = mndProcessGetSmaReq(pMnode, &indexReq, &rsp, &exist);
  if (code) {
    goto _OVER;
  }

  if (!exist) {
    //TODO GET INDEX FROM FULLTEXT
D
dapan1121 已提交
1555 1556
    code = -1;
    terrno = TSDB_CODE_MND_DB_INDEX_NOT_EXIST;
D
dapan1121 已提交
1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574
  } else {
    int32_t contLen = tSerializeSUserIndexRsp(NULL, 0, &rsp);
    void   *pRsp = rpcMallocCont(contLen);
    if (pRsp == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      code = -1;
      goto _OVER;
    }
    
    tSerializeSUserIndexRsp(pRsp, contLen, &rsp);
    
    pReq->pRsp = pRsp;
    pReq->rspLen = contLen;

    code = 0;
  }

_OVER:
D
dapan1121 已提交
1575
  if (code != 0) {
D
dapan1121 已提交
1576 1577 1578 1579 1580 1581
    mError("failed to get index %s since %s", indexReq.indexFName, terrstr());
  }

  return code;
}