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

taosdata / TDengine / #3548

04 Dec 2024 01:03PM UTC coverage: 59.846% (-0.8%) from 60.691%
#3548

push

travis-ci

web-flow
Merge pull request #29033 from taosdata/fix/calculate-vnode-memory-used

fix/calculate-vnode-memory-used

118484 of 254183 branches covered (46.61%)

Branch coverage included in aggregate %.

199691 of 277471 relevant lines covered (71.97%)

18794141.86 hits per line

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

68.06
/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) {
1,701✔
63
  SSdbTable table = {
1,701✔
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);
1,701✔
75
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
1,701✔
76
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
1,701✔
77
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
1,701✔
78
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
1,701✔
79
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
1,701✔
80
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
1,701✔
81
  mndSetMsgHandle(pMnode, TDMT_MND_S3MIGRATE_DB, mndProcessS3MigrateDbReq);
1,701✔
82

83
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
1,701✔
84
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
1,701✔
85

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

89
void mndCleanupDb(SMnode *pMnode) {}
1,701✔
90

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

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

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

155
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
20,030!
156
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
20,030!
157

158
  terrno = 0;
20,030✔
159

160
_OVER:
20,030✔
161
  if (terrno != 0) {
20,030!
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);
20,030✔
168
  return pRaw;
20,030✔
169
}
170

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

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

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

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

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

192
  int32_t dataPos = 0;
17,243✔
193
  SDB_GET_BINARY(pRaw, dataPos, pDb->name, TSDB_DB_FNAME_LEN, _OVER)
17,243!
194
  SDB_GET_BINARY(pRaw, dataPos, pDb->acct, TSDB_USER_LEN, _OVER)
17,243!
195
  SDB_GET_BINARY(pRaw, dataPos, pDb->createUser, TSDB_USER_LEN, _OVER)
17,243!
196
  SDB_GET_INT64(pRaw, dataPos, &pDb->createdTime, _OVER)
17,243!
197
  SDB_GET_INT64(pRaw, dataPos, &pDb->updateTime, _OVER)
17,243!
198
  SDB_GET_INT64(pRaw, dataPos, &pDb->uid, _OVER)
17,243!
199
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfgVersion, _OVER)
17,243!
200
  SDB_GET_INT32(pRaw, dataPos, &pDb->vgVersion, _OVER)
17,243!
201
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfVgroups, _OVER)
17,243!
202
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfStables, _OVER)
17,243!
203
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.buffer, _OVER)
17,243!
204
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.pageSize, _OVER)
17,243!
205
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.pages, _OVER)
17,243!
206
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.cacheLastSize, _OVER)
17,243!
207
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysPerFile, _OVER)
17,243!
208
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep0, _OVER)
17,243!
209
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep1, _OVER)
17,243!
210
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep2, _OVER)
17,243!
211
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.minRows, _OVER)
17,243!
212
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.maxRows, _OVER)
17,243!
213
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walFsyncPeriod, _OVER)
17,243!
214
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.walLevel, _OVER)
17,243!
215
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.precision, _OVER)
17,243!
216
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.compression, _OVER)
17,243!
217
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.replications, _OVER)
17,243!
218
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.strict, _OVER)
17,243!
219
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.cacheLast, _OVER)
17,243!
220
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.hashMethod, _OVER)
17,243!
221
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfRetensions, _OVER)
17,243!
222
  if (pDb->cfg.numOfRetensions > 0) {
17,243✔
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)
17,243!
237
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walRetentionPeriod, _OVER)
17,243!
238
  SDB_GET_INT64(pRaw, dataPos, &pDb->cfg.walRetentionSize, _OVER)
17,243!
239
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walRollPeriod, _OVER)
17,243!
240
  SDB_GET_INT64(pRaw, dataPos, &pDb->cfg.walSegmentSize, _OVER)
17,243!
241
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.sstTrigger, _OVER)
17,243!
242
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.hashPrefix, _OVER)
17,243!
243
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.hashSuffix, _OVER)
17,243!
244
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.tsdbPageSize, _OVER)
17,243!
245
  SDB_GET_INT64(pRaw, dataPos, &pDb->compactStartTime, _OVER)
