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

taosdata / TDengine / #3543

29 Nov 2024 02:58AM UTC coverage: 60.842% (+0.02%) from 60.819%
#3543

push

travis-ci

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

merge: from main to 3.0

120460 of 253224 branches covered (47.57%)

Branch coverage included in aggregate %.

706 of 908 new or added lines in 18 files covered. (77.75%)

2401 existing lines in 137 files now uncovered.

201633 of 276172 relevant lines covered (73.01%)

19045673.23 hits per line

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

69.39
/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 "mndDb.h"
18
#include "audit.h"
19
#include "mndArbGroup.h"
20
#include "mndCluster.h"
21
#include "mndDnode.h"
22
#include "mndIndex.h"
23
#include "mndPrivilege.h"
24
#include "mndShow.h"
25
#include "mndSma.h"
26
#include "mndStb.h"
27
#include "mndStream.h"
28
#include "mndSubscribe.h"
29
#include "mndTopic.h"
30
#include "mndTrans.h"
31
#include "mndUser.h"
32
#include "mndVgroup.h"
33
#include "mndView.h"
34
#include "systable.h"
35
#include "thttp.h"
36
#include "tjson.h"
37
#include "command.h"
38

39
#define DB_VER_NUMBER   1
40
#define DB_RESERVE_SIZE 27
41

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

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

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

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

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

83
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
2,007✔
84
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
2,007✔
85

86
  return sdbSetTable(pMnode->pSdb, table);
2,007✔
87
}
88

89
void mndCleanupDb(SMnode *pMnode) {}
2,006✔
90

91
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
24,464✔
92
  int32_t code = 0;
24,464✔
93
  int32_t lino = 0;
24,464✔
94
  terrno = TSDB_CODE_OUT_OF_MEMORY;
24,464✔
95

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

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

155
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
24,464!
156
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
24,464!
157

158
  terrno = 0;
24,464✔
159

160
_OVER:
24,464✔
161
  if (terrno != 0) {
24,464!
162
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
163
    sdbFreeRaw(pRaw);
×
164
    return NULL;
×
165
  }
166

167
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
24,464✔
168
  return pRaw;
24,464✔
169
}
170

171
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
21,369✔
172
  int32_t code = 0;
21,369✔
173
  int32_t lino = 0;
21,369✔
174
  terrno = TSDB_CODE_OUT_OF_MEMORY;
21,369✔
175
  SSdbRow *pRow = NULL;
21,369✔
176
  SDbObj  *pDb = NULL;
21,369✔
177

178
  int8_t sver = 0;
21,369✔
179
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
21,369!
180

181
  if (sver != DB_VER_NUMBER) {
21,369!
182
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
183
    goto _OVER;
×
184
  }
185

186
  pRow = sdbAllocRow(sizeof(SDbObj));
21,369✔
187
  if (pRow == NULL) goto _OVER;
21,369!
188

189
  pDb = sdbGetRowObj(pRow);
21,369✔
190
  if (pDb == NULL) goto _OVER;
21,369!
191

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

254
  SDB_GET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
21,369!
255
  taosInitRWLatch(&pDb->lock);
21,369✔
256

257
  if (pDb->cfg.s3ChunkSize == 0) {
21,369!
258
    pDb->cfg.s3ChunkSize = TSDB_DEFAULT_S3_CHUNK_SIZE;
×
259

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

263
  if (pDb->cfg.s3KeepLocal == 0) {
21,369!
264
    pDb->cfg.s3KeepLocal = TSDB_DEFAULT_S3_KEEP_LOCAL;
×
265

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

269
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
21,369!
270
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
21,369!
271
          TSDB_DEFAULT_TSDB_PAGESIZE);
272
  }
273

274
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
21,369✔
275
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
21,297!
276
  }
277

278
  terrno = 0;
21,369✔
279

280
_OVER:
21,369✔
281
  if (terrno != 0) {
21,369!
282
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
283
    taosMemoryFreeClear(pRow);
×
284
    return NULL;
×
285
  }
286

287
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
21,369✔
288
  return pRow;
21,369✔
289
}
290

291
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
4,863✔
292
  SSdb    *pSdb = pMnode->pSdb;
4,863✔
293
  SSdbRow *pRow = NULL;
4,863✔
294
  SDbObj  *pNewDb = NULL;
4,863✔
295
  int      code = -1;
4,863✔
296

297
  pRow = mndDbActionDecode(pRaw);
4,863✔
298
  if (pRow == NULL) {
4,863!
299
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
300
    if (terrno != 0) code = terrno;
×
301
    goto _OVER;
×
302
  }
303
  pNewDb = sdbGetRowObj(pRow);
4,863✔
304
  if (pNewDb == NULL) {
4,863!
305
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
306
    if (terrno != 0) code = terrno;
×
307
    goto _OVER;
×
308
  }
309

310
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
4,863✔
311
  if (pOldDb != NULL) {
4,863!
312
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
313
    sdbRelease(pMnode->pSdb, pOldDb);
×
314
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
315
    if (terrno != 0) code = terrno;
×
316
    goto _OVER;
×
317
  }
318

319
  code = 0;
4,863✔
320
_OVER:
4,863✔
321
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
4,863!
322
  taosMemoryFreeClear(pRow);
4,863!
323
  return code;
4,863✔
324
}
325

326
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
5,348✔
327
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
5,348✔
328
  return 0;
5,348✔
329
}
330

331
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
21,367✔
332
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
21,367✔
333
  taosArrayDestroy(pDb->cfg.pRetensions);
21,367✔
334
  return 0;
21,367✔
335
}
336

337
static int32_t mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew) {
8,791✔
338
  mTrace("db:%s, perform update action, old row:%p new row:%p", pOld->name, pOld, pNew);
8,791✔
339
  taosWLockLatch(&pOld->lock);
8,791✔
340
  pOld->updateTime = pNew->updateTime;
8,791✔
341
  pOld->cfgVersion = pNew->cfgVersion;
8,791✔
342
  pOld->vgVersion = pNew->vgVersion;
8,791✔
343
  pOld->cfg.numOfVgroups = pNew->cfg.numOfVgroups;
8,791✔
344
  pOld->cfg.buffer = pNew->cfg.buffer;
8,791✔
345
  pOld->cfg.pageSize = pNew->cfg.pageSize;
8,791✔
346
  pOld->cfg.pages = pNew->cfg.pages;
8,791✔
347
  pOld->cfg.cacheLastSize = pNew->cfg.cacheLastSize;
8,791✔
348
  pOld->cfg.daysPerFile = pNew->cfg.daysPerFile;
8,791✔
349
  pOld->cfg.daysToKeep0 = pNew->cfg.daysToKeep0;
8,791✔
350
  pOld->cfg.daysToKeep1 = pNew->cfg.daysToKeep1;
8,791✔
351
  pOld->cfg.daysToKeep2 = pNew->cfg.daysToKeep2;
8,791✔
352
  pOld->cfg.keepTimeOffset = pNew->cfg.keepTimeOffset;
8,791✔
353
  pOld->cfg.walFsyncPeriod = pNew->cfg.walFsyncPeriod;
8,791✔
354
  pOld->cfg.walLevel = pNew->cfg.walLevel;
8,791✔
355
  pOld->cfg.walRetentionPeriod = pNew->cfg.walRetentionPeriod;
8,791✔
356
  pOld->cfg.walRetentionSize = pNew->cfg.walRetentionSize;
8,791✔
357
  pOld->cfg.strict = pNew->cfg.strict;
8,791✔
358
  pOld->cfg.cacheLast = pNew->cfg.cacheLast;
8,791✔
359
  pOld->cfg.replications = pNew->cfg.replications;
8,791✔
360
  pOld->cfg.sstTrigger = pNew->cfg.sstTrigger;
8,791✔
361
  pOld->cfg.minRows = pNew->cfg.minRows;
8,791✔
362
  pOld->cfg.maxRows = pNew->cfg.maxRows;
8,791✔
363
  pOld->cfg.tsdbPageSize = pNew->cfg.tsdbPageSize;
8,791✔
364
  pOld->cfg.s3ChunkSize = pNew->cfg.s3ChunkSize;
8,791✔
365
  pOld->cfg.s3KeepLocal = pNew->cfg.s3KeepLocal;
8,791✔
366
  pOld->cfg.s3Compact = pNew->cfg.s3Compact;
8,791✔
367
  pOld->cfg.withArbitrator = pNew->cfg.withArbitrator;
8,791✔
368
  pOld->compactStartTime = pNew->compactStartTime;
8,791✔
369
  pOld->tsmaVersion = pNew->tsmaVersion;
8,791✔
370
  taosWUnLockLatch(&pOld->lock);
8,791✔
371
  return 0;
8,791✔
372
}
373

374
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
283,452✔
375
  SSdb *pSdb = pMnode->pSdb;
283,452✔
376
  return sdbGetTableVer(pSdb, SDB_VGROUP);
283,452✔
377
}
378

379
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
3,626,390✔
380
  SSdb   *pSdb = pMnode->pSdb;
3,626,390✔
381
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
3,626,390✔
382
  if (pDb == NULL) {
3,626,390✔
383
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
8,814✔
384
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
7,472✔
385
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
1,342✔
386
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
108✔
387
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
1,234!
388
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
1,234✔
389
    } else {
390
      terrno = TSDB_CODE_APP_ERROR;
×
391
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
392
    }
393
  }
394
  return pDb;
3,626,381✔
395
}
396

397
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
4,002,861✔
398
  SSdb *pSdb = pMnode->pSdb;
4,002,861✔
399
  sdbRelease(pSdb, pDb);
4,002,861✔
400
}
4,002,905✔
401

402
bool mndDbIsExist(SMnode *pMnode, const char *db) {
22✔
403
  SDbObj *pDb = mndAcquireDb(pMnode, db);
22✔
404
  if (pDb == NULL) {
22!
405
    return false;
×
406
  } else {
407
    mndReleaseDb(pMnode, pDb);
22✔
408
    pDb = NULL;
22✔
409
    return true;
22✔
410
  }
411
}
412

413
static int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
5,092✔
414
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
5,092✔
415
  if (pos == NULL) {
5,092!
416
    return TSDB_CODE_MND_INVALID_DB;
×
417
  }
418

419
  int32_t acctId = atoi(dbName);
5,092✔
420
  if (acctId != pUser->acctId) {
5,092!
421
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
422
  }
423

424
  return 0;
5,092✔
425
}
426

427
static int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
5,092✔
428
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
5,092✔
429

430
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
5,092!
431
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
5,092!
432
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
5,092!
433
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
5,092!
434
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
5,092!
435
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
5,092!
436
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
5,092!
437
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
5,092!
438
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
5,092!
439
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
5,092!
440
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
5,092!
441
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
5,092!
442
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
5,092!
443
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
5,092!
444
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
5,092!
445
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
5,092!
446
  if (pCfg->minRows > pCfg->maxRows) return code;
5,092✔
447
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
5,091!
448
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
5,091!
449
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
5,091!
450
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
5,091!
451
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
5,091!
452
#ifdef TD_ENTERPRISE
453
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
5,091!
454
  if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
