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

taosdata / TDengine / #5035

24 Apr 2026 11:25AM UTC coverage: 73.06% (+0.002%) from 73.058%
#5035

push

travis-ci

web-flow
merge: from main to 3.0 branch #35224

merge: from main to 3.0 branch[manual-only]

1344 of 1975 new or added lines in 48 files covered. (68.05%)

14127 existing lines in 142 files now uncovered.

275902 of 377640 relevant lines covered (73.06%)

132208813.58 hits per line

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

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

16
#define _DEFAULT_SOURCE
17
#include "mndDb.h"
18
#include "audit.h"
19
#include "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_INITIAL                   1
49
#define DB_VER_SUPPORT_ADVANCED_SECURITY 2
50
#define DB_VER_SUPPORT_MAC               3
51
#define DB_VER_NUMBER                    DB_VER_SUPPORT_MAC
52

53
#define DB_RESERVE_SIZE 8
54

55
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw);
56
static int32_t  mndDbActionInsert(SSdb *pSdb, SDbObj *pDb);
57
static int32_t  mndDbActionDelete(SSdb *pSdb, SDbObj *pDb);
58
static int32_t  mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew);
59
static int32_t  mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw);
60

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

73
#ifndef TD_ENTERPRISE
74
int32_t mndProcessCompactDbReq(SRpcMsg *pReq) { return TSDB_CODE_OPS_NOT_SUPPORT; }
75
#endif
76

77
int32_t mndInitDb(SMnode *pMnode) {
496,669✔
78
  SSdbTable table = {
496,669✔
79
      .sdbType = SDB_DB,
80
      .keyType = SDB_KEY_BINARY,
81
      .encodeFp = (SdbEncodeFp)mndDbActionEncode,
82
      .decodeFp = (SdbDecodeFp)mndDbActionDecode,
83
      .insertFp = (SdbInsertFp)mndDbActionInsert,
84
      .updateFp = (SdbUpdateFp)mndDbActionUpdate,
85
      .deleteFp = (SdbDeleteFp)mndDbActionDelete,
86
      .validateFp = (SdbValidateFp)mndNewDbActionValidate,
87
  };
88

89
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_DB, mndProcessCreateDbReq);
496,669✔
90
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
496,669✔
91
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
496,669✔
92
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
496,669✔
93
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
496,669✔
94
  mndSetMsgHandle(pMnode, TDMT_MND_SCAN_DB, mndProcessScanDbReq);
496,669✔
95
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
496,669✔
96
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB_WAL, mndProcessTrimDbWalReq);
496,669✔
97
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
496,669✔
98
  mndSetMsgHandle(pMnode, TDMT_MND_SSMIGRATE_DB, mndProcessSsMigrateDbReq);
496,669✔
99
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
496,669✔
100

101
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
496,669✔
102
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
496,669✔
103

104
  return sdbSetTable(pMnode->pSdb, table);
496,669✔
105
}
106

107
void mndCleanupDb(SMnode *pMnode) {}
496,600✔
108

109
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
7,127,027✔
110
  int32_t code = 0;
7,127,027✔
111
  int32_t lino = 0;
7,127,027✔
112
  terrno = TSDB_CODE_OUT_OF_MEMORY;
7,127,027✔
113

114
  int32_t  size = sizeof(SDbObj) + pDb->cfg.numOfRetensions * sizeof(SRetention) + DB_RESERVE_SIZE;
7,127,027✔
115
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, DB_VER_NUMBER, size);
7,127,027✔
116
  if (pRaw == NULL) goto _OVER;
7,127,027✔
117

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

180
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
7,127,027✔
181
  SDB_SET_UINT8(pRaw, dataPos, pDb->cfg.flags, _OVER)
7,127,027✔
182
  SDB_SET_INT64(pRaw, dataPos, pDb->ownerId, _OVER)
7,127,027✔
183

184
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
7,127,027✔
185

186
  terrno = 0;
7,127,027✔
187

188
_OVER:
7,127,027✔
189
  if (terrno != 0) {
7,127,027✔
UNCOV
190
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
191
    sdbFreeRaw(pRaw);
×
192
    return NULL;
×
193
  }
194

195
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
7,127,027✔
196
  return pRaw;
7,127,027✔
197
}
198

199
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
6,428,089✔
200
  int32_t code = 0;
6,428,089✔
201
  int32_t lino = 0;
6,428,089✔
202
  terrno = TSDB_CODE_OUT_OF_MEMORY;
6,428,089✔
203
  SSdbRow *pRow = NULL;
6,428,089✔
204
  SDbObj  *pDb = NULL;
6,428,089✔
205

206
  int8_t sver = 0;
6,428,089✔
207
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
6,428,089✔
208

209
  if (sver < 1 || sver > DB_VER_NUMBER) {
6,428,089✔
UNCOV
210
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
211
    goto _OVER;
×
212
  }
213

214
  pRow = sdbAllocRow(sizeof(SDbObj));
6,428,089✔
215
  if (pRow == NULL) goto _OVER;
6,428,089✔
216

217
  pDb = sdbGetRowObj(pRow);
6,428,089✔
218
  if (pDb == NULL) goto _OVER;
6,428,089✔
219

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

289
  SDB_GET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
6,428,089✔
290
  if (dataPos + sizeof(uint8_t) <= pRaw->dataLen) {
6,428,089✔
291
    SDB_GET_UINT8(pRaw, dataPos, &pDb->cfg.flags, _OVER)
6,428,089✔
292
  }
293
  if (dataPos + sizeof(int64_t) <= pRaw->dataLen) {
6,428,089✔
294
    SDB_GET_INT64(pRaw, dataPos, &pDb->ownerId, _OVER)
6,428,089✔
295
  }
296

297
  taosInitRWLatch(&pDb->lock);
6,428,089✔
298

299
  if (pDb->cfg.ssChunkSize == 0) {
6,428,089✔
UNCOV
300
    pDb->cfg.ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
×
301

UNCOV
302
    mInfo("db:%s, ssChunkSize set from %d to default %d", pDb->name, pDb->cfg.ssChunkSize, TSDB_DEFAULT_SS_CHUNK_SIZE);
×
303
  }
304

305
  if (pDb->cfg.ssKeepLocal == 0) {
6,428,089✔
UNCOV
306
    pDb->cfg.ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
×
307

UNCOV
308
    mInfo("db:%s, ssKeepLocal set from %d to default %d", pDb->name, pDb->cfg.ssKeepLocal, TSDB_DEFAULT_SS_KEEP_LOCAL);
×
309
  }
310

311
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
6,428,089✔
312
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
6,428,089✔
313
          TSDB_DEFAULT_TSDB_PAGESIZE);
314
  }
315

316
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
6,428,089✔
317
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
6,287,968✔
318
  }
319

320
  if (sver < DB_VER_SUPPORT_ADVANCED_SECURITY) {
6,428,089✔
UNCOV
321
    pDb->cfg.allowDrop = TSDB_DEFAULT_DB_ALLOW_DROP;
×
322
  }
323

324
  if (sver < DB_VER_SUPPORT_MAC) {
6,428,089✔
NEW
325
    pDb->cfg.securityLevel = TSDB_DEFAULT_SECURITY_LEVEL;
×
326
  }
327

328
  terrno = 0;
6,428,089✔
329

330
_OVER:
6,428,089✔
331
  if (terrno != 0) {
6,428,089✔
UNCOV
332
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
UNCOV
333
    taosMemoryFreeClear(pRow);
×
UNCOV
334
    return NULL;
×
335
  }
336

337
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
6,428,089✔
338
  return pRow;
6,428,089✔
339
}
340

341
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
1,474,602✔
342
  SSdb    *pSdb = pMnode->pSdb;
1,474,602✔
343
  SSdbRow *pRow = NULL;
1,474,602✔
344
  SDbObj  *pNewDb = NULL;
1,474,602✔
345
  int      code = -1;
1,474,602✔
346

347
  pRow = mndDbActionDecode(pRaw);
1,474,602✔
348
  if (pRow == NULL) {
1,474,602✔
UNCOV
349
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
350
    if (terrno != 0) code = terrno;
×
UNCOV
351
    goto _OVER;
×
352
  }
353
  pNewDb = sdbGetRowObj(pRow);
1,474,602✔
354
  if (pNewDb == NULL) {
1,474,602✔
355
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
356
    if (terrno != 0) code = terrno;
×
UNCOV
357
    goto _OVER;
×
358
  }
359

360
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
1,474,602✔
361
  if (pOldDb != NULL) {
1,474,602✔
362
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
UNCOV
363
    sdbRelease(pMnode->pSdb, pOldDb);
×
UNCOV
364
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
365
    if (terrno != 0) code = terrno;
×
UNCOV
366
    goto _OVER;
×
367
  }
368

369
  code = 0;
1,474,602✔
370
_OVER:
1,474,602✔
371
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
1,474,602✔
372
  taosMemoryFreeClear(pRow);
1,474,602✔
373
  return code;
1,474,602✔
374
}
375

376
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
1,645,320✔
377
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
1,645,320✔
378
  return 0;
1,645,320✔
379
}
380

381
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
6,428,089✔
382
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
6,428,089✔
383
  taosArrayDestroy(pDb->cfg.pRetensions);
6,428,089✔
384
  return 0;
6,428,089✔
385
}
386

387
static int32_t mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew) {
2,525,019✔
388
  mTrace("db:%s, perform update action, old row:%p new row:%p", pOld->name, pOld, pNew);
2,525,019✔
389
  taosWLockLatch(&pOld->lock);
2,525,019✔
390
  pOld->updateTime = pNew->updateTime;
2,525,019✔
391
  pOld->cfgVersion = pNew->cfgVersion;
2,525,019✔
392
  pOld->vgVersion = pNew->vgVersion;
2,525,019✔
393
  pOld->cfg.numOfVgroups = pNew->cfg.numOfVgroups;
2,525,019✔
394
  pOld->cfg.buffer = pNew->cfg.buffer;
2,525,019✔
395
  pOld->cfg.pageSize = pNew->cfg.pageSize;
2,525,019✔
396
  pOld->cfg.pages = pNew->cfg.pages;
2,525,019✔
397
  pOld->cfg.cacheLastSize = pNew->cfg.cacheLastSize;
2,525,019✔
398
  pOld->cfg.cacheLastShardBits = pNew->cfg.cacheLastShardBits;
2,525,019✔
399
  pOld->cfg.daysPerFile = pNew->cfg.daysPerFile;
2,525,019✔
400
  pOld->cfg.daysToKeep0 = pNew->cfg.daysToKeep0;
2,525,019✔
401
  pOld->cfg.daysToKeep1 = pNew->cfg.daysToKeep1;
2,525,019✔
402
  pOld->cfg.daysToKeep2 = pNew->cfg.daysToKeep2;
2,525,019✔
403
  pOld->cfg.keepTimeOffset = pNew->cfg.keepTimeOffset;
2,525,019✔
404
  pOld->cfg.walFsyncPeriod = pNew->cfg.walFsyncPeriod;
2,525,019✔
405
  pOld->cfg.walLevel = pNew->cfg.walLevel;
2,525,019✔
406
  pOld->cfg.walRetentionPeriod = pNew->cfg.walRetentionPeriod;
2,525,019✔
407
  pOld->cfg.walRetentionSize = pNew->cfg.walRetentionSize;
2,525,019✔
408
  pOld->cfg.strict = pNew->cfg.strict;
2,525,019✔
409
  pOld->cfg.cacheLast = pNew->cfg.cacheLast;
2,525,019✔
410
  pOld->cfg.replications = pNew->cfg.replications;
2,525,019✔
411
  pOld->cfg.sstTrigger = pNew->cfg.sstTrigger;
2,525,019✔
412
  pOld->cfg.minRows = pNew->cfg.minRows;
2,525,019✔
413
  pOld->cfg.maxRows = pNew->cfg.maxRows;
2,525,019✔
414
  pOld->cfg.tsdbPageSize = pNew->cfg.tsdbPageSize;
2,525,019✔
415
  pOld->cfg.ssChunkSize = pNew->cfg.ssChunkSize;
2,525,019✔
416
  pOld->cfg.ssKeepLocal = pNew->cfg.ssKeepLocal;
2,525,019✔
417
  pOld->cfg.ssCompact = pNew->cfg.ssCompact;
2,525,019✔
418
  pOld->cfg.withArbitrator = pNew->cfg.withArbitrator;
2,525,019✔
419
  pOld->cfg.compactInterval = pNew->cfg.compactInterval;
2,525,019✔
420
  pOld->cfg.compactStartTime = pNew->cfg.compactStartTime;
2,525,019✔
421
  pOld->cfg.compactEndTime = pNew->cfg.compactEndTime;
2,525,019✔
422
  pOld->cfg.compactTimeOffset = pNew->cfg.compactTimeOffset;
2,525,019✔
423
  pOld->compactStartTime = pNew->compactStartTime;
2,525,019✔
424
  pOld->tsmaVersion = pNew->tsmaVersion;
2,525,019✔
425
  pOld->cfg.isAudit = pNew->cfg.isAudit;
2,525,019✔
426
  pOld->cfg.secureDelete = pNew->cfg.secureDelete;
2,525,019✔
427
  pOld->cfg.flags = pNew->cfg.flags;
2,525,019✔
428
  pOld->ownerId = pNew->ownerId;
2,525,019✔
429
  taosWUnLockLatch(&pOld->lock);
2,525,019✔
430
  return 0;
2,525,019✔
431
}
432

433
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
10,481,038✔
434
  SSdb *pSdb = pMnode->pSdb;
10,481,038✔
435
  return sdbGetTableVer(pSdb, SDB_VGROUP);
10,481,038✔
436
}
437

438
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
107,067,226✔
439
  SSdb   *pSdb = pMnode->pSdb;
107,067,226✔
440
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
107,068,048✔
441
  if (pDb == NULL) {
107,067,295✔
442
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
3,767,613✔
443
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
3,537,823✔
444
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
229,790✔
445
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
12,616✔
446
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
217,174✔
447
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
217,174✔
448
    } else {
UNCOV
449
      terrno = TSDB_CODE_APP_ERROR;
×
UNCOV
450
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
451
    }
452
  }
453
  return pDb;
107,067,257✔
454
}
455

456
SDbObj *mndAcquireAuditDb(SMnode *pMnode) {
53,361,281✔
457
  SSdb      *pSdb = pMnode->pSdb;
53,361,281✔
458
  SDbObj    *pDb = NULL;
53,361,281✔
459
  void      *pIter = NULL;
53,361,281✔
460
  ESdbStatus objStatus = 0;
53,361,281✔
461

462
  while (1) {
463
    pIter = sdbFetchAll(pSdb, SDB_DB, pIter, (void **)&pDb, &objStatus, true);
154,074,141✔
464
    if (pIter == NULL) break;
154,074,141✔
465
    if (pDb->cfg.isAudit == 1) {
100,716,606✔
466
      sdbCancelFetch(pSdb, pIter);
3,746✔
467
      break;
3,746✔
468
    }
469

470
    sdbRelease(pSdb, pDb);
100,712,860✔
471
  }
472
  return pDb;
53,361,281✔
473
}
474

475
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
117,315,026✔
476
  SSdb *pSdb = pMnode->pSdb;
117,315,026✔
477
  sdbRelease(pSdb, pDb);
117,315,070✔
478
}
117,315,790✔
479

480
bool mndDbIsExist(SMnode *pMnode, const char *db, int64_t uid) {
195,330✔
481
  SDbObj *pDb = mndAcquireDb(pMnode, db);
195,330✔
482
  if (pDb == NULL) {
195,330✔
483
    return false;
2,840✔
484
  } else {
485
    bool result = (uid == 0) || (uid == pDb->uid);
192,490✔
486
    mndReleaseDb(pMnode, pDb);
192,490✔
487
    pDb = NULL;
192,490✔
488
    return result;
192,490✔
489
  }
490
}
491

492
int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
1,436,957✔
493
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
1,436,957✔
494
  if (pos == NULL) {
1,436,957✔
UNCOV
495
    return TSDB_CODE_MND_INVALID_DB;
×
496
  }
497

498
  int32_t acctId;
1,434,587✔
499
  int32_t code = taosStr2int32(dbName, &acctId);
1,436,957✔
500
  if (code != 0) {
1,436,957✔
UNCOV
501
    return code;
×
502
  }
503

504
  if (acctId != pUser->acctId) {
1,436,957✔
UNCOV
505
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
506
  }
507

508
  return 0;
1,436,957✔
509
}
510

511
int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
1,436,957✔
512
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
1,436,957✔
513

514
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
1,436,957✔
515
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
1,436,957✔
516
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
1,436,957✔
517
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
1,436,957✔
518
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
1,436,957✔
519
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
1,436,957✔
520
  if (pCfg->cacheLastShardBits < TSDB_MIN_DB_CACHE_SHARD_BITS ||
1,436,957✔
521
      pCfg->cacheLastShardBits > TSDB_MAX_DB_CACHE_SHARD_BITS)
1,435,375✔
522
    return code;
3,955✔
523
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
1,433,002✔
524
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
1,433,002✔
525
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
1,433,002✔
526
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
1,433,002✔
527
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
1,433,002✔
528
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
1,433,002✔
529
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
1,433,002✔
530
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
1,433,002✔
531
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
1,433,002✔
532
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
1,433,002✔
533
  if (pCfg->minRows > pCfg->maxRows) return code;
1,433,002✔
534
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
1,432,211✔
535
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
1,432,211✔
536
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
1,432,211✔
537
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
1,432,211✔
538
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
1,432,211✔
539
#ifdef TD_ENTERPRISE
540
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
1,432,211✔
541
  // if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
542
#else
543
  if (pCfg->replications != 1 && pCfg->replications != 3) return code;
544
  if (pCfg->encryptAlgorithm != TSDB_DEFAULT_ENCRYPT_ALGO) return code;
545
#endif
546

547
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
1,432,211✔
548
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
1,432,211✔
549
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
1,432,211✔
550
  if (pCfg->hashMethod != 1) return code;
1,432,211✔
551
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
1,432,211✔
UNCOV
552
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
UNCOV
553
    TAOS_RETURN(code);
×
554
  }
555
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
1,432,211✔
556
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
1,432,211✔
557
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
1,432,211✔
558
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
1,432,211✔
559
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
1,432,211✔
560
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
1,432,211✔
561
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
1,432,211✔
562
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
1,432,211✔
563
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
1,432,211✔
564
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
1,432,211✔
565
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
1,432,211✔
566

