• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #4506

15 Jul 2025 12:33AM UTC coverage: 62.026% (-0.7%) from 62.706%
#4506

push

travis-ci

web-flow
docs: update stream docs (#31874)

155391 of 320094 branches covered (48.55%)

Branch coverage included in aggregate %.

240721 of 318525 relevant lines covered (75.57%)

6529048.03 hits per line

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

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

16
#define _DEFAULT_SOURCE
17
#include "audit.h"
18
#include "command.h"
19
#include "mndArbGroup.h"
20
#include "mndCluster.h"
21
#include "mndConfig.h"
22
#include "mndDb.h"
23
#include "mndDnode.h"
24
#include "mndIndex.h"
25
#include "mndPrivilege.h"
26
#include "mndShow.h"
27
#include "mndSma.h"
28
#include "mndStb.h"
29
#include "mndStream.h"
30
#include "mndSubscribe.h"
31
#include "mndTopic.h"
32
#include "mndTrans.h"
33
#include "mndUser.h"
34
#include "mndVgroup.h"
35
#include "mndView.h"
36
#include "systable.h"
37
#include "thttp.h"
38
#include "tjson.h"
39
#include "mndSsMigrate.h"
40

41
#define DB_VER_NUMBER   1
42
#define DB_RESERVE_SIZE 14
43

44
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw);
45
static int32_t  mndDbActionInsert(SSdb *pSdb, SDbObj *pDb);
46
static int32_t  mndDbActionDelete(SSdb *pSdb, SDbObj *pDb);
47
static int32_t  mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew);
48
static int32_t  mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw);
49

50
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq);
51
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq);
52
static int32_t mndProcessDropDbReq(SRpcMsg *pReq);
53
static int32_t mndProcessUseDbReq(SRpcMsg *pReq);
54
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq);
55
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq);
56
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity);
57
static void    mndCancelGetNextDb(SMnode *pMnode, void *pIter);
58
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq);
59

60
#ifndef TD_ENTERPRISE
61
int32_t mndProcessCompactDbReq(SRpcMsg *pReq) { return TSDB_CODE_OPS_NOT_SUPPORT; }
62
#endif
63

64
int32_t mndInitDb(SMnode *pMnode) {
2,477✔
65
  SSdbTable table = {
2,477✔
66
      .sdbType = SDB_DB,
67
      .keyType = SDB_KEY_BINARY,
68
      .encodeFp = (SdbEncodeFp)mndDbActionEncode,
69
      .decodeFp = (SdbDecodeFp)mndDbActionDecode,
70
      .insertFp = (SdbInsertFp)mndDbActionInsert,
71
      .updateFp = (SdbUpdateFp)mndDbActionUpdate,
72
      .deleteFp = (SdbDeleteFp)mndDbActionDelete,
73
      .validateFp = (SdbValidateFp)mndNewDbActionValidate,
74
  };
75

76
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_DB, mndProcessCreateDbReq);
2,477✔
77
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
2,477✔
78
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
2,477✔
79
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
2,477✔
80
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
2,477✔
81
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
2,477✔
82
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
2,477✔
83
  mndSetMsgHandle(pMnode, TDMT_MND_SSMIGRATE_DB, mndProcessSsMigrateDbReq);
2,477✔
84
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
2,477✔
85

86
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
2,477✔
87
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
2,477✔
88

89
  return sdbSetTable(pMnode->pSdb, table);
2,477✔
90
}
91

92
void mndCleanupDb(SMnode *pMnode) {}
2,476✔
93

94
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
22,381✔
95
  int32_t code = 0;
22,381✔
96
  int32_t lino = 0;
22,381✔
97
  terrno = TSDB_CODE_OUT_OF_MEMORY;
22,381✔
98

99
  int32_t  size = sizeof(SDbObj) + pDb->cfg.numOfRetensions * sizeof(SRetention) + DB_RESERVE_SIZE;
22,381✔
100
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, DB_VER_NUMBER, size);
22,381✔
101
  if (pRaw == NULL) goto _OVER;
22,381!
102

103
  int32_t dataPos = 0;
22,381✔
104
  SDB_SET_BINARY(pRaw, dataPos, pDb->name, TSDB_DB_FNAME_LEN, _OVER)
22,381!
105
  SDB_SET_BINARY(pRaw, dataPos, pDb->acct, TSDB_USER_LEN, _OVER)
22,381!
106
  SDB_SET_BINARY(pRaw, dataPos, pDb->createUser, TSDB_USER_LEN, _OVER)
22,381!
107
  SDB_SET_INT64(pRaw, dataPos, pDb->createdTime, _OVER)
22,381!
108
  SDB_SET_INT64(pRaw, dataPos, pDb->updateTime, _OVER)
22,381!
109
  SDB_SET_INT64(pRaw, dataPos, pDb->uid, _OVER)
22,381!
110
  SDB_SET_INT32(pRaw, dataPos, pDb->cfgVersion, _OVER)
22,381!
111
  SDB_SET_INT32(pRaw, dataPos, pDb->vgVersion, _OVER)
22,381!
112
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.numOfVgroups, _OVER)
22,381!
113
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.numOfStables, _OVER)
22,381!
114
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.buffer, _OVER)
22,381!
115
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.pageSize, _OVER)
22,381!
116
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.pages, _OVER)
22,381!
117
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.cacheLastSize, _OVER)
22,381!
118
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysPerFile, _OVER)
22,381!
119
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysToKeep0, _OVER)
22,381!
120
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysToKeep1, _OVER)
22,381!
121
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.daysToKeep2, _OVER)
22,381!
122
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.minRows, _OVER)
22,381!
123
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.maxRows, _OVER)
22,381!
124
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.walFsyncPeriod, _OVER)
22,381!
125
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.walLevel, _OVER)
22,381!
126
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.precision, _OVER)
22,381!
127
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.compression, _OVER)
22,381!
128
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.replications, _OVER)
22,381!
129
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.strict, _OVER)
22,381!
130
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.cacheLast, _OVER)
22,381!
131
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.hashMethod, _OVER)
22,381!
132
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.numOfRetensions, _OVER)
22,381!
133
  for (int32_t i = 0; i < pDb->cfg.numOfRetensions; ++i) {
22,453✔
134
    SRetention *pRetension = taosArrayGet(pDb->cfg.pRetensions, i);
72✔
135
    SDB_SET_INT64(pRaw, dataPos, pRetension->freq, _OVER)
72!
136
    SDB_SET_INT64(pRaw, dataPos, pRetension->keep, _OVER)
72!
137
    SDB_SET_INT8(pRaw, dataPos, pRetension->freqUnit, _OVER)
72!
138
    SDB_SET_INT8(pRaw, dataPos, pRetension->keepUnit, _OVER)
72!
139
  }
140
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.schemaless, _OVER)
22,381!
141
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.walRetentionPeriod, _OVER)
22,381!
142
  SDB_SET_INT64(pRaw, dataPos, pDb->cfg.walRetentionSize, _OVER)
22,381!
143
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.walRollPeriod, _OVER)
22,381!
144
  SDB_SET_INT64(pRaw, dataPos, pDb->cfg.walSegmentSize, _OVER)
22,381!
145
  SDB_SET_INT16(pRaw, dataPos, pDb->cfg.sstTrigger, _OVER)
22,381!
146
  SDB_SET_INT16(pRaw, dataPos, pDb->cfg.hashPrefix, _OVER)
22,381!
147
  SDB_SET_INT16(pRaw, dataPos, pDb->cfg.hashSuffix, _OVER)
22,381!
148
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.tsdbPageSize, _OVER)
22,381!
149
  SDB_SET_INT64(pRaw, dataPos, pDb->compactStartTime, _OVER)
22,381!
150
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.keepTimeOffset, _OVER)
22,381!
151
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.ssChunkSize, _OVER)
22,381!
152
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.ssKeepLocal, _OVER)
22,381!
153
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.ssCompact, _OVER)
22,381!
154
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.withArbitrator, _OVER)
22,381!
155
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.encryptAlgorithm, _OVER)
22,381!
156
  SDB_SET_INT32(pRaw, dataPos, pDb->tsmaVersion, _OVER);
22,381!
157
  SDB_SET_INT8(pRaw, dataPos, pDb->cfg.compactTimeOffset, _OVER)
22,381!
158
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.compactStartTime, _OVER)
22,381!
159
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.compactEndTime, _OVER)
22,381!
160
  SDB_SET_INT32(pRaw, dataPos, pDb->cfg.compactInterval, _OVER)
22,381!
161

162
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
22,381!
163
  SDB_SET_UINT8(pRaw, dataPos, pDb->cfg.flags, _OVER)
22,381!
164
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
22,381!
165

166
  terrno = 0;
22,381✔
167

168
_OVER:
22,381✔
169
  if (terrno != 0) {
22,381!
170
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
171
    sdbFreeRaw(pRaw);
×
172
    return NULL;
×
173
  }
174

175
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
22,381✔
176
  return pRaw;
22,381✔
177
}
178

179
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
20,065✔
180
  int32_t code = 0;
20,065✔
181
  int32_t lino = 0;
20,065✔
182
  terrno = TSDB_CODE_OUT_OF_MEMORY;
20,065✔
183
  SSdbRow *pRow = NULL;
20,065✔
184
  SDbObj  *pDb = NULL;
20,065✔
185

186
  int8_t sver = 0;
20,065✔
187
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
20,065!
188

189
  if (sver != DB_VER_NUMBER) {
20,065!
190
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
191
    goto _OVER;
×
192
  }
193

194
  pRow = sdbAllocRow(sizeof(SDbObj));
20,065✔
195
  if (pRow == NULL) goto _OVER;
20,065!
196

197
  pDb = sdbGetRowObj(pRow);
20,065✔
198
  if (pDb == NULL) goto _OVER;
20,065!
199

200
  int32_t dataPos = 0;
20,065✔
201
  SDB_GET_BINARY(pRaw, dataPos, pDb->name, TSDB_DB_FNAME_LEN, _OVER)
20,065!
202
  SDB_GET_BINARY(pRaw, dataPos, pDb->acct, TSDB_USER_LEN, _OVER)
20,065!
203
  SDB_GET_BINARY(pRaw, dataPos, pDb->createUser, TSDB_USER_LEN, _OVER)
20,065!
204
  SDB_GET_INT64(pRaw, dataPos, &pDb->createdTime, _OVER)
20,065!
205
  SDB_GET_INT64(pRaw, dataPos, &pDb->updateTime, _OVER)
20,065!
206
  SDB_GET_INT64(pRaw, dataPos, &pDb->uid, _OVER)
20,065!
207
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfgVersion, _OVER)
20,065!
208
  SDB_GET_INT32(pRaw, dataPos, &pDb->vgVersion, _OVER)
20,065!
209
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfVgroups, _OVER)
20,065!
210
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfStables, _OVER)
20,065!
211
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.buffer, _OVER)
20,065!
212
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.pageSize, _OVER)
20,065!
213
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.pages, _OVER)
20,065!
214
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.cacheLastSize, _OVER)
20,065!
215
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysPerFile, _OVER)
20,065!
216
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep0, _OVER)
20,065!
217
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep1, _OVER)
20,065!
218
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep2, _OVER)
20,065!
219
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.minRows, _OVER)
20,065!
220
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.maxRows, _OVER)
20,065!
221
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walFsyncPeriod, _OVER)
20,065!
222
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.walLevel, _OVER)
20,065!
223
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.precision, _OVER)
20,065!
224
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.compression, _OVER)
20,065!
225
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.replications, _OVER)
20,065!
226
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.strict, _OVER)
20,065!
227
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.cacheLast, _OVER)
20,065!
228
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.hashMethod, _OVER)
20,065!
229
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfRetensions, _OVER)
20,065!
230
  if (pDb->cfg.numOfRetensions > 0) {
20,065✔
231
    pDb->cfg.pRetensions = taosArrayInit(pDb->cfg.numOfRetensions, sizeof(SRetention));
20✔
232
    if (pDb->cfg.pRetensions == NULL) goto _OVER;
20!
233
    for (int32_t i = 0; i < pDb->cfg.numOfRetensions; ++i) {
80✔
234
      SRetention retention = {0};
60✔
235
      SDB_GET_INT64(pRaw, dataPos, &retention.freq, _OVER)
60!
236
      SDB_GET_INT64(pRaw, dataPos, &retention.keep, _OVER)
60!
237
      SDB_GET_INT8(pRaw, dataPos, &retention.freqUnit, _OVER)
60!
238
      SDB_GET_INT8(pRaw, dataPos, &retention.keepUnit, _OVER)
60!
239
      if (taosArrayPush(pDb->cfg.pRetensions, &retention) == NULL) {
120!
240
        goto _OVER;
×
241
      }
242
    }
243
  }
244
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.schemaless, _OVER)
20,065!
245
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walRetentionPeriod, _OVER)
20,065!
246
  SDB_GET_INT64(pRaw, dataPos, &pDb->cfg.walRetentionSize, _OVER)
20,065!
247
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walRollPeriod, _OVER)
20,065!
248
  SDB_GET_INT64(pRaw, dataPos, &pDb->cfg.walSegmentSize, _OVER)
20,065!
249
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.sstTrigger, _OVER)
20,065!
250
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.hashPrefix, _OVER)
20,065!
251
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.hashSuffix, _OVER)
20,065!
252
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.tsdbPageSize, _OVER)
20,065!
253
  SDB_GET_INT64(pRaw, dataPos, &pDb->compactStartTime, _OVER)
