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

taosdata / TDengine / #4887

16 Dec 2025 08:27AM UTC coverage: 65.289% (-0.003%) from 65.292%
#4887

push

travis-ci

web-flow
feat[TS-7233]: audit (#33850)

377 of 536 new or added lines in 28 files covered. (70.34%)

1025 existing lines in 111 files now uncovered.

178977 of 274129 relevant lines covered (65.29%)

102580217.43 hits per line

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

78.51
/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 "command.h"
20
#include "mndArbGroup.h"
21
#include "mndCluster.h"
22
#include "mndCompact.h"
23
#include "mndCompactDetail.h"
24
#include "mndConfig.h"
25
#include "mndDnode.h"
26
#include "mndEncryptAlgr.h"
27
#include "mndIndex.h"
28
#include "mndPrivilege.h"
29
#include "mndRetention.h"
30
#include "mndRetentionDetail.h"
31
#include "mndRsma.h"
32
#include "mndScan.h"
33
#include "mndShow.h"
34
#include "mndSma.h"
35
#include "mndSsMigrate.h"
36
#include "mndStb.h"
37
#include "mndStream.h"
38
#include "mndSubscribe.h"
39
#include "mndTopic.h"
40
#include "mndTrans.h"
41
#include "mndUser.h"
42
#include "mndVgroup.h"
43
#include "mndView.h"
44
#include "systable.h"
45
#include "thttp.h"
46
#include "tjson.h"
47

48
#define DB_VER_NUMBER   1
49
#define DB_RESERVE_SIZE 13
50

51
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw);
52
static int32_t  mndDbActionInsert(SSdb *pSdb, SDbObj *pDb);
53
static int32_t  mndDbActionDelete(SSdb *pSdb, SDbObj *pDb);
54
static int32_t  mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew);
55
static int32_t  mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw);
56

57
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq);
58
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq);
59
static int32_t mndProcessDropDbReq(SRpcMsg *pReq);
60
static int32_t mndProcessUseDbReq(SRpcMsg *pReq);
61
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq);
62
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq);
63
static int32_t mndProcessTrimDbWalReq(SRpcMsg *pReq);
64
static int32_t mndProcessS3MigrateDbReq(SRpcMsg *pReq);
65
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity);
66
static void    mndCancelGetNextDb(SMnode *pMnode, void *pIter);
67
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq);
68

69
#ifndef TD_ENTERPRISE
70
int32_t mndProcessCompactDbReq(SRpcMsg *pReq) { return TSDB_CODE_OPS_NOT_SUPPORT; }
71
#endif
72

73
int32_t mndInitDb(SMnode *pMnode) {
516,920✔
74
  SSdbTable table = {
516,920✔
75
      .sdbType = SDB_DB,
76
      .keyType = SDB_KEY_BINARY,
77
      .encodeFp = (SdbEncodeFp)mndDbActionEncode,
78
      .decodeFp = (SdbDecodeFp)mndDbActionDecode,
79
      .insertFp = (SdbInsertFp)mndDbActionInsert,
80
      .updateFp = (SdbUpdateFp)mndDbActionUpdate,
81
      .deleteFp = (SdbDeleteFp)mndDbActionDelete,
82
      .validateFp = (SdbValidateFp)mndNewDbActionValidate,
83
  };
84

85
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_DB, mndProcessCreateDbReq);
516,920✔
86
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
516,920✔
87
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
516,920✔
88
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
516,920✔
89
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
516,920✔
90
  mndSetMsgHandle(pMnode, TDMT_MND_SCAN_DB, mndProcessScanDbReq);
516,920✔
91
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
516,920✔
92
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB_WAL, mndProcessTrimDbWalReq);
516,920✔
93
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
516,920✔
94
  mndSetMsgHandle(pMnode, TDMT_MND_SSMIGRATE_DB, mndProcessSsMigrateDbReq);
516,920✔
95
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
516,920✔
96

97
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
516,920✔
98
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
516,920✔
99

100
  return sdbSetTable(pMnode->pSdb, table);
516,920✔
101
}
102

103
void mndCleanupDb(SMnode *pMnode) {}
516,803✔
104

105
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
6,246,029✔
106
  int32_t code = 0;
6,246,029✔
107
  int32_t lino = 0;
6,246,029✔
108
  terrno = TSDB_CODE_OUT_OF_MEMORY;
6,246,029✔
109

110
  int32_t  size = sizeof(SDbObj) + pDb->cfg.numOfRetensions * sizeof(SRetention) + DB_RESERVE_SIZE;
6,246,029✔
111
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, DB_VER_NUMBER, size);
6,246,029✔
112
  if (pRaw == NULL) goto _OVER;
6,246,029✔
113

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

174
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
6,246,029✔
175
  SDB_SET_UINT8(pRaw, dataPos, pDb->cfg.flags, _OVER)
6,246,029✔
176
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
6,246,029✔
177

178
  terrno = 0;
6,246,029✔
179

180
_OVER:
6,246,029✔
181
  if (terrno != 0) {
6,246,029✔
182
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
183
    sdbFreeRaw(pRaw);
×
184
    return NULL;
×
185
  }
186

187
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
6,246,029✔
188
  return pRaw;
6,246,029✔
189
}
190

191
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
5,804,909✔
192
  int32_t code = 0;
5,804,909✔
193
  int32_t lino = 0;
5,804,909✔
194
  terrno = TSDB_CODE_OUT_OF_MEMORY;
5,804,909✔
195
  SSdbRow *pRow = NULL;
5,804,909✔
196
  SDbObj  *pDb = NULL;
5,804,909✔
197

198
  int8_t sver = 0;
5,804,909✔
199
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
5,804,909✔
200

201
  if (sver != DB_VER_NUMBER) {
5,804,909✔
202
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
203
    goto _OVER;
×
204
  }
205

206
  pRow = sdbAllocRow(sizeof(SDbObj));
5,804,909✔
207
  if (pRow == NULL) goto _OVER;
5,804,909✔
208

209
  pDb = sdbGetRowObj(pRow);
5,804,909✔
210
  if (pDb == NULL) goto _OVER;
5,804,909✔
211

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

283
  taosInitRWLatch(&pDb->lock);
5,804,909✔
284

285
  if (pDb->cfg.ssChunkSize == 0) {
5,804,909✔
286
    pDb->cfg.ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
×
287

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

291
  if (pDb->cfg.ssKeepLocal == 0) {
5,804,909✔
292
    pDb->cfg.ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
×
293

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

297
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
5,804,909✔
298
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
5,804,909✔
299
          TSDB_DEFAULT_TSDB_PAGESIZE);
300
  }
301

302
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
5,804,909✔
303
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
5,694,330✔
304
  }
305

306
  terrno = 0;
5,804,909✔
307

308
_OVER:
5,804,909✔
309
  if (terrno != 0) {
5,804,909✔
310
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
311
    taosMemoryFreeClear(pRow);
×
312
    return NULL;
×
313
  }
314

315
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
5,804,909✔
316
  return pRow;
5,804,909✔
317
}
318

319
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
1,310,418✔
320
  SSdb    *pSdb = pMnode->pSdb;
1,310,418✔
321
  SSdbRow *pRow = NULL;
1,310,418✔
322
  SDbObj  *pNewDb = NULL;
1,310,418✔
323
  int      code = -1;
1,310,418✔
324

325
  pRow = mndDbActionDecode(pRaw);
1,310,418✔
326
  if (pRow == NULL) {
1,310,418✔
327
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
328
    if (terrno != 0) code = terrno;
×
329
    goto _OVER;
×
330
  }
331
  pNewDb = sdbGetRowObj(pRow);
1,310,418✔
332
  if (pNewDb == NULL) {
1,310,418✔
333
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
334
    if (terrno != 0) code = terrno;
×
335
    goto _OVER;
×
336
  }
337

338
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
1,310,418✔
339
  if (pOldDb != NULL) {
1,310,418✔
340
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
341
    sdbRelease(pMnode->pSdb, pOldDb);
×
342
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
343
    if (terrno != 0) code = terrno;
×
344
    goto _OVER;
×
345
  }
346

347
  code = 0;
1,310,418✔
348
_OVER:
1,310,418✔
349
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
1,310,418✔
350
  taosMemoryFreeClear(pRow);
1,310,418✔
351
  return code;
1,310,418✔
352
}
353

354
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
1,493,609✔
355
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
1,493,609✔
356
  return 0;
1,493,609✔
357
}
358

359
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
5,804,909✔
360
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
5,804,909✔
361
  taosArrayDestroy(pDb->cfg.pRetensions);
5,804,909✔
362
  return 0;
5,804,909✔
363
}
364

365
static int32_t mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew) {
2,269,432✔
366
  mTrace("db:%s, perform update action, old row:%p new row:%p", pOld->name, pOld, pNew);
2,269,432✔
367
  taosWLockLatch(&pOld->lock);
2,269,432✔
368
  pOld->updateTime = pNew->updateTime;
2,269,432✔
369
  pOld->cfgVersion = pNew->cfgVersion;
2,269,432✔
370
  pOld->vgVersion = pNew->vgVersion;
2,269,432✔
371
  pOld->cfg.numOfVgroups = pNew->cfg.numOfVgroups;
2,269,432✔
372
  pOld->cfg.buffer = pNew->cfg.buffer;
2,269,432✔
373
  pOld->cfg.pageSize = pNew->cfg.pageSize;
2,269,432✔
374
  pOld->cfg.pages = pNew->cfg.pages;
2,269,432✔
375
  pOld->cfg.cacheLastSize = pNew->cfg.cacheLastSize;
2,269,432✔
376
  pOld->cfg.daysPerFile = pNew->cfg.daysPerFile;
2,269,432✔
377
  pOld->cfg.daysToKeep0 = pNew->cfg.daysToKeep0;
2,269,432✔
378
  pOld->cfg.daysToKeep1 = pNew->cfg.daysToKeep1;
2,269,432✔
379
  pOld->cfg.daysToKeep2 = pNew->cfg.daysToKeep2;
2,269,432✔
380
  pOld->cfg.keepTimeOffset = pNew->cfg.keepTimeOffset;
2,269,432✔
381
  pOld->cfg.walFsyncPeriod = pNew->cfg.walFsyncPeriod;
2,269,432✔
382
  pOld->cfg.walLevel = pNew->cfg.walLevel;
2,269,432✔
383
  pOld->cfg.walRetentionPeriod = pNew->cfg.walRetentionPeriod;
2,269,432✔
384
  pOld->cfg.walRetentionSize = pNew->cfg.walRetentionSize;
2,269,432✔
385
  pOld->cfg.strict = pNew->cfg.strict;
2,269,432✔
386
  pOld->cfg.cacheLast = pNew->cfg.cacheLast;
2,269,432✔
387
  pOld->cfg.replications = pNew->cfg.replications;
2,269,432✔
388
  pOld->cfg.sstTrigger = pNew->cfg.sstTrigger;
2,269,432✔
389
  pOld->cfg.minRows = pNew->cfg.minRows;
2,269,432✔
390
  pOld->cfg.maxRows = pNew->cfg.maxRows;
2,269,432✔
391
  pOld->cfg.tsdbPageSize = pNew->cfg.tsdbPageSize;
2,269,432✔
392
  pOld->cfg.ssChunkSize = pNew->cfg.ssChunkSize;
2,269,432✔
393
  pOld->cfg.ssKeepLocal = pNew->cfg.ssKeepLocal;
2,269,432✔
394
  pOld->cfg.ssCompact = pNew->cfg.ssCompact;
2,269,432✔
395
  pOld->cfg.withArbitrator = pNew->cfg.withArbitrator;
2,269,432✔
396
  pOld->cfg.compactInterval = pNew->cfg.compactInterval;
2,269,432✔
397
  pOld->cfg.compactStartTime = pNew->cfg.compactStartTime;
2,269,432✔
398
  pOld->cfg.compactEndTime = pNew->cfg.compactEndTime;
2,269,432✔
399
  pOld->cfg.compactTimeOffset = pNew->cfg.compactTimeOffset;
2,269,432✔
400
  pOld->compactStartTime = pNew->compactStartTime;
2,269,432✔
401
  pOld->tsmaVersion = pNew->tsmaVersion;
2,269,432✔
402
  pOld->cfg.isAudit = pNew->cfg.isAudit;
2,269,432✔
403
  taosWUnLockLatch(&pOld->lock);
2,269,432✔
404
  return 0;
2,269,432✔
405
}
406

407
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
8,072,791✔
408
  SSdb *pSdb = pMnode->pSdb;
8,072,791✔
409
  return sdbGetTableVer(pSdb, SDB_VGROUP);
8,072,791✔
410
}
411

412
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
79,104,434✔
413
  SSdb   *pSdb = pMnode->pSdb;
79,104,434✔
414
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
79,104,700✔
415
  if (pDb == NULL) {
79,104,700✔
416
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
2,303,626✔
417
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
2,133,723✔
418
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
169,903✔
419
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
9,954✔
420
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
159,949✔
421
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
159,949✔
422
    } else {
423
      terrno = TSDB_CODE_APP_ERROR;
×
424
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
425
    }
426
  }
427
  return pDb;
79,104,700✔
428
}
429

430
SDbObj *mndAcquireAuditDb(SMnode *pMnode) {
41,303,294✔
431
  SSdb      *pSdb = pMnode->pSdb;
41,303,294✔
432
  SDbObj    *pDb = NULL;
41,303,294✔
433
  void      *pIter = NULL;
41,303,294✔
434
  ESdbStatus objStatus = 0;
41,303,294✔
435

436
  while (1) {
437
    pIter = sdbFetchAll(pSdb, SDB_DB, pIter, (void **)&pDb, &objStatus, true);
111,344,872✔
438
    if (pIter == NULL) break;
111,344,872✔
439
    if (pDb->cfg.isAudit == 1) {
70,042,303✔
440
      sdbCancelFetch(pSdb, pIter);
725✔
441
      break;
725✔
442
    }
443

444
    sdbRelease(pSdb, pDb);
70,041,578✔
445
  }
446
  return pDb;
41,303,294✔
447
}
448

449
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
88,271,858✔
450
  SSdb *pSdb = pMnode->pSdb;
88,271,858✔
451
  sdbRelease(pSdb, pDb);
88,271,858✔
452
}
88,273,058✔
453

454
bool mndDbIsExist(SMnode *pMnode, const char *db, int64_t uid) {
141,672✔
455
  SDbObj *pDb = mndAcquireDb(pMnode, db);
141,672✔
456
  if (pDb == NULL) {
141,672✔
457
    return false;
2,964✔
458
  } else {
459
    bool result = (uid == 0) || (uid == pDb->uid);
138,708✔
460
    mndReleaseDb(pMnode, pDb);
138,708✔
461
    pDb = NULL;
138,708✔
462
    return result;
138,708✔
463
  }
464
}
465

466
int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
1,233,337✔
467
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
1,233,337✔
468
  if (pos == NULL) {
1,233,337✔
469
    return TSDB_CODE_MND_INVALID_DB;
×
470
  }
471

472
  int32_t acctId;
1,233,027✔
473
  int32_t code = taosStr2int32(dbName, &acctId);
1,233,337✔
474
  if (code != 0) {
1,233,337✔
475
    return code;
×
476
  }
477

478
  if (acctId != pUser->acctId) {
1,233,337✔
479
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
480
  }
481

482
  return 0;
1,233,337✔
483
}
484

485
int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
1,233,337✔
486
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
1,233,337✔
487

