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

taosdata / TDengine / #4908

30 Dec 2025 10:52AM UTC coverage: 65.386% (-0.2%) from 65.541%
#4908

push

travis-ci

web-flow
enh: drop multi-stream (#33962)

60 of 106 new or added lines in 4 files covered. (56.6%)

1330 existing lines in 113 files now uncovered.

193461 of 295877 relevant lines covered (65.39%)

115765274.47 hits per line

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

77.68
/source/dnode/mnode/impl/src/mndDb.c
1
/*
2
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3
 *
4
 * This program is free software: you can use, redistribute, and/or modify
5
 * it under the terms of the GNU Affero General Public License, version 3
6
 * or later ("AGPL"), as published by the Free Software Foundation.
7
 *
8
 * This program is distributed in the hope that it will be useful, but WITHOUT
9
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10
 * FITNESS FOR A PARTICULAR PURPOSE.
11
 *
12
 * You should have received a copy of the GNU Affero General Public License
13
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14
 */
15

16
#define _DEFAULT_SOURCE
17
#include "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) {
396,893✔
74
  SSdbTable table = {
396,893✔
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);
396,893✔
86
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
396,893✔
87
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
396,893✔
88
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
396,893✔
89
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
396,893✔
90
  mndSetMsgHandle(pMnode, TDMT_MND_SCAN_DB, mndProcessScanDbReq);
396,893✔
91
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
396,893✔
92
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB_WAL, mndProcessTrimDbWalReq);
396,893✔
93
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
396,893✔
94
  mndSetMsgHandle(pMnode, TDMT_MND_SSMIGRATE_DB, mndProcessSsMigrateDbReq);
396,893✔
95
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
396,893✔
96

97
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
396,893✔
98
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
396,893✔
99

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

103
void mndCleanupDb(SMnode *pMnode) {}
396,829✔
104

105
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
5,394,214✔
106
  int32_t code = 0;
5,394,214✔
107
  int32_t lino = 0;
5,394,214✔
108
  terrno = TSDB_CODE_OUT_OF_MEMORY;
5,394,214✔
109

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

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

174
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
5,394,214✔
175
  SDB_SET_UINT8(pRaw, dataPos, pDb->cfg.flags, _OVER)
5,394,214✔
176
  SDB_SET_INT64(pRaw, dataPos, pDb->ownerId, _OVER)
5,394,214✔
177

178
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
5,394,214✔
179

180
  terrno = 0;
5,394,214✔
181

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

189
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
5,394,214✔
190
  return pRaw;
5,394,214✔
191
}
192

193
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
4,885,375✔
194
  int32_t code = 0;
4,885,375✔
195
  int32_t lino = 0;
4,885,375✔
196
  terrno = TSDB_CODE_OUT_OF_MEMORY;
4,885,375✔
197
  SSdbRow *pRow = NULL;
4,885,375✔
198
  SDbObj  *pDb = NULL;
4,885,375✔
199

200
  int8_t sver = 0;
4,885,375✔
201
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
4,885,375✔
202

203
  if (sver != DB_VER_NUMBER) {
4,885,375✔
204
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
205
    goto _OVER;
×
206
  }
207

208
  pRow = sdbAllocRow(sizeof(SDbObj));
4,885,375✔
209
  if (pRow == NULL) goto _OVER;
4,885,375✔
210

211
  pDb = sdbGetRowObj(pRow);
4,885,375✔
212
  if (pDb == NULL) goto _OVER;
4,885,375✔
213

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

288
  taosInitRWLatch(&pDb->lock);
4,885,375✔
289

290
  if (pDb->cfg.ssChunkSize == 0) {
4,885,375✔
291
    pDb->cfg.ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
×
292

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

296
  if (pDb->cfg.ssKeepLocal == 0) {
4,885,375✔
297
    pDb->cfg.ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
×
298

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

302
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
4,885,375✔
303
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
4,885,375✔
304
          TSDB_DEFAULT_TSDB_PAGESIZE);
305
  }
306

307
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
4,885,375✔
308
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
4,787,707✔
309
  }
310

311
  terrno = 0;
4,885,375✔
312

313
_OVER:
4,885,375✔
314
  if (terrno != 0) {
4,885,375✔
315
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
316
    taosMemoryFreeClear(pRow);
×
317
    return NULL;
×
318
  }
319

320
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
4,885,375✔
321
  return pRow;
4,885,375✔
322
}
323

324
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
1,105,769✔
325
  SSdb    *pSdb = pMnode->pSdb;
1,105,769✔
326
  SSdbRow *pRow = NULL;
1,105,769✔
327
  SDbObj  *pNewDb = NULL;
1,105,769✔
328
  int      code = -1;
1,105,769✔
329

330
  pRow = mndDbActionDecode(pRaw);
1,105,769✔
331
  if (pRow == NULL) {
1,105,769✔
332
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
333
    if (terrno != 0) code = terrno;
×
334
    goto _OVER;
×
335
  }
336
  pNewDb = sdbGetRowObj(pRow);
1,105,769✔
337
  if (pNewDb == NULL) {
1,105,769✔
338
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
339
    if (terrno != 0) code = terrno;
×
340
    goto _OVER;
×
341
  }
342

343
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
1,105,769✔
344
  if (pOldDb != NULL) {
1,105,769✔
345
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
346
    sdbRelease(pMnode->pSdb, pOldDb);
×
347
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
348
    if (terrno != 0) code = terrno;
×
349
    goto _OVER;
×
350
  }
351

352
  code = 0;
1,105,769✔
353
_OVER:
1,105,769✔
354
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
1,105,769✔
355
  taosMemoryFreeClear(pRow);
1,105,769✔
356
  return code;
1,105,769✔
357
}
358

359
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
1,241,295✔
360
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
1,241,295✔
361
  return 0;
1,241,295✔
362
}
363

364
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
4,885,375✔
365
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
4,885,375✔
366
  taosArrayDestroy(pDb->cfg.pRetensions);
4,885,375✔
367
  return 0;
4,885,375✔
368
}
369

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

413
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
7,552,645✔
414
  SSdb *pSdb = pMnode->pSdb;
7,552,645✔
415
  return sdbGetTableVer(pSdb, SDB_VGROUP);
7,552,645✔
416
}
417

418
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
104,810,995✔
419
  SSdb   *pSdb = pMnode->pSdb;
104,810,995✔
420
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
104,811,477✔
421
  if (pDb == NULL) {
104,811,491✔
422
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
5,136,131✔
423
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
4,992,221✔
424
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
143,428✔
425
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
3,682✔
426
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
139,746✔
427
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
139,746✔
428
    } else {
429
      terrno = TSDB_CODE_APP_ERROR;
×
430
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
431
    }
432
  }
433
  return pDb;
104,810,803✔
434
}
435

436
SDbObj *mndAcquireAuditDb(SMnode *pMnode) {
39,456,373✔
437
  SSdb      *pSdb = pMnode->pSdb;
39,456,373✔
438
  SDbObj    *pDb = NULL;
39,456,373✔
439
  void      *pIter = NULL;
39,456,373✔
440
  ESdbStatus objStatus = 0;
39,456,373✔
441

442
  while (1) {
443
    pIter = sdbFetchAll(pSdb, SDB_DB, pIter, (void **)&pDb, &objStatus, true);
111,201,573✔
444
    if (pIter == NULL) break;
111,201,573✔
445
    if (pDb->cfg.isAudit == 1) {
71,745,200✔
446
      sdbCancelFetch(pSdb, pIter);
×
447
      break;
×
448
    }
449

450
    sdbRelease(pSdb, pDb);
71,745,200✔
451
  }
452
  return pDb;
39,456,373✔
453
}
454

455
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
112,486,018✔
456
  SSdb *pSdb = pMnode->pSdb;
112,486,018✔
457
  sdbRelease(pSdb, pDb);
112,486,571✔
458
}
112,487,617✔
459

460
bool mndDbIsExist(SMnode *pMnode, const char *db, int64_t uid) {
118,966✔
461
  SDbObj *pDb = mndAcquireDb(pMnode, db);
118,966✔
462
  if (pDb == NULL) {
118,966✔
UNCOV
463
    return false;
×
464
  } else {
465
    bool result = (uid == 0) || (uid == pDb->uid);
118,966✔
466
    mndReleaseDb(pMnode, pDb);
118,966✔
467
    pDb = NULL;
118,966✔
468
    return result;
118,966✔
469
  }
470
}
471

472
int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
1,073,690✔
473
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
1,073,690✔
474
  if (pos == NULL) {
1,073,690✔
475
    return TSDB_CODE_MND_INVALID_DB;
×
476
  }
477

478
  int32_t acctId;
1,073,485✔
479
  int32_t code = taosStr2int32(dbName, &acctId);
1,073,690✔
480
  if (code != 0) {
1,073,690✔
481
    return code;
×
482
  }
483

484
  if (acctId != pUser->acctId) {
1,073,690✔
485
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
486
  }
487

488
  return 0;
1,073,690✔
489
}
490

491
int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
1,073,690✔
492
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
1,073,690✔
493

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

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

544
  if (pCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
1,072,987✔
545
  if (pCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
1,072,987✔
546
  if (pCfg->ssCompact < TSDB_MIN_SS_COMPACT || pCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
1,072,987✔
547

548
  if (pCfg->compactInterval != 0 &&
1,072,987✔
549
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
3,364✔
550
    return code;
×
551
  if (pCfg->compactStartTime != 0 &&
1,072,987✔
552
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
3,719✔
553
    return code;
×
554
  if (pCfg->compactEndTime != 0 &&
1,072,987✔
555
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
3,719✔
556
    return code;
×
557
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
1,072,987✔
558
    return code;
×
559
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
1,072,987✔
560
    return code;
×
561
  if (pCfg->isAudit < 0 || pCfg->isAudit > 1) return code;
1,072,987✔
562

563
  code = 0;
1,072,987✔
564
  TAOS_RETURN(code);
1,072,987✔
565
}
566

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

602
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
83,028✔
603
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
604
    return code;
×
605
  }
606
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
83,028✔
607
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
608
    return code;
×
609
  }
610

611
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
83,028✔
612
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
1,221✔
613
    return code;
×
614
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
83,028✔
615
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
83,028✔
616
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
83,028✔
617
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
82,328✔
618
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
82,328✔
619
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
82,328✔
620
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
82,328✔
621
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
622
    return code;
×
623
  }
624
  if (pNewCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pNewCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
82,328✔
625
  if (pNewCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pNewCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
82,328✔
626
  if (pNewCfg->ssCompact < TSDB_MIN_SS_COMPACT || pNewCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
82,328✔
627

628
  if (pNewCfg->compactInterval != 0 &&
82,328✔
629
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
6,174✔
630
    return code;
×
631
  if (pNewCfg->compactStartTime != 0 &&
82,328✔
632
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
4,074✔
633
    return code;
×
634
  if (pNewCfg->compactEndTime != 0 &&
82,328✔
635
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
4,074✔
636
    return code;
×
637
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
82,328✔
638
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
4,074✔
639
    return code;
×
640
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
82,328✔
641
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
82,328✔
642
    return code;
×
643
  if (pNewCfg->isAudit < 0 || pNewCfg->isAudit > 1) return code;
82,328✔
644

645
  code = 0;
82,328✔
646
  TAOS_RETURN(code);
82,328✔
647
}
648

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

687
int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,068,685✔
688
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,068,685✔
689
  if (pDbRaw == NULL) return -1;
1,068,685✔
690

691
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
1,068,685✔
692
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
1,068,685✔
693
  return 0;
1,068,685✔
694
}
695

696
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,068,177✔
697
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,351,326✔
698
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
2,283,149✔
699
  }
700
  return 0;
1,068,177✔
701
}
702

703
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
704
  int32_t  code = 0;
×
705
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
706
  if (pDbRaw == NULL) {
×
707
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
708
    if (terrno != 0) code = terrno;
×
709
    TAOS_RETURN(code);
×
710
  }
711
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
712
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
713

714
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
715
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
716
    if (pVgRaw == NULL) {
×
717
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
718
      if (terrno != 0) code = terrno;
×
719
      TAOS_RETURN(code);
×
720
    }
721
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
722
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
723
  }
724

725
  if (pDb->cfg.withArbitrator) {
×
726
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
727
      SVgObj   *pVgObj = pVgroups + v;
×
728
      SArbGroup arbGroup = {0};
×
729
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
730
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
731
    }
732
  }
733

734
  TAOS_RETURN(code);
×
735
}
736

737
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,068,177✔
738
  int32_t  code = 0;
1,068,177✔
739
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,068,177✔
740
  if (pDbRaw == NULL) {
1,068,177✔
741
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
742
    if (terrno != 0) code = terrno;
×
743
    TAOS_RETURN(code);
×
744
  }
745
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
1,068,177✔
746
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
1,068,177✔
747

748
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,351,326✔
749
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
2,283,149✔
750
    if (pVgRaw == NULL) {
2,283,149✔
751
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
752
      if (terrno != 0) code = terrno;
×
753
      TAOS_RETURN(code);
×
754
    }
755
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
2,283,149✔
756
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
2,283,149✔
757
  }
758

759
  if (pDb->cfg.withArbitrator) {
1,068,177✔
760
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
9,870✔
761
      SVgObj   *pVgObj = pVgroups + v;
5,985✔
762
      SArbGroup arbGroup = {0};
5,985✔
763
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
5,985✔
764
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
5,985✔
765
    }
766
  }
767

768
  TAOS_RETURN(code);
1,068,177✔
769
}
770