20,065!
254
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.keepTimeOffset, _OVER)
20,065!
255
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.ssChunkSize, _OVER)
20,065!
256
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.ssKeepLocal, _OVER)
20,065!
257
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.ssCompact, _OVER)
20,065!
258
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.withArbitrator, _OVER)
20,065!
259
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.encryptAlgorithm, _OVER)
20,065!
260
  SDB_GET_INT32(pRaw, dataPos, &pDb->tsmaVersion, _OVER);
20,065!
261
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.compactTimeOffset, _OVER)
20,065!
262
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.compactStartTime, _OVER)
20,065!
263
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.compactEndTime, _OVER)
20,065!
264
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.compactInterval, _OVER)
20,065!
265
  SDB_GET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
20,065!
266
  if (dataPos + sizeof(uint8_t) <= pRaw->dataLen) {
20,065!
267
    SDB_GET_UINT8(pRaw, dataPos, &pDb->cfg.flags, _OVER)
20,065!
268
  }
269

270
  taosInitRWLatch(&pDb->lock);
20,065✔
271

272
  if (pDb->cfg.ssChunkSize == 0) {
20,065!
273
    pDb->cfg.ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
×
274

275
    mInfo("db:%s, ssChunkSize set from %d to default %d", pDb->name, pDb->cfg.ssChunkSize, TSDB_DEFAULT_SS_CHUNK_SIZE);
×
276
  }
277

278
  if (pDb->cfg.ssKeepLocal == 0) {
20,065!
279
    pDb->cfg.ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
×
280

281
    mInfo("db:%s, ssKeepLocal set from %d to default %d", pDb->name, pDb->cfg.ssKeepLocal, TSDB_DEFAULT_SS_KEEP_LOCAL);
×
282
  }
283

284
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
20,065!
285
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
20,065!
286
          TSDB_DEFAULT_TSDB_PAGESIZE);
287
  }
288

289
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
20,065✔
290
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
18,104!
291
  }
292

293
  terrno = 0;
20,065✔
294

295
_OVER:
20,065✔
296
  if (terrno != 0) {
20,065!
297
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
298
    taosMemoryFreeClear(pRow);
×
299
    return NULL;
×
300
  }
301

302
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
20,065✔
303
  return pRow;
20,065✔
304
}
305

306
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
4,883✔
307
  SSdb    *pSdb = pMnode->pSdb;
4,883✔
308
  SSdbRow *pRow = NULL;
4,883✔
309
  SDbObj  *pNewDb = NULL;
4,883✔
310
  int      code = -1;
4,883✔
311

312
  pRow = mndDbActionDecode(pRaw);
4,883✔
313
  if (pRow == NULL) {
4,883!
314
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
315
    if (terrno != 0) code = terrno;
×
316
    goto _OVER;
×
317
  }
318
  pNewDb = sdbGetRowObj(pRow);
4,883✔
319
  if (pNewDb == NULL) {
4,883!
320
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
321
    if (terrno != 0) code = terrno;
×
322
    goto _OVER;
×
323
  }
324

325
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
4,883✔
326
  if (pOldDb != NULL) {
4,883!
327
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
328
    sdbRelease(pMnode->pSdb, pOldDb);
×
329
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
330
    if (terrno != 0) code = terrno;
×
331
    goto _OVER;
×
332
  }
333

334
  code = 0;
4,883✔
335
_OVER:
4,883✔
336
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
4,883!
337
  taosMemoryFreeClear(pRow);
4,883!
338
  return code;
4,883✔
339
}
340

341
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
5,417✔
342
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
5,417✔
343
  return 0;
5,417✔
344
}
345

346
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
20,065✔
347
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
20,065✔
348
  taosArrayDestroy(pDb->cfg.pRetensions);
20,065✔
349
  return 0;
20,065✔
350
}
351

352
static int32_t mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew) {
7,666✔
353
  mTrace("db:%s, perform update action, old row:%p new row:%p", pOld->name, pOld, pNew);
7,666✔
354
  taosWLockLatch(&pOld->lock);
7,666✔
355
  pOld->updateTime = pNew->updateTime;
7,666✔
356
  pOld->cfgVersion = pNew->cfgVersion;
7,666✔
357
  pOld->vgVersion = pNew->vgVersion;
7,666✔
358
  pOld->cfg.numOfVgroups = pNew->cfg.numOfVgroups;
7,666✔
359
  pOld->cfg.buffer = pNew->cfg.buffer;
7,666✔
360
  pOld->cfg.pageSize = pNew->cfg.pageSize;
7,666✔
361
  pOld->cfg.pages = pNew->cfg.pages;
7,666✔
362
  pOld->cfg.cacheLastSize = pNew->cfg.cacheLastSize;
7,666✔
363
  pOld->cfg.daysPerFile = pNew->cfg.daysPerFile;
7,666✔
364
  pOld->cfg.daysToKeep0 = pNew->cfg.daysToKeep0;
7,666✔
365
  pOld->cfg.daysToKeep1 = pNew->cfg.daysToKeep1;
7,666✔
366
  pOld->cfg.daysToKeep2 = pNew->cfg.daysToKeep2;
7,666✔
367
  pOld->cfg.keepTimeOffset = pNew->cfg.keepTimeOffset;
7,666✔
368
  pOld->cfg.walFsyncPeriod = pNew->cfg.walFsyncPeriod;
7,666✔
369
  pOld->cfg.walLevel = pNew->cfg.walLevel;
7,666✔
370
  pOld->cfg.walRetentionPeriod = pNew->cfg.walRetentionPeriod;
7,666✔
371
  pOld->cfg.walRetentionSize = pNew->cfg.walRetentionSize;
7,666✔
372
  pOld->cfg.strict = pNew->cfg.strict;
7,666✔
373
  pOld->cfg.cacheLast = pNew->cfg.cacheLast;
7,666✔
374
  pOld->cfg.replications = pNew->cfg.replications;
7,666✔
375
  pOld->cfg.sstTrigger = pNew->cfg.sstTrigger;
7,666✔
376
  pOld->cfg.minRows = pNew->cfg.minRows;
7,666✔
377
  pOld->cfg.maxRows = pNew->cfg.maxRows;
7,666✔
378
  pOld->cfg.tsdbPageSize = pNew->cfg.tsdbPageSize;
7,666✔
379
  pOld->cfg.ssChunkSize = pNew->cfg.ssChunkSize;
7,666✔
380
  pOld->cfg.ssKeepLocal = pNew->cfg.ssKeepLocal;
7,666✔
381
  pOld->cfg.ssCompact = pNew->cfg.ssCompact;
7,666✔
382
  pOld->cfg.withArbitrator = pNew->cfg.withArbitrator;
7,666✔
383
  pOld->cfg.compactInterval = pNew->cfg.compactInterval;
7,666✔
384
  pOld->cfg.compactStartTime = pNew->cfg.compactStartTime;
7,666✔
385
  pOld->cfg.compactEndTime = pNew->cfg.compactEndTime;
7,666✔
386
  pOld->cfg.compactTimeOffset = pNew->cfg.compactTimeOffset;
7,666✔
387
  pOld->compactStartTime = pNew->compactStartTime;
7,666✔
388
  pOld->tsmaVersion = pNew->tsmaVersion;
7,666✔
389
  taosWUnLockLatch(&pOld->lock);
7,666✔
390
  return 0;
7,666✔
391
}
392

393
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
25,101✔
394
  SSdb *pSdb = pMnode->pSdb;
25,101✔
395
  return sdbGetTableVer(pSdb, SDB_VGROUP);
25,101✔
396
}
397

398
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
293,502✔
399
  SSdb   *pSdb = pMnode->pSdb;
293,502✔
400
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
293,502✔
401
  if (pDb == NULL) {
293,503✔
402
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
9,763✔
403
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
8,194✔
404
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
1,569✔
405
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
133✔
406
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
1,436!
407
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
1,436✔
408
    } else {
409
      terrno = TSDB_CODE_APP_ERROR;
×
410
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
411
    }
412
  }
413
  return pDb;
293,501✔
414
}
415

416
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
332,374✔
417
  SSdb *pSdb = pMnode->pSdb;
332,374✔
418
  sdbRelease(pSdb, pDb);
332,374✔
419
}
332,374✔
420

421
bool mndDbIsExist(SMnode *pMnode, const char *db) {
12✔
422
  SDbObj *pDb = mndAcquireDb(pMnode, db);
12✔
423
  if (pDb == NULL) {
12!
424
    return false;
×
425
  } else {
426
    mndReleaseDb(pMnode, pDb);
12✔
427
    pDb = NULL;
12✔
428
    return true;
12✔
429
  }
430
}
431

432
int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
5,141✔
433
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
5,141✔
434
  if (pos == NULL) {
5,141!
435
    return TSDB_CODE_MND_INVALID_DB;
×
436
  }
437

438
  int32_t acctId;
439
  int32_t code = taosStr2int32(dbName, &acctId);
5,141✔
440
  if (code != 0) {
5,141!
441
    return code;
×
442
  }
443

444
  if (acctId != pUser->acctId) {
5,141!
445
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
446
  }
447

448
  return 0;
5,141✔
449
}
450

451
int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
5,141✔
452
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
5,141✔
453

454
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
5,141!
455
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
5,141!
456
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
5,141!
457
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
5,141!
458
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
5,141!
459
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
5,141!
460
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
5,141!
461
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
5,141!
462
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
5,141!
463
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
5,141!
464
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
5,141!
465
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
5,141!
466
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
5,141!
467
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
5,141!
468
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
5,141!
469
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
5,141!
470
  if (pCfg->minRows > pCfg->maxRows) return code;
5,141✔
471
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
5,139!
472
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
5,139!
473
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
5,139!
474
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
5,139!
475
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
5,139!
476
#ifdef TD_ENTERPRISE
477
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
5,139!
478
  if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
5,139!
479
#else
480
  if (pCfg->replications != 1 && pCfg->replications != 3) return code;
481
  if (pCfg->encryptAlgorithm != TSDB_DEFAULT_ENCRYPT_ALGO) return code;
482
#endif
483

484
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
5,139!
485
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
5,139!
486
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
5,139!
487
  if (pCfg->hashMethod != 1) return code;
5,139!
488
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
5,139!
489
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
490
    TAOS_RETURN(code);
×
491
  }
492
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
5,139!
493
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
5,139!
494
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
5,139!
495
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
5,139!
496
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
5,139!
497
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
5,139!
498
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
5,139!
499
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
5,139!
500
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
5,139!
501
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
5,139!
502
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
5,139!
503

504
  if (pCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
5,139!
505
  if (pCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
5,139!
506
  if (pCfg->ssCompact < TSDB_MIN_SS_COMPACT || pCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
5,139!
507

508
  if (pCfg->compactInterval != 0 &&
5,139✔
509
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
2!
510
    return code;
×
511
  if (pCfg->compactStartTime != 0 &&
5,139✔
512
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
1!
513
    return code;
×
514
  if (pCfg->compactEndTime != 0 &&
5,139✔
515
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
1!
516
    return code;
×
517
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
5,139!
518
    return code;
×
519
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
5,139!
520
    return code;
×
521

522
  code = 0;
5,139✔
523
  TAOS_RETURN(code);
5,139✔
524
}
525

526
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
139✔
527
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
139✔
528
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
139!
529
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
139!
530
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
139!
531
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
139!
532
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
139!
533
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
139!
534
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
139!
535
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
139!
536
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
139!
537
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
139!
538
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
139!
539
    return code;
×
540
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
139!
541
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
139!
542
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
139!
543
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
139!
544
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
139!
545
#ifdef TD_ENTERPRISE
546
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
139!
547
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
139!
548
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
2!
549
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
2✔
550
      return code;
2✔
551
    }
552
  }
553
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
137!
554
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
555
    return code;
×
556
  }
557
#else
558
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
559
#endif
560

561
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
137!
562
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
563
    return code;
×
564
  }
565
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
137!
566
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
567
    return code;
×
568
  }
569

570
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
137!
571
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
×
572
    return code;
×
573
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
137!
574
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
137!
575
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
137!
576
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
137!
577
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
137!
578
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
137!
579
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
137!
580
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
581
    return code;
×
582
  }
