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

taosdata / TDengine / #4488

12 Jul 2025 07:47AM UTC coverage: 62.207% (-0.7%) from 62.948%
#4488

push

travis-ci

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

157961 of 324087 branches covered (48.74%)

Branch coverage included in aggregate %.

244465 of 322830 relevant lines covered (75.73%)

6561668.76 hits per line

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

67.68
/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

40
#define DB_VER_NUMBER   1
41
#define DB_RESERVE_SIZE 14
42

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

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

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

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

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

85
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
2,467✔
86
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
2,467✔
87

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

91
void mndCleanupDb(SMnode *pMnode) {}
2,466✔
92

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

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

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

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

165
  terrno = 0;
22,769✔
166

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

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

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

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

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

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

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

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

269
  taosInitRWLatch(&pDb->lock);
20,223✔
270

271
  if (pDb->cfg.s3ChunkSize == 0) {
20,223!
272
    pDb->cfg.s3ChunkSize = TSDB_DEFAULT_S3_CHUNK_SIZE;
×
273

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

277
  if (pDb->cfg.s3KeepLocal == 0) {
20,223!
278
    pDb->cfg.s3KeepLocal = TSDB_DEFAULT_S3_KEEP_LOCAL;
×
279

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

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

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

292
  terrno = 0;
20,223✔
293

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

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

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

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

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

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

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

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

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

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

397
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
336,352✔
398
  SSdb   *pSdb = pMnode->pSdb;
336,352✔
399
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
336,352✔
400
  if (pDb == NULL) {
336,353✔
401
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
9,954✔
402
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
8,286✔
403
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
1,668✔
404
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
190✔
405
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
1,478!
406
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
1,478✔
407
    } else {
408
      terrno = TSDB_CODE_APP_ERROR;
×
409
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
410
    }
411
  }
412
  return pDb;
336,351✔
413
}
414

415
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
376,426✔
416
  SSdb *pSdb = pMnode->pSdb;
376,426✔
417
  sdbRelease(pSdb, pDb);
376,426✔
418
}
376,424✔
419

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

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

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

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

447
  return 0;
5,200✔
448
}
449

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

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

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

503
  if (pCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
5,198!
504
  if (pCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
5,198!
505
  if (pCfg->s3Compact < TSDB_MIN_S3_COMPACT || pCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
5,198!
506

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

521
  code = 0;
5,198✔
522
  TAOS_RETURN(code);
5,198✔
523
}
524

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

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

569
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
169!
570
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
×
571
    return code;
×
572
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
169!
573
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
169!
574
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
169!
575
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
169!
576
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
169!
577
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
169!
578
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
169!
579
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
580
    return code;
×
581
  }
582
  if (pNewCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pNewCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
169!
583
  if (pNewCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pNewCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
169!
584
  if (pNewCfg->s3Compact < TSDB_MIN_S3_COMPACT || pNewCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
169!
585

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

602
  code = 0;
169✔
603
  TAOS_RETURN(code);
169✔
604
}
605

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

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

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

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

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

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

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

691
  TAOS_RETURN(code);
×
692
}
693

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

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

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

725
  TAOS_RETURN(code);
4,676✔
726
}
727

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

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

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

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

771
  TAOS_RETURN(code);
4,676✔
772
}
773

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

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

785
  TAOS_RETURN(code);
4,676✔
786
}
787

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

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

799
  TAOS_RETURN(code);
4,676✔
800
}
801

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

857
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
5,200✔
858
  dbObj.cfg.pRetensions = pCreate->pRetensions;
5,200✔
859

860
  mndSetDefaultDbCfg(&dbObj.cfg);
5,200✔
861

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1026
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
5,232!
1027

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

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

1035
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
5,232!
1036

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

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

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

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

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

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

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

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

1064
  TAOS_RETURN(code);
5,750✔
1065
}
1066

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

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

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

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

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

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

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

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

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

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

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

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

1129
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
197!
1130
    pDb->cfg.cacheLast = pAlter->cacheLast;
108✔
1131
    code = 0;
108✔
1132
  }
1133

1134
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
197!
1135
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
56✔
1136
    code = 0;
56✔
1137
  }
1138

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

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

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

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

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