17,243!
246
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.keepTimeOffset, _OVER)
17,243!
247
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.s3ChunkSize, _OVER)
17,243!
248
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.s3KeepLocal, _OVER)
17,243!
249
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.s3Compact, _OVER)
17,243!
250
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.withArbitrator, _OVER)
17,243!
251
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.encryptAlgorithm, _OVER)
17,243!
252
  SDB_GET_INT32(pRaw, dataPos, &pDb->tsmaVersion, _OVER);
17,243!
253

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

257
  if (pDb->cfg.s3ChunkSize == 0) {
17,243!
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) {
17,243!
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) {
17,243!
270
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
17,243!
271
          TSDB_DEFAULT_TSDB_PAGESIZE);
272
  }
273

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

278
  terrno = 0;
17,243✔
279

280
_OVER:
17,243✔
281
  if (terrno != 0) {
17,243!
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);
17,243✔
288
  return pRow;
17,243✔
289
}
290

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

297
  pRow = mndDbActionDecode(pRaw);
3,963✔
298
  if (pRow == NULL) {
3,963!
299
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
300
    if (terrno != 0) code = terrno;
×
301
    goto _OVER;
×
302
  }
303
  pNewDb = sdbGetRowObj(pRow);
3,963✔
304
  if (pNewDb == NULL) {
3,963!
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);
3,963✔
311
  if (pOldDb != NULL) {
3,963!
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;
3,963✔
320
_OVER:
3,963✔
321
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
3,963!
322
  taosMemoryFreeClear(pRow);
3,963!
323
  return code;
3,963✔
324
}
325

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

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

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

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

379
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
3,592,313✔
380
  SSdb   *pSdb = pMnode->pSdb;
3,592,313✔
381
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
3,592,313✔
382
  if (pDb == NULL) {
3,592,398✔
383
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
7,538✔
384
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
6,272✔
385
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
1,266✔
386
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
102✔
387
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
1,164!
388
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
1,164✔
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,592,356✔
395
}
396

397
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
3,963,046✔
398
  SSdb *pSdb = pMnode->pSdb;
3,963,046✔
399
  sdbRelease(pSdb, pDb);
3,963,046✔
400
}
3,963,112✔
401

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

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

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

424
  return 0;
4,197✔
425
}
426

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

430
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
4,197!
431
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
4,197!
432
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
4,197!
433
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
4,197!
434
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
4,197!
435
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
4,197!
436
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
4,197!
437
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
4,197!
438
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
4,197!
439
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
4,197!
440
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
4,197!
441
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
4,197!
442
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
4,197!
443
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
4,197!
444
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
4,197!
445
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
4,197!
446
  if (pCfg->minRows > pCfg->maxRows) return code;
4,197✔
447
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
4,196!
448
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
4,196!
449
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
4,196!
450
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
4,196!
451
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
4,196!
452
#ifdef TD_ENTERPRISE
453
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
4,196!
454
  if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
4,196!
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;
4,196!
461
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
4,196!
462
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
4,196!
463
  if (pCfg->hashMethod != 1) return code;
4,196!
464
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
4,196!
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;
4,196!
469
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
4,196!
470
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
4,196!
471
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
4,196!
472
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
4,196!
473
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
4,196!
474
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
4,196!
475
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
4,196!
476
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
4,196!
477
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
4,196!
478
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
4,196!
479

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

484
  code = 0;
4,196✔
485
  TAOS_RETURN(code);
4,196✔
486
}
487

488
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
181✔
489
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
181✔
490
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
181!
491
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
181!
492
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
181!
493
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
181!
494
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
181!
495
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
181!
496
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
181!
497
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
181!
498
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
181!
499
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
181!
500
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
181!
501
    return code;
×
502
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
181!
503
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
181!
504
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
181!
505
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
181!
506
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
181!
507
#ifdef TD_ENTERPRISE
508
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
181!
509
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
181!
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) {
180!
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) {
180!
524
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
525
    return code;
×
526
  }
527
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
180!
528
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
529
    return code;
×
530
  }
531

