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

taosdata / TDengine / #4791

13 Oct 2025 06:50AM UTC coverage: 57.628% (-0.8%) from 58.476%
#4791

push

travis-ci

web-flow
Merge pull request #33213 from taosdata/fix/huoh/timemoe_model_directory

fix: fix tdgpt timemoe model directory

136628 of 303332 branches covered (45.04%)

Branch coverage included in aggregate %.

208121 of 294900 relevant lines covered (70.57%)

4250784.02 hits per line

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

67.55
/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 "mndIndex.h"
27
#include "mndPrivilege.h"
28
#include "mndRetention.h"
29
#include "mndRetentionDetail.h"
30
#include "mndRsma.h"
31
#include "mndScan.h"
32
#include "mndShow.h"
33
#include "mndSma.h"
34
#include "mndSsMigrate.h"
35
#include "mndStb.h"
36
#include "mndStream.h"
37
#include "mndSubscribe.h"
38
#include "mndTopic.h"
39
#include "mndTrans.h"
40
#include "mndUser.h"
41
#include "mndVgroup.h"
42
#include "mndView.h"
43
#include "systable.h"
44
#include "thttp.h"
45
#include "tjson.h"
46

47
#define DB_VER_NUMBER   1
48
#define DB_RESERVE_SIZE 14
49

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

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

66
#ifndef TD_ENTERPRISE
67
int32_t mndProcessCompactDbReq(SRpcMsg *pReq) { return TSDB_CODE_OPS_NOT_SUPPORT; }
68
#endif
69

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

82
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_DB, mndProcessCreateDbReq);
1,332✔
83
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
1,332✔
84
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
1,332✔
85
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
1,332✔
86
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
1,332✔
87
  mndSetMsgHandle(pMnode, TDMT_MND_SCAN_DB, mndProcessScanDbReq);
1,332✔
88
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
1,332✔
89
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
1,332✔
90
  mndSetMsgHandle(pMnode, TDMT_MND_SSMIGRATE_DB, mndProcessSsMigrateDbReq);
1,332✔
91
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
1,332✔
92

93
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
1,332✔
94
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
1,332✔
95

96
  return sdbSetTable(pMnode->pSdb, table);
1,332✔
97
}
98

99
void mndCleanupDb(SMnode *pMnode) {}
1,332✔
100

101
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
13,995✔
102
  int32_t code = 0;
13,995✔
103
  int32_t lino = 0;
13,995✔
104
  terrno = TSDB_CODE_OUT_OF_MEMORY;
13,995✔
105

106
  int32_t  size = sizeof(SDbObj) + pDb->cfg.numOfRetensions * sizeof(SRetention) + DB_RESERVE_SIZE;
13,995✔
107
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, DB_VER_NUMBER, size);
13,995✔
108
  if (pRaw == NULL) goto _OVER;
13,995!
109

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

169
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
13,995!
170
  SDB_SET_UINT8(pRaw, dataPos, pDb->cfg.flags, _OVER)
13,995!
171
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
13,995!
172

173
  terrno = 0;
13,995✔
174

175
_OVER:
13,995✔
176
  if (terrno != 0) {
13,995!
177
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
178
    sdbFreeRaw(pRaw);
×
179
    return NULL;
×
180
  }
181

182
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
13,995✔
183
  return pRaw;
13,995✔
184
}
185

186
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
12,836✔
187
  int32_t code = 0;
12,836✔
188
  int32_t lino = 0;
12,836✔
189
  terrno = TSDB_CODE_OUT_OF_MEMORY;
12,836✔
190
  SSdbRow *pRow = NULL;
12,836✔
191
  SDbObj  *pDb = NULL;
12,836✔
192

193
  int8_t sver = 0;
12,836✔
194
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
12,836!
195

196
  if (sver != DB_VER_NUMBER) {
12,836!
197
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
198
    goto _OVER;
×
199
  }
200

201
  pRow = sdbAllocRow(sizeof(SDbObj));
12,836✔
202
  if (pRow == NULL) goto _OVER;
12,836!
203

204
  pDb = sdbGetRowObj(pRow);
12,836✔
205
  if (pDb == NULL) goto _OVER;
12,836!
206

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

277
  taosInitRWLatch(&pDb->lock);
12,836✔
278

279
  if (pDb->cfg.ssChunkSize == 0) {
12,836!
280
    pDb->cfg.ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
×
281

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

285
  if (pDb->cfg.ssKeepLocal == 0) {
12,836!
286
    pDb->cfg.ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
×
287

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

291
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
12,836!
292
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
12,836!
293
          TSDB_DEFAULT_TSDB_PAGESIZE);
294
  }
295

296
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
12,836✔
297
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
12,673!
298
  }
299

300
  terrno = 0;
12,836✔
301

302
_OVER:
12,836✔
303
  if (terrno != 0) {
12,836!
304
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
305
    taosMemoryFreeClear(pRow);
×
306
    return NULL;
×
307
  }
308

309
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
12,836✔
310
  return pRow;
12,836✔
311
}
312

313
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
3,019✔
314
  SSdb    *pSdb = pMnode->pSdb;
3,019✔
315
  SSdbRow *pRow = NULL;
3,019✔
316
  SDbObj  *pNewDb = NULL;
3,019✔
317
  int      code = -1;
3,019✔
318

319
  pRow = mndDbActionDecode(pRaw);
3,019✔
320
  if (pRow == NULL) {
3,019!
321
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
322
    if (terrno != 0) code = terrno;
×
323
    goto _OVER;
×
324
  }
325
  pNewDb = sdbGetRowObj(pRow);
3,019✔
326
  if (pNewDb == NULL) {
3,019!
327
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
328
    if (terrno != 0) code = terrno;
×
329
    goto _OVER;
×
330
  }
331

332
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
3,019✔
333
  if (pOldDb != NULL) {
3,019!
334
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
335
    sdbRelease(pMnode->pSdb, pOldDb);
×
336
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
337
    if (terrno != 0) code = terrno;
×
338
    goto _OVER;
×
339
  }
340

341
  code = 0;
3,019✔
342
_OVER:
3,019✔
343
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
3,019!
344
  taosMemoryFreeClear(pRow);
3,019!
345
  return code;
3,019✔
346
}
347

348
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
3,504✔
349
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
3,504✔
350
  return 0;
3,504✔
351
}
352

353
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
12,836✔
354
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
12,836✔
355
  taosArrayDestroy(pDb->cfg.pRetensions);
12,836✔
356
  return 0;
12,836✔
357
}
358

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

400
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
16,179✔
401
  SSdb *pSdb = pMnode->pSdb;
16,179✔
402
  return sdbGetTableVer(pSdb, SDB_VGROUP);
16,179✔
403
}
404

405
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
181,791✔
406
  SSdb   *pSdb = pMnode->pSdb;
181,791✔
407
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
181,791✔
408
  if (pDb == NULL) {
181,793✔
409
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
5,226✔
410
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
5,122✔
411
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
104✔
412
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
25✔
413
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
79!
414
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
79✔
415
    } else {
416
      terrno = TSDB_CODE_APP_ERROR;
×
417
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
418
    }
419
  }
420
  return pDb;
181,791✔
421
}
422

423
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
200,381✔
424
  SSdb *pSdb = pMnode->pSdb;
200,381✔
425
  sdbRelease(pSdb, pDb);
200,381✔
426
}
200,381✔
427

428
bool mndDbIsExist(SMnode *pMnode, const char *db) {
180✔
429
  SDbObj *pDb = mndAcquireDb(pMnode, db);
180✔
430
  if (pDb == NULL) {
180!
431
    return false;
×
432
  } else {
433
    mndReleaseDb(pMnode, pDb);
180✔
434
    pDb = NULL;
180✔
435
    return true;
180✔
436
  }
437
}
438

439
int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
2,803✔
440
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
2,803✔
441
  if (pos == NULL) {
2,803!
442
    return TSDB_CODE_MND_INVALID_DB;
×
443
  }
444

445
  int32_t acctId;
446
  int32_t code = taosStr2int32(dbName, &acctId);
2,803✔
447
  if (code != 0) {
2,803!
448
    return code;
×
449
  }
450

451
  if (acctId != pUser->acctId) {
2,803!
452
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
453
  }
454

455
  return 0;
2,803✔
456
}
457

458
int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
2,803✔
459
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
2,803✔
460

461
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
2,803!
462
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
2,803!
463
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
2,803!
464
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
2,803!
465
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
2,803!
466
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
2,803!
467
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
2,803!
468
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
2,803!
469
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
2,803!
470
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
2,803!
471
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
2,803!
472
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
2,803!
473
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
2,803!
474
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
2,803!
475
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
2,803!
476
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
2,803!
477
  if (pCfg->minRows > pCfg->maxRows) return code;
2,803✔
478
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
2,802!
479
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
2,802!
480
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
2,802!
481
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
2,802!
482
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
2,802!
483
#ifdef TD_ENTERPRISE
484
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
2,802!
485
  if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
2,802!
486
#else
487
  if (pCfg->replications != 1 && pCfg->replications != 3) return code;
488
  if (pCfg->encryptAlgorithm != TSDB_DEFAULT_ENCRYPT_ALGO) return code;
489
#endif
490

491
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
2,802!
492
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
2,802!
493
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
2,802!
494
  if (pCfg->hashMethod != 1) return code;
2,802!
495
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
2,802!
496
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
497
    TAOS_RETURN(code);
×
498
  }
499
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
2,802!
500
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
2,802!
501
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
2,802!
502
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
2,802!
503
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
2,802!
504
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
2,802!
505
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
2,802!
506
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
2,802!
507
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
2,802!
508
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
2,802!
509
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
2,802!
510

511
  if (pCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
2,802!
512
  if (pCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
2,802!
513
  if (pCfg->ssCompact < TSDB_MIN_SS_COMPACT || pCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
2,802!
514

515
  if (pCfg->compactInterval != 0 &&
2,802✔
516
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
4!
517
    return code;
×
518
  if (pCfg->compactStartTime != 0 &&
2,802✔
519
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
5!
520
    return code;
×
521
  if (pCfg->compactEndTime != 0 &&
2,802✔
522
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
5!
523
    return code;
×
524
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
2,802!
525
    return code;
×
526
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
2,802!
527
    return code;
×
528

529
  code = 0;
2,802✔
530
  TAOS_RETURN(code);
2,802✔
531
}
532

533
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
135✔
534
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
135✔
535
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
135!
536
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
135!
537
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
135!
538
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
135!
539
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
135!
540
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
135!
541
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
135!
542
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
135!
543
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
135!
544
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
135!
545
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
135!
546
    return code;
×
547
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
135!
548
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
135!
549
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
135!
550
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
135!
551
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
135!
552
#ifdef TD_ENTERPRISE
553
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
135!
554
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
135!
555
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
1!
556
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1✔
557
      return code;
1✔
558
    }
559
  }
560
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
134!
561
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
562
    return code;
×
563
  }
564
#else
565
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
566
#endif
567

568
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
134!
569
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
570
    return code;
×
571
  }
572
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
134!
573
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
574
    return code;
×
575
  }
576

577
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
134✔
578
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
2!
579
    return code;
×
580
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
134!
581
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
134!
582
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
134✔
583
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
133!
584
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
133!
585
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
133!
586
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
133!
587
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
588
    return code;
×
589
  }
590
  if (pNewCfg->ssChunkSize < TSDB_MIN_SS_CHUNK_SIZE || pNewCfg->ssChunkSize > TSDB_MAX_SS_CHUNK_SIZE) return code;
133!
591
  if (pNewCfg->ssKeepLocal < TSDB_MIN_SS_KEEP_LOCAL || pNewCfg->ssKeepLocal > TSDB_MAX_SS_KEEP_LOCAL) return code;
133!
592
  if (pNewCfg->ssCompact < TSDB_MIN_SS_COMPACT || pNewCfg->ssCompact > TSDB_MAX_SS_COMPACT) return code;
133!
593