1171
  if (pAlter->s3KeepLocal > TSDB_MIN_S3_KEEP_LOCAL && pAlter->s3KeepLocal != pDb->cfg.s3KeepLocal) {
197!
1172
    pDb->cfg.s3KeepLocal = pAlter->s3KeepLocal;
×
1173
    pDb->vgVersion++;
×
1174
    code = 0;
×
1175
  }
1176

1177
  if (pAlter->s3Compact >= TSDB_MIN_S3_COMPACT && pAlter->s3Compact != pDb->cfg.s3Compact) {
197!
1178
    pDb->cfg.s3Compact = pAlter->s3Compact;
×
1179
    pDb->vgVersion++;
×
1180
    code = 0;
×
1181
  }
1182

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

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

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

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

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

1221
  TAOS_RETURN(code);
197✔
1222
}
1223

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

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

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

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

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

1271
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
168!
1272

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

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

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

1295
    sdbRelease(pSdb, pVgroup);
2,018✔
1296
  }
1297

1298
  taosArrayDestroy(pArray);
167✔
1299
  TAOS_RETURN(code);
167✔
1300

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

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

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

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

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

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

1334
_OVER:
169✔
1335
  mndTransDrop(pTrans);
169✔
1336
  TAOS_RETURN(code);
169✔
1337
}
1338

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1419
  TAOS_RETURN(code);
198✔
1420
}
1421

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

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

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

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

1487
    mndDumpDbCfgInfo(&cfgRsp, pDb);
5,778✔
1488
  }
1489

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

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

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

1507
  code = 0;
5,778✔
1508

1509
_OVER:
6,620✔
1510

1511
  tFreeSDbCfgRsp(&cfgRsp);
6,620✔
1512

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

1517
  mndReleaseDb(pMnode, pDb);
6,620✔
1518

1519
  TAOS_RETURN(code);
6,620✔
1520
}
1521

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

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

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

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

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

1552
  TAOS_RETURN(code);
2,083✔
1553
}
1554

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

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

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

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

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

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

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

1611
    sdbRelease(pSdb, pVgroup);
7,565✔
1612
  }
1613

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

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

1641
    sdbRelease(pSdb, pStb);
4,772✔
1642
  }
1643

1644
  TAOS_RETURN(code);
2,083✔
1645
}
1646

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

1654
  TAOS_RETURN(code);
4,195✔
1655
}
1656

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

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

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

1675
    sdbRelease(pSdb, pVgroup);
7,565✔
1676
  }
1677

1678
  TAOS_RETURN(code);
2,083✔
1679
}
1680

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

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

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

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

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

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

1726
    sdbRelease(pMnode->pSdb, pStb);
4,772✔
1727
  }
1728

1729
  code = 0;
2,083✔
1730

1731
_OVER:
2,083✔
1732
  return code;
2,083✔
1733
}
1734

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1834
    sdbRelease(pSdb, pVgroup);
3,371✔
1835
  }
1836

1837
  code = mndDropDb(pMnode, pReq, pDb);
2,084✔
1838
  if (code == TSDB_CODE_SUCCESS) {
2,084✔
1839
    code = TSDB_CODE_ACTION_IN_PROGRESS;
2,083✔
1840
  }
1841

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

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

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

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

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

1863
  void *pIter = NULL;
177,493✔
1864
  while (vindex < pDb->cfg.numOfVgroups) {
1,141,243✔
1865
    SVgObj *pVgroup = NULL;
963,702✔
1866
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
963,702✔
1867
    if (pIter == NULL) break;
963,712!
1868

1869
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
963,712✔
1870
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
359,140✔
1871
      vindex++;
359,140✔
1872
    }
1873

1874
    sdbRelease(pSdb, pVgroup);
963,696✔
1875
  }
1876

1877
  sdbCancelFetch(pSdb, pIter);
177,541✔
1878
  return numOfTables;
177,497✔
1879
}
1880

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

1885
  void *pIter = NULL;