532
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
180!
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;
180!
536
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
180!
537
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
180!
538
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
180!
539
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
180!
540
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
180!
541
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
180!
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;
180!
546
  if (pNewCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pNewCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
180!
547
  if (pNewCfg->s3Compact < TSDB_MIN_S3_COMPACT || pNewCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
180!
548

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

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

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

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

600
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
3,737✔
601
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
12,118✔
602
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
8,381!
603
  }
604
  return 0;
3,737✔
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) {
3,737✔
642
  int32_t  code = 0;
3,737✔
643
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
3,737✔
644
  if (pDbRaw == NULL) {
3,737!
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));
3,737!
650
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
3,737!
651

652
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
12,118✔
653
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
8,381✔
654
    if (pVgRaw == NULL) {
8,381!
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));
8,381!
660
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
8,381!
661
  }
662

663
  if (pDb->cfg.withArbitrator) {
3,737✔
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);
3,737✔
673
}
674

675
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
3,737✔
676
                                        SUserObj *pUserDuped) {
677
  int32_t  code = 0;
3,737✔
678
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
3,737✔
679
  if (pDbRaw == NULL) {
3,737!
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));
3,737!
685
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
3,737!
686

687
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
12,118✔
688
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
8,381✔
689
    if (pVgRaw == NULL) {
8,381!
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));
8,381!
695
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
8,381!
696
  }
697

698
  if (pDb->cfg.withArbitrator) {
3,737✔
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) {
3,737✔
708
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
4✔
709
    if (pUserRaw == NULL) {
4!
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));
4!
715
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
4!
716
  }
717

718
  TAOS_RETURN(code);
3,737✔
719
}
720

721
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
3,737✔
722
  int32_t code = 0;
3,737✔
723
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
12,118✔
724
    SVgObj *pVgroup = pVgroups + vg;
8,381✔
725

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

732
  TAOS_RETURN(code);
3,737✔
733
}
734

735
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
3,737✔
736
  int32_t code = 0;
3,737✔
737
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
12,118✔
738
    SVgObj *pVgroup = pVgroups + vg;
8,381✔
739

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

746
  TAOS_RETURN(code);
3,737✔
747
}
748

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

800
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
4,197✔
801
  dbObj.cfg.pRetensions = pCreate->pRetensions;
4,197✔
802

803
  mndSetDefaultDbCfg(&dbObj.cfg);
4,197✔
804

805
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
4,197!
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) {
4,197✔
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) {
4,196✔
816
    int32_t dbLen = strlen(dbObj.name) + 1;
11✔
817
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
11!
818
    dbObj.cfg.hashPrefix += dbLen;
11✔
819
  } else if (dbObj.cfg.hashPrefix < 0) {
4,185✔
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;
4,196✔
826
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
4,196✔
827
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
451!
828
    TAOS_RETURN(code);
451✔
829
  }
830

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

842
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
3,745✔
843
  if (pTrans == NULL) {
3,745!
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);
3,745!
850

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

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

863
_OVER:
3,732✔
864
  taosMemoryFree(pVgroups);
3,745✔
865
  mndUserFreeObj(&newUserObj);
3,745✔
866
  mndTransDrop(pTrans);
3,745✔
867
  TAOS_RETURN(code);
3,745✔
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) {
4,213✔
887
  int32_t    code = 0;
4,213✔
888
  SSdb      *pSdb = pMnode->pSdb;
4,213✔
889
  SDnodeObj *pDnode = NULL;
4,213✔
890
  void      *pIter = NULL;
4,213✔
891

892
#ifdef TD_ENTERPRISE
893
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
4,213!
894
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
×
895
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
×
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();
×
903
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
×
904
    bool online = false;
×
905
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
×
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);
×
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:
×
925
  TAOS_RETURN(code);
4,213✔
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) {
4,617✔
940
  SMnode      *pMnode = pReq->info.node;
4,617✔
941
  int32_t      code = -1;
4,617✔
942
  int32_t      lino = 0;
4,617✔
943
  SDbObj      *pDb = NULL;
4,617✔
944
  SUserObj    *pUser = NULL;
4,617✔
945
  SCreateDbReq createReq = {0};
4,617✔
946
  SArray      *dnodeList = NULL;
4,617✔
947

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

951
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
4,617!
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);
4,617!
959

960
  pDb = mndAcquireDb(pMnode, createReq.db);