583
  if (pNewCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pNewCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
137!
584
  if (pNewCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pNewCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
137!
585
  if (pNewCfg->ssCompact < TSDB_MIN_SS_COMPACT || pNewCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
137!
586

587
  if (pNewCfg->compactInterval != 0 &&
137!
588
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
×
589
    return code;
×
590
  if (pNewCfg->compactStartTime != 0 &&
137!
591
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
×
592
    return code;
×
593
  if (pNewCfg->compactEndTime != 0 &&
137!
594
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
×
595
    return code;
×
596
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
137!
597
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
×
598
    return code;
×
599
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
137!
600
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
137!
601
    return code;
×
602

603
  code = 0;
137✔
604
  TAOS_RETURN(code);
137✔
605
}
606

607
static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
5,141✔
608
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
5,141!
609
  if (pCfg->numOfStables < 0) pCfg->numOfStables = TSDB_DEFAULT_DB_SINGLE_STABLE;
5,141!
610
  if (pCfg->buffer < 0) pCfg->buffer = TSDB_DEFAULT_BUFFER_PER_VNODE;
5,141✔
611
  if (pCfg->pageSize < 0) pCfg->pageSize = TSDB_DEFAULT_PAGESIZE_PER_VNODE;
5,141✔
612
  if (pCfg->pages < 0) pCfg->pages = TSDB_DEFAULT_PAGES_PER_VNODE;
5,141✔
613
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DURATION_PER_FILE;
5,141!
614
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
5,141!
615
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
5,141!
616
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
5,141!
617
  if (pCfg->keepTimeOffset < 0) pCfg->keepTimeOffset = TSDB_DEFAULT_KEEP_TIME_OFFSET;
5,141!
618
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MINROWS_FBLOCK;
5,141!
619
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAXROWS_FBLOCK;
5,141!
620
  if (pCfg->walFsyncPeriod < 0) pCfg->walFsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
5,141!
621
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
5,141!
622
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
5,141!
623
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
5,141!
624
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA;
5,141!
625
  if (pCfg->strict < 0) pCfg->strict = TSDB_DEFAULT_DB_STRICT;
5,141!
626
  if (pCfg->cacheLast < 0) pCfg->cacheLast = TSDB_DEFAULT_CACHE_MODEL;
5,141!
627
  if (pCfg->cacheLastSize <= 0) pCfg->cacheLastSize = TSDB_DEFAULT_CACHE_SIZE;
5,141✔
628
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
5,141!
629
  if (pCfg->schemaless < 0) pCfg->schemaless = TSDB_DB_SCHEMALESS_OFF;
5,141!
630
  if (pCfg->walRetentionPeriod < 0 && pCfg->walRetentionPeriod != -1)
5,141!
631
    pCfg->walRetentionPeriod = TSDB_REPS_DEF_DB_WAL_RET_PERIOD;
×
632
  if (pCfg->walRetentionSize < 0 && pCfg->walRetentionSize != -1)
5,141!
633
    pCfg->walRetentionSize = TSDB_REPS_DEF_DB_WAL_RET_SIZE;
×
634
  if (pCfg->walRollPeriod < 0) pCfg->walRollPeriod = TSDB_REPS_DEF_DB_WAL_ROLL_PERIOD;
5,141!
635
  if (pCfg->walSegmentSize < 0) pCfg->walSegmentSize = TSDB_DEFAULT_DB_WAL_SEGMENT_SIZE;
5,141!
636
  if (pCfg->sstTrigger <= 0) pCfg->sstTrigger = TSDB_DEFAULT_SST_TRIGGER;
5,141✔
637
  if (pCfg->tsdbPageSize <= 0) pCfg->tsdbPageSize = TSDB_DEFAULT_TSDB_PAGESIZE;
5,141✔
638
  if (pCfg->ssChunkSize <= 0) pCfg->ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
5,141✔
639
  if (pCfg->ssKeepLocal <= 0) pCfg->ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
5,141✔
640
  if (pCfg->ssCompact < 0) pCfg->ssCompact = TSDB_DEFAULT_SS_COMPACT;
5,141!
641
  if (pCfg->withArbitrator < 0) pCfg->withArbitrator = TSDB_DEFAULT_DB_WITH_ARBITRATOR;
5,141!
642
  if (pCfg->encryptAlgorithm < 0) pCfg->encryptAlgorithm = TSDB_DEFAULT_ENCRYPT_ALGO;
5,141!
643
}
5,141✔
644

645
int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
4,643✔
646
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,643✔
647
  if (pDbRaw == NULL) return -1;
4,643!
648

649
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
4,643!
650
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
4,643!
651
  return 0;
4,643✔
652
}
653

654
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,643✔
655
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,596✔
656
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
9,953!
657
  }
658
  return 0;
4,643✔
659
}
660

661
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
662
  int32_t  code = 0;
×
663
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
664
  if (pDbRaw == NULL) {
×
665
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
666
    if (terrno != 0) code = terrno;
×
667
    TAOS_RETURN(code);
×
668
  }
669
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
670
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
671

672
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
673
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
674
    if (pVgRaw == NULL) {
×
675
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
676
      if (terrno != 0) code = terrno;
×
677
      TAOS_RETURN(code);
×
678
    }
679
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
680
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
681
  }
682

683
  if (pDb->cfg.withArbitrator) {
×
684
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
685
      SVgObj   *pVgObj = pVgroups + v;
×
686
      SArbGroup arbGroup = {0};
×
687
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
688
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
689
    }
690
  }
691

692
  TAOS_RETURN(code);
×
693
}
694

695
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,643✔
696
  int32_t  code = 0;
4,643✔
697
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,643✔
698
  if (pDbRaw == NULL) {
4,643!
699
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
700
    if (terrno != 0) code = terrno;
×
701
    TAOS_RETURN(code);
×
702
  }
703
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
4,643!
704
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
4,643!
705

706
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,596✔
707
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
9,953✔
708
    if (pVgRaw == NULL) {
9,953!
709
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
710
      if (terrno != 0) code = terrno;
×
711
      TAOS_RETURN(code);
×
712
    }
713
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
9,953!
714
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
9,953!
715
  }
716

717
  if (pDb->cfg.withArbitrator) {
4,643✔
718
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
20✔
719
      SVgObj   *pVgObj = pVgroups + v;
13✔
720
      SArbGroup arbGroup = {0};
13✔
721
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
13!
722
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
13!
723
    }
724
  }
725

726
  TAOS_RETURN(code);
4,643✔
727
}
728

729
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
4,643✔
730
                                        SUserObj *pUserDuped) {
731
  int32_t  code = 0;
4,643✔
732
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,643✔
733
  if (pDbRaw == NULL) {
4,643!
734
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
735
    if (terrno != 0) code = terrno;
×
736
    TAOS_RETURN(code);
×
737
  }
738
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
4,643!
739
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
4,643!
740

741
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,596✔
742
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
9,953✔
743
    if (pVgRaw == NULL) {
9,953!
744
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
745
      if (terrno != 0) code = terrno;
×
746
      TAOS_RETURN(code);
×
747
    }
748
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
9,953!
749
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
9,953!
750
  }
751

752
  if (pDb->cfg.withArbitrator) {
4,643✔
753
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
20✔
754
      SVgObj   *pVgObj = pVgroups + v;
13✔
755
      SArbGroup arbGroup = {0};
13✔
756
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
13!
757
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
13!
758
    }
759
  }
760

761
  if (pUserDuped) {
4,643✔
762
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
4✔
763
    if (pUserRaw == NULL) {
4!
764
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
765
      if (terrno != 0) code = terrno;
×
766
      TAOS_RETURN(code);
×
767
    }
768
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
4!
769
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
4!
770
  }
771

772
  TAOS_RETURN(code);
4,643✔
773
}
774

775
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,643✔
776
  int32_t code = 0;
4,643✔
777
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
14,596✔
778
    SVgObj *pVgroup = pVgroups + vg;
9,953✔
779

780
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
20,847✔
781
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
10,894✔
782
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
10,894!
783
    }
784
  }
785

786
  TAOS_RETURN(code);
4,643✔
787
}
788

789
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,643✔
790
  int32_t code = 0;
4,643✔
791
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
14,596✔
792
    SVgObj *pVgroup = pVgroups + vg;
9,953✔
793

794
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
20,847✔
795
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
10,894✔
796
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
10,894!
797
    }
798
  }
799

800
  TAOS_RETURN(code);
4,643✔
801
}
802

803
static int32_t mndCreateDb(SMnode *pMnode, SRpcMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser, SArray *dnodeList) {
5,141✔
804
  int32_t  code = 0;
5,141✔
805
  SUserObj newUserObj = {0};
5,141✔
806
  SDbObj   dbObj = {0};
5,141✔
807
  (void)memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
5,141✔
808
  (void)memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
5,141✔
809
  dbObj.createdTime = taosGetTimestampMs();
5,141✔
810
  dbObj.updateTime = dbObj.createdTime;
5,141✔
811
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
5,141✔
812
  dbObj.cfgVersion = 1;
5,141✔
813
  dbObj.vgVersion = 1;
5,141✔
814
  dbObj.tsmaVersion = 1;
5,141✔
815
  (void)memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
5,141✔
816
  dbObj.cfg = (SDbCfg){
5,141✔
817
      .numOfVgroups = pCreate->numOfVgroups,
5,141✔
818
      .numOfStables = pCreate->numOfStables,
5,141✔
819
      .buffer = pCreate->buffer,
5,141✔
820
      .pageSize = pCreate->pageSize,
5,141✔
821
      .pages = pCreate->pages,
5,141✔
822
      .cacheLastSize = pCreate->cacheLastSize,
5,141✔
823
      .daysPerFile = pCreate->daysPerFile,
5,141✔
824
      .daysToKeep0 = pCreate->daysToKeep0,
5,141✔
825
      .daysToKeep1 = pCreate->daysToKeep1,
5,141✔
826
      .daysToKeep2 = pCreate->daysToKeep2,
5,141✔
827
      .keepTimeOffset = pCreate->keepTimeOffset,
5,141✔
828
      .minRows = pCreate->minRows,
5,141✔
829
      .maxRows = pCreate->maxRows,
5,141✔
830
      .walFsyncPeriod = pCreate->walFsyncPeriod,
5,141✔
831
      .walLevel = pCreate->walLevel,
5,141✔
832
      .precision = pCreate->precision,
5,141✔
833
      .compression = pCreate->compression,
5,141✔
834
      .replications = pCreate->replications,
5,141✔
835
      .strict = pCreate->strict,
5,141✔
836
      .cacheLast = pCreate->cacheLast,
5,141✔
837
      .hashMethod = 1,
838
      .schemaless = pCreate->schemaless,
5,141✔
839
      .walRetentionPeriod = pCreate->walRetentionPeriod,
5,141✔
840
      .walRetentionSize = pCreate->walRetentionSize,
5,141✔
841
      .walRollPeriod = pCreate->walRollPeriod,
5,141✔
842
      .walSegmentSize = pCreate->walSegmentSize,
5,141✔
843
      .sstTrigger = pCreate->sstTrigger,
5,141✔
844
      .hashPrefix = pCreate->hashPrefix,
5,141✔
845
      .hashSuffix = pCreate->hashSuffix,
5,141✔
846
      .ssChunkSize = pCreate->ssChunkSize,
5,141✔
847
      .ssKeepLocal = pCreate->ssKeepLocal,
5,141✔
848
      .ssCompact = pCreate->ssCompact,
5,141✔
849
      .tsdbPageSize = pCreate->tsdbPageSize,
5,141✔
850
      .withArbitrator = pCreate->withArbitrator,
5,141✔
851
      .encryptAlgorithm = pCreate->encryptAlgorithm,
5,141✔
852
      .compactInterval = pCreate->compactInterval,
5,141✔
853
      .compactStartTime = pCreate->compactStartTime,
5,141✔
854
      .compactEndTime = pCreate->compactEndTime,
5,141✔
855
      .compactTimeOffset = pCreate->compactTimeOffset,
5,141✔
856
  };
857

858
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
5,141✔
859
  dbObj.cfg.pRetensions = pCreate->pRetensions;
5,141✔
860

861
  mndSetDefaultDbCfg(&dbObj.cfg);
5,141✔
862

863
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
5,141!
864
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, terrstr());
×
865
    TAOS_RETURN(code);
×
866
  }
867

868
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
5,141✔
869
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, terrstr());
2!
870
    TAOS_RETURN(code);
2✔
871
  }
872

873
  if (dbObj.cfg.hashPrefix > 0) {
5,139✔
874
    int32_t dbLen = strlen(dbObj.name) + 1;
10✔
875
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
10!
876
    dbObj.cfg.hashPrefix += dbLen;
10✔
877
  } else if (dbObj.cfg.hashPrefix < 0) {
5,129✔
878
    int32_t dbLen = strlen(dbObj.name) + 1;
14✔
879
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
14!
880
    dbObj.cfg.hashPrefix -= dbLen;
14✔
881
  }
882

883
  SVgObj *pVgroups = NULL;
5,139✔
884
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
5,139✔
885
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
471!
886
    TAOS_RETURN(code);
471✔
887
  }
888

889
  // add database privileges for user
890
  SUserObj *pNewUserDuped = NULL;
4,668✔
891
  if (!pUser->superUser) {
4,668✔
892
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
4!
893
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
4!
894
                    NULL, _OVER);
895
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
4!
896
                    NULL, _OVER);
897
    pNewUserDuped = &newUserObj;
4✔
898
  }
899

900
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
4,668✔
901
  if (pTrans == NULL) {
4,668!
902
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
903
    if (terrno != 0) code = terrno;
×
904
    goto _OVER;
×
905
  }
906
  // mndTransSetSerial(pTrans);
907
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
4,668!
908

909
  mndTransSetDbName(pTrans, dbObj.name, NULL);
4,668✔
910
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
4,668✔
911

912
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
4,643✔
913
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
4,643!
914
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,643!
915
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,643!
916
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,643!
917
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
4,643!
918
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,643!
919
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
4,643✔
920

921
_OVER:
4,638✔
922
  taosMemoryFree(pVgroups);
4,668!
923
  mndUserFreeObj(&newUserObj);
4,668✔
924
  mndTransDrop(pTrans);
4,668✔
925
  TAOS_RETURN(code);
4,668✔
926
}
927

928
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
5,173✔
929
  int32_t    code = 0;
5,173✔
930
  SSdb      *pSdb = pMnode->pSdb;
5,173✔
931
  SDnodeObj *pDnode = NULL;
5,173✔
932
  void      *pIter = NULL;
5,173✔
933

934
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
935
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
5,173✔
936
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
1!
937
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
1!
938
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
939
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
940
           tsEncryptionKeyStat);
941
    goto _exit;
×
942
  }
943

944
  int64_t curMs = taosGetTimestampMs();
1✔
945
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
2✔
946
    bool online = false;