190,043✔
1886
  while (1) {
817,101✔
1887
    SVgObj *pVgroup = NULL;
1,007,144✔
1888
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
1,007,144✔
1889
    if (pIter == NULL) break;
1,007,118✔
1890

1891
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
986,128!
1892
      SVgroupInfo vgInfo = {0};
402,666✔
1893
      vgInfo.vgId = pVgroup->vgId;
402,666✔
1894
      vgInfo.hashBegin = pVgroup->hashBegin;
402,666✔
1895
      vgInfo.hashEnd = pVgroup->hashEnd;
402,666✔
1896
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
402,666✔
1897
      vgInfo.epSet.numOfEps = pVgroup->replica;
402,666✔
1898
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
818,229✔
1899
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
415,558✔
1900
        SEp       *pEp = &vgInfo.epSet.eps[gid];
415,558✔
1901
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
415,558✔
1902
        if (pDnode != NULL) {
415,561✔
1903
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
415,465✔
1904
          pEp->port = pDnode->port;
415,465✔
1905
        }
1906
        mndReleaseDnode(pMnode, pDnode);
415,561✔
1907
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
415,563✔
1908
          vgInfo.epSet.inUse = gid;
401,190✔
1909
        }
1910
      }
1911
      vindex++;
402,671✔
1912
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
402,665!
1913
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
1914
      }
1915
    }
1916

1917
    sdbRelease(pSdb, pVgroup);
986,127✔
1918

1919
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
986,154✔
1920
      sdbCancelFetch(pSdb, pIter);
169,053✔
1921
      break;
169,053✔
1922
    }
1923
  }
1924
}
190,043✔
1925

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

1934
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
168,967✔
1935

1936
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
168,971!
1937
      pReq->stateTs < pDb->stateTs) {
3,183✔
1938
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
166,242✔
1939
  }
1940

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

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

1960
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
187,407!
1961

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

1970
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
20,315✔
1971
      usedbRsp.vgVersion = vgVersion++;
20,315✔
1972
    } else {
1973
      usedbRsp.vgVersion = usedbReq.vgVersion;
37✔
1974
    }
1975
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
20,352✔
1976
    code = 0;
20,352✔
1977
  } else {
1978
    pDb = mndAcquireDb(pMnode, usedbReq.db);
167,060✔
1979
    if (pDb == NULL) {
167,058✔
1980
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
705✔
1981
      usedbRsp.uid = usedbReq.dbId;
705✔
1982
      usedbRsp.vgVersion = usedbReq.vgVersion;
705✔
1983
      usedbRsp.errCode = terrno;
705✔
1984

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

1991
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
166,338!
1992

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

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

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

2013
  pReq->info.rsp = pRsp;
186,689✔
2014
  pReq->info.rspLen = contLen;
186,689✔
2015

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

2021
  mndReleaseDb(pMnode, pDb);
187,411✔
2022
  tFreeSUsedbRsp(&usedbRsp);
187,412✔
2023

2024
  TAOS_RETURN(code);
187,412✔
2025
}
2026

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

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

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

2049
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
13,693✔
2050
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
13,067✔
2051
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
4,894✔
2052
      if (pDbCacheInfo->vgVersion >= vgVersion) {
4,894✔
2053
        continue;
9,883✔
2054
      }
2055

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

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

2063
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
675✔
2064

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

2070
      continue;
675✔
2071
    }
2072

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

2087
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
8,526✔
2088

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

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

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

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

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

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

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

2174
  *ppRsp = pRsp;
11,583✔
2175
  *pRspLen = rspLen;
11,583✔
2176

2177
  tFreeSDbHbBatchRsp(&batchRsp);
11,583✔
2178
  TAOS_RETURN(code);
11,583✔
2179
}
2180

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

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

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

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

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

2224
  return 0;
3✔
2225
}
2226

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

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

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

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

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

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

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

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

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

2262
static int32_t mndS3MigrateDb(SMnode *pMnode, SDbObj *pDb) {
×
2263
  SSdb            *pSdb = pMnode->pSdb;
×
2264
  SVgObj          *pVgroup = NULL;
×
2265
  void            *pIter = NULL;
×
2266
  SVS3MigrateDbReq s3migrateReq = {.timestamp = taosGetTimestampSec()};
×
2267
  int32_t          reqLen = tSerializeSVS3MigrateDbReq(NULL, 0, &s3migrateReq);
×
2268
  int32_t          contLen = reqLen + sizeof(SMsgHead);
×
2269
  int32_t          code = 0;
×
2270

2271
  while (1) {
×
2272
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
×
2273
    if (pIter == NULL) break;
×
2274

2275
    if (pVgroup->dbUid != pDb->uid) continue;
×
2276

2277
    SMsgHead *pHead = rpcMallocCont(contLen);
×
2278
    if (pHead == NULL) {
×
2279
      sdbCancelFetch(pSdb, pVgroup);
×
2280
      sdbRelease(pSdb, pVgroup);
×
2281
      continue;
×
2282
    }
2283
    pHead->contLen = htonl(contLen);
×
2284
    pHead->vgId = htonl(pVgroup->vgId);
×
2285
    int32_t ret = 0;
×
2286
    if ((ret = tSerializeSVS3MigrateDbReq((char *)pHead + sizeof(SMsgHead), contLen, &s3migrateReq)) < 0) {
×
2287
      sdbRelease(pSdb, pVgroup);
×
2288
      return ret;
×
2289
    }
2290

2291
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_S3MIGRATE, .pCont = pHead, .contLen = contLen};
×
2292
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
×
2293
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
×
2294
    if (code != 0) {
×
2295
      mError("vgId:%d, failed to send vnode-s3migrate request to vnode since 0x%x", pVgroup->vgId, code);
×
2296
    } else {
2297
      mInfo("vgId:%d, send vnode-s3migrate request to vnode, time:%d", pVgroup->vgId, s3migrateReq.timestamp);
×
2298
    }
2299
    sdbRelease(pSdb, pVgroup);
×
2300
  }