567
  if (pCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
1,432,211✔
568
  if (pCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
1,432,211✔
569
  if (pCfg->ssCompact < TSDB_MIN_SS_COMPACT || pCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
1,432,211✔
570

571
  if (pCfg->compactInterval != 0 &&
1,432,211✔
572
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
4,052✔
UNCOV
573
    return code;
×
574
  if (pCfg->compactStartTime != 0 &&
1,432,211✔
575
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
4,471✔
UNCOV
576
    return code;
×
577
  if (pCfg->compactEndTime != 0 &&
1,432,211✔
578
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
4,471✔
UNCOV
579
    return code;
×
580
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
1,432,211✔
581
    return code;
×
582
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
1,432,211✔
UNCOV
583
    return code;
×
584
  if (pCfg->isAudit < 0 || pCfg->isAudit > 1) return code;
1,432,211✔
585
  if (pCfg->allowDrop < TSDB_MIN_DB_ALLOW_DROP || pCfg->allowDrop > TSDB_MAX_DB_ALLOW_DROP) return code;
586
  if (pCfg->securityLevel < TSDB_MIN_SECURITY_LEVEL || pCfg->securityLevel > TSDB_MAX_SECURITY_LEVEL) return code;
1,432,211✔
587

588
  code = 0;
1,432,211✔
589
  TAOS_RETURN(code);
1,432,211✔
590
}
591

592
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
97,062✔
593
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
97,062✔
594
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
97,062✔
595
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
97,062✔
596
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
97,062✔
597
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
97,062✔
598
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
97,062✔
599
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
97,062✔
600
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
97,062✔
601
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
97,062✔
602
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
97,062✔
603
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
97,062✔
604
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
97,062✔
UNCOV
605
    return code;
×
606
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
97,062✔
607
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
97,062✔
608
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
97,062✔
609
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
97,062✔
610
  if (pNewCfg->cacheLastShardBits < TSDB_MIN_DB_CACHE_SHARD_BITS ||
97,062✔
611
      pNewCfg->cacheLastShardBits > TSDB_MAX_DB_CACHE_SHARD_BITS)
97,062✔
UNCOV
612
    return code;
×
613
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
97,062✔
614
#ifdef TD_ENTERPRISE
615
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
97,062✔
616
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
97,062✔
617
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
788✔
618
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
788✔
619
      return code;
788✔
620
    }
621
  }
622
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
96,274✔
UNCOV
623
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
UNCOV
624
    return code;
×
625
  }
626
#else
627
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
628
#endif
629

630
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
96,274✔
UNCOV
631
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
UNCOV
632
    return code;
×
633
  }
634
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
96,274✔
UNCOV
635
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
UNCOV
636
    return code;
×
637
  }
638

639
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
96,274✔
640
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
1,379✔
641
    return code;
×
642
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
96,274✔
643
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
96,274✔
644
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
96,274✔
645
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
95,486✔
646
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
95,486✔
647
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
95,486✔
648
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
95,486✔
UNCOV
649
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
UNCOV
650
    return code;
×
651
  }
652
  if (pNewCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pNewCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
95,486✔
653
  if (pNewCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pNewCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
95,486✔
654
  if (pNewCfg->ssCompact < TSDB_MIN_SS_COMPACT || pNewCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
95,486✔
655

656
  if (pNewCfg->compactInterval != 0 &&
95,486✔
657
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
7,254✔
UNCOV
658
    return code;
×
659
  if (pNewCfg->compactStartTime != 0 &&
95,486✔
660
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
4,890✔
UNCOV
661
    return code;
×
662
  if (pNewCfg->compactEndTime != 0 &&
95,486✔
663
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
4,890✔
664
    return code;
×
665
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
95,486✔
666
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
4,890✔
667
    return code;
×
668
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
95,486✔
669
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
95,486✔
670
    return code;
×
671
  if (pNewCfg->isAudit < 0 || pNewCfg->isAudit > 1) return code;
95,486✔
672
  if (pNewCfg->allowDrop < TSDB_MIN_DB_ALLOW_DROP || pNewCfg->allowDrop > TSDB_MAX_DB_ALLOW_DROP) return code;
673
  if (pNewCfg->securityLevel < TSDB_MIN_SECURITY_LEVEL || pNewCfg->securityLevel > TSDB_MAX_SECURITY_LEVEL) return code;
95,486✔
674

675
  code = 0;
95,486✔
676
  TAOS_RETURN(code);
95,486✔
677
}
678

679
static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
1,436,303✔
680
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
1,436,303✔
681
  if (pCfg->numOfStables < 0) pCfg->numOfStables = TSDB_DEFAULT_DB_SINGLE_STABLE;
1,436,303✔
682
  if (pCfg->buffer < 0) pCfg->buffer = TSDB_DEFAULT_BUFFER_PER_VNODE;
1,436,303✔
683
  if (pCfg->pageSize < 0) pCfg->pageSize = TSDB_DEFAULT_PAGESIZE_PER_VNODE;
1,436,303✔
684
  if (pCfg->pages < 0) pCfg->pages = TSDB_DEFAULT_PAGES_PER_VNODE;
1,436,303✔
685
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DURATION_PER_FILE;
1,436,303✔
686
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
1,436,303✔
687
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
1,436,303✔
688
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
1,436,303✔
689
  if (pCfg->keepTimeOffset < 0) pCfg->keepTimeOffset = TSDB_DEFAULT_KEEP_TIME_OFFSET;
1,436,303✔
690
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MINROWS_FBLOCK;
1,436,303✔
691
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAXROWS_FBLOCK;
1,436,303✔
692
  if (pCfg->walFsyncPeriod < 0) pCfg->walFsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
1,436,303✔
693
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
1,436,303✔
694
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
1,436,303✔
695
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
1,436,303✔
696
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA;
1,436,303✔
697
  if (pCfg->strict < 0) pCfg->strict = TSDB_DEFAULT_DB_STRICT;
1,436,303✔
698
  if (pCfg->cacheLast < 0) pCfg->cacheLast = TSDB_DEFAULT_CACHE_MODEL;
1,436,303✔
699
  if (pCfg->cacheLastSize <= 0) pCfg->cacheLastSize = TSDB_DEFAULT_CACHE_SIZE;
1,436,303✔
700
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
1,436,303✔
701
  if (pCfg->schemaless < 0) pCfg->schemaless = TSDB_DB_SCHEMALESS_OFF;
1,436,303✔
702
  if (pCfg->walRetentionPeriod < 0 && pCfg->walRetentionPeriod != -1)
1,436,303✔
UNCOV
703
    pCfg->walRetentionPeriod = TSDB_REPS_DEF_DB_WAL_RET_PERIOD;
×
704
  if (pCfg->walRetentionSize < 0 && pCfg->walRetentionSize != -1)
1,436,303✔
UNCOV
705
    pCfg->walRetentionSize = TSDB_REPS_DEF_DB_WAL_RET_SIZE;
×
706
  if (pCfg->walRollPeriod < 0) pCfg->walRollPeriod = TSDB_REPS_DEF_DB_WAL_ROLL_PERIOD;
1,436,303✔
707
  if (pCfg->walSegmentSize < 0) pCfg->walSegmentSize = TSDB_DEFAULT_DB_WAL_SEGMENT_SIZE;
1,436,303✔
708
  if (pCfg->sstTrigger <= 0) pCfg->sstTrigger = TSDB_DEFAULT_SST_TRIGGER;
1,436,303✔
709
  if (pCfg->tsdbPageSize <= 0) pCfg->tsdbPageSize = TSDB_DEFAULT_TSDB_PAGESIZE;
1,436,303✔
710
  if (pCfg->ssChunkSize <= 0) pCfg->ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
1,436,303✔
711
  if (pCfg->ssKeepLocal <= 0) pCfg->ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
1,436,303✔
712
  if (pCfg->ssCompact < 0) pCfg->ssCompact = TSDB_DEFAULT_SS_COMPACT;
1,436,303✔
713
  if (pCfg->withArbitrator < 0) pCfg->withArbitrator = TSDB_DEFAULT_DB_WITH_ARBITRATOR;
1,436,303✔
714
  if (pCfg->encryptAlgorithm < 0) pCfg->encryptAlgorithm = TSDB_DEFAULT_ENCRYPT_ALGO;
1,436,303✔
715
}
1,436,303✔
716

717
int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,427,838✔
718
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,427,838✔
719
  if (pDbRaw == NULL) return -1;
1,427,838✔
720

721
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
1,427,838✔
722
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
1,427,838✔
723
  return 0;
1,427,838✔
724
}
725

726
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,427,184✔
727
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
4,441,879✔
728
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
3,014,695✔
729
  }
730
  return 0;
1,427,184✔
731
}
732

UNCOV
733
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
UNCOV
734
  int32_t  code = 0;
×
UNCOV
735
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
UNCOV
736
  if (pDbRaw == NULL) {
×
UNCOV
737
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
738
    if (terrno != 0) code = terrno;
×
UNCOV
739
    TAOS_RETURN(code);
×
740
  }
741
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
742
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
743

744
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
745
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
746
    if (pVgRaw == NULL) {
×
UNCOV
747
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
748
      if (terrno != 0) code = terrno;
×
749
      TAOS_RETURN(code);
×
750
    }
751
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
752
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
753
  }
754

755
  if (pDb->cfg.withArbitrator) {
×
756
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
UNCOV
757
      SVgObj   *pVgObj = pVgroups + v;
×
758
      SArbGroup arbGroup = {0};
×
759
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
UNCOV
760
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
761
    }
762
  }
763

764
  TAOS_RETURN(code);
×
765
}
766

767
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,427,184✔
768
  int32_t  code = 0;
1,427,184✔
769
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,427,184✔
770
  if (pDbRaw == NULL) {
1,427,184✔
771
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
772
    if (terrno != 0) code = terrno;
×
UNCOV
773
    TAOS_RETURN(code);
×
774
  }
775
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
1,427,184✔
776
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
1,427,184✔
777

778
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
4,441,879✔
779
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
3,014,695✔
780
    if (pVgRaw == NULL) {
3,014,695✔
UNCOV
781
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
782
      if (terrno != 0) code = terrno;
×
UNCOV
783
      TAOS_RETURN(code);
×
784
    }
785
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
3,014,695✔
786
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
3,014,695✔
787
  }
788

789
  if (pDb->cfg.withArbitrator) {
1,427,184✔
790
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
11,180✔
791
      SVgObj   *pVgObj = pVgroups + v;
6,772✔
792
      SArbGroup arbGroup = {0};
6,772✔
793
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6,772✔
794
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
6,772✔
795
    }
796
  }
797

798
  TAOS_RETURN(code);
1,427,184✔
799
}
800

801
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
1,427,184✔
802
                                        SUserObj *pUserDuped, SArray *pAuditOwnedDbs) {
803
  int32_t  code = 0;
1,427,184✔
804
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,427,184✔
805
  if (pDbRaw == NULL) {
1,427,184✔
UNCOV
806
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
807
    if (terrno != 0) code = terrno;
×
UNCOV
808
    TAOS_RETURN(code);
×
809
  }
810
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
1,427,184✔
811
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
1,427,184✔
812

813
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
4,441,879✔
814
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
3,014,695✔
815
    if (pVgRaw == NULL) {
3,014,695✔
UNCOV
816
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
817
      if (terrno != 0) code = terrno;
×
UNCOV
818
      TAOS_RETURN(code);
×
819
    }
820
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
3,014,695✔
821
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
3,014,695✔
822
  }
823

824
  if (pDb->cfg.withArbitrator) {
1,427,184✔
825
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
11,180✔
826
      SVgObj   *pVgObj = pVgroups + v;
6,772✔
827
      SArbGroup arbGroup = {0};
6,772✔
828
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6,772✔
829
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
6,772✔
830
    }
831
  }
832

833
  if (pUserDuped) {
1,427,184✔
834
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
1,427,184✔
835
    if (pUserRaw == NULL) {
1,427,184✔
UNCOV
836
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
837
      if (terrno != 0) code = terrno;
×
UNCOV
838
      TAOS_RETURN(code);
×
839
    }
840
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
1,427,184✔
841
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
1,427,184✔
842
  }
843

844
  if(pAuditOwnedDbs) {
1,427,184✔
845
    int32_t auditOwnedDbSize = taosArrayGetSize(pAuditOwnedDbs);
×
UNCOV
846
    for (int32_t i = 0; i < auditOwnedDbSize; ++i) {
×
UNCOV
847
      SUserObj *pAuditUser = (SUserObj *)TARRAY_GET_ELEM(pAuditOwnedDbs, i);
×
UNCOV
848
      SSdbRaw  *pAuditUserRaw = mndUserActionEncode(pAuditUser);
×
UNCOV
849
      if (pAuditUserRaw == NULL) {
×
UNCOV
850
        TAOS_RETURN(terrno ? terrno : TSDB_CODE_MND_RETURN_VALUE_NULL);
×
851
      }
852
      TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pAuditUserRaw));
×
853
      TAOS_CHECK_RETURN(sdbSetRawStatus(pAuditUserRaw, SDB_STATUS_READY));
×
854
    }
855
  }
856

857
  TAOS_RETURN(code);
1,427,184✔
858
}
859

860
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,427,184✔
861
  int32_t code = 0;
1,427,184✔
862
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
4,441,879✔
863
    SVgObj *pVgroup = pVgroups + vg;
3,014,695✔
864

865
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
6,324,192✔
866
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
3,309,497✔
867
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
3,309,497✔
868
    }
869
  }
870

871
  TAOS_RETURN(code);
1,427,184✔
872
}
873

874
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,427,184✔
875
  int32_t code = 0;
1,427,184✔
876
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
4,441,879✔
877
    SVgObj *pVgroup = pVgroups + vg;
3,014,695✔
878

879
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
6,324,192✔
880
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
3,309,497✔
881
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
3,309,497✔
882
    }
883
  }
884

885
  TAOS_RETURN(code);
1,427,184✔
886
}
887

888
/**
889
 * @brief fill audit owned dbs for users with role SYSAUDIT or SYSAUDIT_LOG.
890
 *
891
 * @param pMnode
892
 * @param pOperUser
893
 * @param pDb
894
 * @param pAuditOwnedDbs
895
 * @return int32_t
896
 */
897
static int32_t mndSetAuditOwnedDbs(SMnode *pMnode, SUserObj *pOperUser, SDbObj *pDb, SArray **pAuditOwnedDbs) {
319✔
898
  int32_t   code = 0, lino = 0;
319✔
899
  SSdb     *pSdb = pMnode->pSdb;
319✔
900
  SArray   *auditOwnedDbs = NULL;
319✔
901
  SUserObj *pUser = NULL;
319✔
902
  SUserObj  newUserObj = {0};
319✔
903
  int32_t   sysAuditLen = sizeof(TSDB_ROLE_SYSAUDIT);
319✔
904
  int32_t   sysAuditLogLen = sizeof(TSDB_ROLE_SYSAUDIT_LOG);
319✔
905

906
  void *pIter = NULL;
319✔
907
  while ((pIter = sdbFetch(pSdb, SDB_USER, pIter, (void **)&pUser))) {
638✔
908
    if (taosHashGetSize(pUser->roles) <= 0) {
319✔
UNCOV
909
      sdbRelease(pSdb, pUser);
×
UNCOV
910
      pUser = NULL;
×
UNCOV
911
      continue;
×
912
    }
913
    if(strncmp(pUser->name, pOperUser->name, TSDB_USER_LEN) == 0) {
319✔
914
      sdbRelease(pSdb, pUser);
319✔
915
      pUser = NULL;
319✔
916
      continue;
319✔
917
    }
918
    if (!taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT, sysAuditLen) &&
×
UNCOV
919
        !taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT_LOG, sysAuditLogLen)) {
×
UNCOV
920
      sdbRelease(pSdb, pUser);
×
UNCOV
921
      pUser = NULL;
×
UNCOV
922
      continue;
×
923
    }
UNCOV
924
    TAOS_CHECK_EXIT(mndUserDupObj(pUser, &newUserObj));
×
925
    TAOS_CHECK_EXIT(taosHashPut(newUserObj.ownedDbs, pDb->name, strlen(pDb->name) + 1, NULL, 0));
×
926
    if (auditOwnedDbs == NULL) {
×
927
      auditOwnedDbs = taosArrayInit(1, sizeof(SUserObj));
×
928
      if (auditOwnedDbs == NULL) {
×
929
        TAOS_CHECK_EXIT(terrno);
×
930
      }
931
    }
932
    if (!taosArrayPush(auditOwnedDbs, &newUserObj)) {
×
933
      TAOS_CHECK_EXIT(terrno);
×
934
    }
935
    memset(&newUserObj, 0, sizeof(SUserObj));
×
936
    sdbRelease(pSdb, pUser);
×
UNCOV
937
    pUser = NULL;
×
938
  }
939
  *pAuditOwnedDbs = auditOwnedDbs;
319✔
940
  auditOwnedDbs = NULL;
319✔
941
_exit:
319✔
942
  if (auditOwnedDbs != NULL) taosArrayDestroyEx(auditOwnedDbs, (FDelete)mndUserFreeObj);
319✔
943
  sdbRelease(pSdb, pUser);
319✔
944
  sdbCancelFetch(pSdb, pIter);
319✔
945
  mndUserFreeObj(&newUserObj);
319✔
946
  TAOS_RETURN(code);
319✔
947
}
948

