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

taosdata / TDengine / #4945

30 Jan 2026 06:19AM UTC coverage: 66.87% (+0.02%) from 66.849%
#4945

push

travis-ci

web-flow
merge: from main to 3.0 #34453

1126 of 2018 new or added lines in 72 files covered. (55.8%)

13708 existing lines in 159 files now uncovered.

205277 of 306978 relevant lines covered (66.87%)

126353544.65 hits per line

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

77.85
/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_NUMBER                    DB_VER_SUPPORT_ADVANCED_SECURITY
51

52
#define DB_RESERVE_SIZE 13
53

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

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

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

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

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

100
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
400,915✔
101
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
400,915✔
102

103
  return sdbSetTable(pMnode->pSdb, table);
400,915✔
104
}
105

106
void mndCleanupDb(SMnode *pMnode) {}
400,857✔
107

108
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
5,542,262✔
109
  int32_t code = 0;
5,542,262✔
110
  int32_t lino = 0;
5,542,262✔
111
  terrno = TSDB_CODE_OUT_OF_MEMORY;
5,542,262✔
112

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

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

177
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
5,542,262✔
178
  SDB_SET_UINT8(pRaw, dataPos, pDb->cfg.flags, _OVER)
5,542,262✔
179
  SDB_SET_INT64(pRaw, dataPos, pDb->ownerId, _OVER)
5,542,262✔
180

181
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
5,542,262✔
182

183
  terrno = 0;
5,542,262✔
184

185
_OVER:
5,542,262✔
186
  if (terrno != 0) {
5,542,262✔
UNCOV
187
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
UNCOV
188
    sdbFreeRaw(pRaw);
×
UNCOV
189
    return NULL;
×
190
  }
191

192
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
5,542,262✔
193
  return pRaw;
5,542,262✔
194
}
195

196
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
4,994,457✔
197
  int32_t code = 0;
4,994,457✔
198
  int32_t lino = 0;
4,994,457✔
199
  terrno = TSDB_CODE_OUT_OF_MEMORY;
4,994,457✔
200
  SSdbRow *pRow = NULL;
4,994,457✔
201
  SDbObj  *pDb = NULL;
4,994,457✔
202

203
  int8_t sver = 0;
4,994,457✔
204
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
4,994,457✔
205

206
  if (sver < 1 || sver > DB_VER_NUMBER) {
4,994,457✔
UNCOV
207
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
UNCOV
208
    goto _OVER;
×
209
  }
210

211
  pRow = sdbAllocRow(sizeof(SDbObj));
4,994,457✔
212
  if (pRow == NULL) goto _OVER;
4,994,457✔
213

214
  pDb = sdbGetRowObj(pRow);
4,994,457✔
215
  if (pDb == NULL) goto _OVER;
4,994,457✔
216

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

291
  taosInitRWLatch(&pDb->lock);
4,994,457✔
292

293
  if (pDb->cfg.ssChunkSize == 0) {
4,994,457✔
UNCOV
294
    pDb->cfg.ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
×
295

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

299
  if (pDb->cfg.ssKeepLocal == 0) {
4,994,457✔
UNCOV
300
    pDb->cfg.ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
×
301

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

305
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
4,994,457✔
306
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
4,994,457✔
307
          TSDB_DEFAULT_TSDB_PAGESIZE);
308
  }
309

310
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
4,994,457✔
311
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
4,896,449✔
312
  }
313

314
  if (sver < DB_VER_SUPPORT_ADVANCED_SECURITY) {
4,994,457✔
NEW
315
    pDb->cfg.allowDrop = TSDB_DEFAULT_DB_ALLOW_DROP;
×
316
  }
317

318
  terrno = 0;
4,994,457✔
319

320
_OVER:
4,994,457✔
321
  if (terrno != 0) {
4,994,457✔
322
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
UNCOV
323
    taosMemoryFreeClear(pRow);
×
UNCOV
324
    return NULL;
×
325
  }
326

327
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
4,994,457✔
328
  return pRow;
4,994,457✔
329
}
330

331
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
1,138,236✔
332
  SSdb    *pSdb = pMnode->pSdb;
1,138,236✔
333
  SSdbRow *pRow = NULL;
1,138,236✔
334
  SDbObj  *pNewDb = NULL;
1,138,236✔
335
  int      code = -1;
1,138,236✔
336

337
  pRow = mndDbActionDecode(pRaw);
1,138,236✔
338
  if (pRow == NULL) {
1,138,236✔
UNCOV
339
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
340
    if (terrno != 0) code = terrno;
×
UNCOV
341
    goto _OVER;
×
342
  }
343
  pNewDb = sdbGetRowObj(pRow);
1,138,236✔
344
  if (pNewDb == NULL) {
1,138,236✔
UNCOV
345
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
346
    if (terrno != 0) code = terrno;
×
347
    goto _OVER;
×
348
  }
349

350
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
1,138,236✔
351
  if (pOldDb != NULL) {
1,138,236✔
352
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
353
    sdbRelease(pMnode->pSdb, pOldDb);
×
354
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
355
    if (terrno != 0) code = terrno;
×
UNCOV
356
    goto _OVER;
×
357
  }
358

359
  code = 0;
1,138,236✔
360
_OVER:
1,138,236✔
361
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
1,138,236✔
362
  taosMemoryFreeClear(pRow);
1,138,236✔
363
  return code;
1,138,236✔
364
}
365

366
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
1,273,314✔
367
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
1,273,314✔
368
  return 0;
1,273,314✔
369
}
370

371
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
4,994,457✔
372
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
4,994,457✔
373
  taosArrayDestroy(pDb->cfg.pRetensions);
4,994,457✔
374
  return 0;
4,994,457✔
375
}
376

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

421
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
8,114,598✔
422
  SSdb *pSdb = pMnode->pSdb;
8,114,598✔
423
  return sdbGetTableVer(pSdb, SDB_VGROUP);
8,114,598✔
424
}
425

426
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
86,214,171✔
427
  SSdb   *pSdb = pMnode->pSdb;
86,214,171✔
428
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
86,214,568✔
429
  if (pDb == NULL) {
86,214,543✔
430
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
2,273,134✔
431
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
2,113,044✔
432
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
160,090✔
433
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
9,663✔
434
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
150,427✔
435
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
150,427✔
436
    } else {
UNCOV
437
      terrno = TSDB_CODE_APP_ERROR;
×
UNCOV
438
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
439
    }
440
  }
441
  return pDb;
86,214,172✔
442
}
443

444
SDbObj *mndAcquireAuditDb(SMnode *pMnode) {
41,259,495✔
445
  SSdb      *pSdb = pMnode->pSdb;
41,259,495✔
446
  SDbObj    *pDb = NULL;
41,259,495✔
447
  void      *pIter = NULL;
41,259,495✔
448
  ESdbStatus objStatus = 0;
41,259,495✔
449

450
  while (1) {
451
    pIter = sdbFetchAll(pSdb, SDB_DB, pIter, (void **)&pDb, &objStatus, true);
116,108,170✔
452
    if (pIter == NULL) break;
116,108,170✔
453
    if (pDb->cfg.isAudit == 1) {
74,849,699✔
454
      sdbCancelFetch(pSdb, pIter);
1,024✔
455
      break;
1,024✔
456
    }
457

458
    sdbRelease(pSdb, pDb);
74,848,675✔
459
  }
460
  return pDb;
41,259,495✔
461
}
462

463
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
94,916,712✔
464
  SSdb *pSdb = pMnode->pSdb;
94,916,712✔
465
  sdbRelease(pSdb, pDb);
94,916,741✔
466
}
94,916,741✔
467

468
bool mndDbIsExist(SMnode *pMnode, const char *db, int64_t uid) {
180,279✔
469
  SDbObj *pDb = mndAcquireDb(pMnode, db);
180,279✔
470
  if (pDb == NULL) {
180,279✔
UNCOV
471
    return false;
×
472
  } else {
473
    bool result = (uid == 0) || (uid == pDb->uid);
180,279✔
474
    mndReleaseDb(pMnode, pDb);
180,279✔
475
    pDb = NULL;
180,279✔
476
    return result;
180,279✔
477
  }
478
}
479

480
int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
1,105,604✔
481
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
1,105,604✔
482
  if (pos == NULL) {
1,105,604✔
UNCOV
483
    return TSDB_CODE_MND_INVALID_DB;
×
484
  }
485

486
  int32_t acctId;
1,105,425✔
487
  int32_t code = taosStr2int32(dbName, &acctId);
1,105,604✔
488
  if (code != 0) {
1,105,604✔
UNCOV
489
    return code;
×
490
  }
491

492
  if (acctId != pUser->acctId) {
1,105,604✔
UNCOV
493
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
494
  }
495

496
  return 0;
1,105,604✔
497
}
498

499
int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
1,105,604✔
500
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
1,105,604✔
501

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

532
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
1,104,891✔
533
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
1,104,891✔
534
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
1,104,891✔
535
  if (pCfg->hashMethod != 1) return code;
1,104,891✔
536
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
1,104,891✔
UNCOV
537
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
UNCOV
538
    TAOS_RETURN(code);
×
539
  }
540
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
1,104,891✔
541
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
1,104,891✔
542
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
1,104,891✔
543
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
1,104,891✔
544
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
1,104,891✔
545
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
1,104,891✔
546
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
1,104,891✔
547
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
1,104,891✔
548
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
1,104,891✔
549
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
1,104,891✔
550
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
1,104,891✔
551

552
  if (pCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
1,104,891✔
553
  if (pCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
1,104,891✔
554
  if (pCfg->ssCompact < TSDB_MIN_SS_COMPACT || pCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
1,104,891✔
555

556
  if (pCfg->compactInterval != 0 &&
1,104,891✔
557
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
3,540✔
UNCOV
558
    return code;
×
559
  if (pCfg->compactStartTime != 0 &&
1,104,891✔
560
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
3,924✔
UNCOV
561
    return code;
×
562
  if (pCfg->compactEndTime != 0 &&
1,104,891✔
563
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
3,924✔
UNCOV
564
    return code;
×
565
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
1,104,891✔
566
    return code;
×
567
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
1,104,891✔
UNCOV
568
    return code;
×
569
  if (pCfg->isAudit < 0 || pCfg->isAudit > 1) return code;
1,104,891✔
570
  if (pCfg->allowDrop < TSDB_MIN_DB_ALLOW_DROP || pCfg->allowDrop > TSDB_MAX_DB_ALLOW_DROP) return code;
571

572
  code = 0;
1,104,891✔
573
  TAOS_RETURN(code);
1,104,891✔
574
}
575

576
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
83,006✔
577
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
83,006✔
578
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
83,006✔
579
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
83,006✔
580
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
83,006✔
581
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
83,006✔
582
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
83,006✔
583
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
83,006✔
584
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
83,006✔
585
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
83,006✔
586
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
83,006✔
587
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
83,006✔
588
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
83,006✔
UNCOV
589
    return code;
×
590
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
83,006✔
591
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
83,006✔
592
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
83,006✔
593
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
83,006✔
594
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
83,006✔
595
#ifdef TD_ENTERPRISE
596
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
83,006✔
597
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
83,006✔
598
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
697✔
599
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
697✔
600
      return code;
697✔
601
    }
602
  }
603
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
82,309✔
UNCOV
604
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
UNCOV
605
    return code;
×
606
  }
607
#else
608
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
609
#endif
610

611
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
82,309✔
UNCOV
612
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
613
    return code;
×
614
  }
615
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
82,309✔
UNCOV
616
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
UNCOV
617
    return code;
×
618
  }
619

620
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
82,309✔
621
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
1,213✔
622
    return code;
×
623
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
82,309✔
624
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
82,309✔
625
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
82,309✔
626
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
81,612✔
627
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
81,612✔
628
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
81,612✔
629
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
81,612✔
UNCOV
630
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
631
    return code;
×
632
  }
633
  if (pNewCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pNewCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
81,612✔
634
  if (pNewCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pNewCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
81,612✔
635
  if (pNewCfg->ssCompact < TSDB_MIN_SS_COMPACT || pNewCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
81,612✔
636

637
  if (pNewCfg->compactInterval != 0 &&
81,612✔
638
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
6,399✔
639
    return code;
×
640
  if (pNewCfg->compactStartTime != 0 &&
81,612✔
641
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
4,308✔
UNCOV
642
    return code;
×
643
  if (pNewCfg->compactEndTime != 0 &&
81,612✔
644
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
4,308✔
UNCOV
645
    return code;
×
646
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
81,612✔
647
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
4,308✔
648
    return code;
×
649
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
81,612✔
650
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
81,612✔
651
    return code;
×
652
  if (pNewCfg->isAudit < 0 || pNewCfg->isAudit > 1) return code;
81,612✔
653
  if (pNewCfg->allowDrop < TSDB_MIN_DB_ALLOW_DROP || pNewCfg->allowDrop > TSDB_MAX_DB_ALLOW_DROP) return code;
654

655
  code = 0;
81,612✔
656
  TAOS_RETURN(code);
81,612✔
657
}
658

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

697
int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,100,383✔
698
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,100,383✔
699
  if (pDbRaw == NULL) return -1;
1,100,383✔
700

701
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
1,100,383✔
702
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
1,100,383✔
703
  return 0;
1,100,383✔
704
}
705

706
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,099,885✔
707
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,461,355✔
708
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
2,361,470✔
709
  }
710
  return 0;
1,099,885✔
711
}
712

UNCOV
713
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
UNCOV
714
  int32_t  code = 0;
×
UNCOV
715
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
UNCOV
716
  if (pDbRaw == NULL) {
×
UNCOV
717
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
718
    if (terrno != 0) code = terrno;
×
UNCOV
719
    TAOS_RETURN(code);
×
720
  }
UNCOV
721
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
UNCOV
722
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
723

724
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
725
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
726
    if (pVgRaw == NULL) {
×
727
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
728
      if (terrno != 0) code = terrno;
×
729
      TAOS_RETURN(code);
×
730
    }
731
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
732
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
733
  }
734

735
  if (pDb->cfg.withArbitrator) {
×
736
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
737
      SVgObj   *pVgObj = pVgroups + v;
×
738
      SArbGroup arbGroup = {0};
×
739
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
UNCOV
740
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
741
    }
742
  }
743

UNCOV
744
  TAOS_RETURN(code);
×
745
}
746

747
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,099,885✔
748
  int32_t  code = 0;
1,099,885✔
749
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,099,885✔
750
  if (pDbRaw == NULL) {
1,099,885✔
UNCOV
751
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
752
    if (terrno != 0) code = terrno;
×
UNCOV
753
    TAOS_RETURN(code);
×
754
  }
755
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
1,099,885✔
756
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
1,099,885✔
757

758
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,461,355✔
759
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
2,361,470✔
760
    if (pVgRaw == NULL) {
2,361,470✔
761
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
762
      if (terrno != 0) code = terrno;
×
763
      TAOS_RETURN(code);
×
764
    }
765
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
2,361,470✔
766
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
2,361,470✔
767
  }
768

769
  if (pDb->cfg.withArbitrator) {
1,099,885✔
770
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
9,920✔
771
      SVgObj   *pVgObj = pVgroups + v;
6,010✔
772
      SArbGroup arbGroup = {0};
6,010✔
773
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6,010✔
774
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
6,010✔
775
    }
776
  }
777

778
  TAOS_RETURN(code);
1,099,885✔
779
}
780

781
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
1,099,885✔
782
                                        SUserObj *pUserDuped, SArray *pAuditOwnedDbs) {
783
  int32_t  code = 0;
1,099,885✔
784
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
1,099,885✔
785
  if (pDbRaw == NULL) {
1,099,885✔
UNCOV
786
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
787
    if (terrno != 0) code = terrno;
×
UNCOV
788
    TAOS_RETURN(code);
×
789
  }
790
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
1,099,885✔
791
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
1,099,885✔
792

793
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
3,461,355✔
794
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
2,361,470✔
795
    if (pVgRaw == NULL) {
2,361,470✔
796
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
797
      if (terrno != 0) code = terrno;
×
798
      TAOS_RETURN(code);
×
799
    }
800
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
2,361,470✔
801
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
2,361,470✔
802
  }