2301

2302
  return 0;
×
2303
}
2304

2305
static int32_t mndProcessS3MigrateDbReq(SRpcMsg *pReq) {
×
2306
  SMnode         *pMnode = pReq->info.node;
×
2307
  int32_t         code = -1;
×
2308
  SDbObj         *pDb = NULL;
×
2309
  SS3MigrateDbReq s3migrateReq = {0};
×
2310

2311
  TAOS_CHECK_GOTO(tDeserializeSS3MigrateDbReq(pReq->pCont, pReq->contLen, &s3migrateReq), NULL, _OVER);
×
2312

2313
  mInfo("db:%s, start to s3migrate", s3migrateReq.db);
×
2314

2315
  pDb = mndAcquireDb(pMnode, s3migrateReq.db);
×
2316
  if (pDb == NULL) {
×
2317
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2318
    if (terrno != 0) code = terrno;
×
2319
    goto _OVER;
×
2320
  }
2321

2322
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
×
2323

2324
  if(pDb->cfg.isMount) {
×
2325
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
2326
    goto _OVER;
×
2327
  }
2328

2329
  code = mndS3MigrateDb(pMnode, pDb);
×
2330

2331
_OVER:
×
2332
  if (code != 0) {
×
2333
    mError("db:%s, failed to process s3migrate db req since %s", s3migrateReq.db, tstrerror(code));
×
2334
  }
2335

2336
  mndReleaseDb(pMnode, pDb);
×
2337
  TAOS_RETURN(code);
×
2338
}
2339

2340
const char *mndGetDbStr(const char *src) {
91,777✔
2341
  char *pos = strstr(src, TS_PATH_DELIMITER);
91,777✔
2342
  if (pos != NULL) ++pos;
91,777✔
2343
  if (pos == NULL) return src;
91,777✔
2344
  return pos;
89,083✔
2345
}
2346

2347
const char *mndGetStableStr(const char *src) {
×
2348
  char *pos = strstr(src, TS_PATH_DELIMITER);
×
2349
  if (pos != NULL) ++pos;
×
2350
  if (pos == NULL) return src;
×
2351
  return mndGetDbStr(pos);
×
2352
}
2353

2354
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
12✔
2355
  int64_t v = 0;
12✔
2356
  switch (unit) {
12!
2357
    case 's':
4✔
2358
      v = val / 1000;
4✔
2359
      break;
4✔
2360
    case 'm':
2✔
2361
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
2✔
2362
      break;
2✔
2363
    case 'h':
×
2364
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2365
      break;
×
2366
    case 'd':
6✔
2367
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
6✔
2368
      break;
6✔
2369
    case 'w':
×
2370
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2371
      break;
×
2372
    default:
×
2373
      break;
×
2374
  }
2375

2376
  return v;
12✔
2377
}
2378