771
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
1,068,177✔
772
                                        SUserObj *pUserDuped) {
773
  int32_t  code = 0;
1,068,177✔
774
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,068,177✔
775
  if (pDbRaw == NULL) {
1,068,177✔
776
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
777
    if (terrno != 0) code = terrno;
×
778
    TAOS_RETURN(code);
×
779
  }
780
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
1,068,177✔
781
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
1,068,177✔
782

783
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,351,326✔
784
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
2,283,149✔
785
    if (pVgRaw == NULL) {
2,283,149✔
786
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
787
      if (terrno != 0) code = terrno;
×
788
      TAOS_RETURN(code);
×
789
    }
790
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
2,283,149✔
791
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
2,283,149✔
792
  }
793

794
  if (pDb->cfg.withArbitrator) {
1,068,177✔
795
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
9,870✔
796
      SVgObj   *pVgObj = pVgroups + v;
5,985✔
797
      SArbGroup arbGroup = {0};
5,985✔
798
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
5,985✔
799
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
5,985✔
800
    }
801
  }
802

803
  if (pUserDuped) {
1,068,177✔
804
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
×
805
    if (pUserRaw == NULL) {
×
806
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
807
      if (terrno != 0) code = terrno;
×
808
      TAOS_RETURN(code);
×
809
    }
810
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
×
811
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
×
812
  }
813

814
  TAOS_RETURN(code);
1,068,177✔
815
}
816

817
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,068,177✔
818
  int32_t code = 0;
1,068,177✔
819
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
3,351,326✔
820
    SVgObj *pVgroup = pVgroups + vg;
2,283,149✔
821

822
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
4,819,267✔
823
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,536,118✔
824
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
2,536,118✔
825
    }
826
  }
827

828
  TAOS_RETURN(code);
1,068,177✔
829
}
830

831
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,068,177✔
832
  int32_t code = 0;
1,068,177✔
833
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
3,351,326✔
834
    SVgObj *pVgroup = pVgroups + vg;
2,283,149✔
835

836
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
4,819,267✔
837
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,536,118✔
838
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
2,536,118✔
839
    }
840
  }
841

842
  TAOS_RETURN(code);
1,068,177✔
843
}
844

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

937
  if (pCreate->isAudit == 1) {
1,073,182✔
938
    if (dbObj.cfg.daysToKeep2 < 2628000) {
×
939
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
940
      mError("db:%s, failed to create, keep not match for audit db, %d", pCreate->db, dbObj.cfg.daysToKeep2);
×
941
      TAOS_RETURN(code);
×
942
    }
943
    if (dbObj.cfg.walLevel != 2) {
×
944
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
945
      mError("db:%s, failed to create, walLevel not match for audit db, %d", pCreate->db, dbObj.cfg.walLevel);
×
946
      TAOS_RETURN(code);
×
947
    }
948
  }
949

950
  mndSetDefaultDbCfg(&dbObj.cfg);
1,073,182✔
951

952
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
1,073,182✔
953
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, tstrerror(code));
×
954
    TAOS_RETURN(code);
×
955
  }
956

957
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
1,073,182✔
958
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, tstrerror(code));
703✔
959
    TAOS_RETURN(code);
703✔
960
  }
961

962
  if (dbObj.cfg.hashPrefix > 0) {
1,072,479✔
963
    int32_t dbLen = strlen(dbObj.name) + 1;
3,884✔
964
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
3,884✔
965
    dbObj.cfg.hashPrefix += dbLen;
3,884✔
966
  } else if (dbObj.cfg.hashPrefix < 0) {
1,068,595✔
967
    int32_t dbLen = strlen(dbObj.name) + 1;
4,599✔
968
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
4,599✔
969
    dbObj.cfg.hashPrefix -= dbLen;
4,599✔
970
  }
971

972
  SVgObj *pVgroups = NULL;
1,072,479✔
973
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
1,072,479✔
974
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
4,302✔
975
    TAOS_RETURN(code);
4,302✔
976
  }
977

978
  // add database privileges for user
979
  SUserObj *pNewUserDuped = NULL;
1,068,177✔
980
#if 0
981
  // We record the DB's owner id instread. The owner has the permission to create objects under the DB, but does not
982
  // have default permissions on objects created by others .Therefore, readDbs/writeDbs are no longer applicable.
983
  if (!pUser->superUser) {
984
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
985
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
986
                    NULL, _OVER);
987
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
988
                    NULL, _OVER);
989
    pNewUserDuped = &newUserObj;
990
  }
991
#endif
992

993
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
1,068,177✔
994
  if (pTrans == NULL) {
1,068,177✔
995
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
996
    if (terrno != 0) code = terrno;
×
997
    goto _OVER;
×
998
  }
999
  // mndTransSetSerial(pTrans);
1000
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
1,068,177✔
1001

1002
  mndTransSetDbName(pTrans, dbObj.name, NULL);
1,068,177✔
1003
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
1,068,177✔
1004

1005
  void   *pIter = NULL;
1,068,177✔
1006
  STrans *exitTrans = NULL;
1,068,177✔
1007

1008
  if (!pUser->superUser) {
1,068,177✔
1009
    while (1) {
1010
      pIter = sdbFetch(pMnode->pSdb, SDB_TRANS, pIter, (void **)&exitTrans);
550✔
1011
      if (pIter == NULL) break;
550✔
1012

1013
      if (exitTrans->conflict == TRN_CONFLICT_DB &&
×
1014
          strncmp(exitTrans->opername, "create-db", TSDB_TRANS_OPER_LEN) == 0) {
×
1015
        code = TSDB_CODE_MND_TRANS_CONFLICT;
×
1016
        sdbRelease(pMnode->pSdb, exitTrans);
×
1017
        sdbCancelFetch(pMnode->pSdb, pIter);
×
1018
        goto _OVER;
×
1019
      }
1020

1021
      sdbRelease(pMnode->pSdb, exitTrans);
×
1022
    }
1023
  }
1024

1025
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
1,068,177✔
1026
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
1,068,177✔
1027
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,068,177✔
1028
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,068,177✔
1029
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,068,177✔
1030
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
1,068,177✔
1031
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,068,177✔
1032
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,068,177✔
1033

1034
_OVER:
1,068,177✔
1035
  taosMemoryFree(pVgroups);
1,068,177✔
1036
  mndUserFreeObj(&newUserObj);
1,068,177✔
1037
  mndTransDrop(pTrans);
1,068,177✔
1038
  TAOS_RETURN(code);
1,068,177✔
1039
}
1040

1041
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
1,084,382✔
1042
  int32_t    code = 0;
1,084,382✔
1043
  SSdb      *pSdb = pMnode->pSdb;
1,084,382✔
1044
  SDnodeObj *pDnode = NULL;
1,084,382✔
1045
  void      *pIter = NULL;
1,084,382✔
1046

1047
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
1048
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
1,084,382✔
1049
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
×
1050
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
×
1051
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
1052
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
1053
           tsEncryptionKeyStat);
1054
    goto _exit;
×
1055
  }
1056

1057
  int64_t curMs = taosGetTimestampMs();
×
1058
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
×
1059
    bool online = false;
×
1060
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
×
1061
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
1062
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
1063
      mError("db:%s, failed to check encryption key:%" PRIi8
×
1064
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
1065
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
1066
             tsEncryptionKeyChksum);
1067
      sdbCancelFetch(pSdb, pIter);
×
1068
      sdbRelease(pSdb, pDnode);
×
1069
      break;
×
1070
    }
1071
    sdbRelease(pSdb, pDnode);
×
1072
  }
1073
#else
1074
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
1075
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
1076
    goto _exit;
1077
  }
1078
#endif
1079
_exit:
1,084,177✔
1080
  TAOS_RETURN(code);
1,084,382✔
1081
}
1082

1083
#ifndef TD_ENTERPRISE
1084
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
1085
  if (dnodeListStr[0] != 0) {
1086
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1087
    return terrno;
1088
  } else {
1089
    return 0;
1090
  }
1091
}
1092
#endif
1093

1094
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
1,176,659✔
1095
  SMnode      *pMnode = pReq->info.node;
1,176,659✔
1096
  int32_t      code = -1;
1,176,659✔
1097
  int32_t      lino = 0;
1,176,659✔
1098
  SDbObj      *pDb = NULL;
1,176,659✔
1099
  SUserObj    *pUser = NULL;
1,176,659✔
1100
  SCreateDbReq createReq = {0};
1,176,659✔
1101
  SArray      *dnodeList = NULL;
1,176,659✔
1102
  int64_t      tss = taosGetTimestampMs();
1,176,659✔
1103

1104
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
1,176,659✔
1105
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
1,176,659✔
1106

1107
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
1,176,659✔
1108
#ifdef WINDOWS
1109
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
1110
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
1111
    goto _OVER;
1112
  }
1113
#endif
1114
  mInfo("db:%s, start to create, vgroups:%d, isAudit:%d, daysToKeep2:%d", createReq.db, createReq.numOfVgroups, createReq.isAudit, createReq.daysToKeep2);
1,176,659✔
1115

1116
  pDb = mndAcquireDb(pMnode, createReq.db);
1,176,659✔
1117
  if (pDb != NULL) {
1,176,659✔
1118
    if (createReq.ignoreExist) {
89,901✔
1119
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
87,711✔
1120
      code = 0;
87,711✔
1121
      goto _OVER;
87,711✔
1122
    } else {
1123
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
2,190✔
1124
      goto _OVER;
2,190✔
1125
    }
1126
  } else {
1127
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
1,086,758✔
1128
      code = terrno;
2,376✔
1129
      goto _OVER;
2,376✔
1130
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
1,084,382✔
1131
      goto _OVER;
×
1132
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
1,084,382✔
1133
      // continue
1134
    } else {  // TSDB_CODE_APP_ERROR
1135
      goto _OVER;
×
1136
    }
1137
  }
1138

1139
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_CREATE_DB, NULL), &lino, _OVER);
1,084,382✔
1140

1141
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
1,084,382✔
1142

1143
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
1,084,382✔
1144
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
1,084,382✔
1145

1146
  if (createReq.replications == 2) {
1,084,382✔
1147
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
5,285✔
1148
  }
1149

1150
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
1,084,382✔
1151

1152
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
1,084,382✔
1153

1154
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pUser), &lino, _OVER);
1,073,182✔
1155

1156
  if (sdbGetSize(pMnode->pSdb, SDB_MOUNT) > 0) {
1,073,182✔
1157
    TAOS_CHECK_GOTO(TSDB_CODE_MND_MOUNT_NOT_EMPTY, &lino, _OVER);
×
1158
  }
1159

1160
  if (createReq.isAudit == 1) {
1,073,182✔
1161
    SDbObj *pAuditDb = mndAcquireAuditDb(pMnode);
×
1162
    if (pAuditDb != NULL) {
×
1163
      mndReleaseDb(pMnode, pAuditDb);
×
1164
      mError("db:%s, audit db already exist, %s", createReq.db, pAuditDb->name);
×
1165
      TAOS_CHECK_GOTO(TSDB_CODE_AUDIT_DB_ALREADY_EXIST, &lino, _OVER);
×
1166
    }
1167
  }
1168

1169
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
1,073,182✔
1170
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
1,068,177✔
1171

1172
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
1,068,177✔
1173
    SName name = {0};
1,068,177✔
1174
    if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
1,068,177✔
1175
      mError("db:%s, failed to parse db name", createReq.db);
×
1176
    int64_t tse = taosGetTimestampMs();
1,068,177✔
1177
    double  duration = (double)(tse - tss);
1,068,177✔
1178
    duration = duration / 1000;
1,068,177✔
1179
    auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen, duration, 0);
1,068,177✔
1180
  }
1181

1182
_OVER:
1,176,454✔
1183
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,176,659✔
1184
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
20,771✔
1185
  }
1186

1187
  mndReleaseDb(pMnode, pDb);
1,176,659✔
1188
  mndReleaseUser(pMnode, pUser);
1,176,659✔
1189
  tFreeSCreateDbReq(&createReq);
1,176,659✔
1190
  taosArrayDestroy(dnodeList);
1,176,659✔
1191

1192
  TAOS_RETURN(code);
1,176,659✔
1193
}
1194

1195
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
107,592✔
1196
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
107,592✔
1197

1198
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
107,592✔
1199
    pDb->cfg.buffer = pAlter->buffer;
4,335✔
1200
    code = 0;
4,335✔
1201
  }
1202

1203
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
107,592✔
1204
    pDb->cfg.pages = pAlter->pages;
2,181✔
1205
    code = 0;
2,181✔
1206
  }
1207

1208
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
107,592✔
1209
    pDb->cfg.pageSize = pAlter->pageSize;
×
1210
    code = 0;
×
1211
  }
1212

1213
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
107,592✔
1214
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
1215
    code = 0;
×
1216
  }
1217

1218
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
107,592✔
1219
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
14,542✔
1220
    code = 0;
14,542✔
1221
  }
1222

1223
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
107,592✔
1224
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
13,815✔
1225
    code = 0;
13,815✔
1226
  }
1227

1228
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
107,592✔
1229
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
10,959✔
1230
    code = 0;
10,959✔
1231
  }
1232

1233
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
107,592✔
1234
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
3,531✔
1235
    code = 0;
3,531✔
1236
  }
1237

1238
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
107,592✔
1239
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
8,616✔
1240
    code = 0;
8,616✔
1241
  }
1242

1243
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
107,592✔
1244
    pDb->cfg.walLevel = pAlter->walLevel;
8,567✔
1245
    code = 0;
8,567✔
1246
  }
1247

1248
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
107,592✔
1249
#if 1
1250
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1251
#else
1252
    pDb->cfg.strict = pAlter->strict;
1253
    code = 0;