5,091!
455
#else
456
  if (pCfg->replications != 1 && pCfg->replications != 3) return code;
457
  if (pCfg->encryptAlgorithm != TSDB_DEFAULT_ENCRYPT_ALGO) return code;
458
#endif
459

460
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
5,091!
461
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
5,091!
462
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
5,091!
463
  if (pCfg->hashMethod != 1) return code;
5,091!
464
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
5,091!
465
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
466
    TAOS_RETURN(code);
×
467
  }
468
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
5,091!
469
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
5,091!
470
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
5,091!
471
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
5,091!
472
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
5,091!
473
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
5,091!
474
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
5,091!
475
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
5,091!
476
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
5,091!
477
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
5,091!
478
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
5,091!
479

480
  if (pCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
5,091!
481
  if (pCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
5,091!
482
  if (pCfg->s3Compact < TSDB_MIN_S3_COMPACT || pCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
5,091!
483

484
  code = 0;
5,091✔
485
  TAOS_RETURN(code);
5,091✔
486
}
487

488
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
264✔
489
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
264✔
490
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
264!
491
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
264!
492
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
264!
493
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
264!
494
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
264!
495
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
264!
496
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
264!
497
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
264!
498
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
264!
499
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
264!
500
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
264!
501
    return code;
×
502
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
264!
503
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
264!
504
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
264!
505
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
264!
506
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
264!
507
#ifdef TD_ENTERPRISE
508
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
264!
509
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
264!
510
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
1!
511
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1✔
512
      return code;
1✔
513
    }
514
  }
515
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
263!
516
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
517
    return code;
×
518
  }
519
#else
520
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
521
#endif
522

523
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
263!
524
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
525
    return code;
×
526
  }
527
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
263!
528
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
529
    return code;
×
530
  }
531

532
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
263!
533
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
×
534
    return code;
×
535
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
263!
536
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
263!
537
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
263!
538
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
263!
539
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
263!
540
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
263!
541
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
263!
542
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
543
    return code;
×
544
  }
545
  if (pNewCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pNewCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
263!
546
  if (pNewCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pNewCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
263!
547
  if (pNewCfg->s3Compact < TSDB_MIN_S3_COMPACT || pNewCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
263!
548

549
  code = 0;
263✔
550
  TAOS_RETURN(code);
263✔
551
}
552

553
static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
5,092✔
554
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
5,092!
555
  if (pCfg->numOfStables < 0) pCfg->numOfStables = TSDB_DEFAULT_DB_SINGLE_STABLE;
5,092!
556
  if (pCfg->buffer < 0) pCfg->buffer = TSDB_DEFAULT_BUFFER_PER_VNODE;
5,092!
557
  if (pCfg->pageSize < 0) pCfg->pageSize = TSDB_DEFAULT_PAGESIZE_PER_VNODE;
5,092!
558
  if (pCfg->pages < 0) pCfg->pages = TSDB_DEFAULT_PAGES_PER_VNODE;
5,092!
559
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DURATION_PER_FILE;
5,092!
560
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
5,092!
561
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
5,092!
562
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
5,092!
563
  if (pCfg->keepTimeOffset < 0) pCfg->keepTimeOffset = TSDB_DEFAULT_KEEP_TIME_OFFSET;
5,092!
564
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MINROWS_FBLOCK;
5,092!
565
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAXROWS_FBLOCK;
5,092!
566
  if (pCfg->walFsyncPeriod < 0) pCfg->walFsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
5,092!
567
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
5,092!
568
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
5,092!
569
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
5,092!
570
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA;
5,092!
571
  if (pCfg->strict < 0) pCfg->strict = TSDB_DEFAULT_DB_STRICT;
5,092!
572
  if (pCfg->cacheLast < 0) pCfg->cacheLast = TSDB_DEFAULT_CACHE_MODEL;
5,092!
573
  if (pCfg->cacheLastSize <= 0) pCfg->cacheLastSize = TSDB_DEFAULT_CACHE_SIZE;
5,092!
574
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
5,092!
575
  if (pCfg->schemaless < 0) pCfg->schemaless = TSDB_DB_SCHEMALESS_OFF;
5,092!
576
  if (pCfg->walRetentionPeriod < 0 && pCfg->walRetentionPeriod != -1)
5,092!
577
    pCfg->walRetentionPeriod = TSDB_REPS_DEF_DB_WAL_RET_PERIOD;
×
578
  if (pCfg->walRetentionSize < 0 && pCfg->walRetentionSize != -1)
5,092!
579
    pCfg->walRetentionSize = TSDB_REPS_DEF_DB_WAL_RET_SIZE;
×
580
  if (pCfg->walRollPeriod < 0) pCfg->walRollPeriod = TSDB_REPS_DEF_DB_WAL_ROLL_PERIOD;
5,092!
581
  if (pCfg->walSegmentSize < 0) pCfg->walSegmentSize = TSDB_DEFAULT_DB_WAL_SEGMENT_SIZE;
5,092!
582
  if (pCfg->sstTrigger <= 0) pCfg->sstTrigger = TSDB_DEFAULT_SST_TRIGGER;
5,092!
583
  if (pCfg->tsdbPageSize <= 0) pCfg->tsdbPageSize = TSDB_DEFAULT_TSDB_PAGESIZE;
5,092!
584
  if (pCfg->s3ChunkSize <= 0) pCfg->s3ChunkSize = TSDB_DEFAULT_S3_CHUNK_SIZE;
5,092!
585
  if (pCfg->s3KeepLocal <= 0) pCfg->s3KeepLocal = TSDB_DEFAULT_S3_KEEP_LOCAL;
5,092!
586
  if (pCfg->s3Compact < 0) pCfg->s3Compact = TSDB_DEFAULT_S3_COMPACT;
5,092!
587
  if (pCfg->withArbitrator < 0) pCfg->withArbitrator = TSDB_DEFAULT_DB_WITH_ARBITRATOR;
5,092!
588
  if (pCfg->encryptAlgorithm < 0) pCfg->encryptAlgorithm = TSDB_DEFAULT_ENCRYPT_ALGO;
5,092!
589
}
5,092✔
590

591
static int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
4,629✔
592
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,629✔
593
  if (pDbRaw == NULL) return -1;
4,629!
594

595
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
4,629!
596
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
4,629!
597
  return 0;
4,629✔
598
}
599

600
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,629✔
601
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,720✔
602
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
10,091!
603
  }
604
  return 0;
4,629✔
605
}
606

607
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
608
  int32_t  code = 0;
×
609
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
610
  if (pDbRaw == NULL) {
×
611
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
612
    if (terrno != 0) code = terrno;
×
613
    TAOS_RETURN(code);
×
614
  }
615
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
616
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
617

618
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
619
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
620
    if (pVgRaw == NULL) {
×
621
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
622
      if (terrno != 0) code = terrno;
×
623
      TAOS_RETURN(code);
×
624
    }
625
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
626
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
627
  }
628

629
  if (pDb->cfg.withArbitrator) {
×
630
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
631
      SVgObj   *pVgObj = pVgroups + v;
×
632
      SArbGroup arbGroup = {0};
×
633
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
634
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
635
    }
636
  }
637

638
  TAOS_RETURN(code);
×
639
}
640

641
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,629✔
642
  int32_t  code = 0;
4,629✔
643
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,629✔
644
  if (pDbRaw == NULL) {
4,629!
645
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
646
    if (terrno != 0) code = terrno;
×
647
    TAOS_RETURN(code);
×
648
  }
649
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
4,629!
650
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
4,629!
651

652
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,720✔
653
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
10,091✔
654
    if (pVgRaw == NULL) {
10,091!
655
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
656
      if (terrno != 0) code = terrno;
×
657
      TAOS_RETURN(code);
×
658
    }
659
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
10,091!
660
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
10,091!
661
  }
662

663
  if (pDb->cfg.withArbitrator) {
4,629✔
664
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
9✔
665
      SVgObj   *pVgObj = pVgroups + v;
6✔
666
      SArbGroup arbGroup = {0};
6✔
667
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6!
668
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
6!
669
    }
670
  }
671

672
  TAOS_RETURN(code);
4,629✔
673
}
674

675
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
4,629✔
676
                                        SUserObj *pUserDuped) {
677
  int32_t  code = 0;
4,629✔
678
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,629✔
679
  if (pDbRaw == NULL) {
4,629!
680
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
681
    if (terrno != 0) code = terrno;
×
682
    TAOS_RETURN(code);
×
683
  }
684
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
4,629!
685
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
4,629!
686

687
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,720✔
688
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
10,091✔
689
    if (pVgRaw == NULL) {
10,091!
690
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
691
      if (terrno != 0) code = terrno;
×
692
      TAOS_RETURN(code);
×
693
    }
694
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
10,091!
695
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
10,091!
696
  }
697

698
  if (pDb->cfg.withArbitrator) {
4,629✔
699
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
9✔
700
      SVgObj   *pVgObj = pVgroups + v;
6✔
701
      SArbGroup arbGroup = {0};
6✔
702
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6!
703
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
6!
704
    }
705
  }
706

707
  if (pUserDuped) {
4,629✔
708
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
7✔
709
    if (pUserRaw == NULL) {
7!
710
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
711
      if (terrno != 0) code = terrno;
×
712
      TAOS_RETURN(code);
×
713
    }
714
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
7!
715
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
7!
716
  }
717

718
  TAOS_RETURN(code);
4,629✔
719
}
720

721
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,629✔
722
  int32_t code = 0;
4,629✔
723
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
14,720✔
724
    SVgObj *pVgroup = pVgroups + vg;
10,091✔
725

726
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
20,996✔
727
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
10,905✔
728
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
10,905!
729
    }
730
  }
731

732
  TAOS_RETURN(code);
4,629✔
733
}
734

735
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,629✔
736
  int32_t code = 0;
4,629✔
737
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
14,720✔
738
    SVgObj *pVgroup = pVgroups + vg;
10,091✔
739

740
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
20,996✔
741
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
10,905✔
742
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
10,905!
743
    }
744
  }
745

746
  TAOS_RETURN(code);
4,629✔
747
}
748