803

804
  if (pDb->cfg.withArbitrator) {
1,099,885✔
805
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
9,920✔
806
      SVgObj   *pVgObj = pVgroups + v;
6,010✔
807
      SArbGroup arbGroup = {0};
6,010✔
808
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
6,010✔
809
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
6,010✔
810
    }
811
  }
812

813
  if (pUserDuped) {
1,099,885✔
814
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
1,099,885✔
815
    if (pUserRaw == NULL) {
1,099,885✔
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, pUserRaw));
1,099,885✔
821
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
1,099,885✔
822
  }
823

824
  if(pAuditOwnedDbs) {
1,099,885✔
NEW
825
    int32_t auditOwnedDbSize = taosArrayGetSize(pAuditOwnedDbs);
×
NEW
826
    for (int32_t i = 0; i < auditOwnedDbSize; ++i) {
×
NEW
827
      SUserObj *pAuditUser = (SUserObj *)TARRAY_GET_ELEM(pAuditOwnedDbs, i);
×
NEW
828
      SSdbRaw  *pAuditUserRaw = mndUserActionEncode(pAuditUser);
×
NEW
829
      if (pAuditUserRaw == NULL) {
×
NEW
830
        TAOS_RETURN(terrno ? terrno : TSDB_CODE_MND_RETURN_VALUE_NULL);
×
831
      }
NEW
832
      TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pAuditUserRaw));
×
NEW
833
      TAOS_CHECK_RETURN(sdbSetRawStatus(pAuditUserRaw, SDB_STATUS_READY));
×
834
    }
835
  }
836

837
  TAOS_RETURN(code);
1,099,885✔
838
}
839

840
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,099,885✔
841
  int32_t code = 0;
1,099,885✔
842
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
3,461,355✔
843
    SVgObj *pVgroup = pVgroups + vg;
2,361,470✔
844

845
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
4,970,986✔
846
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,609,516✔
847
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
2,609,516✔
848
    }
849
  }
850

851
  TAOS_RETURN(code);
1,099,885✔
852
}
853

854
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
1,099,885✔
855
  int32_t code = 0;
1,099,885✔
856
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
3,461,355✔
857
    SVgObj *pVgroup = pVgroups + vg;
2,361,470✔
858

859
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
4,970,986✔
860
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,609,516✔
861
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
2,609,516✔
862
    }
863
  }
864

865
  TAOS_RETURN(code);
1,099,885✔
866
}
867

868
/**
869
 * @brief fill audit owned dbs for users with role SYSAUDIT or SYSAUDIT_LOG.
870
 *
871
 * @param pMnode
872
 * @param pOperUser
873
 * @param pDb
874
 * @param pAuditOwnedDbs
875
 * @return int32_t
876
 */
877
static int32_t mndSetAuditOwnedDbs(SMnode *pMnode, SUserObj *pOperUser, SDbObj *pDb, SArray **pAuditOwnedDbs) {
64✔
878
  int32_t   code = 0, lino = 0;
64✔
879
  SSdb     *pSdb = pMnode->pSdb;
64✔
880
  SArray   *auditOwnedDbs = NULL;
64✔
881
  SUserObj *pUser = NULL;
64✔
882
  SUserObj  newUserObj = {0};
64✔
883
  int32_t   sysAuditLen = strlen(TSDB_ROLE_SYSAUDIT) + 1;
64✔
884
  int32_t   sysAuditLogLen = strlen(TSDB_ROLE_SYSAUDIT_LOG) + 1;
64✔
885

886
  void *pIter = NULL;
64✔
887
  while ((pIter = sdbFetch(pSdb, SDB_USER, pIter, (void **)&pUser))) {
128✔
888
    if (taosHashGetSize(pUser->roles) <= 0) {
64✔
NEW
889
      sdbRelease(pSdb, pUser);
×
NEW
890
      pUser = NULL;
×
NEW
891
      continue;
×
892
    }
893
    if(strncmp(pUser->name, pOperUser->name, TSDB_USER_LEN) == 0) {
64✔
894
      sdbRelease(pSdb, pUser);
64✔
895
      pUser = NULL;
64✔
896
      continue;
64✔
897
    }
NEW
898
    if (!taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT, sysAuditLen) &&
×
NEW
899
        !taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT_LOG, sysAuditLogLen)) {
×
NEW
900
      sdbRelease(pSdb, pUser);
×
NEW
901
      pUser = NULL;
×
NEW
902
      continue;
×
903
    }
NEW
904
    TAOS_CHECK_EXIT(mndUserDupObj(pUser, &newUserObj));
×
NEW
905
    TAOS_CHECK_EXIT(taosHashPut(newUserObj.ownedDbs, pDb->name, strlen(pDb->name) + 1, NULL, 0));
×
NEW
906
    if (auditOwnedDbs == NULL) {
×
NEW
907
      auditOwnedDbs = taosArrayInit(1, sizeof(SUserObj));
×
NEW
908
      if (auditOwnedDbs == NULL) {
×
NEW
909
        TAOS_CHECK_EXIT(terrno);
×
910
      }
911
    }
NEW
912
    if (!taosArrayPush(auditOwnedDbs, &newUserObj)) {
×
NEW
913
      TAOS_CHECK_EXIT(terrno);
×
914
    }
NEW
915
    memset(&newUserObj, 0, sizeof(SUserObj));
×
NEW
916
    sdbRelease(pSdb, pUser);
×
NEW
917
    pUser = NULL;
×
918
  }
919
  *pAuditOwnedDbs = auditOwnedDbs;
64✔
920
  auditOwnedDbs = NULL;
64✔
921
_exit:
64✔
922
  if (auditOwnedDbs != NULL) taosArrayDestroyEx(auditOwnedDbs, (FDelete)mndUserFreeObj);
64✔
923
  sdbRelease(pSdb, pUser);
64✔
924
  sdbCancelFetch(pSdb, pIter);
64✔
925
  mndUserFreeObj(&newUserObj);
64✔
926
  TAOS_RETURN(code);
64✔
927
}
928

929
static int32_t mndCreateDb(SMnode *pMnode, SRpcMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser, SArray *dnodeList) {
1,105,847✔
930
  int32_t  code = 0;
1,105,847✔
931
  SArray  *auditOwnedDbs = NULL;
1,105,847✔
932
  SUserObj newUserObj = {0};
1,105,847✔
933
  SDbObj   dbObj = {0};
1,105,847✔
934
  (void)memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
1,105,847✔
935
  (void)memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
1,105,847✔
936
  dbObj.createdTime = taosGetTimestampMs();
1,105,847✔
937
  dbObj.updateTime = dbObj.createdTime;
1,105,847✔
938
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
1,105,847✔
939
  dbObj.cfgVersion = 1;
1,105,847✔
940
  dbObj.vgVersion = 1;
1,105,847✔
941
  dbObj.tsmaVersion = 1;
1,105,847✔
942
  (void)memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
1,105,847✔
943
  dbObj.ownerId = pUser->uid;
1,105,847✔
944
  dbObj.cfg = (SDbCfg){
2,211,515✔
945
      .numOfVgroups = pCreate->numOfVgroups,
1,105,847✔
946
      .numOfStables = pCreate->numOfStables,
1,105,847✔
947
      .buffer = pCreate->buffer,
1,105,847✔
948
      .pageSize = pCreate->pageSize,
1,105,847✔
949
      .pages = pCreate->pages,
1,105,847✔
950
      .cacheLastSize = pCreate->cacheLastSize,
1,105,847✔
951
      .daysPerFile = pCreate->daysPerFile,
1,105,847✔
952
      .daysToKeep0 = pCreate->daysToKeep0,
1,105,847✔
953
      .daysToKeep1 = pCreate->daysToKeep1,
1,105,847✔
954
      .daysToKeep2 = pCreate->daysToKeep2,
1,105,847✔
955
      .keepTimeOffset = pCreate->keepTimeOffset,
1,105,847✔
956
      .minRows = pCreate->minRows,
1,105,847✔
957
      .maxRows = pCreate->maxRows,
1,105,847✔
958
      .walFsyncPeriod = pCreate->walFsyncPeriod,
1,105,847✔
959
      .walLevel = pCreate->walLevel,
1,105,847✔
960
      .precision = pCreate->precision,
1,105,847✔
961
      .compression = pCreate->compression,
1,105,847✔
962
      .replications = pCreate->replications,
1,105,847✔
963
      .strict = pCreate->strict,
1,105,847✔
964
      .cacheLast = pCreate->cacheLast,
1,105,847✔
965
      .hashMethod = 1,
966
      .schemaless = pCreate->schemaless,
1,105,847✔
967
      .walRetentionPeriod = pCreate->walRetentionPeriod,
1,105,847✔
968
      .walRetentionSize = pCreate->walRetentionSize,
1,105,847✔
969
      .walRollPeriod = pCreate->walRollPeriod,
1,105,847✔
970
      .walSegmentSize = pCreate->walSegmentSize,
1,105,847✔
971
      .sstTrigger = pCreate->sstTrigger,
1,105,847✔
972
      .hashPrefix = pCreate->hashPrefix,
1,105,847✔
973
      .hashSuffix = pCreate->hashSuffix,
1,105,847✔
974
      .ssChunkSize = pCreate->ssChunkSize,
1,105,847✔
975
      .ssKeepLocal = pCreate->ssKeepLocal,
1,105,847✔
976
      .ssCompact = pCreate->ssCompact,
1,105,847✔
977
      .tsdbPageSize = pCreate->tsdbPageSize,
1,105,847✔
978
      .withArbitrator = pCreate->withArbitrator,
1,105,847✔
979
      .encryptAlgorithm = pCreate->encryptAlgorithm,
1,105,847✔
980
      .compactInterval = pCreate->compactInterval,
1,105,847✔
981
      .compactStartTime = pCreate->compactStartTime,
1,105,847✔
982
      .compactEndTime = pCreate->compactEndTime,
1,105,847✔
983
      .compactTimeOffset = pCreate->compactTimeOffset,
1,105,847✔
984
      .isAudit = pCreate->isAudit,
1,105,847✔
985
  };
986
  if (strlen(pCreate->encryptAlgrName) > 0) {
1,105,847✔
987
    if (strncasecmp(pCreate->encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
5,729✔
988
      if (pCreate->isAudit == 1) {
3,442✔
989
        code = TSDB_CODE_AUDIT_MUST_ENCRYPT;
×
990
        mError("db:%s, failed to create, encrypt algorithm not match for audit db, %s", pCreate->db,
×
991
               pCreate->encryptAlgrName);
992
        TAOS_RETURN(code);
×
993
      }
994
      dbObj.cfg.encryptAlgorithm = 0;
3,442✔
995
    } else {
996
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, pCreate->encryptAlgrName);
2,287✔
997
      if (pEncryptAlgr != NULL) {
2,287✔
998
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
1,546✔
UNCOV
999
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
1000
          mError("db:%s, failed to create, encrypt algorithm not match, %s, expect type %d", pCreate->db,
×
1001
                 pCreate->encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
1002
          TAOS_RETURN(code);
×
1003
        }
1004
        dbObj.cfg.encryptAlgorithm = pEncryptAlgr->id;
1,546✔
1005
      } else {
1006
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
741✔
1007
        mError("db:%s, failed to create, encrypt algorithm not exist, %s", pCreate->db, pCreate->encryptAlgrName);
741✔
1008
        TAOS_RETURN(code);
741✔
1009
      }
1010
    }
1011
  } else {
1012
    if (pCreate->isAudit == 1) {
1,100,118✔
UNCOV
1013
      code = TSDB_CODE_AUDIT_MUST_ENCRYPT;
×
UNCOV
1014
      mError("db:%s, failed to create, encrypt algorithm not match for audit db, %s", pCreate->db,
×
1015
             pCreate->encryptAlgrName);
UNCOV
1016
      TAOS_RETURN(code);
×
1017
    }
1018
  }
1019
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
1,105,106✔
1020
  dbObj.cfg.pRetensions = pCreate->pRetensions;
1,105,106✔
1021

1022
  if (pCreate->isAudit == 1) {
1,105,106✔
1023
    if (dbObj.cfg.daysToKeep2 < 2628000) {
64✔
UNCOV
1024
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
UNCOV
1025
      mError("db:%s, failed to create, keep not match for audit db, %d", pCreate->db, dbObj.cfg.daysToKeep2);
×
UNCOV
1026
      TAOS_RETURN(code);
×
1027
    }
1028
    if (dbObj.cfg.walLevel != 2) {
64✔
UNCOV
1029
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
UNCOV
1030
      mError("db:%s, failed to create, walLevel not match for audit db, %d", pCreate->db, dbObj.cfg.walLevel);
×
UNCOV
1031
      TAOS_RETURN(code);
×
1032
    }
1033
    dbObj.cfg.allowDrop = TSDB_MIN_DB_ALLOW_DROP;
64✔
1034
  } else {
1035
    dbObj.cfg.allowDrop = TSDB_DEFAULT_DB_ALLOW_DROP;
1,105,042✔
1036
  }
1037

1038
  mndSetDefaultDbCfg(&dbObj.cfg);
1,105,106✔
1039

1040
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
1,105,106✔
UNCOV
1041
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, tstrerror(code));
×
UNCOV
1042
    TAOS_RETURN(code);
×
1043
  }
1044

1045
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
1,105,106✔
1046
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, tstrerror(code));
713✔
1047
    TAOS_RETURN(code);
713✔
1048
  }
1049

1050
  if (dbObj.cfg.hashPrefix > 0) {
1,104,393✔
1051
    int32_t dbLen = strlen(dbObj.name) + 1;
4,539✔
1052
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
4,539✔
1053
    dbObj.cfg.hashPrefix += dbLen;
4,539✔
1054
  } else if (dbObj.cfg.hashPrefix < 0) {
1,099,854✔
1055
    int32_t dbLen = strlen(dbObj.name) + 1;
4,640✔
1056
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
4,640✔
1057
    dbObj.cfg.hashPrefix -= dbLen;
4,640✔
1058
  }
1059

1060
  SVgObj *pVgroups = NULL;
1,104,393✔
1061
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
1,104,393✔
1062
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
4,508✔
1063
    TAOS_RETURN(code);
4,508✔
1064
  }
1065

1066
  // add database privileges for user
1067
  SUserObj *pNewUserDuped = NULL;
1,099,885✔
1068
#if 0
1069
  // We record the DB's owner id instread. The owner has the permission to create objects under the DB, but does not
1070
  // have default permissions on objects created by others .Therefore, readDbs/writeDbs are no longer applicable.
1071
  if (!pUser->superUser) {
1072
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
1073
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
1074
                    NULL, _OVER);
1075
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
1076
                    NULL, _OVER);
1077
    pNewUserDuped = &newUserObj;
1078
  }
1079
#else
1080
  // Considering the efficiency of use db privileges in some scenarios like insert operation, owned DBs is stored.
1081
  bool addOwned = false;
1,099,885✔
1082
  if (dbObj.cfg.isAudit == 1) {
1,099,885✔
1083
    if (taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT, strlen(TSDB_ROLE_SYSAUDIT) + 1) ||
64✔
NEW
1084
        taosHashGet(pUser->roles, TSDB_ROLE_SYSAUDIT_LOG, strlen(TSDB_ROLE_SYSAUDIT_LOG) + 1)) {
×
1085
      addOwned = true;
64✔
1086
    }
1087
    TAOS_CHECK_GOTO(mndSetAuditOwnedDbs(pMnode, pUser, &dbObj, &auditOwnedDbs), NULL, _OVER);
64✔
1088
  } else {
1089
    addOwned = true;
1,099,821✔
1090
  }