488
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
1,233,337✔
489
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
1,233,337✔
490
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
1,233,337✔
491
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
1,233,337✔
492
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
1,233,337✔
493
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
1,233,337✔
494
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
1,233,337✔
495
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
1,233,337✔
496
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
1,233,337✔
497
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
1,233,337✔
498
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
1,233,337✔
499
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
1,233,337✔
500
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
1,233,337✔
501
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
1,233,337✔
502
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
1,233,337✔
503
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
1,233,337✔
504
  if (pCfg->minRows > pCfg->maxRows) return code;
1,233,337✔
505
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
1,232,564✔
506
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
1,232,564✔
507
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
1,232,564✔
508
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
1,232,564✔
509
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
1,232,564✔
510
#ifdef TD_ENTERPRISE
511
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
1,232,564✔
512
  // if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
513
#else
514
  if (pCfg->replications != 1 && pCfg->replications != 3) return code;
515
  if (pCfg->encryptAlgorithm != TSDB_DEFAULT_ENCRYPT_ALGO) return code;
516
#endif
517

518
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
1,232,564✔
519
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
1,232,564✔
520
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
1,232,564✔
521
  if (pCfg->hashMethod != 1) return code;
1,232,564✔
522
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
1,232,564✔
523
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
524
    TAOS_RETURN(code);
×
525
  }
526
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
1,232,564✔
527
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
1,232,564✔
528
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
1,232,564✔
529
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
1,232,564✔
530
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
1,232,564✔
531
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
1,232,564✔
532
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
1,232,564✔
533
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
1,232,564✔
534
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
1,232,564✔
535
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
1,232,564✔
536
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
1,232,564✔
537

538
  if (pCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
1,232,564✔
539
  if (pCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
1,232,564✔
540
  if (pCfg->ssCompact < TSDB_MIN_SS_COMPACT || pCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
1,232,564✔
541

542
  if (pCfg->compactInterval != 0 &&
1,232,564✔
543
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
4,656✔
544
    return code;
×
545
  if (pCfg->compactStartTime != 0 &&
1,232,564✔
546
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
4,707✔
547
    return code;
×
548
  if (pCfg->compactEndTime != 0 &&
1,232,564✔
549
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
4,707✔
550
    return code;
×
551
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
1,232,564✔
552
    return code;
×
553
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
1,232,564✔
554
    return code;
×
555
  if (pCfg->isAudit < 0 || pCfg->isAudit > 1) return code;
1,232,564✔
556

557
  code = 0;
1,232,564✔
558
  TAOS_RETURN(code);
1,232,564✔
559
}
560

561
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
84,555✔
562
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
84,555✔
563
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
84,555✔
564
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
84,555✔
565
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
84,555✔
566
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
84,555✔
567
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
84,555✔
568
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
84,555✔
569
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
84,555✔
570
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
84,555✔
571
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
84,555✔
572
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
84,555✔
573
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
84,555✔
574
    return code;
×
575
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
84,555✔
576
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
84,555✔
577
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
84,555✔
578
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
84,555✔
579
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
84,555✔
580
#ifdef TD_ENTERPRISE
581
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
84,555✔
582
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
84,555✔
583
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
768✔
584
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
768✔
585
      return code;
768✔
586
    }
587
  }
588
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
83,787✔
589
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
590
    return code;
×
591
  }
592
#else
593
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
594
#endif
595

596
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
83,787✔
597
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
598
    return code;
×
599
  }
600
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
83,787✔
601
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
602
    return code;
×
603
  }
604

605
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
83,787✔
606
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
1,378✔
607
    return code;
×
608
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
83,787✔
609
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
83,787✔
610
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
83,787✔
611
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
83,019✔
612
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
83,019✔
613
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
83,019✔
614
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
83,019✔
615
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
616
    return code;
×
617
  }
618
  if (pNewCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pNewCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
83,019✔
619
  if (pNewCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pNewCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
83,019✔
620
  if (pNewCfg->ssCompact < TSDB_MIN_SS_COMPACT || pNewCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
83,019✔
621

622
  if (pNewCfg->compactInterval != 0 &&
83,019✔
623
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
7,062✔
624
    return code;
×
625
  if (pNewCfg->compactStartTime != 0 &&
83,019✔
626
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
4,758✔
627
    return code;
×
628
  if (pNewCfg->compactEndTime != 0 &&
83,019✔
629
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
4,758✔
630
    return code;
×
631
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
83,019✔
632
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
4,758✔
633
    return code;
×
634
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
83,019✔
635
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
83,019✔
636
    return code;
×
637
  if (pNewCfg->isAudit < 0 || pNewCfg->isAudit > 1) return code;
83,019✔
638

639
  code = 0;
83,019✔
640
  TAOS_RETURN(code);
83,019✔
641
}
642

643
static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
1,231,979✔
644
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
1,231,979✔
645
  if (pCfg->numOfStables < 0) pCfg->numOfStables = TSDB_DEFAULT_DB_SINGLE_STABLE;
1,231,979✔
646
  if (pCfg->buffer < 0) pCfg->buffer = TSDB_DEFAULT_BUFFER_PER_VNODE;
1,231,979✔
647
  if (pCfg->pageSize < 0) pCfg->pageSize = TSDB_DEFAULT_PAGESIZE_PER_VNODE;
1,231,979✔
648
  if (pCfg->pages < 0) pCfg->pages = TSDB_DEFAULT_PAGES_PER_VNODE;
1,231,979✔
649
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DURATION_PER_FILE;
1,231,979✔
650
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
1,231,979✔
651
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
1,231,979✔
652
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
1,231,979✔
653
  if (pCfg->keepTimeOffset < 0) pCfg->keepTimeOffset = TSDB_DEFAULT_KEEP_TIME_OFFSET;
1,231,979✔
654
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MINROWS_FBLOCK;
1,231,979✔
655
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAXROWS_FBLOCK;
1,231,979✔
656
  if (pCfg->walFsyncPeriod < 0) pCfg->walFsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
1,231,979✔
657
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
1,231,979✔
658
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
1,231,979✔
659
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
1,231,979✔
660
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA;
1,231,979✔
661
  if (pCfg->strict < 0) pCfg->strict = TSDB_DEFAULT_DB_STRICT;
1,231,979✔
662
  if (pCfg->cacheLast < 0) pCfg->cacheLast = TSDB_DEFAULT_CACHE_MODEL;
1,231,979✔
663
  if (pCfg->cacheLastSize <= 0) pCfg->cacheLastSize = TSDB_DEFAULT_CACHE_SIZE;
1,231,979✔
664
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
1,231,979✔
665
  if (pCfg->schemaless < 0) pCfg->schemaless = TSDB_DB_SCHEMALESS_OFF;
1,231,979✔
666
  if (pCfg->walRetentionPeriod < 0 && pCfg->walRetentionPeriod != -1)
1,231,979✔
667
    pCfg->walRetentionPeriod = TSDB_REPS_DEF_DB_WAL_RET_PERIOD;
×
668
  if (pCfg->walRetentionSize < 0 && pCfg->walRetentionSize != -1)
1,231,979✔
669
    pCfg->walRetentionSize = TSDB_REPS_DEF_DB_WAL_RET_SIZE;
×
670
  if (pCfg->walRollPeriod < 0) pCfg->walRollPeriod = TSDB_REPS_DEF_DB_WAL_ROLL_PERIOD;
1,231,979✔
671
  if (pCfg->walSegmentSize < 0) pCfg->walSegmentSize = TSDB_DEFAULT_DB_WAL_SEGMENT_SIZE;
1,231,979✔
672
  if (pCfg->sstTrigger <= 0) pCfg->sstTrigger = TSDB_DEFAULT_SST_TRIGGER;
1,231,979✔
673
  if (pCfg->tsdbPageSize <= 0) pCfg->tsdbPageSize = TSDB_DEFAULT_TSDB_PAGESIZE;
1,231,979✔
674
  if (pCfg->ssChunkSize <= 0) pCfg->ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
1,231,979✔
675
  if (pCfg->ssKeepLocal <= 0) pCfg->ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
1,231,979✔
676
  if (pCfg->ssCompact < 0) pCfg->ssCompact = TSDB_DEFAULT_SS_COMPACT;
1,231,979✔
677
  if (pCfg->withArbitrator < 0) pCfg->withArbitrator = TSDB_DEFAULT_DB_WITH_ARBITRATOR;
1,231,979✔
678
  if (pCfg->encryptAlgorithm < 0) pCfg->encryptAlgorithm = TSDB_DEFAULT_ENCRYPT_ALGO;
1,231,979✔
679
}
1,231,979✔
680

681
int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,227,062✔
682
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,227,062✔
683
  if (pDbRaw == NULL) return -1;
1,227,062✔
684

685
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
1,227,062✔
686
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
1,227,062✔
687
  return 0;
1,227,062✔
688
}
689

690
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,225,704✔
691
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,810,358✔
692
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
2,584,654✔
693
  }
694
  return 0;
1,225,704✔
695
}
696

697
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
698
  int32_t  code = 0;
×
699
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
700
  if (pDbRaw == NULL) {
×
701
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
702
    if (terrno != 0) code = terrno;
×
703
    TAOS_RETURN(code);
×
704
  }
705
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
706
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
707

708
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
709
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
710
    if (pVgRaw == NULL) {
×
711
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
712
      if (terrno != 0) code = terrno;
×
713
      TAOS_RETURN(code);
×
714
    }
715
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
716
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
717
  }
718

719
  if (pDb->cfg.withArbitrator) {
×
720
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
721
      SVgObj   *pVgObj = pVgroups + v;
×
722
      SArbGroup arbGroup = {0};
×
723
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
724
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
725
    }
726
  }
727

728
  TAOS_RETURN(code);
×
729
}
730

731
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,225,704✔
732
  int32_t  code = 0;
1,225,704✔
733
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,225,704✔
734
  if (pDbRaw == NULL) {
1,225,704✔
735
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
736
    if (terrno != 0) code = terrno;
×
737
    TAOS_RETURN(code);
×
738
  }
739
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
1,225,704✔
740
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
1,225,704✔
741

742
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,810,358✔
743
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
2,584,654✔
744
    if (pVgRaw == NULL) {
2,584,654✔
745
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
746
      if (terrno != 0) code = terrno;
×
747
      TAOS_RETURN(code);
×
748
    }
749
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
2,584,654✔
750
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
2,584,654✔
751
  }
752

753
  if (pDb->cfg.withArbitrator) {
1,225,704✔
754
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
10,708✔
755
      SVgObj   *pVgObj = pVgroups + v;
6,509✔
756
      SArbGroup arbGroup = {0};
6,509✔
757
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6,509✔
758
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
6,509✔
759
    }
760
  }
761

762
  TAOS_RETURN(code);
1,225,704✔
763
}
764

765
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
1,225,704✔
766
                                        SUserObj *pUserDuped) {
767
  int32_t  code = 0;
1,225,704✔
768
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,225,704✔
769
  if (pDbRaw == NULL) {
1,225,704✔
770
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
771
    if (terrno != 0) code = terrno;
×
772
    TAOS_RETURN(code);
×
773
  }
774
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
1,225,704✔
775
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
1,225,704✔
776

777
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,810,358✔
778
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
2,584,654✔
779
    if (pVgRaw == NULL) {
2,584,654✔
780
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
781
      if (terrno != 0) code = terrno;
×
782
      TAOS_RETURN(code);
×
783
    }
784
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
2,584,654✔
785
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
2,584,654✔
786
  }
787

788
  if (pDb->cfg.withArbitrator) {
1,225,704✔
789
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
10,708✔
790
      SVgObj   *pVgObj = pVgroups + v;
6,509✔
791
      SArbGroup arbGroup = {0};
6,509✔
792
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6,509✔
793
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
6,509✔
794
    }
795
  }
796

797
  if (pUserDuped) {
1,225,704✔
798
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
2,642✔
799
    if (pUserRaw == NULL) {
2,642✔
800
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
801
      if (terrno != 0) code = terrno;
×
802
      TAOS_RETURN(code);
×
803
    }
804
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
2,642✔
805
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
2,642✔
806
  }
807

808
  TAOS_RETURN(code);
1,225,704✔
809
}
810

811
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,225,704✔
812
  int32_t code = 0;
1,225,704✔
813
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
3,810,358✔
814
    SVgObj *pVgroup = pVgroups + vg;
2,584,654✔
815

816
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
5,502,941✔
817
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,918,287✔
818
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
2,918,287✔
819
    }
820
  }
821

822
  TAOS_RETURN(code);
1,225,704✔
823
}
824

825
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,225,704✔
826
  int32_t code = 0;
1,225,704✔
827
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
3,810,358✔
828
    SVgObj *pVgroup = pVgroups + vg;
2,584,654✔
829

830
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
5,502,941✔
831
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,918,287✔
832
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
2,918,287✔
833
    }
834
  }
835

836
  TAOS_RETURN(code);
1,225,704✔
837
}
838