2379
static char *buildRetension(SArray *pRetension) {
1,098✔
2380
  size_t size = taosArrayGetSize(pRetension);
1,098✔
2381
  if (size == 0) {
1,098✔
2382
    return NULL;
1,096✔
2383
  }
2384

2385
  char       *p1 = taosMemoryCalloc(1, 100);
2!
2386
  SRetention *p = taosArrayGet(pRetension, 0);
2✔
2387

2388
  int32_t len = 2;
2✔
2389

2390
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2391
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2392
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2393

2394
  if (size > 1) {
2!
2395
    len += tsnprintf(p1 + len, 100 - len, ",");
2✔
2396
    p = taosArrayGet(pRetension, 1);
2✔
2397

2398
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2399
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2400
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2401
  }
2402

2403
  if (size > 2) {
2!
2404
    len += tsnprintf(p1 + len, 100 - len, ",");
2✔
2405
    p = taosArrayGet(pRetension, 2);
2✔
2406

2407
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2408
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2409
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2410
  }
2411

2412
  varDataSetLen(p1, len);
2✔
2413
  return p1;
2✔
2414
}
2415

2416
static const char *getCacheModelStr(int8_t cacheModel) {
1,098✔
2417
  switch (cacheModel) {
1,098!
2418
    case TSDB_CACHE_MODEL_NONE:
1,065✔
2419
      return TSDB_CACHE_MODEL_NONE_STR;
1,065✔
2420
    case TSDB_CACHE_MODEL_LAST_ROW:
4✔
2421
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
4✔
2422
    case TSDB_CACHE_MODEL_LAST_VALUE:
10✔
2423
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
10✔
2424
    case TSDB_CACHE_MODEL_BOTH:
19✔
2425
      return TSDB_CACHE_MODEL_BOTH_STR;
19✔
2426
    default:
×
2427
      break;
×
2428
  }
2429
  return "unknown";
×
2430
}
2431

2432
static const char *getEncryptAlgorithmStr(int8_t encryptAlgorithm) {
1,098✔
2433
  switch (encryptAlgorithm) {
1,098!
2434
    case TSDB_ENCRYPT_ALGO_NONE:
1,098✔
2435
      return TSDB_ENCRYPT_ALGO_NONE_STR;
1,098✔
2436
    case TSDB_ENCRYPT_ALGO_SM4:
×
2437
      return TSDB_ENCRYPT_ALGO_SM4_STR;
×
2438
    default:
×
2439
      break;
×
2440
  }
2441
  return "unknown";
×
2442
}
2443

2444
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
6,801✔
2445
  if (pDb->cfg.replications == 1) return true;
6,801✔
2446

2447
  SSdb *pSdb = pMnode->pSdb;
1,324✔
2448
  void *pIter = NULL;
1,324✔
2449
  bool  isReady = true;
1,324✔
2450
  while (1) {
6,935✔
2451
    SVgObj *pVgroup = NULL;
8,259✔
2452
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
8,259✔
2453
    if (pIter == NULL) break;
8,259✔
2454

2455
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
6,935✔
2456
      bool hasLeader = false;
3,760✔
2457
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
14,931✔
2458
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
11,171✔
2459
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
9,987!
2460
          hasLeader = true;
1,184✔
2461
        }
2462
      }
2463
      if (!hasLeader) isReady = false;
3,760✔
2464
    }
2465
    sdbRelease(pSdb, pVgroup);
6,935✔
2466
  }
2467

2468
  return isReady;
1,324✔
2469
}
2470

2471
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
3,650✔
2472
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2473
  int32_t cols = 0;
3,650✔
2474
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
3,650✔
2475
  char   *buf = taosMemoryMalloc(bytes);
3,650!
2476
  if (buf == NULL) {
3,650!
2477
    mError("db:%s, failed to malloc buffer", pDb->name);
×
2478
    return;
×
2479
  }
2480
  int32_t code = 0;
3,650✔
2481
  int32_t lino = 0;
3,650✔
2482

2483
  const char *name = mndGetDbStr(pDb->name);
3,650✔
2484
  if (name != NULL) {
3,650!
2485
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
3,650✔
2486
  } else {
2487
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
2488
  }
2489

2490
  const char *precStr = NULL;
3,650✔
2491
  switch (pDb->cfg.precision) {
3,650!
2492
    case TSDB_TIME_PRECISION_MILLI:
3,610✔
2493
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
3,610✔
2494
      break;
3,610✔
2495
    case TSDB_TIME_PRECISION_MICRO:
12✔
2496
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
12✔
2497
      break;
12✔
2498
    case TSDB_TIME_PRECISION_NANO:
28✔
2499
      precStr = TSDB_TIME_PRECISION_NANO_STR;
28✔
2500
      break;
28✔
2501
    default:
×
2502
      precStr = "none";
×
2503
      break;
×
2504
  }