1254
#endif
1255
  }
1256

1257
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
107,592✔
1258
    pDb->cfg.cacheLast = pAlter->cacheLast;
14,069✔
1259
    code = 0;
14,069✔
1260
  }
1261

1262
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
107,592✔
1263
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
2,314✔
1264
    code = 0;
2,314✔
1265
  }
1266

1267
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
107,592✔
1268
    pDb->cfg.replications = pAlter->replications;
6,369✔
1269
    pDb->vgVersion++;
6,369✔
1270
    code = 0;
6,369✔
1271
  }
1272

1273
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
107,592✔
1274
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
1,221✔
1275
    pDb->vgVersion++;
1,221✔
1276
    code = 0;
1,221✔
1277
  }
1278

1279
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
107,592✔
1280
    pDb->cfg.minRows = pAlter->minRows;
1,400✔
1281
    pDb->vgVersion++;
1,400✔
1282
    code = 0;
1,400✔
1283
  }
1284

1285
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
107,592✔
1286
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
11,977✔
1287
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
6,710✔
1288
    pDb->vgVersion++;
6,710✔
1289
    code = 0;
6,710✔
1290
  }
1291

1292
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
107,592✔
1293
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
1,427✔
1294
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
700✔
1295
    pDb->vgVersion++;
700✔
1296
    code = 0;
700✔
1297
  }
1298

1299
  if (pAlter->ssKeepLocal >= TSDB_MIN_SS_KEEP_LOCAL && pAlter->ssKeepLocal <= TSDB_MAX_SS_KEEP_LOCAL && pAlter->ssKeepLocal != pDb->cfg.ssKeepLocal) {
107,592✔
1300
    pDb->cfg.ssKeepLocal = pAlter->ssKeepLocal;
700✔
1301
    pDb->vgVersion++;
700✔
1302
    code = 0;
700✔
1303
  }
1304

1305
  if (pAlter->ssCompact >= TSDB_MIN_SS_COMPACT && pAlter->ssCompact <= TSDB_MAX_SS_COMPACT && pAlter->ssCompact != pDb->cfg.ssCompact) {
107,592✔
1306
    pDb->cfg.ssCompact = pAlter->ssCompact;
700✔
1307
    pDb->vgVersion++;
700✔
1308
    code = 0;
700✔
1309
  }
1310

1311
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
107,592✔
1312
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
700✔
1313
    pDb->vgVersion++;
700✔
1314
    code = 0;
700✔
1315
  }
1316

1317
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
107,592✔
1318
    pDb->cfg.compactInterval = pAlter->compactInterval;
4,795✔
1319
    pDb->vgVersion++;
4,795✔
1320
    code = 0;
4,795✔
1321
  }
1322

1323
  bool compactTimeRangeChanged = false;
107,592✔
1324
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
107,592✔
1325
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
105,507✔
1326
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
105,507✔
1327
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
4,074✔
1328
    compactTimeRangeChanged = true;
4,074✔
1329
    code = 0;
4,074✔
1330
  }
1331

1332
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
107,592✔
1333
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
105,507✔
1334
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
4,074✔
1335
    compactTimeRangeChanged = true;
4,074✔
1336
    code = 0;
4,074✔
1337
  }
1338
  if(compactTimeRangeChanged) {
107,592✔
1339
    pDb->vgVersion++;
4,074✔
1340
  }
1341

1342
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
107,592✔
1343
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
8,217✔
1344
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
3,416✔
1345
    pDb->vgVersion++;
3,416✔
1346
    code = 0;
3,416✔
1347
  }
1348

1349
  if (pAlter->isAudit != pDb->cfg.isAudit) {
107,592✔
1350
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1351
    return code;
×
1352
  }
1353

1354
  TAOS_RETURN(code);
107,592✔
1355
}
1356

1357
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
81,437✔
1358
  int32_t  code = 0;
81,437✔
1359
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
81,437✔
1360
  if (pRedoRaw == NULL) {
81,437✔
1361
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1362
    if (terrno != 0) code = terrno;
×
1363
    TAOS_RETURN(code);
×
1364
  }
1365
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
81,437✔
1366
    sdbFreeRaw(pRedoRaw);
×
1367
    TAOS_RETURN(code);
×
1368
  }
1369

1370
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
81,437✔
1371
    sdbFreeRaw(pRedoRaw);
×
1372
    TAOS_RETURN(code);
×
1373
  }
1374
  return 0;
81,437✔
1375
}
1376

1377
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
81,437✔
1378
  int32_t  code = 0;
81,437✔
1379
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
81,437✔
1380
  if (pCommitRaw == NULL) {
81,437✔
1381
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1382
    if (terrno != 0) code = terrno;
×
1383
    TAOS_RETURN(code);
×
1384
  }
1385
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
81,437✔
1386
    sdbFreeRaw(pCommitRaw);
×
1387
    TAOS_RETURN(code);
×
1388
  }
1389

1390
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
81,437✔
1391
    sdbFreeRaw(pCommitRaw);
×
1392
    TAOS_RETURN(code);
×
1393
  }
1394
  TAOS_RETURN(code);
81,437✔
1395
}
1396

1397
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
81,437✔
1398
  int32_t code = 0, lino = 0;
81,437✔
1399
  SSdb   *pSdb = pMnode->pSdb;
81,437✔
1400
  void   *pIter = NULL;
81,437✔
1401
  SVgObj *pVgroup = NULL;
81,437✔
1402
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
81,437✔
1403

1404
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
81,437✔
1405

1406
  while (1) {
1407
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
560,346✔
1408
    if (pIter == NULL) break;
560,346✔
1409

1410
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
479,586✔
1411
      SVgObj newVgroup = {0};
205,562✔
1412
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
205,562✔
1413
                      _err);
1414

1415
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
204,885✔
1416
        if (pNewDb->cfg.withArbitrator) {
×
1417
          SArbGroup arbGroup = {0};
×
1418
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1419
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1420
        } else {
1421
          SArbGroup arbGroup = {0};
×
1422
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
1423
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1424
        }
1425
      }
1426
    }
1427

1428
    sdbRelease(pSdb, pVgroup);
478,909✔
1429
  }
1430

1431
  taosArrayDestroy(pArray);
80,760✔
1432
  TAOS_RETURN(code);
80,760✔
1433

1434
_err:
677✔
1435
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
677✔
1436

1437
  sdbCancelFetch(pSdb, pIter);
677✔
1438
  sdbRelease(pSdb, pVgroup);
677✔
1439
  taosArrayDestroy(pArray);
677✔
1440
  TAOS_RETURN(code);
677✔
1441
}
1442

1443
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
82,328✔
1444
  int32_t code = -1;
82,328✔
1445
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
82,328✔
1446
  if (pTrans == NULL) {
82,328✔
1447
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1448
    if (terrno != 0) code = terrno;
×
1449
    TAOS_RETURN(code);
×
1450
  }
1451
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
82,328✔
1452
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
82,328✔
1453

1454
  mndTransSetDbName(pTrans, pOld->name, NULL);
82,328✔
1455
  mndTransSetGroupParallel(pTrans);
82,328✔
1456
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
82,328✔
1457
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
81,518✔
1458
  TAOS_CHECK_GOTO(mndTransCheckConflictWithRetention(pMnode, pTrans), NULL, _OVER);
81,437✔
1459

1460
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
81,437✔
1461
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
81,437✔
1462
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
81,437✔
1463

1464
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
80,760✔
1465
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
80,760✔
1466
  code = 0;
80,760✔
1467

1468
_OVER:
82,328✔
1469
  mndTransDrop(pTrans);
82,328✔
1470
  TAOS_RETURN(code);
82,328✔
1471
}
1472

1473
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
110,754✔
1474
  SMnode     *pMnode = pReq->info.node;
110,754✔
1475
  int32_t     code = -1;
110,754✔
1476
  SDbObj     *pDb = NULL;
110,754✔
1477
  SAlterDbReq alterReq = {0};
110,754✔
1478
  SDbObj      dbObj = {0};
110,754✔
1479
  int64_t     tss = taosGetTimestampMs();
110,754✔
1480

1481
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
110,754✔
1482

1483
  mInfo("db:%s, start to alter, encryt_algr:%s", alterReq.db, alterReq.encryptAlgrName);
110,754✔
1484

1485
  pDb = mndAcquireDb(pMnode, alterReq.db);
110,754✔
1486
  if (pDb == NULL) {
110,754✔
1487
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
1,008✔
1488
    if (terrno != 0) code = terrno;
1,008✔
1489
    goto _OVER;
1,008✔
1490
  }
1491

1492
  if (pDb->cfg.isAudit == 1) {
109,746✔
1493
    if (alterReq.daysToKeep2 < 2628000) {
×
1494
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
1495
      mError("db:%s, failed to alter, keep not match for audit db, %d", alterReq.db, alterReq.daysToKeep2);
×
1496
      TAOS_RETURN(code);
×
1497
    }
1498
    if (alterReq.walLevel != 2) {
×
1499
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
1500
      mError("db:%s, failed to alter, walLevel not match for audit db, %d", alterReq.db, alterReq.walLevel);
×
1501
      TAOS_RETURN(code);
×
1502
    }
1503
    if (alterReq.isAudit == 0) {
×
1504
      code = TSDB_CODE_AUDIT_DB_NOT_ALLOW_CHANGE;
×
1505
      mError("db:%s, failed to alter, is not allowed to change audit db, %d", alterReq.db, alterReq.isAudit);
×
1506
      TAOS_RETURN(code);
×
1507
    }
1508
  }
1509

1510
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_ALTER_DB, pDb), NULL, _OVER);
109,746✔
1511

1512
  if (alterReq.replications == 2) {
109,746✔
1513
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
700✔
1514
  }
1515

1516
  int32_t numOfTopics = 0;
109,746✔
1517
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
109,746✔
1518

1519
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
109,746✔
1520
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
1521
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1522
           numOfTopics);
1523
    goto _OVER;
×
1524
  }
1525

1526
  if (pDb->cfg.isMount) {
109,746✔
1527
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
1528
    goto _OVER;
×
1529
  }
1530

1531
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
109,746✔
1532
  if (dbObj.cfg.pRetensions != NULL) {
109,746✔
1533
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
1534
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1535
  }
1536

1537
  if (strlen(alterReq.encryptAlgrName) > 0) {
109,746✔
1538
    mInfo("db:%s, check encryt algr:%s", alterReq.db, alterReq.encryptAlgrName);
2,881✔
1539

1540
    if (strncasecmp(alterReq.encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
2,881✔
1541
      if (pDb->cfg.encryptAlgorithm != 0) {
727✔
1542
        code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
×
1543
        mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
×
1544
        goto _OVER;
×
1545
      }
1546
    } else {
1547
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, alterReq.encryptAlgrName);
2,154✔
1548
      if (pEncryptAlgr != NULL) {
2,154✔
1549
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
1,427✔
1550
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
1551
          mError("db:%s, failed to alter, encrypt algorithm not match, %s, expect type %d", alterReq.db,
×
1552
                 alterReq.encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
1553
          goto _OVER;
×
1554
        }
1555
        if (pEncryptAlgr->id != pDb->cfg.encryptAlgorithm) {
1,427✔
1556
          code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
1,427✔
1557
          mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
1,427✔
1558
          goto _OVER;
1,427✔
1559
        }
1560
      } else {
1561
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
727✔
1562
        mError("db:%s, failed to alter, encrypt algorithm not exist, %s", alterReq.db, alterReq.encryptAlgrName);
727✔
1563
        goto _OVER;
727✔
1564
      }
1565
    }
1566
  }
1567

1568
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
107,592✔
1569
  if (code != 0) {
107,592✔
1570
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
23,864✔
1571
    goto _OVER;
23,864✔
1572
  }
1573

1574
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
83,728✔
1575

1576
  dbObj.cfgVersion++;
82,328✔
1577
  dbObj.updateTime = taosGetTimestampMs();
82,328✔
1578
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
82,328✔
1579

1580
  if (dbObj.cfg.replications != pDb->cfg.replications) {
82,328✔
1581
    // return quickly, operation executed asynchronously
1582
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
5,669✔
1583
  } else {
1584
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
76,659✔
1585
  }
1586

1587
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
82,328✔
1588
    SName name = {0};
82,328✔
1589
    if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
82,328✔
1590
      mError("db:%s, failed to parse db name", alterReq.db);
×
1591

1592
    int64_t tse = taosGetTimestampMs();
82,328✔
1593
    double  duration = (double)(tse - tss);
82,328✔
1594
    duration = duration / 1000;
82,328✔
1595
    auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen, duration, 0);
82,328✔
1596
  }
1597

1598
_OVER:
110,754✔
1599
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
110,754✔
1600
    if (terrno != 0) code = terrno;
6,130✔
1601
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
6,130✔
1602
  }
1603

1604
  mndReleaseDb(pMnode, pDb);
110,754✔
1605
  taosArrayDestroy(dbObj.cfg.pRetensions);
110,754✔
1606
  tFreeSAlterDbReq(&alterReq);
110,754✔
1607

1608
  TAOS_RETURN(code);
110,754✔
1609
}
1610

