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

taosdata / TDengine / #4473

08 Jul 2025 09:38AM UTC coverage: 62.922% (+0.7%) from 62.22%
#4473

push

travis-ci

web-flow
Merge pull request #31712 from taosdata/merge/mainto3.0

merge: from main to 3.0 branch

158525 of 321496 branches covered (49.31%)

Branch coverage included in aggregate %.

56 of 60 new or added lines in 13 files covered. (93.33%)

1333 existing lines in 67 files now uncovered.

245526 of 320647 relevant lines covered (76.57%)

17689640.25 hits per line

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

68.36
/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,492✔
64
  SSdbTable table = {
2,492✔
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,492✔
76
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
2,492✔
77
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
2,492✔
78
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
2,492✔
79
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
2,492✔
80
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
2,492✔
81
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
2,492✔
82
  mndSetMsgHandle(pMnode, TDMT_MND_S3MIGRATE_DB, mndProcessS3MigrateDbReq);
2,492✔
83
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
2,492✔
84

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

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

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

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

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

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

161
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
26,200!
162
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
26,200!
163

164
  terrno = 0;
26,200✔
165

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

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

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

184
  int8_t sver = 0;
22,287✔
185
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
22,287!
186

187
  if (sver != DB_VER_NUMBER) {
22,287!
188
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
189
    goto _OVER;
×
190
  }
191

192
  pRow = sdbAllocRow(sizeof(SDbObj));
22,287✔
193
  if (pRow == NULL) goto _OVER;
22,287!
194

195
  pDb = sdbGetRowObj(pRow);
22,287✔
196
  if (pDb == NULL) goto _OVER;
22,287!
197

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

264
  SDB_GET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
22,287!
265
  taosInitRWLatch(&pDb->lock);
22,287✔
266

267
  if (pDb->cfg.s3ChunkSize == 0) {
22,287!
268
    pDb->cfg.s3ChunkSize = TSDB_DEFAULT_S3_CHUNK_SIZE;
×
269

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

273
  if (pDb->cfg.s3KeepLocal == 0) {
22,287!
274
    pDb->cfg.s3KeepLocal = TSDB_DEFAULT_S3_KEEP_LOCAL;
×
275

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

279
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
22,287!
280
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
22,287!
281
          TSDB_DEFAULT_TSDB_PAGESIZE);
282
  }
283

284
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
22,287✔
285
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
20,324!
286
  }
287

288
  terrno = 0;
22,287✔
289

290
_OVER:
22,287✔
291
  if (terrno != 0) {
22,287!
292
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
293
    taosMemoryFreeClear(pRow);
×
294
    return NULL;
×
295
  }
296

297
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
22,287✔
298
  return pRow;
22,287✔
299
}
300

301
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
5,277✔
302
  SSdb    *pSdb = pMnode->pSdb;
5,277✔
303
  SSdbRow *pRow = NULL;
5,277✔
304
  SDbObj  *pNewDb = NULL;
5,277✔
305
  int      code = -1;
5,277✔
306

307
  pRow = mndDbActionDecode(pRaw);
5,277✔
308
  if (pRow == NULL) {
5,277!
309
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
310
    if (terrno != 0) code = terrno;
×
311
    goto _OVER;
×
312
  }
313
  pNewDb = sdbGetRowObj(pRow);
5,277✔
314
  if (pNewDb == NULL) {
5,277!
315
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
316
    if (terrno != 0) code = terrno;
×
317
    goto _OVER;
×
318
  }
319

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

329
  code = 0;
5,277✔
330
_OVER:
5,277✔
331
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
5,277!
332
  taosMemoryFreeClear(pRow);
5,277!
333
  return code;
5,277✔
334
}
335

336
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
5,831✔
337
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
5,831✔
338
  return 0;
5,831✔
339
}
340

341
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
22,287✔
342
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
22,287✔
343
  taosArrayDestroy(pDb->cfg.pRetensions);
22,287✔
344
  return 0;
22,287✔
345
}
346

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

388
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
305,835✔
389
  SSdb *pSdb = pMnode->pSdb;
305,835✔
390
  return sdbGetTableVer(pSdb, SDB_VGROUP);
305,835✔
391
}
392

393
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
5,105,613✔
394
  SSdb   *pSdb = pMnode->pSdb;
5,105,613✔
395
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
5,105,613✔
396
  if (pDb == NULL) {
5,105,800✔
397
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
11,275✔
398
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
9,420✔
399
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
1,855✔
400
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
113✔
401
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
1,742!
402
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
1,742✔
403
    } else {
404
      terrno = TSDB_CODE_APP_ERROR;
×
405
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
406
    }
407
  }
408
  return pDb;
5,105,740✔
409
}
410

411
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
5,525,017✔
412
  SSdb *pSdb = pMnode->pSdb;
5,525,017✔
413
  sdbRelease(pSdb, pDb);
5,525,017✔
414
}
5,525,228✔
415

416
bool mndDbIsExist(SMnode *pMnode, const char *db) {
15✔
417
  SDbObj *pDb = mndAcquireDb(pMnode, db);
15✔
418
  if (pDb == NULL) {
15!
419
    return false;
×
420
  } else {
421
    mndReleaseDb(pMnode, pDb);
15✔
422
    pDb = NULL;
15✔
423
    return true;
15✔
424
  }
425
}
426

427
static int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
5,593✔
428
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
5,593✔
429
  if (pos == NULL) {
5,593!
430
    return TSDB_CODE_MND_INVALID_DB;
×
431
  }
432

433
  int32_t acctId;
434
  int32_t code = taosStr2int32(dbName, &acctId);
5,593✔
435
  if (code != 0) {
5,593!
436
    return code;
×
437
  }
438

439
  if (acctId != pUser->acctId) {
5,593!
440
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
441
  }
442

443
  return 0;
5,593✔
444
}
445

446
static int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
5,593✔
447
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
5,593✔
448

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

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

499
  if (pCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
5,591!
500
  if (pCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
5,591!
501
  if (pCfg->s3Compact < TSDB_MIN_S3_COMPACT || pCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
5,591!
502

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

517
  code = 0;
5,591✔
518
  TAOS_RETURN(code);
5,591✔
519
}
520

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

556
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
260!
557
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
558
    return code;
×
559
  }
560
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
260!
561
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
562
    return code;
×
563
  }
564

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

582
  if (pNewCfg->compactInterval != 0 &&
260!
583
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
×
584
    return code;
×
585
  if (pNewCfg->compactStartTime != 0 &&
260!
586
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
×
587
    return code;
×
588
  if (pNewCfg->compactEndTime != 0 &&
260!
589
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
×
590
    return code;
×
591
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
260!
592
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
×
593
    return code;
×
594
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
260!
595
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
260!
596
    return code;
×
597

598
  code = 0;
260✔
599
  TAOS_RETURN(code);
260✔
600
}
601

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

640
static int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
5,038✔
641
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
5,038✔
642
  if (pDbRaw == NULL) return -1;
5,038!
643

644
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
5,038!
645
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
5,038!
646
  return 0;
5,038✔
647
}
648

649
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
5,038✔
650
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
15,921✔
651
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
10,883!
652
  }
653
  return 0;
5,038✔
654
}
655

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

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

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

687
  TAOS_RETURN(code);
×
688
}
689

690
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
5,038✔
691
  int32_t  code = 0;
5,038✔
692
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
5,038✔
693
  if (pDbRaw == NULL) {
5,038!
694
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
695
    if (terrno != 0) code = terrno;
×
696
    TAOS_RETURN(code);
×
697
  }
698
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
5,038!
699
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
5,038!
700

701
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
15,921✔
702
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
10,883✔
703
    if (pVgRaw == NULL) {
10,883!
704
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
705
      if (terrno != 0) code = terrno;
×
706
      TAOS_RETURN(code);
×
707
    }
708
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
10,883!
709
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
10,883!
710
  }
711

712
  if (pDb->cfg.withArbitrator) {
5,038✔
713
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
20✔
714
      SVgObj   *pVgObj = pVgroups + v;
13✔
715
      SArbGroup arbGroup = {0};
13✔
716
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
13!
717
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
13!
718
    }
719
  }
720

721
  TAOS_RETURN(code);
5,038✔
722
}
723

724
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
5,038✔
725
                                        SUserObj *pUserDuped) {
726
  int32_t  code = 0;
5,038✔
727
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
5,038✔
728
  if (pDbRaw == NULL) {
5,038!
729
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
730
    if (terrno != 0) code = terrno;
×
731
    TAOS_RETURN(code);
×
732
  }
733
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
5,038!
734
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
5,038!
735

736
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
15,921✔
737
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
10,883✔
738
    if (pVgRaw == NULL) {
10,883!
739
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
740
      if (terrno != 0) code = terrno;
×
741
      TAOS_RETURN(code);
×
742
    }
743
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
10,883!
744
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
10,883!
745
  }
746

747
  if (pDb->cfg.withArbitrator) {
5,038✔
748
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
20✔
749
      SVgObj   *pVgObj = pVgroups + v;
13✔
750
      SArbGroup arbGroup = {0};
13✔
751
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
13!
752
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
13!
753
    }
754
  }
755

756
  if (pUserDuped) {
5,038✔
757
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
4✔
758
    if (pUserRaw == NULL) {
4!
759
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
760
      if (terrno != 0) code = terrno;
×
761
      TAOS_RETURN(code);
×
762
    }
763
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
4!
764
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
4!
765
  }