4,617✔
961
  if (pDb != NULL) {
4,617✔
962
    if (createReq.ignoreExist) {
354✔
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;
13✔
968
      goto _OVER;
13✔
969
    }
970
  } else {
971
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
4,263✔
972
      code = terrno;
49✔
973
      goto _OVER;
49✔
974
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
4,214!
975
      goto _OVER;
×
976
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
4,214!
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);
4,214✔
984

985
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
4,213!
986

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

991
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
4,213!
992

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

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

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

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

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

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

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

1016
  TAOS_RETURN(code);
4,617✔
1017
}
1018

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

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

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

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

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

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

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

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

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

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

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

1072
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
207!
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) {
207!
1082
    pDb->cfg.cacheLast = pAlter->cacheLast;
132✔
1083
    code = 0;
132✔
1084
  }
1085

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

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

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

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

1109
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
207✔
1110
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
44✔
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 &&
207!
1117
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
×
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) {
207!
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) {
207!
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) {
207✔
1136
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
1✔
1137
    pDb->vgVersion++;
1✔
1138
    code = 0;
1✔
1139
  }
1140

1141
  TAOS_RETURN(code);
207✔
1142
}
1143

1144
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
179✔
1145
  int32_t  code = 0;
179✔
1146
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
179✔
1147
  if (pRedoRaw == NULL) {
179!
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) {
179!
1153
    sdbFreeRaw(pRedoRaw);
×
1154
    TAOS_RETURN(code);
×
1155
  }
1156

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

1164
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
179✔
1165
  int32_t  code = 0;
179✔
1166
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
179✔
1167
  if (pCommitRaw == NULL) {
179!
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) {
179!
1173
    sdbFreeRaw(pCommitRaw);
×
1174
    TAOS_RETURN(code);
×
1175
  }
1176

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

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

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

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

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

1202
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
445!
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,742✔
1216
  }
1217

1218
  taosArrayDestroy(pArray);
179✔
1219
  TAOS_RETURN(code);
179✔
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) {
180✔
1231
  int32_t code = -1;
180✔
1232
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
180✔
1233
  if (pTrans == NULL) {
180!
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);
180!
1239

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1307
  if (dbObj.cfg.replications != pDb->cfg.replications) {
180✔
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);
5!
1310
  } else {
1311
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
175!
1312
  }
1313

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

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

1320
_OVER:
208✔
1321
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
208✔
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);
208✔
1327
  taosArrayDestroy(dbObj.cfg.pRetensions);
208✔
1328
  tFreeSAlterDbReq(&alterReq);
208✔
1329

1330
  TAOS_RETURN(code);
208✔
1331
}
1332

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

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

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

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

1393
    mndDumpDbCfgInfo(&cfgRsp, pDb);
71,531✔
1394
  }
1395

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

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

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

1413
  code = 0;
71,531✔
1414

1415
_OVER:
71,579✔
1416

1417
  tFreeSDbCfgRsp(&cfgRsp);
71,579✔
1418

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

1423
  mndReleaseDb(pMnode, pDb);
71,579✔
1424

1425
  TAOS_RETURN(code);
71,579✔
1426
}
1427

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

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

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

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

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

1458
  TAOS_RETURN(code);
1,842✔
1459
}
1460

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

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

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

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

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

1491
  while (1) {
14,769✔
1492
    SVgObj *pVgroup = NULL;
16,611✔
1493
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
16,611✔
1494
    if (pIter == NULL) break;
16,611✔
1495

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

1517
    sdbRelease(pSdb, pVgroup);
14,769✔
1518
  }
1519

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

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

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

1549
  TAOS_RETURN(code);
1,842✔
1550
}
1551

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

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

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

1567
  while (1) {
14,760✔
1568
    SVgObj *pVgroup = NULL;
16,601✔
1569
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
16,601✔
1570
    if (pIter == NULL) break;
16,601✔
1571

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

1580
    sdbRelease(pSdb, pVgroup);
14,760✔
1581
  }
1582

1583
  TAOS_RETURN(code);
1,841✔
1584
}
1585

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

1594
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
3,652✔
1595
  void   *pRsp = NULL;
3,652✔
1596
  if (useRpcMalloc) {
3,652✔
1597
    pRsp = rpcMallocCont(rspLen);
1,811✔
1598
  } else {
1599
    pRsp = taosMemoryMalloc(rspLen);
1,841✔
1600
  }
1601

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

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

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

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

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

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