949
static int32_t mndCreateDb(SMnode *pMnode, SRpcMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser, SArray *dnodeList) {
1,438,698✔
950
  int32_t  code = 0;
1,438,698✔
951
  SArray  *auditOwnedDbs = NULL;
1,438,698✔
952
  SUserObj newUserObj = {0};
1,438,698✔
953
  SDbObj   dbObj = {0};
1,438,698✔
954
  (void)memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
1,438,698✔
955
  (void)memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
1,438,698✔
956
  dbObj.createdTime = taosGetTimestampMs();
1,438,698✔
957
  dbObj.updateTime = dbObj.createdTime;
1,438,698✔
958
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
1,438,698✔
959
  dbObj.cfgVersion = 1;
1,438,698✔
960
  dbObj.vgVersion = 1;
1,438,698✔
961
  dbObj.tsmaVersion = 1;
1,438,698✔
962
  (void)memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
1,438,698✔
963
  dbObj.ownerId = pUser->uid;
1,438,698✔
964
  dbObj.cfg = (SDbCfg){
2,875,026✔
965
      .numOfVgroups = pCreate->numOfVgroups,
1,438,698✔
966
      .numOfStables = pCreate->numOfStables,
1,438,698✔
967
      .buffer = pCreate->buffer,
1,438,698✔
968
      .pageSize = pCreate->pageSize,
1,438,698✔
969
      .pages = pCreate->pages,
1,438,698✔
970
      .cacheLastSize = pCreate->cacheLastSize,
1,438,698✔
971
      .cacheLastShardBits = pCreate->cacheLastShardBits,
1,438,698✔
972
      .daysPerFile = pCreate->daysPerFile,
1,438,698✔
973
      .daysToKeep0 = pCreate->daysToKeep0,
1,438,698✔
974
      .daysToKeep1 = pCreate->daysToKeep1,
1,438,698✔
975
      .daysToKeep2 = pCreate->daysToKeep2,
1,438,698✔
976
      .keepTimeOffset = pCreate->keepTimeOffset,
1,438,698✔
977
      .minRows = pCreate->minRows,
1,438,698✔
978
      .maxRows = pCreate->maxRows,
1,438,698✔
979
      .walFsyncPeriod = pCreate->walFsyncPeriod,
1,438,698✔
980
      .walLevel = pCreate->walLevel,
1,438,698✔
981
      .precision = pCreate->precision,
1,438,698✔
982
      .compression = pCreate->compression,
1,438,698✔
983
      .replications = pCreate->replications,
1,438,698✔
984
      .strict = pCreate->strict,
1,438,698✔
985
      .cacheLast = pCreate->cacheLast,
1,438,698✔
986
      .hashMethod = 1,
987
      .schemaless = pCreate->schemaless,
1,438,698✔
988
      .walRetentionPeriod = pCreate->walRetentionPeriod,
1,438,698✔
989
      .walRetentionSize = pCreate->walRetentionSize,
1,438,698✔
990
      .walRollPeriod = pCreate->walRollPeriod,
1,438,698✔
991
      .walSegmentSize = pCreate->walSegmentSize,
1,438,698✔
992
      .sstTrigger = pCreate->sstTrigger,
1,438,698✔
993
      .hashPrefix = pCreate->hashPrefix,
1,438,698✔
994
      .hashSuffix = pCreate->hashSuffix,
1,438,698✔
995
      .ssChunkSize = pCreate->ssChunkSize,
1,438,698✔
996
      .ssKeepLocal = pCreate->ssKeepLocal,
1,438,698✔
997
      .ssCompact = pCreate->ssCompact,
1,438,698✔
998
      .tsdbPageSize = pCreate->tsdbPageSize,
1,438,698✔
999
      .withArbitrator = pCreate->withArbitrator,
1,438,698✔
1000
      .encryptAlgorithm = pCreate->encryptAlgorithm,
1,438,698✔
1001
      .compactInterval = pCreate->compactInterval,
1,438,698✔
1002
      .compactStartTime = pCreate->compactStartTime,
1,438,698✔
1003
      .compactEndTime = pCreate->compactEndTime,
1,438,698✔
1004
      .compactTimeOffset = pCreate->compactTimeOffset,
1,438,698✔
1005
      .isAudit = pCreate->isAudit,
1,438,698✔
1006
      .secureDelete = pCreate->secureDelete,
1,438,698✔
1007
  };
1008
  if (strlen(pCreate->encryptAlgrName) > 0) {
1,438,698✔
1009
    if (strncasecmp(pCreate->encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
7,433✔
1010
      if (pCreate->isAudit == 1) {
3,862✔
UNCOV
1011
        code = TSDB_CODE_AUDIT_MUST_ENCRYPT;
×
UNCOV
1012
        mError("db:%s, failed to create, encrypt algorithm not match for audit db, %s", pCreate->db,
×
1013
               pCreate->encryptAlgrName);
UNCOV
1014
        TAOS_RETURN(code);
×
1015
      }
1016
      dbObj.cfg.encryptAlgorithm = 0;
3,862✔
1017
    } else {
1018
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, pCreate->encryptAlgrName);
3,571✔
1019
      if (pEncryptAlgr != NULL) {
3,571✔
1020
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
2,756✔
1021
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
UNCOV
1022
          mError("db:%s, failed to create, encrypt algorithm not match, %s, expect type %d", pCreate->db,
×
1023
                 pCreate->encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
UNCOV
1024
          TAOS_RETURN(code);
×
1025
        }
1026
        dbObj.cfg.encryptAlgorithm = pEncryptAlgr->id;
2,756✔
1027
      } else {
1028
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
815✔
1029
        mError("db:%s, failed to create, encrypt algorithm not exist, %s", pCreate->db, pCreate->encryptAlgrName);
815✔
1030
        TAOS_RETURN(code);
815✔
1031
      }
1032
      if (tsDataKey[0] == '\0') {
2,756✔
UNCOV
1033
        code = TSDB_CODE_DNODE_INVALID_ENCRYPTKEY;
×
UNCOV
1034
        mError("db:%s, failed to create db since %s", pCreate->db, tstrerror(code));
×
UNCOV
1035
        TAOS_RETURN(code);
×
1036
      }
1037
    }
1038
  } else {
1039
    if (pCreate->isAudit == 1) {
1,431,265✔
1040
      code = TSDB_CODE_AUDIT_MUST_ENCRYPT;
1,580✔
1041
      mError("db:%s, failed to create, encrypt algorithm not match for audit db, %s", pCreate->db,
1,580✔
1042
             pCreate->encryptAlgrName);
1043
      TAOS_RETURN(code);
1,580✔
1044
    }
1045
  }
1046
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
1,436,303✔
1047
  dbObj.cfg.pRetensions = pCreate->pRetensions;
1,436,303✔
1048

1049
  if (pCreate->isAudit == 1) {
1,436,303✔
1050
    if (dbObj.cfg.daysToKeep2 < 2628000) {
319✔
UNCOV
1051
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
UNCOV
1052
      mError("db:%s, failed to create, keep not match for audit db, %d", pCreate->db, dbObj.cfg.daysToKeep2);
×
UNCOV
1053
      TAOS_RETURN(code);
×
1054
    }
1055
    if (dbObj.cfg.walLevel != 2) {
319✔
UNCOV
1056
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
UNCOV
1057
      mError("db:%s, failed to create, walLevel not match for audit db, %d", pCreate->db, dbObj.cfg.walLevel);
×
1058
      TAOS_RETURN(code);
×
1059
    }
1060
  }
1061
  dbObj.cfg.allowDrop = (uint8_t)pCreate->allowDrop;
1,436,303✔
1062

1063
#ifdef TD_ENTERPRISE  
1064
  // MAC: DB securityLevel
1065
  if (pCreate->securityLevel == 0) {
1,436,303✔
1066
    // Explicitly specified as 0: no need to check privilege, just set it to 0
1067
    dbObj.cfg.securityLevel = TSDB_DEFAULT_SECURITY_LEVEL;
5,197✔
1068
  } else if (pCreate->securityLevel > 0) {
1,431,106✔
1069
    // Require SYSSEC (PRIV_SECURITY_POLICY_ALTER) to set an explicit non-zero security level.
NEW
1070
    if (!mndUserHasMacLabelPriv(pMnode, pUser)) {
×
NEW
1071
      code = TSDB_CODE_MND_NO_RIGHTS;
×
NEW
1072
      mError("db:%s, failed to create, user %s lacks PRIV_SECURITY_POLICY_ALTER to set security_level",
×
1073
             pCreate->db, pUser->user);
NEW
1074
      TAOS_RETURN(code);
×
1075
    }
1076
    // MAC must be active to set db security_level > 0; before activation only user levels can be set.
NEW
1077
    if (pMnode->macActive != MAC_MODE_MANDATORY) {
×
NEW
1078
      code = TSDB_CODE_MAC_INSUFFICIENT_LEVEL;
×
NEW
1079
      mError("db:%s, failed to create, cannot set security_level > 0 before MAC is activated",
×
1080
             pCreate->db);
NEW
1081
      TAOS_RETURN(code);
×
1082
    }
1083
    // NRU: requested level must not exceed operator's clearance.
1084
    // (SYSSEC requires maxSecLevel=4 under MAC, so this check always passes in practice.)
NEW
1085
    if ((uint8_t)pCreate->securityLevel > pUser->maxSecLevel) {
×
NEW
1086
      code = TSDB_CODE_MAC_INSUFFICIENT_LEVEL;
×
NEW
1087
      mError("db:%s, failed to create, securityLevel(%d) exceeds user %s maxSecLevel(%d)",
×
1088
             pCreate->db, pCreate->securityLevel, pUser->user, pUser->maxSecLevel);
NEW
1089
      TAOS_RETURN(code);
×
1090
    }
NEW
1091
    dbObj.cfg.securityLevel = (uint8_t)pCreate->securityLevel;
×
1092
  } else {
1093
    // Not specified: MAC active inherit creator's maxSecLevel as default, otherwise default security_level = 0
1094
    dbObj.cfg.securityLevel =
1,431,106✔
1095
        pMnode->macActive == MAC_MODE_MANDATORY ? pUser->maxSecLevel : TSDB_DEFAULT_SECURITY_LEVEL;
1,431,106✔
1096
  }
1097
#endif
1098
  mndSetDefaultDbCfg(&dbObj.cfg);
1,436,303✔
1099

1100
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
1,436,303✔
UNCOV
1101
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, tstrerror(code));
×
UNCOV
1102
    TAOS_RETURN(code);
×
1103
  }
1104

1105
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
1,436,303✔
1106
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, tstrerror(code));
4,746✔
1107
    TAOS_RETURN(code);
4,746✔
1108
  }
1109

1110
  if (dbObj.cfg.hashPrefix > 0) {
1,431,557✔
1111
    int32_t dbLen = strlen(dbObj.name) + 1;
5,082✔
1112
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
5,082✔
1113
    dbObj.cfg.hashPrefix += dbLen;
5,082✔
1114
  } else if (dbObj.cfg.hashPrefix < 0) {
1,426,475✔
1115
    int32_t dbLen = strlen(dbObj.name) + 1;
5,185✔
1116
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
5,185✔
1117
    dbObj.cfg.hashPrefix -= dbLen;
5,185✔
1118
  }
1119

1120
  SVgObj *pVgroups = NULL;
1,431,557✔
1121
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
1,431,557✔
1122
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
4,373✔
1123
    TAOS_RETURN(code);
4,373✔
1124
  }
1125

1126
  // add database privileges for user
1127
  SUserObj *pNewUserDuped = NULL;
1,427,184✔
1128
#if 0
1129
  // We record the DB's owner id instread. The owner has the permission to create objects under the DB, but does not
1130
  // have default permissions on objects created by others .Therefore, readDbs/writeDbs are no longer applicable.
1131
  if (!pUser->superUser) {
1132
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
1133
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
1134
                    NULL, _OVER);
1135
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
1136
                    NULL, _OVER);
1137
    pNewUserDuped = &newUserObj;
1138
  }
1139
#else
1140
  // Considering the efficiency of use db privileges in some scenarios like insert operation, owned DBs is stored.
1141
  bool addOwned = false;
1,427,184✔
1142
  if (dbObj.cfg.isAudit == 1) {
1,427,184✔
1143
    if (taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT, sizeof(TSDB_ROLE_SYSAUDIT)) ||
319✔
NEW
1144
        taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT_LOG, sizeof(TSDB_ROLE_SYSAUDIT_LOG))) {
×
1145
      addOwned = true;
319✔
1146
    }
1147
    TAOS_CHECK_GOTO(mndSetAuditOwnedDbs(pMnode, pUser, &dbObj, &auditOwnedDbs), NULL, _OVER);
319✔
1148
  } else {
1149
    addOwned = true;
1,426,865✔
1150
  }
1151
  if (addOwned) {
1,427,184✔
1152
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
1,427,184✔
1153
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.ownedDbs, dbObj.name, strlen(dbObj.name) + 1, NULL, 0), NULL, _OVER);
1,427,184✔
1154
    pNewUserDuped = &newUserObj;
1,427,184✔
1155
  }
1156
#endif
1157

1158
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
1,427,184✔
1159
  if (pTrans == NULL) {
1,427,184✔
UNCOV
1160
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1161
    if (terrno != 0) code = terrno;
×
UNCOV
1162
    goto _OVER;
×
1163
  }
1164
  // mndTransSetSerial(pTrans);
1165
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
1,427,184✔
1166

1167
  mndTransSetDbName(pTrans, dbObj.name, NULL);
1,427,184✔
1168
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
1,427,184✔
1169

1170
  void   *pIter = NULL;
1,427,184✔
1171
  STrans *exitTrans = NULL;
1,427,184✔
1172

1173
  if (!pUser->superUser) {
1,427,184✔
1174
    while (1) {
1175
      pIter = sdbFetch(pMnode->pSdb, SDB_TRANS, pIter, (void **)&exitTrans);
5,369✔
1176
      if (pIter == NULL) break;
5,369✔
1177

UNCOV
1178
      if (exitTrans->conflict == TRN_CONFLICT_DB &&
×
UNCOV
1179
          strncmp(exitTrans->opername, "create-db", TSDB_TRANS_OPER_LEN) == 0) {
×
UNCOV
1180
        code = TSDB_CODE_MND_TRANS_CONFLICT;
×
UNCOV
1181
        sdbRelease(pMnode->pSdb, exitTrans);
×
UNCOV
1182
        sdbCancelFetch(pMnode->pSdb, pIter);
×
UNCOV
1183
        goto _OVER;
×
1184
      }
1185

UNCOV
1186
      sdbRelease(pMnode->pSdb, exitTrans);
×
1187
    }
1188
  }
1189

1190
  if (dbObj.cfg.isAudit) {
1,427,184✔
1191
    if (dbObj.cfg.numOfVgroups > 1) {
319✔
UNCOV
1192
      code = TSDB_CODE_AUDIT_DB_NOT_MULTI_VGROUP;
×
UNCOV
1193
      TAOS_CHECK_GOTO(code, NULL, _OVER);
×
1194
    }
1195
  }
1196

1197
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
1,427,184✔
1198
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
1,427,184✔
1199
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,427,184✔
1200
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,427,184✔
1201
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,427,184✔
1202
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped, auditOwnedDbs), NULL, _OVER);
1,427,184✔
1203
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,427,184✔
1204

1205
  if (dbObj.cfg.isAudit) {
1,427,184✔
1206
    // when create audit database, pVgroups num always is 1
1207
    TAOS_CHECK_GOTO(mndCreateAuditStb(pMnode, &dbObj, pUser, pTrans, pVgroups), NULL, _OVER);
319✔
1208
  }
1209
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,427,184✔
1210

1211
_OVER:
1,427,184✔
1212
  taosMemoryFree(pVgroups);
1,427,184✔
1213
  mndUserFreeObj(&newUserObj);
1,427,184✔
1214
  taosArrayDestroyEx(auditOwnedDbs, (FDelete)mndUserFreeObj);
1,427,184✔
1215
  mndTransDrop(pTrans);
1,427,184✔
1216
  TAOS_RETURN(code);
1,427,184✔
1217
}
1218

1219
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
1,451,306✔
1220
  int32_t    code = 0;
1,451,306✔
1221
  SSdb      *pSdb = pMnode->pSdb;
1,451,306✔
1222
  SDnodeObj *pDnode = NULL;
1,451,306✔
1223
  void      *pIter = NULL;
1,451,306✔
1224

1225
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
1226
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
1,451,306✔
UNCOV
1227
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
×
UNCOV
1228
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
×
UNCOV
1229
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
UNCOV
1230
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
1231
           tsEncryptionKeyStat);
1232
    goto _exit;
×
1233
  }
1234

UNCOV
1235
  int64_t curMs = taosGetTimestampMs();
×
UNCOV
1236
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
×
UNCOV
1237
    bool online = false;
×
UNCOV
1238
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
×
UNCOV
1239
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
UNCOV
1240
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
UNCOV
1241
      mError("db:%s, failed to check encryption key:%" PRIi8
×
1242
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
1243
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
1244
             tsEncryptionKeyChksum);
UNCOV
1245
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
1246
      sdbRelease(pSdb, pDnode);
×
UNCOV
1247
      break;
×
1248
    }
UNCOV
1249
    sdbRelease(pSdb, pDnode);
×
1250
  }
1251
#else
1252
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
1253
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
1254
    goto _exit;
1255
  }
1256
#endif
1257
_exit:
1,448,936✔
1258
  TAOS_RETURN(code);
1,451,306✔
1259
}
1260

1261
#ifndef TD_ENTERPRISE
1262
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
1263
  if (dnodeListStr[0] != 0) {
1264
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1265
    return terrno;
1266
  } else {
1267
    return 0;
1268
  }
1269
}
1270
#endif
1271

1272
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
1,569,565✔
1273
  SMnode      *pMnode = pReq->info.node;
1,569,565✔
1274
  int32_t      code = -1;
1,569,565✔
1275
  int32_t      lino = 0;
1,569,565✔
1276
  SDbObj      *pDb = NULL;
1,569,565✔
1277
  SUserObj    *pUser = NULL;
1,569,565✔
1278
  SCreateDbReq createReq = {0};
1,569,565✔
1279
  SArray      *dnodeList = NULL;
1,569,565✔
1280
  int64_t      tss = taosGetTimestampMs();
1,569,565✔
1281

1282
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
1,569,565✔
1283
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
1,569,565✔
1284

1285
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
1,569,565✔
1286
#ifdef WINDOWS
1287
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
1288
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
1289
    goto _OVER;
1290
  }
1291
#endif
1292
  mInfo("db:%s, start to create, vgroups:%d, isAudit:%d, daysToKeep2:%d", createReq.db, createReq.numOfVgroups, createReq.isAudit, createReq.daysToKeep2);
1,569,565✔
1293

1294
  pDb = mndAcquireDb(pMnode, createReq.db);
1,569,565✔
1295
  if (pDb != NULL) {
1,569,565✔
1296
    if (createReq.ignoreExist) {
106,257✔
1297
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
103,700✔
1298
      code = 0;
103,700✔
1299
      goto _OVER;
103,700✔
1300
    } else {
1301
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
2,557✔
1302
      goto _OVER;
2,557✔
1303
    }
1304
  } else {
1305
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
1,463,308✔
1306
      code = terrno;
11,635✔
1307
      goto _OVER;
11,635✔
1308
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
1,451,673✔
UNCOV
1309
      goto _OVER;
×
1310
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
1,451,673✔
1311
      // continue
1312
    } else {  // TSDB_CODE_APP_ERROR
UNCOV
1313
      goto _OVER;
×
1314
    }
1315
  }
1316

1317
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_CREATE_DB, NULL), &lino, _OVER);
1,451,673✔
1318

1319
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
1,451,306✔
1320

1321
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
1,451,306✔
1322
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
1,451,306✔
1323

1324
  if (createReq.replications == 2) {
1,451,306✔
1325
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
5,984✔
1326
  }
1327

1328
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
1,451,306✔
1329

1330
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
1,451,306✔
1331

1332
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pUser), &lino, _OVER);
1,438,698✔
1333

1334
#ifdef TD_ENTERPRISE
1335
  // MAC security checks on DB securityLevel (only when MAC is mandatory and level is explicitly set)
1336
  if (pMnode->macActive == MAC_MODE_MANDATORY && createReq.securityLevel >= 0) {
1,438,698✔
1337
    // NRU: user cannot create DB with securityLevel above their own clearance
NEW
1338
    if (!pUser->superUser && pUser->maxSecLevel < createReq.securityLevel) {
×
NEW
1339
      mError("user:%s, MAC NRU denied: cannot create DB with secLevel(%d) > maxSecLevel(%d)",
×
1340
             RPC_MSG_USER(pReq), createReq.securityLevel, pUser->maxSecLevel);
NEW
1341
      TAOS_CHECK_GOTO(TSDB_CODE_MAC_INSUFFICIENT_LEVEL, &lino, _OVER);
×
1342
    }
1343
    // NWD: user cannot create DB with securityLevel below their mandatory floor.
1344
    // Trusted principals (PRIV_SECURITY_POLICY_ALTER / SYSSEC) are exempt as trusted subjects.
NEW
1345
    if (!mndUserHasMacLabelPriv(pMnode, pUser) && createReq.securityLevel < pUser->minSecLevel) {
×
NEW
1346
      mError("user:%s, MAC NWD denied: cannot create DB with secLevel(%d) < minSecLevel(%d)",
×
1347
             RPC_MSG_USER(pReq), createReq.securityLevel, pUser->minSecLevel);
NEW
1348
      TAOS_CHECK_GOTO(TSDB_CODE_MAC_NO_WRITE_DOWN, &lino, _OVER);
×
1349
    }
1350
  }