766

767
  TAOS_RETURN(code);
5,038✔
768
}
769

770
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
5,038✔
771
  int32_t code = 0;
5,038✔
772
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
15,921✔
773
    SVgObj *pVgroup = pVgroups + vg;
10,883✔
774

775
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
22,699✔
776
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
11,816✔
777
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
11,816!
778
    }
779
  }
780

781
  TAOS_RETURN(code);
5,038✔
782
}
783

784
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
5,038✔
785
  int32_t code = 0;
5,038✔
786
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
15,921✔
787
    SVgObj *pVgroup = pVgroups + vg;
10,883✔
788

789
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
22,699✔
790
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
11,816✔
791
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
11,816!
792
    }
793
  }
794

795
  TAOS_RETURN(code);
5,038✔
796
}
797

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

853
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
5,593✔
854
  dbObj.cfg.pRetensions = pCreate->pRetensions;
5,593✔
855

856
  mndSetDefaultDbCfg(&dbObj.cfg);
5,593✔
857

858
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
5,593!
859
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, terrstr());
×
860
    TAOS_RETURN(code);
×
861
  }
862

863
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
5,593✔
864
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, terrstr());
2!
865
    TAOS_RETURN(code);
2✔
866
  }
867

868
  if (dbObj.cfg.hashPrefix > 0) {
5,591✔
869
    int32_t dbLen = strlen(dbObj.name) + 1;
14✔
870
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
14!
871
    dbObj.cfg.hashPrefix += dbLen;
14✔
872
  } else if (dbObj.cfg.hashPrefix < 0) {
5,577✔
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
  }
877

878
  SVgObj *pVgroups = NULL;
5,591✔
879
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
5,591✔
880
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
492!
881
    TAOS_RETURN(code);
492✔
882
  }
883

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

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

904
  mndTransSetDbName(pTrans, dbObj.name, NULL);
5,099✔
905
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
5,099✔
906

907
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
5,038✔
908
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
5,038!
909
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
5,038!
910
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
5,038!
911
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
5,038!
912
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
5,038!
913
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
5,038!
914
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
5,038✔
915

916
_OVER:
5,034✔
917
  taosMemoryFree(pVgroups);
5,099!
918
  mndUserFreeObj(&newUserObj);
5,099✔
919
  mndTransDrop(pTrans);
5,099✔
920
  TAOS_RETURN(code);
5,099✔
921
}
922

923
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
5,625✔
924
  int32_t    code = 0;
5,625✔
925
  SSdb      *pSdb = pMnode->pSdb;
5,625✔
926
  SDnodeObj *pDnode = NULL;
5,625✔
927
  void      *pIter = NULL;
5,625✔
928

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

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

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

976
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
6,061✔
977
  SMnode      *pMnode = pReq->info.node;
6,061✔
978
  int32_t      code = -1;
6,061✔
979
  int32_t      lino = 0;
6,061✔
980
  SDbObj      *pDb = NULL;
6,061✔
981
  SUserObj    *pUser = NULL;
6,061✔
982
  SCreateDbReq createReq = {0};
6,061✔
983
  SArray      *dnodeList = NULL;
6,061✔
984

985
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
6,061✔
986
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
6,061!
987

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

997
  pDb = mndAcquireDb(pMnode, createReq.db);
6,061✔
998
  if (pDb != NULL) {
6,061✔
999
    if (createReq.ignoreExist) {
380✔
1000
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
365!
1001
      code = 0;
365✔
1002
      goto _OVER;
365✔
1003
    } else {
1004
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
15✔
1005
      goto _OVER;
15✔
1006
    }
1007
  } else {
1008
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
5,681✔
1009
      code = terrno;
55✔
1010
      goto _OVER;
55✔
1011
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
5,626!
1012
      goto _OVER;
×
1013
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
5,626!
1014
      // continue
1015
    } else {  // TSDB_CODE_APP_ERROR
1016
      goto _OVER;
×
1017
    }
1018
  }
1019

1020
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_DB, NULL), &lino, _OVER);
5,626✔
1021

1022
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
5,625!
1023

1024
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
5,625✔
1025
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
5,625!
1026

1027
  if (createReq.replications == 2) {
5,625✔
1028
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
11!
1029
  }
1030

1031
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
5,625!
1032

1033
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
5,625✔
1034

1035
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pUser), &lino, _OVER);
5,593!
1036

1037
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
5,593✔
1038
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
5,034!
1039

1040
  SName name = {0};
5,034✔
1041
  if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
5,034!
UNCOV
1042
    mError("db:%s, failed to parse db name", createReq.db);
×
1043

1044
  auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen);
5,034✔
1045

1046
_OVER:
6,061✔
1047
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
6,061✔
1048
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
662!
1049
  }
1050

1051
  mndReleaseDb(pMnode, pDb);
6,061✔
1052
  mndReleaseUser(pMnode, pUser);
6,061✔
1053
  tFreeSCreateDbReq(&createReq);
6,061✔
1054
  taosArrayDestroy(dnodeList);
6,061✔
1055

1056
  TAOS_RETURN(code);
6,061✔
1057
}
1058

1059
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
288✔
1060
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
288✔
1061

1062
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
288!
UNCOV
1063
    pDb->cfg.buffer = pAlter->buffer;
×
UNCOV
1064
    code = 0;
×
1065
  }
1066

1067
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
288!
UNCOV
1068
    pDb->cfg.pages = pAlter->pages;
×
UNCOV
1069
    code = 0;
×
1070
  }
1071

1072
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
288!
UNCOV
1073
    pDb->cfg.pageSize = pAlter->pageSize;
×
UNCOV
1074
    code = 0;
×
1075
  }
1076

1077
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
288!
UNCOV
1078
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
UNCOV
1079
    code = 0;
×
1080
  }
1081

1082
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
288✔
1083
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
17✔
1084
    code = 0;
17✔
1085
  }
1086

1087
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
288✔
1088
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
17✔
1089
    code = 0;
17✔
1090
  }
1091

1092
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
288✔
1093
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
18✔
1094
    code = 0;
18✔
1095
  }
1096

1097
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
288!
1098
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
1✔
1099
    code = 0;
1✔
1100
  }
1101

1102
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
288!
1103
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
6✔
1104
    code = 0;
6✔
1105
  }
1106

1107
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
288!
1108
    pDb->cfg.walLevel = pAlter->walLevel;
4✔
1109
    code = 0;
4✔
1110
  }
1111

1112
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
288!
1113
#if 1
UNCOV
1114
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1115
#else
1116
    pDb->cfg.strict = pAlter->strict;
1117
    code = 0;
1118
#endif
1119
  }
1120

1121
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
288!
1122
    pDb->cfg.cacheLast = pAlter->cacheLast;
199✔
1123
    code = 0;
199✔
1124
  }
1125

1126
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
288✔
1127
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
146✔
1128
    code = 0;
146✔
1129
  }
1130

1131
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
288!
1132
    pDb->cfg.replications = pAlter->replications;
13✔
1133
    pDb->vgVersion++;
13✔
1134
    code = 0;
13✔
1135
  }
1136

1137
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
288!
UNCOV
1138
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
×
UNCOV
1139
    pDb->vgVersion++;
×
UNCOV
1140
    code = 0;
×
1141
  }
1142

1143
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
288!
UNCOV
1144
    pDb->cfg.minRows = pAlter->minRows;
×
UNCOV
1145
    pDb->vgVersion++;
×
UNCOV
1146
    code = 0;
×
1147
  }
1148

1149
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
288✔
1150
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
44✔
1151
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
20✔
1152
    pDb->vgVersion++;
20✔
1153
    code = 0;
20✔
1154
  }
1155

1156
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
288!
UNCOV
1157
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
×
UNCOV
1158
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
×
UNCOV
1159
    pDb->vgVersion++;
×
1160
    code = 0;
×
1161
  }
1162

1163
  if (pAlter->s3KeepLocal > TSDB_MIN_S3_KEEP_LOCAL && pAlter->s3KeepLocal != pDb->cfg.s3KeepLocal) {
288!
UNCOV
1164
    pDb->cfg.s3KeepLocal = pAlter->s3KeepLocal;
×
UNCOV
1165
    pDb->vgVersion++;
×
UNCOV
1166
    code = 0;
×
1167
  }
1168

1169
  if (pAlter->s3Compact >= TSDB_MIN_S3_COMPACT && pAlter->s3Compact != pDb->cfg.s3Compact) {
288!
UNCOV
1170
    pDb->cfg.s3Compact = pAlter->s3Compact;
×
UNCOV
1171
    pDb->vgVersion++;
×
UNCOV
1172
    code = 0;
×
1173
  }
1174

1175
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
288✔
1176
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
2✔
1177
    pDb->vgVersion++;
2✔
1178
    code = 0;
2✔
1179
  }
1180

1181
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
288!
UNCOV
1182
    pDb->cfg.compactInterval = pAlter->compactInterval;
×
UNCOV
1183
    pDb->vgVersion++;
×
UNCOV
1184
    code = 0;
×
1185
  }
1186

1187
  bool compactTimeRangeChanged = false;
288✔
1188
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
288!
1189
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
288!
1190
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
288!
UNCOV
1191
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
×
UNCOV
1192
    compactTimeRangeChanged = true;
×
UNCOV
1193
    code = 0;
×
1194
  }
1195

1196
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
288!
1197
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
288!
UNCOV
1198
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
×
UNCOV
1199
    compactTimeRangeChanged = true;