1✔
947
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
1!
948
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
949
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
950
      mError("db:%s, failed to check encryption key:%" PRIi8
×
951
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
952
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
953
             tsEncryptionKeyChksum);
954
      sdbCancelFetch(pSdb, pIter);
×
955
      sdbRelease(pSdb, pDnode);
×
956
      break;
×
957
    }
958
    sdbRelease(pSdb, pDnode);
1✔
959
  }
960
#else
961
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
962
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
963
    goto _exit;
964
  }
965
#endif
966
_exit:
1✔
967
  TAOS_RETURN(code);
5,173✔
968
}
969

970
#ifndef TD_ENTERPRISE
971
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
972
  if (dnodeListStr[0] != 0) {
973
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
974
    return terrno;
975
  } else {
976
    return 0;
977
  }
978
}
979
#endif
980

981
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
5,628✔
982
  SMnode      *pMnode = pReq->info.node;
5,628✔
983
  int32_t      code = -1;
5,628✔
984
  int32_t      lino = 0;
5,628✔
985
  SDbObj      *pDb = NULL;
5,628✔
986
  SUserObj    *pUser = NULL;
5,628✔
987
  SCreateDbReq createReq = {0};
5,628✔
988
  SArray      *dnodeList = NULL;
5,628✔
989

990
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
5,628✔
991
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
5,628!
992

993
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
5,628!
994
#ifdef WINDOWS
995
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
996
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
997
    goto _OVER;
998
  }
999
#endif
1000
  mInfo("db:%s, start to create, vgroups:%d", createReq.db, createReq.numOfVgroups);
5,628!
1001

1002
  pDb = mndAcquireDb(pMnode, createReq.db);
5,628✔
1003
  if (pDb != NULL) {
5,628✔
1004
    if (createReq.ignoreExist) {
382✔
1005
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
367!
1006
      code = 0;
367✔
1007
      goto _OVER;
367✔
1008
    } else {
1009
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
15✔
1010
      goto _OVER;
15✔
1011
    }
1012
  } else {
1013
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
5,246✔
1014
      code = terrno;
72✔
1015
      goto _OVER;
72✔
1016
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
5,174!
1017
      goto _OVER;
×
1018
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
5,174!
1019
      // continue
1020
    } else {  // TSDB_CODE_APP_ERROR
1021
      goto _OVER;
×
1022
    }
1023
  }
1024

1025
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_DB, NULL), &lino, _OVER);
5,174✔
1026

1027
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
5,173!
1028

1029
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
5,173✔
1030
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
5,173!
1031

1032
  if (createReq.replications == 2) {
5,173✔
1033
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
11!
1034
  }
1035

1036
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
5,173!
1037

1038
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
5,173✔
1039

1040
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pUser), &lino, _OVER);
5,141!
1041

1042
  if (sdbGetSize(pMnode->pSdb, SDB_MOUNT) > 0) {
5,141!
1043
    TAOS_CHECK_GOTO(TSDB_CODE_MND_MOUNT_NOT_EMPTY, &lino, _OVER);
×
1044
  }
1045

1046
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
5,141✔
1047
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
4,638!
1048

1049
  SName name = {0};
4,638✔
1050
  if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
4,638!
1051
    mError("db:%s, failed to parse db name", createReq.db);
×
1052

1053
  auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen);
4,638✔
1054

1055
_OVER:
5,628✔
1056
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
5,628✔
1057
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
623!
1058
  }
1059

1060
  mndReleaseDb(pMnode, pDb);
5,628✔
1061
  mndReleaseUser(pMnode, pUser);
5,628✔
1062
  tFreeSCreateDbReq(&createReq);
5,628✔
1063
  taosArrayDestroy(dnodeList);
5,628✔
1064

1065
  TAOS_RETURN(code);
5,628✔
1066
}
1067

1068
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
165✔
1069
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
165✔
1070

1071
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
165!
1072
    pDb->cfg.buffer = pAlter->buffer;
×
1073
    code = 0;
×
1074
  }
1075

1076
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
165!
1077
    pDb->cfg.pages = pAlter->pages;
×
1078
    code = 0;
×
1079
  }
1080

1081
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
165!
1082
    pDb->cfg.pageSize = pAlter->pageSize;
×
1083
    code = 0;
×
1084
  }
1085

1086
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
165!
1087
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
1088
    code = 0;
×
1089
  }
1090

1091
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
165✔
1092
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
16✔
1093
    code = 0;
16✔
1094
  }
1095

1096
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
165✔
1097
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
16✔
1098
    code = 0;
16✔
1099
  }
1100

1101
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
165✔
1102
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
17✔
1103
    code = 0;
17✔
1104
  }
1105

1106
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
165!
1107
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
1✔
1108
    code = 0;
1✔
1109
  }
1110

1111
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
165!
1112
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
6✔
1113
    code = 0;
6✔
1114
  }
1115

1116
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
165!
1117
    pDb->cfg.walLevel = pAlter->walLevel;
4✔
1118
    code = 0;
4✔
1119
  }
1120

1121
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
165!
1122
#if 1
1123
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1124
#else
1125
    pDb->cfg.strict = pAlter->strict;
1126
    code = 0;
1127
#endif
1128
  }
1129

1130
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
165!
1131
    pDb->cfg.cacheLast = pAlter->cacheLast;
77✔
1132
    code = 0;
77✔
1133
  }
1134

1135
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
165✔
1136
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
24✔
1137
    code = 0;
24✔
1138
  }
1139

1140
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
165!
1141
    pDb->cfg.replications = pAlter->replications;
13✔
1142
    pDb->vgVersion++;
13✔
1143
    code = 0;
13✔
1144
  }
1145

1146
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
165!
1147
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
×
1148
    pDb->vgVersion++;
×
1149
    code = 0;
×
1150
  }
1151

1152
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
165!
1153
    pDb->cfg.minRows = pAlter->minRows;
×
1154
    pDb->vgVersion++;
×
1155
    code = 0;
×
1156
  }
1157

1158
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
165✔
1159
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
44✔
1160
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
20✔
1161
    pDb->vgVersion++;
20✔
1162
    code = 0;
20✔
1163
  }
1164

1165
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
165!
1166
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
×
1167
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
×
1168
    pDb->vgVersion++;
×
1169
    code = 0;
×
1170
  }
1171

1172
  if (pAlter->ssKeepLocal > TSDB_MIN_SS_KEEP_LOCAL && pAlter->ssKeepLocal != pDb->cfg.ssKeepLocal) {
165!
1173
    pDb->cfg.ssKeepLocal = pAlter->ssKeepLocal;
×
1174
    pDb->vgVersion++;
×
1175
    code = 0;
×
1176
  }
1177

1178
  if (pAlter->ssCompact >= TSDB_MIN_SS_COMPACT && pAlter->ssCompact != pDb->cfg.ssCompact) {
165!
1179
    pDb->cfg.ssCompact = pAlter->ssCompact;
×
1180
    pDb->vgVersion++;
×
1181
    code = 0;
×
1182
  }
1183

1184
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
165✔
1185
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
2✔
1186
    pDb->vgVersion++;
2✔
1187
    code = 0;
2✔
1188
  }
1189

1190
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
165!
1191
    pDb->cfg.compactInterval = pAlter->compactInterval;
×
1192
    pDb->vgVersion++;
×
1193
    code = 0;
×
1194
  }
1195

1196
  bool compactTimeRangeChanged = false;
165✔
1197
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
165!
1198
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
165!
1199
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
165!
1200
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
×
1201
    compactTimeRangeChanged = true;
×
1202
    code = 0;
×
1203
  }
1204

1205
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
165!
1206
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
165!
1207
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
×
1208
    compactTimeRangeChanged = true;
×
1209
    code = 0;
×
1210
  }
1211
  if(compactTimeRangeChanged) {
165!
1212
    pDb->vgVersion++;
×
1213
  }
1214

1215
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
165!
1216
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
×
1217
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
×
1218
    pDb->vgVersion++;
×
1219
    code = 0;
×
1220
  }
1221

1222
  TAOS_RETURN(code);
165✔
1223
}
1224

1225
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
136✔
1226
  int32_t  code = 0;
136✔
1227
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
136✔
1228
  if (pRedoRaw == NULL) {
136!
1229
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1230
    if (terrno != 0) code = terrno;
×
1231
    TAOS_RETURN(code);
×
1232
  }
1233
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
136!
1234
    sdbFreeRaw(pRedoRaw);
×
1235
    TAOS_RETURN(code);
×
1236
  }
1237

1238
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
136!
1239
    sdbFreeRaw(pRedoRaw);
×
1240
    TAOS_RETURN(code);
×
1241
  }
1242
  return 0;
136✔
1243
}
1244

1245
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
136✔
1246
  int32_t  code = 0;
136✔
1247
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
136✔
1248
  if (pCommitRaw == NULL) {
136!
1249
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1250
    if (terrno != 0) code = terrno;
×
1251
    TAOS_RETURN(code);
×
1252
  }
1253
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
136!
1254
    sdbFreeRaw(pCommitRaw);
×
1255
    TAOS_RETURN(code);
×
1256
  }
1257

1258
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
136!
1259
    sdbFreeRaw(pCommitRaw);
×
1260
    TAOS_RETURN(code);
×
1261
  }
1262
  TAOS_RETURN(code);
136✔
1263
}
1264

1265
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
136✔
1266
  int32_t code = 0, lino = 0;
136✔
1267
  SSdb   *pSdb = pMnode->pSdb;
136✔
1268
  void   *pIter = NULL;
136✔
1269
  SVgObj *pVgroup = NULL;
136✔
1270
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
136✔
1271

1272
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
136!
1273

1274
  while (1) {
1275
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
833✔
1276
    if (pIter == NULL) break;
833✔
1277

1278
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
698✔
1279
      SVgObj newVgroup = {0};
517✔
1280
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
517✔
1281
                      _err);
1282

1283
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
516!
1284
        if (pNewDb->cfg.withArbitrator) {
×
1285
          SArbGroup arbGroup = {0};
×
1286
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1287
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1288
        } else {
1289
          SArbGroup arbGroup = {0};
×
1290
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
1291
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1292
        }
1293
      }
1294
    }
1295

1296
    sdbRelease(pSdb, pVgroup);
697✔
1297
  }
1298

1299
  taosArrayDestroy(pArray);
135✔
1300
  TAOS_RETURN(code);
135✔
1301

1302
_err:
1✔
1303
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
1!
1304

1305
  sdbCancelFetch(pSdb, pIter);
1✔
1306
  sdbRelease(pSdb, pVgroup);
1✔
1307
  taosArrayDestroy(pArray);
1✔
1308
  TAOS_RETURN(code);
1✔
1309
}
1310

1311
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
137✔
1312
  int32_t code = -1;
137✔
1313
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
137✔
1314
  if (pTrans == NULL) {
137!
1315
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1316
    if (terrno != 0) code = terrno;
×
1317
    TAOS_RETURN(code);
×
1318
  }
1319
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
137!
1320
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
137!
1321

1322
  mndTransSetDbName(pTrans, pOld->name, NULL);
137✔
1323
  mndTransSetGroupParallel(pTrans);
137✔
1324
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
137!
1325
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
137✔
1326

1327
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
136!
1328
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
136!
1329
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
136✔
1330

1331
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
135!
1332
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
135!
1333
  code = 0;
135✔
1334

1335
_OVER:
137✔
1336
  mndTransDrop(pTrans);
137✔
1337
  TAOS_RETURN(code);
137✔
1338
}
1339

1340
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
166✔
1341
  SMnode     *pMnode = pReq->info.node;
166✔
1342
  int32_t     code = -1;
166✔
1343
  SDbObj     *pDb = NULL;
166✔
1344
  SAlterDbReq alterReq = {0};
166✔
1345
  SDbObj      dbObj = {0};
166✔
1346

1347
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
166!
1348

1349
  mInfo("db:%s, start to alter", alterReq.db);
166!
1350

1351
  pDb = mndAcquireDb(pMnode, alterReq.db);
166✔
1352
  if (pDb == NULL) {
166!
1353
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1354
    if (terrno != 0) code = terrno;
×
1355
    goto _OVER;
×
1356
  }
1357

1358
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_ALTER_DB, pDb), NULL, _OVER);
166✔
1359

1360
  if (alterReq.replications == 2) {
165✔
1361
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
2!
1362
  }
1363

1364
  int32_t numOfTopics = 0;
165✔
1365
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
165!
1366

1367
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
165!
1368
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
1369
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1370
           numOfTopics);
1371
    goto _OVER;
×
1372
  }
1373

1374
  if (pDb->cfg.isMount) {
165!
1375
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
1376
    goto _OVER;
×
1377
  }
1378

1379
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
165✔
1380
  if (dbObj.cfg.pRetensions != NULL) {
165!
1381
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
1382
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1383
  }
1384

1385
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
165✔
1386
  if (code != 0) {
165✔
1387
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
26!
1388
    goto _OVER;
26✔
1389
  }
1390

1391
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
139✔
1392

1393
  dbObj.cfgVersion++;
137✔
1394
  dbObj.updateTime = taosGetTimestampMs();
137✔
1395
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
137✔
1396

1397
  if (dbObj.cfg.replications != pDb->cfg.replications) {
137✔
1398
    // return quickly, operation executed asynchronously
1399
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
11!
1400
  } else {
1401
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
126!
1402
  }
1403

1404
  SName name = {0};
137✔
1405
  if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
137!
1406
    mError("db:%s, failed to parse db name", alterReq.db);
×
1407

1408
  auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen);
137✔
1409