594
  if (pNewCfg->compactInterval != 0 &&
133✔
595
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
9!
596
    return code;
×
597
  if (pNewCfg->compactStartTime != 0 &&
133✔
598
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
6!
599
    return code;
×
600
  if (pNewCfg->compactEndTime != 0 &&
133✔
601
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
6!
602
    return code;
×
603
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
133!
604
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
6!
605
    return code;
×
606
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
133!
607
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
133!
608
    return code;
×
609

610
  code = 0;
133✔
611
  TAOS_RETURN(code);
133✔
612
}
613

614
static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
2,799✔
615
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
2,799!
616
  if (pCfg->numOfStables < 0) pCfg->numOfStables = TSDB_DEFAULT_DB_SINGLE_STABLE;
2,799!
617
  if (pCfg->buffer < 0) pCfg->buffer = TSDB_DEFAULT_BUFFER_PER_VNODE;
2,799✔
618
  if (pCfg->pageSize < 0) pCfg->pageSize = TSDB_DEFAULT_PAGESIZE_PER_VNODE;
2,799✔
619
  if (pCfg->pages < 0) pCfg->pages = TSDB_DEFAULT_PAGES_PER_VNODE;
2,799✔
620
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DURATION_PER_FILE;
2,799!
621
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
2,799!
622
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
2,799!
623
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
2,799!
624
  if (pCfg->keepTimeOffset < 0) pCfg->keepTimeOffset = TSDB_DEFAULT_KEEP_TIME_OFFSET;
2,799!
625
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MINROWS_FBLOCK;
2,799!
626
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAXROWS_FBLOCK;
2,799!
627
  if (pCfg->walFsyncPeriod < 0) pCfg->walFsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
2,799!
628
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
2,799!
629
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
2,799!
630
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
2,799!
631
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA;
2,799!
632
  if (pCfg->strict < 0) pCfg->strict = TSDB_DEFAULT_DB_STRICT;
2,799!
633
  if (pCfg->cacheLast < 0) pCfg->cacheLast = TSDB_DEFAULT_CACHE_MODEL;
2,799!
634
  if (pCfg->cacheLastSize <= 0) pCfg->cacheLastSize = TSDB_DEFAULT_CACHE_SIZE;
2,799✔
635
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
2,799!
636
  if (pCfg->schemaless < 0) pCfg->schemaless = TSDB_DB_SCHEMALESS_OFF;
2,799!
637
  if (pCfg->walRetentionPeriod < 0 && pCfg->walRetentionPeriod != -1)
2,799!
638
    pCfg->walRetentionPeriod = TSDB_REPS_DEF_DB_WAL_RET_PERIOD;
×
639
  if (pCfg->walRetentionSize < 0 && pCfg->walRetentionSize != -1)
2,799!
640
    pCfg->walRetentionSize = TSDB_REPS_DEF_DB_WAL_RET_SIZE;
×
641
  if (pCfg->walRollPeriod < 0) pCfg->walRollPeriod = TSDB_REPS_DEF_DB_WAL_ROLL_PERIOD;
2,799!
642
  if (pCfg->walSegmentSize < 0) pCfg->walSegmentSize = TSDB_DEFAULT_DB_WAL_SEGMENT_SIZE;
2,799!
643
  if (pCfg->sstTrigger <= 0) pCfg->sstTrigger = TSDB_DEFAULT_SST_TRIGGER;
2,799✔
644
  if (pCfg->tsdbPageSize <= 0) pCfg->tsdbPageSize = TSDB_DEFAULT_TSDB_PAGESIZE;
2,799✔
645
  if (pCfg->ssChunkSize <= 0) pCfg->ssChunkSize = TSDB_DEFAULT_SS_CHUNK_SIZE;
2,799✔
646
  if (pCfg->ssKeepLocal <= 0) pCfg->ssKeepLocal = TSDB_DEFAULT_SS_KEEP_LOCAL;
2,799✔
647
  if (pCfg->ssCompact < 0) pCfg->ssCompact = TSDB_DEFAULT_SS_COMPACT;
2,799!
648
  if (pCfg->withArbitrator < 0) pCfg->withArbitrator = TSDB_DEFAULT_DB_WITH_ARBITRATOR;
2,799!
649
  if (pCfg->encryptAlgorithm < 0) pCfg->encryptAlgorithm = TSDB_DEFAULT_ENCRYPT_ALGO;
2,799!
650
}
2,799✔
651

652
int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
2,790✔
653
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
2,790✔
654
  if (pDbRaw == NULL) return -1;
2,790!
655

656
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
2,790!
657
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
2,790!
658
  return 0;
2,790✔
659
}
660

661
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
2,786✔
662
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
8,562✔
663
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
5,776!
664
  }
665
  return 0;
2,786✔
666
}
667

668
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
669
  int32_t  code = 0;
×
670
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
671
  if (pDbRaw == NULL) {
×
672
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
673
    if (terrno != 0) code = terrno;
×
674
    TAOS_RETURN(code);
×
675
  }
676
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
677
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
678

679
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
680
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
681
    if (pVgRaw == NULL) {
×
682
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
683
      if (terrno != 0) code = terrno;
×
684
      TAOS_RETURN(code);
×
685
    }
686
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
687
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
688
  }
689

690
  if (pDb->cfg.withArbitrator) {
×
691
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
692
      SVgObj   *pVgObj = pVgroups + v;
×
693
      SArbGroup arbGroup = {0};
×
694
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
695
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
696
    }
697
  }
698

699
  TAOS_RETURN(code);
×
700
}
701

702
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
2,786✔
703
  int32_t  code = 0;
2,786✔
704
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
2,786✔
705
  if (pDbRaw == NULL) {
2,786!
706
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
707
    if (terrno != 0) code = terrno;
×
708
    TAOS_RETURN(code);
×
709
  }
710
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
2,786!
711
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
2,786!
712

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

724
  if (pDb->cfg.withArbitrator) {
2,786✔
725
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
19✔
726
      SVgObj   *pVgObj = pVgroups + v;
11✔
727
      SArbGroup arbGroup = {0};
11✔
728
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
11!
729
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
11!
730
    }
731
  }
732

733
  TAOS_RETURN(code);
2,786✔
734
}
735

736
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
2,786✔
737
                                        SUserObj *pUserDuped) {
738
  int32_t  code = 0;
2,786✔
739
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
2,786✔
740
  if (pDbRaw == NULL) {
2,786!
741
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
742
    if (terrno != 0) code = terrno;
×
743
    TAOS_RETURN(code);
×
744
  }
745
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
2,786!
746
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
2,786!
747

748
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
8,562✔
749
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
5,776✔
750
    if (pVgRaw == NULL) {
5,776!
751
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
752
      if (terrno != 0) code = terrno;
×
753
      TAOS_RETURN(code);
×
754
    }
755
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
5,776!
756
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
5,776!
757
  }
758

759
  if (pDb->cfg.withArbitrator) {
2,786✔
760
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
19✔
761
      SVgObj   *pVgObj = pVgroups + v;
11✔
762
      SArbGroup arbGroup = {0};
11✔
763
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
11!
764
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
11!
765
    }
766
  }
767

768
  if (pUserDuped) {
2,786✔
769
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
10✔
770
    if (pUserRaw == NULL) {
10!
771
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
772
      if (terrno != 0) code = terrno;
×
773
      TAOS_RETURN(code);
×
774
    }
775
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
10!
776
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
10!
777
  }
778

779
  TAOS_RETURN(code);
2,786✔
780
}
781

782
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
2,786✔
783
  int32_t code = 0;
2,786✔
784
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
8,562✔
785
    SVgObj *pVgroup = pVgroups + vg;
5,776✔
786

787
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
12,343✔
788
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
6,567✔
789
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
6,567!
790
    }
791
  }
792

793
  TAOS_RETURN(code);
2,786✔
794
}
795

796
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
2,786✔
797
  int32_t code = 0;
2,786✔
798
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
8,562✔
799
    SVgObj *pVgroup = pVgroups + vg;
5,776✔
800

801
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
12,343✔
802
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
6,567✔
803
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
6,567!
804
    }
805
  }
806

807
  TAOS_RETURN(code);
2,786✔
808
}
809

810
static int32_t mndCreateDb(SMnode *pMnode, SRpcMsg *pReq, SCreateDbReq *pCreate, SUserObj *pUser, SArray *dnodeList) {
2,799✔
811
  int32_t  code = 0;
2,799✔
812
  SUserObj newUserObj = {0};
2,799✔
813
  SDbObj   dbObj = {0};
2,799✔
814
  (void)memcpy(dbObj.name, pCreate->db, TSDB_DB_FNAME_LEN);
2,799✔
815
  (void)memcpy(dbObj.acct, pUser->acct, TSDB_USER_LEN);
2,799✔
816
  dbObj.createdTime = taosGetTimestampMs();
2,799✔
817
  dbObj.updateTime = dbObj.createdTime;
2,799✔
818
  dbObj.uid = mndGenerateUid(dbObj.name, TSDB_DB_FNAME_LEN);
2,799✔
819
  dbObj.cfgVersion = 1;
2,799✔
820
  dbObj.vgVersion = 1;
2,799✔
821
  dbObj.tsmaVersion = 1;
2,799✔
822
  (void)memcpy(dbObj.createUser, pUser->user, TSDB_USER_LEN);
2,799✔
823
  dbObj.cfg = (SDbCfg){
2,799✔
824
      .numOfVgroups = pCreate->numOfVgroups,
2,799✔
825
      .numOfStables = pCreate->numOfStables,
2,799✔
826
      .buffer = pCreate->buffer,
2,799✔
827
      .pageSize = pCreate->pageSize,
2,799✔
828
      .pages = pCreate->pages,
2,799✔
829
      .cacheLastSize = pCreate->cacheLastSize,
2,799✔
830
      .daysPerFile = pCreate->daysPerFile,
2,799✔
831
      .daysToKeep0 = pCreate->daysToKeep0,
2,799✔
832
      .daysToKeep1 = pCreate->daysToKeep1,
2,799✔
833
      .daysToKeep2 = pCreate->daysToKeep2,
2,799✔
834
      .keepTimeOffset = pCreate->keepTimeOffset,
2,799✔
835
      .minRows = pCreate->minRows,
2,799✔
836
      .maxRows = pCreate->maxRows,
2,799✔
837
      .walFsyncPeriod = pCreate->walFsyncPeriod,
2,799✔
838
      .walLevel = pCreate->walLevel,
2,799✔
839
      .precision = pCreate->precision,
2,799✔
840
      .compression = pCreate->compression,
2,799✔
841
      .replications = pCreate->replications,
2,799✔
842
      .strict = pCreate->strict,
2,799✔
843
      .cacheLast = pCreate->cacheLast,
2,799✔
844
      .hashMethod = 1,
845
      .schemaless = pCreate->schemaless,
2,799✔
846
      .walRetentionPeriod = pCreate->walRetentionPeriod,
2,799✔
847
      .walRetentionSize = pCreate->walRetentionSize,
2,799✔
848
      .walRollPeriod = pCreate->walRollPeriod,
2,799✔
849
      .walSegmentSize = pCreate->walSegmentSize,
2,799✔
850
      .sstTrigger = pCreate->sstTrigger,
2,799✔
851
      .hashPrefix = pCreate->hashPrefix,
2,799✔
852
      .hashSuffix = pCreate->hashSuffix,
2,799✔
853
      .ssChunkSize = pCreate->ssChunkSize,
2,799✔
854
      .ssKeepLocal = pCreate->ssKeepLocal,
2,799✔
855
      .ssCompact = pCreate->ssCompact,
2,799✔
856
      .tsdbPageSize = pCreate->tsdbPageSize,
2,799✔
857
      .withArbitrator = pCreate->withArbitrator,
2,799✔
858
      .encryptAlgorithm = pCreate->encryptAlgorithm,
2,799✔
859
      .compactInterval = pCreate->compactInterval,
2,799✔
860
      .compactStartTime = pCreate->compactStartTime,
2,799✔
861
      .compactEndTime = pCreate->compactEndTime,
2,799✔
862
      .compactTimeOffset = pCreate->compactTimeOffset,
2,799✔
863
  };
864

865
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
2,799✔
866
  dbObj.cfg.pRetensions = pCreate->pRetensions;
2,799✔
867

868
  mndSetDefaultDbCfg(&dbObj.cfg);
2,799✔
869

870
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
2,799!
871
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, terrstr());
×
872
    TAOS_RETURN(code);
×
873
  }
874

875
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
2,799✔
876
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, terrstr());
1!
877
    TAOS_RETURN(code);
1✔
878
  }
879

880
  if (dbObj.cfg.hashPrefix > 0) {
2,798✔
881
    int32_t dbLen = strlen(dbObj.name) + 1;
7✔
882
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
7!
883
    dbObj.cfg.hashPrefix += dbLen;
7✔
884
  } else if (dbObj.cfg.hashPrefix < 0) {
2,791✔
885
    int32_t dbLen = strlen(dbObj.name) + 1;
7✔
886
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
7!
887
    dbObj.cfg.hashPrefix -= dbLen;
7✔
888
  }
889

890
  SVgObj *pVgroups = NULL;
2,798✔
891
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
2,798✔
892
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
12!
893
    TAOS_RETURN(code);
12✔
894
  }
895

896
  // add database privileges for user
897
  SUserObj *pNewUserDuped = NULL;
2,786✔
898
  if (!pUser->superUser) {
2,786✔
899
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
10!
900
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
10!
901
                    NULL, _OVER);