1351
#endif
1352

1353
  if (sdbGetSize(pMnode->pSdb, SDB_MOUNT) > 0) {
1,438,698✔
UNCOV
1354
    TAOS_CHECK_GOTO(TSDB_CODE_MND_MOUNT_NOT_EMPTY, &lino, _OVER);
×
1355
  }
1356

1357
  if (createReq.isAudit == 1) {
1,438,698✔
1358
    SDbObj *pAuditDb = mndAcquireAuditDb(pMnode);
1,899✔
1359
    if (pAuditDb != NULL) {
1,899✔
UNCOV
1360
      mndReleaseDb(pMnode, pAuditDb);
×
UNCOV
1361
      mError("db:%s, audit db already exist, %s", createReq.db, pAuditDb->name);
×
UNCOV
1362
      TAOS_CHECK_GOTO(TSDB_CODE_AUDIT_DB_ALREADY_EXIST, &lino, _OVER);
×
1363
    }
1364
    char *realDbName = strchr(createReq.db, '.');
1,899✔
1365
    if (realDbName && strcmp(realDbName + 1, "log") == 0) {
1,899✔
UNCOV
1366
      mError("db:%s, failed to create, db name not allowed for audit db, %s", createReq.db, realDbName + 1);
×
UNCOV
1367
      TAOS_CHECK_GOTO(TSDB_CODE_MND_INVALID_DB, &lino, _OVER);
×
1368
    }
1369
  }
1370

1371
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
1,438,698✔
1372
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
1,427,184✔
1373

1374
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
1,427,184✔
1375
    SName name = {0};
1,427,184✔
1376
    if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
1,427,184✔
UNCOV
1377
      mError("db:%s, failed to parse db name", createReq.db);
×
1378
    int64_t tse = taosGetTimestampMs();
1,427,184✔
1379
    double  duration = (double)(tse - tss);
1,427,184✔
1380
    duration = duration / 1000;
1,427,184✔
1381
    auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen, duration, 0);
1,427,184✔
1382
  }
1383

1384
_OVER:
1,567,195✔
1385
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,569,565✔
1386
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
38,681✔
1387
  }
1388

1389
  mndReleaseDb(pMnode, pDb);
1,569,565✔
1390
  mndReleaseUser(pMnode, pUser);
1,569,565✔
1391
  tFreeSCreateDbReq(&createReq);
1,569,565✔
1392
  taosArrayDestroy(dnodeList);
1,569,565✔
1393

1394
  TAOS_RETURN(code);
1,569,565✔
1395
}
1396

1397
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
125,191✔
1398
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
125,191✔
1399

1400
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
125,191✔
1401
    pDb->cfg.buffer = pAlter->buffer;
963✔
1402
    code = 0;
963✔
1403
  }
1404

1405
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
125,191✔
1406
    pDb->cfg.pages = pAlter->pages;
2,445✔
1407
    code = 0;
2,445✔
1408
  }
1409

1410
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
125,191✔
UNCOV
1411
    pDb->cfg.pageSize = pAlter->pageSize;
×
UNCOV
1412
    code = 0;
×
1413
  }
1414

1415
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
125,191✔
UNCOV
1416
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
UNCOV
1417
    code = 0;
×
1418
  }
1419

1420
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
125,191✔
1421
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
17,342✔
1422
    code = 0;
17,342✔
1423
  }
1424

1425
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
125,191✔
1426
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
16,554✔
1427
    code = 0;
16,554✔
1428
  }
1429

1430
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
125,191✔
1431
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
13,451✔
1432
    code = 0;
13,451✔
1433
  }
1434

1435
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
125,191✔
1436
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
3,967✔
1437
    code = 0;
3,967✔
1438
  }
1439

1440
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
125,191✔
1441
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
9,480✔
1442
    code = 0;
9,480✔
1443
  }
1444

1445
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
125,191✔
1446
    pDb->cfg.walLevel = pAlter->walLevel;
9,477✔
1447
    code = 0;
9,477✔
1448
  }
1449

1450
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
125,191✔
1451
#if 1
UNCOV
1452
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1453
#else
1454
    pDb->cfg.strict = pAlter->strict;
1455
    code = 0;
1456
#endif
1457
  }
1458

1459
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
125,191✔
1460
    pDb->cfg.cacheLast = pAlter->cacheLast;
16,632✔
1461
    code = 0;
16,632✔
1462
  }
1463

1464
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
125,191✔
1465
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
2,615✔
1466
    code = 0;
2,615✔
1467
  }
1468

1469
  if (pAlter->cacheLastShardBits > 0 && pAlter->cacheLastShardBits != pDb->cfg.cacheLastShardBits) {
125,191✔
1470
    pDb->cfg.cacheLastShardBits = pAlter->cacheLastShardBits;
791✔
1471
    code = 0;
791✔
1472
  }
1473

1474
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
125,191✔
1475
    pDb->cfg.replications = pAlter->replications;
6,584✔
1476
    pDb->vgVersion++;
6,584✔
1477
    code = 0;
6,584✔
1478
  }
1479

1480
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
125,191✔
1481
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
1,379✔
1482
    pDb->vgVersion++;
1,379✔
1483
    code = 0;
1,379✔
1484
  }
1485

1486
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
125,191✔
1487
    pDb->cfg.minRows = pAlter->minRows;
1,576✔
1488
    pDb->vgVersion++;
1,576✔
1489
    code = 0;
1,576✔
1490
  }
1491

1492
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
125,191✔
1493
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
16,939✔
1494
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
10,860✔
1495
    pDb->vgVersion++;
10,860✔
1496
    code = 0;
10,860✔
1497
  }
1498

1499
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
125,191✔
1500
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
1,603✔
1501
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
788✔
1502
    pDb->vgVersion++;
788✔
1503
    code = 0;
788✔
1504
  }
1505

1506
  if (pAlter->ssKeepLocal >= TSDB_MIN_SS_KEEP_LOCAL && pAlter->ssKeepLocal <= TSDB_MAX_SS_KEEP_LOCAL && pAlter->ssKeepLocal != pDb->cfg.ssKeepLocal) {
125,191✔
1507
    pDb->cfg.ssKeepLocal = pAlter->ssKeepLocal;
788✔
1508
    pDb->vgVersion++;
788✔
1509
    code = 0;
788✔
1510
  }
1511

1512
  if (pAlter->ssCompact >= TSDB_MIN_SS_COMPACT && pAlter->ssCompact <= TSDB_MAX_SS_COMPACT && pAlter->ssCompact != pDb->cfg.ssCompact) {
125,191✔
1513
    pDb->cfg.ssCompact = pAlter->ssCompact;
788✔
1514
    pDb->vgVersion++;
788✔
1515
    code = 0;
788✔
1516
  }
1517

1518
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
125,191✔
1519
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
788✔
1520
    pDb->vgVersion++;
788✔
1521
    code = 0;
788✔
1522
  }
1523

1524
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
125,191✔
1525
    pDb->cfg.compactInterval = pAlter->compactInterval;
5,651✔
1526
    pDb->vgVersion++;
5,651✔
1527
    code = 0;
5,651✔
1528
  }
1529

1530
  bool compactTimeRangeChanged = false;
125,191✔
1531
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
125,191✔
1532
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
122,746✔
1533
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
122,746✔
1534
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
4,890✔
1535
    compactTimeRangeChanged = true;
4,890✔
1536
    code = 0;
4,890✔
1537
  }
1538

1539
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
125,191✔
1540
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
122,746✔
1541
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
4,890✔
1542
    compactTimeRangeChanged = true;
4,890✔
1543
    code = 0;
4,890✔
1544
  }
1545
  if(compactTimeRangeChanged) {
125,191✔
1546
    pDb->vgVersion++;
4,890✔
1547
  }
1548

1549
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
125,191✔
1550
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
9,753✔
1551
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
4,048✔
1552
    pDb->vgVersion++;
4,048✔
1553
    code = 0;
4,048✔
1554
  }
1555

1556
  if (pAlter->isAudit > -1 && pAlter->isAudit != pDb->cfg.isAudit) {
125,191✔
UNCOV
1557
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
UNCOV
1558
    return code;
×
1559
  }
1560

1561
  if (pAlter->allowDrop > -1 && pAlter->allowDrop != pDb->cfg.allowDrop) {
125,191✔
UNCOV
1562
    pDb->cfg.allowDrop = pAlter->allowDrop;
×
NEW
1563
    pDb->vgVersion++;
×
UNCOV
1564
    code = 0;
×
1565
  }
1566

1567
  if (pAlter->secureDelete > -1 && pAlter->secureDelete != pDb->cfg.secureDelete) {
125,191✔
1568
    pDb->cfg.secureDelete = pAlter->secureDelete;
620✔
1569
    pDb->vgVersion++;
620✔
1570
    code = 0;
620✔
1571
  }
1572

1573
  if (pAlter->securityLevel > -1 && ((uint8_t)pAlter->securityLevel != pDb->cfg.securityLevel)) {
125,191✔
1574
    pDb->cfg.securityLevel = (uint8_t)pAlter->securityLevel;
865✔
1575
    pDb->vgVersion++;
865✔
1576
    code = 0;
865✔
1577
  }
1578

1579
  TAOS_RETURN(code);
125,191✔
1580
}
1581

1582
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
92,415✔
1583
  int32_t  code = 0;
92,415✔
1584
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
92,415✔
1585
  if (pRedoRaw == NULL) {
92,415✔
UNCOV
1586
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1587
    if (terrno != 0) code = terrno;
×
UNCOV
1588
    TAOS_RETURN(code);
×
1589
  }
1590
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
92,415✔
UNCOV
1591
    sdbFreeRaw(pRedoRaw);
×
UNCOV
1592
    TAOS_RETURN(code);
×
1593
  }
1594

1595
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
92,415✔
UNCOV
1596
    sdbFreeRaw(pRedoRaw);
×
UNCOV
1597
    TAOS_RETURN(code);
×
1598
  }
1599
  return 0;
92,415✔
1600
}
1601

1602
int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
92,415✔
1603
  int32_t  code = 0;
92,415✔
1604
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
92,415✔
1605
  if (pCommitRaw == NULL) {
92,415✔
UNCOV
1606
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1607
    if (terrno != 0) code = terrno;
×
UNCOV
1608
    TAOS_RETURN(code);
×
1609
  }
1610
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
92,415✔
UNCOV
1611
    sdbFreeRaw(pCommitRaw);
×
UNCOV
1612
    TAOS_RETURN(code);
×
1613
  }
1614

1615
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
92,415✔
UNCOV
1616
    sdbFreeRaw(pCommitRaw);
×
UNCOV
1617
    TAOS_RETURN(code);
×
1618
  }
1619
  TAOS_RETURN(code);
92,415✔
1620
}
1621

1622
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
92,415✔
1623
  int32_t code = 0, lino = 0;
92,415✔
1624
  SSdb   *pSdb = pMnode->pSdb;
92,415✔
1625
  void   *pIter = NULL;
92,415✔
1626
  SVgObj *pVgroup = NULL;
92,415✔
1627
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
92,415✔
1628

1629
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
92,415✔
1630

1631
  while (1) {
1632
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
657,210✔
1633
    if (pIter == NULL) break;
657,210✔
1634

1635
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
565,563✔
1636
      SVgObj newVgroup = {0};
237,551✔
1637
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
237,551✔
1638
                      _err);
1639

1640
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
236,783✔
UNCOV
1641
        if (pNewDb->cfg.withArbitrator) {
×
UNCOV
1642
          SArbGroup arbGroup = {0};
×
UNCOV
1643
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
UNCOV
1644
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1645
        } else {
UNCOV
1646
          SArbGroup arbGroup = {0};
×
UNCOV
1647
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
UNCOV
1648
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1649
        }
1650
      }
1651
    }
1652

1653
    sdbRelease(pSdb, pVgroup);
564,795✔
1654
  }
1655

1656
  taosArrayDestroy(pArray);
91,647✔
1657
  TAOS_RETURN(code);
91,647✔
1658

1659
_err:
768✔
1660
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
768✔
1661

1662
  sdbCancelFetch(pSdb, pIter);
768✔
1663
  sdbRelease(pSdb, pVgroup);
768✔
1664
  taosArrayDestroy(pArray);
768✔
1665
  TAOS_RETURN(code);
768✔
1666
}
1667

1668
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
95,486✔
1669
  int32_t code = -1;
95,486✔
1670
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
95,486✔
1671
  if (pTrans == NULL) {
95,486✔
1672
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1673
    if (terrno != 0) code = terrno;
×
1674
    TAOS_RETURN(code);
×
1675
  }
1676
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
95,486✔
1677
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
95,486✔
1678

1679
  mndTransSetDbName(pTrans, pOld->name, NULL);
95,486✔
1680
  mndTransSetGroupParallel(pTrans);
95,486✔
1681
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
95,486✔
1682
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
95,486✔
1683
  TAOS_CHECK_GOTO(mndTransCheckConflictWithRetention(pMnode, pTrans), NULL, _OVER);
95,403✔
1684

1685
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
92,415✔
1686
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
92,415✔
1687
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
92,415✔
1688

1689
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
91,647✔
1690
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
91,647✔
1691
  code = 0;
91,647✔
1692

1693
_OVER:
95,486✔
1694
  mndTransDrop(pTrans);
95,486✔
1695
  TAOS_RETURN(code);
95,486✔
1696
}
1697

1698
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
128,118✔
1699
  SMnode     *pMnode = pReq->info.node;
128,118✔
1700
  int32_t     code = -1;
128,118✔
1701
  SDbObj     *pDb = NULL;
128,118✔
1702
  SAlterDbReq alterReq = {0};
128,118✔
1703
  SDbObj      dbObj = {0};
128,118✔
1704
  int64_t     tss = taosGetTimestampMs();
128,118✔
1705

1706
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
128,118✔
1707

1708
  mInfo("db:%s, start to alter, encryt_algr:%s", alterReq.db, alterReq.encryptAlgrName);
128,118✔
1709

1710
  pDb = mndAcquireDb(pMnode, alterReq.db);
128,118✔
1711
  if (pDb == NULL) {
128,118✔
1712
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1713
    if (terrno != 0) code = terrno;
×
1714
    goto _OVER;
×
1715
  }
1716

1717
#ifdef TD_ENTERPRISE
1718
  // MAC: only superUser or user with PRIV_SECURITY_POLICY_ALTER can ALTER DATABASE ... SECURITY_LEVEL
1719
  // Check this BEFORE general mndCheckDbPrivilege, since holder may not have ALTER grant on the DB.
1720
  if (alterReq.securityLevel > -1) {
128,118✔
1721
    SUserObj *pUser = NULL;
2,090✔
1722
    TAOS_CHECK_GOTO(mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pUser), NULL, _OVER);
2,090✔
1723
    if (!mndUserHasMacLabelPriv(pMnode, pUser)) {
2,090✔
NEW
1724
      mndReleaseUser(pMnode, pUser);
×
NEW
1725
      code = TSDB_CODE_MND_NO_RIGHTS;
×
NEW
1726
      mError("db:%s, failed to alter security_level, user %s lacks PRIV_SECURITY_POLICY_ALTER", alterReq.db,
×
1727
             RPC_MSG_USER(pReq));
NEW
1728
      goto _OVER;
×
1729
    }
1730
    // MAC must be active to set db security_level > 0; before activation only user levels can be set.
1731
    if (alterReq.securityLevel > 0 && pMnode->macActive != MAC_MODE_MANDATORY) {
2,090✔
1732
      mndReleaseUser(pMnode, pUser);
178✔
1733
      code = TSDB_CODE_MAC_INSUFFICIENT_LEVEL;
178✔
1734
      mError("db:%s, failed to alter, cannot set security_level > 0 before MAC is activated", alterReq.db);
178✔
1735
      goto _OVER;
178✔
1736
    }
1737
    // NRU: when MAC is mandatory, new level must not exceed operator's clearance.
1738
    // (SYSSEC requires maxSecLevel=4 under MAC, so this check always passes in practice.)
1739
    if (pMnode->macActive == MAC_MODE_MANDATORY && (uint8_t)alterReq.securityLevel > pUser->maxSecLevel) {
1,912✔
NEW
1740
      mndReleaseUser(pMnode, pUser);
×
NEW
1741
      code = TSDB_CODE_MAC_INSUFFICIENT_LEVEL;
×
NEW
1742
      mError("db:%s, failed to alter security_level(%d): exceeds user %s maxSecLevel(%d)", alterReq.db,
×
1743
             alterReq.securityLevel, RPC_MSG_USER(pReq), pUser->maxSecLevel);
NEW
1744
      goto _OVER;
×
1745
    }
1746
    mndReleaseUser(pMnode, pUser);
1,912✔
1747
  } else {
1748
    TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_ALTER_DB, pDb), NULL,
126,028✔
1749
                    _OVER);
1750
  }
1751
#endif
1752
  if (alterReq.replications == 2) {
127,782✔
1753
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
788✔
1754
  }
1755

1756
  int32_t numOfTopics = 0;
127,782✔
1757
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
127,782✔
1758

1759
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
127,782✔
UNCOV
1760
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
UNCOV
1761
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1762
           numOfTopics);
UNCOV
1763
    goto _OVER;
×
1764
  }
1765

1766
  if (pDb->cfg.isMount) {
127,782✔
UNCOV
1767
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
UNCOV
1768
    goto _OVER;
×
1769
  }
1770

1771
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
127,782✔
1772
  if (dbObj.cfg.pRetensions != NULL) {
127,782✔
1773
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
UNCOV
1774
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1775
  }
1776

1777
  if (pDb->cfg.isAudit == 1) {
127,782✔
UNCOV
1778
    if (alterReq.daysToKeep2 > -1 && alterReq.daysToKeep2 < 2628000) {
×
UNCOV
1779
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
UNCOV
1780
      mError("db:%s, failed to alter, keep not match for audit db, %d", alterReq.db, alterReq.daysToKeep2);
×
UNCOV
1781
      goto _OVER;
×
1782
    }
UNCOV
1783
    if (alterReq.walLevel > -1 && alterReq.walLevel != 2) {
×
UNCOV
1784
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
UNCOV
1785
      mError("db:%s, failed to alter, walLevel not match for audit db, %d", alterReq.db, alterReq.walLevel);
×
UNCOV
1786
      goto _OVER;
×
1787
    }
UNCOV
1788
    if (alterReq.isAudit == 0) {
×
UNCOV
1789
      code = TSDB_CODE_AUDIT_DB_NOT_ALLOW_CHANGE;
×
UNCOV
1790
      mError("db:%s, failed to alter, is not allowed to change audit db, %d", alterReq.db, alterReq.isAudit);
×
UNCOV
1791
      goto _OVER;
×
1792
    }
NEW
1793
    if (alterReq.securityLevel > -1) {
×
NEW
1794
      code = TSDB_CODE_AUDIT_DB_NOT_ALLOW_CHANGE;
×
NEW
1795
      mError("db:%s, failed to alter, security_level of audit db is immutable (fixed at 4)", alterReq.db);
×
NEW
1796
      goto _OVER;
×
1797
    }
1798
  }