1410
_OVER:
166✔
1411
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
166✔
1412
    if (terrno != 0) code = terrno;
5!
1413
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
5!
1414
  }
1415

1416
  mndReleaseDb(pMnode, pDb);
166✔
1417
  taosArrayDestroy(dbObj.cfg.pRetensions);
166✔
1418
  tFreeSAlterDbReq(&alterReq);
166✔
1419

1420
  TAOS_RETURN(code);
166✔
1421
}
1422

1423
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb) {
6,290✔
1424
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
6,290✔
1425
  cfgRsp->dbId = pDb->uid;
6,290✔
1426
  cfgRsp->cfgVersion = pDb->cfgVersion;
6,290✔
1427
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
6,290✔
1428
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
6,290✔
1429
  cfgRsp->buffer = pDb->cfg.buffer;
6,290✔
1430
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
6,290✔
1431
  cfgRsp->pageSize = pDb->cfg.pageSize;
6,290✔
1432
  cfgRsp->pages = pDb->cfg.pages;
6,290✔
1433
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
6,290✔
1434
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
6,290✔
1435
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
6,290✔
1436
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
6,290✔
1437
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
6,290✔
1438
  cfgRsp->minRows = pDb->cfg.minRows;
6,290✔
1439
  cfgRsp->maxRows = pDb->cfg.maxRows;
6,290✔
1440
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
6,290✔
1441
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
6,290✔
1442
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
6,290✔
1443
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
6,290✔
1444
  cfgRsp->walLevel = pDb->cfg.walLevel;
6,290✔
1445
  cfgRsp->precision = pDb->cfg.precision;
6,290✔
1446
  cfgRsp->compression = pDb->cfg.compression;
6,290✔
1447
  cfgRsp->replications = pDb->cfg.replications;
6,290✔
1448
  cfgRsp->strict = pDb->cfg.strict;
6,290✔
1449
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
6,290✔
1450
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
6,290✔
1451
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
6,290✔
1452
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
6,290✔
1453
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
6,290✔
1454
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
6,290✔
1455
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
6,290✔
1456
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
6,290✔
1457
  cfgRsp->schemaless = pDb->cfg.schemaless;
6,290✔
1458
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
6,290✔
1459
  cfgRsp->ssChunkSize = pDb->cfg.ssChunkSize;
6,290✔
1460
  cfgRsp->ssKeepLocal = pDb->cfg.ssKeepLocal;
6,290✔
1461
  cfgRsp->ssCompact = pDb->cfg.ssCompact;
6,290✔
1462
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
6,290✔
1463
  cfgRsp->encryptAlgorithm = pDb->cfg.encryptAlgorithm;
6,290✔
1464
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
6,290✔
1465
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
6,290✔
1466
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
6,290✔
1467
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
6,290✔
1468
  cfgRsp->flags = pDb->cfg.flags;
6,290✔
1469
}
6,290✔
1470

1471
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
6,581✔
1472
  SMnode   *pMnode = pReq->info.node;
6,581✔
1473
  int32_t   code = -1;
6,581✔
1474
  SDbObj   *pDb = NULL;
6,581✔
1475
  SDbCfgReq cfgReq = {0};
6,581✔
1476
  SDbCfgRsp cfgRsp = {0};
6,581✔
1477

1478
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
6,581!
1479

1480
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
6,581!
1481
    pDb = mndAcquireDb(pMnode, cfgReq.db);
6,581✔
1482
    if (pDb == NULL) {
6,581✔
1483
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
845✔
1484
      if (terrno != 0) code = terrno;
845!
1485
      goto _OVER;
845✔
1486
    }
1487

1488
    mndDumpDbCfgInfo(&cfgRsp, pDb);
5,736✔
1489
  }
1490

1491
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
5,736✔
1492
  void   *pRsp = rpcMallocCont(contLen);
5,736✔
1493
  if (pRsp == NULL) {
5,736!
1494
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1495
    if (terrno != 0) code = terrno;
×
1496
    goto _OVER;
×
1497
  }
1498

1499
  int32_t ret = 0;
5,736✔
1500
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
5,736!
1501
    code = ret;
×
1502
    goto _OVER;
×
1503
  }
1504

1505
  pReq->info.rsp = pRsp;
5,736✔
1506
  pReq->info.rspLen = contLen;
5,736✔
1507

1508
  code = 0;
5,736✔
1509

1510
_OVER:
6,581✔
1511

1512
  tFreeSDbCfgRsp(&cfgRsp);
6,581✔
1513

1514
  if (code != 0) {
6,581✔
1515
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
845!
1516
  }
1517

1518
  mndReleaseDb(pMnode, pDb);
6,581✔
1519

1520
  TAOS_RETURN(code);
6,581✔
1521
}
1522

1523
int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,078✔
1524
  int32_t  code = 0;
2,078✔
1525
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
2,078✔
1526
  if (pRedoRaw == NULL) {
2,078!
1527
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1528
    if (terrno != 0) code = terrno;
×
1529
    return -1;
×
1530
  }
1531
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
2,078!
1532
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
2,078!
1533

1534
  SSdb *pSdb = pMnode->pSdb;
2,078✔
1535
  void *pIter = NULL;
2,078✔
1536

1537
  while (1) {
12✔
1538
    SArbGroup *pArbGroup = NULL;
2,090✔
1539
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
2,090✔
1540
    if (pIter == NULL) break;
2,090✔
1541

1542
    if (pArbGroup->dbUid == pDb->uid) {
12!
1543
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
12!
1544
        sdbCancelFetch(pSdb, pIter);
×
1545
        sdbRelease(pSdb, pArbGroup);
×
1546
        TAOS_RETURN(code);
×
1547
      }
1548
    }
1549

1550
    sdbRelease(pSdb, pArbGroup);
12✔
1551
  }
1552

1553
  TAOS_RETURN(code);
2,078✔
1554
}
1555

1556
int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,078✔
1557
  int32_t  code = 0;
2,078✔
1558
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
2,078✔
1559
  if (pCommitRaw == NULL) {
2,078!
1560
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1561
    if (terrno != 0) code = terrno;
×
1562
    return -1;
×
1563
  }
1564
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
2,078!
1565
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
2,078!
1566

1567
  SSdb *pSdb = pMnode->pSdb;
2,078✔
1568
  void *pIter = NULL;
2,078✔
1569

1570
  while (1) {
12✔
1571
    SArbGroup *pArbGroup = NULL;
2,090✔
1572
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
2,090✔
1573
    if (pIter == NULL) break;
2,090✔
1574

1575
    if (pArbGroup->dbUid == pDb->uid) {
12!
1576
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
12!
1577
        sdbCancelFetch(pSdb, pIter);
×
1578
        sdbRelease(pSdb, pArbGroup);
×
1579
        TAOS_RETURN(code);
×
1580
      }
1581
    }
1582

1583
    sdbRelease(pSdb, pArbGroup);
12✔
1584
  }
1585

1586
  while (1) {
7,473✔
1587
    SVgObj *pVgroup = NULL;
9,551✔
1588
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
9,551✔
1589
    if (pIter == NULL) break;
9,551✔
1590

1591
    if (pVgroup->dbUid == pDb->uid) {
7,473✔
1592
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
4,187✔
1593
      if (pVgRaw == NULL) {
4,187!
1594
        sdbCancelFetch(pSdb, pIter);
×
1595
        sdbRelease(pSdb, pVgroup);
×
1596
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1597
        if (terrno != 0) code = terrno;
×
1598
        TAOS_RETURN(code);
×
1599
      }
1600
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
4,187!
1601
        sdbCancelFetch(pSdb, pIter);
×
1602
        sdbRelease(pSdb, pVgroup);
×
1603
        TAOS_RETURN(code);
×
1604
      }
1605
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
4,187!
1606
        sdbCancelFetch(pSdb, pIter);
×
1607
        sdbRelease(pSdb, pVgroup);
×
1608
        TAOS_RETURN(code);
×
1609
      }
1610
    }
1611

1612
    sdbRelease(pSdb, pVgroup);
7,473✔
1613
  }
1614

1615
  while (1) {
4,559✔
1616
    SStbObj *pStb = NULL;
6,637✔
1617
    ESdbStatus status;
1618
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
6,637✔
1619
    if (pIter == NULL) break;
6,637✔
1620

1621
    if (pStb->dbUid == pDb->uid) {
4,559✔
1622
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
2,651✔
1623
      if (pStbRaw == NULL) {
2,651!
1624
        sdbCancelFetch(pSdb, pIter);
×
1625
        sdbRelease(pSdb, pStbRaw);
×
1626
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1627
        if (terrno != 0) code = terrno;
×
1628
        return -1;
×
1629
      }
1630
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
2,651!
1631
        sdbCancelFetch(pSdb, pIter);
×
1632
        sdbRelease(pSdb, pStbRaw);
×
1633
        return -1;
×
1634
      }
1635
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
2,651!
1636
        sdbCancelFetch(pSdb, pIter);
×
1637
        sdbRelease(pSdb, pStbRaw);
×
1638
        return -1;
×
1639
      }
1640
    }
1641

1642
    sdbRelease(pSdb, pStb);
4,559✔
1643
  }
1644

1645
  TAOS_RETURN(code);
2,078✔
1646
}
1647

1648
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
4,187✔
1649
  int32_t code = 0;
4,187✔
1650
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
8,532✔
1651
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
4,345✔
1652
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
4,345!
1653
  }
1654

1655
  TAOS_RETURN(code);
4,187✔
1656
}
1657

1658
int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,078✔
1659
  int32_t code = 0;
2,078✔
1660
  SSdb   *pSdb = pMnode->pSdb;
2,078✔
1661
  void   *pIter = NULL;
2,078✔
1662

1663
  while (1) {
7,473✔
1664
    SVgObj *pVgroup = NULL;
9,551✔
1665
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
9,551✔
1666
    if (pIter == NULL) break;
9,551✔
1667

1668
    if (pVgroup->dbUid == pDb->uid) {
7,473✔
1669
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
4,187!
1670
        sdbCancelFetch(pSdb, pIter);
×
1671
        sdbRelease(pSdb, pVgroup);
×
1672
        TAOS_RETURN(code);
×
1673
      }
1674
    }
1675

1676
    sdbRelease(pSdb, pVgroup);
7,473✔
1677
  }
1678

1679
  TAOS_RETURN(code);
2,078✔
1680
}
1681

1682
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
3,922✔
1683
  int32_t    code = 0;
3,922✔
1684
  SDropDbRsp dropRsp = {0};
3,922✔
1685
  if (pDb != NULL) {
3,922✔
1686
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
2,078✔
1687
    dropRsp.uid = pDb->uid;
2,078✔
1688
  }
1689

1690
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
3,922✔
1691
  void   *pRsp = NULL;
3,922✔
1692
  if (useRpcMalloc) {
3,922✔
1693
    pRsp = rpcMallocCont(rspLen);
1,844✔
1694
  } else {
1695
    pRsp = taosMemoryMalloc(rspLen);
2,078!
1696
  }
1697

1698
  if (pRsp == NULL) {
3,922!
1699
    code = terrno;
×
1700
    TAOS_RETURN(code);
×
1701
  }
1702

1703
  int32_t ret = 0;
3,922✔
1704
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
3,922!
1705
  *pRspLen = rspLen;
3,922✔
1706
  *ppRsp = pRsp;
3,922✔
1707
  TAOS_RETURN(code);
3,922✔
1708
}
1709

1710
int32_t mndRemoveAllStbUser(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,078✔
1711
  int32_t code = -1;
2,078✔
1712
  void   *pIter = NULL;
2,078✔
1713
  while (1) {
4,559✔
1714
    SStbObj   *pStb = NULL;
6,637✔
1715
    ESdbStatus status;
1716
    pIter = sdbFetchAll(pMnode->pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
6,637✔
1717
    if (pIter == NULL) break;
6,637✔
1718

1719
    if (pStb->dbUid == pDb->uid) {
4,559✔
1720
      if ((code = mndUserRemoveStb(pMnode, pTrans, pStb->name)) != 0) {
2,651!
1721
        sdbCancelFetch(pMnode->pSdb, pIter);
×
1722
        sdbRelease(pMnode->pSdb, pStb);
×
1723
        return code;
×
1724
      }
1725
    }
1726

1727
    sdbRelease(pMnode->pSdb, pStb);
4,559✔
1728
  }
1729

1730
  code = 0;
2,078✔
1731

1732
_OVER:
2,078✔
1733
  return code;
2,078✔
1734
}
1735

1736
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
2,079✔
1737
  int32_t code = -1;
2,079✔
1738

1739
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
2,079✔
1740
  if (pTrans == NULL) {
2,079!
1741
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1742
    if (terrno != 0) code = terrno;
×
1743
    goto _OVER;
×
1744
  }
1745

1746
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
2,079!
1747

1748
  mndTransSetDbName(pTrans, pDb->name, NULL);
2,079✔
1749
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
2,079!
1750

1751
  if (mndTopicExistsForDb(pMnode, pDb)) {
2,079✔
1752
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
1✔
1753
    goto _OVER;
1✔
1754
  }
1755

1756
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1757
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1758
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1759
#ifdef TD_ENTERPRISE
1760
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1761
#endif
1762
  TAOS_CHECK_GOTO(mndDropSmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1763
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1764
  TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
2,078!
1765
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1766
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb->name), NULL, _OVER);
2,078!
1767
  TAOS_CHECK_GOTO(mndRemoveAllStbUser(pMnode, pTrans, pDb), NULL, _OVER);
2,078!
1768

1769
  int32_t rspLen = 0;