749
static int32_t mndCreateDb(SMnode *pMnode, SRpcMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser, SArray *dnodeList) {
5,092✔
750
  int32_t  code = 0;
5,092✔
751
  SUserObj newUserObj = {0};
5,092✔
752
  SDbObj   dbObj = {0};
5,092✔
753
  (void)memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
5,092✔
754
  (void)memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
5,092✔
755
  dbObj.createdTime = taosGetTimestampMs();
5,092✔
756
  dbObj.updateTime = dbObj.createdTime;
5,092✔
757
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
5,092✔
758
  dbObj.cfgVersion = 1;
5,092✔
759
  dbObj.vgVersion = 1;
5,092✔
760
  dbObj.tsmaVersion = 1;
5,092✔
761
  (void)memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
5,092✔
762
  dbObj.cfg = (SDbCfg){
5,092✔
763
      .numOfVgroups = pCreate->numOfVgroups,
5,092✔
764
      .numOfStables = pCreate->numOfStables,
5,092✔
765
      .buffer = pCreate->buffer,
5,092✔
766
      .pageSize = pCreate->pageSize,
5,092✔
767
      .pages = pCreate->pages,
5,092✔
768
      .cacheLastSize = pCreate->cacheLastSize,
5,092✔
769
      .daysPerFile = pCreate->daysPerFile,
5,092✔
770
      .daysToKeep0 = pCreate->daysToKeep0,
5,092✔
771
      .daysToKeep1 = pCreate->daysToKeep1,
5,092✔
772
      .daysToKeep2 = pCreate->daysToKeep2,
5,092✔
773
      .keepTimeOffset = pCreate->keepTimeOffset,
5,092✔
774
      .minRows = pCreate->minRows,
5,092✔
775
      .maxRows = pCreate->maxRows,
5,092✔
776
      .walFsyncPeriod = pCreate->walFsyncPeriod,
5,092✔
777
      .walLevel = pCreate->walLevel,
5,092✔
778
      .precision = pCreate->precision,
5,092✔
779
      .compression = pCreate->compression,
5,092✔
780
      .replications = pCreate->replications,
5,092✔
781
      .strict = pCreate->strict,
5,092✔
782
      .cacheLast = pCreate->cacheLast,
5,092✔
783
      .hashMethod = 1,
784
      .schemaless = pCreate->schemaless,
5,092✔
785
      .walRetentionPeriod = pCreate->walRetentionPeriod,
5,092✔
786
      .walRetentionSize = pCreate->walRetentionSize,
5,092✔
787
      .walRollPeriod = pCreate->walRollPeriod,
5,092✔
788
      .walSegmentSize = pCreate->walSegmentSize,
5,092✔
789
      .sstTrigger = pCreate->sstTrigger,
5,092✔
790
      .hashPrefix = pCreate->hashPrefix,
5,092✔
791
      .hashSuffix = pCreate->hashSuffix,
5,092✔
792
      .s3ChunkSize = pCreate->s3ChunkSize,
5,092✔
793
      .s3KeepLocal = pCreate->s3KeepLocal,
5,092✔
794
      .s3Compact = pCreate->s3Compact,
5,092✔
795
      .tsdbPageSize = pCreate->tsdbPageSize,
5,092✔
796
      .withArbitrator = pCreate->withArbitrator,
5,092✔
797
      .encryptAlgorithm = pCreate->encryptAlgorithm,
5,092✔
798
  };
799

800
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
5,092✔
801
  dbObj.cfg.pRetensions = pCreate->pRetensions;
5,092✔
802

803
  mndSetDefaultDbCfg(&dbObj.cfg);
5,092✔
804

805
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
5,092!
806
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, terrstr());
×
807
    TAOS_RETURN(code);
×
808
  }
809

810
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
5,092✔
811
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, terrstr());
1!
812
    TAOS_RETURN(code);
1✔
813
  }
814

815
  if (dbObj.cfg.hashPrefix > 0) {
5,091✔
816
    int32_t dbLen = strlen(dbObj.name) + 1;
10✔
817
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
10!
818
    dbObj.cfg.hashPrefix += dbLen;
10✔
819
  } else if (dbObj.cfg.hashPrefix < 0) {
5,081✔
820
    int32_t dbLen = strlen(dbObj.name) + 1;
11✔
821
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
11!
822
    dbObj.cfg.hashPrefix -= dbLen;
11✔
823
  }
824

825
  SVgObj *pVgroups = NULL;
5,091✔
826
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
5,091✔
827
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
446!
828
    TAOS_RETURN(code);
446✔
829
  }
830

831
  // add database privileges for user
832
  SUserObj *pNewUserDuped = NULL;
4,645✔
833
  if (!pUser->superUser) {
4,645✔
834
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
7!
835
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
7!
836
                    NULL, _OVER);
837
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
7!
838
                    NULL, _OVER);
839
    pNewUserDuped = &newUserObj;
7✔
840
  }
841

842
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
4,645✔
843
  if (pTrans == NULL) {
4,645!
844
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
845
    if (terrno != 0) code = terrno;
×
846
    goto _OVER;
×
847
  }
848
  // mndTransSetSerial(pTrans);
849
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
4,645!
850

851
  mndTransSetDbName(pTrans, dbObj.name, NULL);
4,645✔
852
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
4,645✔
853

854
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
4,629✔
855
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
4,629!
856
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,629!
857
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,629!
858
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,629!
859
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
4,629!
860
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,629!
861
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
4,629✔
862

863
_OVER:
4,625✔
864
  taosMemoryFree(pVgroups);
4,645✔
865
  mndUserFreeObj(&newUserObj);
4,645✔
866
  mndTransDrop(pTrans);
4,645✔
867
  TAOS_RETURN(code);
4,645✔
868
}
869

870
static void mndBuildAuditDetailInt32(char *detail, char *tmp, char *format, int32_t para) {
×
871
  if (para > 0) {
×
872
    if (strlen(detail) > 0) (void)strcat(detail, ", ");
×
873
    (void)sprintf(tmp, format, para);
×
874
    (void)strcat(detail, tmp);
×
875
  }
876
}
×
877

878
static void mndBuildAuditDetailInt64(char *detail, char *tmp, char *format, int64_t para) {
×
879
  if (para > 0) {
×
880
    if (strlen(detail) > 0) (void)strcat(detail, ", ");
×
881
    (void)sprintf(tmp, format, para);
×
882
    (void)strcat(detail, tmp);
×
883
  }
884
}
×
885

886
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
5,108✔
887
  int32_t    code = 0;
5,108✔
888
  SSdb      *pSdb = pMnode->pSdb;
5,108✔
889
  SDnodeObj *pDnode = NULL;
5,108✔
890
  void      *pIter = NULL;
5,108✔
891

892
#ifdef TD_ENTERPRISE
893
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
5,108✔
894
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
1!
895
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
1!
896
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
897
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
898
           tsEncryptionKeyStat);
899
    goto _exit;
×
900
  }
901

902
  int64_t curMs = taosGetTimestampMs();
1✔
903
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
2✔
904
    bool online = false;
1✔
905
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
1!
906
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
907
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
908
      mError("db:%s, failed to check encryption key:%" PRIi8
×
909
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
910
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
911
             tsEncryptionKeyChksum);
912
      sdbCancelFetch(pSdb, pIter);
×
913
      sdbRelease(pSdb, pDnode);
×
914
      break;
×
915
    }
916
    sdbRelease(pSdb, pDnode);
1✔
917
  }
918
#else
919
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
920
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
921
    goto _exit;
922
  }
923
#endif
924
_exit:
1✔
925
  TAOS_RETURN(code);
5,108✔
926
}
927

928
#ifndef TD_ENTERPRISE
929
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
930
  if (dnodeListStr[0] != 0) {
931
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
932
    return terrno;
933
  } else {
934
    return 0;
935
  }
936
}
937
#endif
938

939
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
5,516✔
940
  SMnode      *pMnode = pReq->info.node;
5,516✔
941
  int32_t      code = -1;
5,516✔
942
  int32_t      lino = 0;
5,516✔
943
  SDbObj      *pDb = NULL;
5,516✔
944
  SUserObj    *pUser = NULL;
5,516✔
945
  SCreateDbReq createReq = {0};
5,516✔
946
  SArray      *dnodeList = NULL;
5,516✔
947

948
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
5,516✔
949
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
5,516!
950

951
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
5,516!
952
#ifdef WINDOWS
953
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
954
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
955
    goto _OVER;
956
  }
957
#endif
958
  mInfo("db:%s, start to create, vgroups:%d", createReq.db, createReq.numOfVgroups);
5,516!
959

960
  pDb = mndAcquireDb(pMnode, createReq.db);
5,516✔
961
  if (pDb != NULL) {
5,516✔
962
    if (createReq.ignoreExist) {
353✔
963
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
341!
964
      code = 0;
341✔
965
      goto _OVER;
341✔
966
    } else {
967
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
12✔
968
      goto _OVER;
12✔
969
    }
970
  } else {
971
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
5,163✔
972
      code = terrno;
52✔
973
      goto _OVER;
52✔
974
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
5,111!
975
      goto _OVER;
×
976
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
5,111!
977
      // continue
978
    } else {  // TSDB_CODE_APP_ERROR
979
      goto _OVER;
×
980
    }
981
  }
982

983
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_DB, NULL), &lino, _OVER);
5,111✔
984

985
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
5,108!
986

987
  if (createReq.replications == 2) {
5,108✔
988
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
5!
989
  }
990

991
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
5,108!
992

993
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
5,108✔
994

995
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pUser), &lino, _OVER);
5,092!
996

997
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
5,092✔
998
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
4,625!
999

1000
  SName name = {0};
4,625✔
1001
  if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
4,625!
1002
    mError("db:%s, failed to parse db name", createReq.db);
×
1003

1004
  auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen);
4,625✔
1005

1006
_OVER:
5,516✔
1007
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
5,516✔
1008
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
550!
1009
  }
1010

1011
  mndReleaseDb(pMnode, pDb);
5,516✔
1012
  mndReleaseUser(pMnode, pUser);
5,516✔
1013
  tFreeSCreateDbReq(&createReq);
5,516✔
1014
  taosArrayDestroy(dnodeList);
5,516✔
1015

1016
  TAOS_RETURN(code);
5,516✔
1017
}
1018

1019
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
302✔
1020
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
302✔
1021

1022
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
302✔
1023
    pDb->cfg.buffer = pAlter->buffer;
5✔
1024
    code = 0;
5✔
1025
  }
1026

1027
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
302✔
1028
    pDb->cfg.pages = pAlter->pages;
3✔
1029
    code = 0;
3✔
1030
  }
1031

1032
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
302!
1033
    pDb->cfg.pageSize = pAlter->pageSize;
×
1034
    code = 0;
×
1035
  }
1036

1037
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
302!
1038
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
1039
    code = 0;
×
1040
  }
1041

1042
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
302✔
1043
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
16✔
1044
    code = 0;
16✔
1045
  }
1046

1047
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
302✔
1048
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
16✔
1049
    code = 0;
16✔
1050
  }
1051

1052
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
302✔
1053
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
16✔
1054
    code = 0;
16✔
1055
  }
1056

1057
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
302!
1058
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
1✔
1059
    code = 0;
1✔
1060
  }
1061

1062
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
302✔
1063
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
11✔
1064
    code = 0;
11✔
1065
  }
1066

1067
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
302✔
1068
    pDb->cfg.walLevel = pAlter->walLevel;
11✔
1069
    code = 0;
11✔
1070
  }
1071

1072
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
302!
1073
#if 1
1074
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1075
#else
1076
    pDb->cfg.strict = pAlter->strict;
1077
    code = 0;
1078
#endif
1079
  }
1080

1081
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
302✔
1082
    pDb->cfg.cacheLast = pAlter->cacheLast;
189✔
1083
    code = 0;
189✔
1084
  }
1085

1086
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
302✔
1087
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
143✔
1088
    code = 0;
143✔
1089
  }
1090