839
static int32_t mndCreateDb(SMnode *pMnode, SRpcMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser, SArray *dnodeList) {
1,232,772✔
840
  int32_t  code = 0;
1,232,772✔
841
  SUserObj newUserObj = {0};
1,232,772✔
842
  SDbObj   dbObj = {0};
1,232,772✔
843
  (void)memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
1,232,772✔
844
  (void)memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
1,232,772✔
845
  dbObj.createdTime = taosGetTimestampMs();
1,232,772✔
846
  dbObj.updateTime = dbObj.createdTime;
1,232,772✔
847
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
1,232,772✔
848
  dbObj.cfgVersion = 1;
1,232,772✔
849
  dbObj.vgVersion = 1;
1,232,772✔
850
  dbObj.tsmaVersion = 1;
1,232,772✔
851
  (void)memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
1,232,772✔
852
  dbObj.cfg = (SDbCfg){
2,465,234✔
853
      .numOfVgroups = pCreate->numOfVgroups,
1,232,772✔
854
      .numOfStables = pCreate->numOfStables,
1,232,772✔
855
      .buffer = pCreate->buffer,
1,232,772✔
856
      .pageSize = pCreate->pageSize,
1,232,772✔
857
      .pages = pCreate->pages,
1,232,772✔
858
      .cacheLastSize = pCreate->cacheLastSize,
1,232,772✔
859
      .daysPerFile = pCreate->daysPerFile,
1,232,772✔
860
      .daysToKeep0 = pCreate->daysToKeep0,
1,232,772✔
861
      .daysToKeep1 = pCreate->daysToKeep1,
1,232,772✔
862
      .daysToKeep2 = pCreate->daysToKeep2,
1,232,772✔
863
      .keepTimeOffset = pCreate->keepTimeOffset,
1,232,772✔
864
      .minRows = pCreate->minRows,
1,232,772✔
865
      .maxRows = pCreate->maxRows,
1,232,772✔
866
      .walFsyncPeriod = pCreate->walFsyncPeriod,
1,232,772✔
867
      .walLevel = pCreate->walLevel,
1,232,772✔
868
      .precision = pCreate->precision,
1,232,772✔
869
      .compression = pCreate->compression,
1,232,772✔
870
      .replications = pCreate->replications,
1,232,772✔
871
      .strict = pCreate->strict,
1,232,772✔
872
      .cacheLast = pCreate->cacheLast,
1,232,772✔
873
      .hashMethod = 1,
874
      .schemaless = pCreate->schemaless,
1,232,772✔
875
      .walRetentionPeriod = pCreate->walRetentionPeriod,
1,232,772✔
876
      .walRetentionSize = pCreate->walRetentionSize,
1,232,772✔
877
      .walRollPeriod = pCreate->walRollPeriod,
1,232,772✔
878
      .walSegmentSize = pCreate->walSegmentSize,
1,232,772✔
879
      .sstTrigger = pCreate->sstTrigger,
1,232,772✔
880
      .hashPrefix = pCreate->hashPrefix,
1,232,772✔
881
      .hashSuffix = pCreate->hashSuffix,
1,232,772✔
882
      .ssChunkSize = pCreate->ssChunkSize,
1,232,772✔
883
      .ssKeepLocal = pCreate->ssKeepLocal,
1,232,772✔
884
      .ssCompact = pCreate->ssCompact,
1,232,772✔
885
      .tsdbPageSize = pCreate->tsdbPageSize,
1,232,772✔
886
      .withArbitrator = pCreate->withArbitrator,
1,232,772✔
887
      .encryptAlgorithm = pCreate->encryptAlgorithm,
1,232,772✔
888
      .compactInterval = pCreate->compactInterval,
1,232,772✔
889
      .compactStartTime = pCreate->compactStartTime,
1,232,772✔
890
      .compactEndTime = pCreate->compactEndTime,
1,232,772✔
891
      .compactTimeOffset = pCreate->compactTimeOffset,
1,232,772✔
892
      .isAudit = pCreate->isAudit,
1,232,772✔
893
  };
894
  if (strlen(pCreate->encryptAlgrName) > 0) {
1,232,772✔
895
    if (strncasecmp(pCreate->encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
6,349✔
896
      if (pCreate->isAudit == 1) {
4,490✔
NEW
897
        code = TSDB_CODE_AUDIT_MUST_ENCRYPT;
×
NEW
898
        mError("db:%s, failed to create, encrypt algorithm not match for audit db, %s", pCreate->db,
×
899
               pCreate->encryptAlgrName);
NEW
900
        TAOS_RETURN(code);
×
901
      }
902
      dbObj.cfg.encryptAlgorithm = 0;
4,490✔
903
    } else {
904
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, pCreate->encryptAlgrName);
1,859✔
905
      if (pEncryptAlgr != NULL) {
1,859✔
906
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
1,066✔
907
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
908
          mError("db:%s, failed to create, encrypt algorithm not match, %s, expect type %d", pCreate->db,
×
909
                 pCreate->encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
910
          TAOS_RETURN(code);
×
911
        }
912
        dbObj.cfg.encryptAlgorithm = pEncryptAlgr->id;
1,066✔
913
      } else {
914
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
793✔
915
        mError("db:%s, failed to create, encrypt algorithm not exist, %s", pCreate->db, pCreate->encryptAlgrName);
793✔
916
        TAOS_RETURN(code);
793✔
917
      }
918
    }
919
  } else {
920
    if (pCreate->isAudit == 1) {
1,226,423✔
NEW
921
      code = TSDB_CODE_AUDIT_MUST_ENCRYPT;
×
NEW
922
      mError("db:%s, failed to create, encrypt algorithm not match for audit db, %s", pCreate->db,
×
923
             pCreate->encryptAlgrName);
NEW
924
      TAOS_RETURN(code);
×
925
    }
926
  }
927
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
1,231,979✔
928
  dbObj.cfg.pRetensions = pCreate->pRetensions;
1,231,979✔
929

930
  if (pCreate->isAudit == 1) {
1,231,979✔
931
    if (dbObj.cfg.daysToKeep2 < 2628000) {
145✔
NEW
932
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
NEW
933
      mError("db:%s, failed to create, keep not match for audit db, %d", pCreate->db, dbObj.cfg.daysToKeep2);
×
NEW
934
      TAOS_RETURN(code);
×
935
    }
936
    if (dbObj.cfg.walLevel != 2) {
145✔
NEW
937
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
NEW
938
      mError("db:%s, failed to create, walLevel not match for audit db, %d", pCreate->db, dbObj.cfg.walLevel);
×
NEW
939
      TAOS_RETURN(code);
×
940
    }
941
  }
942

943
  mndSetDefaultDbCfg(&dbObj.cfg);
1,231,979✔
944

945
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
1,231,979✔
946
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, tstrerror(code));
×
947
    TAOS_RETURN(code);
×
948
  }
949

950
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
1,231,979✔
951
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, tstrerror(code));
773✔
952
    TAOS_RETURN(code);
773✔
953
  }
954

955
  if (dbObj.cfg.hashPrefix > 0) {
1,231,206✔
956
    int32_t dbLen = strlen(dbObj.name) + 1;
5,005✔
957
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
5,005✔
958
    dbObj.cfg.hashPrefix += dbLen;
5,005✔
959
  } else if (dbObj.cfg.hashPrefix < 0) {
1,226,201✔
960
    int32_t dbLen = strlen(dbObj.name) + 1;
5,087✔
961
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
5,087✔
962
    dbObj.cfg.hashPrefix -= dbLen;
5,087✔
963
  }
964

965
  SVgObj *pVgroups = NULL;
1,231,206✔
966
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
1,231,206✔
967
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
5,502✔
968
    TAOS_RETURN(code);
5,502✔
969
  }
970

971
  // add database privileges for user
972
  SUserObj *pNewUserDuped = NULL;
1,225,704✔
973
  if (!pUser->superUser) {
1,225,704✔
974
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
2,642✔
975
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
2,642✔
976
                    NULL, _OVER);
977
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
2,642✔
978
                    NULL, _OVER);
979
    pNewUserDuped = &newUserObj;
2,642✔
980
  }
981

982
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
1,225,704✔
983
  if (pTrans == NULL) {
1,225,704✔
984
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
985
    if (terrno != 0) code = terrno;
×
986
    goto _OVER;
×
987
  }
988
  // mndTransSetSerial(pTrans);
989
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
1,225,704✔
990

991
  mndTransSetDbName(pTrans, dbObj.name, NULL);
1,225,704✔
992
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
1,225,704✔
993

994
  void   *pIter = NULL;
1,225,704✔
995
  STrans *exitTrans = NULL;
1,225,704✔
996

997
  if (!pUser->superUser) {
1,225,704✔
998
    while (1) {
999
      pIter = sdbFetch(pMnode->pSdb, SDB_TRANS, pIter, (void **)&exitTrans);
2,642✔
1000
      if (pIter == NULL) break;
2,642✔
1001

1002
      if (exitTrans->conflict == TRN_CONFLICT_DB &&
×
1003
          strncmp(exitTrans->opername, "create-db", TSDB_TRANS_OPER_LEN) == 0) {
×
1004
        code = TSDB_CODE_MND_TRANS_CONFLICT;
×
1005
        sdbRelease(pMnode->pSdb, exitTrans);
×
1006
        sdbCancelFetch(pMnode->pSdb, pIter);
×
1007
        goto _OVER;
×
1008
      }
1009

1010
      sdbRelease(pMnode->pSdb, exitTrans);
×
1011
    }
1012
  }
1013

1014
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
1,225,704✔
1015
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
1,225,704✔
1016
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,225,704✔
1017
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,225,704✔
1018
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,225,704✔
1019
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
1,225,704✔
1020
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,225,704✔
1021
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,225,704✔
1022

1023
_OVER:
1,225,704✔
1024
  taosMemoryFree(pVgroups);
1,225,704✔
1025
  mndUserFreeObj(&newUserObj);
1,225,704✔
1026
  mndTransDrop(pTrans);
1,225,704✔
1027
  TAOS_RETURN(code);
1,225,704✔
1028
}
1029

1030
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
1,245,092✔
1031
  int32_t    code = 0;
1,245,092✔
1032
  SSdb      *pSdb = pMnode->pSdb;
1,245,092✔
1033
  SDnodeObj *pDnode = NULL;
1,245,092✔
1034
  void      *pIter = NULL;
1,245,092✔
1035

1036
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
1037
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
1,245,092✔
1038
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
×
1039
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
×
1040
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
1041
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
1042
           tsEncryptionKeyStat);
1043
    goto _exit;
×
1044
  }
1045

1046
  int64_t curMs = taosGetTimestampMs();
×
1047
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
×
1048
    bool online = false;
×
1049
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
×
1050
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
1051
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
1052
      mError("db:%s, failed to check encryption key:%" PRIi8
×
1053
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
1054
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
1055
             tsEncryptionKeyChksum);
1056
      sdbCancelFetch(pSdb, pIter);
×
1057
      sdbRelease(pSdb, pDnode);
×
1058
      break;
×
1059
    }
1060
    sdbRelease(pSdb, pDnode);
×
1061
  }
1062
#else
1063
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
1064
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
1065
    goto _exit;
1066
  }
1067
#endif
1068
_exit:
1,244,782✔
1069
  TAOS_RETURN(code);
1,245,092✔
1070
}
1071

1072
#ifndef TD_ENTERPRISE
1073
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
1074
  if (dnodeListStr[0] != 0) {
1075
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1076
    return terrno;
1077
  } else {
1078
    return 0;
1079
  }
1080
}
1081
#endif
1082

1083
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
1,301,727✔
1084
  SMnode      *pMnode = pReq->info.node;
1,301,727✔
1085
  int32_t      code = -1;
1,301,727✔
1086
  int32_t      lino = 0;
1,301,727✔
1087
  SDbObj      *pDb = NULL;
1,301,727✔
1088
  SUserObj    *pUser = NULL;
1,301,727✔
1089
  SCreateDbReq createReq = {0};
1,301,727✔
1090
  SArray      *dnodeList = NULL;
1,301,727✔
1091
  int64_t      tss = taosGetTimestampMs();
1,301,727✔
1092

1093
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
1,301,727✔
1094
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
1,301,727✔
1095

1096
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
1,301,727✔
1097
#ifdef WINDOWS
1098
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
1099
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
1100
    goto _OVER;
1101
  }
1102
#endif
1103
  mInfo("db:%s, start to create, vgroups:%d, isAudit:%d, daysToKeep2:%d", createReq.db, createReq.numOfVgroups, createReq.isAudit, createReq.daysToKeep2);
1,301,727✔
1104

1105
  pDb = mndAcquireDb(pMnode, createReq.db);
1,301,727✔
1106
  if (pDb != NULL) {
1,301,727✔
1107
    if (createReq.ignoreExist) {
47,382✔
1108
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
44,000✔
1109
      code = 0;
44,000✔
1110
      goto _OVER;
44,000✔
1111
    } else {
1112
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
3,382✔
1113
      goto _OVER;
3,382✔
1114
    }
1115
  } else {
1116
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
1,254,345✔
1117
      code = terrno;
8,164✔
1118
      goto _OVER;
8,164✔
1119
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
1,246,181✔
1120
      goto _OVER;
×
1121
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
1,246,181✔
1122
      // continue
1123
    } else {  // TSDB_CODE_APP_ERROR
1124
      goto _OVER;
×
1125
    }
1126
  }
1127

1128
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_DB, NULL), &lino, _OVER);
1,246,181✔
1129

1130
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
1,245,092✔
1131

1132
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
1,245,092✔
1133
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
1,245,092✔
1134

1135
  if (createReq.replications == 2) {
1,245,092✔
1136
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
5,739✔
1137
  }
1138

1139
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
1,245,092✔
1140

1141
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
1,245,092✔
1142

1143
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pUser), &lino, _OVER);
1,232,772✔
1144

1145
  if (sdbGetSize(pMnode->pSdb, SDB_MOUNT) > 0) {
1,232,772✔
1146
    TAOS_CHECK_GOTO(TSDB_CODE_MND_MOUNT_NOT_EMPTY, &lino, _OVER);
×
1147
  }
1148

1149
  if (createReq.isAudit == 1) {
1,232,772✔
1150
    SDbObj *pAuditDb = mndAcquireAuditDb(pMnode);
145✔
1151
    if (pAuditDb != NULL) {
145✔
NEW
1152
      mndReleaseDb(pMnode, pAuditDb);
×
NEW
1153
      mError("db:%s, audit db already exist, %s", createReq.db, pAuditDb->name);
×
NEW
1154
      code = TSDB_CODE_AUDIT_DB_ALREADY_EXIST;
×
NEW
1155
      return code;
×
1156
    }
1157
  }
1158

1159
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
1,232,772✔
1160
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
1,225,704✔
1161

1162
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
1,225,704✔
1163
    SName name = {0};
1,225,704✔
1164
    if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
1,225,704✔
NEW
1165
      mError("db:%s, failed to parse db name", createReq.db);
×
1166
    int64_t tse = taosGetTimestampMs();
1,225,704✔
1167
    double  duration = (double)(tse - tss);
1,225,704✔
1168
    duration = duration / 1000;
1,225,704✔
1169
    auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen, duration, 0);
1,225,704✔
1170
  }
1171

1172
_OVER:
1,301,417✔
1173
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,301,727✔
1174
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
32,023✔
1175
  }
1176

1177
  mndReleaseDb(pMnode, pDb);
1,301,727✔
1178
  mndReleaseUser(pMnode, pUser);
1,301,727✔
1179
  tFreeSCreateDbReq(&createReq);
1,301,727✔
1180
  taosArrayDestroy(dnodeList);
1,301,727✔
1181

1182
  TAOS_RETURN(code);
1,301,727✔
1183
}
1184

1185
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
108,175✔
1186
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
108,175✔
1187

1188
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
108,175✔
1189
    pDb->cfg.buffer = pAlter->buffer;
4,733✔
1190
    code = 0;
4,733✔
1191
  }
1192

1193
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
108,175✔
1194
    pDb->cfg.pages = pAlter->pages;
2,379✔
1195
    code = 0;
2,379✔
1196
  }
1197

1198
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
108,175✔
1199
    pDb->cfg.pageSize = pAlter->pageSize;
×
1200
    code = 0;
×
1201
  }
1202

1203
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
108,175✔
1204
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
1205
    code = 0;
×
1206
  }
1207

1208
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
108,175✔
1209
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
15,415✔
1210
    code = 0;
15,415✔
1211
  }
1212

1213
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
108,175✔
1214
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
14,622✔
1215
    code = 0;
14,622✔
1216
  }
1217

1218
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
108,175✔
1219
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
12,105✔
1220
    code = 0;
12,105✔
1221
  }
1222

1223
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
108,175✔
1224
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
3,868✔
1225
    code = 0;
3,868✔
1226
  }
1227

1228
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
108,175✔
1229
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
9,268✔
1230
    code = 0;
9,268✔
1231
  }
1232

1233
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
108,175✔
1234
    pDb->cfg.walLevel = pAlter->walLevel;
9,264✔
1235
    code = 0;
9,264✔
1236
  }
1237

1238
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
108,175✔
1239
#if 1
1240
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1241
#else
1242
    pDb->cfg.strict = pAlter->strict;
1243
    code = 0;
1244
#endif
1245
  }
1246

1247
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
108,175✔
1248
    pDb->cfg.cacheLast = pAlter->cacheLast;
11,888✔
1249
    code = 0;
11,888✔
1250
  }
1251

1252
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
108,175✔
1253
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
1,792✔
1254
    code = 0;
1,792✔
1255
  }
1256

1257
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
108,175✔
1258
    pDb->cfg.replications = pAlter->replications;
8,474✔
1259
    pDb->vgVersion++;
8,474✔
1260
    code = 0;
8,474✔
1261
  }
1262

1263
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
108,175✔
1264
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
1,378✔
1265
    pDb->vgVersion++;
1,378✔
1266
    code = 0;
1,378✔
1267
  }
1268

1269
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
108,175✔
1270
    pDb->cfg.minRows = pAlter->minRows;
1,536✔
1271
    pDb->vgVersion++;
1,536✔
1272
    code = 0;
1,536✔
1273
  }
1274

1275
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
108,175✔
1276
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
5,289✔
1277
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
2,492✔
1278
    pDb->vgVersion++;