×
UNCOV
1200
    code = 0;
×
1201
  }
1202
  if(compactTimeRangeChanged) {
288!
1203
    pDb->vgVersion++;
×
1204
  }
1205

1206
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
288!
UNCOV
1207
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
×
UNCOV
1208
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
×
UNCOV
1209
    pDb->vgVersion++;
×
1210
    code = 0;
×
1211
  }
1212

1213
  TAOS_RETURN(code);
288✔
1214
}
1215

1216
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
259✔
1217
  int32_t  code = 0;
259✔
1218
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
259✔
1219
  if (pRedoRaw == NULL) {
259!
UNCOV
1220
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1221
    if (terrno != 0) code = terrno;
×
UNCOV
1222
    TAOS_RETURN(code);
×
1223
  }
1224
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
259!
1225
    sdbFreeRaw(pRedoRaw);
×
UNCOV
1226
    TAOS_RETURN(code);
×
1227
  }
1228

1229
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
259!
UNCOV
1230
    sdbFreeRaw(pRedoRaw);
×
UNCOV
1231
    TAOS_RETURN(code);
×
1232
  }
1233
  return 0;
259✔
1234
}
1235

1236
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
259✔
1237
  int32_t  code = 0;
259✔
1238
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
259✔
1239
  if (pCommitRaw == NULL) {
259!
UNCOV
1240
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1241
    if (terrno != 0) code = terrno;
×
UNCOV
1242
    TAOS_RETURN(code);
×
1243
  }
1244
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
259!
1245
    sdbFreeRaw(pCommitRaw);
×
UNCOV
1246
    TAOS_RETURN(code);
×
1247
  }
1248

1249
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
259!
UNCOV
1250
    sdbFreeRaw(pCommitRaw);
×
UNCOV
1251
    TAOS_RETURN(code);
×
1252
  }
1253
  TAOS_RETURN(code);
259✔
1254
}
1255

1256
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
259✔
1257
  int32_t code = 0, lino = 0;
259✔
1258
  SSdb   *pSdb = pMnode->pSdb;
259✔
1259
  void   *pIter = NULL;
259✔
1260
  SVgObj *pVgroup = NULL;
259✔
1261
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
259✔
1262

1263
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
259!
1264

1265
  while (1) {
1266
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
7,575✔
1267
    if (pIter == NULL) break;
7,575✔
1268

1269
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
7,317✔
1270
      SVgObj newVgroup = {0};
737✔
1271
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
737✔
1272
                      _err);
1273

1274
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
736!
UNCOV
1275
        if (pNewDb->cfg.withArbitrator) {
×
UNCOV
1276
          SArbGroup arbGroup = {0};
×
UNCOV
1277
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1278
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1279
        } else {
1280
          SArbGroup arbGroup = {0};
×
1281
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
UNCOV
1282
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1283
        }
1284
      }
1285
    }
1286

1287
    sdbRelease(pSdb, pVgroup);
7,316✔
1288
  }
1289

1290
  taosArrayDestroy(pArray);
258✔
1291
  TAOS_RETURN(code);
258✔
1292

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

1296
  sdbCancelFetch(pSdb, pIter);
1✔
1297
  sdbRelease(pSdb, pVgroup);
1✔
1298
  taosArrayDestroy(pArray);
1✔
1299
  TAOS_RETURN(code);
1✔
1300
}
1301

1302
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
260✔
1303
  int32_t code = -1;
260✔
1304
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
260✔
1305
  if (pTrans == NULL) {
260!
UNCOV
1306
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1307
    if (terrno != 0) code = terrno;
×
UNCOV
1308
    TAOS_RETURN(code);
×
1309
  }
1310
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
260!
1311
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
260!
1312

1313
  mndTransSetDbName(pTrans, pOld->name, NULL);
260✔
1314
  mndTransSetGroupParallel(pTrans);
260✔
1315
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
260!
1316
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
260✔
1317

1318
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
259!
1319
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
259!
1320
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
259✔
1321

1322
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
258!
1323
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
258!
1324
  code = 0;
258✔
1325

1326
_OVER:
260✔
1327
  mndTransDrop(pTrans);
260✔
1328
  TAOS_RETURN(code);
260✔
1329
}
1330

1331
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
289✔
1332
  SMnode     *pMnode = pReq->info.node;
289✔
1333
  int32_t     code = -1;
289✔
1334
  SDbObj     *pDb = NULL;
289✔
1335
  SAlterDbReq alterReq = {0};
289✔
1336
  SDbObj      dbObj = {0};
289✔
1337

1338
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
289!
1339

1340
  mInfo("db:%s, start to alter", alterReq.db);
289!
1341

1342
  pDb = mndAcquireDb(pMnode, alterReq.db);
289✔
1343
  if (pDb == NULL) {
289!
UNCOV
1344
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1345
    if (terrno != 0) code = terrno;
×
UNCOV
1346
    goto _OVER;
×
1347
  }
1348

1349
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_ALTER_DB, pDb), NULL, _OVER);
289✔
1350

1351
  if (alterReq.replications == 2) {
288✔
1352
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
2!
1353
  }
1354

1355
  int32_t numOfTopics = 0;
288✔
1356
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
288!
1357

1358
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
288!
UNCOV
1359
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
UNCOV
1360
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1361
           numOfTopics);
1362
    goto _OVER;
×
1363
  }
1364

1365
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
288✔
1366
  if (dbObj.cfg.pRetensions != NULL) {
288!
UNCOV
1367
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
UNCOV
1368
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1369
  }
1370

1371
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
288✔
1372
  if (code != 0) {
288✔
1373
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
26!
1374
    goto _OVER;
26✔
1375
  }
1376

1377
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
262✔
1378

1379
  dbObj.cfgVersion++;
260✔
1380
  dbObj.updateTime = taosGetTimestampMs();
260✔
1381
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
260✔
1382

1383
  if (dbObj.cfg.replications != pDb->cfg.replications) {
260✔
1384
    // return quickly, operation executed asynchronously
1385
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
11!
1386
  } else {
1387
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
249!
1388
  }
1389

1390
  SName name = {0};
260✔
1391
  if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
260!
UNCOV
1392
    mError("db:%s, failed to parse db name", alterReq.db);
×
1393

1394
  auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen);
260✔
1395

1396
_OVER:
289✔
1397
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
289✔
1398
    if (terrno != 0) code = terrno;
5!
1399
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
5!
1400
  }
1401

1402
  mndReleaseDb(pMnode, pDb);
289✔
1403
  taosArrayDestroy(dbObj.cfg.pRetensions);
289✔
1404
  tFreeSAlterDbReq(&alterReq);
289✔
1405

1406
  TAOS_RETURN(code);
289✔
1407
}
1408

1409
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb) {
82,103✔
1410
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
82,103✔
1411
  cfgRsp->dbId = pDb->uid;
82,103✔
1412
  cfgRsp->cfgVersion = pDb->cfgVersion;
82,103✔
1413
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
82,103✔
1414
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
82,103✔
1415
  cfgRsp->buffer = pDb->cfg.buffer;
82,103✔
1416
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
82,103✔
1417
  cfgRsp->pageSize = pDb->cfg.pageSize;
82,103✔
1418
  cfgRsp->pages = pDb->cfg.pages;
82,103✔
1419
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
82,103✔
1420
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
82,103✔
1421
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
82,103✔
1422
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
82,103✔
1423
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
82,103✔
1424
  cfgRsp->minRows = pDb->cfg.minRows;
82,103✔
1425
  cfgRsp->maxRows = pDb->cfg.maxRows;
82,103✔
1426
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
82,103✔
1427
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
82,103✔
1428
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
82,103✔
1429
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
82,103✔
1430
  cfgRsp->walLevel = pDb->cfg.walLevel;
82,103✔
1431
  cfgRsp->precision = pDb->cfg.precision;
82,103✔
1432
  cfgRsp->compression = pDb->cfg.compression;
82,103✔
1433
  cfgRsp->replications = pDb->cfg.replications;
82,103✔
1434
  cfgRsp->strict = pDb->cfg.strict;
82,103✔
1435
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
82,103✔
1436
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
82,103✔
1437
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
82,103✔
1438
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
82,103✔
1439
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
82,103✔
1440
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
82,103✔
1441
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
82,103✔
1442
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
82,103✔
1443
  cfgRsp->schemaless = pDb->cfg.schemaless;
82,103✔
1444
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
82,103✔
1445
  cfgRsp->s3ChunkSize = pDb->cfg.s3ChunkSize;
82,103✔
1446
  cfgRsp->s3KeepLocal = pDb->cfg.s3KeepLocal;
82,103✔
1447
  cfgRsp->s3Compact = pDb->cfg.s3Compact;
82,103✔
1448
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
82,103✔
1449
  cfgRsp->encryptAlgorithm = pDb->cfg.encryptAlgorithm;
82,103✔
1450
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
82,103✔
1451
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
82,103✔
1452
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
82,103✔
1453
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
82,103✔
1454
}
82,103✔
1455

1456
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
75,165✔
1457
  SMnode   *pMnode = pReq->info.node;
75,165✔
1458
  int32_t   code = -1;
75,165✔
1459
  SDbObj   *pDb = NULL;
75,165✔
1460
  SDbCfgReq cfgReq = {0};
75,165✔
1461
  SDbCfgRsp cfgRsp = {0};
75,165✔
1462

1463
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
75,165!
1464