1091
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
302✔
1092
    pDb->cfg.replications = pAlter->replications;
8✔
1093
    pDb->vgVersion++;
8✔
1094
    code = 0;
8✔
1095
  }
1096

1097
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
302!
1098
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
×
1099
    pDb->vgVersion++;
×
1100
    code = 0;
×
1101
  }
1102

1103
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
302!
1104
    pDb->cfg.minRows = pAlter->minRows;
×
1105
    pDb->vgVersion++;
×
1106
    code = 0;
×
1107
  }
1108

1109
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
302✔
1110
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
45✔
1111
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
20✔
1112
    pDb->vgVersion++;
20✔
1113
    code = 0;
20✔
1114
  }
1115

1116
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
302✔
1117
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
1!
1118
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
×
1119
    pDb->vgVersion++;
×
1120
    code = 0;
×
1121
  }
1122

1123
  if (pAlter->s3KeepLocal > TSDB_MIN_S3_KEEP_LOCAL && pAlter->s3KeepLocal != pDb->cfg.s3KeepLocal) {
302!
1124
    pDb->cfg.s3KeepLocal = pAlter->s3KeepLocal;
×
1125
    pDb->vgVersion++;
×
1126
    code = 0;
×
1127
  }
1128

1129
  if (pAlter->s3Compact > TSDB_MIN_S3_COMPACT && pAlter->s3Compact != pDb->cfg.s3Compact) {
302!
1130
    pDb->cfg.s3Compact = pAlter->s3Compact;
×
1131
    pDb->vgVersion++;
×
1132
    code = 0;
×
1133
  }
1134

1135
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
302✔
1136
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
1✔
1137
    pDb->vgVersion++;
1✔
1138
    code = 0;
1✔
1139
  }
1140

1141
  TAOS_RETURN(code);
302✔
1142
}
1143

1144
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
262✔
1145
  int32_t  code = 0;
262✔
1146
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
262✔
1147
  if (pRedoRaw == NULL) {
262!
1148
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1149
    if (terrno != 0) code = terrno;
×
1150
    TAOS_RETURN(code);
×
1151
  }
1152
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
262!
1153
    sdbFreeRaw(pRedoRaw);
×
1154
    TAOS_RETURN(code);
×
1155
  }
1156

1157
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
262!
1158
    sdbFreeRaw(pRedoRaw);
×
1159
    TAOS_RETURN(code);
×
1160
  }
1161
  return 0;
262✔
1162
}
1163

1164
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
262✔
1165
  int32_t  code = 0;
262✔
1166
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
262✔
1167
  if (pCommitRaw == NULL) {
262!
1168
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1169
    if (terrno != 0) code = terrno;
×
1170
    TAOS_RETURN(code);
×
1171
  }
1172
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
262!
1173
    sdbFreeRaw(pCommitRaw);
×
1174
    TAOS_RETURN(code);
×
1175
  }
1176

1177
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
262!
1178
    sdbFreeRaw(pCommitRaw);
×
1179
    TAOS_RETURN(code);
×
1180
  }
1181
  TAOS_RETURN(code);
262✔
1182
}
1183

1184
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
262✔
1185
  int32_t code = 0, lino = 0;
262✔
1186
  SSdb   *pSdb = pMnode->pSdb;
262✔
1187
  void   *pIter = NULL;
262✔
1188
  SVgObj *pVgroup = NULL;
262✔
1189
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
262✔
1190

1191
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
262!
1192

1193
  while (1) {
1194
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
6,934✔
1195
    if (pIter == NULL) break;
6,934✔
1196

1197
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
6,672✔
1198
      SVgObj newVgroup = {0};
718✔
1199
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
718!
1200
                      _err);
1201

1202
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
718!
1203
        if (pNewDb->cfg.withArbitrator) {
×
1204
          SArbGroup arbGroup = {0};
×
1205
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1206
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1207
        } else {
1208
          SArbGroup arbGroup = {0};
×
1209
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
1210
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1211
        }
1212
      }
1213
    }
1214

1215
    sdbRelease(pSdb, pVgroup);
6,672✔
1216
  }
1217

1218
  taosArrayDestroy(pArray);
262✔
1219
  TAOS_RETURN(code);
262✔
1220

1221
_err:
×
1222
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
×
1223

1224
  sdbCancelFetch(pSdb, pIter);
×
1225
  sdbRelease(pSdb, pVgroup);
×
1226
  taosArrayDestroy(pArray);
×
1227
  TAOS_RETURN(code);
×
1228
}
1229

1230
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
263✔
1231
  int32_t code = -1;
263✔
1232
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
263✔
1233
  if (pTrans == NULL) {
263!
1234
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1235
    if (terrno != 0) code = terrno;
×
1236
    TAOS_RETURN(code);
×
1237
  }
1238
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
263!
1239

1240
  mndTransSetDbName(pTrans, pOld->name, NULL);
263✔
1241
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
263!
1242
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
263✔
1243

1244
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
262!
1245
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
262!
1246
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
262!
1247
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
262!
1248
  code = 0;
262✔
1249

1250
_OVER:
263✔
1251
  mndTransDrop(pTrans);
263✔
1252
  TAOS_RETURN(code);
263✔
1253
}
1254

1255
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
303✔
1256
  SMnode     *pMnode = pReq->info.node;
303✔
1257
  int32_t     code = -1;
303✔
1258
  SDbObj     *pDb = NULL;
303✔
1259
  SAlterDbReq alterReq = {0};
303✔
1260
  SDbObj      dbObj = {0};
303✔
1261

1262
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
303!
1263

1264
  mInfo("db:%s, start to alter", alterReq.db);
303!
1265

1266
  pDb = mndAcquireDb(pMnode, alterReq.db);
303✔
1267
  if (pDb == NULL) {
303!
1268
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1269
    if (terrno != 0) code = terrno;
×
UNCOV
1270
    goto _OVER;
×
1271
  }
1272

1273
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_ALTER_DB, pDb), NULL, _OVER);
303✔
1274

1275
  if (alterReq.replications == 2) {
302✔
1276
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
1!
1277
  }
1278

1279
  int32_t numOfTopics = 0;
302✔
1280
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
302!
1281

1282
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
302!
1283
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
UNCOV
1284
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1285
           numOfTopics);
UNCOV
1286
    goto _OVER;
×
1287
  }
1288

1289
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
302✔
1290
  if (dbObj.cfg.pRetensions != NULL) {
302!
1291
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
UNCOV
1292
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1293
  }
1294

1295
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
302✔
1296
  if (code != 0) {
302✔
1297
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
38!
1298
    goto _OVER;
38✔
1299
  }
1300

1301
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
264✔
1302

1303
  dbObj.cfgVersion++;
263✔
1304
  dbObj.updateTime = taosGetTimestampMs();
263✔
1305
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
263✔
1306

1307
  if (dbObj.cfg.replications != pDb->cfg.replications) {
263✔
1308
    // return quickly, operation executed asynchronously
1309
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
7!
1310
  } else {
1311
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
256!
1312
  }
1313

1314
  SName name = {0};
263✔
1315
  if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
263!
UNCOV
1316
    mError("db:%s, failed to parse db name", alterReq.db);
×
1317

1318
  auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen);
263✔
1319

1320
_OVER:
303✔
1321
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
303✔
1322
    if (terrno != 0) code = terrno;
3!
1323
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
3!
1324
  }
1325

1326
  mndReleaseDb(pMnode, pDb);
303✔
1327
  taosArrayDestroy(dbObj.cfg.pRetensions);
303✔
1328
  tFreeSAlterDbReq(&alterReq);
303✔
1329

1330
  TAOS_RETURN(code);
303✔
1331
}
1332

1333
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb) {
79,000✔
1334
  (void)strcpy(cfgRsp->db, pDb->name);
79,000✔
1335
  cfgRsp->dbId = pDb->uid;
79,000✔
1336
  cfgRsp->cfgVersion = pDb->cfgVersion;
79,000✔
1337
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
79,000✔
1338
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
79,000✔
1339
  cfgRsp->buffer = pDb->cfg.buffer;
79,000✔
1340
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
79,000✔
1341
  cfgRsp->pageSize = pDb->cfg.pageSize;
79,000✔
1342
  cfgRsp->pages = pDb->cfg.pages;
79,000✔
1343
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
79,000✔
1344
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
79,000✔
1345
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
79,000✔
1346
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
79,000✔
1347
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
79,000✔
1348
  cfgRsp->minRows = pDb->cfg.minRows;
79,000✔
1349
  cfgRsp->maxRows = pDb->cfg.maxRows;
79,000✔
1350
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
79,000✔
1351
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
79,000✔
1352
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
79,000✔
1353
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
79,000✔
1354
  cfgRsp->walLevel = pDb->cfg.walLevel;
79,000✔
1355
  cfgRsp->precision = pDb->cfg.precision;
79,000✔
1356
  cfgRsp->compression = pDb->cfg.compression;
79,000✔
1357
  cfgRsp->replications = pDb->cfg.replications;
79,000✔
1358
  cfgRsp->strict = pDb->cfg.strict;
79,000✔
1359
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
79,000✔
1360
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
79,000✔
1361
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
79,000✔
1362
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
79,000✔
1363
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
79,000✔
1364
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
79,000✔
1365
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
79,000✔
1366
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
79,000✔
1367
  cfgRsp->schemaless = pDb->cfg.schemaless;
79,000✔
1368
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
79,000✔
1369
  cfgRsp->s3ChunkSize = pDb->cfg.s3ChunkSize;
79,000✔
1370
  cfgRsp->s3KeepLocal = pDb->cfg.s3KeepLocal;
79,000✔
1371
  cfgRsp->s3Compact = pDb->cfg.s3Compact;
79,000✔
1372
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
79,000✔
1373
  cfgRsp->encryptAlgorithm = pDb->cfg.encryptAlgorithm;
79,000✔
1374
}
79,000✔
1375

1376
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
73,946✔
1377
  SMnode   *pMnode = pReq->info.node;
73,946✔
1378
  int32_t   code = -1;
73,946✔
1379
  SDbObj   *pDb = NULL;
73,946✔
1380
  SDbCfgReq cfgReq = {0};
73,946✔
1381
  SDbCfgRsp cfgRsp = {0};
73,946✔
1382

1383
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
73,946!
1384

1385
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
73,946!
1386
    pDb = mndAcquireDb(pMnode, cfgReq.db);
73,946✔
1387
    if (pDb == NULL) {
73,946✔
1388
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
82✔
1389
      if (terrno != 0) code = terrno;
82!
1390
      goto _OVER;
82✔
1391
    }
1392

1393
    mndDumpDbCfgInfo(&cfgRsp, pDb);
73,864✔
1394
  }
1395

1396
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
73,864✔
1397
  void   *pRsp = rpcMallocCont(contLen);
73,863✔
1398
  if (pRsp == NULL) {
73,864!
1399
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1400
    if (terrno != 0) code = terrno;
×
UNCOV
1401
    goto _OVER;
×
1402
  }
1403

1404
  int32_t ret = 0;
73,864✔
1405
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
73,864!
1406
    code = ret;