2,492✔
1279
    code = 0;
2,492✔
1280
  }
1281

1282
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
108,175✔
1283
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
1,561✔
1284
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
768✔
1285
    pDb->vgVersion++;
768✔
1286
    code = 0;
768✔
1287
  }
1288

1289
  if (pAlter->ssKeepLocal >= TSDB_MIN_SS_KEEP_LOCAL && pAlter->ssKeepLocal <= TSDB_MAX_SS_KEEP_LOCAL && pAlter->ssKeepLocal != pDb->cfg.ssKeepLocal) {
108,175✔
1290
    pDb->cfg.ssKeepLocal = pAlter->ssKeepLocal;
768✔
1291
    pDb->vgVersion++;
768✔
1292
    code = 0;
768✔
1293
  }
1294

1295
  if (pAlter->ssCompact >= TSDB_MIN_SS_COMPACT && pAlter->ssCompact <= TSDB_MAX_SS_COMPACT && pAlter->ssCompact != pDb->cfg.ssCompact) {
108,175✔
1296
    pDb->cfg.ssCompact = pAlter->ssCompact;
768✔
1297
    pDb->vgVersion++;
768✔
1298
    code = 0;
768✔
1299
  }
1300

1301
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
108,175✔
1302
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
768✔
1303
    pDb->vgVersion++;
768✔
1304
    code = 0;
768✔
1305
  }
1306

1307
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
108,175✔
1308
    pDb->cfg.compactInterval = pAlter->compactInterval;
5,501✔
1309
    pDb->vgVersion++;
5,501✔
1310
    code = 0;
5,501✔
1311
  }
1312

1313
  bool compactTimeRangeChanged = false;
108,175✔
1314
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
108,175✔
1315
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
105,796✔
1316
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
105,796✔
1317
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
4,758✔
1318
    compactTimeRangeChanged = true;
4,758✔
1319
    code = 0;
4,758✔
1320
  }
1321

1322
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
108,175✔
1323
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
105,796✔
1324
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
4,758✔
1325
    compactTimeRangeChanged = true;
4,758✔
1326
    code = 0;
4,758✔
1327
  }
1328
  if(compactTimeRangeChanged) {
108,175✔
1329
    pDb->vgVersion++;
4,758✔
1330
  }
1331

1332
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
108,175✔
1333
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
9,491✔
1334
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
3,940✔
1335
    pDb->vgVersion++;
3,940✔
1336
    code = 0;
3,940✔
1337
  }
1338

1339
  if (pAlter->isAudit != pDb->cfg.isAudit) {
108,175✔
NEW
1340
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
NEW
1341
    return code;
×
1342
  }
1343

1344
  TAOS_RETURN(code);
108,175✔
1345
}
1346

1347
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
82,003✔
1348
  int32_t  code = 0;
82,003✔
1349
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
82,003✔
1350
  if (pRedoRaw == NULL) {
82,003✔
1351
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1352
    if (terrno != 0) code = terrno;
×
1353
    TAOS_RETURN(code);
×
1354
  }
1355
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
82,003✔
1356
    sdbFreeRaw(pRedoRaw);
×
1357
    TAOS_RETURN(code);
×
1358
  }
1359

1360
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
82,003✔
1361
    sdbFreeRaw(pRedoRaw);
×
1362
    TAOS_RETURN(code);
×
1363
  }
1364
  return 0;
82,003✔
1365
}
1366

1367
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
82,003✔
1368
  int32_t  code = 0;
82,003✔
1369
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
82,003✔
1370
  if (pCommitRaw == NULL) {
82,003✔
1371
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1372
    if (terrno != 0) code = terrno;
×
1373
    TAOS_RETURN(code);
×
1374
  }
1375
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
82,003✔
1376
    sdbFreeRaw(pCommitRaw);
×
1377
    TAOS_RETURN(code);
×
1378
  }
1379

1380
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
82,003✔
1381
    sdbFreeRaw(pCommitRaw);
×
1382
    TAOS_RETURN(code);
×
1383
  }
1384
  TAOS_RETURN(code);
82,003✔
1385
}
1386

1387
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
82,003✔
1388
  int32_t code = 0, lino = 0;
82,003✔
1389
  SSdb   *pSdb = pMnode->pSdb;
82,003✔
1390
  void   *pIter = NULL;
82,003✔
1391
  SVgObj *pVgroup = NULL;
82,003✔
1392
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
82,003✔
1393

1394
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
82,003✔
1395

1396
  while (1) {
1397
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
555,463✔
1398
    if (pIter == NULL) break;
555,463✔
1399

1400
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
474,206✔
1401
      SVgObj newVgroup = {0};
210,432✔
1402
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
210,432✔
1403
                      _err);
1404

1405
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
209,686✔
1406
        if (pNewDb->cfg.withArbitrator) {
×
1407
          SArbGroup arbGroup = {0};
×
1408
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1409
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1410
        } else {
1411
          SArbGroup arbGroup = {0};
×
1412
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
1413
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1414
        }
1415
      }
1416
    }
1417

1418
    sdbRelease(pSdb, pVgroup);
473,460✔
1419
  }
1420

1421
  taosArrayDestroy(pArray);
81,257✔
1422
  TAOS_RETURN(code);
81,257✔
1423

1424
_err:
746✔
1425
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
746✔
1426

1427
  sdbCancelFetch(pSdb, pIter);
746✔
1428
  sdbRelease(pSdb, pVgroup);
746✔
1429
  taosArrayDestroy(pArray);
746✔
1430
  TAOS_RETURN(code);
746✔
1431
}
1432

1433
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
83,019✔
1434
  int32_t code = -1;
83,019✔
1435
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
83,019✔
1436
  if (pTrans == NULL) {
83,019✔
1437
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1438
    if (terrno != 0) code = terrno;
×
1439
    TAOS_RETURN(code);
×
1440
  }
1441
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
83,019✔
1442
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
83,019✔
1443

1444
  mndTransSetDbName(pTrans, pOld->name, NULL);
83,019✔
1445
  mndTransSetGroupParallel(pTrans);
83,019✔
1446
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
83,019✔
1447
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
82,257✔
1448
  TAOS_CHECK_GOTO(mndTransCheckConflictWithRetention(pMnode, pTrans), NULL, _OVER);
82,003✔
1449

1450
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
82,003✔
1451
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
82,003✔
1452
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
82,003✔
1453

1454
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
81,257✔
1455
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
81,257✔
1456
  code = 0;
81,257✔
1457

1458
_OVER:
83,019✔
1459
  mndTransDrop(pTrans);
83,019✔
1460
  TAOS_RETURN(code);
83,019✔
1461
}
1462

1463
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
113,454✔
1464
  SMnode     *pMnode = pReq->info.node;
113,454✔
1465
  int32_t     code = -1;
113,454✔
1466
  SDbObj     *pDb = NULL;
113,454✔
1467
  SAlterDbReq alterReq = {0};
113,454✔
1468
  SDbObj      dbObj = {0};
113,454✔
1469
  int64_t     tss = taosGetTimestampMs();
113,454✔
1470

1471
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
113,454✔
1472

1473
  mInfo("db:%s, start to alter, encryt_algr:%s", alterReq.db, alterReq.encryptAlgrName);
113,454✔
1474

1475
  pDb = mndAcquireDb(pMnode, alterReq.db);
113,454✔
1476
  if (pDb == NULL) {
113,454✔
1477
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
2,560✔
1478
    if (terrno != 0) code = terrno;
2,560✔
1479
    goto _OVER;
2,560✔
1480
  }
1481

1482
  if (pDb->cfg.isAudit == 1) {
110,894✔
NEW
1483
    if (alterReq.daysToKeep2 < 2628000) {
×
NEW
1484
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
NEW
1485
      mError("db:%s, failed to alter, keep not match for audit db, %d", alterReq.db, alterReq.daysToKeep2);
×
NEW
1486
      TAOS_RETURN(code);
×
1487
    }
NEW
1488
    if (alterReq.walLevel != 2) {
×
NEW
1489
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
NEW
1490
      mError("db:%s, failed to alter, walLevel not match for audit db, %d", alterReq.db, alterReq.walLevel);
×
NEW
1491
      TAOS_RETURN(code);
×
1492
    }
NEW
1493
    if (alterReq.isAudit == 0) {
×
NEW
1494
      code = TSDB_CODE_AUDIT_DB_NOT_ALLOW_CHANGE;
×
NEW
1495
      mError("db:%s, failed to alter, is not allowed to change audit db, %d", alterReq.db, alterReq.isAudit);
×
NEW
1496
      TAOS_RETURN(code);
×
1497
    }
1498
  }
1499

1500
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_ALTER_DB, pDb), NULL, _OVER);
110,894✔
1501

1502
  if (alterReq.replications == 2) {
110,529✔
1503
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
768✔
1504
  }
1505

1506
  int32_t numOfTopics = 0;
110,529✔
1507
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
110,529✔
1508

1509
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
110,529✔
1510
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
1511
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1512
           numOfTopics);
1513
    goto _OVER;
×
1514
  }
1515

1516
  if (pDb->cfg.isMount) {
110,529✔
1517
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
1518
    goto _OVER;
×
1519
  }
1520

1521
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
110,529✔
1522
  if (dbObj.cfg.pRetensions != NULL) {
110,529✔
1523
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
1524
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1525
  }
1526

1527
  if (strlen(alterReq.encryptAlgrName) > 0) {
110,529✔
1528
    mInfo("db:%s, check encryt algr:%s", alterReq.db, alterReq.encryptAlgrName);
3,147✔
1529

1530
    if (strncasecmp(alterReq.encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
3,147✔
1531
      if (pDb->cfg.encryptAlgorithm != 0) {
793✔
1532
        code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
×
1533
        mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
×
1534
        goto _OVER;
×
1535
      }
1536
    } else {
1537
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, alterReq.encryptAlgrName);
2,354✔
1538
      if (pEncryptAlgr != NULL) {
2,354✔
1539
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
1,561✔
1540
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
1541
          mError("db:%s, failed to alter, encrypt algorithm not match, %s, expect type %d", alterReq.db,
×
1542
                 alterReq.encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
1543
          goto _OVER;
×
1544
        }
1545
        if (pEncryptAlgr->id != pDb->cfg.encryptAlgorithm) {
1,561✔
1546
          code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
1,561✔
1547
          mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
1,561✔
1548
          goto _OVER;
1,561✔
1549
        }
1550
      } else {
1551
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
793✔
1552
        mError("db:%s, failed to alter, encrypt algorithm not exist, %s", alterReq.db, alterReq.encryptAlgrName);
793✔
1553
        goto _OVER;
793✔
1554
      }
1555
    }
1556
  }
1557

1558
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
108,175✔
1559
  if (code != 0) {
108,175✔
1560
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
23,620✔
1561
    goto _OVER;
23,620✔
1562
  }
1563

1564
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
84,555✔
1565

1566
  dbObj.cfgVersion++;
83,019✔
1567
  dbObj.updateTime = taosGetTimestampMs();
83,019✔
1568
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
83,019✔
1569

1570
  if (dbObj.cfg.replications != pDb->cfg.replications) {
83,019✔
1571
    // return quickly, operation executed asynchronously
1572
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
7,706✔
1573
  } else {
1574
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
75,313✔
1575
  }
1576

1577
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
83,019✔
1578
    SName name = {0};
83,019✔
1579
    if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
83,019✔
NEW
1580
      mError("db:%s, failed to parse db name", alterReq.db);
×
1581

1582
    int64_t tse = taosGetTimestampMs();
83,019✔
1583
    double  duration = (double)(tse - tss);
83,019✔
1584
    duration = duration / 1000;
83,019✔
1585
    auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen, duration, 0);
83,019✔
1586
  }
1587

1588
_OVER:
113,454✔
1589
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
113,454✔
1590
    if (terrno != 0) code = terrno;
8,577✔
1591
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
8,577✔
1592
  }
1593

1594
  mndReleaseDb(pMnode, pDb);
113,454✔
1595
  taosArrayDestroy(dbObj.cfg.pRetensions);
113,454✔
1596
  tFreeSAlterDbReq(&alterReq);
113,454✔
1597

1598
  TAOS_RETURN(code);
113,454✔
1599
}
1600

1601
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb, char *algorithmsId) {
1,501,395✔
1602
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
1,501,395✔
1603
  cfgRsp->dbId = pDb->uid;
1,501,395✔
1604
  cfgRsp->cfgVersion = pDb->cfgVersion;
1,501,188✔
1605
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
1,501,395✔
1606
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
1,501,395✔
1607
  cfgRsp->buffer = pDb->cfg.buffer;
1,501,395✔
1608
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
1,501,188✔
1609
  cfgRsp->pageSize = pDb->cfg.pageSize;
1,501,395✔
1610
  cfgRsp->pages = pDb->cfg.pages;
1,501,395✔
1611
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
1,501,188✔
1612
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
1,501,188✔
1613
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
1,501,395✔
1614
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
1,501,395✔
1615
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
1,501,395✔
1616
  cfgRsp->minRows = pDb->cfg.minRows;
1,501,395✔
1617
  cfgRsp->maxRows = pDb->cfg.maxRows;
1,501,395✔
1618
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
1,501,188✔
1619
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
1,501,395✔
1620
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
1,501,188✔
1621
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
1,501,395✔
1622
  cfgRsp->walLevel = pDb->cfg.walLevel;
1,501,395✔
1623
  cfgRsp->precision = pDb->cfg.precision;
1,501,395✔
1624
  cfgRsp->compression = pDb->cfg.compression;
1,501,188✔
1625
  cfgRsp->replications = pDb->cfg.replications;
1,501,395✔
1626
  cfgRsp->strict = pDb->cfg.strict;
1,501,395✔
1627
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
1,501,395✔
1628
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
1,501,188✔
1629
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
1,501,188✔
1630
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
1,501,188✔
1631
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
1,501,395✔
1632
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
1,501,395✔
1633
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
1,501,188✔
1634
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
1,501,188✔
1635
  cfgRsp->schemaless = pDb->cfg.schemaless;
1,501,188✔
1636
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
1,501,395✔
1637
  cfgRsp->ssChunkSize = pDb->cfg.ssChunkSize;
1,501,395✔
1638
  cfgRsp->ssKeepLocal = pDb->cfg.ssKeepLocal;
1,501,188✔
1639
  cfgRsp->ssCompact = pDb->cfg.ssCompact;
1,501,395✔
1640
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
1,501,395✔
1641
  // cfgRsp->encryptAlgr = pDb->cfg.encryptAlgorithm;
1642
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
1,501,188✔
1643
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
1,501,395✔
1644
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
1,501,188✔
1645
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
1,501,395✔
1646
  cfgRsp->flags = pDb->cfg.flags;
1,501,188✔
1647
  tstrncpy(cfgRsp->algorithmsId, algorithmsId, sizeof(cfgRsp->algorithmsId));
1,501,395✔
1648
  cfgRsp->isAudit = pDb->cfg.isAudit;
1,501,395✔
1649
}
1,501,188✔
1650

1651
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
1,644,366✔
1652
  SMnode   *pMnode = pReq->info.node;
1,644,366✔
1653
  int32_t   code = -1;
1,644,366✔
1654
  SDbObj   *pDb = NULL;
1,644,366✔
1655
  SDbCfgReq cfgReq = {0};
1,644,366✔
1656
  SDbCfgRsp cfgRsp = {0};
1,644,159✔
1657

1658
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
1,644,159✔
1659

1660
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
1,644,366✔
1661
    pDb = mndAcquireDb(pMnode, cfgReq.db);