1611
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb, char *algorithmsId) {
3,587,990✔
1612
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
3,587,990✔
1613
  cfgRsp->dbId = pDb->uid;
3,587,990✔
1614
  cfgRsp->ownerId = pDb->ownerId;
3,586,590✔
1615
  cfgRsp->cfgVersion = pDb->cfgVersion;
3,587,990✔
1616
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
3,587,290✔
1617
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
3,587,290✔
1618
  cfgRsp->buffer = pDb->cfg.buffer;
3,587,290✔
1619
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
3,586,590✔
1620
  cfgRsp->pageSize = pDb->cfg.pageSize;
3,587,290✔
1621
  cfgRsp->pages = pDb->cfg.pages;
3,587,990✔
1622
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
3,587,990✔
1623
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
3,587,990✔
1624
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
3,587,290✔
1625
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
3,587,290✔
1626
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
3,587,290✔
1627
  cfgRsp->minRows = pDb->cfg.minRows;
3,587,290✔
1628
  cfgRsp->maxRows = pDb->cfg.maxRows;
3,587,290✔
1629
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
3,586,590✔
1630
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
3,587,290✔
1631
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
3,587,990✔
1632
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
3,587,990✔
1633
  cfgRsp->walLevel = pDb->cfg.walLevel;
3,587,290✔
1634
  cfgRsp->precision = pDb->cfg.precision;
3,586,590✔
1635
  cfgRsp->compression = pDb->cfg.compression;
3,587,290✔
1636
  cfgRsp->replications = pDb->cfg.replications;
3,587,290✔
1637
  cfgRsp->strict = pDb->cfg.strict;
3,587,290✔
1638
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
3,587,990✔
1639
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
3,587,290✔
1640
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
3,587,290✔
1641
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
3,586,590✔
1642
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
3,587,990✔
1643
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
3,586,590✔
1644
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
3,586,590✔
1645
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
3,587,990✔
1646
  cfgRsp->schemaless = pDb->cfg.schemaless;
3,587,290✔
1647
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
3,587,290✔
1648
  cfgRsp->ssChunkSize = pDb->cfg.ssChunkSize;
3,587,290✔
1649
  cfgRsp->ssKeepLocal = pDb->cfg.ssKeepLocal;
3,586,590✔
1650
  cfgRsp->ssCompact = pDb->cfg.ssCompact;
3,587,290✔
1651
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
3,587,290✔
1652
  // cfgRsp->encryptAlgr = pDb->cfg.encryptAlgorithm;
1653
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
3,587,990✔
1654
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
3,587,990✔
1655
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
3,587,290✔
1656
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
3,587,990✔
1657
  cfgRsp->flags = pDb->cfg.flags;
3,587,290✔
1658
  tstrncpy(cfgRsp->algorithmsId, algorithmsId, sizeof(cfgRsp->algorithmsId));
3,587,290✔
1659
  cfgRsp->isAudit = pDb->cfg.isAudit;
3,587,990✔
1660
}
3,587,990✔
1661

1662
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
6,935,937✔
1663
  SMnode   *pMnode = pReq->info.node;
6,935,937✔
1664
  int32_t   code = -1;
6,936,419✔
1665
  SDbObj   *pDb = NULL;
6,936,419✔
1666
  SDbCfgReq cfgReq = {0};
6,936,419✔
1667
  SDbCfgRsp cfgRsp = {0};
6,936,419✔
1668

1669
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
6,935,937✔
1670

1671
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
6,936,419✔
1672
    pDb = mndAcquireDb(pMnode, cfgReq.db);
6,935,237✔
1673
    if (pDb == NULL) {
6,935,937✔
1674
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
3,391,981✔
1675
      if (terrno != 0) code = terrno;
3,391,981✔
1676
      goto _OVER;
3,391,499✔
1677
    }
1678

1679
    char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
3,543,956✔
1680
    mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
3,544,438✔
1681
    mndDumpDbCfgInfo(&cfgRsp, pDb, algrId);
3,543,738✔
1682
  }
1683

1684
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
3,545,620✔
1685
  void   *pRsp = rpcMallocCont(contLen);
3,543,733✔
1686
  if (pRsp == NULL) {
3,544,438✔
1687
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1688
    if (terrno != 0) code = terrno;
×
1689
    goto _OVER;
×
1690
  }
1691

1692
  int32_t ret = 0;
3,544,438✔
1693
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
3,544,438✔
1694
    code = ret;
×
1695
    goto _OVER;
×
1696
  }
1697

1698
  pReq->info.rsp = pRsp;
3,544,438✔
1699
  pReq->info.rspLen = contLen;
3,544,438✔
1700

1701
  code = 0;
3,543,738✔
1702

1703
_OVER:
6,935,237✔
1704

1705
  tFreeSDbCfgRsp(&cfgRsp);
6,935,932✔
1706

1707
  if (code != 0) {
6,935,237✔
1708
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
3,391,499✔
1709
  }
1710

1711
  mndReleaseDb(pMnode, pDb);
6,935,237✔
1712

1713
  TAOS_RETURN(code);
6,936,419✔
1714
}
1715

1716
int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
599,218✔
1717
  int32_t  code = 0;
599,218✔
1718
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
599,218✔
1719
  if (pRedoRaw == NULL) {
599,218✔
1720
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1721
    if (terrno != 0) code = terrno;
×
1722
    return -1;
×
1723
  }
1724
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
599,218✔
1725
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
599,218✔
1726

1727
  SSdb *pSdb = pMnode->pSdb;
599,218✔
1728
  void *pIter = NULL;
599,218✔
1729

1730
  while (1) {
4,200✔
1731
    SArbGroup *pArbGroup = NULL;
603,418✔
1732
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
603,418✔
1733
    if (pIter == NULL) break;
603,418✔
1734

1735
    if (pArbGroup->dbUid == pDb->uid) {
4,200✔
1736
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
4,200✔
1737
        sdbCancelFetch(pSdb, pIter);
×
1738
        sdbRelease(pSdb, pArbGroup);
×
1739
        TAOS_RETURN(code);
×
1740
      }
1741
    }
1742

1743
    sdbRelease(pSdb, pArbGroup);
4,200✔
1744
  }
1745

1746
  TAOS_RETURN(code);
599,218✔
1747
}
1748

1749
int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
599,218✔
1750
  int32_t  code = 0;
599,218✔
1751
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
599,218✔
1752
  if (pCommitRaw == NULL) {
599,218✔
1753
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1754
    if (terrno != 0) code = terrno;
×
1755
    return -1;
×
1756
  }
1757
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
599,218✔
1758
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
599,218✔
1759

1760
  SSdb *pSdb = pMnode->pSdb;
599,218✔
1761
  void *pIter = NULL;
599,218✔
1762

1763
  while (1) {
4,200✔
1764
    SArbGroup *pArbGroup = NULL;
603,418✔
1765
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
603,418✔
1766
    if (pIter == NULL) break;
603,418✔
1767

1768
    if (pArbGroup->dbUid == pDb->uid) {
4,200✔
1769
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
4,200✔
1770
        sdbCancelFetch(pSdb, pIter);
×
1771
        sdbRelease(pSdb, pArbGroup);
×
1772
        TAOS_RETURN(code);
×
1773
      }
1774
    }
1775

1776
    sdbRelease(pSdb, pArbGroup);
4,200✔
1777
  }
1778

1779
  while (1) {
3,104,931✔
1780
    SVgObj *pVgroup = NULL;
3,704,149✔
1781
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,704,149✔
1782
    if (pIter == NULL) break;
3,704,149✔
1783

1784
    if (pVgroup->dbUid == pDb->uid) {
3,104,931✔
1785
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
1,278,846✔
1786
      if (pVgRaw == NULL) {
1,278,846✔
1787
        sdbCancelFetch(pSdb, pIter);
×
1788
        sdbRelease(pSdb, pVgroup);
×
1789
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1790
        if (terrno != 0) code = terrno;
×
1791
        TAOS_RETURN(code);
×
1792
      }
1793
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
1,278,846✔
1794
        sdbCancelFetch(pSdb, pIter);
×
1795
        sdbRelease(pSdb, pVgroup);
×
1796
        TAOS_RETURN(code);
×
1797
      }
1798
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
1,278,846✔
1799
        sdbCancelFetch(pSdb, pIter);
×
1800
        sdbRelease(pSdb, pVgroup);
×
1801
        TAOS_RETURN(code);
×
1802
      }
1803
    }
1804

1805
    sdbRelease(pSdb, pVgroup);
3,104,931✔
1806
  }
1807

1808
  while (1) {
915,964✔
1809
    SStbObj *pStb = NULL;
1,515,182✔
1810
    ESdbStatus status;
1,515,084✔
1811
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
1,515,182✔
1812
    if (pIter == NULL) break;
1,515,182✔
1813

1814
    if (pStb->dbUid == pDb->uid) {
915,964✔
1815
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
507,112✔
1816
      if (pStbRaw == NULL) {
507,112✔
1817
        sdbCancelFetch(pSdb, pIter);
×
1818
        sdbRelease(pSdb, pStbRaw);
×
1819
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1820
        if (terrno != 0) code = terrno;
×
1821
        return -1;
×
1822
      }
1823
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
507,112✔
1824
        sdbCancelFetch(pSdb, pIter);
×
1825
        sdbRelease(pSdb, pStbRaw);
×
1826
        return -1;
×
1827
      }
1828
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
507,112✔
1829
        sdbCancelFetch(pSdb, pIter);
×
1830
        sdbRelease(pSdb, pStbRaw);
×
1831
        return -1;
×
1832
      }
1833
    }
1834

1835
    sdbRelease(pSdb, pStb);
915,964✔
1836
  }
1837

1838
  TAOS_RETURN(code);
599,218✔
1839
}
1840

1841
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
1,278,846✔
1842
  int32_t code = 0;
1,278,846✔
1843
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
2,622,132✔
1844
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
1,343,286✔
1845
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
1,343,286✔
1846
  }
1847

1848
  TAOS_RETURN(code);
1,278,846✔
1849
}
1850

1851
int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
599,218✔
1852
  int32_t code = 0;
599,218✔
1853
  SSdb   *pSdb = pMnode->pSdb;
599,218✔
1854
  void   *pIter = NULL;
599,218✔
1855

1856
  while (1) {
3,104,931✔
1857
    SVgObj *pVgroup = NULL;
3,704,149✔
1858
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,704,149✔
1859
    if (pIter == NULL) break;
3,704,149✔
1860

1861
    if (pVgroup->dbUid == pDb->uid) {
3,104,931✔
1862
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
1,278,846✔
1863
        sdbCancelFetch(pSdb, pIter);
×
1864
        sdbRelease(pSdb, pVgroup);
×
1865
        TAOS_RETURN(code);
×
1866
      }
1867
    }
1868

1869
    sdbRelease(pSdb, pVgroup);
3,104,931✔
1870
  }
1871

1872
  TAOS_RETURN(code);
599,218✔
1873
}
1874

1875
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
1,004,164✔
1876
  int32_t    code = 0;
1,004,164✔
1877
  SDropDbRsp dropRsp = {0};
1,004,164✔
1878
  if (pDb != NULL) {
1,004,164✔
1879
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
598,712✔
1880
    dropRsp.uid = pDb->uid;
598,712✔
1881
  }
1882

1883
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
1,004,164✔
1884
  void   *pRsp = NULL;
1,004,164✔
1885
  if (useRpcMalloc) {
1,004,164✔
1886
    pRsp = rpcMallocCont(rspLen);
405,452✔
1887
  } else {
1888
    pRsp = taosMemoryMalloc(rspLen);
598,712✔
1889
  }
1890

1891
  if (pRsp == NULL) {
1,004,164✔
1892
    code = terrno;
×
1893
    TAOS_RETURN(code);
×
1894
  }
1895

1896
  int32_t ret = 0;
1,004,164✔
1897
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
1,004,164✔
1898
  *pRspLen = rspLen;
1,004,164✔
1899
  *ppRsp = pRsp;
1,004,164✔
1900
  TAOS_RETURN(code);
1,004,164✔
1901
}
1902

1903
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
599,053✔
1904
  int32_t code = -1;
599,053✔
1905

1906
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
599,053✔
1907
  if (pTrans == NULL) {
599,053✔
1908
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1909
    if (terrno != 0) code = terrno;
×
1910
    goto _OVER;
×
1911
  }
1912

1913
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
599,053✔
1914

1915
  mndTransSetDbName(pTrans, pDb->name, NULL);
599,053✔
1916
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
599,053✔
1917

1918
  if (mndTopicExistsForDb(pMnode, pDb)) {
599,053✔
1919
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
341✔
1920
    goto _OVER;
341✔
1921
  }
1922

1923
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1924
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1925
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1926
#ifdef TD_ENTERPRISE
1927
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1928
#endif
1929
  TAOS_CHECK_GOTO(mndDropTSMAsByDb(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1930
  TAOS_CHECK_GOTO(mndDropRsmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1931
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1932
  //TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
1933
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
598,712✔
1934
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb, NULL), NULL, _OVER);
598,712✔
1935

1936
  int32_t rspLen = 0;
598,712✔
1937
  void   *pRsp = NULL;
598,712✔
1938
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
598,712✔
1939
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
598,712✔
1940

1941
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
598,712✔
1942
  code = 0;
598,712✔
1943

1944
_OVER:
599,053✔
1945
  mndTransDrop(pTrans);
599,053✔
1946
  TAOS_RETURN(code);
599,053✔
1947
}
1948

1949
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
1,010,176✔
1950
  SMnode    *pMnode = pReq->info.node;
1,010,176✔
1951
  int32_t    code = -1;
1,010,176✔
1952
  SDbObj    *pDb = NULL;
1,010,176✔
1953
  SUserObj  *pUser = NULL;
1,010,176✔
1954
  SDropDbReq dropReq = {0};
1,010,176✔
1955
  int64_t    tss = taosGetTimestampMs();
1,010,176✔
1956

1957
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
1,010,176✔
1958

1959
  mInfo("db:%s, start to drop", dropReq.db);
1,010,176✔
1960

1961
  pDb = mndAcquireDb(pMnode, dropReq.db);