×
UNCOV
1407
    goto _OVER;
×
1408
  }
1409

1410
  pReq->info.rsp = pRsp;
73,864✔
1411
  pReq->info.rspLen = contLen;
73,864✔
1412

1413
  code = 0;
73,864✔
1414

1415
_OVER:
73,946✔
1416

1417
  tFreeSDbCfgRsp(&cfgRsp);
73,946✔
1418

1419
  if (code != 0) {
73,945✔
1420
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
82!
1421
  }
1422

1423
  mndReleaseDb(pMnode, pDb);
73,945✔
1424

1425
  TAOS_RETURN(code);
73,946✔
1426
}
1427

1428
static int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,345✔
1429
  int32_t  code = 0;
2,345✔
1430
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
2,345✔
1431
  if (pRedoRaw == NULL) {
2,345!
1432
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1433
    if (terrno != 0) code = terrno;
×
UNCOV
1434
    return -1;
×
1435
  }
1436
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
2,345!
1437
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
2,345!
1438

1439
  SSdb *pSdb = pMnode->pSdb;
2,345✔
1440
  void *pIter = NULL;
2,345✔
1441

1442
  while (1) {
6✔
1443
    SArbGroup *pArbGroup = NULL;
2,351✔
1444
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
2,351✔
1445
    if (pIter == NULL) break;
2,351✔
1446

1447
    if (pArbGroup->dbUid == pDb->uid) {
6!
1448
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
6!
1449
        sdbCancelFetch(pSdb, pIter);
×
1450
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
1451
        TAOS_RETURN(code);
×
1452
      }
1453
    }
1454

1455
    sdbRelease(pSdb, pArbGroup);
6✔
1456
  }
1457

1458
  TAOS_RETURN(code);
2,345✔
1459
}
1460

1461
static int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,345✔
1462
  int32_t  code = 0;
2,345✔
1463
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
2,345✔
1464
  if (pCommitRaw == NULL) {
2,345!
1465
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1466
    if (terrno != 0) code = terrno;
×
UNCOV
1467
    return -1;
×
1468
  }
1469
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
2,345!
1470
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
2,345!
1471

1472
  SSdb *pSdb = pMnode->pSdb;
2,345✔
1473
  void *pIter = NULL;
2,345✔
1474

1475
  while (1) {
6✔
1476
    SArbGroup *pArbGroup = NULL;
2,351✔
1477
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
2,351✔
1478
    if (pIter == NULL) break;
2,351✔
1479

1480
    if (pArbGroup->dbUid == pDb->uid) {
6!
1481
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
6!
1482
        sdbCancelFetch(pSdb, pIter);
×
1483
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
1484
        TAOS_RETURN(code);
×
1485
      }
1486
    }
1487

1488
    sdbRelease(pSdb, pArbGroup);
6✔
1489
  }
1490

1491
  while (1) {
15,407✔
1492
    SVgObj *pVgroup = NULL;
17,752✔
1493
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
17,752✔
1494
    if (pIter == NULL) break;
17,752✔
1495

1496
    if (pVgroup->dbUid == pDb->uid) {
15,407✔
1497
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
4,879✔
1498
      if (pVgRaw == NULL) {
4,879!
1499
        sdbCancelFetch(pSdb, pIter);
×
1500
        sdbRelease(pSdb, pVgroup);
×
1501
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1502
        if (terrno != 0) code = terrno;
×
UNCOV
1503
        TAOS_RETURN(code);
×
1504
      }
1505
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
4,879!
1506
        sdbCancelFetch(pSdb, pIter);
×
1507
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1508
        TAOS_RETURN(code);
×
1509
      }
1510
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
4,879!
1511
        sdbCancelFetch(pSdb, pIter);
×
1512
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1513
        TAOS_RETURN(code);
×
1514
      }
1515
    }
1516

1517
    sdbRelease(pSdb, pVgroup);
15,407✔
1518
  }
1519

1520
  while (1) {
21,249✔
1521
    SStbObj *pStb = NULL;
23,594✔
1522
    pIter = sdbFetch(pSdb, SDB_STB, pIter, (void **)&pStb);
23,594✔
1523
    if (pIter == NULL) break;
23,594✔
1524

1525
    if (pStb->dbUid == pDb->uid) {
21,249✔
1526
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
4,333✔
1527
      if (pStbRaw == NULL) {
4,333!
1528
        sdbCancelFetch(pSdb, pIter);
×
1529
        sdbRelease(pSdb, pStbRaw);
×
1530
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1531
        if (terrno != 0) code = terrno;
×
UNCOV
1532
        return -1;
×
1533
      }
1534
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
4,333!
1535
        sdbCancelFetch(pSdb, pIter);
×
1536
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1537
        return -1;
×
1538
      }
1539
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
4,333!
1540
        sdbCancelFetch(pSdb, pIter);
×
1541
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1542
        return -1;
×
1543
      }
1544
    }
1545

1546
    sdbRelease(pSdb, pStb);
21,249✔
1547
  }
1548

1549
  TAOS_RETURN(code);
2,345✔
1550
}
1551

1552
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
4,876✔
1553
  int32_t code = 0;
4,876✔
1554
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
9,872✔
1555
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
4,996✔
1556
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
4,996!
1557
  }
1558

1559
  TAOS_RETURN(code);
4,876✔
1560
}
1561

1562
static int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,344✔
1563
  int32_t code = 0;
2,344✔
1564
  SSdb *pSdb = pMnode->pSdb;
2,344✔
1565
  void *pIter = NULL;
2,344✔
1566

1567
  while (1) {
15,398✔
1568
    SVgObj *pVgroup = NULL;
17,742✔
1569
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
17,742✔
1570
    if (pIter == NULL) break;
17,742✔
1571

1572
    if (pVgroup->dbUid == pDb->uid) {
15,398✔
1573
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
4,876!
1574
        sdbCancelFetch(pSdb, pIter);
×
1575
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1576
        TAOS_RETURN(code);
×
1577
      }
1578
    }
1579

1580
    sdbRelease(pSdb, pVgroup);
15,398✔
1581
  }
1582

1583
  TAOS_RETURN(code);
2,344✔
1584
}
1585

1586
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
4,420✔
1587
  int32_t    code = 0;
4,420✔
1588
  SDropDbRsp dropRsp = {0};
4,420✔
1589
  if (pDb != NULL) {
4,420✔
1590
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
2,344✔
1591
    dropRsp.uid = pDb->uid;
2,344✔
1592
  }
1593

1594
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
4,420✔
1595
  void   *pRsp = NULL;
4,420✔
1596
  if (useRpcMalloc) {
4,420✔
1597
    pRsp = rpcMallocCont(rspLen);
2,076✔
1598
  } else {
1599
    pRsp = taosMemoryMalloc(rspLen);
2,344✔
1600
  }
1601

1602
  if (pRsp == NULL) {
4,420!
1603
    code = terrno;
×
UNCOV
1604
    TAOS_RETURN(code);
×
1605
  }
1606

1607
  int32_t ret = 0;
4,420✔
1608
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
4,420!
1609
  *pRspLen = rspLen;
4,420✔
1610
  *ppRsp = pRsp;
4,420✔
1611
  TAOS_RETURN(code);
4,420✔
1612
}
1613

1614
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
2,346✔
1615
  int32_t code = -1;
2,346✔
1616

1617
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
2,346✔
1618
  if (pTrans == NULL) {
2,346!
1619
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1620
    if (terrno != 0) code = terrno;
×
UNCOV
1621
    goto _OVER;
×
1622
  }
1623

1624
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
2,346!
1625

1626
  mndTransSetDbName(pTrans, pDb->name, NULL);
2,346✔
1627
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
2,346!
1628

1629
  if (mndTopicExistsForDb(pMnode, pDb)) {
2,346✔
1630
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
1✔
1631
    goto _OVER;
1✔
1632
  }
1633

1634
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
2,345!
1635
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
2,345!
1636
  /*if (mndDropOffsetByDB(pMnode, pTrans, pDb) != 0) goto _OVER;*/
1637
  /*if (mndDropSubByDB(pMnode, pTrans, pDb) != 0) goto _OVER;*/
1638
  /*if (mndDropTopicByDB(pMnode, pTrans, pDb) != 0) goto _OVER;*/
1639
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,345✔
1640
#ifdef TD_ENTERPRISE
1641
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,344!
1642
#endif
1643
  TAOS_CHECK_GOTO(mndDropSmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,344!
1644
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
2,344!
1645
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
2,344!
1646
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb->name), NULL, _OVER);
2,344!
1647

1648
  int32_t rspLen = 0;
2,344✔
1649
  void   *pRsp = NULL;
2,344✔
1650
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
2,344!
1651
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
2,344✔
1652

1653
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
2,344!
1654
  code = 0;
2,344✔
1655

1656
_OVER:
2,346✔
1657
  mndTransDrop(pTrans);
2,346✔
1658
  TAOS_RETURN(code);
2,346✔
1659
}
1660

1661
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
4,505✔
1662
  SMnode    *pMnode = pReq->info.node;
4,505✔
1663
  int32_t    code = -1;
4,505✔
1664
  SDbObj    *pDb = NULL;
4,505✔
1665
  SDropDbReq dropReq = {0};
4,505✔
1666

1667
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
4,505!
1668

1669
  mInfo("db:%s, start to drop", dropReq.db);
4,505!
1670

1671
  pDb = mndAcquireDb(pMnode, dropReq.db);
4,505✔
1672
  if (pDb == NULL) {
4,505✔
1673
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
2,144✔
1674
    if (terrno != 0) code = terrno;
2,144!
1675
    if (dropReq.ignoreNotExists) {
2,144✔
1676
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
2,076✔
1677
    }
1678
    goto _OVER;
2,144✔
1679
  }
1680

1681
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_DB, pDb), NULL, _OVER);
2,361✔
1682

1683
  code = mndDropDb(pMnode, pReq, pDb);
2,346✔
1684
  if (code == TSDB_CODE_SUCCESS) {
2,346✔
1685
    code = TSDB_CODE_ACTION_IN_PROGRESS;
2,344✔
1686
  }
1687

1688
  SName name = {0};
2,346✔
1689
  if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
2,346!
UNCOV
1690
    mError("db:%s, failed to parse db name", dropReq.db);
×
1691

1692
  auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen);
2,346✔
1693

1694
_OVER:
4,505✔
1695
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
4,505✔
1696
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
85!
1697
  }
1698

1699
  mndReleaseDb(pMnode, pDb);
4,505✔
1700
  tFreeSDropDbReq(&dropReq);
4,505✔
1701
  TAOS_RETURN(code);
4,505✔
1702
}
1703

1704
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
2,556,422✔
1705
  int32_t numOfTables = 0;
2,556,422✔
1706
  int32_t vindex = 0;
2,556,422✔
1707
  SSdb   *pSdb = pMnode->pSdb;
2,556,422✔
1708

1709
  void *pIter = NULL;