1,644,366✔
1662
    if (pDb == NULL) {
1,644,366✔
1663
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
294,876✔
1664
      if (terrno != 0) code = terrno;
294,876✔
1665
      goto _OVER;
294,876✔
1666
    }
1667

1668
    char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
1,349,490✔
1669
    mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
1,349,490✔
1670
    mndDumpDbCfgInfo(&cfgRsp, pDb, algrId);
1,349,490✔
1671
  }
1672

1673
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
1,349,490✔
1674
  void   *pRsp = rpcMallocCont(contLen);
1,349,490✔
1675
  if (pRsp == NULL) {
1,349,283✔
1676
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1677
    if (terrno != 0) code = terrno;
×
1678
    goto _OVER;
×
1679
  }
1680

1681
  int32_t ret = 0;
1,349,283✔
1682
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
1,349,283✔
1683
    code = ret;
×
1684
    goto _OVER;
×
1685
  }
1686

1687
  pReq->info.rsp = pRsp;
1,349,490✔
1688
  pReq->info.rspLen = contLen;
1,349,490✔
1689

1690
  code = 0;
1,349,490✔
1691

1692
_OVER:
1,644,366✔
1693

1694
  tFreeSDbCfgRsp(&cfgRsp);
1,644,366✔
1695

1696
  if (code != 0) {
1,644,366✔
1697
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
294,876✔
1698
  }
1699

1700
  mndReleaseDb(pMnode, pDb);
1,644,366✔
1701

1702
  TAOS_RETURN(code);
1,644,366✔
1703
}
1704

1705
int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
723,006✔
1706
  int32_t  code = 0;
723,006✔
1707
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
723,006✔
1708
  if (pRedoRaw == NULL) {
723,006✔
1709
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1710
    if (terrno != 0) code = terrno;
×
1711
    return -1;
×
1712
  }
1713
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
723,006✔
1714
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
723,006✔
1715

1716
  SSdb *pSdb = pMnode->pSdb;
723,006✔
1717
  void *pIter = NULL;
723,006✔
1718

1719
  while (1) {
4,620✔
1720
    SArbGroup *pArbGroup = NULL;
727,626✔
1721
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
727,626✔
1722
    if (pIter == NULL) break;
727,626✔
1723

1724
    if (pArbGroup->dbUid == pDb->uid) {
4,620✔
1725
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
4,620✔
1726
        sdbCancelFetch(pSdb, pIter);
×
1727
        sdbRelease(pSdb, pArbGroup);
×
1728
        TAOS_RETURN(code);
×
1729
      }
1730
    }
1731

1732
    sdbRelease(pSdb, pArbGroup);
4,620✔
1733
  }
1734

1735
  TAOS_RETURN(code);
723,006✔
1736
}
1737

1738
int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
723,006✔
1739
  int32_t  code = 0;
723,006✔
1740
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
723,006✔
1741
  if (pCommitRaw == NULL) {
723,006✔
1742
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1743
    if (terrno != 0) code = terrno;
×
1744
    return -1;
×
1745
  }
1746
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
723,006✔
1747
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
723,006✔
1748

1749
  SSdb *pSdb = pMnode->pSdb;
723,006✔
1750
  void *pIter = NULL;
723,006✔
1751

1752
  while (1) {
4,620✔
1753
    SArbGroup *pArbGroup = NULL;
727,626✔
1754
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
727,626✔
1755
    if (pIter == NULL) break;
727,626✔
1756

1757
    if (pArbGroup->dbUid == pDb->uid) {
4,620✔
1758
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
4,620✔
1759
        sdbCancelFetch(pSdb, pIter);
×
1760
        sdbRelease(pSdb, pArbGroup);
×
1761
        TAOS_RETURN(code);
×
1762
      }
1763
    }
1764

1765
    sdbRelease(pSdb, pArbGroup);
4,620✔
1766
  }
1767

1768
  while (1) {
3,197,550✔
1769
    SVgObj *pVgroup = NULL;
3,920,556✔
1770
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,920,556✔
1771
    if (pIter == NULL) break;
3,920,556✔
1772

1773
    if (pVgroup->dbUid == pDb->uid) {
3,197,550✔
1774
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
1,472,171✔
1775
      if (pVgRaw == NULL) {
1,472,171✔
1776
        sdbCancelFetch(pSdb, pIter);
×
1777
        sdbRelease(pSdb, pVgroup);
×
1778
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1779
        if (terrno != 0) code = terrno;
×
1780
        TAOS_RETURN(code);
×
1781
      }
1782
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
1,472,171✔
1783
        sdbCancelFetch(pSdb, pIter);
×
1784
        sdbRelease(pSdb, pVgroup);
×
1785
        TAOS_RETURN(code);
×
1786
      }
1787
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
1,472,171✔
1788
        sdbCancelFetch(pSdb, pIter);
×
1789
        sdbRelease(pSdb, pVgroup);
×
1790
        TAOS_RETURN(code);
×
1791
      }
1792
    }
1793

1794
    sdbRelease(pSdb, pVgroup);
3,197,550✔
1795
  }
1796

1797
  while (1) {
943,373✔
1798
    SStbObj *pStb = NULL;
1,666,379✔
1799
    ESdbStatus status;
1,666,155✔
1800
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
1,666,379✔
1801
    if (pIter == NULL) break;
1,666,379✔
1802

1803
    if (pStb->dbUid == pDb->uid) {
943,373✔
1804
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
537,925✔
1805
      if (pStbRaw == NULL) {
537,925✔
1806
        sdbCancelFetch(pSdb, pIter);
×
1807
        sdbRelease(pSdb, pStbRaw);
×
1808
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1809
        if (terrno != 0) code = terrno;
×
1810
        return -1;
×
1811
      }
1812
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
537,925✔
1813
        sdbCancelFetch(pSdb, pIter);
×
1814
        sdbRelease(pSdb, pStbRaw);
×
1815
        return -1;
×
1816
      }
1817
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
537,925✔
1818
        sdbCancelFetch(pSdb, pIter);
×
1819
        sdbRelease(pSdb, pStbRaw);
×
1820
        return -1;
×
1821
      }
1822
    }
1823

1824
    sdbRelease(pSdb, pStb);
943,373✔
1825
  }
1826

1827
  TAOS_RETURN(code);
723,006✔
1828
}
1829

1830
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
1,472,171✔
1831
  int32_t code = 0;
1,472,171✔
1832
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
2,999,182✔
1833
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
1,527,011✔
1834
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
1,527,011✔
1835
  }
1836

1837
  TAOS_RETURN(code);
1,472,171✔
1838
}
1839

1840
int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
723,006✔
1841
  int32_t code = 0;
723,006✔
1842
  SSdb   *pSdb = pMnode->pSdb;
723,006✔
1843
  void   *pIter = NULL;
723,006✔
1844

1845
  while (1) {
3,197,550✔
1846
    SVgObj *pVgroup = NULL;
3,920,556✔
1847
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,920,556✔
1848
    if (pIter == NULL) break;
3,920,556✔
1849

1850
    if (pVgroup->dbUid == pDb->uid) {
3,197,550✔
1851
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
1,472,171✔
1852
        sdbCancelFetch(pSdb, pIter);
×
1853
        sdbRelease(pSdb, pVgroup);
×
1854
        TAOS_RETURN(code);
×
1855
      }
1856
    }
1857

1858
    sdbRelease(pSdb, pVgroup);
3,197,550✔
1859
  }
1860

1861
  TAOS_RETURN(code);
723,006✔
1862
}
1863

1864
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
1,153,417✔
1865
  int32_t    code = 0;
1,153,417✔
1866
  SDropDbRsp dropRsp = {0};
1,153,417✔
1867
  if (pDb != NULL) {
1,153,417✔
1868
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
721,658✔
1869
    dropRsp.uid = pDb->uid;
721,658✔
1870
  }
1871

1872
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
1,153,417✔
1873
  void   *pRsp = NULL;
1,153,417✔
1874
  if (useRpcMalloc) {
1,153,417✔
1875
    pRsp = rpcMallocCont(rspLen);
431,759✔
1876
  } else {
1877
    pRsp = taosMemoryMalloc(rspLen);
721,658✔
1878
  }
1879

1880
  if (pRsp == NULL) {
1,153,417✔
1881
    code = terrno;
×
1882
    TAOS_RETURN(code);
×
1883
  }
1884

1885
  int32_t ret = 0;
1,153,417✔
1886
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
1,153,417✔
1887
  *pRspLen = rspLen;
1,153,417✔
1888
  *ppRsp = pRsp;
1,153,417✔
1889
  TAOS_RETURN(code);
1,153,417✔
1890
}
1891

1892
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
721,776✔
1893
  int32_t code = -1;
721,776✔
1894

1895
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
721,776✔
1896
  if (pTrans == NULL) {
721,776✔
1897
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1898
    if (terrno != 0) code = terrno;
×
1899
    goto _OVER;
×
1900
  }
1901

1902
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
721,776✔
1903

1904
  mndTransSetDbName(pTrans, pDb->name, NULL);
721,776✔
1905
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
721,776✔
1906

1907
  if (mndTopicExistsForDb(pMnode, pDb)) {
721,776✔
1908
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
118✔
1909
    goto _OVER;
118✔
1910
  }
1911

1912
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1913
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1914
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1915
#ifdef TD_ENTERPRISE
1916
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1917
#endif
1918
  TAOS_CHECK_GOTO(mndDropTSMAsByDb(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1919
  TAOS_CHECK_GOTO(mndDropRsmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1920
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1921
  //TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
1922
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
721,658✔
1923
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb, NULL), NULL, _OVER);
721,658✔
1924

1925
  int32_t rspLen = 0;
721,658✔
1926
  void   *pRsp = NULL;
721,658✔
1927
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
721,658✔
1928
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
721,658✔
1929

1930
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
721,658✔
1931
  code = 0;
721,658✔
1932

1933
_OVER:
721,776✔
1934
  mndTransDrop(pTrans);
721,776✔
1935
  TAOS_RETURN(code);
721,776✔
1936
}
1937

1938
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
1,163,879✔
1939
  SMnode    *pMnode = pReq->info.node;
1,163,879✔
1940
  int32_t    code = -1;
1,163,879✔
1941
  SDbObj    *pDb = NULL;
1,163,879✔
1942
  SDropDbReq dropReq = {0};
1,163,879✔
1943
  int64_t    tss = taosGetTimestampMs();
1,163,879✔
1944

1945
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
1,163,879✔
1946

1947
  mInfo("db:%s, start to drop", dropReq.db);
1,163,879✔
1948

1949
  pDb = mndAcquireDb(pMnode, dropReq.db);
1,163,879✔
1950
  if (pDb == NULL) {
1,163,879✔
1951
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
436,066✔
1952
    if (terrno != 0) code = terrno;
436,066✔
1953
    if (dropReq.ignoreNotExists) {
436,066✔
1954
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
431,759✔
1955
    }
1956
    goto _OVER;
436,066✔
1957
  }
1958

1959
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_DB, pDb), NULL, _OVER);
727,813✔
1960

1961
  if(pDb->cfg.isMount) {
726,024✔
1962
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
1,348✔
1963
    goto _OVER;
1,348✔
1964
  }
1965

1966
  SSdb *pSdb = pMnode->pSdb;
724,676✔
1967
  void *pIter = NULL;
724,676✔
1968

1969
  while (1) {
3,216,268✔
1970
    SVgObj *pVgroup = NULL;
3,940,944✔
1971
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,940,944✔
1972
    if (pIter == NULL) break;
3,940,944✔
1973

1974
    if (pVgroup->dbUid == pDb->uid) {
3,216,268✔
1975
      bool isFound = false;
1,472,493✔
1976
      for (int32_t i = 0; i < pVgroup->replica; i++) {
2,999,826✔
1977
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
1,527,333✔
UNCOV
1978
          isFound = true;
×
UNCOV
1979
          break;
×
1980
        }
1981
      }
1982
      if (!isFound) {
1,472,493✔
1983
        sdbRelease(pSdb, pVgroup);
1,472,493✔
1984
        continue;
1,472,493✔
1985
      }
UNCOV
1986
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
×
UNCOV
1987
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
1988
      sdbRelease(pSdb, pVgroup);
×
UNCOV
1989
      goto _OVER;
×
1990
    }
1991

1992
    sdbRelease(pSdb, pVgroup);
1,743,775✔
1993
  }
1994

1995
  bool dbStream = false;
724,676✔
1996
  bool vtableStream = false;
724,676✔
1997
  mstCheckDbInUse(pMnode, dropReq.db, &dbStream, &vtableStream, true);
724,676✔
1998
  if (dbStream) {
724,676✔
1999
    code = TSDB_CODE_MND_STREAM_DB_IN_USE;
2,490✔
2000
    mError("db:%s used by streams, drop db not allowed", dropReq.db);
2,490✔
2001
    goto _OVER;
2,490✔
2002
  }
2003

2004
  if (vtableStream && !dropReq.force) {
722,186✔
2005
    code = TSDB_CODE_MND_STREAM_VTABLE_EXITS;
410✔
2006
    mError("db:%s, vtable stream exists, drop db not allowed", dropReq.db);
410✔
2007
    goto _OVER;
410✔
2008
  }
2009

2010
  code = mndDropDb(pMnode, pReq, pDb);
721,776✔
2011
  if (code == TSDB_CODE_SUCCESS) {
721,776✔
2012
    code = TSDB_CODE_ACTION_IN_PROGRESS;
721,658✔
2013
  }
2014

2015
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
721,776✔
2016
    SName name = {0};
721,776✔
2017
    if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
721,776✔
NEW
2018
      mError("db:%s, failed to parse db name", dropReq.db);
×
2019

2020
    int64_t tse = taosGetTimestampMs();
721,776✔
2021
    double  duration = (double)(tse - tss);
721,776✔
2022
    duration = duration / 1000;
721,776✔
2023
    auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen, duration, 0);
721,776✔
2024
  }
2025

2026
_OVER:
1,163,672✔
2027
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,163,879✔
2028
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
10,462✔
2029
  }
2030

2031
  mndReleaseDb(pMnode, pDb);
1,163,879✔
2032
  tFreeSDropDbReq(&dropReq);
1,163,879✔
2033
  TAOS_RETURN(code);
1,163,879✔
2034
}
2035

2036
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
23,922,991✔
2037
  int32_t numOfTables = 0;
23,922,991✔
2038
  int32_t vindex = 0;
23,922,991✔
2039
  SSdb   *pSdb = pMnode->pSdb;
23,922,991✔
2040

2041
  void *pIter = NULL;
23,922,991✔
2042
  while (vindex < pDb->cfg.numOfVgroups) {
92,188,512✔
2043
    SVgObj *pVgroup = NULL;
68,265,521✔
2044
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
68,265,521✔
2045
    if (pIter == NULL) break;
68,265,521✔
2046

2047
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
68,265,521✔
2048
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
48,340,368✔
2049
      vindex++;
48,340,102✔
2050
    }
2051

2052
    sdbRelease(pSdb, pVgroup);
68,265,255✔
2053
  }
2054

2055
  sdbCancelFetch(pSdb, pIter);
23,922,991✔
2056
  return numOfTables;
23,922,991✔
2057
}
2058

2059
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
26,674,969✔
2060
  int32_t vindex = 0;
26,674,969✔
2061
  SSdb   *pSdb = pMnode->pSdb;
26,674,969✔
2062

2063
  void *pIter = NULL;