1465
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
75,165!
1466
    pDb = mndAcquireDb(pMnode, cfgReq.db);
75,165✔
1467
    if (pDb == NULL) {
75,165✔
1468
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
842✔
1469
      if (terrno != 0) code = terrno;
842!
1470
      goto _OVER;
842✔
1471
    }
1472

1473
    mndDumpDbCfgInfo(&cfgRsp, pDb);
74,323✔
1474
  }
1475

1476
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
74,323✔
1477
  void   *pRsp = rpcMallocCont(contLen);
74,315✔
1478
  if (pRsp == NULL) {
74,321!
UNCOV
1479
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1480
    if (terrno != 0) code = terrno;
×
UNCOV
1481
    goto _OVER;
×
1482
  }
1483

1484
  int32_t ret = 0;
74,321✔
1485
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
74,321!
UNCOV
1486
    code = ret;
×
UNCOV
1487
    goto _OVER;
×
1488
  }
1489

1490
  pReq->info.rsp = pRsp;
74,323✔
1491
  pReq->info.rspLen = contLen;
74,323✔
1492

1493
  code = 0;
74,323✔
1494

1495
_OVER:
75,165✔
1496

1497
  tFreeSDbCfgRsp(&cfgRsp);
75,165✔
1498

1499
  if (code != 0) {
75,164✔
1500
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
842!
1501
  }
1502

1503
  mndReleaseDb(pMnode, pDb);
75,164✔
1504

1505
  TAOS_RETURN(code);
75,165✔
1506
}
1507

1508
static int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,462✔
1509
  int32_t  code = 0;
2,462✔
1510
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
2,462✔
1511
  if (pRedoRaw == NULL) {
2,462!
UNCOV
1512
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1513
    if (terrno != 0) code = terrno;
×
UNCOV
1514
    return -1;
×
1515
  }
1516
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
2,462!
1517
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
2,462!
1518

1519
  SSdb *pSdb = pMnode->pSdb;
2,462✔
1520
  void *pIter = NULL;
2,462✔
1521

1522
  while (1) {
12✔
1523
    SArbGroup *pArbGroup = NULL;
2,474✔
1524
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
2,474✔
1525
    if (pIter == NULL) break;
2,474✔
1526

1527
    if (pArbGroup->dbUid == pDb->uid) {
12!
1528
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
12!
UNCOV
1529
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1530
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
1531
        TAOS_RETURN(code);
×
1532
      }
1533
    }
1534

1535
    sdbRelease(pSdb, pArbGroup);
12✔
1536
  }
1537

1538
  TAOS_RETURN(code);
2,462✔
1539
}
1540

1541
static int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,462✔
1542
  int32_t  code = 0;
2,462✔
1543
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
2,462✔
1544
  if (pCommitRaw == NULL) {
2,462!
UNCOV
1545
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1546
    if (terrno != 0) code = terrno;
×
UNCOV
1547
    return -1;
×
1548
  }
1549
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
2,462!
1550
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
2,462!
1551

1552
  SSdb *pSdb = pMnode->pSdb;
2,462✔
1553
  void *pIter = NULL;
2,462✔
1554

1555
  while (1) {
12✔
1556
    SArbGroup *pArbGroup = NULL;
2,474✔
1557
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
2,474✔
1558
    if (pIter == NULL) break;
2,474✔
1559

1560
    if (pArbGroup->dbUid == pDb->uid) {
12!
1561
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
12!
UNCOV
1562
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1563
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
1564
        TAOS_RETURN(code);
×
1565
      }
1566
    }
1567

1568
    sdbRelease(pSdb, pArbGroup);
12✔
1569
  }
1570

1571
  while (1) {
16,153✔
1572
    SVgObj *pVgroup = NULL;
18,615✔
1573
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
18,615✔
1574
    if (pIter == NULL) break;
18,615✔
1575

1576
    if (pVgroup->dbUid == pDb->uid) {
16,153✔
1577
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
5,115✔
1578
      if (pVgRaw == NULL) {
5,115!
UNCOV
1579
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1580
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1581
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1582
        if (terrno != 0) code = terrno;
×
1583
        TAOS_RETURN(code);
×
1584
      }
1585
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
5,115!
1586
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1587
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1588
        TAOS_RETURN(code);
×
1589
      }
1590
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
5,115!
1591
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1592
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1593
        TAOS_RETURN(code);
×
1594
      }
1595
    }
1596

1597
    sdbRelease(pSdb, pVgroup);
16,153✔
1598
  }
1599

1600
  while (1) {
22,378✔
1601
    SStbObj *pStb = NULL;
24,840✔
1602
    ESdbStatus status;
1603
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
24,840✔
1604
    if (pIter == NULL) break;
24,840✔
1605

1606
    if (pStb->dbUid == pDb->uid) {
22,378✔
1607
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
4,299✔
1608
      if (pStbRaw == NULL) {
4,299!
UNCOV
1609
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1610
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1611
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1612
        if (terrno != 0) code = terrno;
×
1613
        return -1;
×
1614
      }
1615
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
4,299!
1616
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1617
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1618
        return -1;
×
1619
      }
1620
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
4,299!
1621
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1622
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1623
        return -1;
×
1624
      }
1625
    }
1626

1627
    sdbRelease(pSdb, pStb);
22,378✔
1628
  }
1629

1630
  TAOS_RETURN(code);
2,462✔
1631
}
1632

1633
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
5,112✔
1634
  int32_t code = 0;
5,112✔
1635
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
10,382✔
1636
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
5,270✔
1637
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
5,270!
1638
  }
1639

1640
  TAOS_RETURN(code);
5,112✔
1641
}
1642

1643
static int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,461✔
1644
  int32_t code = 0;
2,461✔
1645
  SSdb   *pSdb = pMnode->pSdb;
2,461✔
1646
  void   *pIter = NULL;
2,461✔
1647

1648
  while (1) {
16,144✔
1649
    SVgObj *pVgroup = NULL;
18,605✔
1650
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
18,605✔
1651
    if (pIter == NULL) break;
18,605✔
1652

1653
    if (pVgroup->dbUid == pDb->uid) {
16,144✔
1654
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
5,112!
UNCOV
1655
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1656
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1657
        TAOS_RETURN(code);
×
1658
      }
1659
    }
1660

1661
    sdbRelease(pSdb, pVgroup);
16,144✔
1662
  }
1663

1664
  TAOS_RETURN(code);
2,461✔
1665
}
1666

1667
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
4,676✔
1668
  int32_t    code = 0;
4,676✔
1669
  SDropDbRsp dropRsp = {0};
4,676✔
1670
  if (pDb != NULL) {
4,676✔
1671
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
2,461✔
1672
    dropRsp.uid = pDb->uid;
2,461✔
1673
  }
1674

1675
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
4,676✔
1676
  void   *pRsp = NULL;
4,676✔
1677
  if (useRpcMalloc) {
4,676✔
1678
    pRsp = rpcMallocCont(rspLen);
2,215✔
1679
  } else {
1680
    pRsp = taosMemoryMalloc(rspLen);
2,461!
1681
  }
1682

1683
  if (pRsp == NULL) {
4,676!
UNCOV
1684
    code = terrno;
×
UNCOV
1685
    TAOS_RETURN(code);
×
1686
  }
1687

1688
  int32_t ret = 0;
4,676✔
1689
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
4,676!
1690
  *pRspLen = rspLen;
4,676✔
1691
  *ppRsp = pRsp;
4,676✔
1692
  TAOS_RETURN(code);
4,676✔
1693
}
1694

1695
static int32_t mndRemoveAllStbUser(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,461✔
1696
  int32_t code = -1;
2,461✔
1697
  void   *pIter = NULL;
2,461✔
1698
  while (1) {
22,368✔
1699
    SStbObj   *pStb = NULL;
24,829✔
1700
    ESdbStatus status;
1701
    pIter = sdbFetchAll(pMnode->pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
24,829✔
1702
    if (pIter == NULL) break;
24,829✔
1703

1704
    if (pStb->dbUid == pDb->uid) {
22,368✔
1705
      if ((code = mndUserRemoveStb(pMnode, pTrans, pStb->name)) != 0) {
4,295!
UNCOV
1706
        sdbCancelFetch(pMnode->pSdb, pIter);
×
UNCOV
1707
        sdbRelease(pMnode->pSdb, pStb);
×
UNCOV
1708
        return code;
×
1709
      }
1710
    }
1711

1712
    sdbRelease(pMnode->pSdb, pStb);
22,368✔
1713
  }
1714

1715
  code = 0;
2,461✔
1716

1717
_OVER:
2,461✔
1718
  return code;
2,461✔
1719
}
1720

1721
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
2,463✔
1722
  int32_t code = -1;
2,463✔
1723

1724
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
2,463✔
1725
  if (pTrans == NULL) {
2,463!
UNCOV
1726
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1727
    if (terrno != 0) code = terrno;
×
UNCOV
1728
    goto _OVER;
×
1729
  }
1730

1731
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
2,463!
1732

1733
  mndTransSetDbName(pTrans, pDb->name, NULL);
2,463✔
1734
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
2,463!
1735

1736
  if (mndTopicExistsForDb(pMnode, pDb)) {
2,463✔
1737
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
1✔
1738
    goto _OVER;
1✔
1739
  }
1740

1741
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
2,462!
1742
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
2,462!
1743
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,462✔
1744
#ifdef TD_ENTERPRISE
1745
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,461!
1746
#endif
1747
  TAOS_CHECK_GOTO(mndDropSmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,461!
1748
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,461!
1749
  TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
2,461!
1750
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
2,461!
1751
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb->name), NULL, _OVER);
2,461!
1752
  TAOS_CHECK_GOTO(mndRemoveAllStbUser(pMnode, pTrans, pDb), NULL, _OVER);