2,556,422✔
1710
  while (vindex < pDb->cfg.numOfVgroups) {
70,138,295✔
1711
    SVgObj *pVgroup = NULL;
67,580,919✔
1712
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
67,580,919✔
1713
    if (pIter == NULL) break;
67,580,525!
1714

1715
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
67,580,525✔
1716
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
5,134,738✔
1717
      vindex++;
5,134,738✔
1718
    }
1719

1720
    sdbRelease(pSdb, pVgroup);
67,580,238✔
1721
  }
1722

1723
  sdbCancelFetch(pSdb, pIter);
2,557,376✔
1724
  return numOfTables;
2,556,465✔
1725
}
1726

1727
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
2,716,644✔
1728
  int32_t vindex = 0;
2,716,644✔
1729
  SSdb   *pSdb = pMnode->pSdb;
2,716,644✔
1730

1731
  void *pIter = NULL;
2,716,644✔
1732
  while (1) {
74,580,389✔
1733
    SVgObj *pVgroup = NULL;
77,297,033✔
1734
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
77,297,033✔
1735
    if (pIter == NULL) break;
77,295,786✔
1736

1737
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
77,021,667!
1738
      SVgroupInfo vgInfo = {0};
17,217,623✔
1739
      vgInfo.vgId = pVgroup->vgId;
17,217,623✔
1740
      vgInfo.hashBegin = pVgroup->hashBegin;
17,217,623✔
1741
      vgInfo.hashEnd = pVgroup->hashEnd;
17,217,623✔
1742
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
17,217,623✔
1743
      vgInfo.epSet.numOfEps = pVgroup->replica;
17,217,623✔
1744
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
34,447,765✔
1745
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
17,230,087✔
1746
        SEp       *pEp = &vgInfo.epSet.eps[gid];
17,230,087✔
1747
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
17,230,087✔
1748
        if (pDnode != NULL) {
17,230,008✔
1749
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
17,229,894✔
1750
          pEp->port = pDnode->port;
17,229,894✔
1751
        }
1752
        mndReleaseDnode(pMnode, pDnode);
17,230,008✔
1753
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
17,230,142!
1754
          vgInfo.epSet.inUse = gid;
17,215,740✔
1755
        }
1756
      }
1757
      vindex++;
17,217,678✔
1758
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
17,217,519!
UNCOV
1759
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
1760
      }
1761
    }
1762

1763
    sdbRelease(pSdb, pVgroup);
77,021,567✔
1764

1765
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
77,022,925✔
1766
      sdbCancelFetch(pSdb, pIter);
2,442,536✔
1767
      break;
2,442,536✔
1768
    }
1769
  }
1770
}
2,716,655✔
1771

1772
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
2,522,449✔
1773
  int32_t code = 0;
2,522,449✔
1774
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
2,522,449✔
1775
  if (pRsp->pVgroupInfos == NULL) {
2,522,408!
1776
    code = terrno;
×
UNCOV
1777
    TAOS_RETURN(code);
×
1778
  }
1779

1780
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
2,522,408✔
1781

1782
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
2,522,452!
1783
      pReq->stateTs < pDb->stateTs) {
82,894✔
1784
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
2,440,006✔
1785
  }
1786

1787
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
2,522,452✔
1788
  pRsp->uid = pDb->uid;
2,522,452✔
1789
  pRsp->vgVersion = pDb->vgVersion;
2,522,452✔
1790
  pRsp->stateTs = pDb->stateTs;
2,522,452✔
1791
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
2,522,452✔
1792
  pRsp->hashMethod = pDb->cfg.hashMethod;
2,522,440✔
1793
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
2,522,440✔
1794
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
2,522,440✔
1795
  TAOS_RETURN(code);
2,522,440✔
1796
}
1797

1798
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
2,789,822✔
1799
  SMnode   *pMnode = pReq->info.node;
2,789,822✔
1800
  int32_t   code = -1;
2,789,822✔
1801
  SDbObj   *pDb = NULL;
2,789,822✔
1802
  SUseDbReq usedbReq = {0};
2,789,822✔
1803
  SUseDbRsp usedbRsp = {0};
2,789,822✔
1804

1805
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
2,789,822!
1806

1807
  char *p = strchr(usedbReq.db, '.');
2,789,883✔
1808
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
2,789,883✔
1809
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
272,934✔
1810
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
272,934✔
1811
    if (usedbReq.vgVersion < vgVersion) {
272,936✔
1812
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
272,911✔
1813
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
272,911!
1814

1815
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
272,911✔
1816
      usedbRsp.vgVersion = vgVersion++;
272,911✔
1817
    } else {
1818
      usedbRsp.vgVersion = usedbReq.vgVersion;
25✔
1819
    }
1820
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
272,936✔
1821
    code = 0;
272,936✔
1822
  } else {
1823
    pDb = mndAcquireDb(pMnode, usedbReq.db);
2,516,949✔
1824
    if (pDb == NULL) {
2,516,936✔
1825
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
432✔
1826
      usedbRsp.uid = usedbReq.dbId;
432✔
1827
      usedbRsp.vgVersion = usedbReq.vgVersion;
432✔
1828
      usedbRsp.errCode = terrno;
432✔
1829

1830
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
432✔
1831
      if (terrno != 0) code = terrno;
432!
1832
      goto _OVER;
432✔
1833
    } else {
1834
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_USE_DB, pDb), NULL, _OVER);
2,516,504✔
1835

1836
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
2,516,514!
1837

1838
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
2,516,503✔
1839
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
1840
      code = 0;
2,516,503✔
1841
    }
1842
  }
1843

1844
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
2,789,439✔
1845
  void   *pRsp = rpcMallocCont(contLen);
2,789,431✔
1846
  if (pRsp == NULL) {
2,789,396!
1847
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1848
    code = -1;
×
UNCOV
1849
    goto _OVER;
×
1850
  }
1851

1852
  int32_t ret = 0;
2,789,396✔
1853
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
2,789,396!
1854
    code = ret;
×
UNCOV
1855
    goto _OVER;
×
1856
  }
1857

1858
  pReq->info.rsp = pRsp;
2,789,430✔
1859
  pReq->info.rspLen = contLen;
2,789,430✔
1860

1861
_OVER:
2,789,878✔
1862
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
2,789,878!
1863
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
448!
1864
  }
1865

1866
  mndReleaseDb(pMnode, pDb);
2,789,878✔
1867
  tFreeSUsedbRsp(&usedbRsp);
2,789,893✔
1868

1869
  TAOS_RETURN(code);
2,789,886✔
1870
}
1871

1872
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
41,164✔
1873
  int32_t       code = 0;
41,164✔
1874
  SDbHbBatchRsp batchRsp = {0};
41,164✔
1875
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
41,164✔
1876
  if (batchRsp.pArray == NULL) {
41,164!
1877
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
1878
    TAOS_RETURN(code);
×
1879
  }
1880

1881
  for (int32_t i = 0; i < numOfDbs; ++i) {
85,914✔
1882
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
44,750✔
1883
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
44,750✔
1884
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
44,750✔
1885
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
44,750✔
1886
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
44,750✔
1887
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
44,750✔
1888
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
44,750✔
1889

1890
    SDbHbRsp rsp = {0};
44,750✔
1891
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
44,750✔
1892
    rsp.dbId = pDbCacheInfo->dbId;
44,750✔
1893

1894
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
44,750✔
1895
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
40,938✔
1896
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
10,516✔
1897
      if (pDbCacheInfo->vgVersion >= vgVersion) {
10,516✔
1898
        continue;
33,798✔
1899
      }
1900

1901
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
1,208✔
1902
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
1,208✔
1903
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
1,208✔
1904

1905
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
1,208✔
1906
      rsp.useDbRsp->vgVersion = vgVersion++;
1,208✔
1907

1908
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
1,208✔
1909

1910
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
2,416!
1911
        if (terrno != 0) code = terrno;
×
UNCOV
1912
        return code;
×
1913
      }
1914

1915
      continue;
1,208✔
1916
    }
1917

1918
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
34,234✔
1919
    if (pDb == NULL) {
34,234✔
1920
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
220✔
1921
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
220✔
1922
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
220✔
1923
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
220✔
1924
      rsp.useDbRsp->vgVersion = -1;
220✔
1925
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
440!
1926
        if (terrno != 0) code = terrno;
×
UNCOV
1927
        return code;
×
1928
      }
1929
      continue;
220✔
1930
    }
1931

1932
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
34,014✔
1933

1934
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
34,014✔
1935
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
28,811!
1936
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
26,573✔
1937
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
23,060✔
1938
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
1939
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
1940
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
1941
      mndReleaseDb(pMnode, pDb);
23,060✔
1942
      continue;
23,060✔
1943
    } else {
1944
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
10,954✔
1945
            " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
1946
            pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
1947
            pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
1948
    }
1949

1950
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
10,954✔
1951
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
5,136✔
1952
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb);
5,136✔
1953
    }
1954

1955
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
10,954✔
1956
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
9,921✔
1957
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
9,921!
1958
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
9,921!
1959
        bool exist = false;
9,921✔
1960
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
9,921✔
1961
        if (TSDB_CODE_SUCCESS != code) {
9,921✔
1962
          mndReleaseDb(pMnode, pDb);
2✔
1963
          if (code != TSDB_CODE_NEED_RETRY) {
2!
UNCOV
1964
            mError("db:%s, failed to get db tsmas", pDb->name);
×
1965
          } else {
1966
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
2!
1967
          }
1968
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
2✔
1969
          taosMemoryFreeClear(rsp.pTsmaRsp);
2!
1970
          continue;
2✔
1971
        }
1972
        rsp.dbTsmaVersion = pDb->tsmaVersion;
9,919✔
1973
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
9,919✔
1974
      }
1975
    }
1976

1977
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
10,952!
1978
        pDbCacheInfo->stateTs != pDb->stateTs) {
10,882✔
1979
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
2,530✔
1980
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
2,530✔
1981
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
2,530!
1982
        mndReleaseDb(pMnode, pDb);
×
1983
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
1984
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
1985
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
UNCOV
1986
        continue;
×
1987
      }
1988

1989
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
2,530✔
1990
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
2,530✔
1991
      rsp.useDbRsp->uid = pDb->uid;
2,530✔
1992
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
2,530✔
1993
      rsp.useDbRsp->stateTs = pDb->stateTs;
2,530✔
1994
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
2,530✔
1995
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
2,530✔
1996
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
2,530✔
1997
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
2,530✔
1998
    }
1999

2000
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
21,904!
2001
      mndReleaseDb(pMnode, pDb);
×
2002
      if (terrno != 0) code = terrno;
×
UNCOV
2003
      return code;
×
2004
    }
2005
    mndReleaseDb(pMnode, pDb);
10,952✔
2006
  }
2007

2008
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
41,164✔
2009
  void   *pRsp = taosMemoryMalloc(rspLen);
41,164✔
2010
  if (pRsp == NULL) {
41,164!
2011
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2012
    tFreeSDbHbBatchRsp(&batchRsp);
×
UNCOV
2013
    return -1;
×
2014
  }
2015
  int32_t ret = 0;
41,164✔
2016
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
41,164!
2017

2018
  *ppRsp = pRsp;