1799

1800
  if (strlen(alterReq.encryptAlgrName) > 0) {
127,782✔
1801
    mInfo("db:%s, check encryt algr:%s", alterReq.db, alterReq.encryptAlgrName);
3,233✔
1802

1803
    if (strncasecmp(alterReq.encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
3,233✔
1804
      if (pDb->cfg.encryptAlgorithm != 0) {
815✔
UNCOV
1805
        code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
×
UNCOV
1806
        mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
×
UNCOV
1807
        goto _OVER;
×
1808
      }
1809
    } else {
1810
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, alterReq.encryptAlgrName);
2,418✔
1811
      if (pEncryptAlgr != NULL) {
2,418✔
1812
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
1,603✔
UNCOV
1813
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
UNCOV
1814
          mError("db:%s, failed to alter, encrypt algorithm not match, %s, expect type %d", alterReq.db,
×
1815
                 alterReq.encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
1816
          goto _OVER;
×
1817
        }
1818
        if (pEncryptAlgr->id != pDb->cfg.encryptAlgorithm) {
1,603✔
1819
          code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
1,603✔
1820
          mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
1,603✔
1821
          goto _OVER;
1,603✔
1822
        }
1823
      } else {
1824
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
815✔
1825
        mError("db:%s, failed to alter, encrypt algorithm not exist, %s", alterReq.db, alterReq.encryptAlgrName);
815✔
1826
        goto _OVER;
815✔
1827
      }
1828
    }
1829
  }
1830

1831
  // SYSSEC check for securityLevel already done above
1832
  // MAC: When raising DB security_level, all STBs in the DB must have level >= new level.
1833
  if (alterReq.securityLevel > -1 && (uint8_t)alterReq.securityLevel > pDb->cfg.securityLevel) {
125,364✔
1834
    SSdb *pSdb2 = pMnode->pSdb;
346✔
1835
    void *pStbIter = NULL;
346✔
1836
    while (1) {
1,211✔
1837
      SStbObj   *pStb = NULL;
1,557✔
1838
      ESdbStatus stbStatus;
1,557✔
1839
      pStbIter = sdbFetchAll(pSdb2, SDB_STB, pStbIter, (void **)&pStb, &stbStatus, true);
1,557✔
1840
      if (pStbIter == NULL) break;
1,557✔
1841
      if (pStb->dbUid == pDb->uid && pStb->securityLevel < (uint8_t)alterReq.securityLevel) {
1,384✔
1842
        sdbCancelFetch(pSdb2, pStbIter);
173✔
1843
        sdbRelease(pSdb2, pStb);
173✔
1844
        code = TSDB_CODE_MAC_OBJ_LEVEL_BELOW_DB;
173✔
1845
        mError("db:%s, failed to raise security_level to %d: stb %s has level %d", alterReq.db, alterReq.securityLevel,
173✔
1846
               pStb->name, pStb->securityLevel);
1847
        goto _OVER;
173✔
1848
      }
1849
      sdbRelease(pSdb2, pStb);
1,211✔
1850
    }
1851
  }
1852

1853
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
125,191✔
1854
  if (code != 0) {
125,191✔
1855
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
28,129✔
1856
    goto _OVER;
28,129✔
1857
  }
1858

1859
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
97,062✔
1860

1861
  dbObj.cfgVersion++;
95,486✔
1862
  dbObj.updateTime = taosGetTimestampMs();
95,486✔
1863
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
95,486✔
1864

1865
  if (dbObj.cfg.replications != pDb->cfg.replications) {
95,486✔
1866
    // return quickly, operation executed asynchronously
1867
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
5,796✔
1868
  } else {
1869
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
89,690✔
1870
  }
1871

1872
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
95,486✔
1873
    SName name = {0};
95,486✔
1874
    if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
95,486✔
UNCOV
1875
      mError("db:%s, failed to parse db name", alterReq.db);
×
1876

1877
    int64_t tse = taosGetTimestampMs();
95,486✔
1878
    double  duration = (double)(tse - tss);
95,486✔
1879
    duration = duration / 1000;
95,486✔
1880
    auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen, duration, 0);
95,486✔
1881
  }
1882

1883
_OVER:
128,118✔
1884
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
128,118✔
1885
    if (terrno != 0) code = terrno;
8,342✔
1886
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
8,342✔
1887
  }
1888

1889
  mndReleaseDb(pMnode, pDb);
128,118✔
1890
  taosArrayDestroy(dbObj.cfg.pRetensions);
128,118✔
1891
  tFreeSAlterDbReq(&alterReq);
128,118✔
1892

1893
  TAOS_RETURN(code);
128,118✔
1894
}
1895

1896
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb, char *algorithmsId) {
4,462,241✔
1897
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
4,462,241✔
1898
  cfgRsp->dbId = pDb->uid;
4,462,285✔
1899
  cfgRsp->ownerId = pDb->ownerId;
4,462,241✔
1900
  cfgRsp->cfgVersion = pDb->cfgVersion;
4,462,241✔
1901
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
4,462,241✔
1902
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
4,462,241✔
1903
  cfgRsp->buffer = pDb->cfg.buffer;
4,462,241✔
1904
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
4,462,285✔
1905
  cfgRsp->cacheShardBits = pDb->cfg.cacheLastShardBits;
4,461,497✔
1906
  cfgRsp->pageSize = pDb->cfg.pageSize;
4,462,241✔
1907
  cfgRsp->pages = pDb->cfg.pages;
4,461,453✔
1908
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
4,462,285✔
1909
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
4,462,241✔
1910
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
4,462,241✔
1911
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
4,461,453✔
1912
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
4,462,285✔
1913
  cfgRsp->minRows = pDb->cfg.minRows;
4,462,241✔
1914
  cfgRsp->maxRows = pDb->cfg.maxRows;
4,462,241✔
1915
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
4,462,241✔
1916
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
4,461,497✔
1917
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
4,461,453✔
1918
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
4,460,665✔
1919
  cfgRsp->walLevel = pDb->cfg.walLevel;
4,460,665✔
1920
  cfgRsp->precision = pDb->cfg.precision;
4,462,241✔
1921
  cfgRsp->compression = pDb->cfg.compression;
4,462,241✔
1922
  cfgRsp->replications = pDb->cfg.replications;
4,462,241✔
1923
  cfgRsp->strict = pDb->cfg.strict;
4,462,241✔
1924
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
4,460,709✔
1925
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
4,461,453✔
1926
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
4,461,453✔
1927
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
4,461,453✔
1928
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
4,461,453✔
1929
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
4,462,241✔
1930
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
4,460,665✔
1931
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
4,461,453✔
1932
  cfgRsp->schemaless = pDb->cfg.schemaless;
4,462,241✔
1933
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
4,460,665✔
1934
  cfgRsp->ssChunkSize = pDb->cfg.ssChunkSize;
4,461,453✔
1935
  cfgRsp->ssKeepLocal = pDb->cfg.ssKeepLocal;
4,460,665✔
1936
  cfgRsp->ssCompact = pDb->cfg.ssCompact;
4,460,665✔
1937
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
4,461,453✔
1938
  // cfgRsp->encryptAlgr = pDb->cfg.encryptAlgorithm;
1939
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
4,462,241✔
1940
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
4,462,241✔
1941
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
4,460,709✔
1942
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
4,462,241✔
1943
  cfgRsp->flags = pDb->cfg.flags;
4,461,453✔
1944
  tstrncpy(cfgRsp->algorithmsId, algorithmsId, sizeof(cfgRsp->algorithmsId));
4,460,665✔
1945
  cfgRsp->isAudit = pDb->cfg.isAudit;
4,462,285✔
1946
  cfgRsp->secureDelete = pDb->cfg.secureDelete;
4,460,665✔
1947
}
4,461,453✔
1948

1949
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
5,453,289✔
1950
  SMnode   *pMnode = pReq->info.node;
5,453,289✔
1951
  int32_t   code = -1;
5,453,289✔
1952
  SDbObj   *pDb = NULL;
5,453,289✔
1953
  SDbCfgReq cfgReq = {0};
5,453,289✔
1954
  SDbCfgRsp cfgRsp = {0};
5,453,289✔
1955

1956
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
5,453,289✔
1957

1958
  const char *pRealDb = strstr(cfgReq.db, ".");
5,453,333✔
1959
  if (pRealDb) {
5,453,333✔
1960
    ++pRealDb;
5,453,333✔
1961
  } else {
UNCOV
1962
    pRealDb = &cfgReq.db[0];
×
1963
  }
1964
  if (!IS_SYS_DBNAME(pRealDb)) {
5,453,333✔
1965
    pDb = mndAcquireDb(pMnode, cfgReq.db);
4,981,830✔
1966
    if (pDb == NULL) {
4,981,830✔
1967
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
604,842✔
1968
      if (terrno != 0) code = terrno;
604,842✔
1969
      goto _OVER;
604,842✔
1970
    }
1971

1972
    char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
4,376,988✔
1973
    mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
4,376,988✔
1974
    mndDumpDbCfgInfo(&cfgRsp, pDb, algrId);
4,376,988✔
1975
  }
1976

1977
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
4,847,703✔
1978
  void   *pRsp = rpcMallocCont(contLen);
4,847,659✔
1979
  if (pRsp == NULL) {
4,847,659✔
UNCOV
1980
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1981
    if (terrno != 0) code = terrno;
×
UNCOV
1982
    goto _OVER;
×
1983
  }
1984

1985
  int32_t ret = 0;
4,847,659✔
1986
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
4,847,659✔
UNCOV
1987
    code = ret;
×
UNCOV
1988
    goto _OVER;
×
1989
  }
1990

1991
  pReq->info.rsp = pRsp;
4,848,413✔
1992
  pReq->info.rspLen = contLen;
4,848,413✔
1993

1994
  code = 0;
4,847,625✔
1995

1996
_OVER:
5,452,467✔
1997

1998
  tFreeSDbCfgRsp(&cfgRsp);
5,453,329✔
1999

2000
  if (code != 0) {
5,453,251✔
2001
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
604,842✔
2002
  }
2003

2004
  mndReleaseDb(pMnode, pDb);
5,453,251✔
2005

2006
  TAOS_RETURN(code);
5,453,299✔
2007
}
2008

2009
int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
778,481✔
2010
  int32_t  code = 0;
778,481✔
2011
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
778,481✔
2012
  if (pRedoRaw == NULL) {
778,481✔
UNCOV
2013
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2014
    if (terrno != 0) code = terrno;
×
UNCOV
2015
    return -1;
×
2016
  }
2017
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
778,481✔
2018
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
778,481✔
2019

2020
  SSdb *pSdb = pMnode->pSdb;
778,481✔
2021
  void *pIter = NULL;
778,481✔
2022

2023
  while (1) {
4,728✔
2024
    SArbGroup *pArbGroup = NULL;
783,209✔
2025
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
783,209✔
2026
    if (pIter == NULL) break;
783,209✔
2027

2028
    if (pArbGroup->dbUid == pDb->uid) {
4,728✔
2029
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
4,728✔
UNCOV
2030
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2031
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
2032
        TAOS_RETURN(code);
×
2033
      }
2034
    }
2035

2036
    sdbRelease(pSdb, pArbGroup);
4,728✔
2037
  }
2038

2039
  TAOS_RETURN(code);
778,481✔
2040
}
2041

2042
int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
778,481✔
2043
  int32_t  code = 0;
778,481✔
2044
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
778,481✔
2045
  if (pCommitRaw == NULL) {
778,481✔
UNCOV
2046
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2047
    if (terrno != 0) code = terrno;
×
UNCOV
2048
    return -1;
×
2049
  }
2050
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
778,481✔
2051
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
778,481✔
2052

2053
  SSdb *pSdb = pMnode->pSdb;
778,481✔
2054
  void *pIter = NULL;
778,481✔
2055

2056
  while (1) {
4,728✔
2057
    SArbGroup *pArbGroup = NULL;
783,209✔
2058
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
783,209✔
2059
    if (pIter == NULL) break;
783,209✔
2060

2061
    if (pArbGroup->dbUid == pDb->uid) {
4,728✔
2062
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
4,728✔
UNCOV
2063
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2064
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
2065
        TAOS_RETURN(code);
×
2066
      }
2067
    }
2068

2069
    sdbRelease(pSdb, pArbGroup);
4,728✔
2070
  }
2071

2072
  while (1) {
3,881,922✔
2073
    SVgObj *pVgroup = NULL;
4,660,403✔
2074
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
4,660,403✔
2075
    if (pIter == NULL) break;
4,660,403✔
2076

2077
    if (pVgroup->dbUid == pDb->uid) {
3,881,922✔
2078
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
1,651,797✔
2079
      if (pVgRaw == NULL) {
1,651,797✔
UNCOV
2080
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2081
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2082
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2083
        if (terrno != 0) code = terrno;
×
UNCOV
2084
        TAOS_RETURN(code);
×
2085
      }
2086
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
1,651,797✔
UNCOV
2087
        sdbCancelFetch(pSdb, pIter);
×
2088
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2089
        TAOS_RETURN(code);
×
2090
      }
2091
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
1,651,797✔
UNCOV
2092
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2093
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2094
        TAOS_RETURN(code);
×
2095
      }
2096
    }
2097

2098
    sdbRelease(pSdb, pVgroup);
3,881,922✔
2099
  }
2100

2101
  while (1) {
1,353,841✔
2102
    SStbObj *pStb = NULL;
2,132,322✔
2103
    ESdbStatus status;
2,126,187✔
2104
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
2,132,322✔
2105
    if (pIter == NULL) break;
2,132,322✔
2106

2107
    if (pStb->dbUid == pDb->uid) {
1,353,841✔
2108
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
663,677✔
2109
      if (pStbRaw == NULL) {
663,677✔
UNCOV
2110
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2111
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
2112
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2113
        if (terrno != 0) code = terrno;
×
2114
        return -1;
×
2115
      }
2116
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
663,677✔
UNCOV
2117
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2118
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
2119
        return -1;
×
2120
      }
2121
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
663,677✔
UNCOV
2122
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
2123
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
2124
        return -1;
×
2125
      }
2126
    }
2127

2128
    sdbRelease(pSdb, pStb);
1,353,841✔
2129
  }
2130

2131
  TAOS_RETURN(code);
778,481✔
2132
}
2133

2134
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
1,651,797✔
2135
  int32_t code = 0;
1,651,797✔
2136
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
3,377,610✔
2137
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
1,725,813✔
2138
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
1,725,813✔
2139
  }
2140

2141
  TAOS_RETURN(code);
1,651,797✔
2142
}
2143

2144
int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
778,481✔
2145
  int32_t code = 0;
778,481✔
2146
  SSdb   *pSdb = pMnode->pSdb;
778,481✔
2147
  void   *pIter = NULL;
778,481✔
2148

2149
  while (1) {
3,881,922✔
2150
    SVgObj *pVgroup = NULL;
4,660,403✔
2151
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
4,660,403✔
2152
    if (pIter == NULL) break;
4,660,403✔
2153

2154
    if (pVgroup->dbUid == pDb->uid) {
3,881,922✔
2155
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
1,651,797✔
2156
        sdbCancelFetch(pSdb, pIter);
×
2157
        sdbRelease(pSdb, pVgroup);
×
2158
        TAOS_RETURN(code);
×
2159
      }
2160
    }
2161

2162
    sdbRelease(pSdb, pVgroup);
3,881,922✔
2163
  }
2164

2165
  TAOS_RETURN(code);
778,481✔
2166
}
2167

2168
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
1,360,861✔
2169
  int32_t    code = 0;
1,360,861✔
2170
  SDropDbRsp dropRsp = {0};
1,360,861✔
2171
  if (pDb != NULL) {
1,360,861✔
2172
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
777,833✔
2173
    dropRsp.uid = pDb->uid;
777,833✔
2174
  }
2175

2176
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
1,360,861✔
2177
  void   *pRsp = NULL;
1,360,861✔
2178
  if (useRpcMalloc) {
1,360,861✔
2179
    pRsp = rpcMallocCont(rspLen);
583,028✔
2180
  } else {
2181
    pRsp = taosMemoryMalloc(rspLen);
777,833✔
2182
  }
2183

2184
  if (pRsp == NULL) {
1,360,861✔
UNCOV
2185
    code = terrno;
×
UNCOV
2186
    TAOS_RETURN(code);
×
2187
  }
2188

2189
  int32_t ret = 0;
1,360,861✔
2190
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
1,360,861✔
2191
  *pRspLen = rspLen;
1,360,861✔
2192
  *ppRsp = pRsp;
1,360,861✔
2193
  TAOS_RETURN(code);
1,360,861✔
2194
}
2195

2196
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
778,225✔
2197
  int32_t code = -1;
778,225✔
2198

2199
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
778,225✔
2200
  if (pTrans == NULL) {
778,225✔
UNCOV
2201
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2202
    if (terrno != 0) code = terrno;
×
UNCOV
2203
    goto _OVER;
×
2204
  }
2205

2206
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
778,225✔
2207

2208
  mndTransSetDbName(pTrans, pDb->name, NULL);
778,225✔
2209
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
778,225✔
2210

2211
  if (mndTopicExistsForDb(pMnode, pDb)) {
778,225✔
2212
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
392✔
2213
    goto _OVER;
392✔
2214
  }
2215