1629
  if (mndTopicExistsForDb(pMnode, pDb)) {
1,843✔
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);
1,842!
1635
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
1,842!
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);
1,842✔
1640
#ifdef TD_ENTERPRISE
1641
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,841!
1642
#endif
1643
  TAOS_CHECK_GOTO(mndDropSmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,841!
1644
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,841!
1645
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
1,841!
1646
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb->name), NULL, _OVER);
1,841!
1647

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

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

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

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

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

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

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

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

1683
  code = mndDropDb(pMnode, pReq, pDb);
1,843✔
1684
  if (code == TSDB_CODE_SUCCESS) {
1,843✔
1685
    code = TSDB_CODE_ACTION_IN_PROGRESS;
1,841✔
1686
  }
1687

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

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

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

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

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

1709
  void *pIter = NULL;
2,504,272✔
1710
  while (vindex < pDb->cfg.numOfVgroups) {
75,946,251✔
1711
    SVgObj *pVgroup = NULL;
73,440,541✔
1712
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
73,440,541✔
1713
    if (pIter == NULL) break;
73,440,906!
1714

1715
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
73,440,906✔
1716
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
5,028,610✔
1717
      vindex++;
5,028,610✔
1718
    }
1719

1720
    sdbRelease(pSdb, pVgroup);
73,440,501✔
1721
  }
1722

1723
  sdbCancelFetch(pSdb, pIter);
2,505,710✔
1724
  return numOfTables;
2,504,308✔
1725
}
1726

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

1731
  void *pIter = NULL;
2,656,621✔
1732
  while (1) {
79,894,054✔
1733
    SVgObj *pVgroup = NULL;
82,550,675✔
1734
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
82,550,675✔
1735
    if (pIter == NULL) break;
82,549,492✔
1736

1737
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
82,278,707!
1738
      SVgroupInfo vgInfo = {0};
17,105,371✔
1739
      vgInfo.vgId = pVgroup->vgId;
17,105,371✔
1740
      vgInfo.hashBegin = pVgroup->hashBegin;
17,105,371✔
1741
      vgInfo.hashEnd = pVgroup->hashEnd;
17,105,371✔
1742
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
17,105,371✔
1743
      vgInfo.epSet.numOfEps = pVgroup->replica;
17,105,371✔
1744
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
34,221,685✔
1745
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
17,116,240✔
1746
        SEp       *pEp = &vgInfo.epSet.eps[gid];
17,116,240✔
1747
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
17,116,240✔
1748
        if (pDnode != NULL) {
17,116,237✔
1749
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
17,116,127✔
1750
          pEp->port = pDnode->port;
17,116,127✔
1751
        }
1752
        mndReleaseDnode(pMnode, pDnode);
17,116,237✔
1753
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
17,116,314!
1754
          vgInfo.epSet.inUse = gid;
17,103,750✔
1755
        }
1756
      }
1757
      vindex++;
17,105,445✔
1758
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
17,105,269!
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);
82,278,606✔
1764

1765
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
82,279,901✔
1766
      sdbCancelFetch(pSdb, pIter);
2,385,847✔
1767
      break;
2,385,844✔
1768
    }
1769
  }
1770
}
2,656,629✔
1771

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

1780
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
2,465,712✔
1781

1782
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
2,465,750!
1783
      pReq->stateTs < pDb->stateTs) {
82,495✔
1784
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
2,383,598✔
1785
  }
1786

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

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

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

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

1815
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
269,660✔
1816
      usedbRsp.vgVersion = vgVersion++;
269,660✔
1817
    } else {
1818
      usedbRsp.vgVersion = usedbReq.vgVersion;
25✔
1819
    }
1820
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
269,685✔
1821
    code = 0;
269,685✔
1822
  } else {
1823
    pDb = mndAcquireDb(pMnode, usedbReq.db);
2,460,452✔
1824
    if (pDb == NULL) {
2,460,443✔
1825
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
429✔
1826
      usedbRsp.uid = usedbReq.dbId;
429✔
1827
      usedbRsp.vgVersion = usedbReq.vgVersion;
429✔
1828
      usedbRsp.errCode = terrno;
429✔
1829

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

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

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

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

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

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

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

1866
  mndReleaseDb(pMnode, pDb);
2,730,136✔
1867
  tFreeSUsedbRsp(&usedbRsp);
2,730,147✔
1868

1869
  TAOS_RETURN(code);
2,730,139✔
1870
}
1871

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

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

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

1894
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
48,450✔
1895
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
44,833✔
1896
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
9,681✔
1897
      if (pDbCacheInfo->vgVersion >= vgVersion) {
9,681✔
1898
        continue;
31,849✔
1899
      }
1900

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

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

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

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

1915
      continue;
1,125✔
1916
    }