41,164✔
2019
  *pRspLen = rspLen;
41,164✔
2020

2021
  tFreeSDbHbBatchRsp(&batchRsp);
41,164✔
2022
  TAOS_RETURN(code);
41,164✔
2023
}
2024

2025
static int32_t mndTrimDb(SMnode *pMnode, SDbObj *pDb) {
3✔
2026
  SSdb       *pSdb = pMnode->pSdb;
3✔
2027
  SVgObj     *pVgroup = NULL;
3✔
2028
  void       *pIter = NULL;
3✔
2029
  int32_t     code = 0;
3✔
2030
  SVTrimDbReq trimReq = {.timestamp = taosGetTimestampSec()};
3✔
2031
  int32_t     reqLen = tSerializeSVTrimDbReq(NULL, 0, &trimReq);
3✔
2032
  int32_t     contLen = reqLen + sizeof(SMsgHead);
3✔
2033

2034
  while (1) {
5✔
2035
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
8✔
2036
    if (pIter == NULL) break;
8✔
2037

2038
    SMsgHead *pHead = rpcMallocCont(contLen);
5✔
2039
    if (pHead == NULL) {
5!
2040
      sdbCancelFetch(pSdb, pVgroup);
×
2041
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2042
      continue;
×
2043
    }
2044
    pHead->contLen = htonl(contLen);
5✔
2045
    pHead->vgId = htonl(pVgroup->vgId);
5✔
2046
    int32_t ret = 0;
5✔
2047
    if ((ret = tSerializeSVTrimDbReq((char *)pHead + sizeof(SMsgHead), contLen, &trimReq)) < 0) {
5!
2048
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2049
      return ret;
×
2050
    }
2051

2052
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM, .pCont = pHead, .contLen = contLen};
5✔
2053
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
5✔
2054
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
5✔
2055
    if (code != 0) {
5!
UNCOV
2056
      mError("vgId:%d, failed to send vnode-trim request to vnode since 0x%x", pVgroup->vgId, code);
×
2057
    } else {
2058
      mInfo("vgId:%d, send vnode-trim request to vnode, time:%d", pVgroup->vgId, trimReq.timestamp);
5!
2059
    }
2060
    sdbRelease(pSdb, pVgroup);
5✔
2061
  }
2062

2063
  return 0;
3✔
2064
}
2065

2066
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
3✔
2067
  SMnode    *pMnode = pReq->info.node;
3✔
2068
  int32_t    code = -1;
3✔
2069
  SDbObj    *pDb = NULL;
3✔
2070
  STrimDbReq trimReq = {0};
3✔
2071

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

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

2076
  pDb = mndAcquireDb(pMnode, trimReq.db);
3✔
2077
  if (pDb == NULL) {
3!
2078
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2079
    if (terrno != 0) code = terrno;
×
UNCOV
2080
    goto _OVER;
×
2081
  }
2082

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

2085
  code = mndTrimDb(pMnode, pDb);
3✔
2086

2087
_OVER:
3✔
2088
  if (code != 0) {
3!
UNCOV
2089
    mError("db:%s, failed to process trim db req since %s", trimReq.db, terrstr());
×
2090
  }
2091

2092
  mndReleaseDb(pMnode, pDb);
3✔
2093
  TAOS_RETURN(code);
3✔
2094
}
2095

2096
static int32_t mndS3MigrateDb(SMnode *pMnode, SDbObj *pDb) {
×
2097
  SSdb            *pSdb = pMnode->pSdb;
×
2098
  SVgObj          *pVgroup = NULL;
×
2099
  void            *pIter = NULL;
×
2100
  SVS3MigrateDbReq s3migrateReq = {.timestamp = taosGetTimestampSec()};
×
2101
  int32_t          reqLen = tSerializeSVS3MigrateDbReq(NULL, 0, &s3migrateReq);
×
2102
  int32_t          contLen = reqLen + sizeof(SMsgHead);
×
UNCOV
2103
  int32_t          code = 0;
×
2104

2105
  while (1) {
×
2106
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
×
UNCOV
2107
    if (pIter == NULL) break;
×
2108

UNCOV
2109
    if (pVgroup->dbUid != pDb->uid) continue;
×
2110

2111
    SMsgHead *pHead = rpcMallocCont(contLen);
×
2112
    if (pHead == NULL) {
×
2113
      sdbCancelFetch(pSdb, pVgroup);
×
2114
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2115
      continue;
×
2116
    }
2117
    pHead->contLen = htonl(contLen);
×
2118
    pHead->vgId = htonl(pVgroup->vgId);
×
2119
    int32_t ret = 0;
×
2120
    if ((ret = tSerializeSVS3MigrateDbReq((char *)pHead + sizeof(SMsgHead), contLen, &s3migrateReq)) < 0) {
×
2121
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2122
      return ret;
×
2123
    }
2124

2125
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_S3MIGRATE, .pCont = pHead, .contLen = contLen};
×
2126
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
×
2127
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
×
2128
    if (code != 0) {
×
UNCOV
2129
      mError("vgId:%d, failed to send vnode-s3migrate request to vnode since 0x%x", pVgroup->vgId, code);
×
2130
    } else {
UNCOV
2131
      mInfo("vgId:%d, send vnode-s3migrate request to vnode, time:%d", pVgroup->vgId, s3migrateReq.timestamp);
×
2132
    }
UNCOV
2133
    sdbRelease(pSdb, pVgroup);
×
2134
  }
2135

UNCOV
2136
  return 0;
×
2137
}
2138

2139
static int32_t mndProcessS3MigrateDbReq(SRpcMsg *pReq) {
×
2140
  SMnode         *pMnode = pReq->info.node;
×
2141
  int32_t         code = -1;
×
2142
  SDbObj         *pDb = NULL;
×
UNCOV
2143
  SS3MigrateDbReq s3migrateReq = {0};
×
2144

UNCOV
2145
  TAOS_CHECK_GOTO(tDeserializeSS3MigrateDbReq(pReq->pCont, pReq->contLen, &s3migrateReq), NULL, _OVER);
×
2146

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

2149
  pDb = mndAcquireDb(pMnode, s3migrateReq.db);
×
2150
  if (pDb == NULL) {
×
2151
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2152
    if (terrno != 0) code = terrno;
×
UNCOV
2153
    goto _OVER;
×
2154
  }
2155

UNCOV
2156
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
×
2157

UNCOV
2158
  code = mndS3MigrateDb(pMnode, pDb);
×
2159

2160
_OVER:
×
2161
  if (code != 0) {
×
UNCOV
2162
    mError("db:%s, failed to process s3migrate db req since %s", s3migrateReq.db, terrstr());
×
2163
  }
2164

2165
  mndReleaseDb(pMnode, pDb);
×
UNCOV
2166
  TAOS_RETURN(code);
×
2167
}
2168

2169
const char *mndGetDbStr(const char *src) {
1,663,521✔
2170
  char *pos = strstr(src, TS_PATH_DELIMITER);
1,663,521✔
2171
  if (pos != NULL) ++pos;
1,663,521✔
2172
  if (pos == NULL) return src;
1,663,521✔
2173
  return pos;
1,610,785✔
2174
}
2175

2176
const char *mndGetStableStr(const char *src) {
1✔
2177
  char *pos = strstr(src, TS_PATH_DELIMITER);
1✔
2178
  if (pos != NULL) ++pos;
1!
2179
  if (pos == NULL) return src;
1!
UNCOV
2180
  return mndGetDbStr(pos);
×
2181
}
2182

2183
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
6✔
2184
  int64_t v = 0;
6✔
2185
  switch (unit) {
6!
2186
    case 's':
2✔
2187
      v = val / 1000;
2✔
2188
      break;
2✔
2189
    case 'm':
1✔
2190
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
1✔
2191
      break;
1✔
2192
    case 'h':
×
2193
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
UNCOV
2194
      break;
×
2195
    case 'd':
3✔
2196
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
3✔
2197
      break;
3✔
2198
    case 'w':
×
2199
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2200
      break;
×
2201
    default:
×
UNCOV
2202
      break;
×
2203
  }
2204

2205
  return v;
6✔
2206
}
2207

2208
static char *buildRetension(SArray *pRetension) {
584,905✔
2209
  size_t size = taosArrayGetSize(pRetension);
584,905✔
2210
  if (size == 0) {
584,796!
2211
    return NULL;
584,956✔
2212
  }
2213

UNCOV
2214
  char       *p1 = taosMemoryCalloc(1, 100);
×
2215
  SRetention *p = taosArrayGet(pRetension, 0);
1✔
2216

2217
  int32_t len = 2;
1✔
2218

2219
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
1✔
2220
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
1✔
2221
  len += sprintf(p1 + len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
1✔
2222

2223
  if (size > 1) {
1!
2224
    len += sprintf(p1 + len, ",");
1✔
2225
    p = taosArrayGet(pRetension, 1);
1✔
2226

2227
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
1✔
2228
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
1✔
2229
    len += sprintf(p1 + len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
1✔
2230
  }
2231

2232
  if (size > 2) {
1!
2233
    len += sprintf(p1 + len, ",");
1✔
2234
    p = taosArrayGet(pRetension, 2);
1✔
2235

2236
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
1✔
2237
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
1✔
2238
    len += sprintf(p1 + len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
1✔
2239
  }
2240

2241
  varDataSetLen(p1, len);
1✔
2242
  return p1;
1✔
2243
}
2244

2245
static const char *getCacheModelStr(int8_t cacheModel) {
584,647✔
2246
  switch (cacheModel) {
584,647!
2247
    case TSDB_CACHE_MODEL_NONE:
223,270✔
2248
      return TSDB_CACHE_MODEL_NONE_STR;
223,270✔
2249
    case TSDB_CACHE_MODEL_LAST_ROW:
114,532✔
2250
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
114,532✔
2251
    case TSDB_CACHE_MODEL_LAST_VALUE:
31,991✔
2252
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
31,991✔
2253
    case TSDB_CACHE_MODEL_BOTH:
216,655✔
2254
      return TSDB_CACHE_MODEL_BOTH_STR;
216,655✔
2255
    default:
×
UNCOV
2256
      break;
×
2257
  }
UNCOV
2258
  return "unknown";
×
2259
}
2260

2261
static const char *getEncryptAlgorithmStr(int8_t encryptAlgorithm) {
583,820✔
2262
  switch (encryptAlgorithm) {
583,820!
2263
    case TSDB_ENCRYPT_ALGO_NONE:
583,835✔
2264
      return TSDB_ENCRYPT_ALGO_NONE_STR;
583,835✔
2265
    case TSDB_ENCRYPT_ALGO_SM4:
×
2266
      return TSDB_ENCRYPT_ALGO_SM4_STR;
×
2267
    default:
×
UNCOV
2268
      break;
×
2269
  }
UNCOV
2270
  return "unknown";
×
2271
}
2272

2273
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
592,651✔
2274
  if (pDb->cfg.replications == 1) return true;
592,651✔
2275

2276
  SSdb *pSdb = pMnode->pSdb;
1,006✔
2277
  void *pIter = NULL;
1,006✔
2278
  bool  isReady = true;
1,006✔
2279
  while (1) {
6,295✔
2280
    SVgObj *pVgroup = NULL;
7,301✔
2281
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
7,301✔
2282
    if (pIter == NULL) break;
7,336✔
2283

2284
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
6,295✔
2285
      bool hasLeader = false;
3,053✔
2286
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
12,151✔
2287
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
9,098✔
2288
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
8,117!
2289
          hasLeader = true;
981✔
2290
        }
2291
      }
2292
      if (!hasLeader) isReady = false;
3,053✔
2293
    }
2294
    sdbRelease(pSdb, pVgroup);
6,295✔
2295
  }