2216
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2217
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2218
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2219
#ifdef TD_ENTERPRISE
2220
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2221
#endif
2222
  TAOS_CHECK_GOTO(mndDropTSMAsByDb(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2223
  TAOS_CHECK_GOTO(mndDropRsmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2224
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2225
  //TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
2226
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
777,833✔
2227
  TAOS_CHECK_GOTO(mndPrincipalRemoveDb(pMnode, pTrans, pDb, NULL, NULL), NULL, _OVER);
777,833✔
2228

2229
  int32_t rspLen = 0;
777,833✔
2230
  void   *pRsp = NULL;
777,833✔
2231
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
777,833✔
2232
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
777,833✔
2233

2234
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
777,833✔
2235
  code = 0;
777,833✔
2236

2237
_OVER:
778,225✔
2238
  mndTransDrop(pTrans);
778,225✔
2239
  TAOS_RETURN(code);
778,225✔
2240
}
2241

2242
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
1,368,167✔
2243
  SMnode    *pMnode = pReq->info.node;
1,368,167✔
2244
  int32_t    code = -1;
1,368,167✔
2245
  SDbObj    *pDb = NULL;
1,368,167✔
2246
  SDropDbReq dropReq = {0};
1,368,167✔
2247
  int64_t    tss = taosGetTimestampMs();
1,368,167✔
2248

2249
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
1,368,167✔
2250

2251
  mInfo("db:%s, start to drop", dropReq.db);
1,368,167✔
2252

2253
  pDb = mndAcquireDb(pMnode, dropReq.db);
1,368,167✔
2254
  if (pDb == NULL) {
1,368,167✔
2255
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
587,274✔
2256
    if (terrno != 0) code = terrno;
587,274✔
2257
    int32_t ret =
2258
        mndCheckDbPrivilegeByName(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_DROP_DB, dropReq.db, true);
587,274✔
2259
    if (ret != 0) {
587,274✔
UNCOV
2260
      code = ret;
×
UNCOV
2261
      goto _OVER;
×
2262
    }
2263
    if (dropReq.ignoreNotExists) {
587,274✔
2264
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
583,028✔
2265
    }
2266
    goto _OVER;
587,274✔
2267
  }
2268

2269
  if ((code = mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_DROP_DB, pDb))) {
780,893✔
UNCOV
2270
    goto _OVER;
×
2271
  }
2272

2273
  if(pDb->cfg.isMount) {
780,893✔
2274
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
648✔
2275
    goto _OVER;
648✔
2276
  }
2277

2278
  if(!pDb->cfg.allowDrop) {
780,245✔
UNCOV
2279
    code = TSDB_CODE_MND_DB_DROP_NOT_ALLOWED;
×
UNCOV
2280
    goto _OVER;
×
2281
  }
2282

2283
  SSdb *pSdb = pMnode->pSdb;
780,245✔
2284
  void *pIter = NULL;
780,245✔
2285

2286
  while (1) {
3,889,486✔
2287
    SVgObj *pVgroup = NULL;
4,669,731✔
2288
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
4,669,731✔
2289
    if (pIter == NULL) break;
4,669,731✔
2290

2291
    if (pVgroup->dbUid == pDb->uid) {
3,890,236✔
2292
      bool isFound = false;
1,652,913✔
2293
      for (int32_t i = 0; i < pVgroup->replica; i++) {
3,379,092✔
2294
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
1,726,929✔
2295
          isFound = true;
750✔
2296
          break;
750✔
2297
        }
2298
      }
2299
      if (!isFound) {
1,652,913✔
2300
        sdbRelease(pSdb, pVgroup);
1,652,163✔
2301
        continue;
1,652,163✔
2302
      }
2303
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
750✔
2304
      sdbCancelFetch(pSdb, pIter);
750✔
2305
      sdbRelease(pSdb, pVgroup);
750✔
2306
      goto _OVER;
750✔
2307
    }
2308

2309
    sdbRelease(pSdb, pVgroup);
2,237,323✔
2310
  }
2311

2312
  bool dbStream = false;
779,495✔
2313
  bool vtableStream = false;
779,495✔
2314
  mstCheckDbInUse(pMnode, dropReq.db, &dbStream, &vtableStream, true);
779,495✔
2315
  if (dbStream) {
779,495✔
2316
    code = TSDB_CODE_MND_STREAM_DB_IN_USE;
1,016✔
2317
    mError("db:%s used by streams, drop db not allowed", dropReq.db);
1,016✔
2318
    goto _OVER;
1,016✔
2319
  }
2320

2321
  if (vtableStream && !dropReq.force) {
778,479✔
2322
    code = TSDB_CODE_MND_STREAM_VTABLE_EXITS;
254✔
2323
    mError("db:%s, vtable stream exists, drop db not allowed", dropReq.db);
254✔
2324
    goto _OVER;
254✔
2325
  }
2326

2327
  code = mndDropDb(pMnode, pReq, pDb);
778,225✔
2328
  if (code == TSDB_CODE_SUCCESS) {
778,225✔
2329
    code = TSDB_CODE_ACTION_IN_PROGRESS;
777,833✔
2330
  }
2331

2332
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
778,225✔
2333
    SName name = {0};
778,225✔
2334
    if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
778,225✔
UNCOV
2335
      mError("db:%s, failed to parse db name", dropReq.db);
×
2336

2337
    int64_t tse = taosGetTimestampMs();
778,225✔
2338
    double  duration = (double)(tse - tss);
778,225✔
2339
    duration = duration / 1000;
778,225✔
2340
    auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen, duration, 0);
778,225✔
2341
  }
2342

2343
_OVER:
1,366,342✔
2344
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,368,167✔
2345
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
7,306✔
2346
  }
2347

2348
  mndReleaseDb(pMnode, pDb);
1,368,167✔
2349
  tFreeSDropDbReq(&dropReq);
1,368,167✔
2350
  TAOS_RETURN(code);
1,368,167✔
2351
}
2352

2353
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
31,665,473✔
2354
  int32_t numOfTables = 0;
31,665,473✔
2355
  int32_t vindex = 0;
31,665,473✔
2356
  SSdb   *pSdb = pMnode->pSdb;
31,665,473✔
2357

2358
  void *pIter = NULL;
31,665,507✔
2359
  while (vindex < pDb->cfg.numOfVgroups) {
134,728,301✔
2360
    SVgObj *pVgroup = NULL;
103,062,749✔
2361
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
103,062,749✔
2362
    if (pIter == NULL) break;
103,062,794✔
2363

2364
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
103,062,794✔
2365
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
63,618,345✔
2366
      vindex++;
63,618,345✔
2367
    }
2368

2369
    sdbRelease(pSdb, pVgroup);
103,062,760✔
2370
  }
2371

2372
  sdbCancelFetch(pSdb, pIter);
31,665,473✔
2373
  return numOfTables;
31,665,507✔
2374
}
2375

2376
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
30,862,408✔
2377
  int32_t vindex = 0;
30,862,408✔
2378
  SSdb   *pSdb = pMnode->pSdb;
30,862,408✔
2379

2380
  void *pIter = NULL;
30,861,620✔
2381
  while (1) {
69,209,182✔
2382
    SVgObj *pVgroup = NULL;
100,070,802✔
2383
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
100,071,590✔
2384
    if (pIter == NULL) break;
100,071,590✔
2385

2386
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
92,932,174✔
2387
      SVgroupInfo vgInfo = {0};
71,885,860✔
2388
      vgInfo.vgId = pVgroup->vgId;
71,885,891✔
2389
      vgInfo.hashBegin = pVgroup->hashBegin;
71,885,891✔
2390
      vgInfo.hashEnd = pVgroup->hashEnd;
71,885,891✔
2391
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
71,885,891✔
2392
      vgInfo.epSet.numOfEps = pVgroup->replica;
71,885,486✔
2393
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
148,476,165✔
2394
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
76,590,568✔
2395
        SEp       *pEp = &vgInfo.epSet.eps[gid];
76,590,568✔
2396
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
76,590,095✔
2397
        if (pDnode != NULL) {
76,590,385✔
2398
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
76,553,101✔
2399
          pEp->port = pDnode->port;
76,553,101✔
2400
        }
2401
        mndReleaseDnode(pMnode, pDnode);
76,590,385✔
2402
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
76,590,679✔
2403
          vgInfo.epSet.inUse = gid;
71,373,054✔
2404
        }
2405
      }
2406
      vindex++;
71,885,891✔
2407
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
71,885,891✔
UNCOV
2408
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
2409
      }
2410
    }
2411

2412
    sdbRelease(pSdb, pVgroup);
92,932,106✔
2413

2414
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
92,932,174✔
2415
      sdbCancelFetch(pSdb, pIter);
23,722,992✔
2416
      break;
23,722,992✔
2417
    }
2418
  }
2419
}
30,862,408✔
2420

2421
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
24,097,292✔
2422
  int32_t code = 0;
24,097,292✔
2423
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
24,097,292✔
2424
  if (pRsp->pVgroupInfos == NULL) {
24,097,288✔
UNCOV
2425
    code = terrno;
×
UNCOV
2426
    TAOS_RETURN(code);
×
2427
  }
2428

2429
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
24,097,288✔
2430

2431
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
24,097,292✔
2432
      pReq->stateTs < pDb->stateTs) {
1,557,950✔
2433
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
22,698,992✔
2434
  }
2435

2436
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
24,097,292✔
2437
  pRsp->uid = pDb->uid;
24,097,255✔
2438
  pRsp->vgVersion = pDb->vgVersion;
24,097,255✔
2439
  pRsp->stateTs = pDb->stateTs;
24,097,292✔
2440
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
24,097,255✔
2441
  pRsp->hashMethod = pDb->cfg.hashMethod;
24,097,292✔
2442
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
24,097,255✔
2443
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
24,097,292✔
2444
  pRsp->flags = pDb->cfg.flags;
24,097,292✔
2445
  TAOS_RETURN(code);
24,097,255✔
2446
}
2447

2448
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
31,098,841✔
2449
  SMnode   *pMnode = pReq->info.node;
31,098,841✔
2450
  int32_t   code = -1;
31,099,629✔
2451
  SDbObj   *pDb = NULL;
31,099,629✔
2452
  SUseDbReq usedbReq = {0};
31,099,629✔
2453
  SUseDbRsp usedbRsp = {0};
31,098,841✔
2454

2455
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
31,098,765✔
2456

2457
  char *p = strchr(usedbReq.db, '.');
31,099,542✔
2458
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
31,099,542✔
2459
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
6,793,467✔
2460
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
6,793,467✔
2461
    if (usedbReq.vgVersion < vgVersion) {
6,793,471✔
2462
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
6,785,229✔
2463
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
6,785,229✔
2464

2465
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
6,785,229✔
2466
      usedbRsp.vgVersion = vgVersion++;
6,785,229✔
2467
    } else {
2468
      usedbRsp.vgVersion = usedbReq.vgVersion;
8,242✔
2469
    }
2470
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
6,793,471✔
2471
    code = 0;
6,793,471✔
2472
  } else {
2473
    pDb = mndAcquireDb(pMnode, usedbReq.db);
24,306,109✔
2474
    if (pDb == NULL) {
24,305,332✔
2475
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
201,950✔
2476
      usedbRsp.uid = usedbReq.dbId;
201,950✔
2477
      usedbRsp.vgVersion = usedbReq.vgVersion;
201,950✔
2478
      usedbRsp.errCode = terrno;
201,950✔
2479

2480
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
201,950✔
2481
      if (terrno != 0) code = terrno;
201,950✔
2482
      goto _OVER;
201,950✔
2483
    } else {
2484
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_USE_DB, pDb), NULL, _OVER);
24,103,382✔
2485

2486
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
24,097,292✔
2487

2488
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
24,097,255✔
2489
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
2490
      code = 0;
24,097,292✔
2491
    }
2492
  }
2493

2494
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
30,890,763✔
2495
  void   *pRsp = rpcMallocCont(contLen);
30,890,763✔
2496
  if (pRsp == NULL) {
30,890,763✔
UNCOV
2497
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2498
    code = -1;
×
UNCOV
2499
    goto _OVER;
×
2500
  }
2501

2502
  int32_t ret = 0;
30,890,763✔
2503
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
30,890,763✔
UNCOV
2504
    code = ret;
×
UNCOV
2505
    goto _OVER;
×
2506
  }
2507

2508
  pReq->info.rsp = pRsp;
30,890,763✔
2509
  pReq->info.rspLen = contLen;
30,890,763✔
2510

2511
_OVER:
31,099,629✔
2512
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
31,099,629✔
2513
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
208,866✔
2514
  }
2515

2516
  mndReleaseDb(pMnode, pDb);
31,099,629✔
2517
  tFreeSUsedbRsp(&usedbRsp);
31,099,629✔
2518

2519
  TAOS_RETURN(code);
31,099,581✔
2520
}
2521

2522
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
9,171,180✔
2523
  int32_t       code = 0;
9,171,180✔
2524
  SDbHbBatchRsp batchRsp = {0};
9,171,180✔
2525
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
9,171,180✔
2526
  if (batchRsp.pArray == NULL) {
9,171,180✔
UNCOV
2527
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2528
    TAOS_RETURN(code);
×
2529
  }
2530

2531
  for (int32_t i = 0; i < numOfDbs; ++i) {
20,479,515✔
2532
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
11,308,335✔
2533
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
11,308,335✔
2534
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
11,308,335✔
2535
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
11,308,335✔
2536
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
11,308,335✔
2537
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
11,308,335✔
2538
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
11,308,335✔
2539

2540
    SDbHbRsp rsp = {0};
11,308,335✔
2541
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
11,308,335✔
2542
    rsp.dbId = pDbCacheInfo->dbId;
11,308,335✔
2543

2544
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
11,308,335✔
2545
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
10,608,208✔
2546
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
3,687,567✔
2547
      if (pDbCacheInfo->vgVersion >= vgVersion) {
3,687,567✔
2548
        continue;
3,333,758✔
2549
      }
2550

2551
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
354,187✔
2552
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
354,187✔
2553
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
354,187✔
2554

2555
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
354,187✔
2556
      rsp.useDbRsp->vgVersion = vgVersion++;
354,187✔
2557

2558
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
354,187✔
2559

2560
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
708,374✔
UNCOV
2561
        if (terrno != 0) code = terrno;
×
UNCOV
2562
        return code;
×
2563
      }
2564

2565
      continue;
354,187✔
2566
    }
2567

2568
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
7,620,768✔
2569
    if (pDb == NULL) {
7,620,768✔
2570
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
52,553✔
2571
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
52,553✔
2572
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
52,553✔
2573
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
52,553✔
2574
      rsp.useDbRsp->vgVersion = -1;
52,553✔
2575
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
105,106✔
UNCOV
2576
        if (terrno != 0) code = terrno;
×
UNCOV
2577
        return code;
×
2578
      }
2579
      continue;
52,553✔
2580
    }
2581

2582
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
7,568,215✔
2583

2584
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
7,568,215✔
2585
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
7,457,520✔
2586
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
6,488,361✔
2587
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
6,050,756✔
2588
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2589
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2590
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2591
      mndReleaseDb(pMnode, pDb);
6,050,756✔
2592
      continue;
6,050,756✔
2593
    } else {
2594
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
1,517,459✔
2595
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2596
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2597
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2598
    }
2599

2600
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
1,517,459✔
2601
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
85,253✔
2602
      char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
85,253✔
2603
      mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
85,253✔
2604
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb, algrId);
85,253✔
2605
    }
2606

2607
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
1,517,459✔
2608
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
1,005,114✔
2609
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
1,005,114✔
2610
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
1,005,114✔
2611
        bool    exist = false;
1,005,114✔
2612
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
1,005,114✔
2613
        if (TSDB_CODE_SUCCESS != code) {
1,005,114✔
2614
          mndReleaseDb(pMnode, pDb);
2,958✔
2615
          if (code != TSDB_CODE_NEED_RETRY) {
2,958✔
2616
            mError("db:%s, failed to get db tsmas", pDb->name);
2,958✔
2617
          } else {
UNCOV
2618
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2619
          }
2620
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
2,958✔
2621
          taosMemoryFreeClear(rsp.pTsmaRsp);
2,958✔
2622
          continue;
2,958✔
2623
        }
2624
        rsp.dbTsmaVersion = pDb->tsmaVersion;
1,002,156✔
2625
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
1,002,156✔
2626
      }
2627
    }
2628

2629
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
1,514,501✔
2630
        pDbCacheInfo->stateTs != pDb->stateTs) {
1,471,036✔
2631
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
1,024,037✔
2632
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
1,024,037✔
2633
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
1,024,037✔
UNCOV
2634
        mndReleaseDb(pMnode, pDb);
×
UNCOV
2635
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
UNCOV
2636
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
UNCOV
2637
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
UNCOV
2638
        continue;
×
2639
      }
2640

2641
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
1,024,037✔
2642
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
1,024,037✔
2643
      rsp.useDbRsp->uid = pDb->uid;
1,024,037✔
2644
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
1,024,037✔
2645
      rsp.useDbRsp->stateTs = pDb->stateTs;
1,024,037✔
2646
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
1,024,037✔
2647
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
1,024,037✔
2648
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
1,024,037✔
2649
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
1,024,037✔
2650
      rsp.useDbRsp->flags = pDb->cfg.flags;
1,024,037✔
2651
    }
2652

2653
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
3,029,002✔
2654
      mndReleaseDb(pMnode, pDb);
×
UNCOV
2655
      if (terrno != 0) code = terrno;
×
UNCOV
2656
      return code;
×
2657
    }
2658
    mndReleaseDb(pMnode, pDb);
1,514,501✔
2659
  }
2660

2661
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
9,171,180✔
2662
  void   *pRsp = taosMemoryMalloc(rspLen);
9,171,180✔
2663
  if (pRsp == NULL) {
9,171,180✔
UNCOV
2664
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2665
    tFreeSDbHbBatchRsp(&batchRsp);
×
UNCOV
2666
    return -1;
×
2667
  }
2668
  int32_t ret = 0;
9,171,180✔
2669
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
9,171,180✔
2670

2671
  *ppRsp = pRsp;
9,171,180✔
2672
  *pRspLen = rspLen;
9,171,180✔
2673

2674
  tFreeSDbHbBatchRsp(&batchRsp);
9,171,180✔
2675
  TAOS_RETURN(code);
9,171,180✔
2676
}
2677

2678
static int32_t mndSetTrimDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t startTs, STimeWindow tw,
39,275✔
2679
                                       SArray *vgroupIds, ETsdbOpType type, ETriggerType triggerType,
2680
                                       STrimDbRsp *pRsp) {
2681
  int32_t code = 0, lino = 0;
39,275✔
2682
  SSdb   *pSdb = pMnode->pSdb;
39,275✔
2683
  void   *pIter = NULL;
39,275✔
2684

2685
  SRetentionObj obj = {.optrType = type, .triggerType = triggerType};  // reuse SCompactObj struct
39,275✔
2686
  TAOS_CHECK_EXIT(mndAddRetentionToTrans(pMnode, pTrans, &obj, pDb, pRsp));
39,275✔
2687

2688
  int32_t j = 0;
39,275✔
2689
  int32_t numOfVgroups = taosArrayGetSize(vgroupIds);
39,275✔
2690
  if (numOfVgroups > 0) {
39,275✔
2691
    for (int32_t i = 0; i < numOfVgroups; i++) {
9,456✔
2692
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
7,092✔
2693
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
7,092✔
2694

2695
      if (pVgroup == NULL) {
7,092✔
2696
        mError("db:%s, vgroup:%" PRId64 " not exist", pDb->name, vgId);
2,364✔
2697
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
2,364✔
2698
      } else if (pVgroup->dbUid != pDb->uid) {
4,728✔
UNCOV
2699
        mError("db:%s, vgroup:%" PRId64 " not belong to db:%s", pDb->name, vgId, pDb->name);
×
UNCOV
2700
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2701
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
×
2702
      }
2703
      sdbRelease(pSdb, pVgroup);
4,728✔
2704
    }
2705

2706
    for (int32_t i = 0; i < numOfVgroups; i++) {
7,092✔
2707
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
4,728✔
2708
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
4,728✔
2709

2710
      if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
4,728✔
UNCOV
2711
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2712
        TAOS_RETURN(code);
×
2713
      }
2714

2715
      for (int32_t i = 0; i < pVgroup->replica; i++) {
9,456✔
2716
        SVnodeGid *gid = &pVgroup->vnodeGid[i];
4,728✔
2717
        if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
4,728✔
UNCOV
2718
          sdbRelease(pSdb, pVgroup);
×
UNCOV
2719
          TAOS_RETURN(code);
×
2720
        }
2721
        j++;
4,728✔
2722
      }
2723
      sdbRelease(pSdb, pVgroup);
4,728✔
2724
    }