2,078✔
1770
  void   *pRsp = NULL;
2,078✔
1771
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
2,078!
1772
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
2,078✔
1773

1774
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
2,078!
1775
  code = 0;
2,078✔
1776

1777
_OVER:
2,079✔
1778
  mndTransDrop(pTrans);
2,079✔
1779
  TAOS_RETURN(code);
2,079✔
1780
}
1781

1782
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
4,010✔
1783
  SMnode    *pMnode = pReq->info.node;
4,010✔
1784
  int32_t    code = -1;
4,010✔
1785
  SDbObj    *pDb = NULL;
4,010✔
1786
  SDropDbReq dropReq = {0};
4,010✔
1787

1788
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
4,010!
1789

1790
  mInfo("db:%s, start to drop", dropReq.db);
4,010!
1791

1792
  pDb = mndAcquireDb(pMnode, dropReq.db);
4,010✔
1793
  if (pDb == NULL) {
4,010✔
1794
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
1,914✔
1795
    if (terrno != 0) code = terrno;
1,914!
1796
    if (dropReq.ignoreNotExists) {
1,914✔
1797
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
1,844✔
1798
    }
1799
    goto _OVER;
1,914✔
1800
  }
1801

1802
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_DB, pDb), NULL, _OVER);
2,096✔
1803

1804
  if(pDb->cfg.isMount) {
2,081!
1805
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
1806
    goto _OVER;
×
1807
  }
1808

1809
  SSdb *pSdb = pMnode->pSdb;
2,081✔
1810
  void *pIter = NULL;
2,081✔
1811

1812
  while (1) {
7,475✔
1813
    SVgObj *pVgroup = NULL;
9,556✔
1814
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
9,556✔
1815
    if (pIter == NULL) break;
9,556✔
1816

1817
    if (pVgroup->dbUid == pDb->uid) {
7,477✔
1818
      bool isFound = false;
4,190✔
1819
      for (int32_t i = 0; i < pVgroup->replica; i++) {
8,536✔
1820
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
4,348✔
1821
          isFound = true;
2✔
1822
          break;
2✔
1823
        }
1824
      }
1825
      if (!isFound) {
4,190✔
1826
        sdbRelease(pSdb, pVgroup);
4,188✔
1827
        continue;
4,188✔
1828
      }
1829
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
2✔
1830
      sdbCancelFetch(pSdb, pIter);
2✔
1831
      sdbRelease(pSdb, pVgroup);
2✔
1832
      goto _OVER;
2✔
1833
    }
1834

1835
    sdbRelease(pSdb, pVgroup);
3,287✔
1836
  }
1837

1838
  code = mndDropDb(pMnode, pReq, pDb);
2,079✔
1839
  if (code == TSDB_CODE_SUCCESS) {
2,079✔
1840
    code = TSDB_CODE_ACTION_IN_PROGRESS;
2,078✔
1841
  }
1842

1843
  SName name = {0};
2,079✔
1844
  if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
2,079!
1845
    mError("db:%s, failed to parse db name", dropReq.db);
×
1846

1847
  auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen);
2,079✔
1848

1849
_OVER:
4,010✔
1850
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
4,010✔
1851
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
88!
1852
  }
1853

1854
  mndReleaseDb(pMnode, pDb);
4,010✔
1855
  tFreeSDropDbReq(&dropReq);
4,010✔
1856
  TAOS_RETURN(code);
4,010✔
1857
}
1858

1859
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
151,811✔
1860
  int32_t numOfTables = 0;
151,811✔
1861
  int32_t vindex = 0;
151,811✔
1862
  SSdb   *pSdb = pMnode->pSdb;
151,811✔
1863

1864
  void *pIter = NULL;
151,811✔
1865
  while (vindex < pDb->cfg.numOfVgroups) {
556,180✔
1866
    SVgObj *pVgroup = NULL;
404,363✔
1867
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
404,363✔
1868
    if (pIter == NULL) break;
404,369!
1869

1870
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
404,369✔
1871
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
308,279✔
1872
      vindex++;
308,279✔
1873
    }
1874

1875
    sdbRelease(pSdb, pVgroup);
404,369✔
1876
  }
1877

1878
  sdbCancelFetch(pSdb, pIter);
151,817✔
1879
  return numOfTables;
151,810✔
1880
}
1881

1882
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
164,562✔
1883
  int32_t vindex = 0;
164,562✔
1884
  SSdb   *pSdb = pMnode->pSdb;
164,562✔
1885

1886
  void *pIter = NULL;
164,562✔
1887
  while (1) {
287,606✔
1888
    SVgObj *pVgroup = NULL;
452,168✔
1889
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
452,168✔
1890
    if (pIter == NULL) break;
452,166✔
1891

1892
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
431,265!
1893
      SVgroupInfo vgInfo = {0};
348,764✔
1894
      vgInfo.vgId = pVgroup->vgId;
348,764✔
1895
      vgInfo.hashBegin = pVgroup->hashBegin;
348,764✔
1896
      vgInfo.hashEnd = pVgroup->hashEnd;
348,764✔
1897
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
348,764✔
1898
      vgInfo.epSet.numOfEps = pVgroup->replica;
348,764✔
1899
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
710,854✔
1900
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
362,088✔
1901
        SEp       *pEp = &vgInfo.epSet.eps[gid];
362,088✔
1902
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
362,088✔
1903
        if (pDnode != NULL) {
362,089✔
1904
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
361,974✔
1905
          pEp->port = pDnode->port;
361,974✔
1906
        }
1907
        mndReleaseDnode(pMnode, pDnode);
362,089✔
1908
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
362,090✔
1909
          vgInfo.epSet.inUse = gid;
347,134✔
1910
        }
1911
      }
1912
      vindex++;
348,766✔
1913
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
348,763!
1914
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
1915
      }
1916
    }
1917

1918
    sdbRelease(pSdb, pVgroup);
431,264✔
1919

1920
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
431,267✔
1921
      sdbCancelFetch(pSdb, pIter);
143,661✔
1922
      break;
143,661✔
1923
    }
1924
  }
1925
}
164,562✔
1926

1927
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
143,255✔
1928
  int32_t code = 0;
143,255✔
1929
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
143,255✔
1930
  if (pRsp->pVgroupInfos == NULL) {
143,255!
1931
    code = terrno;
×
1932
    TAOS_RETURN(code);
×
1933
  }
1934

1935
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
143,255✔
1936

1937
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
143,254!
1938
      pReq->stateTs < pDb->stateTs) {
2,836✔
1939
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
140,838✔
1940
  }
1941

1942
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
143,254✔
1943
  pRsp->uid = pDb->uid;
143,254✔
1944
  pRsp->vgVersion = pDb->vgVersion;
143,254✔
1945
  pRsp->stateTs = pDb->stateTs;
143,254✔
1946
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
143,254✔
1947
  pRsp->hashMethod = pDb->cfg.hashMethod;
143,254✔
1948
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
143,254✔
1949
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
143,254✔
1950
  pRsp->flags = pDb->cfg.flags;
143,254✔
1951
  TAOS_RETURN(code);
143,254✔
1952
}
1953

1954
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
161,567✔
1955
  SMnode   *pMnode = pReq->info.node;
161,567✔
1956
  int32_t   code = -1;
161,567✔
1957
  SDbObj   *pDb = NULL;
161,567✔
1958
  SUseDbReq usedbReq = {0};
161,567✔
1959
  SUseDbRsp usedbRsp = {0};
161,567✔
1960

1961
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
161,567!
1962

1963
  char *p = strchr(usedbReq.db, '.');
161,568✔
1964
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
161,568!
1965
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
20,225✔
1966
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
20,225✔
1967
    if (usedbReq.vgVersion < vgVersion) {
20,225✔
1968
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
20,187✔
1969
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
20,187!
1970

1971
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
20,187✔
1972
      usedbRsp.vgVersion = vgVersion++;
20,187✔
1973
    } else {
1974
      usedbRsp.vgVersion = usedbReq.vgVersion;
38✔
1975
    }
1976
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
20,225✔
1977
    code = 0;
20,225✔
1978
  } else {
1979
    pDb = mndAcquireDb(pMnode, usedbReq.db);
141,343✔
1980
    if (pDb == NULL) {
141,343✔
1981
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
707✔
1982
      usedbRsp.uid = usedbReq.dbId;
707✔
1983
      usedbRsp.vgVersion = usedbReq.vgVersion;
707✔
1984
      usedbRsp.errCode = terrno;
707✔
1985

1986
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
707✔
1987
      if (terrno != 0) code = terrno;
707!
1988
      goto _OVER;
707✔
1989
    } else {
1990
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_USE_DB, pDb), NULL, _OVER);
140,636✔
1991

1992
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
140,620!
1993

1994
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
140,618✔
1995
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
1996
      code = 0;
140,620✔
1997
    }
1998
  }
1999

2000
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
160,845✔
2001
  void   *pRsp = rpcMallocCont(contLen);
160,845✔
2002
  if (pRsp == NULL) {
160,844!
2003
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2004
    code = -1;
×
2005
    goto _OVER;
×
2006
  }
2007

2008
  int32_t ret = 0;
160,844✔
2009
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
160,844!
2010
    code = ret;
×
2011
    goto _OVER;
×
2012
  }
2013

2014
  pReq->info.rsp = pRsp;
160,845✔
2015
  pReq->info.rspLen = contLen;
160,845✔
2016

2017
_OVER:
161,569✔
2018
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
161,569!
2019
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
724!
2020
  }
2021

2022
  mndReleaseDb(pMnode, pDb);
161,569✔
2023
  tFreeSUsedbRsp(&usedbRsp);
161,569✔
2024

2025
  TAOS_RETURN(code);
161,569✔
2026
}
2027

2028
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
11,639✔
2029
  int32_t       code = 0;
11,639✔
2030
  SDbHbBatchRsp batchRsp = {0};
11,639✔
2031
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
11,639✔
2032
  if (batchRsp.pArray == NULL) {
11,639!
2033
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2034
    TAOS_RETURN(code);
×
2035
  }
2036

2037
  for (int32_t i = 0; i < numOfDbs; ++i) {
25,325✔
2038
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
13,686✔
2039
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
13,686✔
2040
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
13,686✔
2041
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
13,686✔
2042
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
13,686✔
2043
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
13,686✔
2044
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
13,686✔
2045

2046
    SDbHbRsp rsp = {0};
13,686✔
2047
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
13,686✔
2048
    rsp.dbId = pDbCacheInfo->dbId;
13,686✔
2049

2050
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
13,686✔
2051
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
13,076✔
2052
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
4,876✔
2053
      if (pDbCacheInfo->vgVersion >= vgVersion) {
4,876✔
2054
        continue;
9,828✔
2055
      }
2056

2057
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
714!
2058
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
714✔
2059
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
714✔
2060

2061
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
714✔
2062
      rsp.useDbRsp->vgVersion = vgVersion++;
714✔
2063

2064
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
714✔
2065

2066
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
1,428!
2067
        if (terrno != 0) code = terrno;
×
2068
        return code;
×
2069
      }
2070

2071
      continue;
714✔
2072
    }
2073

2074
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
8,810✔
2075
    if (pDb == NULL) {
8,810✔
2076
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
254✔
2077
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
254!
2078
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
254✔
2079
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
254✔
2080
      rsp.useDbRsp->vgVersion = -1;
254✔
2081
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
508!
2082
        if (terrno != 0) code = terrno;
×
2083
        return code;
×
2084
      }
2085
      continue;
254✔
2086
    }
2087

2088
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
8,556✔
2089

2090
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
8,556✔
2091
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
7,956!
2092
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
5,409✔
2093
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
4,698✔
2094
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2095
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2096
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2097
      mndReleaseDb(pMnode, pDb);
4,698✔
2098
      continue;
4,698✔
2099
    } else {
2100
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
3,858✔
2101
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2102
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2103
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2104
    }
2105

2106
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
3,858✔
2107
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
554!
2108
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb);
554✔
2109
    }
2110

2111
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
3,858✔
2112
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
2,534!
2113
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
2,534!
2114
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
2,534!
2115
        bool    exist = false;
2,534✔
2116
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
2,534✔
2117
        if (TSDB_CODE_SUCCESS != code) {
2,534!
2118
          mndReleaseDb(pMnode, pDb);
×
2119
          if (code != TSDB_CODE_NEED_RETRY) {
×
2120
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2121
          } else {
2122
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2123
          }
2124
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2125
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
2126
          continue;
×
2127
        }
2128
        rsp.dbTsmaVersion = pDb->tsmaVersion;
2,534✔
2129
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
2,534✔
2130
      }
2131
    }
2132

2133
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
3,858!
2134
        pDbCacheInfo->stateTs != pDb->stateTs) {
3,805✔
2135
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
2,823!
2136
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
2,823✔
2137
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
2,823!
2138
        mndReleaseDb(pMnode, pDb);
×
2139
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
2140
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2141
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2142
        continue;
×
2143
      }
2144

2145
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
2,823✔
2146
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
2,823✔
2147
      rsp.useDbRsp->uid = pDb->uid;
2,823✔
2148
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
2,823✔
2149
      rsp.useDbRsp->stateTs = pDb->stateTs;
2,823✔
2150
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
2,823✔
2151
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
2,823✔
2152
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
2,823✔
2153
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
2,823✔
2154
      rsp.useDbRsp->flags = pDb->cfg.flags;
2,823✔
2155
    }
2156

2157
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
7,716!
2158
      mndReleaseDb(pMnode, pDb);
×
2159
      if (terrno != 0) code = terrno;