26,674,969✔
2064
  while (1) {
50,601,750✔
2065
    SVgObj *pVgroup = NULL;
77,276,719✔
2066
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
77,276,719✔
2067
    if (pIter == NULL) break;
77,276,719✔
2068

2069
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
70,724,116✔
2070
      SVgroupInfo vgInfo = {0};
58,633,297✔
2071
      vgInfo.vgId = pVgroup->vgId;
58,633,297✔
2072
      vgInfo.hashBegin = pVgroup->hashBegin;
58,633,297✔
2073
      vgInfo.hashEnd = pVgroup->hashEnd;
58,633,297✔
2074
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
58,633,297✔
2075
      vgInfo.epSet.numOfEps = pVgroup->replica;
58,633,297✔
2076
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
122,143,398✔
2077
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
63,511,058✔
2078
        SEp       *pEp = &vgInfo.epSet.eps[gid];
63,511,058✔
2079
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
63,511,058✔
2080
        if (pDnode != NULL) {
63,511,058✔
2081
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
63,473,040✔
2082
          pEp->port = pDnode->port;
63,472,721✔
2083
        }
2084
        mndReleaseDnode(pMnode, pDnode);
63,510,430✔
2085
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
63,511,058✔
2086
          vgInfo.epSet.inUse = gid;
58,131,606✔
2087
        }
2088
      }
2089
      vindex++;
58,633,297✔
2090
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
58,633,297✔
2091
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
2092
      }
2093
    }
2094

2095
    sdbRelease(pSdb, pVgroup);
70,724,116✔
2096

2097
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
70,724,116✔
2098
      sdbCancelFetch(pSdb, pIter);
20,122,366✔
2099
      break;
20,122,366✔
2100
    }
2101
  }
2102
}
26,674,969✔
2103

2104
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
20,679,693✔
2105
  int32_t code = 0;
20,679,693✔
2106
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
20,679,693✔
2107
  if (pRsp->pVgroupInfos == NULL) {
20,679,693✔
2108
    code = terrno;
×
2109
    TAOS_RETURN(code);
×
2110
  }
2111

2112
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
20,679,693✔
2113

2114
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
20,679,693✔
2115
      pReq->stateTs < pDb->stateTs) {
1,456,621✔
2116
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
19,362,993✔
2117
  }
2118

2119
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
20,679,693✔
2120
  pRsp->uid = pDb->uid;
20,679,693✔
2121
  pRsp->vgVersion = pDb->vgVersion;
20,679,693✔
2122
  pRsp->stateTs = pDb->stateTs;
20,679,693✔
2123
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
20,679,693✔
2124
  pRsp->hashMethod = pDb->cfg.hashMethod;
20,679,693✔
2125
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
20,679,693✔
2126
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
20,679,693✔
2127
  pRsp->flags = pDb->cfg.flags;
20,679,693✔
2128
  TAOS_RETURN(code);
20,679,693✔
2129
}
2130

2131
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
27,153,410✔
2132
  SMnode   *pMnode = pReq->info.node;
27,153,410✔
2133
  int32_t   code = -1;
27,153,410✔
2134
  SDbObj   *pDb = NULL;
27,153,410✔
2135
  SUseDbReq usedbReq = {0};
27,153,410✔
2136
  SUseDbRsp usedbRsp = {0};
27,152,642✔
2137

2138
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
27,153,410✔
2139

2140
  char *p = strchr(usedbReq.db, '.');
27,152,908✔
2141
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
27,152,908✔
2142
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
6,328,479✔
2143
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
6,328,479✔
2144
    if (usedbReq.vgVersion < vgVersion) {
6,327,445✔
2145
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
6,314,712✔
2146
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
6,314,712✔
2147

2148
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
6,314,712✔
2149
      usedbRsp.vgVersion = vgVersion++;
6,314,712✔
2150
    } else {
2151
      usedbRsp.vgVersion = usedbReq.vgVersion;
12,733✔
2152
    }
2153
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
6,327,445✔
2154
    code = 0;
6,327,445✔
2155
  } else {
2156
    pDb = mndAcquireDb(pMnode, usedbReq.db);
20,824,429✔
2157
    if (pDb == NULL) {
20,826,231✔
2158
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
131,985✔
2159
      usedbRsp.uid = usedbReq.dbId;
131,985✔
2160
      usedbRsp.vgVersion = usedbReq.vgVersion;
131,985✔
2161
      usedbRsp.errCode = terrno;
131,985✔
2162

2163
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
131,985✔
2164
      if (terrno != 0) code = terrno;
131,985✔
2165
      goto _OVER;
131,985✔
2166
    } else {
2167
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_USE_DB, pDb), NULL, _OVER);
20,694,246✔
2168

2169
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
20,679,693✔
2170

2171
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
20,679,693✔
2172
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
2173
      code = 0;
20,679,693✔
2174
    }
2175
  }
2176

2177
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
27,007,138✔
2178
  void   *pRsp = rpcMallocCont(contLen);
27,007,138✔
2179
  if (pRsp == NULL) {
27,007,138✔
2180
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2181
    code = -1;
×
2182
    goto _OVER;
×
2183
  }
2184

2185
  int32_t ret = 0;
27,007,138✔
2186
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
27,007,138✔
2187
    code = ret;
×
2188
    goto _OVER;
×
2189
  }
2190

2191
  pReq->info.rsp = pRsp;
27,007,138✔
2192
  pReq->info.rspLen = contLen;
27,007,138✔
2193

2194
_OVER:
27,153,676✔
2195
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
27,153,676✔
2196
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
146,538✔
2197
  }
2198

2199
  mndReleaseDb(pMnode, pDb);
27,153,676✔
2200
  tFreeSUsedbRsp(&usedbRsp);
27,153,676✔
2201

2202
  TAOS_RETURN(code);
27,153,676✔
2203
}
2204

2205
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
4,092,520✔
2206
  int32_t       code = 0;
4,092,520✔
2207
  SDbHbBatchRsp batchRsp = {0};
4,092,520✔
2208
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
4,092,520✔
2209
  if (batchRsp.pArray == NULL) {
4,092,520✔
2210
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2211
    TAOS_RETURN(code);
×
2212
  }
2213

2214
  for (int32_t i = 0; i < numOfDbs; ++i) {
9,104,467✔
2215
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
5,011,947✔
2216
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
5,011,947✔
2217
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
5,011,947✔
2218
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
5,011,947✔
2219
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
5,011,947✔
2220
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
5,011,947✔
2221
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
5,011,947✔
2222

2223
    SDbHbRsp rsp = {0};
5,011,947✔
2224
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
5,011,947✔
2225
    rsp.dbId = pDbCacheInfo->dbId;
5,011,947✔
2226

2227
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
5,011,947✔
2228
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
4,813,603✔
2229
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
1,745,346✔
2230
      if (pDbCacheInfo->vgVersion >= vgVersion) {
1,745,346✔
2231
        continue;
1,507,835✔
2232
      }
2233

2234
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
237,891✔
2235
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
237,891✔
2236
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
237,891✔
2237

2238
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
237,891✔
2239
      rsp.useDbRsp->vgVersion = vgVersion++;
237,891✔
2240

2241
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
237,891✔
2242

2243
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
475,782✔
2244
        if (terrno != 0) code = terrno;
×
2245
        return code;
×
2246
      }
2247

2248
      continue;
237,891✔
2249
    }
2250

2251
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
3,266,601✔
2252
    if (pDb == NULL) {
3,266,601✔
2253
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
23,303✔
2254
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
23,303✔
2255
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
23,303✔
2256
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
23,303✔
2257
      rsp.useDbRsp->vgVersion = -1;
23,303✔
2258
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
46,606✔
2259
        if (terrno != 0) code = terrno;
×
2260
        return code;
×
2261
      }
2262
      continue;
23,303✔
2263
    }
2264

2265
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
3,243,298✔
2266

2267
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
3,243,298✔
2268
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
3,065,929✔
2269
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
2,410,815✔
2270
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
2,276,352✔
2271
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2272
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2273
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2274
      mndReleaseDb(pMnode, pDb);
2,276,352✔
2275
      continue;
2,276,352✔
2276
    } else {
2277
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
966,946✔
2278
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2279
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2280
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2281
    }
2282

2283
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
966,946✔
2284
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
151,905✔
2285
      char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
151,905✔
2286
      mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
151,905✔
2287
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb, algrId);
151,905✔
2288
    }
2289

2290
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
966,946✔
2291
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
570,424✔
2292
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
570,424✔
2293
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
570,424✔
2294
        bool    exist = false;
570,424✔
2295
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
570,424✔
2296
        if (TSDB_CODE_SUCCESS != code) {
570,424✔
2297
          mndReleaseDb(pMnode, pDb);
×
2298
          if (code != TSDB_CODE_NEED_RETRY) {
×
2299
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2300
          } else {
2301
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2302
          }
2303
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2304
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
2305
          continue;
×
2306
        }
2307
        rsp.dbTsmaVersion = pDb->tsmaVersion;
570,424✔
2308
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
570,424✔
2309
      }
2310
    }
2311

2312
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
966,946✔
2313
        pDbCacheInfo->stateTs != pDb->stateTs) {
924,989✔
2314
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
759,373✔
2315
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
759,373✔
2316
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
759,373✔
2317
        mndReleaseDb(pMnode, pDb);
×
2318
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
2319
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2320
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2321
        continue;
×
2322
      }
2323

2324
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
759,373✔
2325
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
759,373✔
2326
      rsp.useDbRsp->uid = pDb->uid;
759,373✔
2327
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
759,373✔
2328
      rsp.useDbRsp->stateTs = pDb->stateTs;
759,373✔
2329
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
759,373✔
2330
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
759,373✔
2331
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
759,373✔
2332
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
759,373✔
2333
      rsp.useDbRsp->flags = pDb->cfg.flags;
759,373✔
2334
    }
2335

2336
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
1,933,892✔
2337
      mndReleaseDb(pMnode, pDb);
×
2338
      if (terrno != 0) code = terrno;
×
2339
      return code;
×
2340
    }
2341
    mndReleaseDb(pMnode, pDb);
966,946✔
2342
  }
2343

2344
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
4,092,520✔
2345
  void   *pRsp = taosMemoryMalloc(rspLen);
4,092,520✔
2346
  if (pRsp == NULL) {
4,092,520✔
2347
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2348
    tFreeSDbHbBatchRsp(&batchRsp);
×
2349
    return -1;
×
2350
  }
2351
  int32_t ret = 0;
4,092,520✔
2352
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
4,092,520✔
2353

2354
  *ppRsp = pRsp;
4,092,520✔
2355
  *pRspLen = rspLen;
4,092,520✔
2356

2357
  tFreeSDbHbBatchRsp(&batchRsp);
4,092,520✔
2358
  TAOS_RETURN(code);
4,092,520✔
2359
}
2360

2361
static int32_t mndSetTrimDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t startTs, STimeWindow tw,
19,275✔
2362
                                       SArray *vgroupIds, ETsdbOpType type, ETriggerType triggerType,
2363
                                       STrimDbRsp *pRsp) {
2364
  int32_t code = 0, lino = 0;
19,275✔
2365
  SSdb   *pSdb = pMnode->pSdb;
19,275✔
2366
  void   *pIter = NULL;
19,275✔
2367

2368
  SRetentionObj obj = {.optrType = type, .triggerType = triggerType};  // reuse SCompactObj struct
19,275✔
2369
  TAOS_CHECK_EXIT(mndAddRetentionToTrans(pMnode, pTrans, &obj, pDb, pRsp));
19,275✔
2370

2371
  int32_t j = 0;
19,275✔
2372
  int32_t numOfVgroups = taosArrayGetSize(vgroupIds);
19,275✔
2373
  if (numOfVgroups > 0) {
19,275✔
2374
    for (int32_t i = 0; i < numOfVgroups; i++) {
9,516✔
2375
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
7,137✔
2376
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
7,137✔
2377

2378
      if (pVgroup == NULL) {
7,137✔
2379
        mError("db:%s, vgroup:%" PRId64 " not exist", pDb->name, vgId);
2,379✔
2380
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
2,379✔
2381
      } else if (pVgroup->dbUid != pDb->uid) {
4,758✔
2382
        mError("db:%s, vgroup:%" PRId64 " not belong to db:%s", pDb->name, vgId, pDb->name);
×
2383
        sdbRelease(pSdb, pVgroup);
×
2384
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
×
2385
      }
2386
      sdbRelease(pSdb, pVgroup);
4,758✔
2387
    }
2388

2389
    for (int32_t i = 0; i < numOfVgroups; i++) {
7,137✔
2390
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
4,758✔
2391
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
4,758✔
2392

2393
      if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
4,758✔
2394
        sdbRelease(pSdb, pVgroup);
×
2395
        TAOS_RETURN(code);
×
2396
      }
2397

2398
      for (int32_t i = 0; i < pVgroup->replica; i++) {
9,516✔
2399
        SVnodeGid *gid = &pVgroup->vnodeGid[i];
4,758✔
2400
        if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
4,758✔
2401
          sdbRelease(pSdb, pVgroup);
×
2402
          TAOS_RETURN(code);
×
2403
        }
2404
        j++;
4,758✔
2405
      }
2406
      sdbRelease(pSdb, pVgroup);
4,758✔
2407
    }
2408
  } else {
2409
    while (1) {
59,725✔
2410
      SVgObj *pVgroup = NULL;
74,242✔
2411
      pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
74,242✔
2412
      if (pIter == NULL) break;
74,242✔
2413

2414
      if (pVgroup->dbUid == pDb->uid) {
59,725✔
2415
        if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
37,615✔
2416
          sdbCancelFetch(pSdb, pIter);
×
2417
          sdbRelease(pSdb, pVgroup);
×
2418
          TAOS_RETURN(code);
×
2419
        }
2420

2421
        for (int32_t i = 0; i < pVgroup->replica; i++) {
79,428✔
2422
          SVnodeGid *gid = &pVgroup->vnodeGid[i];
41,813✔
2423
          if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
41,813✔
2424
            sdbCancelFetch(pSdb, pIter);
×
2425
            sdbRelease(pSdb, pVgroup);
×
2426
            TAOS_RETURN(code);
×
2427
          }
2428
          j++;
41,813✔
2429
        }
2430
      }
2431

2432
      sdbRelease(pSdb, pVgroup);
59,725✔
2433
    }
2434
  }
2435
_exit:
19,275✔
2436
  TAOS_RETURN(code);
19,275✔
2437
}
2438

2439
static int32_t mndBuildTrimDbRsp(STrimDbRsp *rsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
16,896✔
2440
  int32_t code = 0;
16,896✔
2441
  int32_t rspLen = tSerializeSCompactDbRsp(NULL, 0, (SCompactDbRsp *)rsp);
16,896✔
2442
  void   *pRsp = NULL;
16,896✔
2443
  if (useRpcMalloc) {
16,896✔
2444
    pRsp = rpcMallocCont(rspLen);
×
2445
  } else {
2446
    pRsp = taosMemoryMalloc(rspLen);
16,896✔
2447
  }
2448

2449
  if (pRsp == NULL) {
16,896✔
2450
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2451
    TAOS_RETURN(code);
×
2452
  }
2453

2454
  (void)tSerializeSCompactDbRsp(pRsp, rspLen, (SCompactDbRsp *)rsp);
16,896✔
2455
  *pRspLen = rspLen;
16,896✔
2456
  *ppRsp = pRsp;
16,896✔
2457
  TAOS_RETURN(code);
16,896✔
2458
}
2459