1917

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

1932
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
38,558✔
1933

1934
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
38,558✔
1935
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
25,622!
1936
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
23,607✔
1937
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
21,955✔
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);
21,955✔
1942
      continue;
21,955✔
1943
    } else {
1944
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
16,603✔
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) {
16,603✔
1951
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
12,889✔
1952
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb);
12,889✔
1953
    }
1954

1955
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
16,603✔
1956
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
15,707✔
1957
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
15,707!
1958
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
15,707!
1959
        bool exist = false;
15,707✔
1960
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
15,707✔
1961
        if (TSDB_CODE_SUCCESS != code) {
15,707✔
1962
          mndReleaseDb(pMnode, pDb);
2✔
1963
          if (code != TSDB_CODE_NEED_RETRY) {
2!
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;
15,705✔
1973
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
15,705✔
1974
      }
1975
    }
1976

1977
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
16,601!
1978
        pDbCacheInfo->stateTs != pDb->stateTs) {
16,552✔
1979
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
2,247✔
1980
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
2,247✔
1981
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
2,247!
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);
×
1986
        continue;
×
1987
      }
1988

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

2000
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
33,202!
2001
      mndReleaseDb(pMnode, pDb);
×
2002
      if (terrno != 0) code = terrno;
×
2003
      return code;
×
2004
    }
2005
    mndReleaseDb(pMnode, pDb);
16,601✔
2006
  }
2007

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

2018
  *ppRsp = pRsp;
45,365✔
2019
  *pRspLen = rspLen;
45,365✔
2020

2021
  tFreeSDbHbBatchRsp(&batchRsp);
45,365✔
2022
  TAOS_RETURN(code);
45,365✔
2023
}
2024

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

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

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

2052
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM, .pCont = pHead, .contLen = contLen};
2✔
2053
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
2✔
2054
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
2✔
2055
    if (code != 0) {
2!
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);
2!
2059
    }
2060
    sdbRelease(pSdb, pVgroup);
2✔
2061
  }
2062

2063
  return 0;
1✔
2064
}
2065

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

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

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

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

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

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

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

2092
  mndReleaseDb(pMnode, pDb);
1✔
2093
  TAOS_RETURN(code);
1✔
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);
×
2103
  int32_t          code = 0;
×
2104

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

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);
×
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);
×
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) {
×
2129
      mError("vgId:%d, failed to send vnode-s3migrate request to vnode since 0x%x", pVgroup->vgId, code);
×
2130
    } else {
2131
      mInfo("vgId:%d, send vnode-s3migrate request to vnode, time:%d", pVgroup->vgId, s3migrateReq.timestamp);
×
2132
    }
2133
    sdbRelease(pSdb, pVgroup);
×
2134
  }
2135

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;
×
2143
  SS3MigrateDbReq s3migrateReq = {0};
×
2144

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

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;
×
2153
    goto _OVER;
×
2154
  }
2155

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

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

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

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

2169
const char *mndGetDbStr(const char *src) {
1,655,384✔
2170
  char *pos = strstr(src, TS_PATH_DELIMITER);
1,655,384✔
2171
  if (pos != NULL) ++pos;
1,655,384✔
2172
  if (pos == NULL) return src;
1,655,384✔
2173
  return pos;
1,604,431✔
2174
}
2175

2176
const char *mndGetStableStr(const char *src) {
×
2177
  char *pos = strstr(src, TS_PATH_DELIMITER);
×
2178
  if (pos != NULL) ++pos;
×
2179
  if (pos == NULL) return src;
×
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);
×
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:
×
2202
      break;
×
2203
  }
2204

2205
  return v;