1091
  if (addOwned) {
1,099,885✔
1092
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
1,099,885✔
1093
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.ownedDbs, dbObj.name, strlen(dbObj.name) + 1, NULL, 0), NULL, _OVER);
1,099,885✔
1094
    pNewUserDuped = &newUserObj;
1,099,885✔
1095
  }
1096
#endif
1097

1098
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
1,099,885✔
1099
  if (pTrans == NULL) {
1,099,885✔
UNCOV
1100
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1101
    if (terrno != 0) code = terrno;
×
UNCOV
1102
    goto _OVER;
×
1103
  }
1104
  // mndTransSetSerial(pTrans);
1105
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
1,099,885✔
1106

1107
  mndTransSetDbName(pTrans, dbObj.name, NULL);
1,099,885✔
1108
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
1,099,885✔
1109

1110
  void   *pIter = NULL;
1,099,885✔
1111
  STrans *exitTrans = NULL;
1,099,885✔
1112

1113
  if (!pUser->superUser) {
1,099,885✔
1114
    while (1) {
1115
      pIter = sdbFetch(pMnode->pSdb, SDB_TRANS, pIter, (void **)&exitTrans);
1,366✔
1116
      if (pIter == NULL) break;
1,366✔
1117

1118
      if (exitTrans->conflict == TRN_CONFLICT_DB &&
×
1119
          strncmp(exitTrans->opername, "create-db", TSDB_TRANS_OPER_LEN) == 0) {
×
UNCOV
1120
        code = TSDB_CODE_MND_TRANS_CONFLICT;
×
1121
        sdbRelease(pMnode->pSdb, exitTrans);
×
UNCOV
1122
        sdbCancelFetch(pMnode->pSdb, pIter);
×
UNCOV
1123
        goto _OVER;
×
1124
      }
1125

UNCOV
1126
      sdbRelease(pMnode->pSdb, exitTrans);
×
1127
    }
1128
  }
1129

1130
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
1,099,885✔
1131
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
1,099,885✔
1132
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,099,885✔
1133
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,099,885✔
1134
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,099,885✔
1135
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped, auditOwnedDbs), NULL, _OVER);
1,099,885✔
1136
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
1,099,885✔
1137
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,099,885✔
1138

1139
_OVER:
1,099,885✔
1140
  taosMemoryFree(pVgroups);
1,099,885✔
1141
  mndUserFreeObj(&newUserObj);
1,099,885✔
1142
  taosArrayDestroyEx(auditOwnedDbs, (FDelete)mndUserFreeObj);
1,099,885✔
1143
  mndTransDrop(pTrans);
1,099,885✔
1144
  TAOS_RETURN(code);
1,099,885✔
1145
}
1146

1147
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
1,117,047✔
1148
  int32_t    code = 0;
1,117,047✔
1149
  SSdb      *pSdb = pMnode->pSdb;
1,117,047✔
1150
  SDnodeObj *pDnode = NULL;
1,117,047✔
1151
  void      *pIter = NULL;
1,117,047✔
1152

1153
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
1154
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
1,117,047✔
UNCOV
1155
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
×
UNCOV
1156
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
×
UNCOV
1157
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
UNCOV
1158
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
1159
           tsEncryptionKeyStat);
UNCOV
1160
    goto _exit;
×
1161
  }
1162

UNCOV
1163
  int64_t curMs = taosGetTimestampMs();
×
UNCOV
1164
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
×
UNCOV
1165
    bool online = false;
×
UNCOV
1166
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
×
UNCOV
1167
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
UNCOV
1168
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
UNCOV
1169
      mError("db:%s, failed to check encryption key:%" PRIi8
×
1170
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
1171
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
1172
             tsEncryptionKeyChksum);
UNCOV
1173
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
1174
      sdbRelease(pSdb, pDnode);
×
UNCOV
1175
      break;
×
1176
    }
UNCOV
1177
    sdbRelease(pSdb, pDnode);
×
1178
  }
1179
#else
1180
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
1181
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
1182
    goto _exit;
1183
  }
1184
#endif
1185
_exit:
1,116,868✔
1186
  TAOS_RETURN(code);
1,117,047✔
1187
}
1188

1189
#ifndef TD_ENTERPRISE
1190
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
1191
  if (dnodeListStr[0] != 0) {
1192
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1193
    return terrno;
1194
  } else {
1195
    return 0;
1196
  }
1197
}
1198
#endif
1199

1200
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
1,215,012✔
1201
  SMnode      *pMnode = pReq->info.node;
1,215,012✔
1202
  int32_t      code = -1;
1,215,012✔
1203
  int32_t      lino = 0;
1,215,012✔
1204
  SDbObj      *pDb = NULL;
1,215,012✔
1205
  SUserObj    *pUser = NULL;
1,215,012✔
1206
  SCreateDbReq createReq = {0};
1,215,012✔
1207
  SArray      *dnodeList = NULL;
1,215,012✔
1208
  int64_t      tss = taosGetTimestampMs();
1,215,012✔
1209

1210
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
1,215,012✔
1211
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
1,215,012✔
1212

1213
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
1,215,012✔
1214
#ifdef WINDOWS
1215
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
1216
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
1217
    goto _OVER;
1218
  }
1219
#endif
1220
  mInfo("db:%s, start to create, vgroups:%d, isAudit:%d, daysToKeep2:%d", createReq.db, createReq.numOfVgroups, createReq.isAudit, createReq.daysToKeep2);
1,215,012✔
1221

1222
  pDb = mndAcquireDb(pMnode, createReq.db);
1,215,012✔
1223
  if (pDb != NULL) {
1,215,012✔
1224
    if (createReq.ignoreExist) {
89,920✔
1225
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
87,709✔
1226
      code = 0;
87,709✔
1227
      goto _OVER;
87,709✔
1228
    } else {
1229
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
2,211✔
1230
      goto _OVER;
2,211✔
1231
    }
1232
  } else {
1233
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
1,125,092✔
1234
      code = terrno;
7,896✔
1235
      goto _OVER;
7,896✔
1236
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
1,117,196✔
UNCOV
1237
      goto _OVER;
×
1238
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
1,117,196✔
1239
      // continue
1240
    } else {  // TSDB_CODE_APP_ERROR
UNCOV
1241
      goto _OVER;
×
1242
    }
1243
  }
1244

1245
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_CREATE_DB, NULL), &lino, _OVER);
1,117,196✔
1246

1247
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
1,117,047✔
1248

1249
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
1,117,047✔
1250
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
1,117,047✔
1251

1252
  if (createReq.replications == 2) {
1,117,047✔
1253
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
5,310✔
1254
  }
1255

1256
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
1,117,047✔
1257

1258
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
1,117,047✔
1259

1260
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pUser), &lino, _OVER);
1,105,847✔
1261

1262
  if (sdbGetSize(pMnode->pSdb, SDB_MOUNT) > 0) {
1,105,847✔
1263
    TAOS_CHECK_GOTO(TSDB_CODE_MND_MOUNT_NOT_EMPTY, &lino, _OVER);
×
1264
  }
1265

1266
  if (createReq.isAudit == 1) {
1,105,847✔
1267
    SDbObj *pAuditDb = mndAcquireAuditDb(pMnode);
64✔
1268
    if (pAuditDb != NULL) {
64✔
1269
      mndReleaseDb(pMnode, pAuditDb);
×
1270
      mError("db:%s, audit db already exist, %s", createReq.db, pAuditDb->name);
×
1271
      TAOS_CHECK_GOTO(TSDB_CODE_AUDIT_DB_ALREADY_EXIST, &lino, _OVER);
×
1272
    }
1273
  }
1274

1275
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
1,105,847✔
1276
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
1,099,885✔
1277

1278
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
1,099,885✔
1279
    SName name = {0};
1,099,885✔
1280
    if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
1,099,885✔
1281
      mError("db:%s, failed to parse db name", createReq.db);
×
1282
    int64_t tse = taosGetTimestampMs();
1,099,885✔
1283
    double  duration = (double)(tse - tss);
1,099,885✔
1284
    duration = duration / 1000;
1,099,885✔
1285
    auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen, duration, 0);
1,099,885✔
1286
  }
1287

1288
_OVER:
1,214,833✔
1289
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,215,012✔
1290
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
27,418✔
1291
  }
1292

1293
  mndReleaseDb(pMnode, pDb);
1,215,012✔
1294
  mndReleaseUser(pMnode, pUser);
1,215,012✔
1295
  tFreeSCreateDbReq(&createReq);
1,215,012✔
1296
  taosArrayDestroy(dnodeList);
1,215,012✔
1297

1298
  TAOS_RETURN(code);
1,215,012✔
1299
}
1300

1301
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
107,251✔
1302
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
107,251✔
1303

1304
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
107,251✔
1305
    pDb->cfg.buffer = pAlter->buffer;
4,402✔
1306
    code = 0;
4,402✔
1307
  }
1308

1309
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
107,251✔
1310
    pDb->cfg.pages = pAlter->pages;
2,223✔
1311
    code = 0;
2,223✔
1312
  }
1313

1314
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
107,251✔
UNCOV
1315
    pDb->cfg.pageSize = pAlter->pageSize;
×
UNCOV
1316
    code = 0;
×
1317
  }
1318

1319
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
107,251✔
UNCOV
1320
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
UNCOV
1321
    code = 0;
×
1322
  }
1323

1324
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
107,251✔
1325
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
14,615✔
1326
    code = 0;
14,615✔
1327
  }
1328

1329
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
107,251✔
1330
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
13,897✔
1331
    code = 0;
13,897✔
1332
  }
1333

1334
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
107,251✔
1335
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
11,080✔
1336
    code = 0;
11,080✔
1337
  }
1338

1339
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
107,251✔
1340
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
3,559✔
1341
    code = 0;
3,559✔
1342
  }
1343

1344
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
107,251✔
1345
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
8,488✔
1346
    code = 0;
8,488✔
1347
  }
1348

1349
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
107,251✔
1350
    pDb->cfg.walLevel = pAlter->walLevel;
8,485✔
1351
    code = 0;
8,485✔
1352
  }
1353

1354
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
107,251✔
1355
#if 1
UNCOV
1356
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1357
#else
1358
    pDb->cfg.strict = pAlter->strict;
1359
    code = 0;
1360
#endif
1361
  }
1362

1363
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
107,251✔
1364
    pDb->cfg.cacheLast = pAlter->cacheLast;
13,353✔
1365
    code = 0;
13,353✔
1366
  }
1367

1368
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
107,251✔
1369
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
1,716✔
1370
    code = 0;
1,716✔
1371
  }
1372

1373
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
107,251✔
1374
    pDb->cfg.replications = pAlter->replications;
6,126✔
1375
    pDb->vgVersion++;
6,126✔
1376
    code = 0;
6,126✔
1377
  }
1378

1379
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
107,251✔
1380
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
1,213✔
1381
    pDb->vgVersion++;
1,213✔
1382
    code = 0;
1,213✔
1383
  }
1384

1385
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
107,251✔
1386
    pDb->cfg.minRows = pAlter->minRows;
1,394✔
1387
    pDb->vgVersion++;
1,394✔
1388
    code = 0;
1,394✔
1389
  }
1390

1391
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
107,251✔
1392
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
11,867✔
1393
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
6,661✔
1394
    pDb->vgVersion++;
6,661✔
1395
    code = 0;
6,661✔
1396
  }
1397

1398
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
107,251✔
1399
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
1,438✔
1400
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
697✔
1401
    pDb->vgVersion++;
697✔
1402
    code = 0;
697✔
1403
  }
1404

1405
  if (pAlter->ssKeepLocal >= TSDB_MIN_SS_KEEP_LOCAL && pAlter->ssKeepLocal <= TSDB_MAX_SS_KEEP_LOCAL && pAlter->ssKeepLocal != pDb->cfg.ssKeepLocal) {
107,251✔
1406
    pDb->cfg.ssKeepLocal = pAlter->ssKeepLocal;
697✔
1407
    pDb->vgVersion++;
697✔
1408
    code = 0;
697✔
1409
  }
1410

1411
  if (pAlter->ssCompact >= TSDB_MIN_SS_COMPACT && pAlter->ssCompact <= TSDB_MAX_SS_COMPACT && pAlter->ssCompact != pDb->cfg.ssCompact) {
107,251✔
1412
    pDb->cfg.ssCompact = pAlter->ssCompact;
697✔
1413
    pDb->vgVersion++;
697✔
1414
    code = 0;
697✔
1415
  }
1416

1417
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
107,251✔
1418
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
697✔
1419
    pDb->vgVersion++;
697✔
1420
    code = 0;
697✔
1421
  }
1422

1423
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
107,251✔
1424
    pDb->cfg.compactInterval = pAlter->compactInterval;
4,984✔
1425
    pDb->vgVersion++;
4,984✔
1426
    code = 0;
4,984✔
1427
  }
1428

1429
  bool compactTimeRangeChanged = false;
107,251✔
1430
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
107,251✔
1431
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
105,074✔
1432
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
105,074✔
1433
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
4,308✔
1434
    compactTimeRangeChanged = true;
4,308✔
1435
    code = 0;
4,308✔
1436
  }
1437

1438
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
107,251✔
1439
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
105,074✔
1440
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
4,308✔
1441
    compactTimeRangeChanged = true;
4,308✔
1442
    code = 0;
4,308✔
1443
  }
1444
  if(compactTimeRangeChanged) {
107,251✔
1445
    pDb->vgVersion++;
4,308✔
1446
  }
1447

1448
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
107,251✔
1449
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
8,618✔
1450
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
3,569✔
1451
    pDb->vgVersion++;
3,569✔
1452
    code = 0;
3,569✔
1453
  }
1454

1455
  if (pAlter->isAudit > -1 && pAlter->isAudit != pDb->cfg.isAudit) {
107,251✔
UNCOV
1456
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
UNCOV
1457
    return code;
×
1458
  }
1459

1460
  if(pAlter->allowDrop > -1 && pAlter->allowDrop != pDb->cfg.allowDrop) {
107,251✔
NEW
1461
    pDb->cfg.allowDrop = pAlter->allowDrop;
×
NEW
1462
    code = 0;
×
1463
  }
1464

1465
  TAOS_RETURN(code);
107,251✔
1466
}
1467

1468
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
80,952✔
1469
  int32_t  code = 0;
80,952✔
1470
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
80,952✔
1471
  if (pRedoRaw == NULL) {
80,952✔
UNCOV
1472
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1473
    if (terrno != 0) code = terrno;
×
UNCOV
1474
    TAOS_RETURN(code);
×
1475
  }
1476
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
80,952✔
UNCOV
1477
    sdbFreeRaw(pRedoRaw);
×
UNCOV
1478
    TAOS_RETURN(code);
×
1479
  }
1480

1481
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
80,952✔
UNCOV
1482
    sdbFreeRaw(pRedoRaw);
×
UNCOV
1483
    TAOS_RETURN(code);
×
1484
  }
1485
  return 0;
80,952✔
1486
}
1487

1488
int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
80,952✔
1489
  int32_t  code = 0;
80,952✔
1490
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
80,952✔
1491
  if (pCommitRaw == NULL) {
80,952✔
UNCOV
1492
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1493
    if (terrno != 0) code = terrno;
×
UNCOV
1494
    TAOS_RETURN(code);
×
1495
  }
1496
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
80,952✔
UNCOV
1497
    sdbFreeRaw(pCommitRaw);
×
UNCOV
1498
    TAOS_RETURN(code);
×
1499
  }
1500

1501
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
80,952✔
UNCOV
1502
    sdbFreeRaw(pCommitRaw);
×
UNCOV
1503
    TAOS_RETURN(code);
×
1504
  }
1505
  TAOS_RETURN(code);
80,952✔
1506
}
1507

1508
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
80,952✔
1509
  int32_t code = 0, lino = 0;
80,952✔
1510
  SSdb   *pSdb = pMnode->pSdb;
80,952✔
1511
  void   *pIter = NULL;
80,952✔
1512
  SVgObj *pVgroup = NULL;
80,952✔
1513
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
80,952✔
1514