2460
int32_t mndTrimDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb, STimeWindow tw, SArray *vgroupIds,
24,033✔
2461
                         ETsdbOpType type, ETriggerType triggerType) {
2462
  SSdb      *pSdb = pMnode->pSdb;
24,033✔
2463
  SVgObj    *pVgroup = NULL;
24,033✔
2464
  void      *pIter = NULL;
24,033✔
2465
  int32_t    code = 0, lino = 0;
24,033✔
2466
  STrimDbRsp rsp = {0};
24,033✔
2467
  bool       isExist = false;
24,033✔
2468

2469
  while (1) {
5,078✔
2470
    SRetentionObj *pObj = NULL;
29,111✔
2471
    pIter = sdbFetch(pMnode->pSdb, SDB_RETENTION, pIter, (void **)&pObj);
29,111✔
2472
    if (pIter == NULL) break;
29,111✔
2473

2474
    if (strcmp(pObj->dbname, pDb->name) == 0) {
5,078✔
2475
      isExist = true;
4,758✔
2476
    }
2477
    sdbRelease(pMnode->pSdb, pObj);
5,078✔
2478
  }
2479
  if (isExist) {
24,033✔
2480
    mInfo("trim db:%s already exist", pDb->name);
4,758✔
2481
    TAOS_RETURN(TSDB_CODE_MND_TRIM_ALREADY_EXIST);
4,758✔
2482
  }
2483

2484
  int64_t startTs = taosGetTimestampMs();
19,275✔
2485
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "trim-db");
19,275✔
2486
  if (pTrans == NULL) goto _exit;
19,275✔
2487
  mInfo("trans:%d, used to trim db:%s", pTrans->id, pDb->name);
19,275✔
2488
  mndTransSetDbName(pTrans, pDb->name, NULL);
19,275✔
2489
  TAOS_CHECK_EXIT(mndTrancCheckConflict(pMnode, pTrans));
19,275✔
2490
  TAOS_CHECK_EXIT(mndSetTrimDbRedoActions(pMnode, pTrans, pDb, startTs, tw, vgroupIds, type, triggerType, &rsp));
19,275✔
2491
  if (pReq) {
16,896✔
2492
    int32_t rspLen = 0;
16,896✔
2493
    void   *pRsp = NULL;
16,896✔
2494
    rsp.bAccepted = true;
16,896✔
2495
    TAOS_CHECK_EXIT(mndBuildTrimDbRsp(&rsp, &rspLen, &pRsp, false));
16,896✔
2496
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
16,896✔
2497
  }
2498
  TAOS_CHECK_EXIT(mndTransPrepare(pMnode, pTrans));
16,896✔
2499
_exit:
19,275✔
2500
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
19,275✔
2501
    mError("db:%s, failed at line %d to trim db since %s", pDb->name, lino, tstrerror(code));
2,379✔
2502
  }
2503
  mndTransDrop(pTrans);
19,275✔
2504
  TAOS_RETURN(code);
19,275✔
2505
}
2506

2507
#ifdef TD_ENTERPRISE
2508
static int32_t mndCheckRsmaInDb(SMnode *pMnode, SDbObj *pDb) {
15,067✔
2509
  int32_t   code = 0;
15,067✔
2510
  void     *pIter = NULL;
15,067✔
2511
  SRsmaObj *pRsma = NULL;
15,067✔
2512
  while ((pIter = sdbFetch(pMnode->pSdb, SDB_RSMA, pIter, (void **)&pRsma))) {
15,067✔
2513
    if (pRsma->dbUid == pDb->uid) {
15,067✔
2514
      sdbRelease(pMnode->pSdb, pRsma);
15,067✔
2515
      sdbCancelFetch(pMnode->pSdb, pIter);
15,067✔
2516
      TAOS_RETURN(code);
15,067✔
2517
    }
2518
    sdbRelease(pMnode->pSdb, pRsma);
×
2519
  }
2520
  TAOS_RETURN(TSDB_CODE_RSMA_NOT_EXIST);
×
2521
}
2522
#endif
2523

2524
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
24,033✔
2525
  SMnode    *pMnode = pReq->info.node;
24,033✔
2526
  int32_t    code = 0, lino = 0;
24,033✔
2527
  SDbObj    *pDb = NULL;
24,033✔
2528
  STrimDbReq trimReq = {0};
24,033✔
2529
  int64_t    tss = taosGetTimestampMs();
24,033✔
2530

2531
  TAOS_CHECK_EXIT(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq));
24,033✔
2532

2533
  mInfo("db:%s, start to trim, optr:%u, tw:%" PRId64 ",%" PRId64, trimReq.db, trimReq.optrType, trimReq.tw.skey,
24,033✔
2534
        trimReq.tw.ekey);
2535

2536
  pDb = mndAcquireDb(pMnode, trimReq.db);
24,033✔
2537
  if (pDb == NULL) {
24,033✔
2538
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2539
    if (terrno != 0) code = terrno;
×
2540
    TAOS_CHECK_EXIT(code);
×
2541
  }
2542

2543
  TAOS_CHECK_EXIT(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb));
24,033✔
2544

2545
  if (pDb->cfg.isMount) {
24,033✔
2546
    TAOS_CHECK_EXIT(TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT);
×
2547
  }
2548

2549
#ifdef TD_ENTERPRISE
2550
  if (trimReq.optrType == TSDB_OPTR_ROLLUP) {
24,033✔
2551
    TAOS_CHECK_EXIT(mndCheckRsmaInDb(pMnode, pDb));
15,067✔
2552
  }
2553
#endif
2554

2555
  TAOS_CHECK_EXIT(mndTrimDb(pMnode, pReq, pDb, trimReq.tw, trimReq.vgroupIds, trimReq.optrType, trimReq.triggerType));
24,033✔
2556
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
16,896✔
2557

2558
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
16,896✔
2559
    SName name = {0};
16,896✔
2560
    if (tNameFromString(&name, trimReq.db, T_NAME_ACCT | T_NAME_DB) < 0) {
16,896✔
NEW
2561
      mWarn("db:%s, failed at line %d to parse db name", trimReq.db, __LINE__);
×
2562
    }
2563

2564
    char optrType[16] = {0};
16,896✔
2565
    (void)snprintf(optrType, sizeof(optrType), "%u", trimReq.optrType);
16,896✔
2566
    int64_t tse = taosGetTimestampMs();
16,896✔
2567
    double  duration = (double)(tse - tss);
16,896✔
2568
    duration = duration / 1000;
16,896✔
2569
    auditRecord(pReq, pMnode->clusterId, "trimDB", name.dbname, optrType, trimReq.sql, trimReq.sqlLen, duration, 0);
16,896✔
2570
  }
2571

2572
_exit:
24,033✔
2573
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
24,033✔
2574
    mError("db:%s, failed at line %d to process trim db req since %s, optr:%u", trimReq.db, lino, tstrerror(code),
7,137✔
2575
           trimReq.optrType);
2576
  }
2577

2578
  mndReleaseDb(pMnode, pDb);
24,033✔
2579
  tFreeSTrimDbReq(&trimReq);
24,033✔
2580
  TAOS_RETURN(code);
24,033✔
2581
}
2582

2583
static int32_t mndBuildSsMigrateDbRsp(SSsMigrateDbRsp *pSsMigrateRsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
×
2584
  int32_t code = 0;
×
2585
  int32_t rspLen = tSerializeSSsMigrateDbRsp(NULL, 0, pSsMigrateRsp);
×
2586
  void   *pRsp = NULL;
×
2587
  if (useRpcMalloc) {
×
2588
    pRsp = rpcMallocCont(rspLen);
×
2589
  } else {
2590
    pRsp = taosMemoryMalloc(rspLen);
×
2591
  }
2592

2593
  if (pRsp == NULL) {
×
2594
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2595
    TAOS_RETURN(code);
×
2596
  }
2597

2598
  (void)tSerializeSSsMigrateDbRsp(pRsp, rspLen, pSsMigrateRsp);
×
2599
  *pRspLen = rspLen;
×
2600
  *ppRsp = pRsp;
×
2601
  TAOS_RETURN(code);
×
2602
}
2603

2604
static int32_t mndTrimDbWal(SMnode *pMnode, SDbObj *pDb) {
612✔
2605
  SSdb   *pSdb = pMnode->pSdb;
612✔
2606
  SVgObj *pVgroup = NULL;
612✔
2607
  void   *pIter = NULL;
612✔
2608
  int32_t code = 0;
612✔
2609

2610
  // Iterate through all vgroups in this database
2611
  while (1) {
1,224✔
2612
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
1,836✔
2613
    if (pIter == NULL) break;
1,836✔
2614

2615
    // Check if this vgroup belongs to the database
2616
    if (pVgroup->dbUid != pDb->uid) {
1,224✔
2617
      sdbRelease(pSdb, pVgroup);
×
2618
      continue;
×
2619
    }
2620

2621
    // Prepare message to send to vnode
2622
    int32_t   contLen = sizeof(SMsgHead);
1,224✔
2623
    SMsgHead *pHead = rpcMallocCont(contLen);
1,224✔
2624
    if (pHead == NULL) {
1,224✔
2625
      sdbCancelFetch(pSdb, pVgroup);
×
2626
      sdbRelease(pSdb, pVgroup);
×
2627
      code = TSDB_CODE_OUT_OF_MEMORY;
×
2628
      break;
×
2629
    }
2630
    pHead->contLen = htonl(contLen);
1,224✔
2631
    pHead->vgId = htonl(pVgroup->vgId);
1,224✔
2632

2633
    // Send TRIM WAL request to vnode
2634
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM_WAL, .pCont = pHead, .contLen = contLen};
1,224✔
2635
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
1,224✔
2636
    code = tmsgSendReq(&epSet, &rpcMsg);
1,224✔
2637
    if (code != 0) {
1,224✔
2638
      mError("vgId:%d, failed to send vnode-trim-wal request since 0x%x", pVgroup->vgId, code);
×
2639
    } else {
2640
      mInfo("vgId:%d, send vnode-trim-wal request to vnode", pVgroup->vgId);
1,224✔
2641
    }
2642
    sdbRelease(pSdb, pVgroup);
1,224✔
2643
  }
2644

2645
  return code;
612✔
2646
}
2647

2648
static int32_t mndProcessTrimDbWalReq(SRpcMsg *pReq) {
612✔
2649
  SMnode    *pMnode = pReq->info.node;
612✔
2650
  int32_t    code = -1;
612✔
2651
  SDbObj    *pDb = NULL;
612✔
2652
  STrimDbReq trimReq = {0};
612✔
2653

2654
  TAOS_CHECK_GOTO(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq), NULL, _OVER);
612✔
2655

2656
  mInfo("db:%s, start to trim WAL", trimReq.db);
612✔
2657

2658
  pDb = mndAcquireDb(pMnode, trimReq.db);
612✔
2659
  if (pDb == NULL) {
612✔
2660
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2661
    if (terrno != 0) code = terrno;
×
2662
    goto _OVER;
×
2663
  }
2664

2665
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
612✔
2666

2667
  code = mndTrimDbWal(pMnode, pDb);
612✔
2668

2669
_OVER:
612✔
2670
  if (code != 0) {
612✔
2671
    mError("db:%s, failed to process trim db wal req since %s", trimReq.db, terrstr());
×
2672
  }
2673

2674
  mndReleaseDb(pMnode, pDb);
612✔
2675
  TAOS_RETURN(code);
612✔
2676
}
2677

2678
int32_t mndSsMigrateDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
×
2679
  int32_t       code = 0;
×
2680
  SSsMigrateDbRsp ssMigrateRsp = {0};
×
2681

2682
  bool  isExist = false;
×
2683
  void *pIter = NULL;
×
2684
  while (1) {
×
2685
    SSsMigrateObj *pSsMigrate = NULL;
×
2686
    pIter = sdbFetch(pMnode->pSdb, SDB_SSMIGRATE, pIter, (void **)&pSsMigrate);
×
2687
    if (pIter == NULL) break;
×
2688

2689
    if (strcmp(pSsMigrate->dbname, pDb->name) == 0) {
×
2690
      isExist = true;
×
2691
    }
2692
    sdbRelease(pMnode->pSdb, pSsMigrate);
×
2693
  }
2694
  if (isExist) {
×
2695
    mInfo("ssmigrate db:%s already exist", pDb->name);
×
2696

2697
    if (pReq) {
×
2698
      int32_t rspLen = 0;
×
2699
      void   *pRsp = NULL;
×
2700
      ssMigrateRsp.ssMigrateId = 0;
×
2701
      ssMigrateRsp.bAccepted = false;
×
2702
      TAOS_CHECK_RETURN(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, true));
×
2703

2704
      pReq->info.rsp = pRsp;
×
2705
      pReq->info.rspLen = rspLen;
×
2706
    }
2707

2708
    return TSDB_CODE_MND_SSMIGRATE_ALREADY_EXIST;
×
2709
  }
2710

2711
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "ssmigrate-db");
×
2712
  if (pTrans == NULL) {
×
2713
    mError("failed to create ssmigrate-db trans since %s", terrstr());
×
2714
    goto _OVER;
×
2715
  }
2716

2717
  mInfo("trans:%d, used to ssmigrate db:%s", pTrans->id, pDb->name);
×
2718
  mndTransSetDbName(pTrans, pDb->name, NULL);
×
2719
  TAOS_CHECK_GOTO(mndTrancCheckConflict(pMnode, pTrans), NULL, _OVER);
×
2720

2721
  SSsMigrateObj ssMigrate = { .startTime = taosGetTimestampMs() };
×
2722
  TAOS_CHECK_GOTO(mndAddSsMigrateToTran(pMnode, pTrans, &ssMigrate, pDb), NULL, _OVER);
×
2723

2724
  if (pReq) {
×
2725
    int32_t rspLen = 0;
×
2726
    void   *pRsp = NULL;
×
2727
    ssMigrateRsp.ssMigrateId = ssMigrate.id;
×
2728
    ssMigrateRsp.bAccepted = true;
×
2729
    TAOS_CHECK_GOTO(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, false), NULL, _OVER);
×
2730
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
×
2731
  }
2732

2733
  if (mndTransPrepare(pMnode, pTrans) != 0) goto _OVER;
×
2734
  code = 0;
×
2735

2736
_OVER:
×
2737
  mndTransDrop(pTrans);
×
2738
  TAOS_RETURN(code);
×
2739
}
2740

2741

2742
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq) {
×
2743
  SMnode         *pMnode = pReq->info.node;
×
2744
  int32_t         code = -1;
×
2745
  SDbObj         *pDb = NULL;
×
2746
  SSsMigrateDbReq ssMigrateReq = {0};
×
2747

2748
  if (!tsSsEnabled) {
×
2749
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
2750
    goto _OVER;
×
2751
  }
2752

2753
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_SHARED_STORAGE), NULL, _OVER);
×
2754
  TAOS_CHECK_GOTO(tDeserializeSSsMigrateDbReq(pReq->pCont, pReq->contLen, &ssMigrateReq), NULL, _OVER);
×
2755

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

2758
  pDb = mndAcquireDb(pMnode, ssMigrateReq.db);
×
2759
  if (pDb == NULL) {
×
2760
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2761
    if (terrno != 0) code = terrno;
×
2762
    goto _OVER;
×
2763
  }
2764

2765
  // TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_SSMIGRATE_DB, pDb), NULL, _OVER);
2766

2767
  if(pDb->cfg.isMount) {
×
2768
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
2769
    goto _OVER;
×
2770
  }
2771

2772
  code = mndSsMigrateDb(pMnode, pReq, pDb);
×
2773

2774
_OVER:
×
2775
  if (code != 0) {
×
2776
    mError("db:%s, failed to process ssmigrate db req since %s", ssMigrateReq.db, tstrerror(code));
×
2777
  }
2778