6✔
2206
}
2207

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

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,762✔
2246
  switch (cacheModel) {
584,762!
2247
    case TSDB_CACHE_MODEL_NONE:
247,481✔
2248
      return TSDB_CACHE_MODEL_NONE_STR;
247,481✔
2249
    case TSDB_CACHE_MODEL_LAST_ROW:
115,632✔
2250
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
115,632✔
2251
    case TSDB_CACHE_MODEL_LAST_VALUE:
81,155✔
2252
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
81,155✔
2253
    case TSDB_CACHE_MODEL_BOTH:
141,869✔
2254
      return TSDB_CACHE_MODEL_BOTH_STR;
141,869✔
2255
    default:
×
2256
      break;
×
2257
  }
2258
  return "unknown";
×
2259
}
2260

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

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

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

2284
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
6,087!
2285
      bool hasLeader = false;
2,926✔
2286
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
11,653✔
2287
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
8,727✔
2288
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
7,745!
2289
          hasLeader = true;
982✔
2290
        }
2291
      }
2292
      if (!hasLeader) isReady = false;
2,926✔
2293
    }
2294
    sdbRelease(pSdb, pVgroup);
6,087✔
2295
  }
2296

2297
  return isReady;
975✔
2298
}
2299

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

2312
  const char *name = mndGetDbStr(pDb->name);
637,900✔
2313
  if (name != NULL) {
637,847✔
2314
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
637,693✔
2315
  } else {
2316
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
154✔
2317
  }
2318

2319
  const char *precStr = NULL;
637,847✔
2320
  switch (pDb->cfg.precision) {
637,847✔
2321
    case TSDB_TIME_PRECISION_MILLI:
637,734✔
2322
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
637,734✔
2323
      break;
637,734✔
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:
83✔
2331
      precStr = "none";
83✔
2332
      break;
83✔
2333
  }
2334
  char precVstr[10] = {0};
637,847✔
2335
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
637,847✔
2336

2337
  char *statusStr = "ready";
637,847✔
2338
  if (objStatus == SDB_STATUS_CREATING) {
637,847✔
2339
    statusStr = "creating";
17✔
2340
  } else if (objStatus == SDB_STATUS_DROPPING) {
637,830✔
2341
    statusStr = "dropping";
161✔
2342
  } else {
2343
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
637,669✔
2344
      statusStr = "unsynced";
14✔
2345
    }
2346
  }
2347
  char statusVstr[24] = {0};
637,792✔
2348
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
637,792✔
2349

2350
  if (sysDb || !sysinfo) {
637,792✔
2351
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
1,711,386✔
2352
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
1,656,239✔
2353
      if (i == 0) {
1,651,977✔
2354
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
42,232!
2355
      } else if (i == 1) {
1,609,745✔
2356
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
50,594!
2357
      } else if (i == 3) {
1,559,151✔
2358
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
50,577!
2359
      } else if (i == 14) {
1,508,574✔
2360
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
50,648!
2361
      } else if (i == 15) {
1,457,926✔
2362
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
50,574!
2363
      } else {
2364
        colDataSetNULL(pColInfo, rows);
1,407,352✔
2365
      }
2366
    }
2367
  } else {
2368
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
586,913✔
2369
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
586,169!
2370

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

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

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

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

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

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

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

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

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

2405
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
587,178!
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,178✔
2409
    }
2410
    varDataSetLen(keepVstr, len);
587,178✔
2411
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
587,178✔
2412
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
585,866!
2413

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

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

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

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

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

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

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

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

2438
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
584,659✔
2439
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
584,912✔
2440
    if (rentensionVstr == NULL) {
584,566✔
2441
      colDataSetNULL(pColInfo, rows);
584,565!
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,566✔
2448
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
584,254!
2449

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

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

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

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

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

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

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

2474
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
583,889✔
2475
    int16_t hashPrefix = pDb->cfg.hashPrefix;
583,615✔
2476
    if (hashPrefix > 0) {
583,615✔
2477
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
13✔
2478
    } else if (hashPrefix < 0) {
583,602✔
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,615!
2482

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2566
    numOfRows += 1;
25,365✔
2567

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

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

2578
  while (numOfRows < rowsCapacity) {
612,751✔
2579
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
612,740✔
2580
    if (pShow->pIter == NULL) break;
612,703✔
2581

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

2589
    sdbRelease(pSdb, pDb);
586,506✔
2590
  }
2591

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

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