902
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
10!
903
                    NULL, _OVER);
904
    pNewUserDuped = &newUserObj;
10✔
905
  }
906

907
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
2,786✔
908
  if (pTrans == NULL) {
2,786!
909
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
910
    if (terrno != 0) code = terrno;
×
911
    goto _OVER;
×
912
  }
913
  // mndTransSetSerial(pTrans);
914
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
2,786!
915

916
  mndTransSetDbName(pTrans, dbObj.name, NULL);
2,786✔
917
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
2,786!
918

919
  void   *pIter = NULL;
2,786✔
920
  STrans *exitTrans = NULL;
2,786✔
921

922
  if (!pUser->superUser) {
2,786✔
923
    while (1) {
924
      pIter = sdbFetch(pMnode->pSdb, SDB_TRANS, pIter, (void **)&exitTrans);
10✔
925
      if (pIter == NULL) break;
10!
926

927
      if (exitTrans->conflict == TRN_CONFLICT_DB &&
×
928
          strncmp(exitTrans->opername, "create-db", TSDB_TRANS_OPER_LEN) == 0) {
×
929
        code = TSDB_CODE_MND_TRANS_CONFLICT;
×
930
        sdbRelease(pMnode->pSdb, exitTrans);
×
931
        sdbCancelFetch(pMnode->pSdb, pIter);
×
932
        goto _OVER;
×
933
      }
934

935
      sdbRelease(pMnode->pSdb, exitTrans);
×
936
    }
937
  }
938

939
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
2,786✔
940
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
2,786!
941
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
2,786!
942
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
2,786!
943
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
2,786!
944
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
2,786!
945
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
2,786!
946
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
2,786!
947

948
_OVER:
2,786✔
949
  taosMemoryFree(pVgroups);
2,786!
950
  mndUserFreeObj(&newUserObj);
2,786✔
951
  mndTransDrop(pTrans);
2,786✔
952
  TAOS_RETURN(code);
2,786✔
953
}
954

955
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
2,817✔
956
  int32_t    code = 0;
2,817✔
957
  SSdb      *pSdb = pMnode->pSdb;
2,817✔
958
  SDnodeObj *pDnode = NULL;
2,817✔
959
  void      *pIter = NULL;
2,817✔
960

961
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
962
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
2,817✔
963
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
4!
964
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
4✔
965
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
2✔
966
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
2!
967
           tsEncryptionKeyStat);
968
    goto _exit;
2✔
969
  }
970

971
  int64_t curMs = taosGetTimestampMs();
2✔
972
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
6✔
973
    bool online = false;
4✔
974
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
4!
975
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
976
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
977
      mError("db:%s, failed to check encryption key:%" PRIi8
×
978
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
979
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
980
             tsEncryptionKeyChksum);
981
      sdbCancelFetch(pSdb, pIter);
×
982
      sdbRelease(pSdb, pDnode);
×
983
      break;
×
984
    }
985
    sdbRelease(pSdb, pDnode);
4✔
986
  }
987
#else
988
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
989
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
990
    goto _exit;
991
  }
992
#endif
993
_exit:
2✔
994
  TAOS_RETURN(code);
2,817✔
995
}
996

997
#ifndef TD_ENTERPRISE
998
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
999
  if (dnodeListStr[0] != 0) {
1000
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1001
    return terrno;
1002
  } else {
1003
    return 0;
1004
  }
1005
}
1006
#endif
1007

1008
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
3,120✔
1009
  SMnode      *pMnode = pReq->info.node;
3,120✔
1010
  int32_t      code = -1;
3,120✔
1011
  int32_t      lino = 0;
3,120✔
1012
  SDbObj      *pDb = NULL;
3,120✔
1013
  SUserObj    *pUser = NULL;
3,120✔
1014
  SCreateDbReq createReq = {0};
3,120✔
1015
  SArray      *dnodeList = NULL;
3,120✔
1016

1017
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
3,120✔
1018
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
3,120!
1019

1020
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
3,120!
1021
#ifdef WINDOWS
1022
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
1023
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
1024
    goto _OVER;
1025
  }
1026
#endif
1027
  mInfo("db:%s, start to create, vgroups:%d", createReq.db, createReq.numOfVgroups);
3,120!
1028

1029
  pDb = mndAcquireDb(pMnode, createReq.db);
3,120✔
1030
  if (pDb != NULL) {
3,120✔
1031
    if (createReq.ignoreExist) {
277✔
1032
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
269!
1033
      code = 0;
269✔
1034
      goto _OVER;
269✔
1035
    } else {
1036
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
8✔
1037
      goto _OVER;
8✔
1038
    }
1039
  } else {
1040
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
2,843✔
1041
      code = terrno;
21✔
1042
      goto _OVER;
21✔
1043
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
2,822!
1044
      goto _OVER;
×
1045
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
2,822!
1046
      // continue
1047
    } else {  // TSDB_CODE_APP_ERROR
1048
      goto _OVER;
×
1049
    }
1050
  }
1051

1052
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_DB, NULL), &lino, _OVER);
2,822✔
1053

1054
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
2,817!
1055

1056
  int32_t nVnodes = createReq.numOfVgroups * createReq.replications;
2,817✔
1057
  TAOS_CHECK_GOTO(grantCheckEx(TSDB_GRANT_VNODE, &nVnodes), &lino, _OVER);
2,817!
1058

1059
  if (createReq.replications == 2) {
2,817✔
1060
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
10!
1061
  }
1062

1063
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
2,817✔
1064

1065
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
2,815✔
1066

1067
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pUser), &lino, _OVER);
2,799!
1068

1069
  if (sdbGetSize(pMnode->pSdb, SDB_MOUNT) > 0) {
2,799!
1070
    TAOS_CHECK_GOTO(TSDB_CODE_MND_MOUNT_NOT_EMPTY, &lino, _OVER);
×
1071
  }
1072

1073
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
2,799✔
1074
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
2,786!
1075

1076
  SName name = {0};
2,786✔
1077
  if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
2,786!
1078
    mError("db:%s, failed to parse db name", createReq.db);
×
1079

1080
  auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen);
2,786✔
1081

1082
_OVER:
3,120✔
1083
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
3,120✔
1084
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
65!
1085
  }
1086

1087
  mndReleaseDb(pMnode, pDb);
3,120✔
1088
  mndReleaseUser(pMnode, pUser);
3,120✔
1089
  tFreeSCreateDbReq(&createReq);
3,120✔
1090
  taosArrayDestroy(dnodeList);
3,120✔
1091

1092
  TAOS_RETURN(code);
3,120✔
1093
}
1094

1095
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
175✔
1096
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
175✔
1097

1098
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
175✔
1099
    pDb->cfg.buffer = pAlter->buffer;
7✔
1100
    code = 0;
7✔
1101
  }
1102

1103
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
175✔
1104
    pDb->cfg.pages = pAlter->pages;
3✔
1105
    code = 0;
3✔
1106
  }
1107

1108
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
175!
1109
    pDb->cfg.pageSize = pAlter->pageSize;
×
1110
    code = 0;
×
1111
  }
1112

1113
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
175!
1114
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
1115
    code = 0;
×
1116
  }
1117

1118
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
175✔
1119
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
20✔
1120
    code = 0;
20✔
1121
  }
1122

1123
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
175✔
1124
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
20✔
1125
    code = 0;
20✔
1126
  }
1127

1128
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
175✔
1129
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
16✔
1130
    code = 0;
16✔
1131
  }
1132

1133
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
175!
1134
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
5✔
1135
    code = 0;
5✔
1136
  }
1137

1138
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
175✔
1139
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
12✔
1140
    code = 0;
12✔
1141
  }
1142

1143
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
175✔
1144
    pDb->cfg.walLevel = pAlter->walLevel;
9✔
1145
    code = 0;
9✔
1146
  }
1147

1148
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
175!
1149
#if 1
1150
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1151
#else
1152
    pDb->cfg.strict = pAlter->strict;
1153
    code = 0;
1154
#endif
1155
  }
1156

1157
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
175✔
1158
    pDb->cfg.cacheLast = pAlter->cacheLast;
29✔
1159
    code = 0;
29✔
1160
  }
1161

1162
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
175✔
1163
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
5✔
1164
    code = 0;
5✔
1165
  }
1166

1167
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
175✔
1168
    pDb->cfg.replications = pAlter->replications;
10✔
1169
    pDb->vgVersion++;
10✔
1170
    code = 0;
10✔
1171
  }
1172

1173
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
175✔
1174
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
2✔
1175
    pDb->vgVersion++;
2✔
1176
    code = 0;
2✔
1177
  }
1178

1179
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
175✔
1180
    pDb->cfg.minRows = pAlter->minRows;
2✔
1181
    pDb->vgVersion++;
2✔
1182
    code = 0;
2✔
1183
  }
1184

1185
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
175✔
1186
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
32✔
1187
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
18✔
1188
    pDb->vgVersion++;
18✔
1189
    code = 0;
18✔
1190
  }
1191

1192
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
175✔
1193
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
2✔
1194
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
1✔
1195
    pDb->vgVersion++;
1✔
1196
    code = 0;
1✔
1197
  }
1198

1199
  if (pAlter->ssKeepLocal >= TSDB_MIN_SS_KEEP_LOCAL && pAlter->ssKeepLocal <= TSDB_MAX_SS_KEEP_LOCAL && pAlter->ssKeepLocal != pDb->cfg.ssKeepLocal) {
175!
1200
    pDb->cfg.ssKeepLocal = pAlter->ssKeepLocal;
1✔
1201
    pDb->vgVersion++;
1✔
1202
    code = 0;
1✔
1203
  }
1204

1205
  if (pAlter->ssCompact >= TSDB_MIN_SS_COMPACT && pAlter->ssCompact <= TSDB_MAX_SS_COMPACT && pAlter->ssCompact != pDb->cfg.ssCompact) {
175!
1206
    pDb->cfg.ssCompact = pAlter->ssCompact;
1✔
1207
    pDb->vgVersion++;
1✔
1208
    code = 0;
1✔
1209
  }
1210

1211
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
175✔
1212
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
1✔
1213
    pDb->vgVersion++;
1✔
1214
    code = 0;
1✔
1215
  }
1216

1217
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
175✔
1218
    pDb->cfg.compactInterval = pAlter->compactInterval;
7✔
1219
    pDb->vgVersion++;
7✔
1220
    code = 0;
7✔
1221
  }
1222

1223
  bool compactTimeRangeChanged = false;
175✔
1224
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
175✔
1225
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
172!
1226
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
172✔
1227
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
6✔
1228
    compactTimeRangeChanged = true;
6✔
1229
    code = 0;
6✔
1230
  }
1231

1232
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
175✔
1233
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
172!
1234
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
6✔
1235
    compactTimeRangeChanged = true;
6✔
1236
    code = 0;
6✔
1237
  }
1238
  if(compactTimeRangeChanged) {
175✔
1239
    pDb->vgVersion++;
6✔
1240
  }
1241

1242
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
175✔
1243
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
12✔
1244
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
5✔
1245
    pDb->vgVersion++;
5✔
1246
    code = 0;
5✔
1247
  }
1248

1249
  TAOS_RETURN(code);
175✔
1250
}
1251

1252
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
132✔
1253
  int32_t  code = 0;
132✔
1254
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
132✔
1255
  if (pRedoRaw == NULL) {
132!
1256
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1257
    if (terrno != 0) code = terrno;
×
1258
    TAOS_RETURN(code);
×
1259
  }
1260
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
132!
1261
    sdbFreeRaw(pRedoRaw);
×
1262
    TAOS_RETURN(code);
×
1263
  }
1264

1265
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
132!
1266
    sdbFreeRaw(pRedoRaw);
×
1267
    TAOS_RETURN(code);
×
1268
  }
1269
  return 0;
132✔
1270
}
1271

1272
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
132✔
1273
  int32_t  code = 0;
132✔
1274
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
132✔
1275
  if (pCommitRaw == NULL) {
132!
1276
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1277
    if (terrno != 0) code = terrno;
×
1278
    TAOS_RETURN(code);
×
1279
  }
1280
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
132!
1281
    sdbFreeRaw(pCommitRaw);
×
1282
    TAOS_RETURN(code);
×
1283
  }
1284

1285
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
132!
1286
    sdbFreeRaw(pCommitRaw);
×
1287
    TAOS_RETURN(code);
×
1288
  }
1289
  TAOS_RETURN(code);
132✔
1290
}
1291

1292
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
132✔
1293
  int32_t code = 0, lino = 0;
132✔
1294
  SSdb   *pSdb = pMnode->pSdb;
132✔
1295
  void   *pIter = NULL;
132✔
1296
  SVgObj *pVgroup = NULL;