1,010,176✔
1962
  if (pDb == NULL) {
1,010,176✔
1963
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
409,128✔
1964
    if (terrno != 0) code = terrno;
409,128✔
1965
    if (dropReq.ignoreNotExists) {
409,128✔
1966
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
405,452✔
1967
    }
1968
    goto _OVER;
409,128✔
1969
  }
1970

1971
  if ((code = mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_DROP_DB, pDb))) {
601,048✔
1972
    if ((code = mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pUser))) {
×
1973
      goto _OVER;
×
1974
    }
1975
    char objFName[TSDB_PRIV_MAX_KEY_LEN] = {0};
×
1976
    (void)snprintf(objFName, sizeof(objFName), "%d.*", pUser->acctId);
×
1977
    if ((code = mndCheckSysObjPrivilege(pMnode, pUser, PRIV_CM_DROP, PRIV_OBJ_DB, 0, objFName, NULL))) {
×
1978
      goto _OVER;
×
1979
    }
1980
  }
1981

1982
  if(pDb->cfg.isMount) {
601,048✔
1983
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
506✔
1984
    goto _OVER;
506✔
1985
  }
1986

1987
  SSdb *pSdb = pMnode->pSdb;
600,542✔
1988
  void *pIter = NULL;
600,542✔
1989

1990
  while (1) {
3,116,213✔
1991
    SVgObj *pVgroup = NULL;
3,716,755✔
1992
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,716,755✔
1993
    if (pIter == NULL) break;
3,716,755✔
1994

1995
    if (pVgroup->dbUid == pDb->uid) {
3,116,213✔
1996
      bool isFound = false;
1,279,664✔
1997
      for (int32_t i = 0; i < pVgroup->replica; i++) {
2,623,768✔
1998
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
1,344,104✔
1999
          isFound = true;
×
2000
          break;
×
2001
        }
2002
      }
2003
      if (!isFound) {
1,279,664✔
2004
        sdbRelease(pSdb, pVgroup);
1,279,664✔
2005
        continue;
1,279,664✔
2006
      }
2007
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
×
2008
      sdbCancelFetch(pSdb, pIter);
×
2009
      sdbRelease(pSdb, pVgroup);
×
2010
      goto _OVER;
×
2011
    }
2012

2013
    sdbRelease(pSdb, pVgroup);
1,836,549✔
2014
  }
2015

2016
  bool dbStream = false;
600,542✔
2017
  bool vtableStream = false;
600,542✔
2018
  mstCheckDbInUse(pMnode, dropReq.db, &dbStream, &vtableStream, true);
600,542✔
2019
  if (dbStream) {
600,542✔
2020
    code = TSDB_CODE_MND_STREAM_DB_IN_USE;
1,278✔
2021
    mError("db:%s used by streams, drop db not allowed", dropReq.db);
1,278✔
2022
    goto _OVER;
1,278✔
2023
  }
2024

2025
  if (vtableStream && !dropReq.force) {
599,264✔
2026
    code = TSDB_CODE_MND_STREAM_VTABLE_EXITS;
211✔
2027
    mError("db:%s, vtable stream exists, drop db not allowed", dropReq.db);
211✔
2028
    goto _OVER;
211✔
2029
  }
2030

2031
  code = mndDropDb(pMnode, pReq, pDb);
599,053✔
2032
  if (code == TSDB_CODE_SUCCESS) {
599,053✔
2033
    code = TSDB_CODE_ACTION_IN_PROGRESS;
598,712✔
2034
  }
2035

2036
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
599,053✔
2037
    SName name = {0};
599,053✔
2038
    if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
599,053✔
2039
      mError("db:%s, failed to parse db name", dropReq.db);
×
2040

2041
    int64_t tse = taosGetTimestampMs();
599,053✔
2042
    double  duration = (double)(tse - tss);
599,053✔
2043
    duration = duration / 1000;
599,053✔
2044
    auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen, duration, 0);
599,053✔
2045
  }
2046

2047
_OVER:
1,010,091✔
2048
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,010,176✔
2049
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
6,012✔
2050
  }
2051

2052
  mndReleaseUser(pMnode, pUser);
1,010,176✔
2053
  mndReleaseDb(pMnode, pDb);
1,010,176✔
2054
  tFreeSDropDbReq(&dropReq);
1,010,176✔
2055
  TAOS_RETURN(code);
1,010,176✔
2056
}
2057

2058
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
21,043,195✔
2059
  int32_t numOfTables = 0;
21,043,195✔
2060
  int32_t vindex = 0;
21,043,195✔
2061
  SSdb   *pSdb = pMnode->pSdb;
21,043,195✔
2062

2063
  void *pIter = NULL;
21,043,195✔
2064
  while (vindex < pDb->cfg.numOfVgroups) {
84,485,375✔
2065
    SVgObj *pVgroup = NULL;
63,442,180✔
2066
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
63,442,180✔
2067
    if (pIter == NULL) break;
63,442,180✔
2068

2069
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
63,442,180✔
2070
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
41,726,402✔
2071
      vindex++;
41,726,444✔
2072
    }
2073

2074
    sdbRelease(pSdb, pVgroup);
63,442,222✔
2075
  }
2076

2077
  sdbCancelFetch(pSdb, pIter);
21,043,190✔
2078
  return numOfTables;
21,043,195✔
2079
}
2080

2081
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
21,882,321✔
2082
  int32_t vindex = 0;
21,882,321✔
2083
  SSdb   *pSdb = pMnode->pSdb;
21,882,321✔
2084

2085
  void *pIter = NULL;
21,882,321✔
2086
  while (1) {
46,164,175✔
2087
    SVgObj *pVgroup = NULL;
68,046,496✔
2088
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
68,046,496✔
2089
    if (pIter == NULL) break;
68,046,496✔
2090

2091
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
62,474,824✔
2092
      SVgroupInfo vgInfo = {0};
50,491,015✔
2093
      vgInfo.vgId = pVgroup->vgId;
50,491,015✔
2094
      vgInfo.hashBegin = pVgroup->hashBegin;
50,491,015✔
2095
      vgInfo.hashEnd = pVgroup->hashEnd;
50,491,015✔
2096
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
50,491,015✔
2097
      vgInfo.epSet.numOfEps = pVgroup->replica;
50,491,015✔
2098
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
104,861,772✔
2099
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
54,370,757✔
2100
        SEp       *pEp = &vgInfo.epSet.eps[gid];
54,370,757✔
2101
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
54,370,757✔
2102
        if (pDnode != NULL) {
54,370,658✔
2103
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
54,343,168✔
2104
          pEp->port = pDnode->port;
54,343,168✔
2105
        }
2106
        mndReleaseDnode(pMnode, pDnode);
54,370,757✔
2107
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
54,370,757✔
2108
          vgInfo.epSet.inUse = gid;
50,065,131✔
2109
        }
2110
      }
2111
      vindex++;
50,491,015✔
2112
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
50,491,015✔
2113
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
2114
      }
2115
    }
2116

2117
    sdbRelease(pSdb, pVgroup);
62,474,824✔
2118

2119
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
62,474,824✔
2120
      sdbCancelFetch(pSdb, pIter);
16,310,649✔
2121
      break;
16,310,649✔
2122
    }
2123
  }
2124
}
21,882,321✔
2125

2126
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
16,416,219✔
2127
  int32_t code = 0;
16,416,219✔
2128
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
16,416,219✔
2129
  if (pRsp->pVgroupInfos == NULL) {
16,416,219✔
2130
    code = terrno;
×
2131
    TAOS_RETURN(code);
×
2132
  }
2133

2134
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
16,416,219✔
2135

2136
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
16,416,219✔
2137
      pReq->stateTs < pDb->stateTs) {
1,040,970✔
2138
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
15,491,160✔
2139
  }
2140

2141
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
16,415,666✔
2142
  pRsp->uid = pDb->uid;
16,416,219✔
2143
  pRsp->vgVersion = pDb->vgVersion;
16,416,219✔
2144
  pRsp->stateTs = pDb->stateTs;
16,415,624✔
2145
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
16,415,666✔
2146
  pRsp->hashMethod = pDb->cfg.hashMethod;
16,415,666✔
2147
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
16,416,177✔
2148
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
16,415,666✔
2149
  pRsp->flags = pDb->cfg.flags;
16,415,666✔
2150
  TAOS_RETURN(code);
16,415,666✔
2151
}
2152

2153
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
21,841,297✔
2154
  SMnode   *pMnode = pReq->info.node;
21,841,297✔
2155
  int32_t   code = -1;
21,841,297✔
2156
  SDbObj   *pDb = NULL;
21,841,297✔
2157
  SUseDbReq usedbReq = {0};
21,841,297✔
2158
  SUseDbRsp usedbRsp = {0};
21,841,297✔
2159
  SUserObj *pOperUser = NULL;
21,841,255✔
2160

2161
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
21,841,255✔
2162

2163
  char *p = strchr(usedbReq.db, '.');
21,841,297✔
2164
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
21,841,297✔
2165
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
5,339,294✔
2166
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
5,339,294✔
2167
    if (usedbReq.vgVersion < vgVersion) {
5,339,290✔
2168
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
5,335,852✔
2169
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
5,335,852✔
2170

2171
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
5,335,852✔
2172
      usedbRsp.vgVersion = vgVersion++;
5,335,852✔
2173
    } else {
2174
      usedbRsp.vgVersion = usedbReq.vgVersion;
3,438✔
2175
    }
2176
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
5,339,290✔
2177
    code = 0;
5,339,290✔
2178
  } else {
2179
    pDb = mndAcquireDb(pMnode, usedbReq.db);
16,502,011✔
2180
    if (pDb == NULL) {
16,502,007✔
2181
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
85,788✔
2182
      usedbRsp.uid = usedbReq.dbId;
85,788✔
2183
      usedbRsp.vgVersion = usedbReq.vgVersion;
85,788✔
2184
      usedbRsp.errCode = terrno;
85,788✔
2185

2186
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
85,788✔
2187
      if (terrno != 0) code = terrno;
85,788✔
2188
      goto _OVER;
85,788✔
2189
    } else {
2190
      TAOS_CHECK_GOTO(mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pOperUser), NULL, _OVER);
16,416,219✔
2191
      TAOS_CHECK_GOTO(mndCheckDbPrivilegeByNameRecF(pMnode, pOperUser, PRIV_DB_USE, PRIV_OBJ_DB, usedbReq.db, NULL), NULL, _OVER);
16,416,219✔
2192

2193
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
16,416,219✔
2194

2195
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
16,416,219✔
2196
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
2197
      code = 0;
16,416,219✔
2198
    }
2199
  }
2200

2201
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
21,755,509✔
2202
  void   *pRsp = rpcMallocCont(contLen);
21,755,469✔
2203
  if (pRsp == NULL) {
21,755,509✔
2204
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2205
    code = -1;
×
2206
    goto _OVER;
×
2207
  }
2208

2209
  int32_t ret = 0;
21,755,509✔
2210
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
21,755,509✔
2211
    code = ret;
×
2212
    goto _OVER;
×
2213
  }
2214

2215
  pReq->info.rsp = pRsp;
21,755,509✔
2216
  pReq->info.rspLen = contLen;
21,754,956✔
2217

2218
_OVER:
21,840,044✔
2219
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
21,841,297✔
2220
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
85,788✔
2221
  }
2222

2223
  mndReleaseDb(pMnode, pDb);
21,841,297✔
2224
  mndReleaseUser(pMnode, pOperUser);
21,841,297✔
2225
  tFreeSUsedbRsp(&usedbRsp);
21,841,297✔
2226

2227
  TAOS_RETURN(code);
21,840,739✔
2228
}
2229

2230
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
5,515,832✔
2231
  int32_t       code = 0;
5,515,832✔
2232
  SDbHbBatchRsp batchRsp = {0};
5,515,832✔
2233
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
5,515,832✔
2234
  if (batchRsp.pArray == NULL) {
5,515,832✔
2235
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2236
    TAOS_RETURN(code);
×
2237
  }
2238

2239
  for (int32_t i = 0; i < numOfDbs; ++i) {
12,389,272✔
2240
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
6,873,440✔
2241
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
6,873,440✔
2242
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
6,873,440✔
2243
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
6,873,440✔
2244
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
6,873,440✔
2245
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
6,873,440✔
2246
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
6,873,440✔
2247

2248
    SDbHbRsp rsp = {0};
6,873,440✔
2249
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
6,873,440✔
2250
    rsp.dbId = pDbCacheInfo->dbId;
6,873,440✔
2251

2252
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
6,873,440✔
2253
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
6,623,274✔
2254
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
2,213,355✔
2255
      if (pDbCacheInfo->vgVersion >= vgVersion) {
2,213,355✔
2256
        continue;
1,977,787✔
2257
      }
2258

2259
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
235,820✔
2260
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
235,820✔
2261
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
235,820✔
2262

2263
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
235,820✔
2264
      rsp.useDbRsp->vgVersion = vgVersion++;
235,820✔
2265

2266
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
235,820✔
2267

2268
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
471,640✔
2269
        if (terrno != 0) code = terrno;
×
2270
        return code;
×
2271
      }
2272

2273
      continue;
235,820✔
2274
    }
2275

2276
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
4,660,085✔
2277
    if (pDb == NULL) {
4,660,085✔
2278
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
33,109✔
2279
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
33,109✔
2280
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
33,109✔
2281
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
33,109✔
2282
      rsp.useDbRsp->vgVersion = -1;
33,109✔
2283
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
66,218✔
2284
        if (terrno != 0) code = terrno;
×
2285
        return code;
×
2286
      }