1515
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
80,952✔
1516

1517
  while (1) {
1518
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
562,073✔
1519
    if (pIter == NULL) break;
562,073✔
1520

1521
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
481,806✔
1522
      SVgObj newVgroup = {0};
203,717✔
1523
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
203,717✔
1524
                      _err);
1525

1526
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
203,032✔
UNCOV
1527
        if (pNewDb->cfg.withArbitrator) {
×
UNCOV
1528
          SArbGroup arbGroup = {0};
×
UNCOV
1529
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
UNCOV
1530
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1531
        } else {
UNCOV
1532
          SArbGroup arbGroup = {0};
×
UNCOV
1533
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
UNCOV
1534
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1535
        }
1536
      }
1537
    }
1538

1539
    sdbRelease(pSdb, pVgroup);
481,121✔
1540
  }
1541

1542
  taosArrayDestroy(pArray);
80,267✔
1543
  TAOS_RETURN(code);
80,267✔
1544

1545
_err:
685✔
1546
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
685✔
1547

1548
  sdbCancelFetch(pSdb, pIter);
685✔
1549
  sdbRelease(pSdb, pVgroup);
685✔
1550
  taosArrayDestroy(pArray);
685✔
1551
  TAOS_RETURN(code);
685✔
1552
}
1553

1554
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
81,612✔
1555
  int32_t code = -1;
81,612✔
1556
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
81,612✔
1557
  if (pTrans == NULL) {
81,612✔
UNCOV
1558
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1559
    if (terrno != 0) code = terrno;
×
UNCOV
1560
    TAOS_RETURN(code);
×
1561
  }
1562
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
81,612✔
1563
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
81,612✔
1564

1565
  mndTransSetDbName(pTrans, pOld->name, NULL);
81,612✔
1566
  mndTransSetGroupParallel(pTrans);
81,612✔
1567
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
81,612✔
1568
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
81,012✔
1569
  TAOS_CHECK_GOTO(mndTransCheckConflictWithRetention(pMnode, pTrans), NULL, _OVER);
80,952✔
1570

1571
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
80,952✔
1572
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
80,952✔
1573
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
80,952✔
1574

1575
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
80,267✔
1576
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
80,267✔
1577
  code = 0;
80,267✔
1578

1579
_OVER:
81,612✔
1580
  mndTransDrop(pTrans);
81,612✔
1581
  TAOS_RETURN(code);
81,612✔
1582
}
1583

1584
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
110,151✔
1585
  SMnode     *pMnode = pReq->info.node;
110,151✔
1586
  int32_t     code = -1;
110,151✔
1587
  SDbObj     *pDb = NULL;
110,151✔
1588
  SAlterDbReq alterReq = {0};
110,151✔
1589
  SDbObj      dbObj = {0};
110,151✔
1590
  int64_t     tss = taosGetTimestampMs();
110,151✔
1591

1592
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
110,151✔
1593

1594
  mInfo("db:%s, start to alter, encryt_algr:%s", alterReq.db, alterReq.encryptAlgrName);
110,151✔
1595

1596
  pDb = mndAcquireDb(pMnode, alterReq.db);
110,151✔
1597
  if (pDb == NULL) {
110,151✔
1598
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
721✔
1599
    if (terrno != 0) code = terrno;
721✔
1600
    goto _OVER;
721✔
1601
  }
1602

1603
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_ALTER_DB, pDb), NULL, _OVER);
109,430✔
1604

1605
  if (alterReq.replications == 2) {
109,430✔
1606
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
697✔
1607
  }
1608

1609
  int32_t numOfTopics = 0;
109,430✔
1610
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
109,430✔
1611

1612
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
109,430✔
UNCOV
1613
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
UNCOV
1614
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1615
           numOfTopics);
UNCOV
1616
    goto _OVER;
×
1617
  }
1618

1619
  if (pDb->cfg.isMount) {
109,430✔
1620
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
1621
    goto _OVER;
×
1622
  }
1623

1624
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
109,430✔
1625
  if (dbObj.cfg.pRetensions != NULL) {
109,430✔
1626
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
1627
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1628
  }
1629

1630
  if (pDb->cfg.isAudit == 1) {
109,430✔
NEW
1631
    if (alterReq.daysToKeep2 > -1 && alterReq.daysToKeep2 < 2628000) {
×
NEW
1632
      code = TSDB_CODE_AUDIT_MUST_KEEPFORCE;
×
NEW
1633
      mError("db:%s, failed to alter, keep not match for audit db, %d", alterReq.db, alterReq.daysToKeep2);
×
NEW
1634
      goto _OVER;
×
1635
    }
NEW
1636
    if (alterReq.walLevel > -1 && alterReq.walLevel != 2) {
×
NEW
1637
      code = TSDB_CODE_AUDIT_MUST_WALFORCE;
×
NEW
1638
      mError("db:%s, failed to alter, walLevel not match for audit db, %d", alterReq.db, alterReq.walLevel);
×
NEW
1639
      goto _OVER;
×
1640
    }
NEW
1641
    if (alterReq.isAudit == 0) {
×
NEW
1642
      code = TSDB_CODE_AUDIT_DB_NOT_ALLOW_CHANGE;
×
NEW
1643
      mError("db:%s, failed to alter, is not allowed to change audit db, %d", alterReq.db, alterReq.isAudit);
×
NEW
1644
      goto _OVER;
×
1645
    }
1646
  }
1647

1648
  if (strlen(alterReq.encryptAlgrName) > 0) {
109,430✔
1649
    mInfo("db:%s, check encryt algr:%s", alterReq.db, alterReq.encryptAlgrName);
2,920✔
1650

1651
    if (strncasecmp(alterReq.encryptAlgrName, "none", TSDB_ENCRYPT_ALGR_NAME_LEN) == 0) {
2,920✔
1652
      if (pDb->cfg.encryptAlgorithm != 0) {
741✔
UNCOV
1653
        code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
×
UNCOV
1654
        mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
×
UNCOV
1655
        goto _OVER;
×
1656
      }
1657
    } else {
1658
      SEncryptAlgrObj *pEncryptAlgr = mndAcquireEncryptAlgrByAId(pMnode, alterReq.encryptAlgrName);
2,179✔
1659
      if (pEncryptAlgr != NULL) {
2,179✔
1660
        if (pEncryptAlgr->type != ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS) {
1,438✔
UNCOV
1661
          code = TSDB_CODE_MNODE_ENCRYPT_TYPE_NOT_MATCH;
×
UNCOV
1662
          mError("db:%s, failed to alter, encrypt algorithm not match, %s, expect type %d", alterReq.db,
×
1663
                 alterReq.encryptAlgrName, ENCRYPT_ALGR_TYPE__SYMMETRIC_CIPHERS);
UNCOV
1664
          goto _OVER;
×
1665
        }
1666
        if (pEncryptAlgr->id != pDb->cfg.encryptAlgorithm) {
1,438✔
1667
          code = TSDB_CODE_MND_ENCRYPT_NOT_ALLOW_CHANGE;
1,438✔
1668
          mError("db:%s, failed to alter, not allowed to change, %s", alterReq.db, alterReq.encryptAlgrName);
1,438✔
1669
          goto _OVER;
1,438✔
1670
        }
1671
      } else {
1672
        code = TSDB_CODE_MNODE_ENCRYPT_ALGR_NOT_EXIST;
741✔
1673
        mError("db:%s, failed to alter, encrypt algorithm not exist, %s", alterReq.db, alterReq.encryptAlgrName);
741✔
1674
        goto _OVER;
741✔
1675
      }
1676
    }
1677
  }
1678

1679
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
107,251✔
1680
  if (code != 0) {
107,251✔
1681
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
24,245✔
1682
    goto _OVER;
24,245✔
1683
  }
1684

1685
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
83,006✔
1686

1687
  dbObj.cfgVersion++;
81,612✔
1688
  dbObj.updateTime = taosGetTimestampMs();
81,612✔
1689
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
81,612✔
1690

1691
  if (dbObj.cfg.replications != pDb->cfg.replications) {
81,612✔
1692
    // return quickly, operation executed asynchronously
1693
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
5,429✔
1694
  } else {
1695
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
76,183✔
1696
  }
1697

1698
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
81,612✔
1699
    SName name = {0};
81,612✔
1700
    if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
81,612✔
UNCOV
1701
      mError("db:%s, failed to parse db name", alterReq.db);
×
1702

1703
    int64_t tse = taosGetTimestampMs();
81,612✔
1704
    double  duration = (double)(tse - tss);
81,612✔
1705
    duration = duration / 1000;
81,612✔
1706
    auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen, duration, 0);
81,612✔
1707
  }
1708

1709
_OVER:
110,151✔
1710
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
110,151✔
1711
    if (terrno != 0) code = terrno;
5,639✔
1712
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
5,639✔
1713
  }
1714

1715
  mndReleaseDb(pMnode, pDb);
110,151✔
1716
  taosArrayDestroy(dbObj.cfg.pRetensions);
110,151✔
1717
  tFreeSAlterDbReq(&alterReq);
110,151✔
1718

1719
  TAOS_RETURN(code);
110,151✔
1720
}
1721

1722
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb, char *algorithmsId) {
3,993,773✔
1723
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
3,993,773✔
1724
  cfgRsp->dbId = pDb->uid;
3,993,773✔
1725
  cfgRsp->ownerId = pDb->ownerId;
3,993,773✔
1726
  cfgRsp->cfgVersion = pDb->cfgVersion;
3,993,773✔
1727
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
3,993,773✔
1728
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
3,993,773✔
1729
  cfgRsp->buffer = pDb->cfg.buffer;
3,993,773✔
1730
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
3,993,773✔
1731
  cfgRsp->pageSize = pDb->cfg.pageSize;
3,993,773✔
1732
  cfgRsp->pages = pDb->cfg.pages;
3,993,773✔
1733
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
3,993,773✔
1734
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
3,993,773✔
1735
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
3,993,773✔
1736
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
3,993,773✔
1737
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
3,993,773✔
1738
  cfgRsp->minRows = pDb->cfg.minRows;
3,993,773✔
1739
  cfgRsp->maxRows = pDb->cfg.maxRows;
3,993,773✔
1740
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
3,993,773✔
1741
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
3,993,773✔
1742
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
3,993,773✔
1743
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
3,993,773✔
1744
  cfgRsp->walLevel = pDb->cfg.walLevel;
3,993,077✔
1745
  cfgRsp->precision = pDb->cfg.precision;
3,993,773✔
1746
  cfgRsp->compression = pDb->cfg.compression;
3,993,773✔
1747
  cfgRsp->replications = pDb->cfg.replications;
3,993,773✔
1748
  cfgRsp->strict = pDb->cfg.strict;
3,993,773✔
1749
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
3,993,773✔
1750
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
3,993,773✔
1751
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
3,993,077✔
1752
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
3,993,773✔
1753
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
3,993,773✔
1754
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
3,993,773✔
1755
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
3,993,773✔
1756
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
3,993,773✔
1757
  cfgRsp->schemaless = pDb->cfg.schemaless;
3,993,773✔
1758
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
3,993,773✔
1759
  cfgRsp->ssChunkSize = pDb->cfg.ssChunkSize;
3,993,077✔
1760
  cfgRsp->ssKeepLocal = pDb->cfg.ssKeepLocal;
3,993,773✔
1761
  cfgRsp->ssCompact = pDb->cfg.ssCompact;
3,993,773✔
1762
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
3,993,773✔
1763
  // cfgRsp->encryptAlgr = pDb->cfg.encryptAlgorithm;
1764
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
3,993,773✔
1765
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
3,993,773✔
1766
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
3,993,773✔
1767
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
3,993,773✔
1768
  cfgRsp->flags = pDb->cfg.flags;
3,993,773✔
1769
  tstrncpy(cfgRsp->algorithmsId, algorithmsId, sizeof(cfgRsp->algorithmsId));
3,993,077✔
1770
  cfgRsp->isAudit = pDb->cfg.isAudit;
3,993,773✔
1771
}
3,993,773✔
1772

1773
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
4,772,076✔
1774
  SMnode   *pMnode = pReq->info.node;
4,772,076✔
1775
  int32_t   code = -1;
4,772,076✔
1776
  SDbObj   *pDb = NULL;
4,772,076✔
1777
  SDbCfgReq cfgReq = {0};
4,772,076✔
1778
  SDbCfgRsp cfgRsp = {0};
4,772,076✔
1779

1780
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
4,772,076✔
1781

1782
  const char *pRealDb = strstr(cfgReq.db, ".");
4,772,076✔
1783
  if (pRealDb) {
4,772,076✔
1784
    ++pRealDb;
4,772,076✔
1785
  } else {
NEW
1786
    pRealDb = &cfgReq.db[0];
×
1787
  }
1788
  if (!IS_SYS_DBNAME(pRealDb)) {
4,772,076✔
1789
    pDb = mndAcquireDb(pMnode, cfgReq.db);
4,392,371✔
1790
    if (pDb == NULL) {
4,392,371✔
1791
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
445,468✔
1792
      if (terrno != 0) code = terrno;
445,468✔
1793
      goto _OVER;
445,468✔
1794
    }
1795

1796
    char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
3,946,903✔
1797
    mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
3,946,903✔
1798
    mndDumpDbCfgInfo(&cfgRsp, pDb, algrId);
3,946,903✔
1799
  }
1800

1801
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
4,326,608✔
1802
  void   *pRsp = rpcMallocCont(contLen);
4,326,608✔
1803
  if (pRsp == NULL) {
4,326,608✔
UNCOV
1804
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1805
    if (terrno != 0) code = terrno;
×
UNCOV
1806
    goto _OVER;
×
1807
  }
1808

1809
  int32_t ret = 0;
4,326,608✔
1810
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
4,326,608✔
UNCOV
1811
    code = ret;
×
UNCOV
1812
    goto _OVER;
×
1813
  }
1814

1815
  pReq->info.rsp = pRsp;
4,326,608✔
1816
  pReq->info.rspLen = contLen;
4,326,608✔
1817

1818
  code = 0;
4,326,608✔
1819

1820
_OVER:
4,772,076✔
1821

1822
  tFreeSDbCfgRsp(&cfgRsp);
4,772,076✔
1823

1824
  if (code != 0) {
4,772,076✔
1825
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
445,468✔
1826
  }
1827

1828
  mndReleaseDb(pMnode, pDb);
4,772,076✔
1829

1830
  TAOS_RETURN(code);
4,772,076✔
1831
}
1832

1833
int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
606,208✔
1834
  int32_t  code = 0;
606,208✔
1835
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
606,208✔
1836
  if (pRedoRaw == NULL) {
606,208✔
UNCOV
1837
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1838
    if (terrno != 0) code = terrno;
×
UNCOV
1839
    return -1;
×
1840
  }
1841
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
606,208✔
1842
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
606,208✔
1843

1844
  SSdb *pSdb = pMnode->pSdb;
606,208✔
1845
  void *pIter = NULL;
606,208✔
1846

1847
  while (1) {
4,200✔
1848
    SArbGroup *pArbGroup = NULL;
610,408✔
1849
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
610,408✔
1850
    if (pIter == NULL) break;
610,408✔
1851

1852
    if (pArbGroup->dbUid == pDb->uid) {
4,200✔
1853
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
4,200✔
UNCOV
1854
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1855
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
1856
        TAOS_RETURN(code);
×
1857
      }
1858
    }
1859

1860
    sdbRelease(pSdb, pArbGroup);
4,200✔
1861
  }
1862

1863
  TAOS_RETURN(code);
606,208✔
1864
}
1865

1866
int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
606,208✔
1867
  int32_t  code = 0;
606,208✔
1868
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
606,208✔
1869
  if (pCommitRaw == NULL) {
606,208✔
UNCOV
1870
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1871
    if (terrno != 0) code = terrno;
×
UNCOV
1872
    return -1;
×
1873
  }
1874
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
606,208✔
1875
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
606,208✔
1876