132✔
1297
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
132✔
1298

1299
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
132!
1300

1301
  while (1) {
1302
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
664✔
1303
    if (pIter == NULL) break;
664✔
1304

1305
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
533✔
1306
      SVgObj newVgroup = {0};
345✔
1307
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
345✔
1308
                      _err);
1309

1310
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
344!
1311
        if (pNewDb->cfg.withArbitrator) {
×
1312
          SArbGroup arbGroup = {0};
×
1313
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1314
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1315
        } else {
1316
          SArbGroup arbGroup = {0};
×
1317
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
1318
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1319
        }
1320
      }
1321
    }
1322

1323
    sdbRelease(pSdb, pVgroup);
532✔
1324
  }
1325

1326
  taosArrayDestroy(pArray);
131✔
1327
  TAOS_RETURN(code);
131✔
1328

1329
_err:
1✔
1330
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
1!
1331

1332
  sdbCancelFetch(pSdb, pIter);
1✔
1333
  sdbRelease(pSdb, pVgroup);
1✔
1334
  taosArrayDestroy(pArray);
1✔
1335
  TAOS_RETURN(code);
1✔
1336
}
1337

1338
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
133✔
1339
  int32_t code = -1;
133✔
1340
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
133✔
1341
  if (pTrans == NULL) {
133!
1342
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1343
    if (terrno != 0) code = terrno;
×
1344
    TAOS_RETURN(code);
×
1345
  }
1346
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
133!
1347
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
133!
1348

1349
  mndTransSetDbName(pTrans, pOld->name, NULL);
133✔
1350
  mndTransSetGroupParallel(pTrans);
133✔
1351
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
133!
1352
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
133✔
1353
  TAOS_CHECK_GOTO(mndTransCheckConflictWithRetention(pMnode, pTrans), NULL, _OVER);
132!
1354

1355
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
132!
1356
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
132!
1357
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
132✔
1358

1359
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
131!
1360
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
131!
1361
  code = 0;
131✔
1362

1363
_OVER:
133✔
1364
  mndTransDrop(pTrans);
133✔
1365
  TAOS_RETURN(code);
133✔
1366
}
1367

1368
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
177✔
1369
  SMnode     *pMnode = pReq->info.node;
177✔
1370
  int32_t     code = -1;
177✔
1371
  SDbObj     *pDb = NULL;
177✔
1372
  SAlterDbReq alterReq = {0};
177✔
1373
  SDbObj      dbObj = {0};
177✔
1374

1375
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
177!
1376

1377
  mInfo("db:%s, start to alter", alterReq.db);
177!
1378

1379
  pDb = mndAcquireDb(pMnode, alterReq.db);
177✔
1380
  if (pDb == NULL) {
177!
1381
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1382
    if (terrno != 0) code = terrno;
×
1383
    goto _OVER;
×
1384
  }
1385

1386
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_ALTER_DB, pDb), NULL, _OVER);
177✔
1387

1388
  if (alterReq.replications == 2) {
175✔
1389
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
1!
1390
  }
1391

1392
  int32_t numOfTopics = 0;
175✔
1393
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
175!
1394

1395
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
175!
1396
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
1397
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1398
           numOfTopics);
1399
    goto _OVER;
×
1400
  }
1401

1402
  if (pDb->cfg.isMount) {
175!
1403
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
1404
    goto _OVER;
×
1405
  }
1406

1407
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
175✔
1408
  if (dbObj.cfg.pRetensions != NULL) {
175!
1409
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
1410
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1411
  }
1412

1413
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
175✔
1414
  if (code != 0) {
175✔
1415
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
40!
1416
    goto _OVER;
40✔
1417
  }
1418

1419
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
135✔
1420

1421
  dbObj.cfgVersion++;
133✔
1422
  dbObj.updateTime = taosGetTimestampMs();
133✔
1423
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
133✔
1424

1425
  if (dbObj.cfg.replications != pDb->cfg.replications) {
133✔
1426
    // return quickly, operation executed asynchronously
1427
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
9!
1428
  } else {
1429
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
124!
1430
  }
1431

1432
  SName name = {0};
133✔
1433
  if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
133!
1434
    mError("db:%s, failed to parse db name", alterReq.db);
×
1435

1436
  auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen);
133✔
1437

1438
_OVER:
177✔
1439
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
177✔
1440
    if (terrno != 0) code = terrno;
6✔
1441
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
6!
1442
  }
1443

1444
  mndReleaseDb(pMnode, pDb);
177✔
1445
  taosArrayDestroy(dbObj.cfg.pRetensions);
177✔
1446
  tFreeSAlterDbReq(&alterReq);
177✔
1447

1448
  TAOS_RETURN(code);
177✔
1449
}
1450

1451
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb) {
3,498✔
1452
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
3,498✔
1453
  cfgRsp->dbId = pDb->uid;
3,498✔
1454
  cfgRsp->cfgVersion = pDb->cfgVersion;
3,498✔
1455
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
3,498✔
1456
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
3,498✔
1457
  cfgRsp->buffer = pDb->cfg.buffer;
3,498✔
1458
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
3,498✔
1459
  cfgRsp->pageSize = pDb->cfg.pageSize;
3,498✔
1460
  cfgRsp->pages = pDb->cfg.pages;
3,498✔
1461
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
3,498✔
1462
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
3,498✔
1463
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
3,498✔
1464
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
3,498✔
1465
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
3,498✔
1466
  cfgRsp->minRows = pDb->cfg.minRows;
3,498✔
1467
  cfgRsp->maxRows = pDb->cfg.maxRows;
3,498✔
1468
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
3,498✔
1469
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
3,498✔
1470
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
3,498✔
1471
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
3,498✔
1472
  cfgRsp->walLevel = pDb->cfg.walLevel;
3,498✔
1473
  cfgRsp->precision = pDb->cfg.precision;
3,498✔
1474
  cfgRsp->compression = pDb->cfg.compression;
3,498✔
1475
  cfgRsp->replications = pDb->cfg.replications;
3,498✔
1476
  cfgRsp->strict = pDb->cfg.strict;
3,498✔
1477
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
3,498✔
1478
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
3,498✔
1479
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
3,498✔
1480
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
3,498✔
1481
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
3,498✔
1482
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
3,498✔
1483
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
3,498✔
1484
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
3,498✔
1485
  cfgRsp->schemaless = pDb->cfg.schemaless;
3,498✔
1486
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
3,498✔
1487
  cfgRsp->ssChunkSize = pDb->cfg.ssChunkSize;
3,498✔
1488
  cfgRsp->ssKeepLocal = pDb->cfg.ssKeepLocal;
3,498✔
1489
  cfgRsp->ssCompact = pDb->cfg.ssCompact;
3,498✔
1490
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
3,498✔
1491
  cfgRsp->encryptAlgorithm = pDb->cfg.encryptAlgorithm;
3,498✔
1492
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
3,498✔
1493
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
3,498✔
1494
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
3,498✔
1495
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
3,498✔
1496
  cfgRsp->flags = pDb->cfg.flags;
3,498✔
1497
}
3,498✔
1498

1499
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
3,854✔
1500
  SMnode   *pMnode = pReq->info.node;
3,854✔
1501
  int32_t   code = -1;
3,854✔
1502
  SDbObj   *pDb = NULL;
3,854✔
1503
  SDbCfgReq cfgReq = {0};
3,854✔
1504
  SDbCfgRsp cfgRsp = {0};
3,854✔
1505

1506
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
3,854!
1507

1508
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
3,854!
1509
    pDb = mndAcquireDb(pMnode, cfgReq.db);
3,854✔
1510
    if (pDb == NULL) {
3,854✔
1511
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
812✔
1512
      if (terrno != 0) code = terrno;
812!
1513
      goto _OVER;
812✔
1514
    }
1515

1516
    mndDumpDbCfgInfo(&cfgRsp, pDb);
3,042✔
1517
  }
1518

1519
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
3,042✔
1520
  void   *pRsp = rpcMallocCont(contLen);
3,042✔
1521
  if (pRsp == NULL) {
3,042!
1522
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1523
    if (terrno != 0) code = terrno;
×
1524
    goto _OVER;
×
1525
  }
1526

1527
  int32_t ret = 0;
3,042✔
1528
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
3,042!
1529
    code = ret;
×
1530
    goto _OVER;
×
1531
  }
1532

1533
  pReq->info.rsp = pRsp;
3,042✔
1534
  pReq->info.rspLen = contLen;
3,042✔
1535

1536
  code = 0;
3,042✔
1537

1538
_OVER:
3,854✔
1539

1540
  tFreeSDbCfgRsp(&cfgRsp);
3,854✔
1541

1542
  if (code != 0) {
3,854✔
1543
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
812!
1544
  }
1545

1546
  mndReleaseDb(pMnode, pDb);
3,854✔
1547

1548
  TAOS_RETURN(code);
3,854✔
1549
}
1550

1551
int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,450✔
1552
  int32_t  code = 0;
1,450✔
1553
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
1,450✔
1554
  if (pRedoRaw == NULL) {
1,450!
1555
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1556
    if (terrno != 0) code = terrno;
×
1557
    return -1;
×
1558
  }
1559
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
1,450!
1560
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
1,450!
1561

1562
  SSdb *pSdb = pMnode->pSdb;
1,450✔
1563
  void *pIter = NULL;
1,450✔
1564

1565
  while (1) {
6✔
1566
    SArbGroup *pArbGroup = NULL;
1,456✔
1567
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
1,456✔
1568
    if (pIter == NULL) break;
1,456✔
1569

1570
    if (pArbGroup->dbUid == pDb->uid) {
6!
1571
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
6!
1572
        sdbCancelFetch(pSdb, pIter);
×
1573
        sdbRelease(pSdb, pArbGroup);
×
1574
        TAOS_RETURN(code);
×
1575
      }
1576
    }
1577

1578
    sdbRelease(pSdb, pArbGroup);
6✔
1579
  }
1580

1581
  TAOS_RETURN(code);
1,450✔
1582
}
1583

1584
int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,450✔
1585
  int32_t  code = 0;
1,450✔
1586
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
1,450✔
1587
  if (pCommitRaw == NULL) {
1,450!
1588
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1589
    if (terrno != 0) code = terrno;
×
1590
    return -1;
×
1591
  }
1592
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
1,450!
1593
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
1,450!
1594

1595
  SSdb *pSdb = pMnode->pSdb;
1,450✔
1596
  void *pIter = NULL;
1,450✔
1597

1598
  while (1) {
6✔
1599
    SArbGroup *pArbGroup = NULL;
1,456✔
1600
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
1,456✔
1601
    if (pIter == NULL) break;
1,456✔
1602

1603
    if (pArbGroup->dbUid == pDb->uid) {
6!
1604
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
6!
1605
        sdbCancelFetch(pSdb, pIter);
×
1606
        sdbRelease(pSdb, pArbGroup);
×
1607
        TAOS_RETURN(code);
×
1608
      }
1609
    }
1610

1611
    sdbRelease(pSdb, pArbGroup);
6✔
1612
  }
1613

1614
  while (1) {
5,101✔
1615
    SVgObj *pVgroup = NULL;
6,551✔
1616
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
6,551✔
1617
    if (pIter == NULL) break;
6,551✔
1618

1619
    if (pVgroup->dbUid == pDb->uid) {
5,101✔
1620
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
2,864✔
1621
      if (pVgRaw == NULL) {
2,864!
1622
        sdbCancelFetch(pSdb, pIter);
×
1623
        sdbRelease(pSdb, pVgroup);
×
1624
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1625
        if (terrno != 0) code = terrno;
×
1626
        TAOS_RETURN(code);
×
1627
      }
1628
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
2,864!
1629
        sdbCancelFetch(pSdb, pIter);
×
1630
        sdbRelease(pSdb, pVgroup);
×
1631
        TAOS_RETURN(code);
×
1632
      }
1633
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
2,864!
1634
        sdbCancelFetch(pSdb, pIter);
×
1635
        sdbRelease(pSdb, pVgroup);
×
1636
        TAOS_RETURN(code);
×
1637
      }
1638
    }
1639

1640
    sdbRelease(pSdb, pVgroup);
5,101✔
1641
  }
1642