2,461!
1753

1754
  int32_t rspLen = 0;
2,461✔
1755
  void   *pRsp = NULL;
2,461✔
1756
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
2,461!
1757
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
2,461✔
1758

1759
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
2,461!
1760
  code = 0;
2,461✔
1761

1762
_OVER:
2,463✔
1763
  mndTransDrop(pTrans);
2,463✔
1764
  TAOS_RETURN(code);
2,463✔
1765
}
1766

1767
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
4,764✔
1768
  SMnode    *pMnode = pReq->info.node;
4,764✔
1769
  int32_t    code = -1;
4,764✔
1770
  SDbObj    *pDb = NULL;
4,764✔
1771
  SDropDbReq dropReq = {0};
4,764✔
1772

1773
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
4,764!
1774

1775
  mInfo("db:%s, start to drop", dropReq.db);
4,764!
1776

1777
  pDb = mndAcquireDb(pMnode, dropReq.db);
4,764✔
1778
  if (pDb == NULL) {
4,764✔
1779
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
2,285✔
1780
    if (terrno != 0) code = terrno;
2,285!
1781
    if (dropReq.ignoreNotExists) {
2,285✔
1782
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
2,215✔
1783
    }
1784
    goto _OVER;
2,285✔
1785
  }
1786

1787
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_DB, pDb), NULL, _OVER);
2,479✔
1788

1789
  SSdb *pSdb = pMnode->pSdb;
2,464✔
1790
  void *pIter = NULL;
2,464✔
1791

1792
  while (1) {
16,155✔
1793
    SVgObj *pVgroup = NULL;
18,619✔
1794
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
18,619✔
1795
    if (pIter == NULL) break;
18,619✔
1796

1797
    if (pVgroup->dbUid == pDb->uid) {
16,156✔
1798
      bool isFound = false;
5,117✔
1799
      for (int32_t i = 0; i < pVgroup->replica; i++) {
10,391✔
1800
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
5,275✔
1801
          isFound = true;
1✔
1802
          break;
1✔
1803
        }
1804
      }
1805
      if (!isFound) {
5,117✔
1806
        sdbRelease(pSdb, pVgroup);
5,116✔
1807
        continue;
5,116✔
1808
      }
1809
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
1✔
1810
      sdbCancelFetch(pSdb, pIter);
1✔
1811
      sdbRelease(pSdb, pVgroup);
1✔
1812
      goto _OVER;
1✔
1813
    }
1814

1815
    sdbRelease(pSdb, pVgroup);
11,039✔
1816
  }
1817

1818
  code = mndDropDb(pMnode, pReq, pDb);
2,463✔
1819
  if (code == TSDB_CODE_SUCCESS) {
2,463✔
1820
    code = TSDB_CODE_ACTION_IN_PROGRESS;
2,461✔
1821
  }
1822

1823
  SName name = {0};
2,463✔
1824
  if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
2,463!
UNCOV
1825
    mError("db:%s, failed to parse db name", dropReq.db);
×
1826

1827
  auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen);
2,463✔
1828

1829
_OVER:
4,764✔
1830
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
4,764✔
1831
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
88!
1832
  }
1833

1834
  mndReleaseDb(pMnode, pDb);
4,764✔
1835
  tFreeSDropDbReq(&dropReq);
4,764✔
1836
  TAOS_RETURN(code);
4,764✔
1837
}
1838

1839
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
3,318,442✔
1840
  int32_t numOfTables = 0;
3,318,442✔
1841
  int32_t vindex = 0;
3,318,442✔
1842
  SSdb   *pSdb = pMnode->pSdb;
3,318,442✔
1843

1844
  void *pIter = NULL;
3,318,442✔
1845
  while (vindex < pDb->cfg.numOfVgroups) {
99,827,974✔
1846
    SVgObj *pVgroup = NULL;
96,507,430✔
1847
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
96,507,430✔
1848
    if (pIter == NULL) break;
96,505,512!
1849

1850
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
96,505,512✔
1851
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
6,654,367✔
1852
      vindex++;
6,654,367✔
1853
    }
1854

1855
    sdbRelease(pSdb, pVgroup);
96,504,021✔
1856
  }
1857

1858
  sdbCancelFetch(pSdb, pIter);
3,320,544✔
1859
  return numOfTables;
3,318,523✔
1860
}
1861

1862
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
3,516,644✔
1863
  int32_t vindex = 0;
3,516,644✔
1864
  SSdb   *pSdb = pMnode->pSdb;
3,516,644✔
1865

1866
  void *pIter = NULL;
3,516,644✔
1867
  while (1) {
105,465,608✔
1868
    SVgObj *pVgroup = NULL;
108,982,252✔
1869
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
108,982,252✔
1870
    if (pIter == NULL) break;
108,979,592✔
1871

1872
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
108,679,857!
1873
      SVgroupInfo vgInfo = {0};
21,272,124✔
1874
      vgInfo.vgId = pVgroup->vgId;
21,272,124✔
1875
      vgInfo.hashBegin = pVgroup->hashBegin;
21,272,124✔
1876
      vgInfo.hashEnd = pVgroup->hashEnd;
21,272,124✔
1877
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
21,272,124✔
1878
      vgInfo.epSet.numOfEps = pVgroup->replica;
21,272,124✔
1879
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
42,557,783✔
1880
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
21,285,476✔
1881
        SEp       *pEp = &vgInfo.epSet.eps[gid];
21,285,476✔
1882
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
21,285,476✔
1883
        if (pDnode != NULL) {
21,285,442✔
1884
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
21,285,267✔
1885
          pEp->port = pDnode->port;
21,285,267✔
1886
        }
1887
        mndReleaseDnode(pMnode, pDnode);
21,285,442✔
1888
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
21,285,659✔
1889
          vgInfo.epSet.inUse = gid;
21,270,067✔
1890
        }
1891
      }
1892
      vindex++;
21,272,307✔
1893
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
21,272,044!
UNCOV
1894
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
1895
      }
1896
    }
1897

1898
    sdbRelease(pSdb, pVgroup);
108,679,785✔
1899

1900
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
108,682,550✔
1901
      sdbCancelFetch(pSdb, pIter);
3,216,942✔
1902
      break;
3,216,926✔
1903
    }
1904
  }
1905
}
3,516,661✔
1906

1907
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
3,295,745✔
1908
  int32_t code = 0;
3,295,745✔
1909
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
3,295,745✔
1910
  if (pRsp->pVgroupInfos == NULL) {
3,295,660!
UNCOV
1911
    code = terrno;
×
UNCOV
1912
    TAOS_RETURN(code);
×
1913
  }
1914

1915
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
3,295,660✔
1916

1917
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
3,295,744!
1918
      pReq->stateTs < pDb->stateTs) {
82,545✔
1919
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
3,213,775✔
1920
  }
1921

1922
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
3,295,737✔
1923
  pRsp->uid = pDb->uid;
3,295,737✔
1924
  pRsp->vgVersion = pDb->vgVersion;
3,295,737✔
1925
  pRsp->stateTs = pDb->stateTs;
3,295,737✔
1926
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
3,295,737✔
1927
  pRsp->hashMethod = pDb->cfg.hashMethod;
3,295,682✔
1928
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
3,295,682✔
1929
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
3,295,682✔
1930
  TAOS_RETURN(code);
3,295,682✔
1931
}
1932

1933
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
3,590,220✔
1934
  SMnode   *pMnode = pReq->info.node;
3,590,220✔
1935
  int32_t   code = -1;
3,590,220✔
1936
  SDbObj   *pDb = NULL;
3,590,220✔
1937
  SUseDbReq usedbReq = {0};
3,590,220✔
1938
  SUseDbRsp usedbRsp = {0};
3,590,220✔
1939

1940
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
3,590,220!
1941

1942
  char *p = strchr(usedbReq.db, '.');
3,590,698✔
1943
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
3,590,698✔
1944
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
298,665✔
1945
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
298,665✔
1946
    if (usedbReq.vgVersion < vgVersion) {
298,661✔
1947
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
298,624✔
1948
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
298,625!
1949

1950
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
298,625✔
1951
      usedbRsp.vgVersion = vgVersion++;
298,625✔
1952
    } else {
1953
      usedbRsp.vgVersion = usedbReq.vgVersion;
37✔
1954
    }
1955
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
298,662✔
1956
    code = 0;
298,662✔
1957
  } else {
1958
    pDb = mndAcquireDb(pMnode, usedbReq.db);
3,292,033✔
1959
    if (pDb == NULL) {
3,292,020✔
1960
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
1,109✔
1961
      usedbRsp.uid = usedbReq.dbId;
1,109✔
1962
      usedbRsp.vgVersion = usedbReq.vgVersion;
1,109✔
1963
      usedbRsp.errCode = terrno;
1,109✔
1964

1965
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
1,109✔
1966
      if (terrno != 0) code = terrno;
1,109!
1967
      goto _OVER;
1,109✔
1968
    } else {
1969
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_USE_DB, pDb), NULL, _OVER);
3,290,911✔
1970

1971
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
3,290,955!
1972