2287
      continue;
33,109✔
2288
    }
2289

2290
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
4,626,976✔
2291

2292
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
4,626,976✔
2293
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
4,559,437✔
2294
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
3,790,880✔
2295
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
3,473,716✔
2296
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2297
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2298
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2299
      mndReleaseDb(pMnode, pDb);
3,473,716✔
2300
      continue;
3,473,716✔
2301
    } else {
2302
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
1,153,260✔
2303
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2304
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2305
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2306
    }
2307

2308
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
1,153,260✔
2309
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
43,552✔
2310
      char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
43,552✔
2311
      mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
43,552✔
2312
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb, algrId);
43,552✔
2313
    }
2314

2315
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
1,153,260✔
2316
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
723,544✔
2317
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
723,544✔
2318
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
723,544✔
2319
        bool    exist = false;
723,544✔
2320
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
723,544✔
2321
        if (TSDB_CODE_SUCCESS != code) {
723,544✔
2322
          mndReleaseDb(pMnode, pDb);
×
2323
          if (code != TSDB_CODE_NEED_RETRY) {
×
2324
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2325
          } else {
2326
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2327
          }
2328
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2329
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
2330
          continue;
×
2331
        }
2332
        rsp.dbTsmaVersion = pDb->tsmaVersion;
723,544✔
2333
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
723,544✔
2334
      }
2335
    }
2336

2337
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
1,153,260✔
2338
        pDbCacheInfo->stateTs != pDb->stateTs) {
1,113,182✔
2339
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
819,489✔
2340
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
819,489✔
2341
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
819,489✔
2342
        mndReleaseDb(pMnode, pDb);
×
2343
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
2344
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2345
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2346
        continue;
×
2347
      }
2348

2349
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
819,489✔
2350
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
819,489✔
2351
      rsp.useDbRsp->uid = pDb->uid;
819,489✔
2352
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
819,489✔
2353
      rsp.useDbRsp->stateTs = pDb->stateTs;
819,489✔
2354
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
819,489✔
2355
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
819,489✔
2356
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
819,489✔
2357
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
819,489✔
2358
      rsp.useDbRsp->flags = pDb->cfg.flags;
819,489✔
2359
    }
2360

2361
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
2,306,520✔
2362
      mndReleaseDb(pMnode, pDb);
×
2363
      if (terrno != 0) code = terrno;
×
2364
      return code;
×
2365
    }
2366
    mndReleaseDb(pMnode, pDb);
1,153,260✔
2367
  }
2368

2369
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
5,515,832✔
2370
  void   *pRsp = taosMemoryMalloc(rspLen);
5,515,832✔
2371
  if (pRsp == NULL) {
5,515,832✔
2372
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2373
    tFreeSDbHbBatchRsp(&batchRsp);
×
2374
    return -1;
×
2375
  }
2376
  int32_t ret = 0;
5,515,832✔
2377
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
5,515,832✔
2378

2379
  *ppRsp = pRsp;
5,515,832✔
2380
  *pRspLen = rspLen;
5,515,832✔
2381

2382
  tFreeSDbHbBatchRsp(&batchRsp);
5,515,832✔
2383
  TAOS_RETURN(code);
5,515,832✔
2384
}
2385

2386
static int32_t mndSetTrimDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t startTs, STimeWindow tw,
17,440✔
2387
                                       SArray *vgroupIds, ETsdbOpType type, ETriggerType triggerType,
2388
                                       STrimDbRsp *pRsp) {
2389
  int32_t code = 0, lino = 0;
17,440✔
2390
  SSdb   *pSdb = pMnode->pSdb;
17,440✔
2391
  void   *pIter = NULL;
17,440✔
2392

2393
  SRetentionObj obj = {.optrType = type, .triggerType = triggerType};  // reuse SCompactObj struct
17,440✔
2394
  TAOS_CHECK_EXIT(mndAddRetentionToTrans(pMnode, pTrans, &obj, pDb, pRsp));
17,440✔
2395

2396
  int32_t j = 0;
17,440✔
2397
  int32_t numOfVgroups = taosArrayGetSize(vgroupIds);
17,440✔
2398
  if (numOfVgroups > 0) {
17,440✔
2399
    for (int32_t i = 0; i < numOfVgroups; i++) {
8,724✔
2400
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
6,543✔
2401
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
6,543✔
2402

2403
      if (pVgroup == NULL) {
6,543✔
2404
        mError("db:%s, vgroup:%" PRId64 " not exist", pDb->name, vgId);
2,181✔
2405
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
2,181✔
2406
      } else if (pVgroup->dbUid != pDb->uid) {
4,362✔
2407
        mError("db:%s, vgroup:%" PRId64 " not belong to db:%s", pDb->name, vgId, pDb->name);
×
2408
        sdbRelease(pSdb, pVgroup);
×
2409
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
×
2410
      }
2411
      sdbRelease(pSdb, pVgroup);
4,362✔
2412
    }
2413

2414
    for (int32_t i = 0; i < numOfVgroups; i++) {
6,543✔
2415
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
4,362✔
2416
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
4,362✔
2417

2418
      if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
4,362✔
2419
        sdbRelease(pSdb, pVgroup);
×
2420
        TAOS_RETURN(code);
×
2421
      }
2422

2423
      for (int32_t i = 0; i < pVgroup->replica; i++) {
8,724✔
2424
        SVnodeGid *gid = &pVgroup->vnodeGid[i];
4,362✔
2425
        if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
4,362✔
2426
          sdbRelease(pSdb, pVgroup);
×
2427
          TAOS_RETURN(code);
×
2428
        }
2429
        j++;
4,362✔
2430
      }
2431
      sdbRelease(pSdb, pVgroup);
4,362✔
2432
    }
2433
  } else {
2434
    while (1) {
52,972✔
2435
      SVgObj *pVgroup = NULL;
66,050✔
2436
      pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
66,050✔
2437
      if (pIter == NULL) break;
66,050✔
2438

2439
      if (pVgroup->dbUid == pDb->uid) {
52,972✔
2440
        if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
32,716✔
2441
          sdbCancelFetch(pSdb, pIter);
×
2442
          sdbRelease(pSdb, pVgroup);
×
2443
          TAOS_RETURN(code);
×
2444
        }
2445

2446
        for (int32_t i = 0; i < pVgroup->replica; i++) {
68,032✔
2447
          SVnodeGid *gid = &pVgroup->vnodeGid[i];
35,316✔
2448
          if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
35,316✔
2449
            sdbCancelFetch(pSdb, pIter);
×
2450
            sdbRelease(pSdb, pVgroup);
×
2451
            TAOS_RETURN(code);
×
2452
          }
2453
          j++;
35,316✔
2454
        }
2455
      }
2456

2457
      sdbRelease(pSdb, pVgroup);
52,972✔
2458
    }
2459
  }
2460
_exit:
17,440✔
2461
  TAOS_RETURN(code);
17,440✔
2462
}
2463

2464
static int32_t mndBuildTrimDbRsp(STrimDbRsp *rsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
15,259✔
2465
  int32_t code = 0;
15,259✔
2466
  int32_t rspLen = tSerializeSCompactDbRsp(NULL, 0, (SCompactDbRsp *)rsp);
15,259✔
2467
  void   *pRsp = NULL;
15,259✔
2468
  if (useRpcMalloc) {
15,259✔
2469
    pRsp = rpcMallocCont(rspLen);
×
2470
  } else {
2471
    pRsp = taosMemoryMalloc(rspLen);
15,259✔
2472
  }
2473

2474
  if (pRsp == NULL) {
15,259✔
2475
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2476
    TAOS_RETURN(code);
×
2477
  }
2478

2479
  (void)tSerializeSCompactDbRsp(pRsp, rspLen, (SCompactDbRsp *)rsp);
15,259✔
2480
  *pRspLen = rspLen;
15,259✔
2481
  *ppRsp = pRsp;
15,259✔
2482
  TAOS_RETURN(code);
15,259✔
2483
}
2484

2485
int32_t mndTrimDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb, STimeWindow tw, SArray *vgroupIds,
21,802✔
2486
                         ETsdbOpType type, ETriggerType triggerType) {
2487
  SSdb      *pSdb = pMnode->pSdb;
21,802✔
2488
  SVgObj    *pVgroup = NULL;
21,802✔
2489
  void      *pIter = NULL;
21,802✔
2490
  int32_t    code = 0, lino = 0;
21,802✔
2491
  STrimDbRsp rsp = {0};
21,802✔
2492
  bool       isExist = false;
21,802✔
2493

2494
  while (1) {
4,479✔
2495
    SRetentionObj *pObj = NULL;
26,281✔
2496
    pIter = sdbFetch(pMnode->pSdb, SDB_RETENTION, pIter, (void **)&pObj);
26,281✔
2497
    if (pIter == NULL) break;
26,281✔
2498

2499
    if (strcmp(pObj->dbname, pDb->name) == 0) {
4,479✔
2500
      isExist = true;
4,362✔
2501
    }
2502
    sdbRelease(pMnode->pSdb, pObj);
4,479✔
2503
  }
2504
  if (isExist) {
21,802✔
2505
    mInfo("trim db:%s already exist", pDb->name);
4,362✔
2506
    TAOS_RETURN(TSDB_CODE_MND_TRIM_ALREADY_EXIST);
4,362✔
2507
  }
2508

2509
  int64_t startTs = taosGetTimestampMs();
17,440✔
2510
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "trim-db");
17,440✔
2511
  if (pTrans == NULL) goto _exit;
17,440✔
2512
  mInfo("trans:%d, used to trim db:%s", pTrans->id, pDb->name);
17,440✔
2513
  mndTransSetDbName(pTrans, pDb->name, NULL);
17,440✔
2514
  TAOS_CHECK_EXIT(mndTrancCheckConflict(pMnode, pTrans));
17,440✔
2515
  TAOS_CHECK_EXIT(mndSetTrimDbRedoActions(pMnode, pTrans, pDb, startTs, tw, vgroupIds, type, triggerType, &rsp));
17,440✔
2516
  if (pReq) {
15,259✔
2517
    int32_t rspLen = 0;
15,259✔
2518
    void   *pRsp = NULL;
15,259✔
2519
    rsp.bAccepted = true;
15,259✔
2520
    TAOS_CHECK_EXIT(mndBuildTrimDbRsp(&rsp, &rspLen, &pRsp, false));
15,259✔
2521
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
15,259✔
2522
  }
2523
  TAOS_CHECK_EXIT(mndTransPrepare(pMnode, pTrans));
15,259✔
2524
_exit:
17,440✔
2525
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
17,440✔
2526
    mError("db:%s, failed at line %d to trim db since %s", pDb->name, lino, tstrerror(code));
2,181✔
2527
  }
2528
  mndTransDrop(pTrans);
17,440✔
2529
  TAOS_RETURN(code);
17,440✔
2530
}
2531

2532
#ifdef TD_ENTERPRISE
2533
static int32_t mndCheckRsmaInDb(SMnode *pMnode, SDbObj *pDb) {
13,813✔
2534
  int32_t   code = 0;
13,813✔
2535
  void     *pIter = NULL;
13,813✔
2536
  SRsmaObj *pRsma = NULL;
13,813✔
2537
  while ((pIter = sdbFetch(pMnode->pSdb, SDB_RSMA, pIter, (void **)&pRsma))) {
13,813✔
2538
    if (pRsma->dbUid == pDb->uid) {
13,813✔
2539
      sdbRelease(pMnode->pSdb, pRsma);
13,813✔
2540
      sdbCancelFetch(pMnode->pSdb, pIter);
13,813✔
2541
      TAOS_RETURN(code);
13,813✔
2542
    }
2543
    sdbRelease(pMnode->pSdb, pRsma);
×
2544
  }
2545
  TAOS_RETURN(TSDB_CODE_RSMA_NOT_EXIST);
×
2546
}
2547
#endif
2548

2549
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
21,802✔
2550
  SMnode    *pMnode = pReq->info.node;
21,802✔
2551
  int32_t    code = 0, lino = 0;
21,802✔
2552
  SDbObj    *pDb = NULL;
21,802✔
2553
  STrimDbReq trimReq = {0};
21,802✔
2554
  int64_t    tss = taosGetTimestampMs();
21,802✔
2555

2556
  TAOS_CHECK_EXIT(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq));
21,802✔
2557

2558
  mInfo("db:%s, start to trim, optr:%u, tw:%" PRId64 ",%" PRId64, trimReq.db, trimReq.optrType, trimReq.tw.skey,
21,802✔
2559
        trimReq.tw.ekey);
2560

2561
  pDb = mndAcquireDb(pMnode, trimReq.db);
21,802✔
2562
  if (pDb == NULL) {
21,802✔
2563
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2564
    if (terrno != 0) code = terrno;
×
2565
    TAOS_CHECK_EXIT(code);
×
2566
  }
2567

2568
  TAOS_CHECK_EXIT(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_TRIM_DB, pDb));
21,802✔
2569

2570
  if (pDb->cfg.isMount) {
21,802✔
2571
    TAOS_CHECK_EXIT(TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT);
×
2572
  }
2573

2574
#ifdef TD_ENTERPRISE
2575
  if (trimReq.optrType == TSDB_OPTR_ROLLUP) {
21,802✔
2576
    TAOS_CHECK_EXIT(mndCheckRsmaInDb(pMnode, pDb));
13,813✔
2577
  }
2578
#endif
2579

2580
  TAOS_CHECK_EXIT(mndTrimDb(pMnode, pReq, pDb, trimReq.tw, trimReq.vgroupIds, trimReq.optrType, trimReq.triggerType));
21,802✔
2581
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
15,259✔
2582