1643
  while (1) {
2,870✔
1644
    SStbObj *pStb = NULL;
4,320✔
1645
    ESdbStatus status;
1646
    pIter = sdbFetchAll(pSdb, SDB_STB, pIter, (void **)&pStb, &status, true);
4,320✔
1647
    if (pIter == NULL) break;
4,320✔
1648

1649
    if (pStb->dbUid == pDb->uid) {
2,870✔
1650
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
1,196✔
1651
      if (pStbRaw == NULL) {
1,196!
1652
        sdbCancelFetch(pSdb, pIter);
×
1653
        sdbRelease(pSdb, pStbRaw);
×
1654
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1655
        if (terrno != 0) code = terrno;
×
1656
        return -1;
×
1657
      }
1658
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
1,196!
1659
        sdbCancelFetch(pSdb, pIter);
×
1660
        sdbRelease(pSdb, pStbRaw);
×
1661
        return -1;
×
1662
      }
1663
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
1,196!
1664
        sdbCancelFetch(pSdb, pIter);
×
1665
        sdbRelease(pSdb, pStbRaw);
×
1666
        return -1;
×
1667
      }
1668
    }
1669

1670
    sdbRelease(pSdb, pStb);
2,870✔
1671
  }
1672

1673
  TAOS_RETURN(code);
1,450✔
1674
}
1675

1676
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
2,864✔
1677
  int32_t code = 0;
2,864✔
1678
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
5,832✔
1679
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
2,968✔
1680
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
2,968!
1681
  }
1682

1683
  TAOS_RETURN(code);
2,864✔
1684
}
1685

1686
int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,450✔
1687
  int32_t code = 0;
1,450✔
1688
  SSdb   *pSdb = pMnode->pSdb;
1,450✔
1689
  void   *pIter = NULL;
1,450✔
1690

1691
  while (1) {
5,101✔
1692
    SVgObj *pVgroup = NULL;
6,551✔
1693
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
6,551✔
1694
    if (pIter == NULL) break;
6,551✔
1695

1696
    if (pVgroup->dbUid == pDb->uid) {
5,101✔
1697
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
2,864!
1698
        sdbCancelFetch(pSdb, pIter);
×
1699
        sdbRelease(pSdb, pVgroup);
×
1700
        TAOS_RETURN(code);
×
1701
      }
1702
    }
1703

1704
    sdbRelease(pSdb, pVgroup);
5,101✔
1705
  }
1706

1707
  TAOS_RETURN(code);
1,450✔
1708
}
1709

1710
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
2,549✔
1711
  int32_t    code = 0;
2,549✔
1712
  SDropDbRsp dropRsp = {0};
2,549✔
1713
  if (pDb != NULL) {
2,549✔
1714
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
1,446✔
1715
    dropRsp.uid = pDb->uid;
1,446✔
1716
  }
1717

1718
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
2,549✔
1719
  void   *pRsp = NULL;
2,549✔
1720
  if (useRpcMalloc) {
2,549✔
1721
    pRsp = rpcMallocCont(rspLen);
1,103✔
1722
  } else {
1723
    pRsp = taosMemoryMalloc(rspLen);
1,446!
1724
  }
1725

1726
  if (pRsp == NULL) {
2,549!
1727
    code = terrno;
×
1728
    TAOS_RETURN(code);
×
1729
  }
1730

1731
  int32_t ret = 0;
2,549✔
1732
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
2,549!
1733
  *pRspLen = rspLen;
2,549✔
1734
  *ppRsp = pRsp;
2,549✔
1735
  TAOS_RETURN(code);
2,549✔
1736
}
1737

1738
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
1,447✔
1739
  int32_t code = -1;
1,447✔
1740

1741
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
1,447✔
1742
  if (pTrans == NULL) {
1,447!
1743
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1744
    if (terrno != 0) code = terrno;
×
1745
    goto _OVER;
×
1746
  }
1747

1748
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
1,447!
1749

1750
  mndTransSetDbName(pTrans, pDb->name, NULL);
1,447✔
1751
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
1,447!
1752

1753
  if (mndTopicExistsForDb(pMnode, pDb)) {
1,447✔
1754
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
1✔
1755
    goto _OVER;
1✔
1756
  }
1757

1758
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1759
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1760
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1761
#ifdef TD_ENTERPRISE
1762
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1763
#endif
1764
  TAOS_CHECK_GOTO(mndDropTSMAsByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1765
  TAOS_CHECK_GOTO(mndDropRsmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1766
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1767
  //TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
1768
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
1,446!
1769
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb, NULL), NULL, _OVER);
1,446!
1770

1771
  int32_t rspLen = 0;
1,446✔
1772
  void   *pRsp = NULL;
1,446✔
1773
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
1,446!
1774
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
1,446✔
1775

1776
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,446!
1777
  code = 0;
1,446✔
1778

1779
_OVER:
1,447✔
1780
  mndTransDrop(pTrans);
1,447✔
1781
  TAOS_RETURN(code);
1,447✔
1782
}
1783

1784
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
2,577✔
1785
  SMnode    *pMnode = pReq->info.node;
2,577✔
1786
  int32_t    code = -1;
2,577✔
1787
  SDbObj    *pDb = NULL;
2,577✔
1788
  SDropDbReq dropReq = {0};
2,577✔
1789

1790
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
2,577!
1791

1792
  mInfo("db:%s, start to drop", dropReq.db);
2,577!
1793

1794
  pDb = mndAcquireDb(pMnode, dropReq.db);
2,577✔
1795
  if (pDb == NULL) {
2,577✔
1796
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
1,111✔
1797
    if (terrno != 0) code = terrno;
1,111!
1798
    if (dropReq.ignoreNotExists) {
1,111✔
1799
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
1,103✔
1800
    }
1801
    goto _OVER;
1,111✔
1802
  }
1803

1804
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_DB, pDb), NULL, _OVER);
1,466✔
1805

1806
  if(pDb->cfg.isMount) {
1,458✔
1807
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
4✔
1808
    goto _OVER;
4✔
1809
  }
1810

1811
  SSdb *pSdb = pMnode->pSdb;
1,454✔
1812
  void *pIter = NULL;
1,454✔
1813

1814
  while (1) {
5,143✔
1815
    SVgObj *pVgroup = NULL;
6,597✔
1816
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
6,597✔
1817
    if (pIter == NULL) break;
6,597✔
1818

1819
    if (pVgroup->dbUid == pDb->uid) {
5,143✔
1820
      bool isFound = false;
2,864✔
1821
      for (int32_t i = 0; i < pVgroup->replica; i++) {
5,832✔
1822
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
2,968!
1823
          isFound = true;
×
1824
          break;
×
1825
        }
1826
      }
1827
      if (!isFound) {
2,864!
1828
        sdbRelease(pSdb, pVgroup);
2,864✔
1829
        continue;
2,864✔
1830
      }
1831
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
×
1832
      sdbCancelFetch(pSdb, pIter);
×
1833
      sdbRelease(pSdb, pVgroup);
×
1834
      goto _OVER;
×
1835
    }
1836

1837
    sdbRelease(pSdb, pVgroup);
2,279✔
1838
  }
1839

1840
  bool dbStream = false;
1,454✔
1841
  bool vtableStream = false;
1,454✔
1842
  mstCheckDbInUse(pMnode, dropReq.db, &dbStream, &vtableStream, true);
1,454✔
1843
  if (dbStream) {
1,454✔
1844
    code = TSDB_CODE_MND_STREAM_DB_IN_USE;
6✔
1845
    mError("db:%s used by streams, drop db not allowed", dropReq.db);
6!
1846
    goto _OVER;
6✔
1847
  }
1848

1849
  if (vtableStream && !dropReq.force) {
1,448✔
1850
    code = TSDB_CODE_MND_STREAM_VTABLE_EXITS;
1✔
1851
    mError("db:%s, vtable stream exists, drop db not allowed", dropReq.db);
1!
1852
    goto _OVER;
1✔
1853
  }
1854

1855
  code = mndDropDb(pMnode, pReq, pDb);
1,447✔
1856
  if (code == TSDB_CODE_SUCCESS) {
1,447✔
1857
    code = TSDB_CODE_ACTION_IN_PROGRESS;
1,446✔
1858
  }
1859

1860
  SName name = {0};
1,447✔
1861
  if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
1,447!
1862
    mError("db:%s, failed to parse db name", dropReq.db);
×
1863

1864
  auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen);
1,447✔
1865

1866
_OVER:
2,577✔
1867
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
2,577✔
1868
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
28!
1869
  }
1870

1871
  mndReleaseDb(pMnode, pDb);
2,577✔
1872
  tFreeSDropDbReq(&dropReq);
2,577✔
1873
  TAOS_RETURN(code);
2,577✔
1874
}
1875

1876
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
45,410✔
1877
  int32_t numOfTables = 0;
45,410✔
1878
  int32_t vindex = 0;
45,410✔
1879
  SSdb   *pSdb = pMnode->pSdb;
45,410✔
1880

1881
  void *pIter = NULL;
45,410✔
1882
  while (vindex < pDb->cfg.numOfVgroups) {
186,097✔
1883
    SVgObj *pVgroup = NULL;
140,686✔
1884
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
140,686✔
1885
    if (pIter == NULL) break;
140,685!
1886

1887
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
140,685✔
1888
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
91,646✔
1889
      vindex++;
91,646✔
1890
    }
1891

1892
    sdbRelease(pSdb, pVgroup);
140,685✔
1893
  }
1894

1895
  sdbCancelFetch(pSdb, pIter);
45,411✔
1896
  return numOfTables;
45,411✔
1897
}
1898

1899
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
50,344✔
1900
  int32_t vindex = 0;
50,344✔
1901
  SSdb   *pSdb = pMnode->pSdb;
50,344✔
1902

1903
  void *pIter = NULL;
50,344✔
1904
  while (1) {
106,922✔
1905
    SVgObj *pVgroup = NULL;
157,266✔
1906
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
157,266✔
1907
    if (pIter == NULL) break;
157,266✔
1908

1909
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
143,792!
1910
      SVgroupInfo vgInfo = {0};
112,378✔
1911
      vgInfo.vgId = pVgroup->vgId;
112,378✔
1912
      vgInfo.hashBegin = pVgroup->hashBegin;
112,378✔
1913
      vgInfo.hashEnd = pVgroup->hashEnd;
112,378✔
1914
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
112,378✔
1915
      vgInfo.epSet.numOfEps = pVgroup->replica;
112,378✔
1916
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
237,208✔
1917
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
124,829✔
1918
        SEp       *pEp = &vgInfo.epSet.eps[gid];
124,829✔
1919
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
124,829✔
1920
        if (pDnode != NULL) {
124,830✔
1921
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
124,775✔
1922
          pEp->port = pDnode->port;
124,775✔
1923
        }
1924
        mndReleaseDnode(pMnode, pDnode);
124,830✔
1925
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
124,830!
1926
          vgInfo.epSet.inUse = gid;
110,874✔
1927
        }
1928
      }
1929
      vindex++;
112,379✔
1930
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
112,377!
1931
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
1932
      }
1933
    }
1934

1935
    sdbRelease(pSdb, pVgroup);
143,791✔
1936

1937
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
143,792✔
1938
      sdbCancelFetch(pSdb, pIter);
36,870✔
1939
      break;
36,870✔
1940
    }
1941
  }
1942
}
50,344✔
1943

1944
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
39,019✔
1945
  int32_t code = 0;
39,019✔
1946
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
39,019✔
1947
  if (pRsp->pVgroupInfos == NULL) {
39,018!
1948
    code = terrno;
×
1949
    TAOS_RETURN(code);
×
1950
  }
1951

1952
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
39,018✔
1953

1954
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
39,019!
1955
      pReq->stateTs < pDb->stateTs) {
4,191✔
1956
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
35,123✔
1957
  }
1958

1959
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
39,019✔
1960
  pRsp->uid = pDb->uid;
39,019✔
1961
  pRsp->vgVersion = pDb->vgVersion;
39,019✔
1962
  pRsp->stateTs = pDb->stateTs;
39,019✔
1963
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
39,019✔
1964
  pRsp->hashMethod = pDb->cfg.hashMethod;
39,018✔
1965
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
39,018✔
1966
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
39,018✔
1967
  pRsp->flags = pDb->cfg.flags;
39,018✔
1968
  TAOS_RETURN(code);
39,018✔
1969
}
1970

1971
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
52,401✔
1972
  SMnode   *pMnode = pReq->info.node;
52,401✔
1973
  int32_t   code = -1;
52,401✔
1974
  SDbObj   *pDb = NULL;
52,401✔
1975
  SUseDbReq usedbReq = {0};
52,401✔
1976
  SUseDbRsp usedbRsp = {0};
52,401✔
1977

1978
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
52,401!
1979

1980
  char *p = strchr(usedbReq.db, '.');
52,401✔
1981
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
52,401!
1982
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
13,061✔
1983
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
13,061✔
1984
    if (usedbReq.vgVersion < vgVersion) {
13,061✔
1985
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
13,029✔
1986
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
13,029!
1987

1988
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
13,029✔
1989
      usedbRsp.vgVersion = vgVersion++;
13,029✔
1990
    } else {
1991
      usedbRsp.vgVersion = usedbReq.vgVersion;
32✔
1992
    }