1973
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
3,290,898✔
1974
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
1975
      code = 0;
3,290,901✔
1976
    }
1977
  }
1978

1979
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
3,589,563✔
1980
  void   *pRsp = rpcMallocCont(contLen);
3,589,533✔
1981
  if (pRsp == NULL) {
3,589,473!
UNCOV
1982
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
1983
    code = -1;
×
UNCOV
1984
    goto _OVER;
×
1985
  }
1986

1987
  int32_t ret = 0;
3,589,473✔
1988
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
3,589,473!
UNCOV
1989
    code = ret;
×
UNCOV
1990
    goto _OVER;
×
1991
  }
1992

1993
  pReq->info.rsp = pRsp;
3,589,584✔
1994
  pReq->info.rspLen = contLen;
3,589,584✔
1995

1996
_OVER:
3,590,718✔
1997
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
3,590,718!
1998
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
1,134!
1999
  }
2000

2001
  mndReleaseDb(pMnode, pDb);
3,590,718✔
2002
  tFreeSUsedbRsp(&usedbRsp);
3,590,750✔
2003

2004
  TAOS_RETURN(code);
3,590,721✔
2005
}
2006

2007
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
27,951✔
2008
  int32_t       code = 0;
27,951✔
2009
  SDbHbBatchRsp batchRsp = {0};
27,951✔
2010
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
27,951✔
2011
  if (batchRsp.pArray == NULL) {
27,951!
UNCOV
2012
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2013
    TAOS_RETURN(code);
×
2014
  }
2015

2016
  for (int32_t i = 0; i < numOfDbs; ++i) {
58,210✔
2017
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
30,259✔
2018
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
30,259✔
2019
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
30,259✔
2020
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
30,259✔
2021
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
30,259✔
2022
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
30,259✔
2023
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
30,259✔
2024

2025
    SDbHbRsp rsp = {0};
30,259✔
2026
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
30,259✔
2027
    rsp.dbId = pDbCacheInfo->dbId;
30,259✔
2028

2029
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
30,259✔
2030
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
28,698✔
2031
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
7,174✔
2032
      if (pDbCacheInfo->vgVersion >= vgVersion) {
7,174✔
2033
        continue;
16,328✔
2034
      }
2035

2036
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
1,110!
2037
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
1,110✔
2038
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
1,110✔
2039

2040
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
1,110✔
2041
      rsp.useDbRsp->vgVersion = vgVersion++;
1,110✔
2042

2043
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
1,110✔
2044

2045
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
2,220!
UNCOV
2046
        if (terrno != 0) code = terrno;
×
UNCOV
2047
        return code;
×
2048
      }
2049

2050
      continue;
1,110✔
2051
    }
2052

2053
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
23,085✔
2054
    if (pDb == NULL) {
23,085✔
2055
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
310✔
2056
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
310!
2057
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
310✔
2058
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
310✔
2059
      rsp.useDbRsp->vgVersion = -1;
310✔
2060
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
620!
UNCOV
2061
        if (terrno != 0) code = terrno;
×
UNCOV
2062
        return code;
×
2063
      }
2064
      continue;
310✔
2065
    }
2066

2067
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
22,775✔
2068

2069
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
22,775✔
2070
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
14,931!
2071
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
12,057✔
2072
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
8,844✔
2073
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2074
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2075
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2076
      mndReleaseDb(pMnode, pDb);
8,844✔
2077
      continue;
8,844✔
2078
    } else {
2079
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
13,931✔
2080
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2081
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2082
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2083
    }
2084

2085
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
13,931✔
2086
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
7,780!
2087
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb);
7,780✔
2088
    }
2089

2090
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
13,931✔
2091
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
12,478!
2092
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
12,478!
2093
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
12,478!
2094
        bool    exist = false;
12,478✔
2095
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
12,478✔
2096
        if (TSDB_CODE_SUCCESS != code) {
12,478!
UNCOV
2097
          mndReleaseDb(pMnode, pDb);
×
UNCOV
2098
          if (code != TSDB_CODE_NEED_RETRY) {
×
UNCOV
2099
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2100
          } else {
2101
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2102
          }
UNCOV
2103
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2104
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
UNCOV
2105
          continue;
×
2106
        }
2107
        rsp.dbTsmaVersion = pDb->tsmaVersion;
12,478✔
2108
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
12,478✔
2109
      }
2110
    }
2111

2112
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
13,931!
2113
        pDbCacheInfo->stateTs != pDb->stateTs) {
13,861✔
2114
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
3,157!
2115
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
3,157✔
2116
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
3,157!
UNCOV
2117
        mndReleaseDb(pMnode, pDb);
×
UNCOV
2118
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
UNCOV
2119
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2120
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2121
        continue;
×
2122
      }
2123

2124
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
3,157✔
2125
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
3,157✔
2126
      rsp.useDbRsp->uid = pDb->uid;
3,157✔
2127
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
3,157✔
2128
      rsp.useDbRsp->stateTs = pDb->stateTs;
3,157✔
2129
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
3,157✔
2130
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
3,157✔
2131
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
3,157✔
2132
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
3,157✔
2133
    }
2134

2135
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
27,862!
UNCOV
2136
      mndReleaseDb(pMnode, pDb);
×
UNCOV
2137
      if (terrno != 0) code = terrno;
×
UNCOV
2138
      return code;
×
2139
    }
2140
    mndReleaseDb(pMnode, pDb);
13,931✔
2141
  }
2142

2143
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
27,951✔
2144
  void   *pRsp = taosMemoryMalloc(rspLen);
27,951!
2145
  if (pRsp == NULL) {
27,951!
UNCOV
2146
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2147
    tFreeSDbHbBatchRsp(&batchRsp);
×
UNCOV
2148
    return -1;
×
2149
  }
2150
  int32_t ret = 0;
27,951✔
2151
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
27,951!
2152

2153
  *ppRsp = pRsp;
27,951✔
2154
  *pRspLen = rspLen;
27,951✔
2155

2156
  tFreeSDbHbBatchRsp(&batchRsp);
27,951✔
2157
  TAOS_RETURN(code);
27,951✔
2158
}
2159

2160
static int32_t mndTrimDb(SMnode *pMnode, SDbObj *pDb) {
3✔
2161
  SSdb       *pSdb = pMnode->pSdb;
3✔
2162
  SVgObj     *pVgroup = NULL;
3✔
2163
  void       *pIter = NULL;
3✔
2164
  int32_t     code = 0;
3✔
2165
  SVTrimDbReq trimReq = {.timestamp = taosGetTimestampSec()};
3✔
2166
  int32_t     reqLen = tSerializeSVTrimDbReq(NULL, 0, &trimReq);
3✔
2167
  int32_t     contLen = reqLen + sizeof(SMsgHead);
3✔
2168

2169
  while (1) {
4✔
2170
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
7✔
2171
    if (pIter == NULL) break;
7✔
2172

2173
    SMsgHead *pHead = rpcMallocCont(contLen);
4✔
2174
    if (pHead == NULL) {
4!
UNCOV
2175
      sdbCancelFetch(pSdb, pVgroup);
×
UNCOV
2176
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2177
      continue;
×
2178
    }
2179
    pHead->contLen = htonl(contLen);
4✔
2180
    pHead->vgId = htonl(pVgroup->vgId);
4✔
2181
    int32_t ret = 0;
4✔
2182
    if ((ret = tSerializeSVTrimDbReq((char *)pHead + sizeof(SMsgHead), contLen, &trimReq)) < 0) {
4!
UNCOV
2183
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2184
      return ret;
×
2185
    }
2186

2187
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM, .pCont = pHead, .contLen = contLen};
4✔
2188
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
4✔
2189
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
4✔
2190
    if (code != 0) {
4!
UNCOV
2191
      mError("vgId:%d, failed to send vnode-trim request to vnode since 0x%x", pVgroup->vgId, code);
×
2192
    } else {
2193
      mInfo("vgId:%d, send vnode-trim request to vnode, time:%d", pVgroup->vgId, trimReq.timestamp);
4!
2194
    }
2195
    sdbRelease(pSdb, pVgroup);
4✔
2196
  }
2197

2198
  return 0;
3✔
2199
}
2200

2201
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
3✔
2202
  SMnode    *pMnode = pReq->info.node;
3✔
2203
  int32_t    code = -1;
3✔
2204
  SDbObj    *pDb = NULL;
3✔
2205
  STrimDbReq trimReq = {0};
3✔
2206

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

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

2211
  pDb = mndAcquireDb(pMnode, trimReq.db);
3✔
2212
  if (pDb == NULL) {
3!
UNCOV
2213
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2214
    if (terrno != 0) code = terrno;
×
UNCOV
2215
    goto _OVER;
×
2216
  }
2217

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

2220
  code = mndTrimDb(pMnode, pDb);
3✔
2221

2222
_OVER:
3✔
2223
  if (code != 0) {
3!
UNCOV
2224
    mError("db:%s, failed to process trim db req since %s", trimReq.db, terrstr());
×
2225
  }
2226

2227
  mndReleaseDb(pMnode, pDb);
3✔
2228
  TAOS_RETURN(code);
3✔
2229
}
2230