2296

2297
  return isReady;
1,041✔
2298
}
2299

2300
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
639,588✔
2301
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2302
  int32_t cols = 0;
639,588✔
2303
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
639,588✔
2304
  char   *buf = taosMemoryMalloc(bytes);
639,588✔
2305
  if (buf == NULL) {
639,337!
2306
    mError("db:%s, failed to malloc buffer", pDb->name);
×
UNCOV
2307
    return;
×
2308
  }
2309
  int32_t code = 0;
639,337✔
2310
  int32_t lino = 0;
639,337✔
2311

2312
  const char *name = mndGetDbStr(pDb->name);
639,337✔
2313
  if (name != NULL) {
639,303✔
2314
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
639,170✔
2315
  } else {
2316
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
133✔
2317
  }
2318

2319
  const char *precStr = NULL;
639,303✔
2320
  switch (pDb->cfg.precision) {
639,303✔
2321
    case TSDB_TIME_PRECISION_MILLI:
639,208✔
2322
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
639,208✔
2323
      break;
639,208✔
2324
    case TSDB_TIME_PRECISION_MICRO:
14✔
2325
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
14✔
2326
      break;
14✔
2327
    case TSDB_TIME_PRECISION_NANO:
16✔
2328
      precStr = TSDB_TIME_PRECISION_NANO_STR;
16✔
2329
      break;
16✔
2330
    default:
65✔
2331
      precStr = "none";
65✔
2332
      break;
65✔
2333
  }
2334
  char precVstr[10] = {0};
639,303✔
2335
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
639,303✔
2336

2337
  char *statusStr = "ready";
639,303✔
2338
  if (objStatus == SDB_STATUS_CREATING) {
639,303✔
2339
    statusStr = "creating";
34✔
2340
  } else if (objStatus == SDB_STATUS_DROPPING) {
639,269✔
2341
    statusStr = "dropping";
357✔
2342
  } else {
2343
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
638,912✔
2344
      statusStr = "unsynced";
29✔
2345
    }
2346
  }
2347
  char statusVstr[24] = {0};
639,272✔
2348
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
639,272✔
2349

2350
  if (sysDb || !sysinfo) {
639,272✔
2351
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
1,725,961✔
2352
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
1,669,746✔
2353
      if (i == 0) {
1,663,935✔
2354
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
41,098!
2355
      } else if (i == 1) {
1,622,837✔
2356
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
51,372!
2357
      } else if (i == 3) {
1,571,465✔
2358
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
51,320!
2359
      } else if (i == 14) {
1,520,145✔
2360
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
51,452!
2361
      } else if (i == 15) {
1,468,693✔
2362
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
51,393!
2363
      } else {
2364
        colDataSetNULL(pColInfo, rows);
1,417,300✔
2365
      }
2366
    }
2367
  } else {
2368
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
587,662✔
2369
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
586,644!
2370

2371
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
585,380✔
2372
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
584,766!
2373

2374
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,321✔
2375
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
583,828!
2376

2377
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,101✔
2378
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
583,696!
2379

2380
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,608✔
2381
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
583,461!
2382

2383
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
583,813!
2384
    char        strictVstr[24] = {0};
583,813✔
2385
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
583,813✔
2386
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,813✔
2387
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
584,427!
2388

2389
    char    durationVstr[128] = {0};
585,208✔
2390
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE, pDb->cfg.daysPerFile);
585,208✔
2391

2392
    varDataSetLen(durationVstr, len);
587,636✔
2393
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
587,636✔
2394
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
586,243!
2395

2396
    char keepVstr[512] = {0};
585,488✔
2397
    char keep0Str[128] = {0};
585,488✔
2398
    char keep1Str[128] = {0};
585,488✔
2399
    char keep2Str[128] = {0};
585,488✔
2400

2401
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
585,488✔
2402
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
587,903✔
2403
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
587,999✔
2404

2405
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
587,922!
UNCOV
2406
        len = sprintf(&keepVstr[VARSTR_HEADER_SIZE], "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
2407
    } else {
2408
        len = sprintf(&keepVstr[VARSTR_HEADER_SIZE], "%s,%s,%s", keep0Str, keep1Str, keep2Str);
587,922✔
2409
    }
2410
    varDataSetLen(keepVstr, len);
587,922✔
2411
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
587,922✔
2412
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
586,068!
2413

2414
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,879✔
2415
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
584,257!
2416

2417
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,929✔
2418
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
583,196!
2419

2420
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,308✔
2421
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
582,998!
2422

2423
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,354✔
2424
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
583,018!
2425

2426
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,471✔
2427
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
582,962!
2428

2429
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,307✔
2430
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
583,005!
2431

2432
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,130✔
2433
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
582,768!
2434

2435
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,394✔
2436
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
583,963!
2437

2438
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
584,728✔
2439
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
585,006✔
2440
    if (rentensionVstr == NULL) {
584,610✔
2441
      colDataSetNULL(pColInfo, rows);
584,609!
2442
    } else {
2443
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
1!
2444
      taosMemoryFree(rentensionVstr);
1✔
2445
    }
2446

2447
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,610✔
2448
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
584,224!
2449

2450
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
584,676✔
2451
    char        cacheModelVstr[24] = {0};
584,648✔
2452
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
584,648✔
2453
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,648✔
2454
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
585,018!
2455

2456
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,818✔
2457
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
584,279!
2458

2459
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,840✔
2460
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
583,244!
2461

2462
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,288✔
2463
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
582,809!
2464

2465
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,539✔
2466
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
583,098!
2467

2468
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,494✔
2469
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
582,995!
2470

2471
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,435✔
2472
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
583,153!
2473

2474
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,604✔
2475
    int16_t hashPrefix = pDb->cfg.hashPrefix;
583,264✔
2476
    if (hashPrefix > 0) {
583,264✔
2477
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
9✔
2478
    } else if (hashPrefix < 0) {
583,255✔
2479
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
16✔
2480
    }
2481
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
583,264!
2482

2483
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,618✔
2484
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
583,277!
2485

2486
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,397✔
2487
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
582,977!
2488

2489
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,206✔
2490
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
583,200!
2491

2492
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,671✔
2493
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3ChunkSize, false), &lino, _OVER);
583,273!
2494

2495
    char keeplocalVstr[128] = {0};
583,949✔
2496
    len = sprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], "%dm", pDb->cfg.s3KeepLocal);
583,949✔
2497
    varDataSetLen(keeplocalVstr, len);
583,949✔
2498
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,949✔
2499
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
586,518!
2500

2501
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
585,138✔
2502
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3Compact, false), &lino, _OVER);
584,544!
2503

2504
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,723✔
2505
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
583,081!
2506

2507
    const char *encryptAlgorithmStr = getEncryptAlgorithmStr(pDb->cfg.encryptAlgorithm);
583,194✔
2508
    char        encryptAlgorithmVStr[24] = {0};
583,826✔
2509
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
583,826✔
2510
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,826✔
2511
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
584,857!
2512
  }
2513
_OVER:
641,378✔
2514
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
641,378!
2515
  taosMemoryFree(buf);
641,378✔
2516
}
2517

2518
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
25,779✔
2519
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
25,779✔
2520
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
25,779✔
2521
  pDbObj->cfg.numOfVgroups = 0;
25,804✔
2522
  pDbObj->cfg.strict = 1;
25,804✔
2523
  pDbObj->cfg.replications = 1;
25,804✔
2524
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
25,804✔
2525
}
25,804✔
2526

2527
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
25,789✔
2528
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
25,789✔
2529
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
25,789✔
2530
  pDbObj->cfg.numOfVgroups = 0;
25,804✔
2531
  pDbObj->cfg.strict = 1;
25,804✔
2532
  pDbObj->cfg.replications = 1;
25,804✔
2533
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
25,804✔
2534
}
25,804✔
2535

2536
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
27,440,360✔
2537
  SVgObj  *pVgroup = pObj;
27,440,360✔
2538
  int32_t *numOfTables = p1;
27,440,360✔
2539
  int64_t  uid = *(int64_t *)p2;
27,440,360✔
2540
  if (pVgroup->dbUid == uid) {
27,440,360✔
2541
    *numOfTables += pVgroup->numOfTables;
1,151,773✔
2542
  }
2543
  return true;
27,440,360✔
2544
}
2545

2546
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
25,773✔
2547
  SMnode    *pMnode = pReq->info.node;
25,773✔
2548
  SSdb      *pSdb = pMnode->pSdb;
25,773✔
2549
  int32_t    numOfRows = 0;
25,773✔
2550
  SDbObj    *pDb = NULL;
25,773✔
2551
  SUserObj  *pUser = NULL;
25,773✔
2552
  ESdbStatus objStatus = 0;
25,773✔
2553

2554
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
25,773✔
2555
  if (pUser == NULL) return 0;
25,794!
2556
  bool sysinfo = pUser->sysInfo;
25,794✔
2557

2558
  // Append the information_schema database into the result.
2559
  if (!pShow->sysDbRsp) {
25,794✔
2560
    SDbObj infoschemaDb = {0};
25,792✔
2561
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
25,792✔
2562
    size_t numOfTables = 0;
25,804✔
2563
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
25,804✔
2564
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
25,793✔
2565

2566
    numOfRows += 1;
25,772✔
2567

2568
    SDbObj perfschemaDb = {0};
25,772✔
2569
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
25,772✔
2570
    numOfTables = 0;
25,804✔
2571
    getPerfDbMeta(NULL, &numOfTables);
25,804✔
2572
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
25,804✔
2573

2574
    numOfRows += 1;
25,784✔
2575
    pShow->sysDbRsp = true;
25,784✔
2576
  }
2577

2578
  while (numOfRows < rowsCapacity) {
613,935✔
2579
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
613,911✔
2580
    if (pShow->pIter == NULL) break;
613,842✔
2581

2582
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
588,038✔
2583
      int32_t numOfTables = 0;
588,125✔
2584
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
588,125✔
2585
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
588,092✔
2586
      numOfRows++;
587,037✔
2587
    }
2588

2589
    sdbRelease(pSdb, pDb);
587,059✔
2590
  }
2591

2592
  pShow->numOfRows += numOfRows;
25,828✔
2593
  mndReleaseUser(pMnode, pUser);
25,828✔
2594
  return numOfRows;
25,805✔
2595
}
2596

2597
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
2598
  SSdb *pSdb = pMnode->pSdb;
×
2599
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
UNCOV
2600
}
×
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

© 2025 Coveralls, Inc