1993
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
13,061✔
1994
    code = 0;
13,061✔
1995
  } else {
1996
    pDb = mndAcquireDb(pMnode, usedbReq.db);
39,340✔
1997
    if (pDb == NULL) {
39,340✔
1998
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
278✔
1999
      usedbRsp.uid = usedbReq.dbId;
278✔
2000
      usedbRsp.vgVersion = usedbReq.vgVersion;
278✔
2001
      usedbRsp.errCode = terrno;
278✔
2002

2003
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
278✔
2004
      if (terrno != 0) code = terrno;
278!
2005
      goto _OVER;
278✔
2006
    } else {
2007
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_USE_DB, pDb), NULL, _OVER);
39,062✔
2008

2009
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
39,019!
2010

2011
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
39,018✔
2012
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
2013
      code = 0;
39,019✔
2014
    }
2015
  }
2016

2017
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
52,080✔
2018
  void   *pRsp = rpcMallocCont(contLen);
52,080✔
2019
  if (pRsp == NULL) {
52,080!
2020
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2021
    code = -1;
×
2022
    goto _OVER;
×
2023
  }
2024

2025
  int32_t ret = 0;
52,080✔
2026
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
52,080!
2027
    code = ret;
×
2028
    goto _OVER;
×
2029
  }
2030

2031
  pReq->info.rsp = pRsp;
52,080✔
2032
  pReq->info.rspLen = contLen;
52,080✔
2033

2034
_OVER:
52,401✔
2035
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
52,401!
2036
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
321!
2037
  }
2038

2039
  mndReleaseDb(pMnode, pDb);
52,401✔
2040
  tFreeSUsedbRsp(&usedbRsp);
52,401✔
2041

2042
  TAOS_RETURN(code);
52,401✔
2043
}
2044

2045
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
7,798✔
2046
  int32_t       code = 0;
7,798✔
2047
  SDbHbBatchRsp batchRsp = {0};
7,798✔
2048
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
7,798✔
2049
  if (batchRsp.pArray == NULL) {
7,798!
2050
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2051
    TAOS_RETURN(code);
×
2052
  }
2053

2054
  for (int32_t i = 0; i < numOfDbs; ++i) {
17,348✔
2055
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
9,550✔
2056
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
9,550✔
2057
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
9,550✔
2058
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
9,550✔
2059
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
9,550✔
2060
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
9,550✔
2061
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
9,550✔
2062

2063
    SDbHbRsp rsp = {0};
9,550✔
2064
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
9,550✔
2065
    rsp.dbId = pDbCacheInfo->dbId;
9,550✔
2066

2067
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
9,550✔
2068
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
9,100✔
2069
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
3,118✔
2070
      if (pDbCacheInfo->vgVersion >= vgVersion) {
3,118✔
2071
        continue;
7,242✔
2072
      }
2073

2074
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
445!
2075
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
445✔
2076
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
445✔
2077

2078
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
445✔
2079
      rsp.useDbRsp->vgVersion = vgVersion++;
445✔
2080

2081
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
445✔
2082

2083
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
890!
2084
        if (terrno != 0) code = terrno;
×
2085
        return code;
×
2086
      }
2087

2088
      continue;
445✔
2089
    }
2090

2091
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
6,432✔
2092
    if (pDb == NULL) {
6,432✔
2093
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
40!
2094
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
40!
2095
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
40✔
2096
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
40✔
2097
      rsp.useDbRsp->vgVersion = -1;
40✔
2098
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
80!
2099
        if (terrno != 0) code = terrno;
×
2100
        return code;
×
2101
      }
2102
      continue;
40✔
2103
    }
2104

2105
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
6,392✔
2106

2107
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
6,392✔
2108
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
5,874!
2109
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
4,348✔
2110
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
4,084✔
2111
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2112
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2113
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2114
      mndReleaseDb(pMnode, pDb);
4,084✔
2115
      continue;
4,084✔
2116
    } else {
2117
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
2,308✔
2118
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2119
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2120
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2121
    }
2122

2123
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
2,308✔
2124
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
456!
2125
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb);
456✔
2126
    }
2127

2128
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
2,308✔
2129
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
1,396!
2130
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
1,396!
2131
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
1,396!
2132
        bool    exist = false;
1,396✔
2133
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
1,396✔
2134
        if (TSDB_CODE_SUCCESS != code) {
1,396!
2135
          mndReleaseDb(pMnode, pDb);
×
2136
          if (code != TSDB_CODE_NEED_RETRY) {
×
2137
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2138
          } else {
2139
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2140
          }
2141
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2142
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
2143
          continue;
×
2144
        }
2145
        rsp.dbTsmaVersion = pDb->tsmaVersion;
1,396✔
2146
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
1,396✔
2147
      }
2148
    }
2149

2150
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
2,308!
2151
        pDbCacheInfo->stateTs != pDb->stateTs) {
2,222✔
2152
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
1,747!
2153
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
1,747✔
2154
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
1,747!
2155
        mndReleaseDb(pMnode, pDb);
×
2156
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
2157
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2158
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2159
        continue;
×
2160
      }
2161

2162
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
1,747✔
2163
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
1,747✔
2164
      rsp.useDbRsp->uid = pDb->uid;
1,747✔
2165
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
1,747✔
2166
      rsp.useDbRsp->stateTs = pDb->stateTs;
1,747✔
2167
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
1,747✔
2168
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
1,747✔
2169
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
1,747✔
2170
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
1,747✔
2171
      rsp.useDbRsp->flags = pDb->cfg.flags;
1,747✔
2172
    }
2173

2174
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
4,616!
2175
      mndReleaseDb(pMnode, pDb);
×
2176
      if (terrno != 0) code = terrno;
×
2177
      return code;
×
2178
    }
2179
    mndReleaseDb(pMnode, pDb);
2,308✔
2180
  }
2181

2182
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
7,798✔
2183
  void   *pRsp = taosMemoryMalloc(rspLen);
7,798!
2184
  if (pRsp == NULL) {
7,798!
2185
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2186
    tFreeSDbHbBatchRsp(&batchRsp);
×
2187
    return -1;
×
2188
  }
2189
  int32_t ret = 0;
7,798✔
2190
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
7,798!
2191

2192
  *ppRsp = pRsp;
7,798✔
2193
  *pRspLen = rspLen;
7,798✔
2194

2195
  tFreeSDbHbBatchRsp(&batchRsp);
7,798✔
2196
  TAOS_RETURN(code);
7,798✔
2197
}
2198

2199
static int32_t mndSetTrimDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, int64_t startTs, STimeWindow tw,
26✔
2200
                                       SArray *vgroupIds, ETsdbOpType type, ETriggerType triggerType,
2201
                                       STrimDbRsp *pRsp) {
2202
  int32_t code = 0, lino = 0;
26✔
2203
  SSdb   *pSdb = pMnode->pSdb;
26✔
2204
  void   *pIter = NULL;
26✔
2205

2206
  SRetentionObj obj = {.optrType = type, .triggerType = triggerType};  // reuse SCompactObj struct
26✔
2207
  TAOS_CHECK_EXIT(mndAddRetentionToTrans(pMnode, pTrans, &obj, pDb, pRsp));
26!
2208

2209
  int32_t j = 0;
26✔
2210
  int32_t numOfVgroups = taosArrayGetSize(vgroupIds);
26✔
2211
  if (numOfVgroups > 0) {
26✔
2212
    for (int32_t i = 0; i < numOfVgroups; i++) {
12✔
2213
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
9✔
2214
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
9✔
2215

2216
      if (pVgroup == NULL) {
9✔
2217
        mError("db:%s, vgroup:%" PRId64 " not exist", pDb->name, vgId);
3!
2218
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
3!
2219
      } else if (pVgroup->dbUid != pDb->uid) {
6!
2220
        mError("db:%s, vgroup:%" PRId64 " not belong to db:%s", pDb->name, vgId, pDb->name);
×
2221
        sdbRelease(pSdb, pVgroup);
×
2222
        TAOS_CHECK_EXIT(TSDB_CODE_MND_VGROUP_NOT_EXIST);
×
2223
      }
2224
      sdbRelease(pSdb, pVgroup);
6✔
2225
    }
2226

2227
    for (int32_t i = 0; i < numOfVgroups; i++) {
9✔
2228
      int64_t vgId = *(int64_t *)taosArrayGet(vgroupIds, i);
6✔
2229
      SVgObj *pVgroup = mndAcquireVgroup(pMnode, vgId);
6✔
2230

2231
      if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
6!
2232
        sdbRelease(pSdb, pVgroup);
×
2233
        TAOS_RETURN(code);
×
2234
      }
2235

2236
      for (int32_t i = 0; i < pVgroup->replica; i++) {
12✔
2237
        SVnodeGid *gid = &pVgroup->vnodeGid[i];
6✔
2238
        if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
6!
2239
          sdbRelease(pSdb, pVgroup);
×
2240
          TAOS_RETURN(code);
×
2241
        }
2242
        j++;
6✔
2243
      }
2244
      sdbRelease(pSdb, pVgroup);
6✔
2245
    }
2246
  } else {
2247
    while (1) {
80✔
2248
      SVgObj *pVgroup = NULL;
100✔
2249
      pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
100✔
2250
      if (pIter == NULL) break;
100✔
2251

2252
      if (pVgroup->dbUid == pDb->uid) {
80✔
2253
        if ((code = mndBuildTrimVgroupAction(pMnode, pTrans, pDb, pVgroup, startTs, tw, type, triggerType)) != 0) {
54!
2254
          sdbCancelFetch(pSdb, pIter);
×
2255
          sdbRelease(pSdb, pVgroup);
×
2256
          TAOS_RETURN(code);
×
2257
        }
2258

2259
        for (int32_t i = 0; i < pVgroup->replica; i++) {
112✔
2260
          SVnodeGid *gid = &pVgroup->vnodeGid[i];
58✔
2261
          if ((code = mndAddRetentionDetailToTrans(pMnode, pTrans, &obj, pVgroup, gid, j)) != 0) {
58!
2262
            sdbCancelFetch(pSdb, pIter);
×
2263
            sdbRelease(pSdb, pVgroup);
×
2264
            TAOS_RETURN(code);
×
2265
          }
2266
          j++;
58✔
2267
        }
2268
      }
2269

2270
      sdbRelease(pSdb, pVgroup);
80✔
2271
    }
2272
  }
2273
_exit:
26✔
2274
  TAOS_RETURN(code);
26✔
2275
}
2276

2277
static int32_t mndBuildTrimDbRsp(STrimDbRsp *rsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
23✔
2278
  int32_t code = 0;
23✔
2279
  int32_t rspLen = tSerializeSCompactDbRsp(NULL, 0, (SCompactDbRsp *)rsp);
23✔
2280
  void   *pRsp = NULL;
23✔
2281
  if (useRpcMalloc) {
23!
2282
    pRsp = rpcMallocCont(rspLen);
×
2283
  } else {
2284
    pRsp = taosMemoryMalloc(rspLen);
23!
2285
  }
2286

2287
  if (pRsp == NULL) {
23!
2288
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2289
    TAOS_RETURN(code);
×
2290
  }
2291

2292
  (void)tSerializeSCompactDbRsp(pRsp, rspLen, (SCompactDbRsp *)rsp);
23✔
2293
  *pRspLen = rspLen;
23✔
2294
  *ppRsp = pRsp;
23✔
2295
  TAOS_RETURN(code);
23✔
2296
}
2297

2298
int32_t mndTrimDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb, STimeWindow tw, SArray *vgroupIds,
32✔
2299
                         ETsdbOpType type, ETriggerType triggerType) {
2300
  SSdb      *pSdb = pMnode->pSdb;
32✔
2301
  SVgObj    *pVgroup = NULL;
32✔
2302
  void      *pIter = NULL;
32✔
2303
  int32_t    code = 0, lino = 0;
32✔
2304
  STrimDbRsp rsp = {0};
32✔
2305
  bool       isExist = false;
32✔
2306

2307
  while (1) {
7✔
2308
    SRetentionObj *pObj = NULL;
39✔
2309
    pIter = sdbFetch(pMnode->pSdb, SDB_RETENTION, pIter, (void **)&pObj);
39✔
2310
    if (pIter == NULL) break;
39✔
2311

2312
    if (strcmp(pObj->dbname, pDb->name) == 0) {
7✔
2313
      isExist = true;
6✔
2314
    }
2315
    sdbRelease(pMnode->pSdb, pObj);
7✔
2316
  }
2317
  if (isExist) {
32✔
2318
    mInfo("trim db:%s already exist", pDb->name);
6!
2319
    TAOS_RETURN(TSDB_CODE_MND_TRIM_ALREADY_EXIST);
6✔
2320
  }
2321

2322
  int64_t startTs = taosGetTimestampMs();
26✔
2323
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "trim-db");
26✔
2324
  if (pTrans == NULL) goto _exit;