2505
  char precVstr[10] = {0};
3,650✔
2506
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
3,650✔
2507

2508
  char *statusStr = "ready";
3,650✔
2509
  if (objStatus == SDB_STATUS_CREATING) {
3,650!
2510
    statusStr = "creating";
×
2511
  } else if (objStatus == SDB_STATUS_DROPPING) {
3,650!
2512
    statusStr = "dropping";
×
2513
  } else {
2514
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
3,650✔
2515
      statusStr = "unsynced";
19✔
2516
    }
2517
  }
2518
  char statusVstr[24] = {0};
3,650✔
2519
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
3,650✔
2520

2521
  if (sysDb || !sysinfo) {
3,650✔
2522
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
96,976✔
2523
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
94,424✔
2524
      if (i == 0) {
94,424✔
2525
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
2,552!
2526
      } else if (i == 1) {
91,872✔
2527
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
2,552!
2528
      } else if (i == 3) {
89,320✔
2529
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
2,552!
2530
      } else if (i == 14) {
86,768✔
2531
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
2,552!
2532
      } else if (i == 15) {
84,216✔
2533
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
2,552!
2534
      } else {
2535
        colDataSetNULL(pColInfo, rows);
81,664✔
2536
      }
2537
    }
2538
  } else {
2539
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2540
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
1,098!
2541

2542
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2543
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
1,098!
2544

2545
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2546
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
1,098!
2547

2548
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2549
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
1,098!
2550

2551
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2552
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
1,098!
2553

2554
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
1,098!
2555
    char        strictVstr[24] = {0};
1,098✔
2556
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
1,098✔
2557
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2558
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
1,098!
2559

2560
    char    durationStr[128] = {0};
1,098✔
2561
    char    durationVstr[128] = {0};
1,098✔
2562
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
1,098✔
2563
                                       pDb->cfg.daysPerFile);
2564

2565
    varDataSetLen(durationVstr, len);
1,098✔
2566
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2567
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,098!
2568

2569
    char keepVstr[128] = {0};
1,098✔
2570
    char keep0Str[32] = {0};
1,098✔
2571
    char keep1Str[32] = {0};
1,098✔
2572
    char keep2Str[32] = {0};
1,098✔
2573

2574
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
1,098✔
2575
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
1,098✔
2576
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
1,098✔
2577

2578
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
1,098!
2579
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
2580
    } else {
2581
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
1,098✔
2582
    }
2583
    varDataSetLen(keepVstr, len);
1,098✔
2584
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2585
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
1,098!
2586

2587
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2588
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
1,098!
2589

2590
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2591
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
1,098!
2592

2593
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2594
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
1,098!
2595

2596
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2597
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
1,098!
2598

2599
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2600
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
1,098!
2601

2602
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2603
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
1,098!
2604

2605
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2606
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
1,098!
2607

2608
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2609
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
1,098!
2610

2611
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
1,098✔
2612
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2613
    if (rentensionVstr == NULL) {
1,098✔
2614
      colDataSetNULL(pColInfo, rows);
1,096!
2615
    } else {
2616
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
2!
2617
      taosMemoryFree(rentensionVstr);
2!
2618
    }
2619

2620
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2621
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
1,098!
2622

2623
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
1,098✔
2624
    char        cacheModelVstr[24] = {0};
1,098✔
2625
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
1,098✔
2626
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2627
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
1,098!
2628

2629
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2630
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
1,098!
2631

2632
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2633
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
1,098!
2634

2635
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2636
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
1,098!
2637

2638
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2639
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
1,098!
2640

2641
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2642
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
1,098!
2643

2644
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2645
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
1,098!
2646

2647
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2648
    int16_t hashPrefix = pDb->cfg.hashPrefix;
1,098✔
2649
    if (hashPrefix > 0) {
1,098✔
2650
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
14✔
2651
    } else if (hashPrefix < 0) {
1,084✔
2652
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
12✔
2653
    }
2654
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
1,098!
2655

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

2659
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2660
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
1,098!
2661

2662
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2663
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
1,098!
2664

2665
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2666
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3ChunkSize, false), &lino, _OVER);
1,098!
2667

2668
    char keeplocalVstr[128] = {0};
1,098✔
2669
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.s3KeepLocal);
1,098✔
2670
    varDataSetLen(keeplocalVstr, len);
1,098✔
2671
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2672
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
1,098!
2673