UNCOV
2231
static int32_t mndS3MigrateDb(SMnode *pMnode, SDbObj *pDb) {
×
UNCOV
2232
  SSdb            *pSdb = pMnode->pSdb;
×
UNCOV
2233
  SVgObj          *pVgroup = NULL;
×
2234
  void            *pIter = NULL;
×
2235
  SVS3MigrateDbReq s3migrateReq = {.timestamp = taosGetTimestampSec()};
×
2236
  int32_t          reqLen = tSerializeSVS3MigrateDbReq(NULL, 0, &s3migrateReq);
×
2237
  int32_t          contLen = reqLen + sizeof(SMsgHead);
×
2238
  int32_t          code = 0;
×
2239

2240
  while (1) {
×
2241
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
×
UNCOV
2242
    if (pIter == NULL) break;
×
2243

2244
    if (pVgroup->dbUid != pDb->uid) continue;
×
2245

UNCOV
2246
    SMsgHead *pHead = rpcMallocCont(contLen);
×
2247
    if (pHead == NULL) {
×
UNCOV
2248
      sdbCancelFetch(pSdb, pVgroup);
×
2249
      sdbRelease(pSdb, pVgroup);
×
2250
      continue;
×
2251
    }
2252
    pHead->contLen = htonl(contLen);
×
2253
    pHead->vgId = htonl(pVgroup->vgId);
×
UNCOV
2254
    int32_t ret = 0;
×
2255
    if ((ret = tSerializeSVS3MigrateDbReq((char *)pHead + sizeof(SMsgHead), contLen, &s3migrateReq)) < 0) {
×
2256
      sdbRelease(pSdb, pVgroup);
×
2257
      return ret;
×
2258
    }
2259

2260
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_S3MIGRATE, .pCont = pHead, .contLen = contLen};
×
UNCOV
2261
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
×
UNCOV
2262
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
×
2263
    if (code != 0) {
×
2264
      mError("vgId:%d, failed to send vnode-s3migrate request to vnode since 0x%x", pVgroup->vgId, code);
×
2265
    } else {
2266
      mInfo("vgId:%d, send vnode-s3migrate request to vnode, time:%d", pVgroup->vgId, s3migrateReq.timestamp);
×
2267
    }
UNCOV
2268
    sdbRelease(pSdb, pVgroup);
×
2269
  }
2270

2271
  return 0;
×
2272
}
2273

2274
static int32_t mndProcessS3MigrateDbReq(SRpcMsg *pReq) {
×
UNCOV
2275
  SMnode         *pMnode = pReq->info.node;
×
UNCOV
2276
  int32_t         code = -1;
×
2277
  SDbObj         *pDb = NULL;
×
2278
  SS3MigrateDbReq s3migrateReq = {0};
×
2279

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

UNCOV
2282
  mInfo("db:%s, start to s3migrate", s3migrateReq.db);
×
2283

UNCOV
2284
  pDb = mndAcquireDb(pMnode, s3migrateReq.db);
×
2285
  if (pDb == NULL) {
×
UNCOV
2286
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2287
    if (terrno != 0) code = terrno;
×
2288
    goto _OVER;
×
2289
  }
2290

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

UNCOV
2293
  code = mndS3MigrateDb(pMnode, pDb);
×
2294

UNCOV
2295
_OVER:
×
2296
  if (code != 0) {
×
UNCOV
2297
    mError("db:%s, failed to process s3migrate db req since %s", s3migrateReq.db, terrstr());
×
2298
  }
2299

2300
  mndReleaseDb(pMnode, pDb);
×
UNCOV
2301
  TAOS_RETURN(code);
×
2302
}
2303

2304
const char *mndGetDbStr(const char *src) {
1,922,863✔
2305
  char *pos = strstr(src, TS_PATH_DELIMITER);
1,922,863✔
2306
  if (pos != NULL) ++pos;
1,922,863✔
2307
  if (pos == NULL) return src;
1,922,863✔
2308
  return pos;
1,866,404✔
2309
}
2310

UNCOV
2311
const char *mndGetStableStr(const char *src) {
×
UNCOV
2312
  char *pos = strstr(src, TS_PATH_DELIMITER);
×
UNCOV
2313
  if (pos != NULL) ++pos;
×
2314
  if (pos == NULL) return src;
×
2315
  return mndGetDbStr(pos);
×
2316
}
2317

2318
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
12✔
2319
  int64_t v = 0;
12✔
2320
  switch (unit) {
12!
2321
    case 's':
4✔
2322
      v = val / 1000;
4✔
2323
      break;
4✔
2324
    case 'm':
2✔
2325
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
2✔
2326
      break;
2✔
UNCOV
2327
    case 'h':
×
UNCOV
2328
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
UNCOV
2329
      break;
×
2330
    case 'd':
6✔
2331
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
6✔
2332
      break;
6✔
UNCOV
2333
    case 'w':
×
UNCOV
2334
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
UNCOV
2335
      break;
×
2336
    default:
×
2337
      break;
×
2338
  }
2339

2340
  return v;
12✔
2341
}
2342

2343
static char *buildRetension(SArray *pRetension) {
679,319✔
2344
  size_t size = taosArrayGetSize(pRetension);
679,319✔
2345
  if (size == 0) {
679,291!
2346
    return NULL;
679,323✔
2347
  }
2348

UNCOV
2349
  char       *p1 = taosMemoryCalloc(1, 100);
×
2350
  SRetention *p = taosArrayGet(pRetension, 0);
2✔
2351

2352
  int32_t len = 2;
2✔
2353

2354
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2355
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2356
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2357

2358
  if (size > 1) {
2!
2359
    len += tsnprintf(p1 + len, 100 - len, ",");
2✔
2360
    p = taosArrayGet(pRetension, 1);
2✔
2361

2362
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2363
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2364
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2365
  }
2366

2367
  if (size > 2) {
2!
2368
    len += tsnprintf(p1 + len, 100 - len, ",");
2✔
2369
    p = taosArrayGet(pRetension, 2);
2✔
2370

2371
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
2✔
2372
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
2✔
2373
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
2✔
2374
  }
2375

2376
  varDataSetLen(p1, len);
2✔
2377
  return p1;
2✔
2378
}
2379

2380
static const char *getCacheModelStr(int8_t cacheModel) {
679,016✔
2381
  switch (cacheModel) {
679,016!
2382
    case TSDB_CACHE_MODEL_NONE:
211,187✔
2383
      return TSDB_CACHE_MODEL_NONE_STR;
211,187✔
2384
    case TSDB_CACHE_MODEL_LAST_ROW:
93,561✔
2385
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
93,561✔
2386
    case TSDB_CACHE_MODEL_LAST_VALUE:
52,228✔
2387
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
52,228✔
2388
    case TSDB_CACHE_MODEL_BOTH:
323,382✔
2389
      return TSDB_CACHE_MODEL_BOTH_STR;
323,382✔
UNCOV
2390
    default:
×
UNCOV
2391
      break;
×
2392
  }
2393
  return "unknown";
×
2394
}
2395

2396
static const char *getEncryptAlgorithmStr(int8_t encryptAlgorithm) {
678,349✔
2397
  switch (encryptAlgorithm) {
678,349!
2398
    case TSDB_ENCRYPT_ALGO_NONE:
678,359✔
2399
      return TSDB_ENCRYPT_ALGO_NONE_STR;
678,359✔
UNCOV
2400
    case TSDB_ENCRYPT_ALGO_SM4:
×
UNCOV
2401
      return TSDB_ENCRYPT_ALGO_SM4_STR;
×
UNCOV
2402
    default:
×
2403
      break;
×
2404
  }
2405
  return "unknown";
×
2406
}
2407

2408
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
685,411✔
2409
  if (pDb->cfg.replications == 1) return true;
685,411✔
2410

2411
  SSdb *pSdb = pMnode->pSdb;
1,288✔
2412
  void *pIter = NULL;
1,288✔
2413
  bool  isReady = true;
1,288✔
2414
  while (1) {
6,804✔
2415
    SVgObj *pVgroup = NULL;
8,092✔
2416
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
8,092✔
2417
    if (pIter == NULL) break;
8,115✔
2418

2419
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
6,804✔
2420
      bool hasLeader = false;
3,761✔
2421
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
14,935✔
2422
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
11,174✔
2423
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
9,954!
2424
          hasLeader = true;
1,220✔
2425
        }
2426
      }
2427
      if (!hasLeader) isReady = false;
3,761✔
2428
    }
2429
    sdbRelease(pSdb, pVgroup);
6,804✔
2430
  }
2431

2432
  return isReady;
1,311✔
2433
}
2434

2435
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
738,015✔
2436
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2437
  int32_t cols = 0;
738,015✔
2438
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
738,015✔
2439
  char   *buf = taosMemoryMalloc(bytes);
738,015!
2440
  if (buf == NULL) {
737,874!
UNCOV
2441
    mError("db:%s, failed to malloc buffer", pDb->name);
×
UNCOV
2442
    return;
×
2443
  }
2444
  int32_t code = 0;
737,874✔
2445
  int32_t lino = 0;
737,874✔
2446

2447
  const char *name = mndGetDbStr(pDb->name);
737,874✔
2448
  if (name != NULL) {
737,834✔
2449
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
737,753✔
2450
  } else {
2451
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
81✔
2452
  }
2453

2454
  const char *precStr = NULL;
737,834✔
2455
  switch (pDb->cfg.precision) {
737,834✔
2456
    case TSDB_TIME_PRECISION_MILLI:
737,751✔
2457
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
737,751✔
2458
      break;
737,751✔
2459
    case TSDB_TIME_PRECISION_MICRO:
21✔
2460
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
21✔
2461
      break;
21✔
2462
    case TSDB_TIME_PRECISION_NANO:
30✔
2463
      precStr = TSDB_TIME_PRECISION_NANO_STR;
30✔
2464
      break;
30✔
2465
    default:
32✔
2466
      precStr = "none";
32✔
2467
      break;
32✔
2468
  }