2725
  } else {
2726
    while (1) {
354,168✔
2727
      SVgObj *pVgroup = NULL;
388,715✔
2728
      pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
388,715✔
2729
      if (pIter == NULL) break;
388,715✔
2730

2731
      if (pVgroup->dbUid == pDb->uid) {
354,168✔
2732
        if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
88,048✔
UNCOV
2733
          sdbCancelFetch(pSdb, pIter);
×
UNCOV
2734
          sdbRelease(pSdb, pVgroup);
×
UNCOV
2735
          TAOS_RETURN(code);
×
2736
        }
2737

2738
        for (int32_t i = 0; i < pVgroup->replica; i++) {
199,264✔
2739
          SVnodeGid *gid = &pVgroup->vnodeGid[i];
111,216✔
2740
          if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
111,216✔
UNCOV
2741
            sdbCancelFetch(pSdb, pIter);
×
UNCOV
2742
            sdbRelease(pSdb, pVgroup);
×
UNCOV
2743
            TAOS_RETURN(code);
×
2744
          }
2745
          j++;
111,216✔
2746
        }
2747
      }
2748

2749
      sdbRelease(pSdb, pVgroup);
354,168✔
2750
    }
2751
  }
2752
_exit:
39,275✔
2753
  TAOS_RETURN(code);
39,275✔
2754
}
2755

2756
static int32_t mndBuildTrimDbRsp(STrimDbRsp *rsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
36,911✔
2757
  int32_t code = 0;
36,911✔
2758
  int32_t rspLen = tSerializeSCompactDbRsp(NULL, 0, (SCompactDbRsp *)rsp);
36,911✔
2759
  void   *pRsp = NULL;
36,911✔
2760
  if (useRpcMalloc) {
36,911✔
2761
    pRsp = rpcMallocCont(rspLen);
×
2762
  } else {
2763
    pRsp = taosMemoryMalloc(rspLen);
36,911✔
2764
  }
2765

2766
  if (pRsp == NULL) {
36,911✔
UNCOV
2767
    code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2768
    TAOS_RETURN(code);
×
2769
  }
2770

2771
  (void)tSerializeSCompactDbRsp(pRsp, rspLen, (SCompactDbRsp *)rsp);
36,911✔
2772
  *pRspLen = rspLen;
36,911✔
2773
  *ppRsp = pRsp;
36,911✔
2774
  TAOS_RETURN(code);
36,911✔
2775
}
2776

2777
int32_t mndTrimDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb, STimeWindow tw, SArray *vgroupIds,
44,725✔
2778
                         ETsdbOpType type, ETriggerType triggerType) {
2779
  SSdb      *pSdb = pMnode->pSdb;
44,725✔
2780
  SVgObj    *pVgroup = NULL;
44,725✔
2781
  void      *pIter = NULL;
44,725✔
2782
  int32_t    code = 0, lino = 0;
44,725✔
2783
  STrimDbRsp rsp = {0};
44,725✔
2784
  bool       isExist = false;
44,725✔
2785

2786
  while (1) {
50,448✔
2787
    SRetentionObj *pObj = NULL;
95,173✔
2788
    pIter = sdbFetch(pMnode->pSdb, SDB_RETENTION, pIter, (void **)&pObj);
95,173✔
2789
    if (pIter == NULL) break;
95,173✔
2790

2791
    if (strcmp(pObj->dbname, pDb->name) == 0) {
50,448✔
2792
      isExist = true;
4,728✔
2793
    }
2794
    sdbRelease(pMnode->pSdb, pObj);
50,448✔
2795
  }
2796
  if (isExist) {
44,725✔
2797
    mInfo("trim db:%s already exist", pDb->name);
4,728✔
2798
    TAOS_RETURN(TSDB_CODE_MND_TRIM_ALREADY_EXIST);
4,728✔
2799
  }
2800

2801
  int64_t startTs = taosGetTimestampMs();
39,997✔
2802
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "trim-db");
39,997✔
2803
  if (pTrans == NULL) goto _exit;
39,997✔
2804
  mInfo("trans:%d, used to trim db:%s", pTrans->id, pDb->name);
39,997✔
2805
  mndTransSetDbName(pTrans, pDb->name, NULL);
39,997✔
2806
  TAOS_CHECK_EXIT(mndTrancCheckConflict(pMnode, pTrans));
39,997✔
2807
  TAOS_CHECK_EXIT(mndSetTrimDbRedoActions(pMnode, pTrans, pDb, startTs, tw, vgroupIds, type, triggerType, &rsp));
39,275✔
2808
  if (pReq) {
36,911✔
2809
    int32_t rspLen = 0;
36,911✔
2810
    void   *pRsp = NULL;
36,911✔
2811
    rsp.bAccepted = true;
36,911✔
2812
    TAOS_CHECK_EXIT(mndBuildTrimDbRsp(&rsp, &rspLen, &pRsp, false));
36,911✔
2813
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
36,911✔
2814
  }
2815
  TAOS_CHECK_EXIT(mndTransPrepare(pMnode, pTrans));
36,911✔
2816
_exit:
39,997✔
2817
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
39,997✔
2818
    mError("db:%s, failed at line %d to trim db since %s", pDb->name, lino, tstrerror(code));
3,086✔
2819
  }
2820
  mndTransDrop(pTrans);
39,997✔
2821
  TAOS_RETURN(code);
39,997✔
2822
}
2823

2824
#ifdef TD_ENTERPRISE
2825
static int32_t mndCheckRsmaInDb(SMnode *pMnode, SDbObj *pDb) {
14,972✔
2826
  int32_t   code = 0;
14,972✔
2827
  void     *pIter = NULL;
14,972✔
2828
  SRsmaObj *pRsma = NULL;
14,972✔
2829
  while ((pIter = sdbFetch(pMnode->pSdb, SDB_RSMA, pIter, (void **)&pRsma))) {
14,972✔
2830
    if (pRsma->dbUid == pDb->uid) {
14,972✔
2831
      sdbRelease(pMnode->pSdb, pRsma);
14,972✔
2832
      sdbCancelFetch(pMnode->pSdb, pIter);
14,972✔
2833
      TAOS_RETURN(code);
14,972✔
2834
    }
UNCOV
2835
    sdbRelease(pMnode->pSdb, pRsma);
×
2836
  }
2837
  TAOS_RETURN(TSDB_CODE_RSMA_NOT_EXIST);
×
2838
}
2839
#endif
2840

2841
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
23,997✔
2842
  SMnode    *pMnode = pReq->info.node;
23,997✔
2843
  int32_t    code = 0, lino = 0;
23,997✔
2844
  SDbObj    *pDb = NULL;
23,997✔
2845
  STrimDbReq trimReq = {0};
23,997✔
2846
  int64_t    tss = taosGetTimestampMs();
23,997✔
2847

2848
  TAOS_CHECK_EXIT(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq));
23,997✔
2849

2850
  mInfo("db:%s, start to trim, optr:%u, tw:%" PRId64 ",%" PRId64, trimReq.db, trimReq.optrType, trimReq.tw.skey,
23,997✔
2851
        trimReq.tw.ekey);
2852

2853
  pDb = mndAcquireDb(pMnode, trimReq.db);
23,997✔
2854
  if (pDb == NULL) {
23,997✔
UNCOV
2855
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2856
    if (terrno != 0) code = terrno;
×
UNCOV
2857
    TAOS_CHECK_EXIT(code);
×
2858
  }
2859

2860
  TAOS_CHECK_EXIT(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_TRIM_DB, pDb));
23,997✔
2861

2862
  if (pDb->cfg.isMount) {
23,997✔
UNCOV
2863
    TAOS_CHECK_EXIT(TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT);
×
2864
  }
2865

2866
#ifdef TD_ENTERPRISE
2867
  if (trimReq.optrType == TSDB_OPTR_ROLLUP) {
23,997✔
2868
    TAOS_CHECK_EXIT(mndCheckRsmaInDb(pMnode, pDb));
14,972✔
2869
  }
2870
#endif
2871

2872
  TAOS_CHECK_EXIT(mndTrimDb(pMnode, pReq, pDb, trimReq.tw, trimReq.vgroupIds, trimReq.optrType, trimReq.triggerType));
23,997✔
2873
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
16,905✔
2874

2875
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
16,905✔
2876
    SName name = {0};
16,905✔
2877
    if (tNameFromString(&name, trimReq.db, T_NAME_ACCT | T_NAME_DB) < 0) {
16,905✔
UNCOV
2878
      mWarn("db:%s, failed at line %d to parse db name", trimReq.db, __LINE__);
×
2879
    }
2880

2881
    char optrType[16] = {0};
16,905✔
2882
    (void)snprintf(optrType, sizeof(optrType), "%u", trimReq.optrType);
16,905✔
2883
    int64_t tse = taosGetTimestampMs();
16,905✔
2884
    double  duration = (double)(tse - tss);
16,905✔
2885
    duration = duration / 1000;
16,905✔
2886
    auditRecord(pReq, pMnode->clusterId, "trimDB", name.dbname, optrType, trimReq.sql, trimReq.sqlLen, duration, 0);
16,905✔
2887
  }
2888

2889
_exit:
23,997✔
2890
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
23,997✔
2891
    mError("db:%s, failed at line %d to process trim db req since %s, optr:%u", trimReq.db, lino, tstrerror(code),
7,092✔
2892
           trimReq.optrType);
2893
  }
2894

2895
  mndReleaseDb(pMnode, pDb);
23,997✔
2896
  tFreeSTrimDbReq(&trimReq);
23,997✔
2897
  TAOS_RETURN(code);
23,997✔
2898
}
2899

2900
static int32_t mndBuildSsMigrateDbRsp(SSsMigrateDbRsp *pSsMigrateRsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
2,364✔
2901
  int32_t code = 0;
2,364✔
2902
  int32_t rspLen = tSerializeSSsMigrateDbRsp(NULL, 0, pSsMigrateRsp);
2,364✔
2903
  void   *pRsp = NULL;
2,364✔
2904
  if (useRpcMalloc) {
2,364✔
UNCOV
2905
    pRsp = rpcMallocCont(rspLen);
×
2906
  } else {
2907
    pRsp = taosMemoryMalloc(rspLen);
2,364✔
2908
  }
2909

2910
  if (pRsp == NULL) {
2,364✔
UNCOV
2911
    code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2912
    TAOS_RETURN(code);
×
2913
  }
2914

2915
  (void)tSerializeSSsMigrateDbRsp(pRsp, rspLen, pSsMigrateRsp);
2,364✔
2916
  *pRspLen = rspLen;
2,364✔
2917
  *ppRsp = pRsp;
2,364✔
2918
  TAOS_RETURN(code);
2,364✔
2919
}
2920

2921
static int32_t mndTrimDbWal(SMnode *pMnode, SDbObj *pDb) {
1,206✔
2922
  SSdb   *pSdb = pMnode->pSdb;
1,206✔
2923
  SVgObj *pVgroup = NULL;
1,206✔
2924
  void   *pIter = NULL;
1,206✔
2925
  int32_t code = 0;
1,206✔
2926

2927
  // Iterate through all vgroups in this database
2928
  while (1) {
2,412✔
2929
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,618✔
2930
    if (pIter == NULL) break;
3,618✔
2931

2932
    // Check if this vgroup belongs to the database
2933
    if (pVgroup->dbUid != pDb->uid) {
2,412✔
UNCOV
2934
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2935
      continue;
×
2936
    }
2937

2938
    // Prepare message to send to vnode
2939
    int32_t   contLen = sizeof(SMsgHead);
2,412✔
2940
    SMsgHead *pHead = rpcMallocCont(contLen);
2,412✔
2941
    if (pHead == NULL) {
2,412✔
UNCOV
2942
      sdbCancelFetch(pSdb, pVgroup);
×
UNCOV
2943
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2944
      code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2945
      break;
×
2946
    }
2947
    pHead->contLen = htonl(contLen);
2,412✔
2948
    pHead->vgId = htonl(pVgroup->vgId);
2,412✔
2949

2950
    // Send TRIM WAL request to vnode
2951
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM_WAL, .pCont = pHead, .contLen = contLen};
2,412✔
2952
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
2,412✔
2953
    code = tmsgSendReq(&epSet, &rpcMsg);
2,412✔
2954
    if (code != 0) {
2,412✔
UNCOV
2955
      mError("vgId:%d, failed to send vnode-trim-wal request since 0x%x", pVgroup->vgId, code);
×
2956
    } else {
2957
      mInfo("vgId:%d, send vnode-trim-wal request to vnode", pVgroup->vgId);
2,412✔
2958
    }
2959
    sdbRelease(pSdb, pVgroup);
2,412✔
2960
  }
2961

2962
  return code;
1,206✔
2963
}
2964

2965
static int32_t mndProcessTrimDbWalReq(SRpcMsg *pReq) {
1,206✔
2966
  SMnode    *pMnode = pReq->info.node;
1,206✔
2967
  int32_t    code = -1;
1,206✔
2968
  SDbObj    *pDb = NULL;
1,206✔
2969
  STrimDbReq trimReq = {0};
1,206✔
2970

2971
  TAOS_CHECK_GOTO(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq), NULL, _OVER);
1,206✔
2972

2973
  mInfo("db:%s, start to trim WAL", trimReq.db);
1,206✔
2974

2975
  pDb = mndAcquireDb(pMnode, trimReq.db);
1,206✔
2976
  if (pDb == NULL) {
1,206✔
UNCOV
2977
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2978
    if (terrno != 0) code = terrno;
×
UNCOV
2979
    goto _OVER;
×
2980
  }
2981

2982
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_TRIM_DB, pDb), NULL, _OVER);
1,206✔
2983

2984
  code = mndTrimDbWal(pMnode, pDb);
1,206✔
2985

2986
_OVER:
1,206✔
2987
  if (code != 0) {
1,206✔
UNCOV
2988
    mError("db:%s, failed to process trim db wal req since %s", trimReq.db, terrstr());
×
2989
  }
2990

2991
  mndReleaseDb(pMnode, pDb);
1,206✔
2992
  TAOS_RETURN(code);
1,206✔
2993
}
2994

2995
int32_t mndSsMigrateDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
2,364✔
2996
  int32_t       code = 0;
2,364✔
2997
  SSsMigrateDbRsp ssMigrateRsp = {0};
2,364✔
2998

2999
  bool  isExist = false;
2,364✔
3000
  void *pIter = NULL;
2,364✔
UNCOV
3001
  while (1) {
×
3002
    SSsMigrateObj *pSsMigrate = NULL;
2,364✔
3003
    pIter = sdbFetch(pMnode->pSdb, SDB_SSMIGRATE, pIter, (void **)&pSsMigrate);
2,364✔
3004
    if (pIter == NULL) break;
2,364✔
3005

UNCOV
3006
    if (strcmp(pSsMigrate->dbname, pDb->name) == 0) {
×
UNCOV
3007
      isExist = true;
×
3008
    }
UNCOV
3009
    sdbRelease(pMnode->pSdb, pSsMigrate);
×
3010
  }
3011
  if (isExist) {
2,364✔
UNCOV
3012
    mInfo("ssmigrate db:%s already exist", pDb->name);
×
3013

UNCOV
3014
    if (pReq) {
×
UNCOV
3015
      int32_t rspLen = 0;
×
UNCOV
3016
      void   *pRsp = NULL;
×
UNCOV
3017
      ssMigrateRsp.ssMigrateId = 0;
×
UNCOV
3018
      ssMigrateRsp.bAccepted = false;
×
UNCOV
3019
      TAOS_CHECK_RETURN(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, true));
×
3020

UNCOV
3021
      pReq->info.rsp = pRsp;
×
UNCOV
3022
      pReq->info.rspLen = rspLen;
×
3023
    }
3024

UNCOV
3025
    return TSDB_CODE_MND_SSMIGRATE_ALREADY_EXIST;
×
3026
  }
3027

3028
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "ssmigrate-db");
2,364✔
3029
  if (pTrans == NULL) {
2,364✔
UNCOV
3030
    mError("failed to create ssmigrate-db trans since %s", terrstr());
×
3031
    goto _OVER;
×
3032
  }
3033

3034
  mInfo("trans:%d, used to ssmigrate db:%s", pTrans->id, pDb->name);
2,364✔
3035
  mndTransSetDbName(pTrans, pDb->name, NULL);
2,364✔
3036
  TAOS_CHECK_GOTO(mndTrancCheckConflict(pMnode, pTrans), NULL, _OVER);
2,364✔
3037

3038
  SSsMigrateObj ssMigrate = { .startTime = taosGetTimestampMs() };
2,364✔
3039
  TAOS_CHECK_GOTO(mndAddSsMigrateToTran(pMnode, pTrans, &ssMigrate, pDb), NULL, _OVER);
2,364✔
3040

3041
  if (pReq) {
2,364✔
3042
    int32_t rspLen = 0;
2,364✔
3043
    void   *pRsp = NULL;
2,364✔
3044
    ssMigrateRsp.ssMigrateId = ssMigrate.id;
2,364✔
3045
    ssMigrateRsp.bAccepted = true;
2,364✔
3046
    TAOS_CHECK_GOTO(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, false), NULL, _OVER);
2,364✔
3047
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
2,364✔
3048
  }
3049

3050
  if (mndTransPrepare(pMnode, pTrans) != 0) goto _OVER;
2,364✔
3051
  code = 0;
2,364✔
3052

3053
_OVER:
2,364✔
3054
  mndTransDrop(pTrans);
2,364✔
3055
  TAOS_RETURN(code);
2,364✔
3056
}
3057

3058

3059
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq) {
2,838✔
3060
  SMnode         *pMnode = pReq->info.node;
2,838✔
3061
  int32_t         code = -1;
2,838✔
3062
  SDbObj         *pDb = NULL;
2,838✔
3063
  SSsMigrateDbReq ssMigrateReq = {0};
2,838✔
3064

3065
  if (!tsSsEnabled) {
2,838✔
3066
    code = TSDB_CODE_OPS_NOT_SUPPORT;
474✔
3067
    goto _OVER;
474✔
3068
  }
3069

3070
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_SHARED_STORAGE), NULL, _OVER);
2,364✔
3071
  TAOS_CHECK_GOTO(tDeserializeSSsMigrateDbReq(pReq->pCont, pReq->contLen, &ssMigrateReq), NULL, _OVER);
2,364✔
3072

3073
  mInfo("db:%s, start to ssmigrate", ssMigrateReq.db);
2,364✔
3074

3075
  pDb = mndAcquireDb(pMnode, ssMigrateReq.db);
2,364✔
3076
  if (pDb == NULL) {
2,364✔
UNCOV
3077
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
3078
    if (terrno != 0) code = terrno;
×
UNCOV
3079
    goto _OVER;
×
3080
  }
3081

3082
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_SSMIGRATE_DB, pDb), NULL, _OVER);
2,364✔
3083

3084
  if(pDb->cfg.isMount) {
2,364✔
UNCOV
3085
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
UNCOV
3086
    goto _OVER;
×
3087
  }
3088

3089
  code = mndSsMigrateDb(pMnode, pReq, pDb);