26!
2325
  mInfo("trans:%d, used to trim db:%s", pTrans->id, pDb->name);
26!
2326
  mndTransSetDbName(pTrans, pDb->name, NULL);
26✔
2327
  TAOS_CHECK_EXIT(mndTrancCheckConflict(pMnode, pTrans));
26!
2328
  TAOS_CHECK_EXIT(mndSetTrimDbRedoActions(pMnode, pTrans, pDb, startTs, tw, vgroupIds, type, triggerType, &rsp));
26✔
2329
  if (pReq) {
23!
2330
    int32_t rspLen = 0;
23✔
2331
    void   *pRsp = NULL;
23✔
2332
    rsp.bAccepted = true;
23✔
2333
    TAOS_CHECK_EXIT(mndBuildTrimDbRsp(&rsp, &rspLen, &pRsp, false));
23!
2334
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
23✔
2335
  }
2336
  TAOS_CHECK_EXIT(mndTransPrepare(pMnode, pTrans));
23!
2337
_exit:
23✔
2338
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
26!
2339
    mError("db:%s, failed at line %d to trim db since %s", pDb->name, lino, tstrerror(code));
3!
2340
  }
2341
  mndTransDrop(pTrans);
26✔
2342
  TAOS_RETURN(code);
26✔
2343
}
2344

2345
#ifdef TD_ENTERPRISE
2346
static int32_t mndCheckRsmaInDb(SMnode *pMnode, SDbObj *pDb) {
18✔
2347
  int32_t   code = 0;
18✔
2348
  void     *pIter = NULL;
18✔
2349
  SRsmaObj *pRsma = NULL;
18✔
2350
  while ((pIter = sdbFetch(pMnode->pSdb, SDB_RSMA, pIter, (void **)&pRsma))) {
18!
2351
    if (pRsma->dbUid == pDb->uid) {
18!
2352
      sdbRelease(pMnode->pSdb, pRsma);
18✔
2353
      sdbCancelFetch(pMnode->pSdb, pIter);
18✔
2354
      TAOS_RETURN(code);
18✔
2355
    }
2356
    sdbRelease(pMnode->pSdb, pRsma);
×
2357
  }
2358
  TAOS_RETURN(TSDB_CODE_RSMA_NOT_EXIST);
×
2359
}
2360
#endif
2361

2362
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
32✔
2363
  SMnode    *pMnode = pReq->info.node;
32✔
2364
  int32_t    code = 0, lino = 0;
32✔
2365
  SDbObj    *pDb = NULL;
32✔
2366
  STrimDbReq trimReq = {0};
32✔
2367

2368
  TAOS_CHECK_EXIT(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq));
32!
2369

2370
  mInfo("db:%s, start to trim, optr:%u, tw:%" PRId64 ",%" PRId64, trimReq.db, trimReq.optrType, trimReq.tw.skey,
32!
2371
        trimReq.tw.ekey);
2372

2373
  pDb = mndAcquireDb(pMnode, trimReq.db);
32✔
2374
  if (pDb == NULL) {
32!
2375
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2376
    if (terrno != 0) code = terrno;
×
2377
    TAOS_CHECK_EXIT(code);
×
2378
  }
2379

2380
  TAOS_CHECK_EXIT(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb));
32!
2381

2382
  if (pDb->cfg.isMount) {
32!
2383
    TAOS_CHECK_EXIT(TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT);
×
2384
  }
2385

2386
#ifdef TD_ENTERPRISE
2387
  if (trimReq.optrType == TSDB_OPTR_ROLLUP) {
32✔
2388
    TAOS_CHECK_EXIT(mndCheckRsmaInDb(pMnode, pDb));
18!
2389
  }
2390
#endif
2391

2392
  TAOS_CHECK_EXIT(mndTrimDb(pMnode, pReq, pDb, trimReq.tw, trimReq.vgroupIds, trimReq.optrType, trimReq.triggerType));
32✔
2393
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
23!
2394

2395
  SName name = {0};
23✔
2396
  if (tNameFromString(&name, trimReq.db, T_NAME_ACCT | T_NAME_DB) < 0) {
23!
2397
    mWarn("db:%s, failed at line %d to parse db name", trimReq.db, __LINE__);
×
2398
  }
2399

2400
  char optrType[16] = {0};
23✔
2401
  (void)snprintf(optrType, sizeof(optrType), "%u", trimReq.optrType);
23✔
2402

2403
  auditRecord(pReq, pMnode->clusterId, "trimDB", name.dbname, optrType, trimReq.sql, trimReq.sqlLen);
23✔
2404

2405
_exit:
32✔
2406
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
32!
2407
    mError("db:%s, failed at line %d to process trim db req since %s, optr:%u", trimReq.db, lino, tstrerror(code),
9!
2408
           trimReq.optrType);
2409
  }
2410

2411
  mndReleaseDb(pMnode, pDb);
32✔
2412
  tFreeSTrimDbReq(&trimReq);
32✔
2413
  TAOS_RETURN(code);
32✔
2414
}
2415

2416
static int32_t mndBuildSsMigrateDbRsp(SSsMigrateDbRsp *pSsMigrateRsp, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
×
2417
  int32_t code = 0;
×
2418
  int32_t rspLen = tSerializeSSsMigrateDbRsp(NULL, 0, pSsMigrateRsp);
×
2419
  void   *pRsp = NULL;
×
2420
  if (useRpcMalloc) {
×
2421
    pRsp = rpcMallocCont(rspLen);
×
2422
  } else {
2423
    pRsp = taosMemoryMalloc(rspLen);
×
2424
  }
2425

2426
  if (pRsp == NULL) {
×
2427
    code = TSDB_CODE_OUT_OF_MEMORY;
×
2428
    TAOS_RETURN(code);
×
2429
  }
2430

2431
  (void)tSerializeSSsMigrateDbRsp(pRsp, rspLen, pSsMigrateRsp);
×
2432
  *pRspLen = rspLen;
×
2433
  *ppRsp = pRsp;
×
2434
  TAOS_RETURN(code);
×
2435
}
2436

2437
int32_t mndSsMigrateDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
×
2438
  int32_t       code = 0;
×
2439
  SSsMigrateDbRsp ssMigrateRsp = {0};
×
2440

2441
  bool  isExist = false;
×
2442
  void *pIter = NULL;
×
2443
  while (1) {
×
2444
    SSsMigrateObj *pSsMigrate = NULL;
×
2445
    pIter = sdbFetch(pMnode->pSdb, SDB_SSMIGRATE, pIter, (void **)&pSsMigrate);
×
2446
    if (pIter == NULL) break;
×
2447

2448
    if (strcmp(pSsMigrate->dbname, pDb->name) == 0) {
×
2449
      isExist = true;
×
2450
    }
2451
    sdbRelease(pMnode->pSdb, pSsMigrate);
×
2452
  }
2453
  if (isExist) {
×
2454
    mInfo("ssmigrate db:%s already exist", pDb->name);
×
2455

2456
    if (pReq) {
×
2457
      int32_t rspLen = 0;
×
2458
      void   *pRsp = NULL;
×
2459
      ssMigrateRsp.ssMigrateId = 0;
×
2460
      ssMigrateRsp.bAccepted = false;
×
2461
      TAOS_CHECK_RETURN(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, true));
×
2462

2463
      pReq->info.rsp = pRsp;
×
2464
      pReq->info.rspLen = rspLen;
×
2465
    }
2466

2467
    return TSDB_CODE_MND_SSMIGRATE_ALREADY_EXIST;
×
2468
  }
2469

2470
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "ssmigrate-db");
×
2471
  if (pTrans == NULL) {
×
2472
    mError("failed to create ssmigrate-db trans since %s", terrstr());
×
2473
    goto _OVER;
×
2474
  }
2475

2476
  mInfo("trans:%d, used to ssmigrate db:%s", pTrans->id, pDb->name);
×
2477
  mndTransSetDbName(pTrans, pDb->name, NULL);
×
2478
  TAOS_CHECK_GOTO(mndTrancCheckConflict(pMnode, pTrans), NULL, _OVER);
×
2479

2480
  SSsMigrateObj ssMigrate = { .startTime = taosGetTimestampMs() };
×
2481
  TAOS_CHECK_GOTO(mndAddSsMigrateToTran(pMnode, pTrans, &ssMigrate, pDb), NULL, _OVER);
×
2482

2483
  if (pReq) {
×
2484
    int32_t rspLen = 0;
×
2485
    void   *pRsp = NULL;
×
2486
    ssMigrateRsp.ssMigrateId = ssMigrate.id;
×
2487
    ssMigrateRsp.bAccepted = true;
×
2488
    TAOS_CHECK_GOTO(mndBuildSsMigrateDbRsp(&ssMigrateRsp, &rspLen, &pRsp, false), NULL, _OVER);
×
2489
    mndTransSetRpcRsp(pTrans, pRsp, rspLen);
×
2490
  }
2491

2492
  if (mndTransPrepare(pMnode, pTrans) != 0) goto _OVER;
×
2493
  code = 0;
×
2494

2495
_OVER:
×
2496
  mndTransDrop(pTrans);
×
2497
  TAOS_RETURN(code);
×
2498
}
2499

2500

2501
static int32_t mndProcessSsMigrateDbReq(SRpcMsg *pReq) {
×
2502
  SMnode         *pMnode = pReq->info.node;
×
2503
  int32_t         code = -1;
×
2504
  SDbObj         *pDb = NULL;
×
2505
  SSsMigrateDbReq ssMigrateReq = {0};
×
2506

2507
  if (!tsSsEnabled) {
×
2508
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
2509
    goto _OVER;
×
2510
  }
2511

2512
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_SHARED_STORAGE), NULL, _OVER);
×
2513
  TAOS_CHECK_GOTO(tDeserializeSSsMigrateDbReq(pReq->pCont, pReq->contLen, &ssMigrateReq), NULL, _OVER);
×
2514

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

2517
  pDb = mndAcquireDb(pMnode, ssMigrateReq.db);
×
2518
  if (pDb == NULL) {
×
2519
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2520
    if (terrno != 0) code = terrno;
×
2521
    goto _OVER;
×
2522
  }
2523

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

2526
  if(pDb->cfg.isMount) {
×
2527
    code = TSDB_CODE_MND_MOUNT_OBJ_NOT_SUPPORT;
×
2528
    goto _OVER;
×
2529
  }
2530

2531
  code = mndSsMigrateDb(pMnode, pReq, pDb);
×
2532

2533
_OVER:
×
2534
  if (code != 0) {
×
2535
    mError("db:%s, failed to process ssmigrate db req since %s", ssMigrateReq.db, tstrerror(code));
×
2536
  }
2537

2538
  mndReleaseDb(pMnode, pDb);
×
2539
  TAOS_RETURN(code);
×
2540
}
2541

2542
const char *mndGetDbStr(const char *src) {
34,108✔
2543
  char *pos = strstr(src, TS_PATH_DELIMITER);
34,108✔
2544
  if (pos != NULL) ++pos;
34,108✔
2545
  if (pos == NULL) return src;
34,108✔
2546
  return pos;
30,746✔
2547
}
2548

2549
const char *mndGetStableStr(const char *src) {
26,563✔
2550
  char *pos = strstr(src, TS_PATH_DELIMITER);
26,563✔
2551
  if (pos != NULL) ++pos;
26,563!
2552
  if (pos == NULL) return src;
26,563!
2553
  return mndGetDbStr(pos);
26,563✔
2554
}
2555

2556
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
×
2557
  int64_t v = 0;
×
2558
  switch (unit) {
×
2559
    case 's':
×
2560
      v = val / 1000;
×
2561
      break;
×
2562
    case 'm':
×
2563
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
×
2564
      break;
×
2565
    case 'h':
×
2566
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2567
      break;
×
2568
    case 'd':
×
2569
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
×
2570
      break;
×
2571
    case 'w':
×
2572
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2573
      break;
×
2574
    default:
×
2575
      break;
×
2576
  }
2577

2578
  return v;
×
2579
}
2580

2581
static char *buildRetension(SArray *pRetension) {
1,302✔
2582
  size_t size = taosArrayGetSize(pRetension);
1,302✔
2583
  if (size == 0) {
1,302!
2584
    return NULL;
1,302✔
2585
  }
2586

2587
  char       *p1 = taosMemoryCalloc(1, 100);
×
2588
  SRetention *p = taosArrayGet(pRetension, 0);
×
2589

2590
  int32_t len = 2;
×
2591

2592
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2593
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2594
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2595

2596
  if (size > 1) {
×
2597
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2598
    p = taosArrayGet(pRetension, 1);
×
2599

2600
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2601
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2602
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2603
  }
2604

2605
  if (size > 2) {
×
2606
    len += tsnprintf(p1 + len, 100 - len, ",");
×
2607
    p = taosArrayGet(pRetension, 2);
×
2608

2609
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
×
2610
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
×
2611
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
×
2612
  }
2613

2614
  varDataSetLen(p1, len);
×
2615
  return p1;
×
2616
}
2617