1877
  SSdb *pSdb = pMnode->pSdb;
606,208✔
1878
  void *pIter = NULL;
606,208✔
1879

1880
  while (1) {
4,200✔
1881
    SArbGroup *pArbGroup = NULL;
610,408✔
1882
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
610,408✔
1883
    if (pIter == NULL) break;
610,408✔
1884

1885
    if (pArbGroup->dbUid == pDb->uid) {
4,200✔
1886
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
4,200✔
UNCOV
1887
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1888
        sdbRelease(pSdb, pArbGroup);
×
UNCOV
1889
        TAOS_RETURN(code);
×
1890
      }
1891
    }
1892

1893
    sdbRelease(pSdb, pArbGroup);
4,200✔
1894
  }
1895

1896
  while (1) {
3,148,302✔
1897
    SVgObj *pVgroup = NULL;
3,754,510✔
1898
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,754,510✔
1899
    if (pIter == NULL) break;
3,754,510✔
1900

1901
    if (pVgroup->dbUid == pDb->uid) {
3,148,302✔
1902
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
1,294,963✔
1903
      if (pVgRaw == NULL) {
1,294,963✔
UNCOV
1904
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1905
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1906
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1907
        if (terrno != 0) code = terrno;
×
UNCOV
1908
        TAOS_RETURN(code);
×
1909
      }
1910
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
1,294,963✔
UNCOV
1911
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1912
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1913
        TAOS_RETURN(code);
×
1914
      }
1915
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
1,294,963✔
UNCOV
1916
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1917
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1918
        TAOS_RETURN(code);
×
1919
      }
1920
    }
1921

1922
    sdbRelease(pSdb, pVgroup);
3,148,302✔
1923
  }
1924

1925
  while (1) {
1,002,148✔
1926
    SStbObj *pStb = NULL;
1,608,356✔
1927
    ESdbStatus status;
1,608,272✔
1928
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
1,608,356✔
1929
    if (pIter == NULL) break;
1,608,356✔
1930

1931
    if (pStb->dbUid == pDb->uid) {
1,002,148✔
1932
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
522,044✔
1933
      if (pStbRaw == NULL) {
522,044✔
UNCOV
1934
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1935
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1936
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1937
        if (terrno != 0) code = terrno;
×
UNCOV
1938
        return -1;
×
1939
      }
1940
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
522,044✔
UNCOV
1941
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1942
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1943
        return -1;
×
1944
      }
1945
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
522,044✔
UNCOV
1946
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1947
        sdbRelease(pSdb, pStbRaw);
×
UNCOV
1948
        return -1;
×
1949
      }
1950
    }
1951

1952
    sdbRelease(pSdb, pStb);
1,002,148✔
1953
  }
1954

1955
  TAOS_RETURN(code);
606,208✔
1956
}
1957

1958
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
1,294,963✔
1959
  int32_t code = 0;
1,294,963✔
1960
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
2,654,232✔
1961
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
1,359,269✔
1962
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
1,359,269✔
1963
  }
1964

1965
  TAOS_RETURN(code);
1,294,963✔
1966
}
1967

1968
int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
606,208✔
1969
  int32_t code = 0;
606,208✔
1970
  SSdb   *pSdb = pMnode->pSdb;
606,208✔
1971
  void   *pIter = NULL;
606,208✔
1972

1973
  while (1) {
3,148,302✔
1974
    SVgObj *pVgroup = NULL;
3,754,510✔
1975
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,754,510✔
1976
    if (pIter == NULL) break;
3,754,510✔
1977

1978
    if (pVgroup->dbUid == pDb->uid) {
3,148,302✔
1979
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
1,294,963✔
UNCOV
1980
        sdbCancelFetch(pSdb, pIter);
×
UNCOV
1981
        sdbRelease(pSdb, pVgroup);
×
UNCOV
1982
        TAOS_RETURN(code);
×
1983
      }
1984
    }
1985

1986
    sdbRelease(pSdb, pVgroup);
3,148,302✔
1987
  }
1988

1989
  TAOS_RETURN(code);
606,208✔
1990
}
1991

1992
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
1,032,934✔
1993
  int32_t    code = 0;
1,032,934✔
1994
  SDropDbRsp dropRsp = {0};
1,032,934✔
1995
  if (pDb != NULL) {
1,032,934✔
1996
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
605,716✔
1997
    dropRsp.uid = pDb->uid;
605,716✔
1998
  }
1999

2000
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
1,032,934✔
2001
  void   *pRsp = NULL;
1,032,934✔
2002
  if (useRpcMalloc) {
1,032,934✔
2003
    pRsp = rpcMallocCont(rspLen);
427,218✔
2004
  } else {
2005
    pRsp = taosMemoryMalloc(rspLen);
605,716✔
2006
  }
2007

2008
  if (pRsp == NULL) {
1,032,934✔
UNCOV
2009
    code = terrno;
×
UNCOV
2010
    TAOS_RETURN(code);
×
2011
  }
2012

2013
  int32_t ret = 0;
1,032,934✔
2014
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
1,032,934✔
2015
  *pRspLen = rspLen;
1,032,934✔
2016
  *ppRsp = pRsp;
1,032,934✔
2017
  TAOS_RETURN(code);
1,032,934✔
2018
}
2019

2020
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
606,050✔
2021
  int32_t code = -1;
606,050✔
2022

2023
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
606,050✔
2024
  if (pTrans == NULL) {
606,050✔
2025
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2026
    if (terrno != 0) code = terrno;
×
UNCOV
2027
    goto _OVER;
×
2028
  }
2029

2030
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
606,050✔
2031

2032
  mndTransSetDbName(pTrans, pDb->name, NULL);
606,050✔
2033
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
606,050✔
2034

2035
  if (mndTopicExistsForDb(pMnode, pDb)) {
606,050✔
2036
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
334✔
2037
    goto _OVER;
334✔
2038
  }
2039

2040
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2041
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2042
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2043
#ifdef TD_ENTERPRISE
2044
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2045
#endif
2046
  TAOS_CHECK_GOTO(mndDropTSMAsByDb(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2047
  TAOS_CHECK_GOTO(mndDropRsmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2048
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2049
  //TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
2050
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
605,716✔
2051
  TAOS_CHECK_GOTO(mndPrincipalRemoveDb(pMnode, pTrans, pDb, NULL, NULL), NULL, _OVER);
605,716✔
2052

2053
  int32_t rspLen = 0;
605,716✔
2054
  void   *pRsp = NULL;
605,716✔
2055
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
605,716✔
2056
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
605,716✔
2057

2058
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
605,716✔
2059
  code = 0;
605,716✔
2060

2061
_OVER:
606,050✔
2062
  mndTransDrop(pTrans);
606,050✔
2063
  TAOS_RETURN(code);
606,050✔
2064
}
2065

2066
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
1,038,932✔
2067
  SMnode    *pMnode = pReq->info.node;
1,038,932✔
2068
  int32_t    code = -1;
1,038,932✔
2069
  SDbObj    *pDb = NULL;
1,038,932✔
2070
  SDropDbReq dropReq = {0};
1,038,932✔
2071
  int64_t    tss = taosGetTimestampMs();
1,038,932✔
2072

2073
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
1,038,932✔
2074

2075
  mInfo("db:%s, start to drop", dropReq.db);
1,038,932✔
2076

2077
  pDb = mndAcquireDb(pMnode, dropReq.db);
1,038,932✔
2078
  if (pDb == NULL) {
1,038,932✔
2079
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
430,915✔
2080
    if (terrno != 0) code = terrno;
430,915✔
2081
    if (dropReq.ignoreNotExists) {
430,915✔
2082
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
427,218✔
2083
    }
2084
    goto _OVER;
430,915✔
2085
  }
2086

2087
  if ((code = mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_DROP_DB, pDb))) {
608,017✔
NEW
2088
    goto _OVER;
×
2089
  }
2090

2091
  if(pDb->cfg.isMount) {
608,017✔
2092
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
492✔
2093
    goto _OVER;
492✔
2094
  }
2095

2096
  if(!pDb->cfg.allowDrop) {
607,525✔
NEW
2097
    code = TSDB_CODE_MND_DB_DROP_NOT_ALLOWED;
×
NEW
2098
    goto _OVER;
×
2099
  }
2100

2101
  SSdb *pSdb = pMnode->pSdb;
607,525✔
2102
  void *pIter = NULL;
607,525✔
2103

2104
  while (1) {
3,159,526✔
2105
    SVgObj *pVgroup = NULL;
3,767,051✔
2106
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,767,051✔
2107
    if (pIter == NULL) break;
3,767,051✔
2108

2109
    if (pVgroup->dbUid == pDb->uid) {
3,159,526✔
2110
      bool isFound = false;
1,295,788✔
2111
      for (int32_t i = 0; i < pVgroup->replica; i++) {
2,655,882✔
2112
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
1,360,094✔
UNCOV
2113
          isFound = true;
×
UNCOV
2114
          break;
×
2115
        }
2116
      }
2117
      if (!isFound) {
1,295,788✔
2118
        sdbRelease(pSdb, pVgroup);
1,295,788✔
2119
        continue;
1,295,788✔
2120
      }
UNCOV
2121
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
×
UNCOV
2122
      sdbCancelFetch(pSdb, pIter);
×
2123
      sdbRelease(pSdb, pVgroup);
×
2124
      goto _OVER;
×
2125
    }
2126

2127
    sdbRelease(pSdb, pVgroup);
1,863,738✔
2128
  }
2129

2130
  bool dbStream = false;
607,525✔
2131
  bool vtableStream = false;
607,525✔
2132
  mstCheckDbInUse(pMnode, dropReq.db, &dbStream, &vtableStream, true);
607,525✔
2133
  if (dbStream) {
607,525✔
2134
    code = TSDB_CODE_MND_STREAM_DB_IN_USE;
1,266✔
2135
    mError("db:%s used by streams, drop db not allowed", dropReq.db);
1,266✔
2136
    goto _OVER;
1,266✔
2137
  }
2138

2139
  if (vtableStream && !dropReq.force) {
606,259✔
2140
    code = TSDB_CODE_MND_STREAM_VTABLE_EXITS;
209✔
2141
    mError("db:%s, vtable stream exists, drop db not allowed", dropReq.db);
209✔
2142
    goto _OVER;
209✔
2143
  }
2144

2145
  code = mndDropDb(pMnode, pReq, pDb);
606,050✔
2146
  if (code == TSDB_CODE_SUCCESS) {
606,050✔
2147
    code = TSDB_CODE_ACTION_IN_PROGRESS;
605,716✔
2148
  }
2149

2150
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
606,050✔
2151
    SName name = {0};
606,050✔
2152
    if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
606,050✔
UNCOV
2153
      mError("db:%s, failed to parse db name", dropReq.db);
×
2154

2155
    int64_t tse = taosGetTimestampMs();
606,050✔
2156
    double  duration = (double)(tse - tss);
606,050✔
2157
    duration = duration / 1000;
606,050✔
2158
    auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen, duration, 0);
606,050✔
2159
  }
2160

2161
_OVER:
1,038,859✔
2162
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,038,932✔
2163
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
5,998✔
2164
  }
2165

2166
  mndReleaseDb(pMnode, pDb);
1,038,932✔
2167
  tFreeSDropDbReq(&dropReq);
1,038,932✔
2168
  TAOS_RETURN(code);
1,038,932✔
2169
}
2170

2171
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
26,439,627✔
2172
  int32_t numOfTables = 0;
26,439,627✔
2173
  int32_t vindex = 0;
26,439,627✔
2174
  SSdb   *pSdb = pMnode->pSdb;
26,439,627✔
2175

2176
  void *pIter = NULL;
26,439,630✔
2177
  while (vindex < pDb->cfg.numOfVgroups) {
106,829,124✔
2178
    SVgObj *pVgroup = NULL;
80,389,465✔
2179
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
80,389,465✔
2180
    if (pIter == NULL) break;
80,389,494✔
2181

2182
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
80,389,494✔
2183
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
53,115,781✔
2184
      vindex++;
53,115,810✔
2185
    }
2186

2187
    sdbRelease(pSdb, pVgroup);
80,389,491✔
2188
  }
2189

2190
  sdbCancelFetch(pSdb, pIter);
26,439,630✔
2191
  return numOfTables;
26,439,630✔
2192
}
2193

2194
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
26,512,083✔
2195
  int32_t vindex = 0;
26,512,083✔
2196
  SSdb   *pSdb = pMnode->pSdb;
26,512,083✔
2197

2198
  void *pIter = NULL;
26,512,112✔
2199
  while (1) {
55,273,016✔
2200
    SVgObj *pVgroup = NULL;
81,785,128✔
2201
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
81,785,128✔
2202
    if (pIter == NULL) break;
81,785,099✔
2203

2204
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
76,199,805✔
2205
      SVgroupInfo vgInfo = {0};
60,667,554✔
2206
      vgInfo.vgId = pVgroup->vgId;
60,667,554✔
2207
      vgInfo.hashBegin = pVgroup->hashBegin;
60,667,554✔
2208
      vgInfo.hashEnd = pVgroup->hashEnd;
60,667,554✔
2209
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
60,667,525✔
2210
      vgInfo.epSet.numOfEps = pVgroup->replica;
60,667,525✔
2211
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
125,024,371✔
2212
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
64,356,846✔
2213
        SEp       *pEp = &vgInfo.epSet.eps[gid];
64,356,846✔
2214
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
64,356,846✔
2215
        if (pDnode != NULL) {
64,356,817✔
2216
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
64,325,084✔
2217
          pEp->port = pDnode->port;
64,325,084✔
2218
        }
2219
        mndReleaseDnode(pMnode, pDnode);
64,356,764✔
2220
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
64,356,846✔
2221
          vgInfo.epSet.inUse = gid;
60,197,365✔
2222
        }
2223
      }
2224
      vindex++;
60,667,554✔
2225
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
60,667,554✔
UNCOV
2226
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
2227
      }
2228
    }
2229

2230
    sdbRelease(pSdb, pVgroup);
76,199,834✔
2231

2232
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
76,199,834✔
2233
      sdbCancelFetch(pSdb, pIter);
20,926,818✔
2234
      break;
20,926,818✔
2235
    }
2236
  }
2237
}
26,512,112✔
2238

2239
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
21,087,231✔
2240
  int32_t code = 0;
21,087,231✔
2241
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
21,087,231✔
2242
  if (pRsp->pVgroupInfos == NULL) {
21,087,231✔
UNCOV
2243
    code = terrno;
×
UNCOV
2244
    TAOS_RETURN(code);
×
2245
  }
2246

2247
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
21,087,228✔
2248

2249
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
21,087,231✔
2250
      pReq->stateTs < pDb->stateTs) {
1,106,652✔
2251
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
20,097,286✔
2252
  }
2253

2254
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
21,087,228✔
2255
  pRsp->uid = pDb->uid;
21,087,228✔
2256
  pRsp->vgVersion = pDb->vgVersion;
21,087,170✔
2257
  pRsp->stateTs = pDb->stateTs;
21,087,199✔
2258
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
21,087,170✔
2259
  pRsp->hashMethod = pDb->cfg.hashMethod;
21,087,199✔
2260
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
21,087,202✔
2261
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
21,087,228✔
2262
  pRsp->flags = pDb->cfg.flags;
21,087,196✔
2263
  TAOS_RETURN(code);
21,087,199✔
2264
}
2265

2266
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
26,484,847✔
2267
  SMnode   *pMnode = pReq->info.node;
26,484,847✔
2268
  int32_t   code = -1;
26,484,847✔
2269
  SDbObj   *pDb = NULL;
26,484,847✔
2270
  SUseDbReq usedbReq = {0};
26,484,847✔
2271
  SUseDbRsp usedbRsp = {0};
26,484,847✔
2272

2273
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
26,484,847✔
2274

2275
  char *p = strchr(usedbReq.db, '.');