2583
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
15,259✔
2584
    SName name = {0};
15,259✔
2585
    if (tNameFromString(&name, trimReq.db, T_NAME_ACCT | T_NAME_DB) < 0) {
15,259✔
2586
      mWarn("db:%s, failed at line %d to parse db name", trimReq.db, __LINE__);
×
2587
    }
2588

2589
    char optrType[16] = {0};
15,259✔
2590
    (void)snprintf(optrType, sizeof(optrType), "%u", trimReq.optrType);
15,259✔
2591
    int64_t tse = taosGetTimestampMs();
15,259✔
2592
    double  duration = (double)(tse - tss);
15,259✔
2593
    duration = duration / 1000;
15,259✔
2594
    auditRecord(pReq, pMnode->clusterId, "trimDB", name.dbname, optrType, trimReq.sql, trimReq.sqlLen, duration, 0);
15,259✔
2595
  }
2596

2597
_exit:
21,802✔
2598
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
21,802✔
2599
    mError("db:%s, failed at line %d to process trim db req since %s, optr:%u", trimReq.db, lino, tstrerror(code),
6,543✔
2600
           trimReq.optrType);
2601
  }
2602

2603
  mndReleaseDb(pMnode, pDb);
21,802✔
2604
  tFreeSTrimDbReq(&trimReq);
21,802✔
2605
  TAOS_RETURN(code);
21,802✔
2606
}
2607

2608
static int32_t mndBuildSsMigrateDbRsp(SSsMigrateDbRsp *pSsMigrateRsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
×
2609
  int32_t code = 0;
×
2610
  int32_t rspLen = tSerializeSSsMigrateDbRsp(NULL, 0, pSsMigrateRsp);
×
2611
  void   *pRsp = NULL;
×
2612
  if (useRpcMalloc) {
×
2613
    pRsp = rpcMallocCont(rspLen);
×
2614
  } else {
2615
    pRsp = taosMemoryMalloc(rspLen);
×
2616
  }
2617

2618
  if (pRsp == NULL) {
×
2619
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2620
    TAOS_RETURN(code);
×
2621
  }
2622

2623
  (void)tSerializeSSsMigrateDbRsp(pRsp, rspLen, pSsMigrateRsp);
×
2624
  *pRspLen = rspLen;
×
2625
  *ppRsp = pRsp;
×
2626
  TAOS_RETURN(code);
×
2627
}
2628

2629
static int32_t mndTrimDbWal(SMnode *pMnode, SDbObj *pDb) {
529✔
2630
  SSdb   *pSdb = pMnode->pSdb;
529✔
2631
  SVgObj *pVgroup = NULL;
529✔
2632
  void   *pIter = NULL;
529✔
2633
  int32_t code = 0;
529✔
2634

2635
  // Iterate through all vgroups in this database
2636
  while (1) {
1,058✔
2637
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
1,587✔
2638
    if (pIter == NULL) break;
1,587✔
2639

2640
    // Check if this vgroup belongs to the database
2641
    if (pVgroup->dbUid != pDb->uid) {
1,058✔
2642
      sdbRelease(pSdb, pVgroup);
×
2643
      continue;
×
2644
    }
2645

2646
    // Prepare message to send to vnode
2647
    int32_t   contLen = sizeof(SMsgHead);
1,058✔
2648
    SMsgHead *pHead = rpcMallocCont(contLen);
1,058✔
2649
    if (pHead == NULL) {
1,058✔
2650
      sdbCancelFetch(pSdb, pVgroup);
×
2651
      sdbRelease(pSdb, pVgroup);
×
2652
      code = TSDB_CODE_OUT_OF_MEMORY;
×
2653
      break;
×
2654
    }
2655
    pHead->contLen = htonl(contLen);
1,058✔
2656
    pHead->vgId = htonl(pVgroup->vgId);
1,058✔
2657

2658
    // Send TRIM WAL request to vnode
2659
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM_WAL, .pCont = pHead, .contLen = contLen};
1,058✔
2660
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
1,058✔
2661
    code = tmsgSendReq(&epSet, &rpcMsg);
1,058✔
2662
    if (code != 0) {
1,058✔
2663
      mError("vgId:%d, failed to send vnode-trim-wal request since 0x%x", pVgroup->vgId, code);
×
2664
    } else {
2665
      mInfo("vgId:%d, send vnode-trim-wal request to vnode", pVgroup->vgId);
1,058✔
2666
    }
2667
    sdbRelease(pSdb, pVgroup);
1,058✔
2668
  }
2669

2670
  return code;
529✔
2671
}
2672

2673
static int32_t mndProcessTrimDbWalReq(SRpcMsg *pReq) {
529✔
2674
  SMnode    *pMnode = pReq->info.node;
529✔
2675
  int32_t    code = -1;
529✔
2676
  SDbObj    *pDb = NULL;
529✔
2677
  STrimDbReq trimReq = {0};
529✔
2678

2679
  TAOS_CHECK_GOTO(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq), NULL, _OVER);
529✔
2680

2681
  mInfo("db:%s, start to trim WAL", trimReq.db);
529✔
2682

2683
  pDb = mndAcquireDb(pMnode, trimReq.db);
529✔
2684
  if (pDb == NULL) {
529✔
2685
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2686
    if (terrno != 0) code = terrno;
×
2687
    goto _OVER;
×
2688
  }
2689

2690
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_TRIM_DB, pDb), NULL, _OVER);
529✔
2691

2692
  code = mndTrimDbWal(pMnode, pDb);
529✔
2693

2694
_OVER:
529✔
2695
  if (code != 0) {
529✔
2696
    mError("db:%s, failed to process trim db wal req since %s", trimReq.db, terrstr());
×
2697
  }
2698

2699
  mndReleaseDb(pMnode, pDb);
529✔
2700
  TAOS_RETURN(code);
529✔
2701
}
2702

2703
int32_t mndSsMigrateDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
×
2704
  int32_t       code = 0;
×
2705
  SSsMigrateDbRsp ssMigrateRsp = {0};
×
2706

2707
  bool  isExist = false;
×
2708
  void *pIter = NULL;
×
2709
  while (1) {
×
2710
    SSsMigrateObj *pSsMigrate = NULL;
×
2711
    pIter = sdbFetch(pMnode->pSdb, SDB_SSMIGRATE, pIter, (void **)&pSsMigrate);
×
2712
    if (pIter == NULL) break;
×
2713

2714
    if (strcmp(pSsMigrate->dbname, pDb->name) == 0) {
×
2715
      isExist = true;
×
2716
    }
2717
    sdbRelease(pMnode->pSdb, pSsMigrate);
×
2718
  }
2719
  if (isExist) {
×
2720
    mInfo("ssmigrate db:%s already exist", pDb->name);
×
2721

2722
    if (pReq) {
×
2723
      int32_t rspLen = 0;
×
2724
      void   *pRsp = NULL;
×
2725
      ssMigrateRsp.ssMigrateId = 0;
×
2726
      ssMigrateRsp.bAccepted = false;
×
2727
      TAOS_CHECK_RETURN(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, true));
×
2728

2729
      pReq->info.rsp = pRsp;
×
2730
      pReq->info.rspLen = rspLen;
×
2731
    }
2732

2733
    return TSDB_CODE_MND_SSMIGRATE_ALREADY_EXIST;
×
2734
  }
2735

2736
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "ssmigrate-db");
×
2737
  if (pTrans == NULL) {
×
2738
    mError("failed to create ssmigrate-db trans since %s", terrstr());
×
2739
    goto _OVER;
×
2740
  }
2741

2742
  mInfo("trans:%d, used to ssmigrate db:%s", pTrans->id, pDb->name);
×
2743
  mndTransSetDbName(pTrans, pDb->name, NULL);
×
2744
  TAOS_CHECK_GOTO(mndTrancCheckConflict(pMnode, pTrans), NULL, _OVER);
×
2745

2746
  SSsMigrateObj ssMigrate = { .startTime = taosGetTimestampMs() };
×
2747
  TAOS_CHECK_GOTO(mndAddSsMigrateToTran(pMnode, pTrans, &ssMigrate, pDb), NULL, _OVER);
×
2748

2749
  if (pReq) {
×
2750
    int32_t rspLen = 0;
×
2751
    void   *pRsp = NULL;
×
2752
    ssMigrateRsp.ssMigrateId = ssMigrate.id;
×
2753
    ssMigrateRsp.bAccepted = true;
×
2754
    TAOS_CHECK_GOTO(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, false), NULL, _OVER);
×
2755
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
×
2756
  }
2757

2758
  if (mndTransPrepare(pMnode, pTrans) != 0) goto _OVER;
×
2759
  code = 0;
×
2760

2761
_OVER:
×
2762
  mndTransDrop(pTrans);
×
2763
  TAOS_RETURN(code);
×
2764
}
2765

2766

2767
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq) {
×
2768
  SMnode         *pMnode = pReq->info.node;
×
2769
  int32_t         code = -1;
×
2770
  SDbObj         *pDb = NULL;
×
2771
  SSsMigrateDbReq ssMigrateReq = {0};
×
2772

2773
  if (!tsSsEnabled) {
×
2774
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
2775
    goto _OVER;
×
2776
  }
2777

2778
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_SHARED_STORAGE), NULL, _OVER);
×
2779
  TAOS_CHECK_GOTO(tDeserializeSSsMigrateDbReq(pReq->pCont, pReq->contLen, &ssMigrateReq), NULL, _OVER);
×
2780

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

2783
  pDb = mndAcquireDb(pMnode, ssMigrateReq.db);
×
2784
  if (pDb == NULL) {
×
2785
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2786
    if (terrno != 0) code = terrno;
×
2787
    goto _OVER;
×
2788
  }
2789

2790
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_SSMIGRATE_DB, pDb), NULL, _OVER);
×
2791

2792
  if(pDb->cfg.isMount) {
×
2793
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
2794
    goto _OVER;
×
2795
  }
2796

2797
  code = mndSsMigrateDb(pMnode, pReq, pDb);
×
2798

2799
_OVER:
×
2800
  if (code != 0) {
×
2801
    mError("db:%s, failed to process ssmigrate db req since %s", ssMigrateReq.db, tstrerror(code));
×
2802
  }
2803

2804
  mndReleaseDb(pMnode, pDb);
×
2805
  TAOS_RETURN(code);
×
2806
}
2807

2808
const char *mndGetDbStr(const char *src) {
19,058,983✔
2809
  char *pos = strstr(src, TS_PATH_DELIMITER);
19,058,983✔
2810
  if (pos != NULL) ++pos;
19,058,983✔
2811
  if (pos == NULL) return src;
19,058,983✔
2812
  return pos;
17,816,692✔
2813
}
2814

2815
const char *mndGetStableStr(const char *src) {
16,284,321✔
2816
  char *pos = strstr(src, TS_PATH_DELIMITER);
16,284,321✔
2817
  if (pos != NULL) ++pos;
16,284,321✔
2818
  if (pos == NULL) return src;
16,284,321✔
2819
  return mndGetDbStr(pos);
16,284,321✔
2820
}
2821

2822
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
×
2823
  int64_t v = 0;
×
2824
  switch (unit) {
×
2825
    case 's':
×
2826
      v = val / 1000;
×
2827
      break;
×
2828
    case 'm':
×
2829
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
×
2830
      break;
×
2831
    case 'h':
×
2832
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2833
      break;
×
2834
    case 'd':
×
2835
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
×
2836
      break;
×
2837
    case 'w':
×
2838
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2839
      break;
×
2840
    default:
×
2841
      break;
×
2842
  }
2843

2844
  return v;
×
2845
}
2846

2847
static char *buildRetension(SArray *pRetension) {
496,129✔
2848
  size_t size = taosArrayGetSize(pRetension);
496,129✔
2849
  if (size == 0) {
496,129✔
2850
    return NULL;
496,129✔
2851
  }
2852

2853
  char       *p1 = taosMemoryCalloc(1, 100);
×
2854
  SRetention *p = taosArrayGet(pRetension, 0);
×
2855

2856
  int32_t len = 2;
×
2857

2858
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2859
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2860
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2861

2862
  if (size > 1) {
×
2863
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2864
    p = taosArrayGet(pRetension, 1);
×
2865

2866
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2867
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2868
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2869
  }
2870

2871
  if (size > 2) {
×
2872
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2873
    p = taosArrayGet(pRetension, 2);
×
2874

2875
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2876
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2877
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2878
  }
2879

2880
  varDataSetLen(p1, len);
×
2881
  return p1;
×
2882
}
2883

2884
static const char *getCacheModelStr(int8_t cacheModel) {
496,129✔
2885
  switch (cacheModel) {
496,129✔
2886
    case TSDB_CACHE_MODEL_NONE:
460,965✔
2887
      return TSDB_CACHE_MODEL_NONE_STR;
460,965✔
2888
    case TSDB_CACHE_MODEL_LAST_ROW:
2,103✔
2889
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
2,103✔
2890
    case TSDB_CACHE_MODEL_LAST_VALUE:
5,416✔
2891
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
5,416✔
2892
    case TSDB_CACHE_MODEL_BOTH:
27,645✔
2893
      return TSDB_CACHE_MODEL_BOTH_STR;
27,645✔
2894
    default:
×
2895
      break;
×
2896
  }
2897
  return "unknown";
×
2898
}
2899