2,364✔
3090

3091
_OVER:
2,838✔
3092
  if (code != 0) {
2,838✔
3093
    mError("db:%s, failed to process ssmigrate db req since %s", ssMigrateReq.db, tstrerror(code));
474✔
3094
  }
3095

3096
  mndReleaseDb(pMnode, pDb);
2,838✔
3097
  TAOS_RETURN(code);
2,838✔
3098
}
3099

3100
const char *mndGetDbStr(const char *src) {
21,960,398✔
3101
  char *pos = strstr(src, TS_PATH_DELIMITER);
21,960,398✔
3102
  if (pos != NULL) ++pos;
21,960,398✔
3103
  if (pos == NULL) return src;
21,960,398✔
3104
  return pos;
20,418,147✔
3105
}
3106

3107
const char *mndGetStableStr(const char *src) {
18,168,941✔
3108
  char *pos = strstr(src, TS_PATH_DELIMITER);
18,168,941✔
3109
  if (pos != NULL) ++pos;
18,168,941✔
3110
  if (pos == NULL) return src;
18,168,941✔
3111
  return mndGetDbStr(pos);
18,168,941✔
3112
}
3113

3114
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
×
UNCOV
3115
  int64_t v = 0;
×
UNCOV
3116
  switch (unit) {
×
UNCOV
3117
    case 's':
×
UNCOV
3118
      v = val / 1000;
×
UNCOV
3119
      break;
×
UNCOV
3120
    case 'm':
×
UNCOV
3121
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
×
UNCOV
3122
      break;
×
UNCOV
3123
    case 'h':
×
UNCOV
3124
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
UNCOV
3125
      break;
×
UNCOV
3126
    case 'd':
×
3127
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
×
UNCOV
3128
      break;
×
UNCOV
3129
    case 'w':
×
UNCOV
3130
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
UNCOV
3131
      break;
×
3132
    default:
×
3133
      break;
×
3134
  }
3135

UNCOV
3136
  return v;
×
3137
}
3138

3139
static char *buildRetension(SArray *pRetension) {
648,494✔
3140
  size_t size = taosArrayGetSize(pRetension);
648,494✔
3141
  if (size == 0) {
648,494✔
3142
    return NULL;
648,494✔
3143
  }
3144

3145
  char       *p1 = taosMemoryCalloc(1, 100);
×
UNCOV
3146
  SRetention *p = taosArrayGet(pRetension, 0);
×
3147

3148
  int32_t len = 2;
×
3149

UNCOV
3150
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
3151
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
UNCOV
3152
  len += snprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
3153

UNCOV
3154
  if (size > 1) {
×
UNCOV
3155
    len += snprintf(p1 + len, 100 - len, ",");
×
3156
    p = taosArrayGet(pRetension, 1);
×
3157

UNCOV
3158
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
UNCOV
3159
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
UNCOV
3160
    len += snprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
3161
  }
3162

UNCOV
3163
  if (size > 2) {
×
UNCOV
3164
    len += snprintf(p1 + len, 100 - len, ",");
×
UNCOV
3165
    p = taosArrayGet(pRetension, 2);
×
3166

UNCOV
3167
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
UNCOV
3168
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
UNCOV
3169
    len += snprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
3170
  }
3171

UNCOV
3172
  varDataSetLen(p1, len);
×
UNCOV
3173
  return p1;
×
3174
}
3175

3176
static const char *getCacheModelStr(int8_t cacheModel) {
648,494✔
3177
  switch (cacheModel) {
648,494✔
3178
    case TSDB_CACHE_MODEL_NONE:
587,921✔
3179
      return TSDB_CACHE_MODEL_NONE_STR;
587,921✔
3180
    case TSDB_CACHE_MODEL_LAST_ROW:
3,648✔
3181
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
3,648✔
3182
    case TSDB_CACHE_MODEL_LAST_VALUE:
7,351✔
3183
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
7,351✔
3184
    case TSDB_CACHE_MODEL_BOTH:
49,574✔
3185
      return TSDB_CACHE_MODEL_BOTH_STR;
49,574✔
UNCOV
3186
    default:
×
UNCOV
3187
      break;
×
3188
  }
UNCOV
3189
  return "unknown";
×
3190
}
3191

3192
static void getEncryptAlgorithmStr(SMnode *pMnode, int8_t encryptAlgorithm, char *out) {
648,494✔
3193
  switch (encryptAlgorithm) {
648,494✔
3194
    case TSDB_ENCRYPT_ALGO_NONE: {
647,192✔
3195
      tstrncpy(out, TSDB_ENCRYPT_ALGO_NONE_STR, TSDB_ENCRYPT_ALGR_NAME_LEN);
647,192✔
3196
      break;
647,192✔
3197
    }
3198
    default: {
1,302✔
3199
      SEncryptAlgrObj *obj = mndAcquireEncryptAlgrById(pMnode, encryptAlgorithm);
1,302✔
3200
      if (obj != NULL) {
1,302✔
3201
        tstrncpy(out, obj->algorithm_id, TSDB_ENCRYPT_ALGR_NAME_LEN);
1,302✔
3202
        mndReleaseEncryptAlgr(pMnode, obj);
1,302✔
3203
      }
3204
      break;
1,302✔
3205
    }
3206
  }
3207
}
648,494✔
3208

3209
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
2,450,682✔
3210
  if (pDb->cfg.replications == 1) return true;
2,450,682✔
3211

3212
  SSdb *pSdb = pMnode->pSdb;
474,617✔
3213
  void *pIter = NULL;
474,617✔
3214
  bool  isReady = true;
474,617✔
3215
  while (1) {
2,325,893✔
3216
    SVgObj *pVgroup = NULL;
2,800,510✔
3217
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
2,800,510✔
3218
    if (pIter == NULL) break;
2,800,510✔
3219

3220
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
2,325,893✔
3221
      bool hasLeader = false;
1,386,262✔
3222
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
5,488,564✔
3223
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
4,102,302✔
3224
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
3,614,238✔
3225
          hasLeader = true;
488,064✔
3226
        }
3227
      }
3228
      if (!hasLeader) isReady = false;
1,386,262✔
3229
    }
3230
    sdbRelease(pSdb, pVgroup);
2,325,893✔
3231
  }
3232

3233
  return isReady;
474,617✔
3234
}
3235

3236
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
1,718,653✔
3237
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo,
3238
                              SSHashObj **ppUidNames) {
3239
  int32_t cols = 0;
1,718,653✔
3240
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
1,718,653✔
3241
  char   *buf = taosMemoryMalloc(bytes);
1,718,653✔
3242
  if (buf == NULL) {
1,718,653✔
3243
    mError("db:%s, failed to malloc buffer", pDb->name);
×
3244
    return;
×
3245
  }
3246
  int32_t code = 0;
1,718,653✔
3247
  int32_t lino = 0;
1,718,653✔
3248

3249
  const char *name = mndGetDbStr(pDb->name);
1,718,653✔
3250
  if (name != NULL) {
1,718,653✔
3251
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
1,718,653✔
3252
  } else {
3253
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
3254
  }
3255

3256
  const char *precStr = NULL;
1,718,653✔
3257
  switch (pDb->cfg.precision) {
1,718,653✔
3258
    case TSDB_TIME_PRECISION_MILLI:
1,670,492✔
3259
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
1,670,492✔
3260
      break;
1,670,492✔
3261
    case TSDB_TIME_PRECISION_MICRO:
13,083✔
3262
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
13,083✔
3263
      break;
13,083✔
3264
    case TSDB_TIME_PRECISION_NANO:
35,078✔
3265
      precStr = TSDB_TIME_PRECISION_NANO_STR;
35,078✔
3266
      break;
35,078✔
UNCOV
3267
    default:
×
UNCOV
3268
      precStr = "none";
×
UNCOV
3269
      break;
×
3270
  }
3271
  char precVstr[10] = {0};
1,718,653✔
3272
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
1,718,653✔
3273

3274
  char *statusStr = "ready";
1,718,653✔
3275
  if (objStatus == SDB_STATUS_CREATING) {
1,718,653✔
3276
    statusStr = "creating";
×
3277
  } else if (objStatus == SDB_STATUS_DROPPING) {
1,718,653✔
3278
    statusStr = "dropping";
×
3279
  } else {
3280
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
1,718,653✔
3281
      statusStr = "unsynced";
3,269✔
3282
    }
3283
  }
3284
  char statusVstr[24] = {0};
1,718,653✔
3285
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
1,718,653✔
3286

3287
  if (sysDb || !sysinfo) {
1,718,653✔
3288
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
46,016,837✔
3289
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
44,946,678✔
3290
      if (i == 0) {
44,946,678✔
3291
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
1,070,159✔
3292
      } else if (i == 1) {
43,876,519✔
3293
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
1,070,159✔
3294
      } else if (i == 3) {
42,806,360✔
3295
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
1,070,159✔
3296
      } else if (i == 14) {
41,736,201✔
3297
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
1,070,159✔
3298
      } else if (i == 15) {
40,666,042✔
3299
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
1,070,159✔
3300
      } else if (i == 28) {
39,595,883✔
3301
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
1,070,159✔
3302
      } else {
3303
        colDataSetNULL(pColInfo, rows);
38,525,724✔
3304
      }
3305
    }
3306
  } else {
3307
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3308
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
648,494✔
3309

3310
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3311
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
648,494✔
3312

3313
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3314
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
648,494✔
3315

3316
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3317
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
648,494✔
3318

3319
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3320
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
648,494✔
3321

3322
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
648,494✔
3323
    char        strictVstr[24] = {0};
648,494✔
3324
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
648,494✔
3325
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3326
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
648,494✔
3327

3328
    char    durationStr[128] = {0};
648,494✔
3329
    char    durationVstr[128] = {0};
648,494✔
3330
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
648,494✔
3331
                                       pDb->cfg.daysPerFile);
3332

3333
    varDataSetLen(durationVstr, len);
648,494✔
3334
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3335
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
648,494✔
3336

3337
    char keepVstr[128] = {0};
648,494✔
3338
    char keep0Str[32] = {0};
648,494✔
3339
    char keep1Str[32] = {0};
648,494✔
3340
    char keep2Str[32] = {0};
648,494✔
3341

3342
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
648,494✔
3343
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
648,494✔
3344
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
648,494✔
3345

3346
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
648,494✔
UNCOV
3347
      len = snprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr) - VARSTR_HEADER_SIZE, "%s,%s,%s", keep1Str,
×
3348
                     keep2Str, keep0Str);
3349
    } else {
3350
      len = snprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr) - VARSTR_HEADER_SIZE, "%s,%s,%s", keep0Str,
648,494✔
3351
                     keep1Str, keep2Str);
3352
    }
3353
    varDataSetLen(keepVstr, len);
648,494✔
3354
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3355
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
648,494✔
3356

3357
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3358
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
648,494✔
3359

3360
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3361
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
648,494✔
3362

3363
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3364
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
648,494✔
3365

3366
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3367
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
648,494✔
3368

3369
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3370
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
648,494✔
3371

3372
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3373
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
648,494✔
3374

3375
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3376
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
648,494✔
3377

3378
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3379
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
648,494✔
3380

3381
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
648,494✔
3382
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3383
    if (rentensionVstr == NULL) {
648,494✔
3384
      colDataSetNULL(pColInfo, rows);
648,494✔
3385
    } else {
UNCOV
3386
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
×
UNCOV
3387
      taosMemoryFree(rentensionVstr);
×
3388
    }
3389

3390
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3391
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
648,494✔
3392

3393
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
648,494✔
3394
    char        cacheModelVstr[24] = {0};
648,494✔
3395
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
648,494✔
3396
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3397
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
648,494✔
3398

3399
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3400
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
648,494✔
3401

3402
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3403
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastShardBits, false), &lino, _OVER);
648,494✔
3404

3405
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3406
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
648,494✔
3407

3408
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3409
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
648,494✔
3410

3411
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3412
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
648,494✔
3413

3414
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3415
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
648,494✔
3416

3417
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3418
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
648,494✔
3419

3420
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3421
    int16_t hashPrefix = pDb->cfg.hashPrefix;
648,494✔
3422
    if (hashPrefix > 0) {
648,494✔
3423
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
5,009✔
3424
    } else if (hashPrefix < 0) {
643,485✔
3425
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
5,624✔
3426
    }
3427
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
648,494✔
3428

3429
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3430
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
648,494✔
3431

3432
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3433
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
648,494✔
3434

3435
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3436
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
648,494✔
3437

3438
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3439
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssChunkSize, false), &lino, _OVER);
648,494✔
3440

3441
    char keeplocalVstr[128] = {0};
648,494✔
3442
    len = snprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr) - VARSTR_HEADER_SIZE, "%dm",
648,494✔
3443
                   pDb->cfg.ssKeepLocal);
3444
    varDataSetLen(keeplocalVstr, len);
648,494✔
3445
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3446
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
648,494✔
3447

3448
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3449
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssCompact, false), &lino, _OVER);
648,494✔
3450

3451
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3452
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
648,494✔
3453

3454
    char encryptAlgorithmStr[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
648,494✔
3455
    getEncryptAlgorithmStr(pMnode, pDb->cfg.encryptAlgorithm, encryptAlgorithmStr);
648,494✔
3456
    char        encryptAlgorithmVStr[24] = {0};
648,494✔
3457
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
648,494✔
3458
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
648,494✔
3459
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
648,494✔
3460

3461
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
648,494✔
3462
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
648,494✔
3463
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3464
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
648,494✔
3465
    }
3466

3467
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
648,494✔
3468
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
648,494✔
3469
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
648,494✔
3470
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
648,494✔
3471
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3472
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
648,494✔
3473
    }
3474

3475
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
648,494✔
3476
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
648,494✔
3477
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3478
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
648,494✔
3479
    }
3480

3481
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3482
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.isAudit, false), &lino, _OVER);
648,494✔
3483
    }
3484

3485
    if (!(*ppUidNames)) {
648,494✔
3486
      TAOS_CHECK_GOTO(mndBuildUidNamesHash(pMnode, ppUidNames), &lino, _OVER);
358,592✔
3487
    }
3488
    const char *ownerName = tSimpleHashGet(*ppUidNames, (const char *)&pDb->ownerId, sizeof(pDb->ownerId));
648,494✔
3489
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%s", ownerName ? ownerName : "[unknown]"));
648,494✔
3490
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
648,494✔
3491
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3492
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
648,494✔
3493
    }
3494

3495
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3496
      uint8_t allowDrop = pDb->cfg.allowDrop;
648,494✔
3497
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&allowDrop, false), &lino, _OVER);
648,494✔
3498
    }
3499

3500
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
648,494✔
3501
      uint8_t securityLevel = pDb->cfg.securityLevel;
648,494✔
3502
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&securityLevel, false), &lino, _OVER);
648,494✔
3503
    }
3504
  }
3505
_OVER:
1,718,653✔
3506
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
1,718,653✔
3507
  taosMemoryFree(buf);
1,718,653✔
3508
}
3509

3510
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
532,497✔
3511
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
532,497✔
3512
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
532,497✔
3513
  pDbObj->cfg.numOfVgroups = 0;
532,497✔
3514
  pDbObj->cfg.strict = 1;
532,497✔
3515
  pDbObj->cfg.replications = 1;
532,497✔
3516
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
532,497✔
3517
}
532,497✔
3518

3519
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
532,497✔
3520
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
532,497✔
3521
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
532,497✔
3522
  pDbObj->cfg.numOfVgroups = 0;
532,497✔
3523
  pDbObj->cfg.strict = 1;
532,497✔
3524
  pDbObj->cfg.replications = 1;
532,497✔
3525
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
532,497✔
3526
}
532,497✔
3527

3528
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
5,332,144✔
3529
  SVgObj  *pVgroup = pObj;
5,332,144✔
3530
  int32_t *numOfTables = p1;
5,332,144✔
3531
  int64_t  uid = *(int64_t *)p2;
5,332,144✔
3532
  if (pVgroup->dbUid == uid) {
5,332,144✔
3533
    *numOfTables += pVgroup->numOfTables;
1,703,535✔
3534
  }
3535
  return true;
5,332,144✔
3536
}
3537

3538
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
532,497✔
3539
  SMnode    *pMnode = pReq->info.node;
532,497✔
3540
  SSdb      *pSdb = pMnode->pSdb;
532,497✔
3541
  int32_t    numOfRows = 0;
532,497✔
3542
  SDbObj    *pDb = NULL;
532,497✔
3543
  SUserObj  *pOperUser = NULL;
532,497✔
3544
  ESdbStatus objStatus = 0;
532,497✔
3545
  SSHashObj *pUidNames = NULL;
532,497✔
3546

3547
  (void)mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pOperUser);
532,497✔
3548
  if (pOperUser == NULL) return 0;
532,497✔
3549
  bool sysinfo = pOperUser->sysInfo;
532,497✔
3550
  char objFName[TSDB_OBJ_FNAME_LEN + 1] = {0};
532,497✔
3551
  (void)snprintf(objFName, sizeof(objFName), "%d.*", pOperUser->acctId);
532,497✔
3552
  bool showAnyDb = (0 == mndCheckSysObjPrivilege(pMnode, pOperUser, RPC_MSG_TOKEN(pReq), PRIV_CM_SHOW, PRIV_OBJ_DB, 0,
532,497✔
3553
                                                 objFName, NULL));
3554

3555
  // Append the information_schema database into the result.
3556
  if (!pShow->sysDbRsp) {
532,497✔
3557
    SDbObj infoschemaDb = {0};
532,497✔
3558
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
532,497✔
3559
    size_t numOfTables = 0;
532,497✔
3560
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
532,497✔
3561
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1, &pUidNames);
532,497✔
3562

3563
    numOfRows += 1;
532,497✔
3564

3565
    SDbObj perfschemaDb = {0};
532,497✔
3566
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
532,497✔
3567
    numOfTables = 0;
532,497✔
3568
    getPerfDbMeta(NULL, &numOfTables);
532,497✔
3569
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1, &pUidNames);
532,497✔
3570

3571
    numOfRows += 1;
532,497✔
3572
    pShow->sysDbRsp = true;
532,497✔
3573
  }
3574

3575
  while (numOfRows < rowsCapacity) {
1,186,156✔
3576
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
1,186,156✔
3577
    if (pShow->pIter == NULL) break;
1,186,156✔
3578
    if (showAnyDb || mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_SHOW_DATABASES, pDb) == 0) {
653,659✔
3579
      int32_t numOfTables = 0;
653,659✔
3580
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
653,659✔
3581
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo, &pUidNames);
653,659✔
3582
      numOfRows++;
653,659✔
3583
    }
3584

3585
    sdbRelease(pSdb, pDb);
653,659✔
3586
  }
3587

3588
  pShow->numOfRows += numOfRows;
532,497✔
3589
  mndReleaseUser(pMnode, pOperUser);
532,497✔
3590
  tSimpleHashCleanup(pUidNames);
532,497✔
3591
  return numOfRows;
532,497✔
3592
}
3593

UNCOV
3594
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
UNCOV
3595
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
3596
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
UNCOV
3597
}
×
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