26,484,534✔
2276
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
26,484,534✔
2277
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
5,298,805✔
2278
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
5,298,805✔
2279
    if (usedbReq.vgVersion < vgVersion) {
5,298,805✔
2280
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
5,291,846✔
2281
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
5,291,846✔
2282

2283
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
5,291,846✔
2284
      usedbRsp.vgVersion = vgVersion++;
5,291,846✔
2285
    } else {
2286
      usedbRsp.vgVersion = usedbReq.vgVersion;
6,959✔
2287
    }
2288
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
5,298,805✔
2289
    code = 0;
5,298,805✔
2290
  } else {
2291
    pDb = mndAcquireDb(pMnode, usedbReq.db);
21,185,336✔
2292
    if (pDb == NULL) {
21,186,042✔
2293
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
94,140✔
2294
      usedbRsp.uid = usedbReq.dbId;
94,140✔
2295
      usedbRsp.vgVersion = usedbReq.vgVersion;
94,140✔
2296
      usedbRsp.errCode = terrno;
94,140✔
2297

2298
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
94,140✔
2299
      if (terrno != 0) code = terrno;
94,140✔
2300
      goto _OVER;
94,140✔
2301
    } else {
2302
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_USE_DB, pDb), NULL, _OVER);
21,091,902✔
2303

2304
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
21,087,231✔
2305

2306
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
21,087,167✔
2307
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
2308
      code = 0;
21,087,231✔
2309
    }
2310
  }
2311

2312
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
26,386,036✔
2313
  void   *pRsp = rpcMallocCont(contLen);
26,385,797✔
2314
  if (pRsp == NULL) {
26,385,797✔
UNCOV
2315
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2316
    code = -1;
×
UNCOV
2317
    goto _OVER;
×
2318
  }
2319

2320
  int32_t ret = 0;
26,385,797✔
2321
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
26,385,797✔
UNCOV
2322
    code = ret;
×
UNCOV
2323
    goto _OVER;
×
2324
  }
2325

2326
  pReq->info.rsp = pRsp;
26,385,794✔
2327
  pReq->info.rspLen = contLen;
26,386,007✔
2328

2329
_OVER:
26,485,244✔
2330
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
26,485,244✔
2331
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
99,208✔
2332
  }
2333

2334
  mndReleaseDb(pMnode, pDb);
26,485,244✔
2335
  tFreeSUsedbRsp(&usedbRsp);
26,485,244✔
2336

2337
  TAOS_RETURN(code);
26,485,186✔
2338
}
2339

2340
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
6,627,876✔
2341
  int32_t       code = 0;
6,627,876✔
2342
  SDbHbBatchRsp batchRsp = {0};
6,627,876✔
2343
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
6,627,876✔
2344
  if (batchRsp.pArray == NULL) {
6,627,876✔
UNCOV
2345
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2346
    TAOS_RETURN(code);
×
2347
  }
2348

2349
  for (int32_t i = 0; i < numOfDbs; ++i) {
14,836,482✔
2350
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
8,208,606✔
2351
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
8,208,606✔
2352
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
8,208,606✔
2353
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
8,208,606✔
2354
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
8,208,606✔
2355
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
8,208,606✔
2356
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
8,208,606✔
2357

2358
    SDbHbRsp rsp = {0};
8,208,606✔
2359
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
8,208,606✔
2360
    rsp.dbId = pDbCacheInfo->dbId;
8,208,606✔
2361

2362
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
8,208,606✔
2363
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
7,589,875✔
2364
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
2,815,793✔
2365
      if (pDbCacheInfo->vgVersion >= vgVersion) {
2,815,793✔
2366
        continue;
2,522,564✔
2367
      }
2368

2369
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
293,448✔
2370
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
293,448✔
2371
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
293,448✔
2372

2373
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
293,448✔
2374
      rsp.useDbRsp->vgVersion = vgVersion++;
293,448✔
2375

2376
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
293,448✔
2377

2378
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
586,896✔
UNCOV
2379
        if (terrno != 0) code = terrno;
×
UNCOV
2380
        return code;
×
2381
      }
2382

2383
      continue;
293,448✔
2384
    }
2385

2386
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
5,392,813✔
2387
    if (pDb == NULL) {
5,392,813✔
2388
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
40,414✔
2389
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
40,414✔
2390
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
40,414✔
2391
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
40,414✔
2392
      rsp.useDbRsp->vgVersion = -1;
40,414✔
2393
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
80,828✔
UNCOV
2394
        if (terrno != 0) code = terrno;
×
UNCOV
2395
        return code;
×
2396
      }
2397
      continue;
40,414✔
2398
    }
2399

2400
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
5,352,399✔
2401

2402
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
5,352,399✔
2403
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
5,281,714✔
2404
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
4,502,149✔
2405
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
4,165,607✔
2406
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2407
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2408
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2409
      mndReleaseDb(pMnode, pDb);
4,165,607✔
2410
      continue;
4,165,607✔
2411
    } else {
2412
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
1,186,792✔
2413
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2414
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2415
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2416
    }
2417

2418
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
1,186,792✔
2419
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
46,870✔
2420
      char algrId[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
46,870✔
2421
      mndGetEncryptOsslAlgrNameById(pMnode, pDb->cfg.encryptAlgorithm, algrId);
46,870✔
2422
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb, algrId);
46,870✔
2423
    }
2424

2425
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
1,186,792✔
2426
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
771,088✔
2427
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
771,088✔
2428
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
771,088✔
2429
        bool    exist = false;
771,088✔
2430
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
771,088✔
2431
        if (TSDB_CODE_SUCCESS != code) {
771,088✔
2432
          mndReleaseDb(pMnode, pDb);
×
2433
          if (code != TSDB_CODE_NEED_RETRY) {
×
UNCOV
2434
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2435
          } else {
UNCOV
2436
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2437
          }
UNCOV
2438
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
UNCOV
2439
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
UNCOV
2440
          continue;
×
2441
        }
2442
        rsp.dbTsmaVersion = pDb->tsmaVersion;
771,088✔
2443
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
771,088✔
2444
      }
2445
    }
2446

2447
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
1,186,792✔
2448
        pDbCacheInfo->stateTs != pDb->stateTs) {
1,145,909✔
2449
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
829,532✔
2450
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
829,532✔
2451
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
829,532✔
UNCOV
2452
        mndReleaseDb(pMnode, pDb);
×
UNCOV
2453
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
UNCOV
2454
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2455
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2456
        continue;
×
2457
      }
2458

2459
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
829,532✔
2460
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
829,532✔
2461
      rsp.useDbRsp->uid = pDb->uid;
829,532✔
2462
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
829,532✔
2463
      rsp.useDbRsp->stateTs = pDb->stateTs;
829,532✔
2464
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
829,532✔
2465
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
829,532✔
2466
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
829,532✔
2467
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
829,532✔
2468
      rsp.useDbRsp->flags = pDb->cfg.flags;
829,532✔
2469
    }
2470

2471
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
2,373,584✔
UNCOV
2472
      mndReleaseDb(pMnode, pDb);
×
UNCOV
2473
      if (terrno != 0) code = terrno;
×
UNCOV
2474
      return code;
×
2475
    }
2476
    mndReleaseDb(pMnode, pDb);
1,186,792✔
2477
  }
2478

2479
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
6,627,876✔
2480
  void   *pRsp = taosMemoryMalloc(rspLen);
6,627,876✔
2481
  if (pRsp == NULL) {
6,627,876✔
UNCOV
2482
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2483
    tFreeSDbHbBatchRsp(&batchRsp);
×
UNCOV
2484
    return -1;
×
2485
  }
2486
  int32_t ret = 0;
6,627,876✔
2487
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
6,627,876✔
2488

2489
  *ppRsp = pRsp;
6,627,876✔
2490
  *pRspLen = rspLen;
6,627,876✔
2491

2492
  tFreeSDbHbBatchRsp(&batchRsp);
6,627,876✔
2493
  TAOS_RETURN(code);
6,627,876✔
2494
}
2495

2496
static int32_t mndSetTrimDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t startTs, STimeWindow tw,
32,014✔
2497
                                       SArray *vgroupIds, ETsdbOpType type, ETriggerType triggerType,
2498
                                       STrimDbRsp *pRsp) {
2499
  int32_t code = 0, lino = 0;
32,014✔
2500
  SSdb   *pSdb = pMnode->pSdb;
32,014✔
2501
  void   *pIter = NULL;
32,014✔
2502

2503
  SRetentionObj obj = {.optrType = type, .triggerType = triggerType};  // reuse SCompactObj struct
32,014✔
2504
  TAOS_CHECK_EXIT(mndAddRetentionToTrans(pMnode, pTrans, &obj, pDb, pRsp));
32,014✔
2505

2506
  int32_t j = 0;
32,014✔
2507
  int32_t numOfVgroups = taosArrayGetSize(vgroupIds);
32,014✔
2508
  if (numOfVgroups > 0) {
32,014✔
2509
    for (int32_t i = 0; i < numOfVgroups; i++) {
8,616✔
2510
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
6,462✔
2511
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
6,462✔
2512

2513
      if (pVgroup == NULL) {
6,462✔
2514
        mError("db:%s, vgroup:%" PRId64 " not exist", pDb->name, vgId);
2,154✔
2515
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
2,154✔
2516
      } else if (pVgroup->dbUid != pDb->uid) {
4,308✔
UNCOV
2517
        mError("db:%s, vgroup:%" PRId64 " not belong to db:%s", pDb->name, vgId, pDb->name);
×
UNCOV
2518
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2519
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
×
2520
      }
2521
      sdbRelease(pSdb, pVgroup);
4,308✔
2522
    }
2523

2524
    for (int32_t i = 0; i < numOfVgroups; i++) {
6,462✔
2525
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
4,308✔
2526
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
4,308✔
2527

2528
      if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
4,308✔
UNCOV
2529
        sdbRelease(pSdb, pVgroup);
×
UNCOV
2530
        TAOS_RETURN(code);
×
2531
      }
2532

2533
      for (int32_t i = 0; i < pVgroup->replica; i++) {
8,616✔
2534
        SVnodeGid *gid = &pVgroup->vnodeGid[i];
4,308✔
2535
        if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
4,308✔
UNCOV
2536
          sdbRelease(pSdb, pVgroup);
×
UNCOV
2537
          TAOS_RETURN(code);
×
2538
        }
2539
        j++;
4,308✔
2540
      }
2541
      sdbRelease(pSdb, pVgroup);
4,308✔
2542
    }
2543
  } else {
2544
    while (1) {
110,221✔
2545
      SVgObj *pVgroup = NULL;
137,927✔
2546
      pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
137,927✔
2547
      if (pIter == NULL) break;
137,927✔
2548

2549
      if (pVgroup->dbUid == pDb->uid) {
110,221✔
2550
        if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
62,890✔
UNCOV
2551
          sdbCancelFetch(pSdb, pIter);
×
UNCOV
2552
          sdbRelease(pSdb, pVgroup);
×
UNCOV
2553
          TAOS_RETURN(code);
×
2554
        }
2555

2556
        for (int32_t i = 0; i < pVgroup->replica; i++) {
128,302✔
2557
          SVnodeGid *gid = &pVgroup->vnodeGid[i];
65,412✔
2558
          if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
65,412✔
UNCOV
2559
            sdbCancelFetch(pSdb, pIter);
×
UNCOV
2560
            sdbRelease(pSdb, pVgroup);
×
UNCOV
2561
            TAOS_RETURN(code);
×
2562
          }
2563
          j++;
65,412✔
2564
        }
2565
      }
2566

2567
      sdbRelease(pSdb, pVgroup);
110,221✔
2568
    }
2569
  }
2570
_exit:
32,014✔
2571
  TAOS_RETURN(code);
32,014✔
2572
}
2573

2574
static int32_t mndBuildTrimDbRsp(STrimDbRsp *rsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
29,860✔
2575
  int32_t code = 0;
29,860✔
2576
  int32_t rspLen = tSerializeSCompactDbRsp(NULL, 0, (SCompactDbRsp *)rsp);
29,860✔
2577
  void   *pRsp = NULL;
29,860✔
2578
  if (useRpcMalloc) {
29,860✔
UNCOV
2579
    pRsp = rpcMallocCont(rspLen);
×
2580
  } else {
2581
    pRsp = taosMemoryMalloc(rspLen);
29,860✔
2582
  }
2583

2584
  if (pRsp == NULL) {
29,860✔
UNCOV
2585
    code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2586
    TAOS_RETURN(code);
×
2587
  }
2588

2589
  (void)tSerializeSCompactDbRsp(pRsp, rspLen, (SCompactDbRsp *)rsp);
29,860✔
2590
  *pRspLen = rspLen;
29,860✔
2591
  *ppRsp = pRsp;
29,860✔
2592
  TAOS_RETURN(code);
29,860✔
2593
}
2594

2595
int32_t mndTrimDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb, STimeWindow tw, SArray *vgroupIds,
36,322✔
2596
                         ETsdbOpType type, ETriggerType triggerType) {
2597
  SSdb      *pSdb = pMnode->pSdb;
36,322✔
2598
  SVgObj    *pVgroup = NULL;
36,322✔
2599
  void      *pIter = NULL;
36,322✔
2600
  int32_t    code = 0, lino = 0;
36,322✔
2601
  STrimDbRsp rsp = {0};
36,322✔
2602
  bool       isExist = false;
36,322✔
2603

2604
  while (1) {
11,110✔
2605
    SRetentionObj *pObj = NULL;
47,432✔
2606
    pIter = sdbFetch(pMnode->pSdb, SDB_RETENTION, pIter, (void **)&pObj);
47,432✔
2607
    if (pIter == NULL) break;
47,432✔
2608

2609
    if (strcmp(pObj->dbname, pDb->name) == 0) {
11,110✔
2610
      isExist = true;
4,308✔
2611
    }
2612
    sdbRelease(pMnode->pSdb, pObj);
11,110✔
2613
  }
2614
  if (isExist) {
36,322✔
2615
    mInfo("trim db:%s already exist", pDb->name);
4,308✔
2616
    TAOS_RETURN(TSDB_CODE_MND_TRIM_ALREADY_EXIST);
4,308✔
2617
  }
2618

2619
  int64_t startTs = taosGetTimestampMs();
32,014✔
2620
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "trim-db");
32,014✔
2621
  if (pTrans == NULL) goto _exit;
32,014✔
2622
  mInfo("trans:%d, used to trim db:%s", pTrans->id, pDb->name);
32,014✔
2623
  mndTransSetDbName(pTrans, pDb->name, NULL);
32,014✔
2624
  TAOS_CHECK_EXIT(mndTrancCheckConflict(pMnode, pTrans));
32,014✔
2625
  TAOS_CHECK_EXIT(mndSetTrimDbRedoActions(pMnode, pTrans, pDb, startTs, tw, vgroupIds, type, triggerType, &rsp));
32,014✔
2626
  if (pReq) {
29,860✔
2627
    int32_t rspLen = 0;
29,860✔
2628
    void   *pRsp = NULL;
29,860✔
2629
    rsp.bAccepted = true;
29,860✔
2630
    TAOS_CHECK_EXIT(mndBuildTrimDbRsp(&rsp, &rspLen, &pRsp, false));
29,860✔
2631
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
29,860✔
2632
  }
2633
  TAOS_CHECK_EXIT(mndTransPrepare(pMnode, pTrans));
29,860✔
2634
_exit:
32,014✔
2635
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
32,014✔
2636
    mError("db:%s, failed at line %d to trim db since %s", pDb->name, lino, tstrerror(code));
2,154✔
2637
  }
2638
  mndTransDrop(pTrans);
32,014✔
2639
  TAOS_RETURN(code);
32,014✔
2640
}
2641