2900
static void getEncryptAlgorithmStr(SMnode *pMnode, int8_t encryptAlgorithm, char *out) {
496,129✔
2901
  switch (encryptAlgorithm) {
496,129✔
2902
    case TSDB_ENCRYPT_ALGO_NONE: {
496,129✔
2903
      tstrncpy(out, TSDB_ENCRYPT_ALGO_NONE_STR, TSDB_ENCRYPT_ALGR_NAME_LEN);
496,129✔
2904
      break;
496,129✔
2905
    }
2906
    default: {
×
2907
      SEncryptAlgrObj *obj = mndAcquireEncryptAlgrById(pMnode, encryptAlgorithm);
×
2908
      if (obj != NULL) {
×
2909
        tstrncpy(out, obj->algorithm_id, TSDB_ENCRYPT_ALGR_NAME_LEN);
×
2910
        mndReleaseEncryptAlgr(pMnode, obj);
×
2911
      }
2912
      break;
×
2913
    }
2914
  }
2915
}
496,129✔
2916

2917
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
1,889,083✔
2918
  if (pDb->cfg.replications == 1) return true;
1,889,083✔
2919

2920
  SSdb *pSdb = pMnode->pSdb;
410,635✔
2921
  void *pIter = NULL;
410,635✔
2922
  bool  isReady = true;
410,635✔
2923
  while (1) {
1,978,673✔
2924
    SVgObj *pVgroup = NULL;
2,389,308✔
2925
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
2,389,308✔
2926
    if (pIter == NULL) break;
2,389,308✔
2927

2928
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
1,978,673✔
2929
      bool hasLeader = false;
1,192,558✔
2930
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
4,721,420✔
2931
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
3,528,862✔
2932
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
3,100,925✔
2933
          hasLeader = true;
427,937✔
2934
        }
2935
      }
2936
      if (!hasLeader) isReady = false;
1,192,558✔
2937
    }
2938
    sdbRelease(pSdb, pVgroup);
1,978,673✔
2939
  }
2940

2941
  return isReady;
410,635✔
2942
}
2943

2944
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
1,360,081✔
2945
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo,
2946
                              SSHashObj **ppUidNames) {
2947
  int32_t cols = 0;
1,360,081✔
2948
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
1,360,081✔
2949
  char   *buf = taosMemoryMalloc(bytes);
1,360,081✔
2950
  if (buf == NULL) {
1,360,081✔
2951
    mError("db:%s, failed to malloc buffer", pDb->name);
×
2952
    return;
×
2953
  }
2954
  int32_t code = 0;
1,360,081✔
2955
  int32_t lino = 0;
1,360,081✔
2956

2957
  const char *name = mndGetDbStr(pDb->name);
1,360,081✔
2958
  if (name != NULL) {
1,360,081✔
2959
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
1,360,081✔
2960
  } else {
2961
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
2962
  }
2963

2964
  const char *precStr = NULL;
1,360,081✔
2965
  switch (pDb->cfg.precision) {
1,360,081✔
2966
    case TSDB_TIME_PRECISION_MILLI:
1,318,216✔
2967
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
1,318,216✔
2968
      break;
1,318,216✔
2969
    case TSDB_TIME_PRECISION_MICRO:
11,146✔
2970
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
11,146✔
2971
      break;
11,146✔
2972
    case TSDB_TIME_PRECISION_NANO:
30,719✔
2973
      precStr = TSDB_TIME_PRECISION_NANO_STR;
30,719✔
2974
      break;
30,719✔
2975
    default:
×
2976
      precStr = "none";
×
2977
      break;
×
2978
  }
2979
  char precVstr[10] = {0};
1,360,081✔
2980
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
1,360,081✔
2981

2982
  char *statusStr = "ready";
1,360,081✔
2983
  if (objStatus == SDB_STATUS_CREATING) {
1,360,081✔
2984
    statusStr = "creating";
×
2985
  } else if (objStatus == SDB_STATUS_DROPPING) {
1,360,081✔
2986
    statusStr = "dropping";
×
2987
  } else {
2988
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
1,360,081✔
2989
      statusStr = "unsynced";
1,629✔
2990
    }
2991
  }
2992
  char statusVstr[24] = {0};
1,360,081✔
2993
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
1,360,081✔
2994

2995
  if (sysDb || !sysinfo) {
1,360,081✔
2996
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
34,558,080✔
2997
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
33,694,128✔
2998
      if (i == 0) {
33,694,128✔
2999
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
863,952✔
3000
      } else if (i == 1) {
32,830,176✔
3001
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
863,952✔
3002
      } else if (i == 3) {
31,966,224✔
3003
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
863,952✔
3004
      } else if (i == 14) {
31,102,272✔
3005
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
863,952✔
3006
      } else if (i == 15) {
30,238,320✔
3007
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
863,952✔
3008
      } else {
3009
        colDataSetNULL(pColInfo, rows);
29,374,368✔
3010
      }
3011
    }
3012
  } else {
3013
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3014
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
496,129✔
3015

3016
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3017
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
496,129✔
3018

3019
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3020
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
496,129✔
3021

3022
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3023
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
496,129✔
3024

3025
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3026
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
496,129✔
3027

3028
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
496,129✔
3029
    char        strictVstr[24] = {0};
496,129✔
3030
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
496,129✔
3031
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3032
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
496,129✔
3033

3034
    char    durationStr[128] = {0};
496,129✔
3035
    char    durationVstr[128] = {0};
496,129✔
3036
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
496,129✔
3037
                                       pDb->cfg.daysPerFile);
3038

3039
    varDataSetLen(durationVstr, len);
496,129✔
3040
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3041
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
496,129✔
3042

3043
    char keepVstr[128] = {0};
496,129✔
3044
    char keep0Str[32] = {0};
496,129✔
3045
    char keep1Str[32] = {0};
496,129✔
3046
    char keep2Str[32] = {0};
496,129✔
3047

3048
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
496,129✔
3049
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
496,129✔
3050
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
496,129✔
3051

3052
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
496,129✔
3053
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
3054
    } else {
3055
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
496,129✔
3056
    }
3057
    varDataSetLen(keepVstr, len);
496,129✔
3058
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3059
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
496,129✔
3060

3061
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3062
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
496,129✔
3063

3064
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3065
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
496,129✔
3066

3067
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3068
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
496,129✔
3069

3070
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3071
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
496,129✔
3072

3073
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3074
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
496,129✔
3075

3076
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3077
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
496,129✔
3078

3079
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3080
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
496,129✔
3081

3082
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3083
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
496,129✔
3084

3085
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
496,129✔
3086
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3087
    if (rentensionVstr == NULL) {
496,129✔
3088
      colDataSetNULL(pColInfo, rows);
496,129✔
3089
    } else {
3090
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
×
3091
      taosMemoryFree(rentensionVstr);
×
3092
    }
3093

3094
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3095
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
496,129✔
3096

3097
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
496,129✔
3098
    char        cacheModelVstr[24] = {0};
496,129✔
3099
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
496,129✔
3100
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3101
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
496,129✔
3102

3103
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3104
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
496,129✔
3105

3106
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3107
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
496,129✔
3108

3109
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3110
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
496,129✔
3111

3112
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3113
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
496,129✔
3114

3115
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3116
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
496,129✔
3117

3118
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3119
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
496,129✔
3120

3121
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3122
    int16_t hashPrefix = pDb->cfg.hashPrefix;
496,129✔
3123
    if (hashPrefix > 0) {
496,129✔
3124
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
3,189✔
3125
    } else if (hashPrefix < 0) {
492,940✔
3126
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
4,968✔
3127
    }
3128
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
496,129✔
3129

3130
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3131
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
496,129✔
3132

3133
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3134
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
496,129✔
3135

3136
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3137
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
496,129✔
3138

3139
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3140
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssChunkSize, false), &lino, _OVER);
496,129✔
3141

3142
    char keeplocalVstr[128] = {0};
496,129✔
3143
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.ssKeepLocal);
496,129✔
3144
    varDataSetLen(keeplocalVstr, len);
496,129✔
3145
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3146
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
496,129✔
3147

3148
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3149
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssCompact, false), &lino, _OVER);
496,129✔
3150

3151
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3152
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
496,129✔
3153

3154
    char encryptAlgorithmStr[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
496,129✔
3155
    getEncryptAlgorithmStr(pMnode, pDb->cfg.encryptAlgorithm, encryptAlgorithmStr);
496,129✔
3156
    char        encryptAlgorithmVStr[24] = {0};
496,129✔
3157
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
496,129✔
3158
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
496,129✔
3159
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
496,129✔
3160

3161
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
496,129✔
3162
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
496,129✔
3163
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
496,129✔
3164
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
496,129✔
3165
    }
3166

3167
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
496,129✔
3168
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
496,129✔
3169
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
496,129✔
3170
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
496,129✔
3171
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
496,129✔
3172
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
496,129✔
3173
    }
3174

3175
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
496,129✔
3176
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
496,129✔
3177
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
496,129✔
3178
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
496,129✔
3179
    }
3180

3181
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
496,129✔
3182
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.isAudit, false), &lino, _OVER);
496,129✔
3183
    }
3184

3185
    if (!(*ppUidNames)) {
496,129✔
3186
      TAOS_CHECK_GOTO(mndBuildUidNamesHash(pMnode, ppUidNames), &lino, _OVER);
282,350✔
3187
    }
3188
    const char *ownerName = tSimpleHashGet(*ppUidNames, (const char *)&pDb->ownerId, sizeof(pDb->ownerId));
496,129✔
3189
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%s", ownerName ? ownerName : "[unknown]"));
496,129✔
3190
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
496,129✔
3191
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
496,129✔
3192
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
496,129✔
3193
    }
3194
  }
3195
_OVER:
1,360,081✔
3196
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
1,360,081✔
3197
  taosMemoryFree(buf);
1,360,081✔
3198
}
3199

3200
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
430,000✔
3201
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
430,000✔
3202
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
430,000✔
3203
  pDbObj->cfg.numOfVgroups = 0;
430,000✔
3204
  pDbObj->cfg.strict = 1;
430,000✔
3205
  pDbObj->cfg.replications = 1;
430,000✔
3206
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
430,000✔
3207
}
430,000✔
3208

3209
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
430,000✔
3210
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
430,000✔
3211
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
430,000✔
3212
  pDbObj->cfg.numOfVgroups = 0;
430,000✔
3213
  pDbObj->cfg.strict = 1;
430,000✔
3214
  pDbObj->cfg.replications = 1;
430,000✔
3215
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
430,000✔
3216
}
430,000✔
3217

3218
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
4,010,042✔
3219
  SVgObj  *pVgroup = pObj;
4,010,042✔
3220
  int32_t *numOfTables = p1;
4,010,042✔
3221
  int64_t  uid = *(int64_t *)p2;
4,010,042✔
3222
  if (pVgroup->dbUid == uid) {
4,010,042✔
3223
    *numOfTables += pVgroup->numOfTables;
1,199,454✔
3224
  }
3225
  return true;
4,010,042✔
3226
}
3227

3228
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
430,000✔
3229
  SMnode    *pMnode = pReq->info.node;
430,000✔
3230
  SSdb      *pSdb = pMnode->pSdb;
430,000✔
3231
  int32_t    numOfRows = 0;
430,000✔
3232
  SDbObj    *pDb = NULL;
430,000✔
3233
  SUserObj  *pOperUser = NULL;
430,000✔
3234
  ESdbStatus objStatus = 0;
430,000✔
3235
  SSHashObj *pUidNames = NULL;
430,000✔
3236

3237
  (void)mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pOperUser);
430,000✔
3238
  if (pOperUser == NULL) return 0;
430,000✔
3239
  bool sysinfo = pOperUser->sysInfo;
430,000✔
3240
  char objFName[TSDB_OBJ_FNAME_LEN + 1] = {0};
430,000✔
3241
  (void)snprintf(objFName, sizeof(objFName), "%d.*", pOperUser->acctId);
430,000✔
3242
  bool showAnyDb = (0 == mndCheckSysObjPrivilege(pMnode, pOperUser, PRIV_CM_SHOW, PRIV_OBJ_DB, 0, objFName, NULL));
430,000✔
3243

3244
  // Append the information_schema database into the result.
3245
  if (!pShow->sysDbRsp) {
430,000✔
3246
    SDbObj infoschemaDb = {0};
430,000✔
3247
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
430,000✔
3248
    size_t numOfTables = 0;
430,000✔
3249
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
430,000✔
3250
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1, &pUidNames);
430,000✔
3251

3252
    numOfRows += 1;
430,000✔
3253

3254
    SDbObj perfschemaDb = {0};
430,000✔
3255
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
430,000✔
3256
    numOfTables = 0;
430,000✔
3257
    getPerfDbMeta(NULL, &numOfTables);
430,000✔
3258
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1, &pUidNames);
430,000✔
3259

3260
    numOfRows += 1;
430,000✔
3261
    pShow->sysDbRsp = true;
430,000✔
3262
  }
3263

3264
  while (numOfRows < rowsCapacity) {
930,081✔
3265
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
930,081✔
3266
    if (pShow->pIter == NULL) break;
930,081✔
3267
    if (showAnyDb || mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_SHOW_DATABASES, pDb) == 0) {
500,081✔
3268
      int32_t numOfTables = 0;
500,081✔
3269
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
500,081✔
3270
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo, &pUidNames);
500,081✔
3271
      numOfRows++;
500,081✔
3272
    }
3273

3274
    sdbRelease(pSdb, pDb);
500,081✔
3275
  }
3276

3277
  pShow->numOfRows += numOfRows;
430,000✔
3278
  mndReleaseUser(pMnode, pOperUser);
430,000✔
3279
  tSimpleHashCleanup(pUidNames);
430,000✔
3280
  return numOfRows;
430,000✔
3281
}
3282

3283
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
3284
  SSdb *pSdb = pMnode->pSdb;
×
3285
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
3286
}
×
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