2618
static const char *getCacheModelStr(int8_t cacheModel) {
1,302✔
2619
  switch (cacheModel) {
1,302!
2620
    case TSDB_CACHE_MODEL_NONE:
1,262✔
2621
      return TSDB_CACHE_MODEL_NONE_STR;
1,262✔
2622
    case TSDB_CACHE_MODEL_LAST_ROW:
7✔
2623
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
7✔
2624
    case TSDB_CACHE_MODEL_LAST_VALUE:
12✔
2625
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
12✔
2626
    case TSDB_CACHE_MODEL_BOTH:
21✔
2627
      return TSDB_CACHE_MODEL_BOTH_STR;
21✔
2628
    default:
×
2629
      break;
×
2630
  }
2631
  return "unknown";
×
2632
}
2633

2634
static const char *getEncryptAlgorithmStr(int8_t encryptAlgorithm) {
1,302✔
2635
  switch (encryptAlgorithm) {
1,302!
2636
    case TSDB_ENCRYPT_ALGO_NONE:
1,302✔
2637
      return TSDB_ENCRYPT_ALGO_NONE_STR;
1,302✔
2638
    case TSDB_ENCRYPT_ALGO_SM4:
×
2639
      return TSDB_ENCRYPT_ALGO_SM4_STR;
×
2640
    default:
×
2641
      break;
×
2642
  }
2643
  return "unknown";
×
2644
}
2645

2646
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
4,888✔
2647
  if (pDb->cfg.replications == 1) return true;
4,888✔
2648

2649
  SSdb *pSdb = pMnode->pSdb;
1,024✔
2650
  void *pIter = NULL;
1,024✔
2651
  bool  isReady = true;
1,024✔
2652
  while (1) {
7,480✔
2653
    SVgObj *pVgroup = NULL;
8,504✔
2654
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
8,504✔
2655
    if (pIter == NULL) break;
8,504✔
2656

2657
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
7,480!
2658
      bool hasLeader = false;
3,079✔
2659
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
12,237✔
2660
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
9,158✔
2661
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
8,199!
2662
          hasLeader = true;
959✔
2663
        }
2664
      }
2665
      if (!hasLeader) isReady = false;
3,079✔
2666
    }
2667
    sdbRelease(pSdb, pVgroup);
7,480✔
2668
  }
2669

2670
  return isReady;
1,024✔
2671
}
2672

2673
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
3,896✔
2674
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2675
  int32_t cols = 0;
3,896✔
2676
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
3,896✔
2677
  char   *buf = taosMemoryMalloc(bytes);
3,896!
2678
  if (buf == NULL) {
3,896!
2679
    mError("db:%s, failed to malloc buffer", pDb->name);
×
2680
    return;
×
2681
  }
2682
  int32_t code = 0;
3,896✔
2683
  int32_t lino = 0;
3,896✔
2684

2685
  const char *name = mndGetDbStr(pDb->name);
3,896✔
2686
  if (name != NULL) {
3,896!
2687
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
3,896✔
2688
  } else {
2689
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
×
2690
  }
2691

2692
  const char *precStr = NULL;
3,896✔
2693
  switch (pDb->cfg.precision) {
3,896!
2694
    case TSDB_TIME_PRECISION_MILLI:
3,808✔
2695
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
3,808✔
2696
      break;
3,808✔
2697
    case TSDB_TIME_PRECISION_MICRO:
42✔
2698
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
42✔
2699
      break;
42✔
2700
    case TSDB_TIME_PRECISION_NANO:
46✔
2701
      precStr = TSDB_TIME_PRECISION_NANO_STR;
46✔
2702
      break;
46✔
2703
    default:
×
2704
      precStr = "none";
×
2705
      break;
×
2706
  }
2707
  char precVstr[10] = {0};
3,896✔
2708
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
3,896✔
2709

2710
  char *statusStr = "ready";
3,896✔
2711
  if (objStatus == SDB_STATUS_CREATING) {
3,896!
2712
    statusStr = "creating";
×
2713
  } else if (objStatus == SDB_STATUS_DROPPING) {
3,896!
2714
    statusStr = "dropping";
×
2715
  } else {
2716
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
3,896✔
2717
      statusStr = "unsynced";
9✔
2718
    }
2719
  }
2720
  char statusVstr[24] = {0};
3,896✔
2721
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
3,896✔
2722

2723
  if (sysDb || !sysinfo) {
3,896✔
2724
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
98,572✔
2725
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
95,978✔
2726
      if (i == 0) {
95,978✔
2727
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
2,594!
2728
      } else if (i == 1) {
93,384✔
2729
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
2,594!
2730
      } else if (i == 3) {
90,790✔
2731
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
2,594!
2732
      } else if (i == 14) {
88,196✔
2733
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
2,594!
2734
      } else if (i == 15) {
85,602✔
2735
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
2,594!
2736
      } else {
2737
        colDataSetNULL(pColInfo, rows);
83,008✔
2738
      }
2739
    }
2740
  } else {
2741
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2742
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
1,302!
2743

2744
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2745
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
1,302!
2746

2747
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2748
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
1,302!
2749

2750
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2751
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
1,302!
2752

2753
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2754
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
1,302!
2755

2756
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
1,302!
2757
    char        strictVstr[24] = {0};
1,302✔
2758
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
1,302✔
2759
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2760
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
1,302!
2761

2762
    char    durationStr[128] = {0};
1,302✔
2763
    char    durationVstr[128] = {0};
1,302✔
2764
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
1,302✔
2765
                                       pDb->cfg.daysPerFile);
2766

2767
    varDataSetLen(durationVstr, len);
1,302✔
2768
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2769
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,302!
2770

2771
    char keepVstr[128] = {0};
1,302✔
2772
    char keep0Str[32] = {0};
1,302✔
2773
    char keep1Str[32] = {0};
1,302✔
2774
    char keep2Str[32] = {0};
1,302✔
2775

2776
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
1,302✔
2777
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
1,302✔
2778
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
1,302✔
2779

2780
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
1,302!
2781
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
×
2782
    } else {
2783
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
1,302✔
2784
    }
2785
    varDataSetLen(keepVstr, len);
1,302✔
2786
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2787
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
1,302!
2788

2789
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2790
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
1,302!
2791

2792
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2793
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
1,302!
2794

2795
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2796
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
1,302!
2797

2798
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2799
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
1,302!
2800

2801
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2802
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
1,302!
2803

2804
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2805
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
1,302!
2806

2807
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2808
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
1,302!
2809

2810
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2811
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
1,302!
2812

2813
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
1,302✔
2814
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2815
    if (rentensionVstr == NULL) {
1,302!
2816
      colDataSetNULL(pColInfo, rows);
1,302!
2817
    } else {
2818
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
×
2819
      taosMemoryFree(rentensionVstr);
×
2820
    }
2821

2822
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2823
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
1,302!
2824

2825
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
1,302✔
2826
    char        cacheModelVstr[24] = {0};
1,302✔
2827
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
1,302✔
2828
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2829
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
1,302!
2830

2831
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2832
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
1,302!
2833

2834
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2835
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
1,302!
2836

2837
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2838
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
1,302!
2839

2840
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2841
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
1,302!
2842

2843
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2844
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
1,302!
2845

2846
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2847
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
1,302!
2848

2849
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2850
    int16_t hashPrefix = pDb->cfg.hashPrefix;
1,302✔
2851
    if (hashPrefix > 0) {
1,302✔
2852
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
7✔
2853
    } else if (hashPrefix < 0) {
1,295✔
2854
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
8✔
2855
    }
2856
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
1,302!
2857

2858
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2859
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
1,302!
2860

2861
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2862
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
1,302!
2863

2864
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2865
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
1,302!
2866

2867
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2868
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssChunkSize, false), &lino, _OVER);
1,302!
2869

2870
    char keeplocalVstr[128] = {0};
1,302✔
2871
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.ssKeepLocal);
1,302✔
2872
    varDataSetLen(keeplocalVstr, len);
1,302✔
2873
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2874
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
1,302!
2875

2876
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2877
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.ssCompact, false), &lino, _OVER);
1,302!
2878

2879
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2880
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
1,302!
2881

2882
    const char *encryptAlgorithmStr = getEncryptAlgorithmStr(pDb->cfg.encryptAlgorithm);
1,302✔
2883
    char        encryptAlgorithmVStr[24] = {0};
1,302✔
2884
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
1,302✔
2885
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,302✔
2886
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
1,302!
2887

2888
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
1,302✔
2889
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,302✔
2890
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,302!
2891
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,302!
2892
    }
2893

2894
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
1,302✔
2895
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
1,302✔
2896
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
1,302✔
2897
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,302✔
2898
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,302!
2899
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,302!
2900
    }
2901

2902
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
1,302✔
2903
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
1,302✔
2904
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
1,302!
2905
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
1,302!
2906
    }
2907
  }
2908
_OVER:
3,896✔
2909
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
3,896!
2910
  taosMemoryFree(buf);
3,896!
2911
}
2912

2913
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
1,291✔
2914
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
1,291✔
2915
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
1,291✔
2916
  pDbObj->cfg.numOfVgroups = 0;
1,291✔
2917
  pDbObj->cfg.strict = 1;
1,291✔
2918
  pDbObj->cfg.replications = 1;
1,291✔
2919
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
1,291✔
2920
}
1,291✔
2921

2922
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
1,291✔
2923
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
1,291✔
2924
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
1,291✔
2925
  pDbObj->cfg.numOfVgroups = 0;
1,291✔
2926
  pDbObj->cfg.strict = 1;
1,291✔
2927
  pDbObj->cfg.replications = 1;
1,291✔
2928
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
1,291✔
2929
}
1,291✔
2930

2931
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
6,670✔
2932
  SVgObj  *pVgroup = pObj;
6,670✔
2933
  int32_t *numOfTables = p1;
6,670✔
2934
  int64_t  uid = *(int64_t *)p2;
6,670✔
2935
  if (pVgroup->dbUid == uid) {
6,670✔
2936
    *numOfTables += pVgroup->numOfTables;
2,615✔
2937
  }
2938
  return true;
6,670✔
2939
}
2940

2941
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
1,291✔
2942
  SMnode    *pMnode = pReq->info.node;
1,291✔
2943
  SSdb      *pSdb = pMnode->pSdb;
1,291✔
2944
  int32_t    numOfRows = 0;
1,291✔
2945
  SDbObj    *pDb = NULL;
1,291✔
2946
  SUserObj  *pUser = NULL;
1,291✔
2947
  ESdbStatus objStatus = 0;
1,291✔
2948

2949
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
1,291✔
2950
  if (pUser == NULL) return 0;
1,291!
2951
  bool sysinfo = pUser->sysInfo;
1,291✔
2952

2953
  // Append the information_schema database into the result.
2954
  if (!pShow->sysDbRsp) {
1,291!
2955
    SDbObj infoschemaDb = {0};
1,291✔
2956
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
1,291✔
2957
    size_t numOfTables = 0;
1,291✔
2958
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
1,291✔
2959
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
1,291✔
2960

2961
    numOfRows += 1;
1,291✔
2962

2963
    SDbObj perfschemaDb = {0};
1,291✔
2964
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
1,291✔
2965
    numOfTables = 0;
1,291✔
2966
    getPerfDbMeta(NULL, &numOfTables);
1,291✔
2967
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
1,291✔
2968

2969
    numOfRows += 1;
1,291✔
2970
    pShow->sysDbRsp = true;
1,291✔
2971
  }
2972

2973
  while (numOfRows < rowsCapacity) {
2,637!
2974
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
2,637✔
2975
    if (pShow->pIter == NULL) break;
2,637✔
2976

2977
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
1,346✔
2978
      int32_t numOfTables = 0;
1,314✔
2979
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
1,314✔
2980
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
1,314✔
2981
      numOfRows++;
1,314✔
2982
    }
2983

2984
    sdbRelease(pSdb, pDb);
1,346✔
2985
  }
2986

2987
  pShow->numOfRows += numOfRows;
1,291✔
2988
  mndReleaseUser(pMnode, pUser);
1,291✔
2989
  return numOfRows;
1,291✔
2990
}
2991

2992
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
2993
  SSdb *pSdb = pMnode->pSdb;
×
2994
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
2995
}
×
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