2642
#ifdef TD_ENTERPRISE
2643
static int32_t mndCheckRsmaInDb(SMnode *pMnode, SDbObj *pDb) {
13,642✔
2644
  int32_t   code = 0;
13,642✔
2645
  void     *pIter = NULL;
13,642✔
2646
  SRsmaObj *pRsma = NULL;
13,642✔
2647
  while ((pIter = sdbFetch(pMnode->pSdb, SDB_RSMA, pIter, (void **)&pRsma))) {
13,642✔
2648
    if (pRsma->dbUid == pDb->uid) {
13,642✔
2649
      sdbRelease(pMnode->pSdb, pRsma);
13,642✔
2650
      sdbCancelFetch(pMnode->pSdb, pIter);
13,642✔
2651
      TAOS_RETURN(code);
13,642✔
2652
    }
UNCOV
2653
    sdbRelease(pMnode->pSdb, pRsma);
×
2654
  }
UNCOV
2655
  TAOS_RETURN(TSDB_CODE_RSMA_NOT_EXIST);
×
2656
}
2657
#endif
2658

2659
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
21,566✔
2660
  SMnode    *pMnode = pReq->info.node;
21,566✔
2661
  int32_t    code = 0, lino = 0;
21,566✔
2662
  SDbObj    *pDb = NULL;
21,566✔
2663
  STrimDbReq trimReq = {0};
21,566✔
2664
  int64_t    tss = taosGetTimestampMs();
21,566✔
2665

2666
  TAOS_CHECK_EXIT(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq));
21,566✔
2667

2668
  mInfo("db:%s, start to trim, optr:%u, tw:%" PRId64 ",%" PRId64, trimReq.db, trimReq.optrType, trimReq.tw.skey,
21,566✔
2669
        trimReq.tw.ekey);
2670

2671
  pDb = mndAcquireDb(pMnode, trimReq.db);
21,566✔
2672
  if (pDb == NULL) {
21,566✔
UNCOV
2673
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2674
    if (terrno != 0) code = terrno;
×
UNCOV
2675
    TAOS_CHECK_EXIT(code);
×
2676
  }
2677

2678
  TAOS_CHECK_EXIT(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_TRIM_DB, pDb));
21,566✔
2679

2680
  if (pDb->cfg.isMount) {
21,566✔
UNCOV
2681
    TAOS_CHECK_EXIT(TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT);
×
2682
  }
2683

2684
#ifdef TD_ENTERPRISE
2685
  if (trimReq.optrType == TSDB_OPTR_ROLLUP) {
21,566✔
2686
    TAOS_CHECK_EXIT(mndCheckRsmaInDb(pMnode, pDb));
13,642✔
2687
  }
2688
#endif
2689

2690
  TAOS_CHECK_EXIT(mndTrimDb(pMnode, pReq, pDb, trimReq.tw, trimReq.vgroupIds, trimReq.optrType, trimReq.triggerType));
21,566✔
2691
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
15,104✔
2692

2693
  if (tsAuditLevel >= AUDIT_LEVEL_DATABASE) {
15,104✔
2694
    SName name = {0};
15,104✔
2695
    if (tNameFromString(&name, trimReq.db, T_NAME_ACCT | T_NAME_DB) < 0) {
15,104✔
2696
      mWarn("db:%s, failed at line %d to parse db name", trimReq.db, __LINE__);
×
2697
    }
2698

2699
    char optrType[16] = {0};
15,104✔
2700
    (void)snprintf(optrType, sizeof(optrType), "%u", trimReq.optrType);
15,104✔
2701
    int64_t tse = taosGetTimestampMs();
15,104✔
2702
    double  duration = (double)(tse - tss);
15,104✔
2703
    duration = duration / 1000;
15,104✔
2704
    auditRecord(pReq, pMnode->clusterId, "trimDB", name.dbname, optrType, trimReq.sql, trimReq.sqlLen, duration, 0);
15,104✔
2705
  }
2706

2707
_exit:
21,566✔
2708
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
21,566✔
2709
    mError("db:%s, failed at line %d to process trim db req since %s, optr:%u", trimReq.db, lino, tstrerror(code),
6,462✔
2710
           trimReq.optrType);
2711
  }
2712

2713
  mndReleaseDb(pMnode, pDb);
21,566✔
2714
  tFreeSTrimDbReq(&trimReq);
21,566✔
2715
  TAOS_RETURN(code);
21,566✔
2716
}
2717

UNCOV
2718
static int32_t mndBuildSsMigrateDbRsp(SSsMigrateDbRsp *pSsMigrateRsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
×
UNCOV
2719
  int32_t code = 0;
×
UNCOV
2720
  int32_t rspLen = tSerializeSSsMigrateDbRsp(NULL, 0, pSsMigrateRsp);
×
UNCOV
2721
  void   *pRsp = NULL;
×
UNCOV
2722
  if (useRpcMalloc) {
×
UNCOV
2723
    pRsp = rpcMallocCont(rspLen);
×
2724
  } else {
UNCOV
2725
    pRsp = taosMemoryMalloc(rspLen);
×
2726
  }
2727

UNCOV
2728
  if (pRsp == NULL) {
×
UNCOV
2729
    code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2730
    TAOS_RETURN(code);
×
2731
  }
2732

UNCOV
2733
  (void)tSerializeSSsMigrateDbRsp(pRsp, rspLen, pSsMigrateRsp);
×
UNCOV
2734
  *pRspLen = rspLen;
×
UNCOV
2735
  *ppRsp = pRsp;
×
UNCOV
2736
  TAOS_RETURN(code);
×
2737
}
2738

2739
static int32_t mndTrimDbWal(SMnode *pMnode, SDbObj *pDb) {
1,074✔
2740
  SSdb   *pSdb = pMnode->pSdb;
1,074✔
2741
  SVgObj *pVgroup = NULL;
1,074✔
2742
  void   *pIter = NULL;
1,074✔
2743
  int32_t code = 0;
1,074✔
2744

2745
  // Iterate through all vgroups in this database
2746
  while (1) {
2,148✔
2747
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3,222✔
2748
    if (pIter == NULL) break;
3,222✔
2749

2750
    // Check if this vgroup belongs to the database
2751
    if (pVgroup->dbUid != pDb->uid) {
2,148✔
UNCOV
2752
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2753
      continue;
×
2754
    }
2755

2756
    // Prepare message to send to vnode
2757
    int32_t   contLen = sizeof(SMsgHead);
2,148✔
2758
    SMsgHead *pHead = rpcMallocCont(contLen);
2,148✔
2759
    if (pHead == NULL) {
2,148✔
UNCOV
2760
      sdbCancelFetch(pSdb, pVgroup);
×
UNCOV
2761
      sdbRelease(pSdb, pVgroup);
×
UNCOV
2762
      code = TSDB_CODE_OUT_OF_MEMORY;
×
2763
      break;
×
2764
    }
2765
    pHead->contLen = htonl(contLen);
2,148✔
2766
    pHead->vgId = htonl(pVgroup->vgId);
2,148✔
2767

2768
    // Send TRIM WAL request to vnode
2769
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM_WAL, .pCont = pHead, .contLen = contLen};
2,148✔
2770
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
2,148✔
2771
    code = tmsgSendReq(&epSet, &rpcMsg);
2,148✔
2772
    if (code != 0) {
2,148✔
UNCOV
2773
      mError("vgId:%d, failed to send vnode-trim-wal request since 0x%x", pVgroup->vgId, code);
×
2774
    } else {
2775
      mInfo("vgId:%d, send vnode-trim-wal request to vnode", pVgroup->vgId);
2,148✔
2776
    }
2777
    sdbRelease(pSdb, pVgroup);
2,148✔
2778
  }
2779

2780
  return code;
1,074✔
2781
}
2782

2783
static int32_t mndProcessTrimDbWalReq(SRpcMsg *pReq) {
1,074✔
2784
  SMnode    *pMnode = pReq->info.node;
1,074✔
2785
  int32_t    code = -1;
1,074✔
2786
  SDbObj    *pDb = NULL;
1,074✔
2787
  STrimDbReq trimReq = {0};
1,074✔
2788

2789
  TAOS_CHECK_GOTO(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq), NULL, _OVER);
1,074✔
2790

2791
  mInfo("db:%s, start to trim WAL", trimReq.db);
1,074✔
2792

2793
  pDb = mndAcquireDb(pMnode, trimReq.db);
1,074✔
2794
  if (pDb == NULL) {
1,074✔
UNCOV
2795
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2796
    if (terrno != 0) code = terrno;
×
UNCOV
2797
    goto _OVER;
×
2798
  }
2799

2800
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_TRIM_DB, pDb), NULL, _OVER);
1,074✔
2801

2802
  code = mndTrimDbWal(pMnode, pDb);
1,074✔
2803

2804
_OVER:
1,074✔
2805
  if (code != 0) {
1,074✔
2806
    mError("db:%s, failed to process trim db wal req since %s", trimReq.db, terrstr());
×
2807
  }
2808

2809
  mndReleaseDb(pMnode, pDb);
1,074✔
2810
  TAOS_RETURN(code);
1,074✔
2811
}
2812

UNCOV
2813
int32_t mndSsMigrateDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
×
UNCOV
2814
  int32_t       code = 0;
×
UNCOV
2815
  SSsMigrateDbRsp ssMigrateRsp = {0};
×
2816

UNCOV
2817
  bool  isExist = false;
×
UNCOV
2818
  void *pIter = NULL;
×
UNCOV
2819
  while (1) {
×
UNCOV
2820
    SSsMigrateObj *pSsMigrate = NULL;
×
UNCOV
2821
    pIter = sdbFetch(pMnode->pSdb, SDB_SSMIGRATE, pIter, (void **)&pSsMigrate);
×
UNCOV
2822
    if (pIter == NULL) break;
×
2823

UNCOV
2824
    if (strcmp(pSsMigrate->dbname, pDb->name) == 0) {
×
UNCOV
2825
      isExist = true;
×
2826
    }
UNCOV
2827
    sdbRelease(pMnode->pSdb, pSsMigrate);
×
2828
  }
2829
  if (isExist) {
×
2830
    mInfo("ssmigrate db:%s already exist", pDb->name);
×
2831

2832
    if (pReq) {
×
2833
      int32_t rspLen = 0;
×
UNCOV
2834
      void   *pRsp = NULL;
×
2835
      ssMigrateRsp.ssMigrateId = 0;
×
UNCOV
2836
      ssMigrateRsp.bAccepted = false;
×
UNCOV
2837
      TAOS_CHECK_RETURN(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, true));
×
2838

2839
      pReq->info.rsp = pRsp;
×
2840
      pReq->info.rspLen = rspLen;
×
2841
    }
2842

2843
    return TSDB_CODE_MND_SSMIGRATE_ALREADY_EXIST;
×
2844
  }
2845

2846
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "ssmigrate-db");
×
UNCOV
2847
  if (pTrans == NULL) {
×
UNCOV
2848
    mError("failed to create ssmigrate-db trans since %s", terrstr());
×
UNCOV
2849
    goto _OVER;
×
2850
  }
2851

UNCOV
2852
  mInfo("trans:%d, used to ssmigrate db:%s", pTrans->id, pDb->name);
×
UNCOV
2853
  mndTransSetDbName(pTrans, pDb->name, NULL);
×
UNCOV
2854
  TAOS_CHECK_GOTO(mndTrancCheckConflict(pMnode, pTrans), NULL, _OVER);
×
2855

UNCOV
2856
  SSsMigrateObj ssMigrate = { .startTime = taosGetTimestampMs() };
×
UNCOV
2857
  TAOS_CHECK_GOTO(mndAddSsMigrateToTran(pMnode, pTrans, &ssMigrate, pDb), NULL, _OVER);
×
2858

UNCOV
2859
  if (pReq) {
×
UNCOV
2860
    int32_t rspLen = 0;
×
UNCOV
2861
    void   *pRsp = NULL;
×
2862
    ssMigrateRsp.ssMigrateId = ssMigrate.id;
×
2863
    ssMigrateRsp.bAccepted = true;
×
UNCOV
2864
    TAOS_CHECK_GOTO(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, false), NULL, _OVER);
×
UNCOV
2865
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
×
2866
  }
2867

UNCOV
2868
  if (mndTransPrepare(pMnode, pTrans) != 0) goto _OVER;
×
UNCOV
2869
  code = 0;
×
2870

2871
_OVER:
×
2872
  mndTransDrop(pTrans);
×
2873
  TAOS_RETURN(code);
×
2874
}
2875

2876

UNCOV
2877
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq) {
×
UNCOV
2878
  SMnode         *pMnode = pReq->info.node;
×
UNCOV
2879
  int32_t         code = -1;
×
UNCOV
2880
  SDbObj         *pDb = NULL;
×
UNCOV
2881
  SSsMigrateDbReq ssMigrateReq = {0};
×
2882

2883
  if (!tsSsEnabled) {
×
UNCOV
2884
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
UNCOV
2885
    goto _OVER;
×
2886
  }
2887

UNCOV
2888
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_SHARED_STORAGE), NULL, _OVER);
×
UNCOV
2889
  TAOS_CHECK_GOTO(tDeserializeSSsMigrateDbReq(pReq->pCont, pReq->contLen, &ssMigrateReq), NULL, _OVER);
×
2890

UNCOV
2891
  mInfo("db:%s, start to ssmigrate", ssMigrateReq.db);
×
2892

UNCOV
2893
  pDb = mndAcquireDb(pMnode, ssMigrateReq.db);
×
UNCOV
2894
  if (pDb == NULL) {
×
UNCOV
2895
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2896
    if (terrno != 0) code = terrno;
×
UNCOV
2897
    goto _OVER;
×
2898
  }
2899

UNCOV
2900
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_SSMIGRATE_DB, pDb), NULL, _OVER);
×
2901

UNCOV
2902
  if(pDb->cfg.isMount) {
×
UNCOV
2903
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
UNCOV
2904
    goto _OVER;
×
2905
  }
2906

2907
  code = mndSsMigrateDb(pMnode, pReq, pDb);
×
2908

UNCOV
2909
_OVER:
×
UNCOV
2910
  if (code != 0) {
×
UNCOV
2911
    mError("db:%s, failed to process ssmigrate db req since %s", ssMigrateReq.db, tstrerror(code));
×
2912
  }
2913

UNCOV
2914
  mndReleaseDb(pMnode, pDb);
×
UNCOV
2915
  TAOS_RETURN(code);
×
2916
}
2917

2918
const char *mndGetDbStr(const char *src) {
18,344,318✔
2919
  char *pos = strstr(src, TS_PATH_DELIMITER);
18,344,318✔
2920
  if (pos != NULL) ++pos;
18,344,318✔
2921
  if (pos == NULL) return src;
18,344,318✔
2922
  return pos;
17,091,627✔
2923
}
2924

2925
const char *mndGetStableStr(const char *src) {
15,524,313✔
2926
  char *pos = strstr(src, TS_PATH_DELIMITER);
15,524,313✔
2927
  if (pos != NULL) ++pos;
15,524,313✔
2928
  if (pos == NULL) return src;
15,524,313✔
2929
  return mndGetDbStr(pos);
15,524,313✔
2930
}
2931

2932
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
×
UNCOV
2933
  int64_t v = 0;
×
2934
  switch (unit) {
×
2935
    case 's':
×
UNCOV
2936
      v = val / 1000;
×
2937
      break;
×
UNCOV
2938
    case 'm':
×
2939
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
×
2940
      break;
×
UNCOV
2941
    case 'h':
×
2942
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2943
      break;
×
2944
    case 'd':
×
2945
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
×
2946
      break;
×
2947
    case 'w':
×
UNCOV
2948
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2949
      break;
×
2950
    default:
×
UNCOV
2951
      break;
×
2952
  }
2953

UNCOV
2954
  return v;
×
2955
}
2956