×
2160
      return code;
×
2161
    }
2162
    mndReleaseDb(pMnode, pDb);
3,858✔
2163
  }
2164

2165
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
11,639✔
2166
  void   *pRsp = taosMemoryMalloc(rspLen);
11,639!
2167
  if (pRsp == NULL) {
11,639!
2168
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2169
    tFreeSDbHbBatchRsp(&batchRsp);
×
2170
    return -1;
×
2171
  }
2172
  int32_t ret = 0;
11,639✔
2173
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
11,639!
2174

2175
  *ppRsp = pRsp;
11,639✔
2176
  *pRspLen = rspLen;
11,639✔
2177

2178
  tFreeSDbHbBatchRsp(&batchRsp);
11,639✔
2179
  TAOS_RETURN(code);
11,639✔
2180
}
2181

2182
static int32_t mndTrimDb(SMnode *pMnode, SDbObj *pDb) {
3✔
2183
  SSdb       *pSdb = pMnode->pSdb;
3✔
2184
  SVgObj     *pVgroup = NULL;
3✔
2185
  void       *pIter = NULL;
3✔
2186
  int32_t     code = 0;
3✔
2187
  SVTrimDbReq trimReq = {.timestamp = taosGetTimestampSec()};
3✔
2188
  int32_t     reqLen = tSerializeSVTrimDbReq(NULL, 0, &trimReq);
3✔
2189
  int32_t     contLen = reqLen + sizeof(SMsgHead);
3✔
2190

2191
  while (1) {
4✔
2192
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
7✔
2193
    if (pIter == NULL) break;
7✔
2194

2195
    if (pVgroup->dbUid != pDb->uid) {
4!
2196
      sdbRelease(pSdb, pVgroup);
×
2197
      continue;
×
2198
    }
2199

2200
    SMsgHead *pHead = rpcMallocCont(contLen);
4✔
2201
    if (pHead == NULL) {
4!
2202
      sdbCancelFetch(pSdb, pVgroup);
×
2203
      sdbRelease(pSdb, pVgroup);
×
2204
      continue;
×
2205
    }
2206
    pHead->contLen = htonl(contLen);
4✔
2207
    pHead->vgId = htonl(pVgroup->vgId);
4✔
2208
    int32_t ret = 0;
4✔
2209
    if ((ret = tSerializeSVTrimDbReq((char *)pHead + sizeof(SMsgHead), contLen, &trimReq)) < 0) {
4!
2210
      sdbRelease(pSdb, pVgroup);
×
2211
      return ret;
×
2212
    }
2213

2214
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM, .pCont = pHead, .contLen = contLen};
4✔
2215
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
4✔
2216
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
4✔
2217
    if (code != 0) {
4!
2218
      mError("vgId:%d, failed to send vnode-trim request to vnode since 0x%x", pVgroup->vgId, code);
×
2219
    } else {
2220
      mInfo("vgId:%d, send vnode-trim request to vnode, time:%d", pVgroup->vgId, trimReq.timestamp);
4!
2221
    }
2222
    sdbRelease(pSdb, pVgroup);
4✔
2223
  }
2224

2225
  return 0;
3✔
2226
}
2227

2228
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
3✔
2229
  SMnode    *pMnode = pReq->info.node;
3✔
2230
  int32_t    code = -1;
3✔
2231
  SDbObj    *pDb = NULL;
3✔
2232
  STrimDbReq trimReq = {0};
3✔
2233

2234
  TAOS_CHECK_GOTO(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq), NULL, _OVER);
3!
2235

2236
  mInfo("db:%s, start to trim", trimReq.db);
3!
2237

2238
  pDb = mndAcquireDb(pMnode, trimReq.db);
3✔
2239
  if (pDb == NULL) {
3!
2240
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2241
    if (terrno != 0) code = terrno;
×
2242
    goto _OVER;
×
2243
  }
2244

2245
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
3!
2246

2247
  if (pDb->cfg.isMount) {
3!
2248
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
2249
    goto _OVER;
×
2250
  }
2251

2252
  code = mndTrimDb(pMnode, pDb);
3✔
2253

2254
_OVER:
3✔
2255
  if (code != 0) {
3!
2256
    mError("db:%s, failed to process trim db req since %s", trimReq.db, tstrerror(code));
×
2257
  }
2258

2259
  mndReleaseDb(pMnode, pDb);
3✔
2260
  TAOS_RETURN(code);
3✔
2261
}
2262

2263
static int32_t mndSetSsMigrateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t ssMigrateTs) {
×
2264
  int32_t code = 0;
×
2265
  SDbObj  dbObj = {0};
×
2266
  memcpy(&dbObj, pDb, sizeof(SDbObj));
×
2267
  dbObj.ssMigrateStartTime = ssMigrateTs;
×
2268

2269
  SSdbRaw *pCommitRaw = mndDbActionEncode(&dbObj);
×
2270
  if (pCommitRaw == NULL) {
×
2271
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2272
    if (terrno != 0) code = terrno;
×
2273
    TAOS_RETURN(code);
×
2274
  }
2275
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
×
2276
    sdbFreeRaw(pCommitRaw);
×
2277
    TAOS_RETURN(code);
×
2278
  }
2279

2280
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY));
×
2281
  TAOS_RETURN(code);
×
2282
}
2283

2284
static int32_t mndSetSsMigrateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t ssMigrateTs, SSsMigrateDbRsp *pSsMigrateRsp) {
×
2285
  int32_t code = 0;
×
2286
  SSdb   *pSdb = pMnode->pSdb;
×
2287
  void   *pIter = NULL;
×
2288

2289
  // startTime is in seconds, so convert milliseconds to seconds
2290
  SSsMigrateObj ssMigrate = { .startTime = ssMigrateTs/1000 };
×
2291
  if ((code = mndAddSsMigrateToTran(pMnode, pTrans, &ssMigrate, pDb)) != 0) {
×
2292
    TAOS_RETURN(code);
×
2293
  }
2294
  pSsMigrateRsp->ssMigrateId = ssMigrate.id;
×
2295

2296
  int32_t j = 0;
×
2297
  while (1) {
×
2298
    SVgObj *pVgroup = NULL;
×
2299
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
×
2300
    if (pIter == NULL) break;
×
2301

2302
    if (pVgroup->mountVgId || pVgroup->dbUid != pDb->uid) {
×
2303
      sdbRelease(pSdb, pVgroup);
×
2304
      continue;
×
2305
    }
2306

2307
    if ((code = mndBuildSsMigrateVgroupAction(pMnode, pTrans, pDb, pVgroup, &ssMigrate)) != 0) {
×
2308
      sdbCancelFetch(pSdb, pIter);
×
2309
      sdbRelease(pSdb, pVgroup);
×
2310
      TAOS_RETURN(code);
×
2311
    }
2312

2313
    sdbRelease(pSdb, pVgroup);
×
2314
  }
2315

2316
  TAOS_RETURN(code);
×
2317
}
2318

2319
static int32_t mndBuildSsMigrateDbRsp(SSsMigrateDbRsp *pSsMigrateRsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
×
2320
  int32_t code = 0;
×
2321
  int32_t rspLen = tSerializeSSsMigrateDbRsp(NULL, 0, pSsMigrateRsp);
×
2322
  void   *pRsp = NULL;
×
2323
  if (useRpcMalloc) {
×
2324
    pRsp = rpcMallocCont(rspLen);
×
2325
  } else {
2326
    pRsp = taosMemoryMalloc(rspLen);
×
2327
  }
2328

2329
  if (pRsp == NULL) {
×
2330
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2331
    TAOS_RETURN(code);
×
2332
  }
2333

2334
  (void)tSerializeSSsMigrateDbRsp(pRsp, rspLen, pSsMigrateRsp);
×
2335
  *pRspLen = rspLen;
×
2336
  *ppRsp = pRsp;
×
2337
  TAOS_RETURN(code);
×
2338
}
2339

2340
int32_t mndSsMigrateDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
×
2341
  int32_t       code = 0;
×
2342
  SSsMigrateDbRsp ssMigrateRsp = {0};
×
2343

2344
  bool  isExist = false;
×
2345
  void *pIter = NULL;
×
2346
  while (1) {
×
2347
    SSsMigrateObj *pSsMigrate = NULL;
×
2348
    pIter = sdbFetch(pMnode->pSdb, SDB_SSMIGRATE, pIter, (void **)&pSsMigrate);
×
2349
    if (pIter == NULL) break;
×
2350

2351
    if (strcmp(pSsMigrate->dbname, pDb->name) == 0) {
×
2352
      isExist = true;
×
2353
    }
2354
    sdbRelease(pMnode->pSdb, pSsMigrate);
×
2355
  }
2356
  if (isExist) {
×
2357
    mInfo("ssmigrate db:%s already exist", pDb->name);
×
2358

2359
    if (pReq) {
×
2360
      int32_t rspLen = 0;
×
2361
      void   *pRsp = NULL;
×
2362
      ssMigrateRsp.ssMigrateId = 0;
×
2363
      ssMigrateRsp.bAccepted = false;
×
2364
      TAOS_CHECK_RETURN(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, true));
×
2365

2366
      pReq->info.rsp = pRsp;
×
2367
      pReq->info.rspLen = rspLen;
×
2368
    }
2369

2370
    return TSDB_CODE_MND_SSMIGRATE_ALREADY_EXIST;
×
2371
  }
2372

2373
  int64_t ssMigrateTs = taosGetTimestampMs();
×
2374
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "ssmigrate-db");
×
2375
  if (pTrans == NULL) {
×
2376
    mError("failed to create ssmigrate-db trans since %s", terrstr());
×
2377
    goto _OVER;
×
2378
  }
2379

2380
  mInfo("trans:%d, used to ssmigrate db:%s", pTrans->id, pDb->name);
×
2381
  mndTransSetDbName(pTrans, pDb->name, NULL);
×
2382
  TAOS_CHECK_GOTO(mndTrancCheckConflict(pMnode, pTrans), NULL, _OVER);
×
2383

2384
  TAOS_CHECK_GOTO(mndSetSsMigrateDbCommitLogs(pMnode, pTrans, pDb, ssMigrateTs), NULL, _OVER);
×
2385
  TAOS_CHECK_GOTO(mndSetSsMigrateDbRedoActions(pMnode, pTrans, pDb, ssMigrateTs, &ssMigrateRsp), NULL, _OVER);
×
2386

2387
  if (pReq) {
×
2388
    int32_t rspLen = 0;
×
2389
    void   *pRsp = NULL;
×
2390
    ssMigrateRsp.bAccepted = true;
×
2391
    TAOS_CHECK_GOTO(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, false), NULL, _OVER);
×
2392
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
×
2393
  }
2394

2395
  if (mndTransPrepare(pMnode, pTrans) != 0) goto _OVER;
×
2396
  code = 0;
×
2397

2398
_OVER:
×
2399
  mndTransDrop(pTrans);
×
2400
  TAOS_RETURN(code);
×
2401
}
2402

2403

2404
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq) {
×
2405
  SMnode         *pMnode = pReq->info.node;
×
2406
  int32_t         code = -1;
×
2407
  SDbObj         *pDb = NULL;
×
2408
  SSsMigrateDbReq ssMigrateReq = {0};
×
2409

2410
  if (!tsSsEnabled) {
×
2411
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
2412
    goto _OVER;
×
2413
  }
2414

2415
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_SHARED_STORAGE), NULL, _OVER);
×
2416
  TAOS_CHECK_GOTO(tDeserializeSSsMigrateDbReq(pReq->pCont, pReq->contLen, &ssMigrateReq), NULL, _OVER);
×
2417

2418
  mInfo("db:%s, start to ssmigrate", ssMigrateReq.db);
×
2419

2420
  pDb = mndAcquireDb(pMnode, ssMigrateReq.db);
×
2421
  if (pDb == NULL) {
×
2422
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2423
    if (terrno != 0) code = terrno;
×
2424
    goto _OVER;
×
2425
  }
2426

2427
  // TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
2428

2429
  if(pDb->cfg.isMount) {
×
2430
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
2431
    goto _OVER;
×
2432
  }
2433

2434
  code = mndSsMigrateDb(pMnode, pReq, pDb);
×
2435

2436
_OVER:
×
2437
  if (code != 0) {
×
2438
    mError("db:%s, failed to process ssmigrate db req since %s", ssMigrateReq.db, tstrerror(code));
×
2439
  }
2440

2441
  mndReleaseDb(pMnode, pDb);
×
2442
  TAOS_RETURN(code);
×
2443
}
2444

2445
const char *mndGetDbStr(const char *src) {
53,685✔
2446
  char *pos = strstr(src, TS_PATH_DELIMITER);
53,685✔
2447
  if (pos != NULL) ++pos;
53,685✔
2448
  if (pos == NULL) return src;
53,685✔
2449
  return pos;
50,966✔
2450
}
2451

2452
const char *mndGetStableStr(const char *src) {
×
2453
  char *pos = strstr(src, TS_PATH_DELIMITER);
×
2454
  if (pos != NULL) ++pos;
×
2455
  if (pos == NULL) return src;
×
2456
  return mndGetDbStr(pos);
×
2457
}
2458

2459
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
12✔
2460
  int64_t v = 0;
12✔
2461
  switch (unit) {
12!
2462
    case 's':
4✔
2463
      v = val / 1000;
4✔
2464
      break;
4✔
2465
    case 'm':
2✔
2466
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
2✔
2467
      break;
2✔
2468
    case 'h':
×
2469
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2470
      break;
×
2471
    case 'd':
6✔
2472
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
6✔
2473
      break;
6✔
2474
    case 'w':
×
2475
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2476
      break;
×
2477
    default:
×
2478
      break;
×
2479
  }