2779
  mndReleaseDb(pMnode, pDb);
×
2780
  TAOS_RETURN(code);
×
2781
}
2782

2783
const char *mndGetDbStr(const char *src) {
31,757,012✔
2784
  char *pos = strstr(src, TS_PATH_DELIMITER);
31,757,012✔
2785
  if (pos != NULL) ++pos;
31,757,012✔
2786
  if (pos == NULL) return src;
31,757,012✔
2787
  return pos;
30,078,672✔
2788
}
2789

2790
const char *mndGetStableStr(const char *src) {
27,949,017✔
2791
  char *pos = strstr(src, TS_PATH_DELIMITER);
27,949,017✔
2792
  if (pos != NULL) ++pos;
27,949,017✔
2793
  if (pos == NULL) return src;
27,949,017✔
2794
  return mndGetDbStr(pos);
27,949,017✔
2795
}
2796

2797
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
×
2798
  int64_t v = 0;
×
2799
  switch (unit) {
×
2800
    case 's':
×
2801
      v = val / 1000;
×
2802
      break;
×
2803
    case 'm':
×
2804
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
×
2805
      break;
×
2806
    case 'h':
×
2807
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2808
      break;
×
2809
    case 'd':
×
2810
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
×
2811
      break;
×
2812
    case 'w':
×
2813
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2814
      break;
×
2815
    default:
×
2816
      break;
×
2817
  }
2818

2819
  return v;
×
2820
}
2821

2822
static char *buildRetension(SArray *pRetension) {
767,251✔
2823
  size_t size = taosArrayGetSize(pRetension);
767,251✔
2824
  if (size == 0) {
767,251✔
2825
    return NULL;
767,251✔
2826
  }
2827

2828
  char       *p1 = taosMemoryCalloc(1, 100);
×
2829
  SRetention *p = taosArrayGet(pRetension, 0);
×
2830

2831
  int32_t len = 2;
×
2832

2833
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2834
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2835
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2836

2837
  if (size > 1) {
×
2838
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2839
    p = taosArrayGet(pRetension, 1);
×
2840

2841
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2842
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2843
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2844
  }
2845

2846
  if (size > 2) {
×
2847
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2848
    p = taosArrayGet(pRetension, 2);
×
2849

2850
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2851
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2852
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2853
  }
2854

2855
  varDataSetLen(p1, len);
×
2856
  return p1;
×
2857
}
2858

2859
static const char *getCacheModelStr(int8_t cacheModel) {
767,251✔
2860
  switch (cacheModel) {
767,251✔
2861
    case TSDB_CACHE_MODEL_NONE:
727,093✔
2862
      return TSDB_CACHE_MODEL_NONE_STR;
727,093✔
2863
    case TSDB_CACHE_MODEL_LAST_ROW:
2,309✔
2864
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
2,309✔
2865
    case TSDB_CACHE_MODEL_LAST_VALUE:
7,479✔
2866
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
7,479✔
2867
    case TSDB_CACHE_MODEL_BOTH:
30,370✔
2868
      return TSDB_CACHE_MODEL_BOTH_STR;
30,370✔
2869
    default:
×
2870
      break;
×
2871
  }
2872
  return "unknown";
×
2873
}
2874

2875
static void getEncryptAlgorithmStr(SMnode *pMnode, int8_t encryptAlgorithm, char *out) {
767,251✔
2876
  switch (encryptAlgorithm) {
767,251✔
2877
    case TSDB_ENCRYPT_ALGO_NONE: {
767,251✔
2878
      tstrncpy(out, TSDB_ENCRYPT_ALGO_NONE_STR, TSDB_ENCRYPT_ALGR_NAME_LEN);
767,251✔
2879
      break;
767,251✔
2880
    }
2881
    default: {
×
2882
      SEncryptAlgrObj *obj = mndAcquireEncryptAlgrById(pMnode, encryptAlgorithm);
×
2883
      if (obj != NULL) {
×
2884
        tstrncpy(out, obj->algorithm_id, TSDB_ENCRYPT_ALGR_NAME_LEN);
×
2885
        mndReleaseEncryptAlgr(pMnode, obj);
×
2886
      }
2887
      break;
×
2888
    }
2889
  }
2890
}
767,251✔
2891

2892
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
2,315,814✔
2893
  if (pDb->cfg.replications == 1) return true;
2,315,814✔
2894

2895
  SSdb *pSdb = pMnode->pSdb;
429,803✔
2896
  void *pIter = NULL;
429,803✔
2897
  bool  isReady = true;
429,803✔
2898
  while (1) {
2,740,808✔
2899
    SVgObj *pVgroup = NULL;
3,170,611✔
2900
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,170,611✔
2901
    if (pIter == NULL) break;
3,170,611✔
2902

2903
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
2,740,808✔
2904
      bool hasLeader = false;
1,488,224✔
2905
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
5,901,067✔
2906
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
4,412,843✔
2907
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
3,837,309✔
2908
          hasLeader = true;
575,534✔
2909
        }
2910
      }
2911
      if (!hasLeader) isReady = false;
1,488,224✔
2912
    }
2913
    sdbRelease(pSdb, pVgroup);
2,740,808✔
2914
  }
2915

2916
  return isReady;
429,803✔
2917
}
2918

2919
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
2,032,961✔
2920
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2921
  int32_t cols = 0;
2,032,961✔
2922
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
2,032,961✔
2923
  char   *buf = taosMemoryMalloc(bytes);
2,032,961✔
2924
  if (buf == NULL) {
2,032,961✔
2925
    mError("db:%s, failed to malloc buffer", pDb->name);
×
2926
    return;
×
2927
  }
2928
  int32_t code = 0;
2,032,961✔
2929
  int32_t lino = 0;
2,032,961✔
2930

2931
  const char *name = mndGetDbStr(pDb->name);
2,032,961✔
2932
  if (name != NULL) {
2,032,961✔
2933
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
2,032,961✔
2934
  } else {
2935
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
2936
  }
2937

2938
  const char *precStr = NULL;
2,032,961✔
2939
  switch (pDb->cfg.precision) {
2,032,961✔
2940
    case TSDB_TIME_PRECISION_MILLI:
1,974,365✔
2941
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
1,974,365✔
2942
      break;
1,974,365✔
2943
    case TSDB_TIME_PRECISION_MICRO:
19,684✔
2944
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
19,684✔
2945
      break;
19,684✔
2946
    case TSDB_TIME_PRECISION_NANO:
38,912✔
2947
      precStr = TSDB_TIME_PRECISION_NANO_STR;
38,912✔
2948
      break;
38,912✔
2949
    default:
×
2950
      precStr = "none";
×
2951
      break;
×
2952
  }
2953
  char precVstr[10] = {0};
2,032,961✔
2954
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
2,032,961✔
2955

2956
  char *statusStr = "ready";
2,032,961✔
2957
  if (objStatus == SDB_STATUS_CREATING) {
2,032,961✔
2958
    statusStr = "creating";
×
2959
  } else if (objStatus == SDB_STATUS_DROPPING) {
2,032,961✔
2960
    statusStr = "dropping";
×
2961
  } else {
2962
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
2,032,961✔
2963
      statusStr = "unsynced";
2,576✔
2964
    }
2965
  }
2966
  char statusVstr[24] = {0};
2,032,961✔
2967
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
2,032,961✔
2968

2969
  if (sysDb || !sysinfo) {
2,032,961✔
2970
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
49,362,690✔
2971
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
48,096,980✔
2972
      if (i == 0) {
48,096,980✔
2973
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
1,265,710✔
2974
      } else if (i == 1) {
46,831,270✔
2975
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
1,265,710✔
2976
      } else if (i == 3) {
45,565,560✔
2977
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
1,265,710✔
2978
      } else if (i == 14) {
44,299,850✔
2979
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
1,265,710✔
2980
      } else if (i == 15) {
43,034,140✔
2981
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
1,265,710✔
2982
      } else {
2983
        colDataSetNULL(pColInfo, rows);
41,768,430✔
2984
      }
2985
    }
2986
  } else {
2987
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
2988
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
767,251✔
2989

2990
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
2991
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
767,251✔
2992

2993
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
2994
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
767,251✔
2995

2996
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
2997
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
767,251✔
2998

2999
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3000
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
767,251✔
3001

3002
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
767,251✔
3003
    char        strictVstr[24] = {0};
767,251✔
3004
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
767,251✔
3005
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3006
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
767,251✔
3007

3008
    char    durationStr[128] = {0};
767,251✔
3009
    char    durationVstr[128] = {0};
767,251✔
3010
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
767,251✔
3011
                                       pDb->cfg.daysPerFile);
3012

3013
    varDataSetLen(durationVstr, len);
767,251✔
3014
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3015
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
767,251✔
3016

3017
    char keepVstr[128] = {0};
767,251✔
3018
    char keep0Str[32] = {0};
767,251✔
3019
    char keep1Str[32] = {0};
767,251✔
3020
    char keep2Str[32] = {0};
767,251✔
3021

3022
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
767,251✔
3023
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
767,251✔
3024
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
767,251✔
3025

3026
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
767,251✔
3027
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
3028
    } else {
3029
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
767,251✔
3030
    }
3031
    varDataSetLen(keepVstr, len);
767,251✔
3032
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3033
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
767,251✔
3034

3035
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3036
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
767,251✔
3037

3038
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3039
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
767,251✔
3040

3041
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3042
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
767,251✔
3043

3044
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3045
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
767,251✔
3046

3047
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3048
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
767,251✔
3049

3050
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3051
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
767,251✔
3052

3053
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3054
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
767,251✔
3055

3056
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3057
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
767,251✔
3058

3059
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
767,251✔
3060
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3061
    if (rentensionVstr == NULL) {
767,251✔
3062
      colDataSetNULL(pColInfo, rows);
767,251✔
3063
    } else {
3064
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
×
3065
      taosMemoryFree(rentensionVstr);
×
3066
    }
3067

3068
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3069
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
767,251✔
3070

3071
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
767,251✔
3072
    char        cacheModelVstr[24] = {0};
767,251✔
3073
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
767,251✔
3074
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3075
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
767,251✔
3076

3077
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3078
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
767,251✔
3079

3080
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3081
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
767,251✔
3082

3083
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3084
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
767,251✔
3085

3086
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3087
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
767,251✔
3088

3089
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3090
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
767,251✔
3091

3092
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3093
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
767,251✔
3094

3095
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3096
    int16_t hashPrefix = pDb->cfg.hashPrefix;
767,251✔
3097
    if (hashPrefix > 0) {
767,251✔
3098
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
4,925✔
3099
    } else if (hashPrefix < 0) {
762,326✔
3100
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
5,536✔
3101
    }
3102
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
767,251✔
3103

3104
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3105
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
767,251✔
3106

3107
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3108
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
767,251✔
3109

3110
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3111
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
767,251✔
3112

3113
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3114
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssChunkSize, false), &lino, _OVER);
767,251✔
3115

3116
    char keeplocalVstr[128] = {0};
767,251✔
3117
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.ssKeepLocal);
767,251✔
3118
    varDataSetLen(keeplocalVstr, len);
767,251✔
3119
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3120
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
767,251✔
3121

3122
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3123
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssCompact, false), &lino, _OVER);
767,251✔
3124

3125
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3126
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
767,251✔
3127

3128
    char encryptAlgorithmStr[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
767,251✔
3129
    getEncryptAlgorithmStr(pMnode, pDb->cfg.encryptAlgorithm, encryptAlgorithmStr);
767,251✔
3130
    char        encryptAlgorithmVStr[24] = {0};
767,251✔
3131
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
767,251✔
3132
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3133
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
767,251✔
3134

3135
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
767,251✔
3136
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
767,251✔
3137
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
767,251✔
3138
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
767,251✔
3139
    }
3140

3141
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
767,251✔
3142
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
767,251✔
3143
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
767,251✔
3144
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
767,251✔
3145
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
767,251✔
3146
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
767,251✔
3147
    }
3148

3149
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
767,251✔
3150
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
767,251✔
3151
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
767,251✔
3152
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
767,251✔
3153
    }
3154

3155
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
767,251✔
3156
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.isAudit, false), &lino, _OVER);
767,251✔
3157
  }
3158
_OVER:
2,032,961✔
3159
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
2,032,961✔
3160
  taosMemoryFree(buf);
2,032,961✔
3161
}
3162

3163
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
630,665✔
3164
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
630,665✔
3165
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
630,665✔
3166
  pDbObj->cfg.numOfVgroups = 0;
630,665✔
3167
  pDbObj->cfg.strict = 1;
630,665✔
3168
  pDbObj->cfg.replications = 1;
630,665✔
3169
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
630,665✔
3170
}
630,665✔
3171

3172
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
630,665✔
3173
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
630,665✔
3174
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
630,665✔
3175
  pDbObj->cfg.numOfVgroups = 0;
630,665✔
3176
  pDbObj->cfg.strict = 1;
630,665✔
3177
  pDbObj->cfg.replications = 1;
630,665✔
3178
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
630,665✔
3179
}
630,665✔
3180

3181
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
5,885,558✔
3182
  SVgObj  *pVgroup = pObj;
5,885,558✔
3183
  int32_t *numOfTables = p1;
5,885,558✔
3184
  int64_t  uid = *(int64_t *)p2;
5,885,558✔
3185
  if (pVgroup->dbUid == uid) {
5,885,558✔
3186
    *numOfTables += pVgroup->numOfTables;
1,762,261✔
3187
  }
3188
  return true;
5,885,558✔
3189
}
3190

3191
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
630,665✔
3192
  SMnode    *pMnode = pReq->info.node;
630,665✔
3193
  SSdb      *pSdb = pMnode->pSdb;
630,665✔
3194
  int32_t    numOfRows = 0;
630,665✔
3195
  SDbObj    *pDb = NULL;
630,665✔
3196
  SUserObj  *pUser = NULL;
630,665✔
3197
  ESdbStatus objStatus = 0;
630,665✔
3198

3199
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
630,665✔
3200
  if (pUser == NULL) return 0;
630,665✔
3201
  bool sysinfo = pUser->sysInfo;
630,665✔
3202

3203
  // Append the information_schema database into the result.
3204
  if (!pShow->sysDbRsp) {
630,665✔
3205
    SDbObj infoschemaDb = {0};
630,665✔
3206
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
630,665✔
3207
    size_t numOfTables = 0;
630,665✔
3208
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
630,665✔
3209
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
630,665✔
3210

3211
    numOfRows += 1;
630,665✔
3212

3213
    SDbObj perfschemaDb = {0};
630,665✔
3214
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
630,665✔
3215
    numOfTables = 0;
630,665✔
3216
    getPerfDbMeta(NULL, &numOfTables);
630,665✔
3217
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
630,665✔
3218

3219
    numOfRows += 1;
630,665✔
3220
    pShow->sysDbRsp = true;
630,665✔
3221
  }
3222

3223
  while (numOfRows < rowsCapacity) {
1,414,691✔
3224
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
1,414,691✔
3225
    if (pShow->pIter == NULL) break;
1,414,691✔
3226

3227
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
784,026✔
3228
      int32_t numOfTables = 0;
771,631✔
3229
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
771,631✔
3230
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
771,631✔
3231
      numOfRows++;
771,631✔
3232
    }
3233

3234
    sdbRelease(pSdb, pDb);
784,026✔
3235
  }
3236

3237
  pShow->numOfRows += numOfRows;
630,665✔
3238
  mndReleaseUser(pMnode, pUser);
630,665✔
3239
  return numOfRows;
630,665✔
3240
}
3241

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

© 2026 Coveralls, Inc