2957
static char *buildRetension(SArray *pRetension) {
504,142✔
2958
  size_t size = taosArrayGetSize(pRetension);
504,142✔
2959
  if (size == 0) {
504,142✔
2960
    return NULL;
504,142✔
2961
  }
2962

2963
  char       *p1 = taosMemoryCalloc(1, 100);
×
2964
  SRetention *p = taosArrayGet(pRetension, 0);
×
2965

2966
  int32_t len = 2;
×
2967

UNCOV
2968
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2969
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2970
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2971

2972
  if (size > 1) {
×
2973
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2974
    p = taosArrayGet(pRetension, 1);
×
2975

UNCOV
2976
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
UNCOV
2977
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2978
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2979
  }
2980

2981
  if (size > 2) {
×
2982
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2983
    p = taosArrayGet(pRetension, 2);
×
2984

UNCOV
2985
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
UNCOV
2986
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2987
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2988
  }
2989

2990
  varDataSetLen(p1, len);
×
2991
  return p1;
×
2992
}
2993

2994
static const char *getCacheModelStr(int8_t cacheModel) {
504,142✔
2995
  switch (cacheModel) {
504,142✔
2996
    case TSDB_CACHE_MODEL_NONE:
469,109✔
2997
      return TSDB_CACHE_MODEL_NONE_STR;
469,109✔
2998
    case TSDB_CACHE_MODEL_LAST_ROW:
2,107✔
2999
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
2,107✔
3000
    case TSDB_CACHE_MODEL_LAST_VALUE:
5,384✔
3001
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
5,384✔
3002
    case TSDB_CACHE_MODEL_BOTH:
27,542✔
3003
      return TSDB_CACHE_MODEL_BOTH_STR;
27,542✔
3004
    default:
×
3005
      break;
×
3006
  }
3007
  return "unknown";
×
3008
}
3009

3010
static void getEncryptAlgorithmStr(SMnode *pMnode, int8_t encryptAlgorithm, char *out) {
504,142✔
3011
  switch (encryptAlgorithm) {
504,142✔
3012
    case TSDB_ENCRYPT_ALGO_NONE: {
504,142✔
3013
      tstrncpy(out, TSDB_ENCRYPT_ALGO_NONE_STR, TSDB_ENCRYPT_ALGR_NAME_LEN);
504,142✔
3014
      break;
504,142✔
3015
    }
UNCOV
3016
    default: {
×
3017
      SEncryptAlgrObj *obj = mndAcquireEncryptAlgrById(pMnode, encryptAlgorithm);
×
UNCOV
3018
      if (obj != NULL) {
×
3019
        tstrncpy(out, obj->algorithm_id, TSDB_ENCRYPT_ALGR_NAME_LEN);
×
3020
        mndReleaseEncryptAlgr(pMnode, obj);
×
3021
      }
UNCOV
3022
      break;
×
3023
    }
3024
  }
3025
}
504,142✔
3026

3027
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
1,930,061✔
3028
  if (pDb->cfg.replications == 1) return true;
1,930,061✔
3029

3030
  SSdb *pSdb = pMnode->pSdb;
411,170✔
3031
  void *pIter = NULL;
411,170✔
3032
  bool  isReady = true;
411,170✔
3033
  while (1) {
1,910,951✔
3034
    SVgObj *pVgroup = NULL;
2,322,121✔
3035
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
2,322,121✔
3036
    if (pIter == NULL) break;
2,322,121✔
3037

3038
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
1,910,951✔
3039
      bool hasLeader = false;
1,197,230✔
3040
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
4,742,978✔
3041
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
3,545,748✔
3042
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
3,127,521✔
3043
          hasLeader = true;
418,227✔
3044
        }
3045
      }
3046
      if (!hasLeader) isReady = false;
1,197,230✔
3047
    }
3048
    sdbRelease(pSdb, pVgroup);
1,910,951✔
3049
  }
3050

3051
  return isReady;
411,170✔
3052
}
3053

3054
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
1,383,894✔
3055
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo,
3056
                              SSHashObj **ppUidNames) {
3057
  int32_t cols = 0;
1,383,894✔
3058
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
1,383,894✔
3059
  char   *buf = taosMemoryMalloc(bytes);
1,383,894✔
3060
  if (buf == NULL) {
1,383,894✔
3061
    mError("db:%s, failed to malloc buffer", pDb->name);
×
UNCOV
3062
    return;
×
3063
  }
3064
  int32_t code = 0;
1,383,894✔
3065
  int32_t lino = 0;
1,383,894✔
3066

3067
  const char *name = mndGetDbStr(pDb->name);
1,383,894✔
3068
  if (name != NULL) {
1,383,894✔
3069
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
1,383,894✔
3070
  } else {
UNCOV
3071
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
3072
  }
3073

3074
  const char *precStr = NULL;
1,383,894✔
3075
  switch (pDb->cfg.precision) {
1,383,894✔
3076
    case TSDB_TIME_PRECISION_MILLI:
1,341,930✔
3077
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
1,341,930✔
3078
      break;
1,341,930✔
3079
    case TSDB_TIME_PRECISION_MICRO:
11,223✔
3080
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
11,223✔
3081
      break;
11,223✔
3082
    case TSDB_TIME_PRECISION_NANO:
30,741✔
3083
      precStr = TSDB_TIME_PRECISION_NANO_STR;
30,741✔
3084
      break;
30,741✔
UNCOV
3085
    default:
×
3086
      precStr = "none";
×
3087
      break;
×
3088
  }
3089
  char precVstr[10] = {0};
1,383,894✔
3090
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
1,383,894✔
3091

3092
  char *statusStr = "ready";
1,383,894✔
3093
  if (objStatus == SDB_STATUS_CREATING) {
1,383,894✔
UNCOV
3094
    statusStr = "creating";
×
3095
  } else if (objStatus == SDB_STATUS_DROPPING) {
1,383,894✔
3096
    statusStr = "dropping";
×
3097
  } else {
3098
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
1,383,894✔
3099
      statusStr = "unsynced";
2,364✔
3100
    }
3101
  }
3102
  char statusVstr[24] = {0};
1,383,894✔
3103
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
1,383,894✔
3104

3105
  if (sysDb || !sysinfo) {
1,383,894✔
3106
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
36,069,832✔
3107
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
35,190,080✔
3108
      if (i == 0) {
35,190,080✔
3109
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
879,752✔
3110
      } else if (i == 1) {
34,310,328✔
3111
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
879,752✔
3112
      } else if (i == 3) {
33,430,576✔
3113
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
879,752✔
3114
      } else if (i == 14) {
32,550,824✔
3115
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
879,752✔
3116
      } else if (i == 15) {
31,671,072✔
3117
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
879,752✔
3118
      } else {
3119
        colDataSetNULL(pColInfo, rows);
30,791,320✔
3120
      }
3121
    }
3122
  } else {
3123
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3124
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
504,142✔
3125

3126
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3127
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
504,142✔
3128

3129
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3130
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
504,142✔
3131

3132
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3133
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
504,142✔
3134

3135
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3136
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
504,142✔
3137

3138
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
504,142✔
3139
    char        strictVstr[24] = {0};
504,142✔
3140
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
504,142✔
3141
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3142
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
504,142✔
3143

3144
    char    durationStr[128] = {0};
504,142✔
3145
    char    durationVstr[128] = {0};
504,142✔
3146
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
504,142✔
3147
                                       pDb->cfg.daysPerFile);
3148

3149
    varDataSetLen(durationVstr, len);
504,142✔
3150
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3151
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
504,142✔
3152

3153
    char keepVstr[128] = {0};
504,142✔
3154
    char keep0Str[32] = {0};
504,142✔
3155
    char keep1Str[32] = {0};
504,142✔
3156
    char keep2Str[32] = {0};
504,142✔
3157

3158
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
504,142✔
3159
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
504,142✔
3160
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
504,142✔
3161

3162
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
504,142✔
UNCOV
3163
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
3164
    } else {
3165
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
504,142✔
3166
    }
3167
    varDataSetLen(keepVstr, len);
504,142✔
3168
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3169
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
504,142✔
3170

3171
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3172
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
504,142✔
3173

3174
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3175
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
504,142✔
3176

3177
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3178
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
504,142✔
3179

3180
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3181
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
504,142✔
3182

3183
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3184
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
504,142✔
3185

3186
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3187
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
504,142✔
3188

3189
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3190
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
504,142✔
3191

3192
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3193
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
504,142✔
3194

3195
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
504,142✔
3196
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3197
    if (rentensionVstr == NULL) {
504,142✔
3198
      colDataSetNULL(pColInfo, rows);
504,142✔
3199
    } else {
UNCOV
3200
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
×
UNCOV
3201
      taosMemoryFree(rentensionVstr);
×
3202
    }
3203

3204
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3205
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
504,142✔
3206

3207
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
504,142✔
3208
    char        cacheModelVstr[24] = {0};
504,142✔
3209
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
504,142✔
3210
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3211
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
504,142✔
3212

3213
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3214
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
504,142✔
3215

3216
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3217
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
504,142✔
3218

3219
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3220
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
504,142✔
3221

3222
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3223
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
504,142✔
3224

3225
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3226
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
504,142✔
3227

3228
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3229
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
504,142✔
3230

3231
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3232
    int16_t hashPrefix = pDb->cfg.hashPrefix;
504,142✔
3233
    if (hashPrefix > 0) {
504,142✔
3234
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
4,469✔
3235
    } else if (hashPrefix < 0) {
499,673✔
3236
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
5,008✔
3237
    }
3238
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
504,142✔
3239

3240
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3241
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
504,142✔
3242

3243
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3244
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
504,142✔
3245

3246
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3247
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
504,142✔
3248

3249
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3250
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssChunkSize, false), &lino, _OVER);
504,142✔
3251

3252
    char keeplocalVstr[128] = {0};
504,142✔
3253
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.ssKeepLocal);
504,142✔
3254
    varDataSetLen(keeplocalVstr, len);
504,142✔
3255
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3256
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
504,142✔
3257

3258
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3259
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssCompact, false), &lino, _OVER);
504,142✔
3260

3261
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3262
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
504,142✔
3263

3264
    char encryptAlgorithmStr[TSDB_ENCRYPT_ALGR_NAME_LEN] = {0};
504,142✔
3265
    getEncryptAlgorithmStr(pMnode, pDb->cfg.encryptAlgorithm, encryptAlgorithmStr);
504,142✔
3266
    char        encryptAlgorithmVStr[24] = {0};
504,142✔
3267
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
504,142✔
3268
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
504,142✔
3269
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
504,142✔
3270

3271
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
504,142✔
3272
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
504,142✔
3273
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
504,142✔
3274
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
504,142✔
3275
    }
3276

3277
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
504,142✔
3278
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
504,142✔
3279
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
504,142✔
3280
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
504,142✔
3281
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
504,142✔
3282
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
504,142✔
3283
    }
3284

3285
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
504,142✔
3286
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
504,142✔
3287
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
504,142✔
3288
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
504,142✔
3289
    }
3290

3291
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
504,142✔
3292
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.isAudit, false), &lino, _OVER);
504,142✔
3293
    }
3294

3295
    if (!(*ppUidNames)) {
504,142✔
3296
      TAOS_CHECK_GOTO(mndBuildUidNamesHash(pMnode, ppUidNames), &lino, _OVER);
288,882✔
3297
    }
3298
    const char *ownerName = tSimpleHashGet(*ppUidNames, (const char *)&pDb->ownerId, sizeof(pDb->ownerId));
504,142✔
3299
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%s", ownerName ? ownerName : "[unknown]"));
504,142✔
3300
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
504,142✔
3301
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
504,142✔
3302
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
504,142✔
3303
    }
3304

3305
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
504,142✔
3306
      uint8_t allowDrop = pDb->cfg.allowDrop;
504,142✔
3307
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&allowDrop, false), &lino, _OVER);
504,142✔
3308
    }
3309
  }
3310
_OVER:
1,383,894✔
3311
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
1,383,894✔
3312
  taosMemoryFree(buf);
1,383,894✔
3313
}
3314

3315
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
437,885✔
3316
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
437,885✔
3317
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
437,885✔
3318
  pDbObj->cfg.numOfVgroups = 0;
437,885✔
3319
  pDbObj->cfg.strict = 1;
437,885✔
3320
  pDbObj->cfg.replications = 1;
437,885✔
3321
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
437,885✔
3322
}
437,885✔
3323

3324
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
437,885✔
3325
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
437,885✔
3326
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
437,885✔
3327
  pDbObj->cfg.numOfVgroups = 0;
437,885✔
3328
  pDbObj->cfg.strict = 1;
437,885✔
3329
  pDbObj->cfg.replications = 1;
437,885✔
3330
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
437,885✔
3331
}
437,885✔
3332

3333
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
4,201,960✔
3334
  SVgObj  *pVgroup = pObj;
4,201,960✔
3335
  int32_t *numOfTables = p1;
4,201,960✔
3336
  int64_t  uid = *(int64_t *)p2;
4,201,960✔
3337
  if (pVgroup->dbUid == uid) {
4,201,960✔
3338
    *numOfTables += pVgroup->numOfTables;
1,361,620✔
3339
  }
3340
  return true;
4,201,960✔
3341
}
3342

3343
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
437,885✔
3344
  SMnode    *pMnode = pReq->info.node;
437,885✔
3345
  SSdb      *pSdb = pMnode->pSdb;
437,885✔
3346
  int32_t    numOfRows = 0;
437,885✔
3347
  SDbObj    *pDb = NULL;
437,885✔
3348
  SUserObj  *pOperUser = NULL;
437,885✔
3349
  ESdbStatus objStatus = 0;
437,885✔
3350
  SSHashObj *pUidNames = NULL;
437,885✔
3351

3352
  (void)mndAcquireUser(pMnode, RPC_MSG_USER(pReq), &pOperUser);
437,885✔
3353
  if (pOperUser == NULL) return 0;
437,885✔
3354
  bool sysinfo = pOperUser->sysInfo;
437,885✔
3355
  char objFName[TSDB_OBJ_FNAME_LEN + 1] = {0};
437,885✔
3356
  (void)snprintf(objFName, sizeof(objFName), "%d.*", pOperUser->acctId);
437,885✔
3357
  bool showAnyDb = (0 == mndCheckSysObjPrivilege(pMnode, pOperUser, RPC_MSG_TOKEN(pReq), PRIV_CM_SHOW, PRIV_OBJ_DB, 0,
437,885✔
3358
                                                 objFName, NULL));
3359

3360
  // Append the information_schema database into the result.
3361
  if (!pShow->sysDbRsp) {
437,885✔
3362
    SDbObj infoschemaDb = {0};
437,885✔
3363
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
437,885✔
3364
    size_t numOfTables = 0;
437,885✔
3365
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
437,885✔
3366
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1, &pUidNames);
437,885✔
3367

3368
    numOfRows += 1;
437,885✔
3369

3370
    SDbObj perfschemaDb = {0};
437,885✔
3371
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
437,885✔
3372
    numOfTables = 0;
437,885✔
3373
    getPerfDbMeta(NULL, &numOfTables);
437,885✔
3374
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1, &pUidNames);
437,885✔
3375

3376
    numOfRows += 1;
437,885✔
3377
    pShow->sysDbRsp = true;
437,885✔
3378
  }
3379

3380
  while (numOfRows < rowsCapacity) {
946,009✔
3381
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
946,009✔
3382
    if (pShow->pIter == NULL) break;
946,009✔
3383
    if (showAnyDb || mndCheckDbPrivilege(pMnode, RPC_MSG_USER(pReq), RPC_MSG_TOKEN(pReq), MND_OPER_SHOW_DATABASES, pDb) == 0) {
508,124✔
3384
      int32_t numOfTables = 0;
508,124✔
3385
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
508,124✔
3386
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo, &pUidNames);
508,124✔
3387
      numOfRows++;
508,124✔
3388
    }
3389

3390
    sdbRelease(pSdb, pDb);
508,124✔
3391
  }
3392

3393
  pShow->numOfRows += numOfRows;
437,885✔
3394
  mndReleaseUser(pMnode, pOperUser);
437,885✔
3395
  tSimpleHashCleanup(pUidNames);
437,885✔
3396
  return numOfRows;
437,885✔
3397
}
3398

UNCOV
3399
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
UNCOV
3400
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
3401
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
UNCOV
3402
}
×
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