2674
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2675
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3Compact, false), &lino, _OVER);
1,098!
2676

2677
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2678
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
1,098!
2679

2680
    const char *encryptAlgorithmStr = getEncryptAlgorithmStr(pDb->cfg.encryptAlgorithm);
1,098✔
2681
    char        encryptAlgorithmVStr[24] = {0};
1,098✔
2682
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
1,098✔
2683
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,098✔
2684
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
1,098!
2685

2686
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
1,098✔
2687
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,098✔
2688
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,098!
2689
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,098!
2690
    }
2691

2692
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
1,098✔
2693
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
1,098✔
2694
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
1,098✔
2695
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,098✔
2696
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,098!
2697
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,098!
2698
    }
2699

2700
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
1,098✔
2701
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,098✔
2702
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,098!
2703
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,098!
2704
    }
2705
  }
2706
_OVER:
3,650✔
2707
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
3,650!
2708
  taosMemoryFree(buf);
3,650!
2709
}
2710

2711
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
1,270✔
2712
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
1,270✔
2713
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
1,270✔
2714
  pDbObj->cfg.numOfVgroups = 0;
1,270✔
2715
  pDbObj->cfg.strict = 1;
1,270✔
2716
  pDbObj->cfg.replications = 1;
1,270✔
2717
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
1,270✔
2718
}
1,270✔
2719

2720
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
1,270✔
2721
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
1,270✔
2722
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
1,270✔
2723
  pDbObj->cfg.numOfVgroups = 0;
1,270✔
2724
  pDbObj->cfg.strict = 1;
1,270✔
2725
  pDbObj->cfg.replications = 1;
1,270✔
2726
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
1,270✔
2727
}
1,270✔
2728

2729
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
4,997✔
2730
  SVgObj  *pVgroup = pObj;
4,997✔
2731
  int32_t *numOfTables = p1;
4,997✔
2732
  int64_t  uid = *(int64_t *)p2;
4,997✔
2733
  if (pVgroup->dbUid == uid) {
4,997✔
2734
    *numOfTables += pVgroup->numOfTables;
2,321✔
2735
  }
2736
  return true;
4,997✔
2737
}
2738

2739
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
1,270✔
2740
  SMnode    *pMnode = pReq->info.node;
1,270✔
2741
  SSdb      *pSdb = pMnode->pSdb;
1,270✔
2742
  int32_t    numOfRows = 0;
1,270✔
2743
  SDbObj    *pDb = NULL;
1,270✔
2744
  SUserObj  *pUser = NULL;
1,270✔
2745
  ESdbStatus objStatus = 0;
1,270✔
2746

2747
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
1,270✔
2748
  if (pUser == NULL) return 0;
1,270!
2749
  bool sysinfo = pUser->sysInfo;
1,270✔
2750

2751
  // Append the information_schema database into the result.
2752
  if (!pShow->sysDbRsp) {
1,270!
2753
    SDbObj infoschemaDb = {0};
1,270✔
2754
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
1,270✔
2755
    size_t numOfTables = 0;
1,270✔
2756
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
1,270✔
2757
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
1,270✔
2758

2759
    numOfRows += 1;
1,270✔
2760

2761
    SDbObj perfschemaDb = {0};
1,270✔
2762
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
1,270✔
2763
    numOfTables = 0;
1,270✔
2764
    getPerfDbMeta(NULL, &numOfTables);
1,270✔
2765
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
1,270✔
2766

2767
    numOfRows += 1;
1,270✔
2768
    pShow->sysDbRsp = true;
1,270✔
2769
  }
2770

2771
  while (numOfRows < rowsCapacity) {
2,403!
2772
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
2,403✔
2773
    if (pShow->pIter == NULL) break;
2,403✔
2774

2775
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
1,133✔
2776
      int32_t numOfTables = 0;
1,110✔
2777
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
1,110✔
2778
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
1,110✔
2779
      numOfRows++;
1,110✔
2780
    }
2781

2782
    sdbRelease(pSdb, pDb);
1,133✔
2783
  }
2784

2785
  pShow->numOfRows += numOfRows;
1,270✔
2786
  mndReleaseUser(pMnode, pUser);
1,270✔
2787
  return numOfRows;
1,270✔
2788
}
2789

2790
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
2791
  SSdb *pSdb = pMnode->pSdb;
×
2792
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
2793
}
×
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