2469
  char precVstr[10] = {0};
737,834✔
2470
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
737,834✔
2471

2472
  char *statusStr = "ready";
737,834✔
2473
  if (objStatus == SDB_STATUS_CREATING) {
737,834✔
2474
    statusStr = "creating";
136✔
2475
  } else if (objStatus == SDB_STATUS_DROPPING) {
737,698✔
2476
    statusStr = "dropping";
1,966✔
2477
  } else {
2478
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
735,732✔
2479
      statusStr = "unsynced";
43✔
2480
    }
2481
  }
2482
  char statusVstr[24] = {0};
737,756✔
2483
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
737,756✔
2484

2485
  if (sysDb || !sysinfo) {
737,756✔
2486
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
2,040,738✔
2487
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
1,980,280✔
2488
      if (i == 0) {
1,973,296✔
2489
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
45,047!
2490
      } else if (i == 1) {
1,928,249✔
2491
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
56,006!
2492
      } else if (i == 3) {
1,872,243✔
2493
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
55,995!
2494
      } else if (i == 14) {
1,816,248✔
2495
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
56,167!
2496
      } else if (i == 15) {
1,760,081✔
2497
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
56,068!
2498
      } else {
2499
        colDataSetNULL(pColInfo, rows);
1,704,013✔
2500
      }
2501
    }
2502
  } else {
2503
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
681,514✔
2504
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
680,328!
2505

2506
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
679,318✔
2507
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
678,644!
2508

2509
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,504✔
2510
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
677,900!
2511

2512
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,382✔
2513
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
677,802!
2514

2515
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,367✔
2516
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
677,789!
2517

2518
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
678,126!
2519
    char        strictVstr[24] = {0};
678,126✔
2520
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
678,126✔
2521
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,126✔
2522
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
678,145!
2523

2524
    char    durationStr[128] = {0};
679,476✔
2525
    char    durationVstr[128] = {0};
679,476✔
2526
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
679,476✔
2527
                                       pDb->cfg.daysPerFile);
2528

2529
    varDataSetLen(durationVstr, len);
681,517✔
2530
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
681,517✔
2531
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
680,379!
2532

2533
    char keepVstr[128] = {0};
679,662✔
2534
    char keep0Str[32] = {0};
679,662✔
2535
    char keep1Str[32] = {0};
679,662✔
2536
    char keep2Str[32] = {0};
679,662✔
2537

2538
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
679,662✔
2539
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
681,761✔
2540
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
681,807✔
2541

2542
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
681,816!
2543
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
74✔
2544
    } else {
2545
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
681,742✔
2546
    }
2547
    varDataSetLen(keepVstr, len);
681,735✔
2548
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
681,735✔
2549
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
680,290!
2550

2551
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
679,302✔
2552
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
678,591!
2553

2554
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,535✔
2555
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
677,837!
2556

2557
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,322✔
2558
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
677,692!
2559

2560
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,214✔
2561
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
677,683!
2562

2563
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,193✔
2564
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
677,605!
2565

2566
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,194✔
2567
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
677,605!
2568

2569
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
677,961✔
2570
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
677,450!
2571

2572
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,880✔
2573
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
678,306!
2574

2575
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
679,195✔
2576
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
679,323✔
2577
    if (rentensionVstr == NULL) {
678,940✔
2578
      colDataSetNULL(pColInfo, rows);
678,938!
2579
    } else {
2580
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
2!
2581
      taosMemoryFree(rentensionVstr);
2!
2582
    }
2583

2584
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,940✔
2585
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
678,598!
2586

2587
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
679,017✔
2588
    char        cacheModelVstr[24] = {0};
679,016✔
2589
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
679,016✔
2590
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
679,016✔
2591
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
679,249!
2592

2593
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
679,239✔
2594
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
678,609!
2595

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

2599
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,442✔
2600
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
677,824!
2601

2602
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,143✔
2603
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
677,627!
2604

2605
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,183✔
2606
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
677,680!
2607

2608
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,085✔
2609
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
677,525!
2610

2611
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,048✔
2612
    int16_t hashPrefix = pDb->cfg.hashPrefix;
677,560✔
2613
    if (hashPrefix > 0) {
677,560✔
2614
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
14✔
2615
    } else if (hashPrefix < 0) {
677,546✔
2616
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
16✔
2617
    }
2618
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
677,560!
2619

2620
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,253✔
2621
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
677,667!
2622

2623
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,174✔
2624
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
677,645!
2625

2626
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,221✔
2627
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
677,727!
2628

2629
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,140✔
2630
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3ChunkSize, false), &lino, _OVER);
677,554!
2631

2632
    char keeplocalVstr[128] = {0};
678,311✔
2633
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.s3KeepLocal);
678,311✔
2634
    varDataSetLen(keeplocalVstr, len);
681,779✔
2635
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
681,779✔
2636
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
680,491!
2637

2638
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
679,408✔
2639
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3Compact, false), &lino, _OVER);
678,784!
2640

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

2644
    const char *encryptAlgorithmStr = getEncryptAlgorithmStr(pDb->cfg.encryptAlgorithm);
678,345✔
2645
    char        encryptAlgorithmVStr[24] = {0};
678,377✔
2646
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
678,377✔
2647
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
678,377✔
2648
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
679,230!
2649

2650
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
679,641✔
2651
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
681,751✔
2652
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
681,751!
2653
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
680,535!
2654
    }
2655

2656
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
679,786✔
2657
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
681,749✔
2658
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
681,815✔
2659
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
681,815✔
2660
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
681,815!
2661
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
680,513!
2662
    }
2663

2664
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
679,650✔
2665
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
679,650✔
2666
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
679,650✔
2667
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
680,415!
2668
    }
2669
  }
2670
_OVER:
740,035✔
2671
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
740,035!
2672
  taosMemoryFree(buf);
735,745!
2673
}
2674

2675
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
28,127✔
2676
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
28,127✔
2677
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
28,127✔
2678
  pDbObj->cfg.numOfVgroups = 0;
28,148✔
2679
  pDbObj->cfg.strict = 1;
28,148✔
2680
  pDbObj->cfg.replications = 1;
28,148✔
2681
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
28,148✔
2682
}
28,148✔
2683

2684
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
28,131✔
2685
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
28,131✔
2686
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
28,131✔
2687
  pDbObj->cfg.numOfVgroups = 0;
28,149✔
2688
  pDbObj->cfg.strict = 1;
28,149✔
2689
  pDbObj->cfg.replications = 1;
28,149✔
2690
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
28,149✔
2691
}
28,149✔
2692

2693
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
33,955,861✔
2694
  SVgObj  *pVgroup = pObj;
33,955,861✔
2695
  int32_t *numOfTables = p1;
33,955,861✔
2696
  int64_t  uid = *(int64_t *)p2;
33,955,861✔
2697
  if (pVgroup->dbUid == uid) {
33,955,861✔
2698
    *numOfTables += pVgroup->numOfTables;
1,336,983✔
2699
  }
2700
  return true;
33,955,861✔
2701
}
2702

2703
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
28,136✔
2704
  SMnode    *pMnode = pReq->info.node;
28,136✔
2705
  SSdb      *pSdb = pMnode->pSdb;
28,136✔
2706
  int32_t    numOfRows = 0;
28,136✔
2707
  SDbObj    *pDb = NULL;
28,136✔
2708
  SUserObj  *pUser = NULL;
28,136✔
2709
  ESdbStatus objStatus = 0;
28,136✔
2710

2711
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
28,136✔
2712
  if (pUser == NULL) return 0;
28,136!
2713
  bool sysinfo = pUser->sysInfo;
28,136✔
2714

2715
  // Append the information_schema database into the result.
2716
  if (!pShow->sysDbRsp) {
28,136✔
2717
    SDbObj infoschemaDb = {0};
28,135✔
2718
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
28,135✔
2719
    size_t numOfTables = 0;
28,148✔
2720
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
28,148✔
2721
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
28,141✔
2722

2723
    numOfRows += 1;
28,123✔
2724

2725
    SDbObj perfschemaDb = {0};
28,123✔
2726
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
28,123✔
2727
    numOfTables = 0;
28,149✔
2728
    getPerfDbMeta(NULL, &numOfTables);
28,149✔
2729
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
28,149✔
2730

2731
    numOfRows += 1;
28,126✔
2732
    pShow->sysDbRsp = true;
28,126✔
2733
  }
2734

2735
  while (numOfRows < rowsCapacity) {
710,055✔
2736
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
710,049✔
2737
    if (pShow->pIter == NULL) break;
710,005✔
2738

2739
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
681,856✔
2740
      int32_t numOfTables = 0;
681,921✔
2741
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
681,921✔
2742
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
681,847✔
2743
      numOfRows++;
681,039✔
2744
    }
2745

2746
    sdbRelease(pSdb, pDb);
681,062✔
2747
  }
2748

2749
  pShow->numOfRows += numOfRows;
28,155✔
2750
  mndReleaseUser(pMnode, pUser);
28,155✔
2751
  return numOfRows;
28,149✔
2752
}
2753

UNCOV
2754
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
UNCOV
2755
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
2756
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
2757
}
×
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