2480

2481
  return v;
12✔
2482
}
2483

2484
static char *buildRetension(SArray *pRetension) {
1,096✔
2485
  size_t size = taosArrayGetSize(pRetension);
1,096✔
2486
  if (size == 0) {
1,096✔
2487
    return NULL;
1,094✔
2488
  }
2489

2490
  char       *p1 = taosMemoryCalloc(1, 100);
2!
2491
  SRetention *p = taosArrayGet(pRetension, 0);
2✔
2492

2493
  int32_t len = 2;
2✔
2494

2495
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2496
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2497
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2498

2499
  if (size > 1) {
2!
2500
    len += tsnprintf(p1 + len, 100 - len, ",");
2✔
2501
    p = taosArrayGet(pRetension, 1);
2✔
2502

2503
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2504
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2505
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2506
  }
2507

2508
  if (size > 2) {
2!
2509
    len += tsnprintf(p1 + len, 100 - len, ",");
2✔
2510
    p = taosArrayGet(pRetension, 2);
2✔
2511

2512
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2513
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2514
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2515
  }
2516

2517
  varDataSetLen(p1, len);
2✔
2518
  return p1;
2✔
2519
}
2520

2521
static const char *getCacheModelStr(int8_t cacheModel) {
1,096✔
2522
  switch (cacheModel) {
1,096!
2523
    case TSDB_CACHE_MODEL_NONE:
1,063✔
2524
      return TSDB_CACHE_MODEL_NONE_STR;
1,063✔
2525
    case TSDB_CACHE_MODEL_LAST_ROW:
4✔
2526
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
4✔
2527
    case TSDB_CACHE_MODEL_LAST_VALUE:
10✔
2528
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
10✔
2529
    case TSDB_CACHE_MODEL_BOTH:
19✔
2530
      return TSDB_CACHE_MODEL_BOTH_STR;
19✔
2531
    default:
×
2532
      break;
×
2533
  }
2534
  return "unknown";
×
2535
}
2536

2537
static const char *getEncryptAlgorithmStr(int8_t encryptAlgorithm) {
1,096✔
2538
  switch (encryptAlgorithm) {
1,096!
2539
    case TSDB_ENCRYPT_ALGO_NONE:
1,096✔
2540
      return TSDB_ENCRYPT_ALGO_NONE_STR;
1,096✔
2541
    case TSDB_ENCRYPT_ALGO_SM4:
×
2542
      return TSDB_ENCRYPT_ALGO_SM4_STR;
×
2543
    default:
×
2544
      break;
×
2545
  }
2546
  return "unknown";
×
2547
}
2548

2549
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
6,808✔
2550
  if (pDb->cfg.replications == 1) return true;
6,808✔
2551

2552
  SSdb *pSdb = pMnode->pSdb;
1,366✔
2553
  void *pIter = NULL;
1,366✔
2554
  bool  isReady = true;
1,366✔
2555
  while (1) {
8,281✔
2556
    SVgObj *pVgroup = NULL;
9,647✔
2557
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
9,647✔
2558
    if (pIter == NULL) break;
9,647✔
2559

2560
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
8,281✔
2561
      bool hasLeader = false;
3,833✔
2562
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
15,232✔
2563
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
11,399✔
2564
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
10,182!
2565
          hasLeader = true;
1,217✔
2566
        }
2567
      }
2568
      if (!hasLeader) isReady = false;
3,833✔
2569
    }
2570
    sdbRelease(pSdb, pVgroup);
8,281✔
2571
  }
2572

2573
  return isReady;
1,366✔
2574
}
2575

2576
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
3,642✔
2577
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2578
  int32_t cols = 0;
3,642✔
2579
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
3,642✔
2580
  char   *buf = taosMemoryMalloc(bytes);
3,642!
2581
  if (buf == NULL) {
3,642!
2582
    mError("db:%s, failed to malloc buffer", pDb->name);
×
2583
    return;
×
2584
  }
2585
  int32_t code = 0;
3,642✔
2586
  int32_t lino = 0;
3,642✔
2587

2588
  const char *name = mndGetDbStr(pDb->name);
3,642✔
2589
  if (name != NULL) {
3,642!
2590
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
3,642✔
2591
  } else {
2592
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
2593
  }
2594

2595
  const char *precStr = NULL;
3,642✔
2596
  switch (pDb->cfg.precision) {
3,642!
2597
    case TSDB_TIME_PRECISION_MILLI:
3,602✔
2598
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
3,602✔
2599
      break;
3,602✔
2600
    case TSDB_TIME_PRECISION_MICRO:
12✔
2601
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
12✔
2602
      break;
12✔
2603
    case TSDB_TIME_PRECISION_NANO:
28✔
2604
      precStr = TSDB_TIME_PRECISION_NANO_STR;
28✔
2605
      break;
28✔
2606
    default:
×
2607
      precStr = "none";
×
2608
      break;
×
2609
  }
2610
  char precVstr[10] = {0};
3,642✔
2611
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
3,642✔
2612

2613
  char *statusStr = "ready";
3,642✔
2614
  if (objStatus == SDB_STATUS_CREATING) {
3,642!
2615
    statusStr = "creating";
×
2616
  } else if (objStatus == SDB_STATUS_DROPPING) {
3,642!
2617
    statusStr = "dropping";
×
2618
  } else {
2619
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
3,642✔
2620
      statusStr = "unsynced";
10✔
2621
    }
2622
  }
2623
  char statusVstr[24] = {0};
3,642✔
2624
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
3,642✔
2625

2626
  if (sysDb || !sysinfo) {
3,642✔
2627
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
96,748✔
2628
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
94,202✔
2629
      if (i == 0) {
94,202✔
2630
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
2,546!
2631
      } else if (i == 1) {
91,656✔
2632
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
2,546!
2633
      } else if (i == 3) {
89,110✔
2634
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
2,546!
2635
      } else if (i == 14) {
86,564✔
2636
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
2,546!
2637
      } else if (i == 15) {
84,018✔
2638
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
2,546!
2639
      } else {
2640
        colDataSetNULL(pColInfo, rows);
81,472✔
2641
      }
2642
    }
2643
  } else {
2644
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2645
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
1,096!
2646

2647
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2648
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
1,096!
2649

2650
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2651
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
1,096!
2652

2653
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2654
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
1,096!
2655

2656
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2657
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
1,096!
2658

2659
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
1,096!
2660
    char        strictVstr[24] = {0};
1,096✔
2661
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
1,096✔
2662
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2663
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
1,096!
2664

2665
    char    durationStr[128] = {0};
1,096✔
2666
    char    durationVstr[128] = {0};
1,096✔
2667
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
1,096✔
2668
                                       pDb->cfg.daysPerFile);
2669

2670
    varDataSetLen(durationVstr, len);
1,096✔
2671
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2672
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,096!
2673

2674
    char keepVstr[128] = {0};
1,096✔
2675
    char keep0Str[32] = {0};
1,096✔
2676
    char keep1Str[32] = {0};
1,096✔
2677
    char keep2Str[32] = {0};
1,096✔
2678

2679
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
1,096✔
2680
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
1,096✔
2681
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
1,096✔
2682

2683
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
1,096!
2684
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
2685
    } else {
2686
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
1,096✔
2687
    }
2688
    varDataSetLen(keepVstr, len);
1,096✔
2689
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2690
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
1,096!
2691

2692
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2693
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
1,096!
2694

2695
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2696
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
1,096!
2697

2698
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2699
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
1,096!
2700

2701
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2702
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
1,096!
2703

2704
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2705
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
1,096!
2706

2707
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2708
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
1,096!
2709

2710
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2711
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
1,096!
2712

2713
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2714
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
1,096!
2715

2716
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
1,096✔
2717
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2718
    if (rentensionVstr == NULL) {
1,096✔
2719
      colDataSetNULL(pColInfo, rows);
1,094!
2720
    } else {
2721
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
2!
2722
      taosMemoryFree(rentensionVstr);
2!
2723
    }
2724

2725
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2726
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
1,096!
2727

2728
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
1,096✔
2729
    char        cacheModelVstr[24] = {0};
1,096✔
2730
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
1,096✔
2731
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2732
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
1,096!
2733

2734
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2735
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
1,096!
2736

2737
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2738
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
1,096!
2739

2740
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2741
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
1,096!
2742

2743
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2744
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
1,096!
2745

2746
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2747
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
1,096!
2748

2749
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2750
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
1,096!
2751

2752
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2753
    int16_t hashPrefix = pDb->cfg.hashPrefix;
1,096✔
2754
    if (hashPrefix > 0) {
1,096✔
2755
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
6✔
2756
    } else if (hashPrefix < 0) {
1,090✔
2757
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
16✔
2758
    }
2759
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
1,096!
2760

2761
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2762
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
1,096!
2763

2764
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2765
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
1,096!
2766

2767
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2768
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
1,096!
2769

2770
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2771
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssChunkSize, false), &lino, _OVER);
1,096!
2772

2773
    char keeplocalVstr[128] = {0};
1,096✔
2774
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.ssKeepLocal);
1,096✔
2775
    varDataSetLen(keeplocalVstr, len);
1,096✔
2776
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2777
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
1,096!
2778

2779
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2780
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssCompact, false), &lino, _OVER);
1,096!
2781

2782
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2783
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
1,096!
2784

2785
    const char *encryptAlgorithmStr = getEncryptAlgorithmStr(pDb->cfg.encryptAlgorithm);
1,096✔
2786
    char        encryptAlgorithmVStr[24] = {0};
1,096✔
2787
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
1,096✔
2788
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,096✔
2789
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
1,096!
2790

2791
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
1,096✔
2792
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,096✔
2793
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,096!
2794
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,096!
2795
    }
2796

2797
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
1,096✔
2798
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
1,096✔
2799
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
1,096✔
2800
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,096✔
2801
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,096!
2802
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,096!
2803
    }
2804

2805
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
1,096✔
2806
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,096✔
2807
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,096!
2808
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,096!
2809
    }
2810
  }
2811
_OVER:
3,642✔
2812
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
3,642!
2813
  taosMemoryFree(buf);
3,642!
2814
}
2815

2816
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
1,267✔
2817
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
1,267✔
2818
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
1,267✔
2819
  pDbObj->cfg.numOfVgroups = 0;
1,267✔
2820
  pDbObj->cfg.strict = 1;
1,267✔
2821
  pDbObj->cfg.replications = 1;
1,267✔
2822
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
1,267✔
2823
}
1,267✔
2824

2825
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
1,267✔
2826
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
1,267✔
2827
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
1,267✔
2828
  pDbObj->cfg.numOfVgroups = 0;
1,267✔
2829
  pDbObj->cfg.strict = 1;
1,267✔
2830
  pDbObj->cfg.replications = 1;
1,267✔
2831
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
1,267✔
2832
}
1,267✔
2833

2834
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
4,985✔
2835
  SVgObj  *pVgroup = pObj;
4,985✔
2836
  int32_t *numOfTables = p1;
4,985✔
2837
  int64_t  uid = *(int64_t *)p2;
4,985✔
2838
  if (pVgroup->dbUid == uid) {
4,985✔
2839
    *numOfTables += pVgroup->numOfTables;
2,309✔
2840
  }
2841
  return true;
4,985✔
2842
}
2843

2844
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
1,267✔
2845
  SMnode    *pMnode = pReq->info.node;
1,267✔
2846
  SSdb      *pSdb = pMnode->pSdb;
1,267✔
2847
  int32_t    numOfRows = 0;
1,267✔
2848
  SDbObj    *pDb = NULL;
1,267✔
2849
  SUserObj  *pUser = NULL;
1,267✔
2850
  ESdbStatus objStatus = 0;
1,267✔
2851

2852
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
1,267✔
2853
  if (pUser == NULL) return 0;
1,267!
2854
  bool sysinfo = pUser->sysInfo;
1,267✔
2855

2856
  // Append the information_schema database into the result.
2857
  if (!pShow->sysDbRsp) {
1,267!
2858
    SDbObj infoschemaDb = {0};
1,267✔
2859
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
1,267✔
2860
    size_t numOfTables = 0;
1,267✔
2861
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
1,267✔
2862
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
1,267✔
2863

2864
    numOfRows += 1;
1,267✔
2865

2866
    SDbObj perfschemaDb = {0};
1,267✔
2867
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
1,267✔
2868
    numOfTables = 0;
1,267✔
2869
    getPerfDbMeta(NULL, &numOfTables);
1,267✔
2870
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
1,267✔
2871

2872
    numOfRows += 1;
1,267✔
2873
    pShow->sysDbRsp = true;
1,267✔
2874
  }
2875

2876
  while (numOfRows < rowsCapacity) {
2,398!
2877
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
2,398✔
2878
    if (pShow->pIter == NULL) break;
2,398✔
2879

2880
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
1,131✔
2881
      int32_t numOfTables = 0;
1,108✔
2882
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
1,108✔
2883
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
1,108✔
2884
      numOfRows++;
1,108✔
2885
    }
2886

2887
    sdbRelease(pSdb, pDb);
1,131✔
2888
  }
2889

2890
  pShow->numOfRows += numOfRows;
1,267✔
2891
  mndReleaseUser(pMnode, pUser);
1,267✔
2892
  return numOfRows;
1,267✔
2893
}
2894

2895
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
2896
  SSdb *pSdb = pMnode->pSdb;
×
2897
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
2898
}
×
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc