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

taosdata / TDengine / #3858

17 Apr 2025 01:40PM UTC coverage: 62.968% (+0.5%) from 62.513%
#3858

push

travis-ci

web-flow
docs(opc): add perssit data support (#30783)

156194 of 316378 branches covered (49.37%)

Branch coverage included in aggregate %.

242021 of 316027 relevant lines covered (76.58%)

19473613.85 hits per line

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

67.99
/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 "audit.h"
18
#include "command.h"
19
#include "mndArbGroup.h"
20
#include "mndCluster.h"
21
#include "mndConfig.h"
22
#include "mndDb.h"
23
#include "mndDnode.h"
24
#include "mndIndex.h"
25
#include "mndPrivilege.h"
26
#include "mndShow.h"
27
#include "mndSma.h"
28
#include "mndStb.h"
29
#include "mndStream.h"
30
#include "mndSubscribe.h"
31
#include "mndTopic.h"
32
#include "mndTrans.h"
33
#include "mndUser.h"
34
#include "mndVgroup.h"
35
#include "mndView.h"
36
#include "systable.h"
37
#include "thttp.h"
38
#include "tjson.h"
39

40
#define DB_VER_NUMBER   1
41
#define DB_RESERVE_SIZE 14
42

43
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw);
44
static int32_t  mndDbActionInsert(SSdb *pSdb, SDbObj *pDb);
45
static int32_t  mndDbActionDelete(SSdb *pSdb, SDbObj *pDb);
46
static int32_t  mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew);
47
static int32_t  mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw);
48

49
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq);
50
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq);
51
static int32_t mndProcessDropDbReq(SRpcMsg *pReq);
52
static int32_t mndProcessUseDbReq(SRpcMsg *pReq);
53
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq);
54
static int32_t mndProcessS3MigrateDbReq(SRpcMsg *pReq);
55
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity);
56
static void    mndCancelGetNextDb(SMnode *pMnode, void *pIter);
57
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq);
58

59
#ifndef TD_ENTERPRISE
60
int32_t mndProcessCompactDbReq(SRpcMsg *pReq) { return TSDB_CODE_OPS_NOT_SUPPORT; }
61
#endif
62

63
int32_t mndInitDb(SMnode *pMnode) {
2,011✔
64
  SSdbTable table = {
2,011✔
65
      .sdbType = SDB_DB,
66
      .keyType = SDB_KEY_BINARY,
67
      .encodeFp = (SdbEncodeFp)mndDbActionEncode,
68
      .decodeFp = (SdbDecodeFp)mndDbActionDecode,
69
      .insertFp = (SdbInsertFp)mndDbActionInsert,
70
      .updateFp = (SdbUpdateFp)mndDbActionUpdate,
71
      .deleteFp = (SdbDeleteFp)mndDbActionDelete,
72
      .validateFp = (SdbValidateFp)mndNewDbActionValidate,
73
  };
74

75
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_DB, mndProcessCreateDbReq);
2,011✔
76
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_DB, mndProcessAlterDbReq);
2,011✔
77
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_DB, mndProcessDropDbReq);
2,011✔
78
  mndSetMsgHandle(pMnode, TDMT_MND_USE_DB, mndProcessUseDbReq);
2,011✔
79
  mndSetMsgHandle(pMnode, TDMT_MND_COMPACT_DB, mndProcessCompactDbReq);
2,011✔
80
  mndSetMsgHandle(pMnode, TDMT_MND_TRIM_DB, mndProcessTrimDbReq);
2,011✔
81
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_CFG, mndProcessGetDbCfgReq);
2,011✔
82
  mndSetMsgHandle(pMnode, TDMT_MND_S3MIGRATE_DB, mndProcessS3MigrateDbReq);
2,011✔
83
  mndSetMsgHandle(pMnode, TDMT_MND_GET_DB_INFO, mndProcessUseDbReq);
2,011✔
84

85
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_DB, mndRetrieveDbs);
2,011✔
86
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_DB, mndCancelGetNextDb);
2,011✔
87

88
  return sdbSetTable(pMnode->pSdb, table);
2,011✔
89
}
90

91
void mndCleanupDb(SMnode *pMnode) {}
2,010✔
92

93
SSdbRaw *mndDbActionEncode(SDbObj *pDb) {
23,176✔
94
  int32_t code = 0;
23,176✔
95
  int32_t lino = 0;
23,176✔
96
  terrno = TSDB_CODE_OUT_OF_MEMORY;
23,176✔
97

98
  int32_t  size = sizeof(SDbObj) + pDb->cfg.numOfRetensions * sizeof(SRetention) + DB_RESERVE_SIZE;
23,176✔
99
  SSdbRaw *pRaw = sdbAllocRaw(SDB_DB, DB_VER_NUMBER, size);
23,176✔
100
  if (pRaw == NULL) goto _OVER;
23,176!
101

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

161
  SDB_SET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
23,176!
162
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
23,176!
163

164
  terrno = 0;
23,176✔
165

166
_OVER:
23,176✔
167
  if (terrno != 0) {
23,176!
168
    mError("db:%s, failed to encode to raw:%p since %s", pDb->name, pRaw, terrstr());
×
169
    sdbFreeRaw(pRaw);
×
170
    return NULL;
×
171
  }
172

173
  mTrace("db:%s, encode to raw:%p, row:%p", pDb->name, pRaw, pDb);
23,176✔
174
  return pRaw;
23,176✔
175
}
176

177
static SSdbRow *mndDbActionDecode(SSdbRaw *pRaw) {
19,501✔
178
  int32_t code = 0;
19,501✔
179
  int32_t lino = 0;
19,501✔
180
  terrno = TSDB_CODE_OUT_OF_MEMORY;
19,501✔
181
  SSdbRow *pRow = NULL;
19,501✔
182
  SDbObj  *pDb = NULL;
19,501✔
183

184
  int8_t sver = 0;
19,501✔
185
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
19,501!
186

187
  if (sver != DB_VER_NUMBER) {
19,501!
188
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
189
    goto _OVER;
×
190
  }
191

192
  pRow = sdbAllocRow(sizeof(SDbObj));
19,501✔
193
  if (pRow == NULL) goto _OVER;
19,501!
194

195
  pDb = sdbGetRowObj(pRow);
19,501✔
196
  if (pDb == NULL) goto _OVER;
19,501!
197

198
  int32_t dataPos = 0;
19,501✔
199
  SDB_GET_BINARY(pRaw, dataPos, pDb->name, TSDB_DB_FNAME_LEN, _OVER)
19,501!
200
  SDB_GET_BINARY(pRaw, dataPos, pDb->acct, TSDB_USER_LEN, _OVER)
19,501!
201
  SDB_GET_BINARY(pRaw, dataPos, pDb->createUser, TSDB_USER_LEN, _OVER)
19,501!
202
  SDB_GET_INT64(pRaw, dataPos, &pDb->createdTime, _OVER)
19,501!
203
  SDB_GET_INT64(pRaw, dataPos, &pDb->updateTime, _OVER)
19,501!
204
  SDB_GET_INT64(pRaw, dataPos, &pDb->uid, _OVER)
19,501!
205
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfgVersion, _OVER)
19,501!
206
  SDB_GET_INT32(pRaw, dataPos, &pDb->vgVersion, _OVER)
19,501!
207
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfVgroups, _OVER)
19,501!
208
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfStables, _OVER)
19,501!
209
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.buffer, _OVER)
19,501!
210
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.pageSize, _OVER)
19,501!
211
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.pages, _OVER)
19,501!
212
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.cacheLastSize, _OVER)
19,501!
213
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysPerFile, _OVER)
19,501!
214
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep0, _OVER)
19,501!
215
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep1, _OVER)
19,501!
216
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.daysToKeep2, _OVER)
19,501!
217
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.minRows, _OVER)
19,501!
218
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.maxRows, _OVER)
19,501!
219
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walFsyncPeriod, _OVER)
19,501!
220
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.walLevel, _OVER)
19,501!
221
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.precision, _OVER)
19,501!
222
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.compression, _OVER)
19,501!
223
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.replications, _OVER)
19,501!
224
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.strict, _OVER)
19,501!
225
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.cacheLast, _OVER)
19,501!
226
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.hashMethod, _OVER)
19,501!
227
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.numOfRetensions, _OVER)
19,501!
228
  if (pDb->cfg.numOfRetensions > 0) {
19,501✔
229
    pDb->cfg.pRetensions = taosArrayInit(pDb->cfg.numOfRetensions, sizeof(SRetention));
16✔
230
    if (pDb->cfg.pRetensions == NULL) goto _OVER;
16!
231
    for (int32_t i = 0; i < pDb->cfg.numOfRetensions; ++i) {
64✔
232
      SRetention retention = {0};
48✔
233
      SDB_GET_INT64(pRaw, dataPos, &retention.freq, _OVER)
48!
234
      SDB_GET_INT64(pRaw, dataPos, &retention.keep, _OVER)
48!
235
      SDB_GET_INT8(pRaw, dataPos, &retention.freqUnit, _OVER)
48!
236
      SDB_GET_INT8(pRaw, dataPos, &retention.keepUnit, _OVER)
48!
237
      if (taosArrayPush(pDb->cfg.pRetensions, &retention) == NULL) {
96!
238
        goto _OVER;
×
239
      }
240
    }
241
  }
242
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.schemaless, _OVER)
19,501!
243
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walRetentionPeriod, _OVER)
19,501!
244
  SDB_GET_INT64(pRaw, dataPos, &pDb->cfg.walRetentionSize, _OVER)
19,501!
245
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.walRollPeriod, _OVER)
19,501!
246
  SDB_GET_INT64(pRaw, dataPos, &pDb->cfg.walSegmentSize, _OVER)
19,501!
247
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.sstTrigger, _OVER)
19,501!
248
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.hashPrefix, _OVER)
19,501!
249
  SDB_GET_INT16(pRaw, dataPos, &pDb->cfg.hashSuffix, _OVER)
19,501!
250
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.tsdbPageSize, _OVER)
19,501!
251
  SDB_GET_INT64(pRaw, dataPos, &pDb->compactStartTime, _OVER)
19,501!
252
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.keepTimeOffset, _OVER)
19,501!
253
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.s3ChunkSize, _OVER)
19,501!
254
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.s3KeepLocal, _OVER)
19,501!
255
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.s3Compact, _OVER)
19,501!
256
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.withArbitrator, _OVER)
19,501!
257
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.encryptAlgorithm, _OVER)
19,501!
258
  SDB_GET_INT32(pRaw, dataPos, &pDb->tsmaVersion, _OVER);
19,501!
259
  SDB_GET_INT8(pRaw, dataPos, &pDb->cfg.compactTimeOffset, _OVER)
19,501!
260
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.compactStartTime, _OVER)
19,501!
261
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.compactEndTime, _OVER)
19,501!
262
  SDB_GET_INT32(pRaw, dataPos, &pDb->cfg.compactInterval, _OVER)
19,501!
263

264
  SDB_GET_RESERVE(pRaw, dataPos, DB_RESERVE_SIZE, _OVER)
19,501!
265
  taosInitRWLatch(&pDb->lock);
19,501✔
266

267
  if (pDb->cfg.s3ChunkSize == 0) {
19,501!
268
    pDb->cfg.s3ChunkSize = TSDB_DEFAULT_S3_CHUNK_SIZE;
×
269

270
    mInfo("db:%s, s3ChunkSize set from %d to default %d", pDb->name, pDb->cfg.s3ChunkSize, TSDB_DEFAULT_S3_CHUNK_SIZE);
×
271
  }
272

273
  if (pDb->cfg.s3KeepLocal == 0) {
19,501!
274
    pDb->cfg.s3KeepLocal = TSDB_DEFAULT_S3_KEEP_LOCAL;
×
275

276
    mInfo("db:%s, s3KeepLocal set from %d to default %d", pDb->name, pDb->cfg.s3KeepLocal, TSDB_DEFAULT_S3_KEEP_LOCAL);
×
277
  }
278

279
  if (pDb->cfg.tsdbPageSize != TSDB_MIN_TSDB_PAGESIZE) {
19,501!
280
    mInfo("db:%s, tsdbPageSize set from %d to default %d", pDb->name, pDb->cfg.tsdbPageSize,
19,501!
281
          TSDB_DEFAULT_TSDB_PAGESIZE);
282
  }
283

284
  if (pDb->cfg.sstTrigger != TSDB_MIN_STT_TRIGGER) {
19,501✔
285
    mInfo("db:%s, sstTrigger set from %d to default %d", pDb->name, pDb->cfg.sstTrigger, TSDB_DEFAULT_SST_TRIGGER);
17,708!
286
  }
287

288
  terrno = 0;
19,501✔
289

290
_OVER:
19,501✔
291
  if (terrno != 0) {
19,501!
292
    mError("db:%s, failed to decode from raw:%p since %s", pDb == NULL ? "null" : pDb->name, pRaw, terrstr());
×
293
    taosMemoryFreeClear(pRow);
×
294
    return NULL;
×
295
  }
296

297
  mTrace("db:%s, decode from raw:%p, row:%p", pDb->name, pRaw, pDb);
19,501✔
298
  return pRow;
19,501✔
299
}
300

301
static int32_t mndNewDbActionValidate(SMnode *pMnode, STrans *pTrans, SSdbRaw *pRaw) {
4,548✔
302
  SSdb    *pSdb = pMnode->pSdb;
4,548✔
303
  SSdbRow *pRow = NULL;
4,548✔
304
  SDbObj  *pNewDb = NULL;
4,548✔
305
  int      code = -1;
4,548✔
306

307
  pRow = mndDbActionDecode(pRaw);
4,548✔
308
  if (pRow == NULL) {
4,548!
309
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
310
    if (terrno != 0) code = terrno;
×
311
    goto _OVER;
×
312
  }
313
  pNewDb = sdbGetRowObj(pRow);
4,548✔
314
  if (pNewDb == NULL) {
4,548!
315
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
316
    if (terrno != 0) code = terrno;
×
317
    goto _OVER;
×
318
  }
319

320
  SDbObj *pOldDb = sdbAcquire(pMnode->pSdb, SDB_DB, pNewDb->name);
4,548✔
321
  if (pOldDb != NULL) {
4,548!
322
    mError("trans:%d, db name already in use. name: %s", pTrans->id, pNewDb->name);
×
323
    sdbRelease(pMnode->pSdb, pOldDb);
×
324
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
325
    if (terrno != 0) code = terrno;
×
326
    goto _OVER;
×
327
  }
328

329
  code = 0;
4,548✔
330
_OVER:
4,548✔
331
  if (pNewDb) mndDbActionDelete(pSdb, pNewDb);
4,548!
332
  taosMemoryFreeClear(pRow);
4,548!
333
  return code;
4,548✔
334
}
335

336
static int32_t mndDbActionInsert(SSdb *pSdb, SDbObj *pDb) {
4,972✔
337
  mTrace("db:%s, perform insert action, row:%p", pDb->name, pDb);
4,972✔
338
  return 0;
4,972✔
339
}
340

341
static int32_t mndDbActionDelete(SSdb *pSdb, SDbObj *pDb) {
19,501✔
342
  mTrace("db:%s, perform delete action, row:%p", pDb->name, pDb);
19,501✔
343
  taosArrayDestroy(pDb->cfg.pRetensions);
19,501✔
344
  return 0;
19,501✔
345
}
346

347
static int32_t mndDbActionUpdate(SSdb *pSdb, SDbObj *pOld, SDbObj *pNew) {
7,988✔
348
  mTrace("db:%s, perform update action, old row:%p new row:%p", pOld->name, pOld, pNew);
7,988✔
349
  taosWLockLatch(&pOld->lock);
7,988✔
350
  pOld->updateTime = pNew->updateTime;
7,988✔
351
  pOld->cfgVersion = pNew->cfgVersion;
7,988✔
352
  pOld->vgVersion = pNew->vgVersion;
7,988✔
353
  pOld->cfg.numOfVgroups = pNew->cfg.numOfVgroups;
7,988✔
354
  pOld->cfg.buffer = pNew->cfg.buffer;
7,988✔
355
  pOld->cfg.pageSize = pNew->cfg.pageSize;
7,988✔
356
  pOld->cfg.pages = pNew->cfg.pages;
7,988✔
357
  pOld->cfg.cacheLastSize = pNew->cfg.cacheLastSize;
7,988✔
358
  pOld->cfg.daysPerFile = pNew->cfg.daysPerFile;
7,988✔
359
  pOld->cfg.daysToKeep0 = pNew->cfg.daysToKeep0;
7,988✔
360
  pOld->cfg.daysToKeep1 = pNew->cfg.daysToKeep1;
7,988✔
361
  pOld->cfg.daysToKeep2 = pNew->cfg.daysToKeep2;
7,988✔
362
  pOld->cfg.keepTimeOffset = pNew->cfg.keepTimeOffset;
7,988✔
363
  pOld->cfg.walFsyncPeriod = pNew->cfg.walFsyncPeriod;
7,988✔
364
  pOld->cfg.walLevel = pNew->cfg.walLevel;
7,988✔
365
  pOld->cfg.walRetentionPeriod = pNew->cfg.walRetentionPeriod;
7,988✔
366
  pOld->cfg.walRetentionSize = pNew->cfg.walRetentionSize;
7,988✔
367
  pOld->cfg.strict = pNew->cfg.strict;
7,988✔
368
  pOld->cfg.cacheLast = pNew->cfg.cacheLast;
7,988✔
369
  pOld->cfg.replications = pNew->cfg.replications;
7,988✔
370
  pOld->cfg.sstTrigger = pNew->cfg.sstTrigger;
7,988✔
371
  pOld->cfg.minRows = pNew->cfg.minRows;
7,988✔
372
  pOld->cfg.maxRows = pNew->cfg.maxRows;
7,988✔
373
  pOld->cfg.tsdbPageSize = pNew->cfg.tsdbPageSize;
7,988✔
374
  pOld->cfg.s3ChunkSize = pNew->cfg.s3ChunkSize;
7,988✔
375
  pOld->cfg.s3KeepLocal = pNew->cfg.s3KeepLocal;
7,988✔
376
  pOld->cfg.s3Compact = pNew->cfg.s3Compact;
7,988✔
377
  pOld->cfg.withArbitrator = pNew->cfg.withArbitrator;
7,988✔
378
  pOld->cfg.compactInterval = pNew->cfg.compactInterval;
7,988✔
379
  pOld->cfg.compactStartTime = pNew->cfg.compactStartTime;
7,988✔
380
  pOld->cfg.compactEndTime = pNew->cfg.compactEndTime;
7,988✔
381
  pOld->cfg.compactTimeOffset = pNew->cfg.compactTimeOffset;
7,988✔
382
  pOld->compactStartTime = pNew->compactStartTime;
7,988✔
383
  pOld->tsmaVersion = pNew->tsmaVersion;
7,988✔
384
  taosWUnLockLatch(&pOld->lock);
7,988✔
385
  return 0;
7,988✔
386
}
387

388
static inline int32_t mndGetGlobalVgroupVersion(SMnode *pMnode) {
301,298✔
389
  SSdb *pSdb = pMnode->pSdb;
301,298✔
390
  return sdbGetTableVer(pSdb, SDB_VGROUP);
301,298✔
391
}
392

393
SDbObj *mndAcquireDb(SMnode *pMnode, const char *db) {
4,383,596✔
394
  SSdb   *pSdb = pMnode->pSdb;
4,383,596✔
395
  SDbObj *pDb = sdbAcquire(pSdb, SDB_DB, db);
4,383,596✔
396
  if (pDb == NULL) {
4,383,634✔
397
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
9,971✔
398
      terrno = TSDB_CODE_MND_DB_NOT_EXIST;
8,060✔
399
    } else if (terrno == TSDB_CODE_SDB_OBJ_CREATING) {
1,911✔
400
      terrno = TSDB_CODE_MND_DB_IN_CREATING;
98✔
401
    } else if (terrno == TSDB_CODE_SDB_OBJ_DROPPING) {
1,813!
402
      terrno = TSDB_CODE_MND_DB_IN_DROPPING;
1,813✔
403
    } else {
404
      terrno = TSDB_CODE_APP_ERROR;
×
405
      mFatal("db:%s, failed to acquire db since %s", db, terrstr());
×
406
    }
407
  }
408
  return pDb;
4,383,577✔
409
}
410

411
void mndReleaseDb(SMnode *pMnode, SDbObj *pDb) {
4,790,362✔
412
  SSdb *pSdb = pMnode->pSdb;
4,790,362✔
413
  sdbRelease(pSdb, pDb);
4,790,362✔
414
}
4,790,493✔
415

416
bool mndDbIsExist(SMnode *pMnode, const char *db) {
13✔
417
  SDbObj *pDb = mndAcquireDb(pMnode, db);
13✔
418
  if (pDb == NULL) {
13!
419
    return false;
×
420
  } else {
421
    mndReleaseDb(pMnode, pDb);
13✔
422
    pDb = NULL;
13✔
423
    return true;
13✔
424
  }
425
}
426

427
static int32_t mndCheckDbName(const char *dbName, SUserObj *pUser) {
4,833✔
428
  char *pos = strstr(dbName, TS_PATH_DELIMITER);
4,833✔
429
  if (pos == NULL) {
4,833!
430
    return TSDB_CODE_MND_INVALID_DB;
×
431
  }
432

433
  int32_t acctId;
434
  int32_t code = taosStr2int32(dbName, &acctId);
4,833✔
435
  if (code != 0) {
4,833!
436
    return code;
×
437
  }
438

439
  if (acctId != pUser->acctId) {
4,833!
440
    return TSDB_CODE_MND_INVALID_DB_ACCT;
×
441
  }
442

443
  return 0;
4,833✔
444
}
445

446
static int32_t mndCheckDbCfg(SMnode *pMnode, SDbCfg *pCfg) {
4,833✔
447
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
4,833✔
448

449
  if (pCfg->numOfVgroups < TSDB_MIN_VNODES_PER_DB || pCfg->numOfVgroups > TSDB_MAX_VNODES_PER_DB) return code;
4,833!
450
  if (pCfg->numOfStables < TSDB_DB_STREAM_MODE_OFF || pCfg->numOfStables > TSDB_DB_STREAM_MODE_ON) return code;
4,833!
451
  if (pCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
4,833!
452
  if (pCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
4,833!
453
  if (pCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
4,833!
454
  if (pCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
4,833!
455
  if (pCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
4,833!
456
  if (pCfg->daysToKeep0 < TSDB_MIN_KEEP || pCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
4,833!
457
  if (pCfg->daysToKeep1 < TSDB_MIN_KEEP || pCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
4,833!
458
  if (pCfg->daysToKeep2 < TSDB_MIN_KEEP || pCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
4,833!
459
  if (pCfg->daysToKeep0 < pCfg->daysPerFile) return code;
4,833!
460
  if (pCfg->daysToKeep0 > pCfg->daysToKeep1) return code;
4,833!
461
  if (pCfg->daysToKeep1 > pCfg->daysToKeep2) return code;
4,833!
462
  if (pCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET) return code;
4,833!
463
  if (pCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
4,833!
464
  if (pCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
4,833!
465
  if (pCfg->minRows > pCfg->maxRows) return code;
4,833✔
466
  if (pCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
4,832!
467
  if (pCfg->walLevel < TSDB_MIN_WAL_LEVEL || pCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
4,832!
468
  if (pCfg->precision < TSDB_MIN_PRECISION && pCfg->precision > TSDB_MAX_PRECISION) return code;
4,832!
469
  if (pCfg->compression < TSDB_MIN_COMP_LEVEL || pCfg->compression > TSDB_MAX_COMP_LEVEL) return code;
4,832!
470
  if (pCfg->replications < TSDB_MIN_DB_REPLICA || pCfg->replications > TSDB_MAX_DB_REPLICA) return code;
4,832!
471
#ifdef TD_ENTERPRISE
472
  if ((pCfg->replications == 2) ^ (pCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
4,832!
473
  if (pCfg->encryptAlgorithm < TSDB_MIN_ENCRYPT_ALGO || pCfg->encryptAlgorithm > TSDB_MAX_ENCRYPT_ALGO) return code;
4,832!
474
#else
475
  if (pCfg->replications != 1 && pCfg->replications != 3) return code;
476
  if (pCfg->encryptAlgorithm != TSDB_DEFAULT_ENCRYPT_ALGO) return code;
477
#endif
478

479
  if (pCfg->strict < TSDB_DB_STRICT_OFF || pCfg->strict > TSDB_DB_STRICT_ON) return code;
4,832!
480
  if (pCfg->schemaless < TSDB_DB_SCHEMALESS_OFF || pCfg->schemaless > TSDB_DB_SCHEMALESS_ON) return code;
4,832!
481
  if (pCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
4,832!
482
  if (pCfg->hashMethod != 1) return code;
4,832!
483
  if (pCfg->replications > mndGetDnodeSize(pMnode)) {
4,832!
484
    code = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
485
    TAOS_RETURN(code);
×
486
  }
487
  if (pCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
4,832!
488
  if (pCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
4,832!
489
  if (pCfg->walRollPeriod < TSDB_DB_MIN_WAL_ROLL_PERIOD) return code;
4,832!
490
  if (pCfg->walSegmentSize < TSDB_DB_MIN_WAL_SEGMENT_SIZE) return code;
4,832!
491
  if (pCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pCfg->sstTrigger > TSDB_MAX_STT_TRIGGER) return code;
4,832!
492
  if (pCfg->hashPrefix < TSDB_MIN_HASH_PREFIX || pCfg->hashPrefix > TSDB_MAX_HASH_PREFIX) return code;
4,832!
493
  if (pCfg->hashSuffix < TSDB_MIN_HASH_SUFFIX || pCfg->hashSuffix > TSDB_MAX_HASH_SUFFIX) return code;
4,832!
494
  if ((pCfg->hashSuffix * pCfg->hashPrefix) < 0) return code;
4,832!
495
  if ((pCfg->hashPrefix + pCfg->hashSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) return code;
4,832!
496
  if (pCfg->tsdbPageSize < TSDB_MIN_TSDB_PAGESIZE || pCfg->tsdbPageSize > TSDB_MAX_TSDB_PAGESIZE) return code;
4,832!
497
  if (taosArrayGetSize(pCfg->pRetensions) != pCfg->numOfRetensions) return code;
4,832!
498

499
  if (pCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
4,832!
500
  if (pCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
4,832!
501
  if (pCfg->s3Compact < TSDB_MIN_S3_COMPACT || pCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
4,832!
502

503
  if (pCfg->compactInterval != 0 &&
4,832✔
504
      (pCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pCfg->compactInterval > pCfg->daysToKeep2))
2!
505
    return code;
×
506
  if (pCfg->compactStartTime != 0 &&
4,832✔
507
      (pCfg->compactStartTime < -pCfg->daysToKeep2 || pCfg->compactStartTime > -pCfg->daysPerFile))
1!
508
    return code;
×
509
  if (pCfg->compactEndTime != 0 &&
4,832✔
510
      (pCfg->compactEndTime < -pCfg->daysToKeep2 || pCfg->compactEndTime > -pCfg->daysPerFile))
1!
511
    return code;
×
512
  if (pCfg->compactStartTime != 0 && pCfg->compactEndTime != 0 && pCfg->compactStartTime >= pCfg->compactEndTime)
4,832!
513
    return code;
×
514
  if (pCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET || pCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
4,832!
515
    return code;
×
516

517
  code = 0;
4,832✔
518
  TAOS_RETURN(code);
4,832✔
519
}
520

521
static int32_t mndCheckInChangeDbCfg(SMnode *pMnode, SDbCfg *pOldCfg, SDbCfg *pNewCfg) {
241✔
522
  int32_t code = TSDB_CODE_MND_INVALID_DB_OPTION;
241✔
523
  if (pNewCfg->buffer < TSDB_MIN_BUFFER_PER_VNODE || pNewCfg->buffer > TSDB_MAX_BUFFER_PER_VNODE) return code;
241!
524
  if (pNewCfg->pages < TSDB_MIN_PAGES_PER_VNODE || pNewCfg->pages > TSDB_MAX_PAGES_PER_VNODE) return code;
241!
525
  if (pNewCfg->pageSize < TSDB_MIN_PAGESIZE_PER_VNODE || pNewCfg->pageSize > TSDB_MAX_PAGESIZE_PER_VNODE) return code;
241!
526
  if (pNewCfg->daysPerFile < TSDB_MIN_DAYS_PER_FILE || pNewCfg->daysPerFile > TSDB_MAX_DAYS_PER_FILE) return code;
241!
527
  if (pNewCfg->daysToKeep0 < TSDB_MIN_KEEP || pNewCfg->daysToKeep0 > TSDB_MAX_KEEP) return code;
241!
528
  if (pNewCfg->daysToKeep1 < TSDB_MIN_KEEP || pNewCfg->daysToKeep1 > TSDB_MAX_KEEP) return code;
241!
529
  if (pNewCfg->daysToKeep2 < TSDB_MIN_KEEP || pNewCfg->daysToKeep2 > TSDB_MAX_KEEP) return code;
241!
530
  if (pNewCfg->daysToKeep0 < pNewCfg->daysPerFile) return code;
241!
531
  if (pNewCfg->daysToKeep0 > pNewCfg->daysToKeep1) return code;
241!
532
  if (pNewCfg->daysToKeep1 > pNewCfg->daysToKeep2) return code;
241!
533
  if (pNewCfg->keepTimeOffset < TSDB_MIN_KEEP_TIME_OFFSET || pNewCfg->keepTimeOffset > TSDB_MAX_KEEP_TIME_OFFSET)
241!
534
    return code;
×
535
  if (pNewCfg->walFsyncPeriod < TSDB_MIN_FSYNC_PERIOD || pNewCfg->walFsyncPeriod > TSDB_MAX_FSYNC_PERIOD) return code;
241!
536
  if (pNewCfg->walLevel < TSDB_MIN_WAL_LEVEL || pNewCfg->walLevel > TSDB_MAX_WAL_LEVEL) return code;
241!
537
  if (pNewCfg->cacheLast < TSDB_CACHE_MODEL_NONE || pNewCfg->cacheLast > TSDB_CACHE_MODEL_BOTH) return code;
241!
538
  if (pNewCfg->cacheLastSize < TSDB_MIN_DB_CACHE_SIZE || pNewCfg->cacheLastSize > TSDB_MAX_DB_CACHE_SIZE) return code;
241!
539
  if (pNewCfg->replications < TSDB_MIN_DB_REPLICA || pNewCfg->replications > TSDB_MAX_DB_REPLICA) return code;
241!
540
#ifdef TD_ENTERPRISE
541
  if ((pNewCfg->replications == 2) ^ (pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR)) return code;
241!
542
  if (pNewCfg->replications == 2 && pNewCfg->withArbitrator == TSDB_MAX_DB_WITH_ARBITRATOR) {
241!
543
    if (pOldCfg->replications != 1 && pOldCfg->replications != 2) {
1!
544
      terrno = TSDB_CODE_OPS_NOT_SUPPORT;
1✔
545
      return code;
1✔
546
    }
547
  }
548
  if (pNewCfg->replications != 2 && pOldCfg->replications == 2) {
240!
549
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
×
550
    return code;
×
551
  }
552
#else
553
  if (pNewCfg->replications != 1 && pNewCfg->replications != 3) return code;
554
#endif
555

556
  if (pNewCfg->walLevel == 0 && pOldCfg->replications > 1) {
240!
557
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
558
    return code;
×
559
  }
560
  if (pNewCfg->replications > 1 && pOldCfg->walLevel == 0) {
240!
561
    terrno = TSDB_CODE_MND_INVALID_WAL_LEVEL;
×
562
    return code;
×
563
  }
564

565
  if (pNewCfg->sstTrigger != pOldCfg->sstTrigger &&
240!
566
      (pNewCfg->sstTrigger < TSDB_MIN_STT_TRIGGER || pNewCfg->sstTrigger > TSDB_MAX_STT_TRIGGER))
×
567
    return code;
×
568
  if (pNewCfg->minRows < TSDB_MIN_MINROWS_FBLOCK || pNewCfg->minRows > TSDB_MAX_MINROWS_FBLOCK) return code;
240!
569
  if (pNewCfg->maxRows < TSDB_MIN_MAXROWS_FBLOCK || pNewCfg->maxRows > TSDB_MAX_MAXROWS_FBLOCK) return code;
240!
570
  if (pNewCfg->minRows > pNewCfg->maxRows) return code;
240!
571
  if (pNewCfg->walRetentionPeriod < TSDB_DB_MIN_WAL_RETENTION_PERIOD) return code;
240!
572
  if (pNewCfg->walRetentionSize < TSDB_DB_MIN_WAL_RETENTION_SIZE) return code;
240!
573
  if (pNewCfg->strict < TSDB_DB_STRICT_OFF || pNewCfg->strict > TSDB_DB_STRICT_ON) return code;
240!
574
  if (pNewCfg->replications > mndGetDnodeSize(pMnode)) {
240!
575
    terrno = TSDB_CODE_MND_NO_ENOUGH_DNODES;
×
576
    return code;
×
577
  }
578
  if (pNewCfg->s3ChunkSize < TSDB_MIN_S3_CHUNK_SIZE || pNewCfg->s3ChunkSize > TSDB_MAX_S3_CHUNK_SIZE) return code;
240!
579
  if (pNewCfg->s3KeepLocal < TSDB_MIN_S3_KEEP_LOCAL || pNewCfg->s3KeepLocal > TSDB_MAX_S3_KEEP_LOCAL) return code;
240!
580
  if (pNewCfg->s3Compact < TSDB_MIN_S3_COMPACT || pNewCfg->s3Compact > TSDB_MAX_S3_COMPACT) return code;
240!
581

582
  if (pNewCfg->compactInterval != 0 &&
240!
583
      (pNewCfg->compactInterval < TSDB_MIN_COMPACT_INTERVAL || pNewCfg->compactInterval > pNewCfg->daysToKeep2))
×
584
    return code;
×
585
  if (pNewCfg->compactStartTime != 0 &&
240!
586
      (pNewCfg->compactStartTime < -pNewCfg->daysToKeep2 || pNewCfg->compactStartTime > -pNewCfg->daysPerFile))
×
587
    return code;
×
588
  if (pNewCfg->compactEndTime != 0 &&
240!
589
      (pNewCfg->compactEndTime < -pNewCfg->daysToKeep2 || pNewCfg->compactEndTime > -pNewCfg->daysPerFile))
×
590
    return code;
×
591
  if (pNewCfg->compactStartTime != 0 && pNewCfg->compactEndTime != 0 &&
240!
592
      pNewCfg->compactStartTime >= pNewCfg->compactEndTime)
×
593
    return code;
×
594
  if (pNewCfg->compactTimeOffset < TSDB_MIN_COMPACT_TIME_OFFSET ||
240!
595
      pNewCfg->compactTimeOffset > TSDB_MAX_COMPACT_TIME_OFFSET)
240!
596
    return code;
×
597

598
  code = 0;
240✔
599
  TAOS_RETURN(code);
240✔
600
}
601

602
static void mndSetDefaultDbCfg(SDbCfg *pCfg) {
4,833✔
603
  if (pCfg->numOfVgroups < 0) pCfg->numOfVgroups = TSDB_DEFAULT_VN_PER_DB;
4,833!
604
  if (pCfg->numOfStables < 0) pCfg->numOfStables = TSDB_DEFAULT_DB_SINGLE_STABLE;
4,833!
605
  if (pCfg->buffer < 0) pCfg->buffer = TSDB_DEFAULT_BUFFER_PER_VNODE;
4,833✔
606
  if (pCfg->pageSize < 0) pCfg->pageSize = TSDB_DEFAULT_PAGESIZE_PER_VNODE;
4,833✔
607
  if (pCfg->pages < 0) pCfg->pages = TSDB_DEFAULT_PAGES_PER_VNODE;
4,833✔
608
  if (pCfg->daysPerFile < 0) pCfg->daysPerFile = TSDB_DEFAULT_DURATION_PER_FILE;
4,833!
609
  if (pCfg->daysToKeep0 < 0) pCfg->daysToKeep0 = TSDB_DEFAULT_KEEP;
4,833!
610
  if (pCfg->daysToKeep1 < 0) pCfg->daysToKeep1 = pCfg->daysToKeep0;
4,833!
611
  if (pCfg->daysToKeep2 < 0) pCfg->daysToKeep2 = pCfg->daysToKeep1;
4,833!
612
  if (pCfg->keepTimeOffset < 0) pCfg->keepTimeOffset = TSDB_DEFAULT_KEEP_TIME_OFFSET;
4,833!
613
  if (pCfg->minRows < 0) pCfg->minRows = TSDB_DEFAULT_MINROWS_FBLOCK;
4,833!
614
  if (pCfg->maxRows < 0) pCfg->maxRows = TSDB_DEFAULT_MAXROWS_FBLOCK;
4,833!
615
  if (pCfg->walFsyncPeriod < 0) pCfg->walFsyncPeriod = TSDB_DEFAULT_FSYNC_PERIOD;
4,833!
616
  if (pCfg->walLevel < 0) pCfg->walLevel = TSDB_DEFAULT_WAL_LEVEL;
4,833!
617
  if (pCfg->precision < 0) pCfg->precision = TSDB_DEFAULT_PRECISION;
4,833!
618
  if (pCfg->compression < 0) pCfg->compression = TSDB_DEFAULT_COMP_LEVEL;
4,833!
619
  if (pCfg->replications < 0) pCfg->replications = TSDB_DEFAULT_DB_REPLICA;
4,833!
620
  if (pCfg->strict < 0) pCfg->strict = TSDB_DEFAULT_DB_STRICT;
4,833!
621
  if (pCfg->cacheLast < 0) pCfg->cacheLast = TSDB_DEFAULT_CACHE_MODEL;
4,833!
622
  if (pCfg->cacheLastSize <= 0) pCfg->cacheLastSize = TSDB_DEFAULT_CACHE_SIZE;
4,833✔
623
  if (pCfg->numOfRetensions < 0) pCfg->numOfRetensions = 0;
4,833!
624
  if (pCfg->schemaless < 0) pCfg->schemaless = TSDB_DB_SCHEMALESS_OFF;
4,833!
625
  if (pCfg->walRetentionPeriod < 0 && pCfg->walRetentionPeriod != -1)
4,833!
626
    pCfg->walRetentionPeriod = TSDB_REPS_DEF_DB_WAL_RET_PERIOD;
×
627
  if (pCfg->walRetentionSize < 0 && pCfg->walRetentionSize != -1)
4,833!
628
    pCfg->walRetentionSize = TSDB_REPS_DEF_DB_WAL_RET_SIZE;
×
629
  if (pCfg->walRollPeriod < 0) pCfg->walRollPeriod = TSDB_REPS_DEF_DB_WAL_ROLL_PERIOD;
4,833!
630
  if (pCfg->walSegmentSize < 0) pCfg->walSegmentSize = TSDB_DEFAULT_DB_WAL_SEGMENT_SIZE;
4,833!
631
  if (pCfg->sstTrigger <= 0) pCfg->sstTrigger = TSDB_DEFAULT_SST_TRIGGER;
4,833✔
632
  if (pCfg->tsdbPageSize <= 0) pCfg->tsdbPageSize = TSDB_DEFAULT_TSDB_PAGESIZE;
4,833✔
633
  if (pCfg->s3ChunkSize <= 0) pCfg->s3ChunkSize = TSDB_DEFAULT_S3_CHUNK_SIZE;
4,833✔
634
  if (pCfg->s3KeepLocal <= 0) pCfg->s3KeepLocal = TSDB_DEFAULT_S3_KEEP_LOCAL;
4,833✔
635
  if (pCfg->s3Compact < 0) pCfg->s3Compact = TSDB_DEFAULT_S3_COMPACT;
4,833!
636
  if (pCfg->withArbitrator < 0) pCfg->withArbitrator = TSDB_DEFAULT_DB_WITH_ARBITRATOR;
4,833!
637
  if (pCfg->encryptAlgorithm < 0) pCfg->encryptAlgorithm = TSDB_DEFAULT_ENCRYPT_ALGO;
4,833!
638
}
4,833✔
639

640
static int32_t mndSetCreateDbPrepareAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
4,314✔
641
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,314✔
642
  if (pDbRaw == NULL) return -1;
4,314!
643

644
  if (mndTransAppendPrepareLog(pTrans, pDbRaw) != 0) return -1;
4,314!
645
  if (sdbSetRawStatus(pDbRaw, SDB_STATUS_CREATING) != 0) return -1;
4,314!
646
  return 0;
4,314✔
647
}
648

649
static int32_t mndSetNewVgPrepareActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,314✔
650
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,060✔
651
    if (mndAddNewVgPrepareAction(pMnode, pTrans, (pVgroups + v)) != 0) return -1;
9,746!
652
  }
653
  return 0;
4,314✔
654
}
655

656
static int32_t mndSetCreateDbRedoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
×
657
  int32_t  code = 0;
×
658
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
×
659
  if (pDbRaw == NULL) {
×
660
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
661
    if (terrno != 0) code = terrno;
×
662
    TAOS_RETURN(code);
×
663
  }
664
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pDbRaw));
×
665
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_UPDATE));
×
666

667
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
668
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
×
669
    if (pVgRaw == NULL) {
×
670
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
671
      if (terrno != 0) code = terrno;
×
672
      TAOS_RETURN(code);
×
673
    }
674
    TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pVgRaw));
×
675
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_UPDATE));
×
676
  }
677

678
  if (pDb->cfg.withArbitrator) {
×
679
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
×
680
      SVgObj   *pVgObj = pVgroups + v;
×
681
      SArbGroup arbGroup = {0};
×
682
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
×
683
      TAOS_CHECK_RETURN(mndSetCreateArbGroupRedoLogs(pTrans, &arbGroup));
×
684
    }
685
  }
686

687
  TAOS_RETURN(code);
×
688
}
689

690
static int32_t mndSetCreateDbUndoLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,314✔
691
  int32_t  code = 0;
4,314✔
692
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,314✔
693
  if (pDbRaw == NULL) {
4,314!
694
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
695
    if (terrno != 0) code = terrno;
×
696
    TAOS_RETURN(code);
×
697
  }
698
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pDbRaw));
4,314!
699
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_DROPPED));
4,314!
700

701
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,060✔
702
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
9,746✔
703
    if (pVgRaw == NULL) {
9,746!
704
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
705
      if (terrno != 0) code = terrno;
×
706
      TAOS_RETURN(code);
×
707
    }
708
    TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pVgRaw));
9,746!
709
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED));
9,746!
710
  }
711

712
  if (pDb->cfg.withArbitrator) {
4,314✔
713
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
11✔
714
      SVgObj   *pVgObj = pVgroups + v;
7✔
715
      SArbGroup arbGroup = {0};
7✔
716
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
7!
717
      TAOS_CHECK_RETURN(mndSetCreateArbGroupUndoLogs(pTrans, &arbGroup));
7!
718
    }
719
  }
720

721
  TAOS_RETURN(code);
4,314✔
722
}
723

724
static int32_t mndSetCreateDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups,
4,314✔
725
                                        SUserObj *pUserDuped) {
726
  int32_t  code = 0;
4,314✔
727
  SSdbRaw *pDbRaw = mndDbActionEncode(pDb);
4,314✔
728
  if (pDbRaw == NULL) {
4,314!
729
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
730
    if (terrno != 0) code = terrno;
×
731
    TAOS_RETURN(code);
×
732
  }
733
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pDbRaw));
4,314!
734
  TAOS_CHECK_RETURN(sdbSetRawStatus(pDbRaw, SDB_STATUS_READY));
4,314!
735

736
  for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
14,060✔
737
    SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroups + v);
9,746✔
738
    if (pVgRaw == NULL) {
9,746!
739
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
740
      if (terrno != 0) code = terrno;
×
741
      TAOS_RETURN(code);
×
742
    }
743
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pVgRaw));
9,746!
744
    TAOS_CHECK_RETURN(sdbSetRawStatus(pVgRaw, SDB_STATUS_READY));
9,746!
745
  }
746

747
  if (pDb->cfg.withArbitrator) {
4,314✔
748
    for (int32_t v = 0; v < pDb->cfg.numOfVgroups; ++v) {
11✔
749
      SVgObj   *pVgObj = pVgroups + v;
7✔
750
      SArbGroup arbGroup = {0};
7✔
751
      TAOS_CHECK_RETURN(mndArbGroupInitFromVgObj(pVgObj, &arbGroup));
7!
752
      TAOS_CHECK_RETURN(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup));
7!
753
    }
754
  }
755

756
  if (pUserDuped) {
4,314✔
757
    SSdbRaw *pUserRaw = mndUserActionEncode(pUserDuped);
4✔
758
    if (pUserRaw == NULL) {
4!
759
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
760
      if (terrno != 0) code = terrno;
×
761
      TAOS_RETURN(code);
×
762
    }
763
    TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pUserRaw));
4!
764
    TAOS_CHECK_RETURN(sdbSetRawStatus(pUserRaw, SDB_STATUS_READY));
4!
765
  }
766

767
  TAOS_RETURN(code);
4,314✔
768
}
769

770
static int32_t mndSetCreateDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,314✔
771
  int32_t code = 0;
4,314✔
772
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
14,060✔
773
    SVgObj *pVgroup = pVgroups + vg;
9,746✔
774

775
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
20,273✔
776
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
10,527✔
777
      TAOS_CHECK_RETURN(mndAddCreateVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid));
10,527!
778
    }
779
  }
780

781
  TAOS_RETURN(code);
4,314✔
782
}
783

784
static int32_t mndSetCreateDbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroups) {
4,314✔
785
  int32_t code = 0;
4,314✔
786
  for (int32_t vg = 0; vg < pDb->cfg.numOfVgroups; ++vg) {
14,060✔
787
    SVgObj *pVgroup = pVgroups + vg;
9,746✔
788

789
    for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
20,273✔
790
      SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
10,527✔
791
      TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, false));
10,527!
792
    }
793
  }
794

795
  TAOS_RETURN(code);
4,314✔
796
}
797

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

853
  dbObj.cfg.numOfRetensions = pCreate->numOfRetensions;
4,833✔
854
  dbObj.cfg.pRetensions = pCreate->pRetensions;
4,833✔
855

856
  mndSetDefaultDbCfg(&dbObj.cfg);
4,833✔
857

858
  if ((code = mndCheckDbName(dbObj.name, pUser)) != 0) {
4,833!
859
    mError("db:%s, failed to create, check db name failed, since %s", pCreate->db, terrstr());
×
860
    TAOS_RETURN(code);
×
861
  }
862

863
  if ((code = mndCheckDbCfg(pMnode, &dbObj.cfg)) != 0) {
4,833✔
864
    mError("db:%s, failed to create, check db cfg failed, since %s", pCreate->db, terrstr());
1!
865
    TAOS_RETURN(code);
1✔
866
  }
867

868
  if (dbObj.cfg.hashPrefix > 0) {
4,832✔
869
    int32_t dbLen = strlen(dbObj.name) + 1;
11✔
870
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix + dbLen);
11!
871
    dbObj.cfg.hashPrefix += dbLen;
11✔
872
  } else if (dbObj.cfg.hashPrefix < 0) {
4,821✔
873
    int32_t dbLen = strlen(dbObj.name) + 1;
11✔
874
    mInfo("db:%s, hashPrefix adjust from %d to %d", dbObj.name, dbObj.cfg.hashPrefix, dbObj.cfg.hashPrefix - dbLen);
11!
875
    dbObj.cfg.hashPrefix -= dbLen;
11✔
876
  }
877

878
  SVgObj *pVgroups = NULL;
4,832✔
879
  if ((code = mndAllocVgroup(pMnode, &dbObj, &pVgroups, dnodeList)) != 0) {
4,832✔
880
    mError("db:%s, failed to create, alloc vgroup failed, since %s", pCreate->db, terrstr());
492!
881
    TAOS_RETURN(code);
492✔
882
  }
883

884
  // add database privileges for user
885
  SUserObj *pNewUserDuped = NULL;
4,340✔
886
  if (!pUser->superUser) {
4,340✔
887
    TAOS_CHECK_GOTO(mndUserDupObj(pUser, &newUserObj), NULL, _OVER);
4!
888
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.readDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
4!
889
                    NULL, _OVER);
890
    TAOS_CHECK_GOTO(taosHashPut(newUserObj.writeDbs, dbObj.name, strlen(dbObj.name) + 1, dbObj.name, TSDB_FILENAME_LEN),
4!
891
                    NULL, _OVER);
892
    pNewUserDuped = &newUserObj;
4✔
893
  }
894

895
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "create-db");
4,340✔
896
  if (pTrans == NULL) {
4,340!
897
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
898
    if (terrno != 0) code = terrno;
×
899
    goto _OVER;
×
900
  }
901
  // mndTransSetSerial(pTrans);
902
  mInfo("trans:%d, used to create db:%s", pTrans->id, pCreate->db);
4,340!
903

904
  mndTransSetDbName(pTrans, dbObj.name, NULL);
4,340✔
905
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
4,340✔
906

907
  mndTransSetOper(pTrans, MND_OPER_CREATE_DB);
4,314✔
908
  TAOS_CHECK_GOTO(mndSetCreateDbPrepareAction(pMnode, pTrans, &dbObj), NULL, _OVER);
4,314!
909
  TAOS_CHECK_GOTO(mndSetCreateDbRedoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,314!
910
  TAOS_CHECK_GOTO(mndSetNewVgPrepareActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,314!
911
  TAOS_CHECK_GOTO(mndSetCreateDbUndoLogs(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,314!
912
  TAOS_CHECK_GOTO(mndSetCreateDbCommitLogs(pMnode, pTrans, &dbObj, pVgroups, pNewUserDuped), NULL, _OVER);
4,314!
913
  TAOS_CHECK_GOTO(mndSetCreateDbUndoActions(pMnode, pTrans, &dbObj, pVgroups), NULL, _OVER);
4,314!
914
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
4,314✔
915

916
_OVER:
4,309✔
917
  taosMemoryFree(pVgroups);
4,340!
918
  mndUserFreeObj(&newUserObj);
4,340✔
919
  mndTransDrop(pTrans);
4,340✔
920
  TAOS_RETURN(code);
4,340✔
921
}
922

923
static int32_t mndCheckDbEncryptKey(SMnode *pMnode, SCreateDbReq *pReq) {
4,849✔
924
  int32_t    code = 0;
4,849✔
925
  SSdb      *pSdb = pMnode->pSdb;
4,849✔
926
  SDnodeObj *pDnode = NULL;
4,849✔
927
  void      *pIter = NULL;
4,849✔
928

929
#if defined(TD_ENTERPRISE) || defined(TD_ASTRA_TODO)
930
  if (pReq->encryptAlgorithm == TSDB_ENCRYPT_ALGO_NONE) goto _exit;
4,849✔
931
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB_ENCRYPTION), NULL, _exit);
1!
932
  if (tsEncryptionKeyStat != ENCRYPT_KEY_STAT_LOADED) {
1!
933
    code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
934
    mError("db:%s, failed to check encryption key:%" PRIi8 " in mnode leader since it's not loaded", pReq->db,
×
935
           tsEncryptionKeyStat);
936
    goto _exit;
×
937
  }
938

939
  int64_t curMs = taosGetTimestampMs();
1✔
940
  while ((pIter = sdbFetch(pSdb, SDB_DNODE, pIter, (void **)&pDnode))) {
2✔
941
    bool online = false;
1✔
942
    if ((pDnode->encryptionKeyStat != tsEncryptionKeyStat || pDnode->encryptionKeyChksum != tsEncryptionKeyChksum) &&
1!
943
        (online = mndIsDnodeOnline(pDnode, curMs))) {
×
944
      code = TSDB_CODE_MND_INVALID_ENCRYPT_KEY;
×
945
      mError("db:%s, failed to check encryption key:%" PRIi8
×
946
             "-%u in dnode:%d since it's inconsitent with mnode leader:%" PRIi8 "-%u",
947
             pReq->db, pDnode->encryptionKeyStat, pDnode->encryptionKeyChksum, pDnode->id, tsEncryptionKeyStat,
948
             tsEncryptionKeyChksum);
949
      sdbCancelFetch(pSdb, pIter);
×
950
      sdbRelease(pSdb, pDnode);
×
951
      break;
×
952
    }
953
    sdbRelease(pSdb, pDnode);
1✔
954
  }
955
#else
956
  if (pReq->encryptAlgorithm != TSDB_ENCRYPT_ALGO_NONE) {
957
    code = TSDB_CODE_MND_INVALID_DB_OPTION;
958
    goto _exit;
959
  }
960
#endif
961
_exit:
1✔
962
  TAOS_RETURN(code);
4,849✔
963
}
964

965
#ifndef TD_ENTERPRISE
966
int32_t mndCheckDbDnodeList(SMnode *pMnode, char *db, char *dnodeListStr, SArray *dnodeList) {
967
  if (dnodeListStr[0] != 0) {
968
    terrno = TSDB_CODE_OPS_NOT_SUPPORT;
969
    return terrno;
970
  } else {
971
    return 0;
972
  }
973
}
974
#endif
975

976
static int32_t mndProcessCreateDbReq(SRpcMsg *pReq) {
5,275✔
977
  SMnode      *pMnode = pReq->info.node;
5,275✔
978
  int32_t      code = -1;
5,275✔
979
  int32_t      lino = 0;
5,275✔
980
  SDbObj      *pDb = NULL;
5,275✔
981
  SUserObj    *pUser = NULL;
5,275✔
982
  SCreateDbReq createReq = {0};
5,275✔
983
  SArray      *dnodeList = NULL;
5,275✔
984

985
  dnodeList = taosArrayInit(mndGetDnodeSize(pMnode), sizeof(int32_t));
5,275✔
986
  TSDB_CHECK_NULL(dnodeList, code, lino, _OVER, TSDB_CODE_OUT_OF_MEMORY);
5,275!
987

988
  TAOS_CHECK_GOTO(tDeserializeSCreateDbReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
5,275!
989
#ifdef WINDOWS
990
  if (taosArrayGetSize(createReq.pRetensions) > 0) {
991
    terrno = TSDB_CODE_MND_INVALID_PLATFORM;
992
    goto _OVER;
993
  }
994
#endif
995
  mInfo("db:%s, start to create, vgroups:%d", createReq.db, createReq.numOfVgroups);
5,275!
996

997
  pDb = mndAcquireDb(pMnode, createReq.db);
5,275✔
998
  if (pDb != NULL) {
5,275✔
999
    if (createReq.ignoreExist) {
377✔
1000
      mInfo("db:%s, already exist, ignore exist is set", createReq.db);
364!
1001
      code = 0;
364✔
1002
      goto _OVER;
364✔
1003
    } else {
1004
      code = TSDB_CODE_MND_DB_ALREADY_EXIST;
13✔
1005
      goto _OVER;
13✔
1006
    }
1007
  } else {
1008
    if (terrno == TSDB_CODE_MND_DB_IN_CREATING) {
4,898✔
1009
      code = terrno;
48✔
1010
      goto _OVER;
48✔
1011
    } else if (terrno == TSDB_CODE_MND_DB_IN_DROPPING) {
4,850!
1012
      goto _OVER;
×
1013
    } else if (terrno == TSDB_CODE_MND_DB_NOT_EXIST) {
4,850!
1014
      // continue
1015
    } else {  // TSDB_CODE_APP_ERROR
1016
      goto _OVER;
×
1017
    }
1018
  }
1019

1020
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_DB, NULL), &lino, _OVER);
4,850✔
1021

1022
  TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DB), &lino, _OVER);
4,849!
1023

1024
  if (createReq.replications == 2) {
4,849✔
1025
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), &lino, _OVER);
6!
1026
  }
1027

1028
  TAOS_CHECK_GOTO(mndCheckDbEncryptKey(pMnode, &createReq), &lino, _OVER);
4,849!
1029

1030
  TAOS_CHECK_GOTO(mndCheckDbDnodeList(pMnode, createReq.db, createReq.dnodeListStr, dnodeList), &lino, _OVER);
4,849✔
1031

1032
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pUser), &lino, _OVER);
4,833!
1033

1034
  TAOS_CHECK_GOTO(mndCreateDb(pMnode, pReq, &createReq, pUser, dnodeList), &lino, _OVER);
4,833✔
1035
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
4,309!
1036

1037
  SName name = {0};
4,309✔
1038
  if (tNameFromString(&name, createReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
4,309!
1039
    mError("db:%s, failed to parse db name", createReq.db);
×
1040

1041
  auditRecord(pReq, pMnode->clusterId, "createDB", name.dbname, "", createReq.sql, createReq.sqlLen);
4,309✔
1042

1043
_OVER:
5,275✔
1044
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
5,275✔
1045
    mError("db:%s, failed to create at line:%d since %s", createReq.db, lino, tstrerror(code));
602!
1046
  }
1047

1048
  mndReleaseDb(pMnode, pDb);
5,275✔
1049
  mndReleaseUser(pMnode, pUser);
5,275✔
1050
  tFreeSCreateDbReq(&createReq);
5,275✔
1051
  taosArrayDestroy(dnodeList);
5,275✔
1052

1053
  TAOS_RETURN(code);
5,275✔
1054
}
1055

1056
static int32_t mndSetDbCfgFromAlterDbReq(SDbObj *pDb, SAlterDbReq *pAlter) {
267✔
1057
  int32_t code = TSDB_CODE_MND_DB_OPTION_UNCHANGED;
267✔
1058

1059
  if (pAlter->buffer > 0 && pAlter->buffer != pDb->cfg.buffer) {
267!
1060
    pDb->cfg.buffer = pAlter->buffer;
×
1061
    code = 0;
×
1062
  }
1063

1064
  if (pAlter->pages > 0 && pAlter->pages != pDb->cfg.pages) {
267!
1065
    pDb->cfg.pages = pAlter->pages;
×
1066
    code = 0;
×
1067
  }
1068

1069
  if (pAlter->pageSize > 0 && pAlter->pageSize != pDb->cfg.pageSize) {
267!
1070
    pDb->cfg.pageSize = pAlter->pageSize;
×
1071
    code = 0;
×
1072
  }
1073

1074
  if (pAlter->daysPerFile > 0 && pAlter->daysPerFile != pDb->cfg.daysPerFile) {
267!
1075
    pDb->cfg.daysPerFile = pAlter->daysPerFile;
×
1076
    code = 0;
×
1077
  }
1078

1079
  if (pAlter->daysToKeep0 > 0 && pAlter->daysToKeep0 != pDb->cfg.daysToKeep0) {
267✔
1080
    pDb->cfg.daysToKeep0 = pAlter->daysToKeep0;
16✔
1081
    code = 0;
16✔
1082
  }
1083

1084
  if (pAlter->daysToKeep1 > 0 && pAlter->daysToKeep1 != pDb->cfg.daysToKeep1) {
267✔
1085
    pDb->cfg.daysToKeep1 = pAlter->daysToKeep1;
16✔
1086
    code = 0;
16✔
1087
  }
1088

1089
  if (pAlter->daysToKeep2 > 0 && pAlter->daysToKeep2 != pDb->cfg.daysToKeep2) {
267✔
1090
    pDb->cfg.daysToKeep2 = pAlter->daysToKeep2;
17✔
1091
    code = 0;
17✔
1092
  }
1093

1094
  if (pAlter->keepTimeOffset >= 0 && pAlter->keepTimeOffset != pDb->cfg.keepTimeOffset) {
267!
1095
    pDb->cfg.keepTimeOffset = pAlter->keepTimeOffset;
1✔
1096
    code = 0;
1✔
1097
  }
1098

1099
  if (pAlter->walFsyncPeriod >= 0 && pAlter->walFsyncPeriod != pDb->cfg.walFsyncPeriod) {
267!
1100
    pDb->cfg.walFsyncPeriod = pAlter->walFsyncPeriod;
3✔
1101
    code = 0;
3✔
1102
  }
1103

1104
  if (pAlter->walLevel >= 0 && pAlter->walLevel != pDb->cfg.walLevel) {
267!
1105
    pDb->cfg.walLevel = pAlter->walLevel;
2✔
1106
    code = 0;
2✔
1107
  }
1108

1109
  if (pAlter->strict >= 0 && pAlter->strict != pDb->cfg.strict) {
267!
1110
#if 1
1111
    code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1112
#else
1113
    pDb->cfg.strict = pAlter->strict;
1114
    code = 0;
1115
#endif
1116
  }
1117

1118
  if (pAlter->cacheLast >= 0 && pAlter->cacheLast != pDb->cfg.cacheLast) {
267!
1119
    pDb->cfg.cacheLast = pAlter->cacheLast;
191✔
1120
    code = 0;
191✔
1121
  }
1122

1123
  if (pAlter->cacheLastSize > 0 && pAlter->cacheLastSize != pDb->cfg.cacheLastSize) {
267✔
1124
    pDb->cfg.cacheLastSize = pAlter->cacheLastSize;
144✔
1125
    code = 0;
144✔
1126
  }
1127

1128
  if (pAlter->replications > 0 && pAlter->replications != pDb->cfg.replications) {
267!
1129
    pDb->cfg.replications = pAlter->replications;
6✔
1130
    pDb->vgVersion++;
6✔
1131
    code = 0;
6✔
1132
  }
1133

1134
  if (pAlter->sstTrigger > 0 && pAlter->sstTrigger != pDb->cfg.sstTrigger) {
267!
1135
    pDb->cfg.sstTrigger = pAlter->sstTrigger;
×
1136
    pDb->vgVersion++;
×
1137
    code = 0;
×
1138
  }
1139

1140
  if (pAlter->minRows > 0 && pAlter->minRows != pDb->cfg.minRows) {
267!
1141
    pDb->cfg.minRows = pAlter->minRows;
×
1142
    pDb->vgVersion++;
×
1143
    code = 0;
×
1144
  }
1145

1146
  if (pAlter->walRetentionPeriod > TSDB_DB_MIN_WAL_RETENTION_PERIOD &&
267✔
1147
      pAlter->walRetentionPeriod != pDb->cfg.walRetentionPeriod) {
44✔
1148
    pDb->cfg.walRetentionPeriod = pAlter->walRetentionPeriod;
20✔
1149
    pDb->vgVersion++;
20✔
1150
    code = 0;
20✔
1151
  }
1152

1153
  if (pAlter->walRetentionSize > TSDB_DB_MIN_WAL_RETENTION_SIZE &&
267!
1154
      pAlter->walRetentionSize != pDb->cfg.walRetentionSize) {
×
1155
    pDb->cfg.walRetentionSize = pAlter->walRetentionSize;
×
1156
    pDb->vgVersion++;
×
1157
    code = 0;
×
1158
  }
1159

1160
  if (pAlter->s3KeepLocal > TSDB_MIN_S3_KEEP_LOCAL && pAlter->s3KeepLocal != pDb->cfg.s3KeepLocal) {
267!
1161
    pDb->cfg.s3KeepLocal = pAlter->s3KeepLocal;
×
1162
    pDb->vgVersion++;
×
1163
    code = 0;
×
1164
  }
1165

1166
  if (pAlter->s3Compact >= TSDB_MIN_S3_COMPACT && pAlter->s3Compact != pDb->cfg.s3Compact) {
267!
1167
    pDb->cfg.s3Compact = pAlter->s3Compact;
×
1168
    pDb->vgVersion++;
×
1169
    code = 0;
×
1170
  }
1171

1172
  if (pAlter->withArbitrator >= TSDB_MIN_DB_WITH_ARBITRATOR && pAlter->withArbitrator != pDb->cfg.withArbitrator) {
267✔
1173
    pDb->cfg.withArbitrator = pAlter->withArbitrator;
1✔
1174
    pDb->vgVersion++;
1✔
1175
    code = 0;
1✔
1176
  }
1177

1178
  if (pAlter->compactInterval >= TSDB_DEFAULT_COMPACT_INTERVAL && pAlter->compactInterval != pDb->cfg.compactInterval) {
267!
1179
    pDb->cfg.compactInterval = pAlter->compactInterval;
×
1180
    pDb->vgVersion++;
×
1181
    code = 0;
×
1182
  }
1183

1184
  bool compactTimeRangeChanged = false;
267✔
1185
  if (pAlter->compactStartTime != pDb->cfg.compactStartTime &&
267!
1186
      (pAlter->compactStartTime == TSDB_DEFAULT_COMPACT_START_TIME ||
267!
1187
       pAlter->compactStartTime <= -pDb->cfg.daysPerFile)) {
267!
1188
    pDb->cfg.compactStartTime = pAlter->compactStartTime;
×
1189
    compactTimeRangeChanged = true;
×
1190
    code = 0;
×
1191
  }
1192

1193
  if (pAlter->compactEndTime != pDb->cfg.compactEndTime &&
267!
1194
      (pAlter->compactEndTime == TSDB_DEFAULT_COMPACT_END_TIME || pAlter->compactEndTime <= -pDb->cfg.daysPerFile)) {
267!
1195
    pDb->cfg.compactEndTime = pAlter->compactEndTime;
×
1196
    compactTimeRangeChanged = true;
×
1197
    code = 0;
×
1198
  }
1199
  if(compactTimeRangeChanged) {
267!
1200
    pDb->vgVersion++;
×
1201
  }
1202

1203
  if (pAlter->compactTimeOffset >= TSDB_MIN_COMPACT_TIME_OFFSET &&
267!
1204
      pAlter->compactTimeOffset != pDb->cfg.compactTimeOffset) {
×
1205
    pDb->cfg.compactTimeOffset = pAlter->compactTimeOffset;
×
1206
    pDb->vgVersion++;
×
1207
    code = 0;
×
1208
  }
1209

1210
  TAOS_RETURN(code);
267✔
1211
}
1212

1213
static int32_t mndSetAlterDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
239✔
1214
  int32_t  code = 0;
239✔
1215
  SSdbRaw *pRedoRaw = mndDbActionEncode(pOld);
239✔
1216
  if (pRedoRaw == NULL) {
239!
1217
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1218
    if (terrno != 0) code = terrno;
×
1219
    TAOS_RETURN(code);
×
1220
  }
1221
  if ((code = mndTransAppendPrepareLog(pTrans, pRedoRaw)) != 0) {
239!
1222
    sdbFreeRaw(pRedoRaw);
×
1223
    TAOS_RETURN(code);
×
1224
  }
1225

1226
  if ((code = sdbSetRawStatus(pRedoRaw, SDB_STATUS_READY)) != 0) {
239!
1227
    sdbFreeRaw(pRedoRaw);
×
1228
    TAOS_RETURN(code);
×
1229
  }
1230
  return 0;
239✔
1231
}
1232

1233
static int32_t mndSetAlterDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pOld, SDbObj *pNew) {
239✔
1234
  int32_t  code = 0;
239✔
1235
  SSdbRaw *pCommitRaw = mndDbActionEncode(pNew);
239✔
1236
  if (pCommitRaw == NULL) {
239!
1237
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1238
    if (terrno != 0) code = terrno;
×
1239
    TAOS_RETURN(code);
×
1240
  }
1241
  if ((code = mndTransAppendCommitlog(pTrans, pCommitRaw)) != 0) {
239!
1242
    sdbFreeRaw(pCommitRaw);
×
1243
    TAOS_RETURN(code);
×
1244
  }
1245

1246
  if ((code = sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY)) != 0) {
239!
1247
    sdbFreeRaw(pCommitRaw);
×
1248
    TAOS_RETURN(code);
×
1249
  }
1250
  TAOS_RETURN(code);
239✔
1251
}
1252

1253
static int32_t mndSetAlterDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pOldDb, SDbObj *pNewDb) {
239✔
1254
  int32_t code = 0, lino = 0;
239✔
1255
  SSdb   *pSdb = pMnode->pSdb;
239✔
1256
  void   *pIter = NULL;
239✔
1257
  SVgObj *pVgroup = NULL;
239✔
1258
  SArray *pArray = mndBuildDnodesArray(pMnode, 0, NULL);
239✔
1259

1260
  TSDB_CHECK_NULL(pArray, code, lino, _err, TSDB_CODE_OUT_OF_MEMORY);
239!
1261

1262
  while (1) {
1263
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
7,436✔
1264
    if (pIter == NULL) break;
7,436✔
1265

1266
    if (mndVgroupInDb(pVgroup, pNewDb->uid)) {
7,197✔
1267
      SVgObj newVgroup = {0};
633✔
1268
      TAOS_CHECK_GOTO(mndBuildAlterVgroupAction(pMnode, pTrans, pOldDb, pNewDb, pVgroup, pArray, &newVgroup), &lino,
633!
1269
                      _err);
1270

1271
      if (pNewDb->cfg.withArbitrator != pOldDb->cfg.withArbitrator) {
633!
1272
        if (pNewDb->cfg.withArbitrator) {
×
1273
          SArbGroup arbGroup = {0};
×
1274
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(&newVgroup, &arbGroup), &lino, _err);
×
1275
          TAOS_CHECK_GOTO(mndSetCreateArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1276
        } else {
1277
          SArbGroup arbGroup = {0};
×
1278
          TAOS_CHECK_GOTO(mndArbGroupInitFromVgObj(pVgroup, &arbGroup), &lino, _err);
×
1279
          TAOS_CHECK_GOTO(mndSetDropArbGroupCommitLogs(pTrans, &arbGroup), &lino, _err);
×
1280
        }
1281
      }
1282
    }
1283

1284
    sdbRelease(pSdb, pVgroup);
7,197✔
1285
  }
1286

1287
  taosArrayDestroy(pArray);
239✔
1288
  TAOS_RETURN(code);
239✔
1289

1290
_err:
×
1291
  mError("db:%s, %s failed at %d since %s", pNewDb->name, __func__, lino, tstrerror(code));
×
1292

1293
  sdbCancelFetch(pSdb, pIter);
×
1294
  sdbRelease(pSdb, pVgroup);
×
1295
  taosArrayDestroy(pArray);
×
1296
  TAOS_RETURN(code);
×
1297
}
1298

1299
static int32_t mndAlterDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pOld, SDbObj *pNew) {
240✔
1300
  int32_t code = -1;
240✔
1301
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "alter-db");
240✔
1302
  if (pTrans == NULL) {
240!
1303
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1304
    if (terrno != 0) code = terrno;
×
1305
    TAOS_RETURN(code);
×
1306
  }
1307
  mInfo("trans:%d, used to alter db:%s", pTrans->id, pOld->name);
240!
1308
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
240!
1309

1310
  mndTransSetDbName(pTrans, pOld->name, NULL);
240✔
1311
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
240!
1312
  TAOS_CHECK_GOTO(mndTransCheckConflictWithCompact(pMnode, pTrans), NULL, _OVER);
240✔
1313

1314
  TAOS_CHECK_GOTO(mndSetAlterDbPrepareLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
239!
1315
  TAOS_CHECK_GOTO(mndSetAlterDbCommitLogs(pMnode, pTrans, pOld, pNew), NULL, _OVER);
239!
1316
  TAOS_CHECK_GOTO(mndSetAlterDbRedoActions(pMnode, pTrans, pOld, pNew), NULL, _OVER);
239!
1317

1318
  mInfo("trans:%d, used to alter db, ableToBeKilled:%d, killMode:%d", pTrans->id, pTrans->ableToBeKilled, pTrans->killMode);
239!
1319
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
239!
1320
  code = 0;
239✔
1321

1322
_OVER:
240✔
1323
  mndTransDrop(pTrans);
240✔
1324
  TAOS_RETURN(code);
240✔
1325
}
1326

1327
static int32_t mndProcessAlterDbReq(SRpcMsg *pReq) {
268✔
1328
  SMnode     *pMnode = pReq->info.node;
268✔
1329
  int32_t     code = -1;
268✔
1330
  SDbObj     *pDb = NULL;
268✔
1331
  SAlterDbReq alterReq = {0};
268✔
1332
  SDbObj      dbObj = {0};
268✔
1333

1334
  TAOS_CHECK_GOTO(tDeserializeSAlterDbReq(pReq->pCont, pReq->contLen, &alterReq), NULL, _OVER);
268!
1335

1336
  mInfo("db:%s, start to alter", alterReq.db);
268!
1337

1338
  pDb = mndAcquireDb(pMnode, alterReq.db);
268✔
1339
  if (pDb == NULL) {
268!
1340
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1341
    if (terrno != 0) code = terrno;
×
1342
    goto _OVER;
×
1343
  }
1344

1345
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_ALTER_DB, pDb), NULL, _OVER);
268✔
1346

1347
  if (alterReq.replications == 2) {
267✔
1348
    TAOS_CHECK_GOTO(grantCheck(TSDB_GRANT_DUAL_REPLICA_HA), NULL, _OVER);
1!
1349
  }
1350

1351
  int32_t numOfTopics = 0;
267✔
1352
  TAOS_CHECK_GOTO(mndGetNumOfTopics(pMnode, pDb->name, &numOfTopics), NULL, _OVER);
267!
1353

1354
  if (numOfTopics != 0 && alterReq.walRetentionPeriod == 0) {
267!
1355
    code = TSDB_CODE_MND_DB_RETENTION_PERIOD_ZERO;
×
1356
    mError("db:%s, not allowed to set WAL_RETENTION_PERIOD 0 when there are topics defined. numOfTopics:%d", pDb->name,
×
1357
           numOfTopics);
1358
    goto _OVER;
×
1359
  }
1360

1361
  (void)memcpy(&dbObj, pDb, sizeof(SDbObj));
267✔
1362
  if (dbObj.cfg.pRetensions != NULL) {
267!
1363
    dbObj.cfg.pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
×
1364
    if (dbObj.cfg.pRetensions == NULL) goto _OVER;
×
1365
  }
1366

1367
  code = mndSetDbCfgFromAlterDbReq(&dbObj, &alterReq);
267✔
1368
  if (code != 0) {
267✔
1369
    if (code == TSDB_CODE_MND_DB_OPTION_UNCHANGED) code = 0;
26!
1370
    goto _OVER;
26✔
1371
  }
1372

1373
  TAOS_CHECK_GOTO(mndCheckInChangeDbCfg(pMnode, &pDb->cfg, &dbObj.cfg), NULL, _OVER);
241✔
1374

1375
  dbObj.cfgVersion++;
240✔
1376
  dbObj.updateTime = taosGetTimestampMs();
240✔
1377
  code = mndAlterDb(pMnode, pReq, pDb, &dbObj);
240✔
1378

1379
  if (dbObj.cfg.replications != pDb->cfg.replications) {
240✔
1380
    // return quickly, operation executed asynchronously
1381
    mInfo("db:%s, alter db replica from %d to %d", pDb->name, pDb->cfg.replications, dbObj.cfg.replications);
5!
1382
  } else {
1383
    if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
235!
1384
  }
1385

1386
  SName name = {0};
240✔
1387
  if (tNameFromString(&name, alterReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
240!
1388
    mError("db:%s, failed to parse db name", alterReq.db);
×
1389

1390
  auditRecord(pReq, pMnode->clusterId, "alterDB", name.dbname, "", alterReq.sql, alterReq.sqlLen);
240✔
1391

1392
_OVER:
268✔
1393
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
268✔
1394
    if (terrno != 0) code = terrno;
3!
1395
    mError("db:%s, failed to alter since %s", alterReq.db, tstrerror(code));
3!
1396
  }
1397

1398
  mndReleaseDb(pMnode, pDb);
268✔
1399
  taosArrayDestroy(dbObj.cfg.pRetensions);
268✔
1400
  tFreeSAlterDbReq(&alterReq);
268✔
1401

1402
  TAOS_RETURN(code);
268✔
1403
}
1404

1405
static void mndDumpDbCfgInfo(SDbCfgRsp *cfgRsp, SDbObj *pDb) {
82,028✔
1406
  tstrncpy(cfgRsp->db, pDb->name, sizeof(cfgRsp->db));
82,028✔
1407
  cfgRsp->dbId = pDb->uid;
82,028✔
1408
  cfgRsp->cfgVersion = pDb->cfgVersion;
82,028✔
1409
  cfgRsp->numOfVgroups = pDb->cfg.numOfVgroups;
82,028✔
1410
  cfgRsp->numOfStables = pDb->cfg.numOfStables;
82,028✔
1411
  cfgRsp->buffer = pDb->cfg.buffer;
82,028✔
1412
  cfgRsp->cacheSize = pDb->cfg.cacheLastSize;
82,028✔
1413
  cfgRsp->pageSize = pDb->cfg.pageSize;
82,028✔
1414
  cfgRsp->pages = pDb->cfg.pages;
82,028✔
1415
  cfgRsp->daysPerFile = pDb->cfg.daysPerFile;
82,028✔
1416
  cfgRsp->daysToKeep0 = pDb->cfg.daysToKeep0;
82,028✔
1417
  cfgRsp->daysToKeep1 = pDb->cfg.daysToKeep1;
82,028✔
1418
  cfgRsp->daysToKeep2 = pDb->cfg.daysToKeep2;
82,028✔
1419
  cfgRsp->keepTimeOffset = pDb->cfg.keepTimeOffset;
82,028✔
1420
  cfgRsp->minRows = pDb->cfg.minRows;
82,028✔
1421
  cfgRsp->maxRows = pDb->cfg.maxRows;
82,028✔
1422
  cfgRsp->walFsyncPeriod = pDb->cfg.walFsyncPeriod;
82,028✔
1423
  cfgRsp->hashPrefix = pDb->cfg.hashPrefix;
82,028✔
1424
  cfgRsp->hashSuffix = pDb->cfg.hashSuffix;
82,028✔
1425
  cfgRsp->hashMethod = pDb->cfg.hashMethod;
82,028✔
1426
  cfgRsp->walLevel = pDb->cfg.walLevel;
82,028✔
1427
  cfgRsp->precision = pDb->cfg.precision;
82,028✔
1428
  cfgRsp->compression = pDb->cfg.compression;
82,028✔
1429
  cfgRsp->replications = pDb->cfg.replications;
82,028✔
1430
  cfgRsp->strict = pDb->cfg.strict;
82,028✔
1431
  cfgRsp->cacheLast = pDb->cfg.cacheLast;
82,028✔
1432
  cfgRsp->tsdbPageSize = pDb->cfg.tsdbPageSize;
82,028✔
1433
  cfgRsp->walRetentionPeriod = pDb->cfg.walRetentionPeriod;
82,028✔
1434
  cfgRsp->walRollPeriod = pDb->cfg.walRollPeriod;
82,028✔
1435
  cfgRsp->walRetentionSize = pDb->cfg.walRetentionSize;
82,028✔
1436
  cfgRsp->walSegmentSize = pDb->cfg.walSegmentSize;
82,028✔
1437
  cfgRsp->numOfRetensions = pDb->cfg.numOfRetensions;
82,028✔
1438
  cfgRsp->pRetensions = taosArrayDup(pDb->cfg.pRetensions, NULL);
82,028✔
1439
  cfgRsp->schemaless = pDb->cfg.schemaless;
82,028✔
1440
  cfgRsp->sstTrigger = pDb->cfg.sstTrigger;
82,028✔
1441
  cfgRsp->s3ChunkSize = pDb->cfg.s3ChunkSize;
82,028✔
1442
  cfgRsp->s3KeepLocal = pDb->cfg.s3KeepLocal;
82,028✔
1443
  cfgRsp->s3Compact = pDb->cfg.s3Compact;
82,028✔
1444
  cfgRsp->withArbitrator = pDb->cfg.withArbitrator;
82,028✔
1445
  cfgRsp->encryptAlgorithm = pDb->cfg.encryptAlgorithm;
82,028✔
1446
  cfgRsp->compactInterval = pDb->cfg.compactInterval;
82,028✔
1447
  cfgRsp->compactStartTime = pDb->cfg.compactStartTime;
82,028✔
1448
  cfgRsp->compactEndTime = pDb->cfg.compactEndTime;
82,028✔
1449
  cfgRsp->compactTimeOffset = pDb->cfg.compactTimeOffset;
82,028✔
1450
}
82,028✔
1451

1452
static int32_t mndProcessGetDbCfgReq(SRpcMsg *pReq) {
74,386✔
1453
  SMnode   *pMnode = pReq->info.node;
74,386✔
1454
  int32_t   code = -1;
74,386✔
1455
  SDbObj   *pDb = NULL;
74,386✔
1456
  SDbCfgReq cfgReq = {0};
74,386✔
1457
  SDbCfgRsp cfgRsp = {0};
74,386✔
1458

1459
  TAOS_CHECK_GOTO(tDeserializeSDbCfgReq(pReq->pCont, pReq->contLen, &cfgReq), NULL, _OVER);
74,386!
1460

1461
  if (strcasecmp(cfgReq.db, TSDB_INFORMATION_SCHEMA_DB) && strcasecmp(cfgReq.db, TSDB_PERFORMANCE_SCHEMA_DB)) {
74,386!
1462
    pDb = mndAcquireDb(pMnode, cfgReq.db);
74,386✔
1463
    if (pDb == NULL) {
74,386✔
1464
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
836✔
1465
      if (terrno != 0) code = terrno;
836!
1466
      goto _OVER;
836✔
1467
    }
1468

1469
    mndDumpDbCfgInfo(&cfgRsp, pDb);
73,550✔
1470
  }
1471

1472
  int32_t contLen = tSerializeSDbCfgRsp(NULL, 0, &cfgRsp);
73,550✔
1473
  void   *pRsp = rpcMallocCont(contLen);
73,549✔
1474
  if (pRsp == NULL) {
73,549!
1475
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1476
    if (terrno != 0) code = terrno;
×
1477
    goto _OVER;
×
1478
  }
1479

1480
  int32_t ret = 0;
73,549✔
1481
  if ((ret = tSerializeSDbCfgRsp(pRsp, contLen, &cfgRsp)) < 0) {
73,549!
1482
    code = ret;
×
1483
    goto _OVER;
×
1484
  }
1485

1486
  pReq->info.rsp = pRsp;
73,550✔
1487
  pReq->info.rspLen = contLen;
73,550✔
1488

1489
  code = 0;
73,550✔
1490

1491
_OVER:
74,386✔
1492

1493
  tFreeSDbCfgRsp(&cfgRsp);
74,386✔
1494

1495
  if (code != 0) {
74,385✔
1496
    mError("db:%s, failed to get cfg since %s", cfgReq.db, terrstr());
836!
1497
  }
1498

1499
  mndReleaseDb(pMnode, pDb);
74,385✔
1500

1501
  TAOS_RETURN(code);
74,386✔
1502
}
1503

1504
static int32_t mndSetDropDbPrepareLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,973✔
1505
  int32_t  code = 0;
1,973✔
1506
  SSdbRaw *pRedoRaw = mndDbActionEncode(pDb);
1,973✔
1507
  if (pRedoRaw == NULL) {
1,973!
1508
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1509
    if (terrno != 0) code = terrno;
×
1510
    return -1;
×
1511
  }
1512
  TAOS_CHECK_RETURN(mndTransAppendPrepareLog(pTrans, pRedoRaw));
1,973!
1513
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
1,973!
1514

1515
  SSdb *pSdb = pMnode->pSdb;
1,973✔
1516
  void *pIter = NULL;
1,973✔
1517

1518
  while (1) {
6✔
1519
    SArbGroup *pArbGroup = NULL;
1,979✔
1520
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
1,979✔
1521
    if (pIter == NULL) break;
1,979✔
1522

1523
    if (pArbGroup->dbUid == pDb->uid) {
6!
1524
      if ((code = mndSetDropArbGroupPrepareLogs(pTrans, pArbGroup)) != 0) {
6!
1525
        sdbCancelFetch(pSdb, pIter);
×
1526
        sdbRelease(pSdb, pArbGroup);
×
1527
        TAOS_RETURN(code);
×
1528
      }
1529
    }
1530

1531
    sdbRelease(pSdb, pArbGroup);
6✔
1532
  }
1533

1534
  TAOS_RETURN(code);
1,973✔
1535
}
1536

1537
static int32_t mndSetDropDbCommitLogs(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,973✔
1538
  int32_t  code = 0;
1,973✔
1539
  SSdbRaw *pCommitRaw = mndDbActionEncode(pDb);
1,973✔
1540
  if (pCommitRaw == NULL) {
1,973!
1541
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1542
    if (terrno != 0) code = terrno;
×
1543
    return -1;
×
1544
  }
1545
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
1,973!
1546
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
1,973!
1547

1548
  SSdb *pSdb = pMnode->pSdb;
1,973✔
1549
  void *pIter = NULL;
1,973✔
1550

1551
  while (1) {
6✔
1552
    SArbGroup *pArbGroup = NULL;
1,979✔
1553
    pIter = sdbFetch(pSdb, SDB_ARBGROUP, pIter, (void **)&pArbGroup);
1,979✔
1554
    if (pIter == NULL) break;
1,979✔
1555

1556
    if (pArbGroup->dbUid == pDb->uid) {
6!
1557
      if ((code = mndSetDropArbGroupCommitLogs(pTrans, pArbGroup)) != 0) {
6!
1558
        sdbCancelFetch(pSdb, pIter);
×
1559
        sdbRelease(pSdb, pArbGroup);
×
1560
        TAOS_RETURN(code);
×
1561
      }
1562
    }
1563

1564
    sdbRelease(pSdb, pArbGroup);
6✔
1565
  }
1566

1567
  while (1) {
15,292✔
1568
    SVgObj *pVgroup = NULL;
17,265✔
1569
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
17,265✔
1570
    if (pIter == NULL) break;
17,265✔
1571

1572
    if (pVgroup->dbUid == pDb->uid) {
15,292✔
1573
      SSdbRaw *pVgRaw = mndVgroupActionEncode(pVgroup);
4,441✔
1574
      if (pVgRaw == NULL) {
4,441!
1575
        sdbCancelFetch(pSdb, pIter);
×
1576
        sdbRelease(pSdb, pVgroup);
×
1577
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1578
        if (terrno != 0) code = terrno;
×
1579
        TAOS_RETURN(code);
×
1580
      }
1581
      if ((code = mndTransAppendCommitlog(pTrans, pVgRaw)) != 0) {
4,441!
1582
        sdbCancelFetch(pSdb, pIter);
×
1583
        sdbRelease(pSdb, pVgroup);
×
1584
        TAOS_RETURN(code);
×
1585
      }
1586
      if ((code = sdbSetRawStatus(pVgRaw, SDB_STATUS_DROPPED)) != 0) {
4,441!
1587
        sdbCancelFetch(pSdb, pIter);
×
1588
        sdbRelease(pSdb, pVgroup);
×
1589
        TAOS_RETURN(code);
×
1590
      }
1591
    }
1592

1593
    sdbRelease(pSdb, pVgroup);
15,292✔
1594
  }
1595

1596
  while (1) {
22,004✔
1597
    SStbObj *pStb = NULL;
23,977✔
1598
    pIter = sdbFetch(pSdb, SDB_STB, pIter, (void **)&pStb);
23,977✔
1599
    if (pIter == NULL) break;
23,977✔
1600

1601
    if (pStb->dbUid == pDb->uid) {
22,004✔
1602
      SSdbRaw *pStbRaw = mndStbActionEncode(pStb);
4,221✔
1603
      if (pStbRaw == NULL) {
4,221!
1604
        sdbCancelFetch(pSdb, pIter);
×
1605
        sdbRelease(pSdb, pStbRaw);
×
1606
        code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1607
        if (terrno != 0) code = terrno;
×
1608
        return -1;
×
1609
      }
1610
      if ((code = mndTransAppendCommitlog(pTrans, pStbRaw)) != 0) {
4,221!
1611
        sdbCancelFetch(pSdb, pIter);
×
1612
        sdbRelease(pSdb, pStbRaw);
×
1613
        return -1;
×
1614
      }
1615
      if ((code = sdbSetRawStatus(pStbRaw, SDB_STATUS_DROPPED)) != 0) {
4,221!
1616
        sdbCancelFetch(pSdb, pIter);
×
1617
        sdbRelease(pSdb, pStbRaw);
×
1618
        return -1;
×
1619
      }
1620
    }
1621

1622
    sdbRelease(pSdb, pStb);
22,004✔
1623
  }
1624

1625
  TAOS_RETURN(code);
1,973✔
1626
}
1627

1628
static int32_t mndBuildDropVgroupAction(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SVgObj *pVgroup) {
4,438✔
1629
  int32_t code = 0;
4,438✔
1630
  for (int32_t vn = 0; vn < pVgroup->replica; ++vn) {
8,996✔
1631
    SVnodeGid *pVgid = pVgroup->vnodeGid + vn;
4,558✔
1632
    TAOS_CHECK_RETURN(mndAddDropVnodeAction(pMnode, pTrans, pDb, pVgroup, pVgid, true));
4,558!
1633
  }
1634

1635
  TAOS_RETURN(code);
4,438✔
1636
}
1637

1638
static int32_t mndSetDropDbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj *pDb) {
1,972✔
1639
  int32_t code = 0;
1,972✔
1640
  SSdb   *pSdb = pMnode->pSdb;
1,972✔
1641
  void   *pIter = NULL;
1,972✔
1642

1643
  while (1) {
15,283✔
1644
    SVgObj *pVgroup = NULL;
17,255✔
1645
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
17,255✔
1646
    if (pIter == NULL) break;
17,255✔
1647

1648
    if (pVgroup->dbUid == pDb->uid) {
15,283✔
1649
      if ((code = mndBuildDropVgroupAction(pMnode, pTrans, pDb, pVgroup)) != 0) {
4,438!
1650
        sdbCancelFetch(pSdb, pIter);
×
1651
        sdbRelease(pSdb, pVgroup);
×
1652
        TAOS_RETURN(code);
×
1653
      }
1654
    }
1655

1656
    sdbRelease(pSdb, pVgroup);
15,283✔
1657
  }
1658

1659
  TAOS_RETURN(code);
1,972✔
1660
}
1661

1662
static int32_t mndBuildDropDbRsp(SDbObj *pDb, int32_t *pRspLen, void **ppRsp, bool useRpcMalloc) {
4,011✔
1663
  int32_t    code = 0;
4,011✔
1664
  SDropDbRsp dropRsp = {0};
4,011✔
1665
  if (pDb != NULL) {
4,011✔
1666
    (void)memcpy(dropRsp.db, pDb->name, TSDB_DB_FNAME_LEN);
1,972✔
1667
    dropRsp.uid = pDb->uid;
1,972✔
1668
  }
1669

1670
  int32_t rspLen = tSerializeSDropDbRsp(NULL, 0, &dropRsp);
4,011✔
1671
  void   *pRsp = NULL;
4,011✔
1672
  if (useRpcMalloc) {
4,011✔
1673
    pRsp = rpcMallocCont(rspLen);
2,039✔
1674
  } else {
1675
    pRsp = taosMemoryMalloc(rspLen);
1,972!
1676
  }
1677

1678
  if (pRsp == NULL) {
4,011!
1679
    code = terrno;
×
1680
    TAOS_RETURN(code);
×
1681
  }
1682

1683
  int32_t ret = 0;
4,011✔
1684
  if ((ret = tSerializeSDropDbRsp(pRsp, rspLen, &dropRsp)) < 0) return ret;
4,011!
1685
  *pRspLen = rspLen;
4,011✔
1686
  *ppRsp = pRsp;
4,011✔
1687
  TAOS_RETURN(code);
4,011✔
1688
}
1689

1690
static int32_t mndDropDb(SMnode *pMnode, SRpcMsg *pReq, SDbObj *pDb) {
1,974✔
1691
  int32_t code = -1;
1,974✔
1692

1693
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_DB, pReq, "drop-db");
1,974✔
1694
  if (pTrans == NULL) {
1,974!
1695
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1696
    if (terrno != 0) code = terrno;
×
1697
    goto _OVER;
×
1698
  }
1699

1700
  mInfo("trans:%d start to drop db:%s", pTrans->id, pDb->name);
1,974!
1701

1702
  mndTransSetDbName(pTrans, pDb->name, NULL);
1,974✔
1703
  TAOS_CHECK_GOTO(mndTransCheckConflict(pMnode, pTrans), NULL, _OVER);
1,974!
1704

1705
  if (mndTopicExistsForDb(pMnode, pDb)) {
1,974✔
1706
    code = TSDB_CODE_MND_TOPIC_MUST_BE_DELETED;
1✔
1707
    goto _OVER;
1✔
1708
  }
1709

1710
  TAOS_CHECK_GOTO(mndSetDropDbPrepareLogs(pMnode, pTrans, pDb), NULL, _OVER);
1,973!
1711
  TAOS_CHECK_GOTO(mndSetDropDbCommitLogs(pMnode, pTrans, pDb), NULL, _OVER);
1,973!
1712
  TAOS_CHECK_GOTO(mndDropStreamByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,973✔
1713
#ifdef TD_ENTERPRISE
1714
  TAOS_CHECK_GOTO(mndDropViewByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,972!
1715
#endif
1716
  TAOS_CHECK_GOTO(mndDropSmasByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,972!
1717
  TAOS_CHECK_GOTO(mndDropIdxsByDb(pMnode, pTrans, pDb), NULL, _OVER);
1,972!
1718
  TAOS_CHECK_GOTO(mndStreamSetStopStreamTasksActions(pMnode, pTrans, pDb->uid), NULL, _OVER);
1,972!
1719
  TAOS_CHECK_GOTO(mndSetDropDbRedoActions(pMnode, pTrans, pDb), NULL, _OVER);
1,972!
1720
  TAOS_CHECK_GOTO(mndUserRemoveDb(pMnode, pTrans, pDb->name), NULL, _OVER);
1,972!
1721

1722
  int32_t rspLen = 0;
1,972✔
1723
  void   *pRsp = NULL;
1,972✔
1724
  TAOS_CHECK_GOTO(mndBuildDropDbRsp(pDb, &rspLen, &pRsp, false), NULL, _OVER);
1,972!
1725
  mndTransSetRpcRsp(pTrans, pRsp, rspLen);
1,972✔
1726

1727
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,972!
1728
  code = 0;
1,972✔
1729

1730
_OVER:
1,974✔
1731
  mndTransDrop(pTrans);
1,974✔
1732
  TAOS_RETURN(code);
1,974✔
1733
}
1734

1735
static int32_t mndProcessDropDbReq(SRpcMsg *pReq) {
4,094✔
1736
  SMnode    *pMnode = pReq->info.node;
4,094✔
1737
  int32_t    code = -1;
4,094✔
1738
  SDbObj    *pDb = NULL;
4,094✔
1739
  SDropDbReq dropReq = {0};
4,094✔
1740

1741
  TAOS_CHECK_GOTO(tDeserializeSDropDbReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
4,094!
1742

1743
  mInfo("db:%s, start to drop", dropReq.db);
4,094!
1744

1745
  pDb = mndAcquireDb(pMnode, dropReq.db);
4,094✔
1746
  if (pDb == NULL) {
4,094✔
1747
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
2,104✔
1748
    if (terrno != 0) code = terrno;
2,104!
1749
    if (dropReq.ignoreNotExists) {
2,104✔
1750
      code = mndBuildDropDbRsp(pDb, &pReq->info.rspLen, &pReq->info.rsp, true);
2,039✔
1751
    }
1752
    goto _OVER;
2,104✔
1753
  }
1754

1755
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_DB, pDb), NULL, _OVER);
1,990✔
1756

1757
  SSdb *pSdb = pMnode->pSdb;
1,975✔
1758
  void *pIter = NULL;
1,975✔
1759

1760
  while (1) {
15,294✔
1761
    SVgObj *pVgroup = NULL;
17,269✔
1762
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
17,269✔
1763
    if (pIter == NULL) break;
17,269✔
1764

1765
    if (pVgroup->dbUid == pDb->uid) {
15,295✔
1766
      bool isFound = false;
4,443✔
1767
      for (int32_t i = 0; i < pVgroup->replica; i++) {
9,005✔
1768
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_OFFLINE) {
4,563✔
1769
          isFound = true;
1✔
1770
          break;
1✔
1771
        }
1772
      }
1773
      if (!isFound) {
4,443✔
1774
        sdbRelease(pSdb, pVgroup);
4,442✔
1775
        continue;
4,442✔
1776
      }
1777
      code = TSDB_CODE_MND_VGROUP_OFFLINE;
1✔
1778
      sdbCancelFetch(pSdb, pIter);
1✔
1779
      sdbRelease(pSdb, pVgroup);
1✔
1780
      goto _OVER;
1✔
1781
    }
1782

1783
    sdbRelease(pSdb, pVgroup);
10,852✔
1784
  }
1785

1786
  code = mndDropDb(pMnode, pReq, pDb);
1,974✔
1787
  if (code == TSDB_CODE_SUCCESS) {
1,974✔
1788
    code = TSDB_CODE_ACTION_IN_PROGRESS;
1,972✔
1789
  }
1790

1791
  SName name = {0};
1,974✔
1792
  if (tNameFromString(&name, dropReq.db, T_NAME_ACCT | T_NAME_DB) < 0)
1,974!
1793
    mError("db:%s, failed to parse db name", dropReq.db);
×
1794

1795
  auditRecord(pReq, pMnode->clusterId, "dropDB", name.dbname, "", dropReq.sql, dropReq.sqlLen);
1,974✔
1796

1797
_OVER:
4,094✔
1798
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
4,094✔
1799
    mError("db:%s, failed to drop since %s", dropReq.db, terrstr());
83!
1800
  }
1801

1802
  mndReleaseDb(pMnode, pDb);
4,094✔
1803
  tFreeSDropDbReq(&dropReq);
4,094✔
1804
  TAOS_RETURN(code);
4,094✔
1805
}
1806

1807
static int32_t mndGetDBTableNum(SDbObj *pDb, SMnode *pMnode) {
2,813,740✔
1808
  int32_t numOfTables = 0;
2,813,740✔
1809
  int32_t vindex = 0;
2,813,740✔
1810
  SSdb   *pSdb = pMnode->pSdb;
2,813,740✔
1811

1812
  void *pIter = NULL;
2,813,740✔
1813
  while (vindex < pDb->cfg.numOfVgroups) {
81,483,256✔
1814
    SVgObj *pVgroup = NULL;
78,668,637✔
1815
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
78,668,637✔
1816
    if (pIter == NULL) break;
78,667,229!
1817

1818
    if (mndVgroupInDb(pVgroup, pDb->uid)) {
78,667,229✔
1819
      numOfTables += pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
5,652,582✔
1820
      vindex++;
5,652,582✔
1821
    }
1822

1823
    sdbRelease(pSdb, pVgroup);
78,666,779✔
1824
  }
1825

1826
  sdbCancelFetch(pSdb, pIter);
2,814,619✔
1827
  return numOfTables;
2,813,784✔
1828
}
1829

1830
void mndBuildDBVgroupInfo(SDbObj *pDb, SMnode *pMnode, SArray *pVgList) {
2,989,621✔
1831
  int32_t vindex = 0;
2,989,621✔
1832
  SSdb   *pSdb = pMnode->pSdb;
2,989,621✔
1833

1834
  void *pIter = NULL;
2,989,621✔
1835
  while (1) {
86,118,762✔
1836
    SVgObj *pVgroup = NULL;
89,108,383✔
1837
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
89,108,383✔
1838
    if (pIter == NULL) break;
89,105,954✔
1839

1840
    if ((NULL == pDb || pVgroup->dbUid == pDb->uid) && !pVgroup->isTsma) {
88,815,615!
1841
      SVgroupInfo vgInfo = {0};
18,649,324✔
1842
      vgInfo.vgId = pVgroup->vgId;
18,649,324✔
1843
      vgInfo.hashBegin = pVgroup->hashBegin;
18,649,324✔
1844
      vgInfo.hashEnd = pVgroup->hashEnd;
18,649,324✔
1845
      vgInfo.numOfTable = pVgroup->numOfTables / TSDB_TABLE_NUM_UNIT;
18,649,324✔
1846
      vgInfo.epSet.numOfEps = pVgroup->replica;
18,649,324✔
1847
      for (int32_t gid = 0; gid < pVgroup->replica; ++gid) {
37,309,826✔
1848
        SVnodeGid *pVgid = &pVgroup->vnodeGid[gid];
18,660,407✔
1849
        SEp       *pEp = &vgInfo.epSet.eps[gid];
18,660,407✔
1850
        SDnodeObj *pDnode = mndAcquireDnode(pMnode, pVgid->dnodeId);
18,660,407✔
1851
        if (pDnode != NULL) {
18,660,352✔
1852
          (void)memcpy(pEp->fqdn, pDnode->fqdn, TSDB_FQDN_LEN);
18,660,240✔
1853
          pEp->port = pDnode->port;
18,660,240✔
1854
        }
1855
        mndReleaseDnode(pMnode, pDnode);
18,660,352✔
1856
        if (pVgid->syncState == TAOS_SYNC_STATE_LEADER || pVgid->syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
18,660,502✔
1857
          vgInfo.epSet.inUse = gid;
18,648,335✔
1858
        }
1859
      }
1860
      vindex++;
18,649,419✔
1861
      if (taosArrayPush(pVgList, &vgInfo) == NULL) {
18,649,321!
1862
        mError("db:%s, failed to push vgInfo to array, vgId:%d, but continue next", pDb->name, vgInfo.vgId);
×
1863
      }
1864
    }
1865

1866
    sdbRelease(pSdb, pVgroup);
88,815,613✔
1867

1868
    if (pDb && (vindex >= pDb->cfg.numOfVgroups)) {
88,818,058✔
1869
      sdbCancelFetch(pSdb, pIter);
2,699,296✔
1870
      break;
2,699,293✔
1871
    }
1872
  }
1873
}
2,989,632✔
1874

1875
int32_t mndExtractDbInfo(SMnode *pMnode, SDbObj *pDb, SUseDbRsp *pRsp, const SUseDbReq *pReq) {
2,779,408✔
1876
  int32_t code = 0;
2,779,408✔
1877
  pRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
2,779,408✔
1878
  if (pRsp->pVgroupInfos == NULL) {
2,779,367!
1879
    code = terrno;
×
1880
    TAOS_RETURN(code);
×
1881
  }
1882

1883
  int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
2,779,367✔
1884

1885
  if (pReq == NULL || pReq->vgVersion < pDb->vgVersion || pReq->dbId != pDb->uid || numOfTable != pReq->numOfTable ||
2,779,410!
1886
      pReq->stateTs < pDb->stateTs) {
83,186✔
1887
    mndBuildDBVgroupInfo(pDb, pMnode, pRsp->pVgroupInfos);
2,696,622✔
1888
  }
1889

1890
  (void)memcpy(pRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
2,779,408✔
1891
  pRsp->uid = pDb->uid;
2,779,408✔
1892
  pRsp->vgVersion = pDb->vgVersion;
2,779,408✔
1893
  pRsp->stateTs = pDb->stateTs;
2,779,408✔
1894
  pRsp->vgNum = taosArrayGetSize(pRsp->pVgroupInfos);
2,779,408✔
1895
  pRsp->hashMethod = pDb->cfg.hashMethod;
2,779,397✔
1896
  pRsp->hashPrefix = pDb->cfg.hashPrefix;
2,779,397✔
1897
  pRsp->hashSuffix = pDb->cfg.hashSuffix;
2,779,397✔
1898
  TAOS_RETURN(code);
2,779,397✔
1899
}
1900

1901
static int32_t mndProcessUseDbReq(SRpcMsg *pReq) {
3,062,957✔
1902
  SMnode   *pMnode = pReq->info.node;
3,062,957✔
1903
  int32_t   code = -1;
3,062,957✔
1904
  SDbObj   *pDb = NULL;
3,062,957✔
1905
  SUseDbReq usedbReq = {0};
3,062,957✔
1906
  SUseDbRsp usedbRsp = {0};
3,062,957✔
1907

1908
  TAOS_CHECK_GOTO(tDeserializeSUseDbReq(pReq->pCont, pReq->contLen, &usedbReq), NULL, _OVER);
3,062,957!
1909

1910
  char *p = strchr(usedbReq.db, '.');
3,063,040✔
1911
  if (p && ((0 == strcmp(p + 1, TSDB_INFORMATION_SCHEMA_DB) || (0 == strcmp(p + 1, TSDB_PERFORMANCE_SCHEMA_DB))))) {
3,063,040✔
1912
    (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
289,127✔
1913
    int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
289,127✔
1914
    if (usedbReq.vgVersion < vgVersion) {
289,143✔
1915
      usedbRsp.pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
289,111✔
1916
      if (usedbRsp.pVgroupInfos == NULL) goto _OVER;
289,109!
1917

1918
      mndBuildDBVgroupInfo(NULL, pMnode, usedbRsp.pVgroupInfos);
289,109✔
1919
      usedbRsp.vgVersion = vgVersion++;
289,111✔
1920
    } else {
1921
      usedbRsp.vgVersion = usedbReq.vgVersion;
32✔
1922
    }
1923
    usedbRsp.vgNum = taosArrayGetSize(usedbRsp.pVgroupInfos);
289,143✔
1924
    code = 0;
289,143✔
1925
  } else {
1926
    pDb = mndAcquireDb(pMnode, usedbReq.db);
2,773,913✔
1927
    if (pDb == NULL) {
2,773,910✔
1928
      (void)memcpy(usedbRsp.db, usedbReq.db, TSDB_DB_FNAME_LEN);
705✔
1929
      usedbRsp.uid = usedbReq.dbId;
705✔
1930
      usedbRsp.vgVersion = usedbReq.vgVersion;
705✔
1931
      usedbRsp.errCode = terrno;
705✔
1932

1933
      code = TSDB_CODE_MND_RETURN_VALUE_NULL;
705✔
1934
      if (terrno != 0) code = terrno;
705!
1935
      goto _OVER;
705✔
1936
    } else {
1937
      TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_USE_DB, pDb), NULL, _OVER);
2,773,205✔
1938

1939
      TAOS_CHECK_GOTO(mndExtractDbInfo(pMnode, pDb, &usedbRsp, &usedbReq), NULL, _OVER);
2,773,211!
1940

1941
      mDebug("db:%s, process usedb req vgVersion:%d stateTs:%" PRId64 ", rsp vgVersion:%d stateTs:%" PRId64,
2,773,194✔
1942
             usedbReq.db, usedbReq.vgVersion, usedbReq.stateTs, usedbRsp.vgVersion, usedbRsp.stateTs);
1943
      code = 0;
2,773,194✔
1944
    }
1945
  }
1946

1947
  int32_t contLen = tSerializeSUseDbRsp(NULL, 0, &usedbRsp);
3,062,337✔
1948
  void   *pRsp = rpcMallocCont(contLen);
3,062,325✔
1949
  if (pRsp == NULL) {
3,062,299!
1950
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1951
    code = -1;
×
1952
    goto _OVER;
×
1953
  }
1954

1955
  int32_t ret = 0;
3,062,299✔
1956
  if ((ret = tSerializeSUseDbRsp(pRsp, contLen, &usedbRsp)) < 0) {
3,062,299!
1957
    code = ret;
×
1958
    goto _OVER;
×
1959
  }
1960

1961
  pReq->info.rsp = pRsp;
3,062,341✔
1962
  pReq->info.rspLen = contLen;
3,062,341✔
1963

1964
_OVER:
3,063,063✔
1965
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
3,063,063!
1966
    mError("db:%s, failed to process use db req since %s", usedbReq.db, tstrerror(code));
722!
1967
  }
1968

1969
  mndReleaseDb(pMnode, pDb);
3,063,063✔
1970
  tFreeSUsedbRsp(&usedbRsp);
3,063,075✔
1971

1972
  TAOS_RETURN(code);
3,063,071✔
1973
}
1974

1975
int32_t mndValidateDbInfo(SMnode *pMnode, SDbCacheInfo *pDbs, int32_t numOfDbs, void **ppRsp, int32_t *pRspLen) {
43,480✔
1976
  int32_t       code = 0;
43,480✔
1977
  SDbHbBatchRsp batchRsp = {0};
43,480✔
1978
  batchRsp.pArray = taosArrayInit(numOfDbs, sizeof(SDbHbRsp));
43,480✔
1979
  if (batchRsp.pArray == NULL) {
43,480!
1980
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1981
    TAOS_RETURN(code);
×
1982
  }
1983

1984
  for (int32_t i = 0; i < numOfDbs; ++i) {
90,366✔
1985
    SDbCacheInfo *pDbCacheInfo = &pDbs[i];
46,886✔
1986
    pDbCacheInfo->dbId = be64toh(pDbCacheInfo->dbId);
46,886✔
1987
    pDbCacheInfo->vgVersion = htonl(pDbCacheInfo->vgVersion);
46,885✔
1988
    pDbCacheInfo->cfgVersion = htonl(pDbCacheInfo->cfgVersion);
46,885✔
1989
    pDbCacheInfo->numOfTable = htonl(pDbCacheInfo->numOfTable);
46,885✔
1990
    pDbCacheInfo->stateTs = be64toh(pDbCacheInfo->stateTs);
46,885✔
1991
    pDbCacheInfo->tsmaVersion = htonl(pDbCacheInfo->tsmaVersion);
46,884✔
1992

1993
    SDbHbRsp rsp = {0};
46,884✔
1994
    (void)memcpy(rsp.db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
46,884✔
1995
    rsp.dbId = pDbCacheInfo->dbId;
46,884✔
1996

1997
    if ((0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_INFORMATION_SCHEMA_DB) ||
46,884✔
1998
         (0 == strcasecmp(pDbCacheInfo->dbFName, TSDB_PERFORMANCE_SCHEMA_DB)))) {
42,168✔
1999
      int32_t vgVersion = mndGetGlobalVgroupVersion(pMnode);
12,155✔
2000
      if (pDbCacheInfo->vgVersion >= vgVersion) {
12,157✔
2001
        continue;
33,952✔
2002
      }
2003

2004
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
1,228!
2005
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
1,228✔
2006
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(10, sizeof(SVgroupInfo));
1,228✔
2007

2008
      mndBuildDBVgroupInfo(NULL, pMnode, rsp.useDbRsp->pVgroupInfos);
1,228✔
2009
      rsp.useDbRsp->vgVersion = vgVersion++;
1,228✔
2010

2011
      rsp.useDbRsp->vgNum = taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
1,228✔
2012

2013
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
2,456!
2014
        if (terrno != 0) code = terrno;
×
2015
        return code;
×
2016
      }
2017

2018
      continue;
1,228✔
2019
    }
2020

2021
    SDbObj *pDb = mndAcquireDb(pMnode, pDbCacheInfo->dbFName);
34,729✔
2022
    if (pDb == NULL) {
34,729✔
2023
      mTrace("db:%s, no exist", pDbCacheInfo->dbFName);
357✔
2024
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
357!
2025
      (void)memcpy(rsp.useDbRsp->db, pDbCacheInfo->dbFName, TSDB_DB_FNAME_LEN);
357✔
2026
      rsp.useDbRsp->uid = pDbCacheInfo->dbId;
357✔
2027
      rsp.useDbRsp->vgVersion = -1;
357✔
2028
      if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
714!
2029
        if (terrno != 0) code = terrno;
×
2030
        return code;
×
2031
      }
2032
      continue;
357✔
2033
    }
2034

2035
    int32_t numOfTable = mndGetDBTableNum(pDb, pMnode);
34,372✔
2036

2037
    if (pDbCacheInfo->vgVersion >= pDb->vgVersion && pDbCacheInfo->cfgVersion >= pDb->cfgVersion &&
34,372✔
2038
        numOfTable == pDbCacheInfo->numOfTable && pDbCacheInfo->stateTs == pDb->stateTs &&
25,849!
2039
        pDbCacheInfo->tsmaVersion >= pDb->tsmaVersion) {
23,459✔
2040
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
21,438✔
2041
             " numOfTables:%d, not changed vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2042
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2043
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2044
      mndReleaseDb(pMnode, pDb);
21,438✔
2045
      continue;
21,438✔
2046
    } else {
2047
      mTrace("db:%s, valid dbinfo, vgVersion:%d cfgVersion:%d stateTs:%" PRId64
12,934✔
2048
             " numOfTables:%d, changed to vgVersion:%d cfgVersion:%d stateTs:%" PRId64 " numOfTables:%d",
2049
             pDbCacheInfo->dbFName, pDbCacheInfo->vgVersion, pDbCacheInfo->cfgVersion, pDbCacheInfo->stateTs,
2050
             pDbCacheInfo->numOfTable, pDb->vgVersion, pDb->cfgVersion, pDb->stateTs, numOfTable);
2051
    }
2052

2053
    if (pDbCacheInfo->cfgVersion < pDb->cfgVersion) {
12,934✔
2054
      rsp.cfgRsp = taosMemoryCalloc(1, sizeof(SDbCfgRsp));
8,478!
2055
      mndDumpDbCfgInfo(rsp.cfgRsp, pDb);
8,478✔
2056
    }
2057

2058
    if (pDbCacheInfo->tsmaVersion != pDb->tsmaVersion) {
12,934✔
2059
      rsp.pTsmaRsp = taosMemoryCalloc(1, sizeof(STableTSMAInfoRsp));
11,889!
2060
      if (rsp.pTsmaRsp) rsp.pTsmaRsp->pTsmas = taosArrayInit(4, POINTER_BYTES);
11,889!
2061
      if (rsp.pTsmaRsp && rsp.pTsmaRsp->pTsmas) {
11,889!
2062
        bool    exist = false;
11,889✔
2063
        int32_t code = mndGetDbTsmas(pMnode, 0, pDb->uid, rsp.pTsmaRsp, &exist);
11,889✔
2064
        if (TSDB_CODE_SUCCESS != code) {
11,889!
2065
          mndReleaseDb(pMnode, pDb);
×
2066
          if (code != TSDB_CODE_NEED_RETRY) {
×
2067
            mError("db:%s, failed to get db tsmas", pDb->name);
×
2068
          } else {
2069
            mWarn("db:%s, need retry to get db tsmas", pDb->name);
×
2070
          }
2071
          taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2072
          taosMemoryFreeClear(rsp.pTsmaRsp);
×
2073
          continue;
×
2074
        }
2075
        rsp.dbTsmaVersion = pDb->tsmaVersion;
11,889✔
2076
        mDebug("update tsma version to %d, got tsma num: %ld", pDb->tsmaVersion, rsp.pTsmaRsp->pTsmas->size);
11,889✔
2077
      }
2078
    }
2079

2080
    if (pDbCacheInfo->vgVersion < pDb->vgVersion || numOfTable != pDbCacheInfo->numOfTable ||
12,934!
2081
        pDbCacheInfo->stateTs != pDb->stateTs) {
12,887✔
2082
      rsp.useDbRsp = taosMemoryCalloc(1, sizeof(SUseDbRsp));
2,673!
2083
      rsp.useDbRsp->pVgroupInfos = taosArrayInit(pDb->cfg.numOfVgroups, sizeof(SVgroupInfo));
2,673✔
2084
      if (rsp.useDbRsp->pVgroupInfos == NULL) {
2,673!
2085
        mndReleaseDb(pMnode, pDb);
×
2086
        mError("db:%s, failed to malloc usedb response", pDb->name);
×
2087
        taosArrayDestroyP(rsp.pTsmaRsp->pTsmas, tFreeAndClearTableTSMAInfo);
×
2088
        taosMemoryFreeClear(rsp.pTsmaRsp);
×
2089
        continue;
×
2090
      }
2091

2092
      mndBuildDBVgroupInfo(pDb, pMnode, rsp.useDbRsp->pVgroupInfos);
2,673✔
2093
      (void)memcpy(rsp.useDbRsp->db, pDb->name, TSDB_DB_FNAME_LEN);
2,673✔
2094
      rsp.useDbRsp->uid = pDb->uid;
2,673✔
2095
      rsp.useDbRsp->vgVersion = pDb->vgVersion;
2,673✔
2096
      rsp.useDbRsp->stateTs = pDb->stateTs;
2,673✔
2097
      rsp.useDbRsp->vgNum = (int32_t)taosArrayGetSize(rsp.useDbRsp->pVgroupInfos);
2,673✔
2098
      rsp.useDbRsp->hashMethod = pDb->cfg.hashMethod;
2,673✔
2099
      rsp.useDbRsp->hashPrefix = pDb->cfg.hashPrefix;
2,673✔
2100
      rsp.useDbRsp->hashSuffix = pDb->cfg.hashSuffix;
2,673✔
2101
    }
2102

2103
    if (taosArrayPush(batchRsp.pArray, &rsp) == NULL) {
25,868!
2104
      mndReleaseDb(pMnode, pDb);
×
2105
      if (terrno != 0) code = terrno;
×
2106
      return code;
×
2107
    }
2108
    mndReleaseDb(pMnode, pDb);
12,934✔
2109
  }
2110

2111
  int32_t rspLen = tSerializeSDbHbBatchRsp(NULL, 0, &batchRsp);
43,480✔
2112
  void   *pRsp = taosMemoryMalloc(rspLen);
43,480!
2113
  if (pRsp == NULL) {
43,479!
2114
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
2115
    tFreeSDbHbBatchRsp(&batchRsp);
×
2116
    return -1;
×
2117
  }
2118
  int32_t ret = 0;
43,479✔
2119
  if ((ret = tSerializeSDbHbBatchRsp(pRsp, rspLen, &batchRsp)) < 0) return ret;
43,479!
2120

2121
  *ppRsp = pRsp;
43,480✔
2122
  *pRspLen = rspLen;
43,480✔
2123

2124
  tFreeSDbHbBatchRsp(&batchRsp);
43,480✔
2125
  TAOS_RETURN(code);
43,480✔
2126
}
2127

2128
static int32_t mndTrimDb(SMnode *pMnode, SDbObj *pDb) {
1✔
2129
  SSdb       *pSdb = pMnode->pSdb;
1✔
2130
  SVgObj     *pVgroup = NULL;
1✔
2131
  void       *pIter = NULL;
1✔
2132
  int32_t     code = 0;
1✔
2133
  SVTrimDbReq trimReq = {.timestamp = taosGetTimestampSec()};
1✔
2134
  int32_t     reqLen = tSerializeSVTrimDbReq(NULL, 0, &trimReq);
1✔
2135
  int32_t     contLen = reqLen + sizeof(SMsgHead);
1✔
2136

2137
  while (1) {
2✔
2138
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
3✔
2139
    if (pIter == NULL) break;
3✔
2140

2141
    SMsgHead *pHead = rpcMallocCont(contLen);
2✔
2142
    if (pHead == NULL) {
2!
2143
      sdbCancelFetch(pSdb, pVgroup);
×
2144
      sdbRelease(pSdb, pVgroup);
×
2145
      continue;
×
2146
    }
2147
    pHead->contLen = htonl(contLen);
2✔
2148
    pHead->vgId = htonl(pVgroup->vgId);
2✔
2149
    int32_t ret = 0;
2✔
2150
    if ((ret = tSerializeSVTrimDbReq((char *)pHead + sizeof(SMsgHead), contLen, &trimReq)) < 0) {
2!
2151
      sdbRelease(pSdb, pVgroup);
×
2152
      return ret;
×
2153
    }
2154

2155
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_TRIM, .pCont = pHead, .contLen = contLen};
2✔
2156
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
2✔
2157
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
2✔
2158
    if (code != 0) {
2!
2159
      mError("vgId:%d, failed to send vnode-trim request to vnode since 0x%x", pVgroup->vgId, code);
×
2160
    } else {
2161
      mInfo("vgId:%d, send vnode-trim request to vnode, time:%d", pVgroup->vgId, trimReq.timestamp);
2!
2162
    }
2163
    sdbRelease(pSdb, pVgroup);
2✔
2164
  }
2165

2166
  return 0;
1✔
2167
}
2168

2169
static int32_t mndProcessTrimDbReq(SRpcMsg *pReq) {
1✔
2170
  SMnode    *pMnode = pReq->info.node;
1✔
2171
  int32_t    code = -1;
1✔
2172
  SDbObj    *pDb = NULL;
1✔
2173
  STrimDbReq trimReq = {0};
1✔
2174

2175
  TAOS_CHECK_GOTO(tDeserializeSTrimDbReq(pReq->pCont, pReq->contLen, &trimReq), NULL, _OVER);
1!
2176

2177
  mInfo("db:%s, start to trim", trimReq.db);
1!
2178

2179
  pDb = mndAcquireDb(pMnode, trimReq.db);
1✔
2180
  if (pDb == NULL) {
1!
2181
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2182
    if (terrno != 0) code = terrno;
×
2183
    goto _OVER;
×
2184
  }
2185

2186
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
1!
2187

2188
  code = mndTrimDb(pMnode, pDb);
1✔
2189

2190
_OVER:
1✔
2191
  if (code != 0) {
1!
2192
    mError("db:%s, failed to process trim db req since %s", trimReq.db, terrstr());
×
2193
  }
2194

2195
  mndReleaseDb(pMnode, pDb);
1✔
2196
  TAOS_RETURN(code);
1✔
2197
}
2198

2199
static int32_t mndS3MigrateDb(SMnode *pMnode, SDbObj *pDb) {
×
2200
  SSdb            *pSdb = pMnode->pSdb;
×
2201
  SVgObj          *pVgroup = NULL;
×
2202
  void            *pIter = NULL;
×
2203
  SVS3MigrateDbReq s3migrateReq = {.timestamp = taosGetTimestampSec()};
×
2204
  int32_t          reqLen = tSerializeSVS3MigrateDbReq(NULL, 0, &s3migrateReq);
×
2205
  int32_t          contLen = reqLen + sizeof(SMsgHead);
×
2206
  int32_t          code = 0;
×
2207

2208
  while (1) {
×
2209
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
×
2210
    if (pIter == NULL) break;
×
2211

2212
    if (pVgroup->dbUid != pDb->uid) continue;
×
2213

2214
    SMsgHead *pHead = rpcMallocCont(contLen);
×
2215
    if (pHead == NULL) {
×
2216
      sdbCancelFetch(pSdb, pVgroup);
×
2217
      sdbRelease(pSdb, pVgroup);
×
2218
      continue;
×
2219
    }
2220
    pHead->contLen = htonl(contLen);
×
2221
    pHead->vgId = htonl(pVgroup->vgId);
×
2222
    int32_t ret = 0;
×
2223
    if ((ret = tSerializeSVS3MigrateDbReq((char *)pHead + sizeof(SMsgHead), contLen, &s3migrateReq)) < 0) {
×
2224
      sdbRelease(pSdb, pVgroup);
×
2225
      return ret;
×
2226
    }
2227

2228
    SRpcMsg rpcMsg = {.msgType = TDMT_VND_S3MIGRATE, .pCont = pHead, .contLen = contLen};
×
2229
    SEpSet  epSet = mndGetVgroupEpset(pMnode, pVgroup);
×
2230
    int32_t code = tmsgSendReq(&epSet, &rpcMsg);
×
2231
    if (code != 0) {
×
2232
      mError("vgId:%d, failed to send vnode-s3migrate request to vnode since 0x%x", pVgroup->vgId, code);
×
2233
    } else {
2234
      mInfo("vgId:%d, send vnode-s3migrate request to vnode, time:%d", pVgroup->vgId, s3migrateReq.timestamp);
×
2235
    }
2236
    sdbRelease(pSdb, pVgroup);
×
2237
  }
2238

2239
  return 0;
×
2240
}
2241

2242
static int32_t mndProcessS3MigrateDbReq(SRpcMsg *pReq) {
×
2243
  SMnode         *pMnode = pReq->info.node;
×
2244
  int32_t         code = -1;
×
2245
  SDbObj         *pDb = NULL;
×
2246
  SS3MigrateDbReq s3migrateReq = {0};
×
2247

2248
  TAOS_CHECK_GOTO(tDeserializeSS3MigrateDbReq(pReq->pCont, pReq->contLen, &s3migrateReq), NULL, _OVER);
×
2249

2250
  mInfo("db:%s, start to s3migrate", s3migrateReq.db);
×
2251

2252
  pDb = mndAcquireDb(pMnode, s3migrateReq.db);
×
2253
  if (pDb == NULL) {
×
2254
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2255
    if (terrno != 0) code = terrno;
×
2256
    goto _OVER;
×
2257
  }
2258

2259
  TAOS_CHECK_GOTO(mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_TRIM_DB, pDb), NULL, _OVER);
×
2260

2261
  code = mndS3MigrateDb(pMnode, pDb);
×
2262

2263
_OVER:
×
2264
  if (code != 0) {
×
2265
    mError("db:%s, failed to process s3migrate db req since %s", s3migrateReq.db, terrstr());
×
2266
  }
2267

2268
  mndReleaseDb(pMnode, pDb);
×
2269
  TAOS_RETURN(code);
×
2270
}
2271

2272
const char *mndGetDbStr(const char *src) {
1,871,391✔
2273
  char *pos = strstr(src, TS_PATH_DELIMITER);
1,871,391✔
2274
  if (pos != NULL) ++pos;
1,871,391✔
2275
  if (pos == NULL) return src;
1,871,391✔
2276
  return pos;
1,815,681✔
2277
}
2278

2279
const char *mndGetStableStr(const char *src) {
×
2280
  char *pos = strstr(src, TS_PATH_DELIMITER);
×
2281
  if (pos != NULL) ++pos;
×
2282
  if (pos == NULL) return src;
×
2283
  return mndGetDbStr(pos);
×
2284
}
2285

2286
static int64_t getValOfDiffPrecision(int8_t unit, int64_t val) {
6✔
2287
  int64_t v = 0;
6✔
2288
  switch (unit) {
6!
2289
    case 's':
2✔
2290
      v = val / 1000;
2✔
2291
      break;
2✔
2292
    case 'm':
1✔
2293
      v = val / tsTickPerMin[TSDB_TIME_PRECISION_MILLI];
1✔
2294
      break;
1✔
2295
    case 'h':
×
2296
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 60);
×
2297
      break;
×
2298
    case 'd':
3✔
2299
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60);
3✔
2300
      break;
3✔
2301
    case 'w':
×
2302
      v = val / (tsTickPerMin[TSDB_TIME_PRECISION_MILLI] * 24 * 60 * 7);
×
2303
      break;
×
2304
    default:
×
2305
      break;
×
2306
  }
2307

2308
  return v;
6✔
2309
}
2310

2311
static char *buildRetension(SArray *pRetension) {
637,678✔
2312
  size_t size = taosArrayGetSize(pRetension);
637,678✔
2313
  if (size == 0) {
637,685!
2314
    return NULL;
637,735✔
2315
  }
2316

2317
  char       *p1 = taosMemoryCalloc(1, 100);
×
2318
  SRetention *p = taosArrayGet(pRetension, 0);
1✔
2319

2320
  int32_t len = 2;
1✔
2321

2322
  int64_t v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
1✔
2323
  int64_t v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
1✔
2324
  len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
1✔
2325

2326
  if (size > 1) {
1!
2327
    len += tsnprintf(p1 + len, 100 - len, ",");
1✔
2328
    p = taosArrayGet(pRetension, 1);
1✔
2329

2330
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
1✔
2331
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
1✔
2332
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
1✔
2333
  }
2334

2335
  if (size > 2) {
1!
2336
    len += tsnprintf(p1 + len, 100 - len, ",");
1✔
2337
    p = taosArrayGet(pRetension, 2);
1✔
2338

2339
    v1 = getValOfDiffPrecision(p->freqUnit, p->freq);
1✔
2340
    v2 = getValOfDiffPrecision(p->keepUnit, p->keep);
1✔
2341
    len += tsnprintf(p1 + len, 100 - len, "%" PRId64 "%c:%" PRId64 "%c", v1, p->freqUnit, v2, p->keepUnit);
1✔
2342
  }
2343

2344
  varDataSetLen(p1, len);
1✔
2345
  return p1;
1✔
2346
}
2347

2348
static const char *getCacheModelStr(int8_t cacheModel) {
637,456✔
2349
  switch (cacheModel) {
637,456!
2350
    case TSDB_CACHE_MODEL_NONE:
297,028✔
2351
      return TSDB_CACHE_MODEL_NONE_STR;
297,028✔
2352
    case TSDB_CACHE_MODEL_LAST_ROW:
95,519✔
2353
      return TSDB_CACHE_MODEL_LAST_ROW_STR;
95,519✔
2354
    case TSDB_CACHE_MODEL_LAST_VALUE:
160,307✔
2355
      return TSDB_CACHE_MODEL_LAST_VALUE_STR;
160,307✔
2356
    case TSDB_CACHE_MODEL_BOTH:
85,843✔
2357
      return TSDB_CACHE_MODEL_BOTH_STR;
85,843✔
2358
    default:
×
2359
      break;
×
2360
  }
2361
  return "unknown";
×
2362
}
2363

2364
static const char *getEncryptAlgorithmStr(int8_t encryptAlgorithm) {
636,682✔
2365
  switch (encryptAlgorithm) {
636,682!
2366
    case TSDB_ENCRYPT_ALGO_NONE:
636,689✔
2367
      return TSDB_ENCRYPT_ALGO_NONE_STR;
636,689✔
2368
    case TSDB_ENCRYPT_ALGO_SM4:
×
2369
      return TSDB_ENCRYPT_ALGO_SM4_STR;
×
2370
    default:
×
2371
      break;
×
2372
  }
2373
  return "unknown";
×
2374
}
2375

2376
bool mndIsDbReady(SMnode *pMnode, SDbObj *pDb) {
644,037✔
2377
  if (pDb->cfg.replications == 1) return true;
644,037✔
2378

2379
  SSdb *pSdb = pMnode->pSdb;
1,033✔
2380
  void *pIter = NULL;
1,033✔
2381
  bool  isReady = true;
1,033✔
2382
  while (1) {
6,930✔
2383
    SVgObj *pVgroup = NULL;
7,963✔
2384
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
7,963✔
2385
    if (pIter == NULL) break;
7,980✔
2386

2387
    if (pVgroup->dbUid == pDb->uid && pVgroup->replica > 1) {
6,930!
2388
      bool hasLeader = false;
3,086✔
2389
      for (int32_t i = 0; i < pVgroup->replica; ++i) {
12,291✔
2390
        if (pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_LEADER ||
9,205✔
2391
            pVgroup->vnodeGid[i].syncState == TAOS_SYNC_STATE_ASSIGNED_LEADER) {
8,154!
2392
          hasLeader = true;
1,051✔
2393
        }
2394
      }
2395
      if (!hasLeader) isReady = false;
3,086✔
2396
    }
2397
    sdbRelease(pSdb, pVgroup);
6,930✔
2398
  }
2399

2400
  return isReady;
1,050✔
2401
}
2402

2403
static void mndDumpDbInfoData(SMnode *pMnode, SSDataBlock *pBlock, SDbObj *pDb, SShowObj *pShow, int32_t rows,
694,949✔
2404
                              int64_t numOfTables, bool sysDb, ESdbStatus objStatus, bool sysinfo) {
2405
  int32_t cols = 0;
694,949✔
2406
  int32_t bytes = pShow->pMeta->pSchemas[cols].bytes;
694,949✔
2407
  char   *buf = taosMemoryMalloc(bytes);
694,949!
2408
  if (buf == NULL) {
694,768!
2409
    mError("db:%s, failed to malloc buffer", pDb->name);
×
2410
    return;
×
2411
  }
2412
  int32_t code = 0;
694,768✔
2413
  int32_t lino = 0;
694,768✔
2414

2415
  const char *name = mndGetDbStr(pDb->name);
694,768✔
2416
  if (name != NULL) {
694,735✔
2417
    STR_WITH_MAXSIZE_TO_VARSTR(buf, name, bytes);
694,645✔
2418
  } else {
2419
    STR_WITH_MAXSIZE_TO_VARSTR(buf, "NULL", bytes);
90✔
2420
  }
2421

2422
  const char *precStr = NULL;
694,735✔
2423
  switch (pDb->cfg.precision) {
694,735✔
2424
    case TSDB_TIME_PRECISION_MILLI:
694,679✔
2425
      precStr = TSDB_TIME_PRECISION_MILLI_STR;
694,679✔
2426
      break;
694,679✔
2427
    case TSDB_TIME_PRECISION_MICRO:
14✔
2428
      precStr = TSDB_TIME_PRECISION_MICRO_STR;
14✔
2429
      break;
14✔
2430
    case TSDB_TIME_PRECISION_NANO:
15✔
2431
      precStr = TSDB_TIME_PRECISION_NANO_STR;
15✔
2432
      break;
15✔
2433
    default:
27✔
2434
      precStr = "none";
27✔
2435
      break;
27✔
2436
  }
2437
  char precVstr[10] = {0};
694,735✔
2438
  STR_WITH_MAXSIZE_TO_VARSTR(precVstr, precStr, 10);
694,735✔
2439

2440
  char *statusStr = "ready";
694,735✔
2441
  if (objStatus == SDB_STATUS_CREATING) {
694,735✔
2442
    statusStr = "creating";
37✔
2443
  } else if (objStatus == SDB_STATUS_DROPPING) {
694,698✔
2444
    statusStr = "dropping";
535✔
2445
  } else {
2446
    if (!sysDb && !mndIsDbReady(pMnode, pDb)) {
694,163✔
2447
      statusStr = "unsynced";
21✔
2448
    }
2449
  }
2450
  char statusVstr[24] = {0};
694,710✔
2451
  STR_WITH_MAXSIZE_TO_VARSTR(statusVstr, statusStr, 24);
694,710✔
2452

2453
  if (sysDb || !sysinfo) {
694,710✔
2454
    for (int32_t i = 0; i < pShow->numOfColumns; ++i) {
2,032,380✔
2455
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, i);
1,973,823✔
2456
      if (i == 0) {
1,970,255✔
2457
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
48,265!
2458
      } else if (i == 1) {
1,921,990✔
2459
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
55,071!
2460
      } else if (i == 3) {
1,866,919✔
2461
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
55,004!
2462
      } else if (i == 14) {
1,811,915✔
2463
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, precVstr, false), &lino, _OVER);
55,077!
2464
      } else if (i == 15) {
1,756,838✔
2465
        TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, statusVstr, false), &lino, _OVER);
55,054!
2466
      } else {
2467
        colDataSetNULL(pColInfo, rows);
1,701,784✔
2468
      }
2469
    }
2470
  } else {
2471
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
639,548✔
2472
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, buf, false), &lino, _OVER);
638,844!
2473

2474
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,956✔
2475
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->createdTime, false), &lino, _OVER);
637,547!
2476

2477
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,281✔
2478
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfVgroups, false), &lino, _OVER);
636,872!
2479

2480
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,107✔
2481
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&numOfTables, false), &lino, _OVER);
636,756!
2482

2483
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,911✔
2484
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.replications, false), &lino, _OVER);
636,644!
2485

2486
    const char *strictStr = pDb->cfg.strict ? "on" : "off";
637,214!
2487
    char        strictVstr[24] = {0};
637,214✔
2488
    STR_WITH_MAXSIZE_TO_VARSTR(strictVstr, strictStr, 24);
637,214✔
2489
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,214✔
2490
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)strictVstr, false), &lino, _OVER);
637,326!
2491

2492
    char    durationStr[128] = {0};
637,932✔
2493
    char    durationVstr[128] = {0};
637,932✔
2494
    int32_t len = formatDurationOrKeep(&durationVstr[VARSTR_HEADER_SIZE], sizeof(durationVstr) - VARSTR_HEADER_SIZE,
637,932✔
2495
                                       pDb->cfg.daysPerFile);
2496

2497
    varDataSetLen(durationVstr, len);
639,262✔
2498
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
639,262✔
2499
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
638,528!
2500

2501
    char keepVstr[128] = {0};
638,064✔
2502
    char keep0Str[32] = {0};
638,064✔
2503
    char keep1Str[32] = {0};
638,064✔
2504
    char keep2Str[32] = {0};
638,064✔
2505

2506
    int32_t lenKeep0 = formatDurationOrKeep(keep0Str, sizeof(keep0Str), pDb->cfg.daysToKeep0);
638,064✔
2507
    int32_t lenKeep1 = formatDurationOrKeep(keep1Str, sizeof(keep1Str), pDb->cfg.daysToKeep1);
639,567✔
2508
    int32_t lenKeep2 = formatDurationOrKeep(keep2Str, sizeof(keep2Str), pDb->cfg.daysToKeep2);
639,596✔
2509

2510
    if (pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep1 || pDb->cfg.daysToKeep0 > pDb->cfg.daysToKeep2) {
639,581!
2511
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep1Str, keep2Str, keep0Str);
64✔
2512
    } else {
2513
      len = tsnprintf(&keepVstr[VARSTR_HEADER_SIZE], sizeof(keepVstr), "%s,%s,%s", keep0Str, keep1Str, keep2Str);
639,517✔
2514
    }
2515
    varDataSetLen(keepVstr, len);
639,604✔
2516
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
639,604✔
2517
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keepVstr, false), &lino, _OVER);
638,490!
2518

2519
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,594✔
2520
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.buffer, false), &lino, _OVER);
637,186!
2521

2522
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,001✔
2523
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pageSize, false), &lino, _OVER);
636,553!
2524

2525
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,859✔
2526
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.pages, false), &lino, _OVER);
636,488!
2527

2528
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,817✔
2529
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.minRows, false), &lino, _OVER);
636,412!
2530

2531
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,764✔
2532
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.maxRows, false), &lino, _OVER);
636,411!
2533

2534
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,798✔
2535
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.compression, false), &lino, _OVER);
636,469!
2536

2537
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,904✔
2538
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)precVstr, false), &lino, _OVER);
636,495!
2539

2540
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,348✔
2541
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)statusVstr, false), &lino, _OVER);
637,009!
2542

2543
    char *rentensionVstr = buildRetension(pDb->cfg.pRetensions);
637,746✔
2544
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,699✔
2545
    if (rentensionVstr == NULL) {
637,260✔
2546
      colDataSetNULL(pColInfo, rows);
637,259!
2547
    } else {
2548
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)rentensionVstr, false), &lino, _OVER);
1!
2549
      taosMemoryFree(rentensionVstr);
1!
2550
    }
2551

2552
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,260✔
2553
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.numOfStables, false), &lino, _OVER);
636,960!
2554

2555
    const char *cacheModelStr = getCacheModelStr(pDb->cfg.cacheLast);
637,504✔
2556
    char        cacheModelVstr[24] = {0};
637,426✔
2557
    STR_WITH_MAXSIZE_TO_VARSTR(cacheModelVstr, cacheModelStr, 24);
637,426✔
2558
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,426✔
2559
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)cacheModelVstr, false), &lino, _OVER);
637,497!
2560

2561
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,453✔
2562
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.cacheLastSize, false), &lino, _OVER);
637,061!
2563

2564
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,959✔
2565
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walLevel, false), &lino, _OVER);
636,528!
2566

2567
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,703✔
2568
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walFsyncPeriod, false), &lino, _OVER);
636,346!
2569

2570
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,491✔
2571
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionPeriod, false), &lino, _OVER);
636,199!
2572

2573
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,598✔
2574
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.walRetentionSize, false), &lino, _OVER);
636,274!
2575

2576
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,577✔
2577
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.sstTrigger, false), &lino, _OVER);
636,225!
2578

2579
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,702✔
2580
    int16_t hashPrefix = pDb->cfg.hashPrefix;
636,393✔
2581
    if (hashPrefix > 0) {
636,393✔
2582
      hashPrefix = pDb->cfg.hashPrefix - strlen(pDb->name) - 1;
13✔
2583
    } else if (hashPrefix < 0) {
636,380✔
2584
      hashPrefix = pDb->cfg.hashPrefix + strlen(pDb->name) + 1;
16✔
2585
    }
2586
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&hashPrefix, false), &lino, _OVER);
636,393!
2587

2588
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,716✔
2589
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.hashSuffix, false), &lino, _OVER);
636,334!
2590

2591
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,539✔
2592
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.tsdbPageSize, false), &lino, _OVER);
636,241!
2593

2594
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,595✔
2595
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.keepTimeOffset, false), &lino, _OVER);
636,278!
2596

2597
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,619✔
2598
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3ChunkSize, false), &lino, _OVER);
636,269!
2599

2600
    char keeplocalVstr[128] = {0};
636,822✔
2601
    len = tsnprintf(&keeplocalVstr[VARSTR_HEADER_SIZE], sizeof(keeplocalVstr), "%dm", pDb->cfg.s3KeepLocal);
636,822✔
2602
    varDataSetLen(keeplocalVstr, len);
639,540✔
2603
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
639,540✔
2604
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)keeplocalVstr, false), &lino, _OVER);
638,640!
2605

2606
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
637,719✔
2607
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.s3Compact, false), &lino, _OVER);
637,236!
2608

2609
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,892✔
2610
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)&pDb->cfg.withArbitrator, false), &lino, _OVER);
636,463!
2611

2612
    const char *encryptAlgorithmStr = getEncryptAlgorithmStr(pDb->cfg.encryptAlgorithm);
636,689✔
2613
    char        encryptAlgorithmVStr[24] = {0};
636,714✔
2614
    STR_WITH_MAXSIZE_TO_VARSTR(encryptAlgorithmVStr, encryptAlgorithmStr, 24);
636,714✔
2615
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
636,714✔
2616
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)encryptAlgorithmVStr, false), &lino, _OVER);
637,335!
2617

2618
    TAOS_UNUSED(formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactInterval));
637,668✔
2619
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
639,450✔
2620
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
639,450!
2621
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
638,498!
2622
    }
2623

2624
    len = formatDurationOrKeep(durationStr, sizeof(durationStr), pDb->cfg.compactStartTime);
637,774✔
2625
    TAOS_UNUSED(formatDurationOrKeep(durationVstr, sizeof(durationVstr), pDb->cfg.compactEndTime));
639,437✔
2626
    TAOS_UNUSED(snprintf(durationStr + len, sizeof(durationStr) - len, ",%s", durationVstr));
639,559✔
2627
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
639,559✔
2628
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
639,559!
2629
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
638,499!
2630
    }
2631

2632
    TAOS_UNUSED(snprintf(durationStr, sizeof(durationStr), "%dh", pDb->cfg.compactTimeOffset));
637,652✔
2633
    STR_WITH_MAXSIZE_TO_VARSTR(durationVstr, durationStr, sizeof(durationVstr));
637,652✔
2634
    if ((pColInfo = taosArrayGet(pBlock->pDataBlock, cols++))) {
637,652✔
2635
      TAOS_CHECK_GOTO(colDataSetVal(pColInfo, rows, (const char *)durationVstr, false), &lino, _OVER);
638,399!
2636
    }
2637
  }
2638
_OVER:
696,315✔
2639
  if (code != 0) mError("failed to retrieve at line:%d, since %s", lino, tstrerror(code));
696,315!
2640
  taosMemoryFree(buf);
692,857!
2641
}
2642

2643
static void setInformationSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
27,596✔
2644
  tstrncpy(pDbObj->name, TSDB_INFORMATION_SCHEMA_DB, tListLen(pDbObj->name));
27,596✔
2645
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
27,596✔
2646
  pDbObj->cfg.numOfVgroups = 0;
27,620✔
2647
  pDbObj->cfg.strict = 1;
27,620✔
2648
  pDbObj->cfg.replications = 1;
27,620✔
2649
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
27,620✔
2650
}
27,620✔
2651

2652
static void setPerfSchemaDbCfg(SMnode *pMnode, SDbObj *pDbObj) {
27,592✔
2653
  tstrncpy(pDbObj->name, TSDB_PERFORMANCE_SCHEMA_DB, tListLen(pDbObj->name));
27,592✔
2654
  pDbObj->createdTime = mndGetClusterCreateTime(pMnode);
27,592✔
2655
  pDbObj->cfg.numOfVgroups = 0;
27,622✔
2656
  pDbObj->cfg.strict = 1;
27,622✔
2657
  pDbObj->cfg.replications = 1;
27,622✔
2658
  pDbObj->cfg.precision = TSDB_TIME_PRECISION_MILLI;
27,622✔
2659
}
27,622✔
2660

2661
static bool mndGetTablesOfDbFp(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
29,878,777✔
2662
  SVgObj  *pVgroup = pObj;
29,878,777✔
2663
  int32_t *numOfTables = p1;
29,878,777✔
2664
  int64_t  uid = *(int64_t *)p2;
29,878,777✔
2665
  if (pVgroup->dbUid == uid) {
29,878,777✔
2666
    *numOfTables += pVgroup->numOfTables;
1,253,286✔
2667
  }
2668
  return true;
29,878,777✔
2669
}
2670

2671
static int32_t mndRetrieveDbs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rowsCapacity) {
27,604✔
2672
  SMnode    *pMnode = pReq->info.node;
27,604✔
2673
  SSdb      *pSdb = pMnode->pSdb;
27,604✔
2674
  int32_t    numOfRows = 0;
27,604✔
2675
  SDbObj    *pDb = NULL;
27,604✔
2676
  SUserObj  *pUser = NULL;
27,604✔
2677
  ESdbStatus objStatus = 0;
27,604✔
2678

2679
  (void)mndAcquireUser(pMnode, pReq->info.conn.user, &pUser);
27,604✔
2680
  if (pUser == NULL) return 0;
27,609!
2681
  bool sysinfo = pUser->sysInfo;
27,609✔
2682

2683
  // Append the information_schema database into the result.
2684
  if (!pShow->sysDbRsp) {
27,609✔
2685
    SDbObj infoschemaDb = {0};
27,608✔
2686
    setInformationSchemaDbCfg(pMnode, &infoschemaDb);
27,608✔
2687
    size_t numOfTables = 0;
27,619✔
2688
    getVisibleInfosTablesNum(sysinfo, &numOfTables);
27,619✔
2689
    mndDumpDbInfoData(pMnode, pBlock, &infoschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
27,611✔
2690

2691
    numOfRows += 1;
27,585✔
2692

2693
    SDbObj perfschemaDb = {0};
27,585✔
2694
    setPerfSchemaDbCfg(pMnode, &perfschemaDb);
27,585✔
2695
    numOfTables = 0;
27,622✔
2696
    getPerfDbMeta(NULL, &numOfTables);
27,622✔
2697
    mndDumpDbInfoData(pMnode, pBlock, &perfschemaDb, pShow, numOfRows, numOfTables, true, 0, 1);
27,621✔
2698

2699
    numOfRows += 1;
27,588✔
2700
    pShow->sysDbRsp = true;
27,588✔
2701
  }
2702

2703
  while (numOfRows < rowsCapacity) {
667,422✔
2704
    pShow->pIter = sdbFetchAll(pSdb, SDB_DB, pShow->pIter, (void **)&pDb, &objStatus, true);
667,407✔
2705
    if (pShow->pIter == NULL) break;
667,349✔
2706

2707
    if (mndCheckDbPrivilege(pMnode, pReq->info.conn.user, MND_OPER_READ_OR_WRITE_DB, pDb) == 0) {
639,729✔
2708
      int32_t numOfTables = 0;
639,816✔
2709
      sdbTraverse(pSdb, SDB_VGROUP, mndGetTablesOfDbFp, &numOfTables, &pDb->uid, NULL);
639,816✔
2710
      mndDumpDbInfoData(pMnode, pBlock, pDb, pShow, numOfRows, numOfTables, false, objStatus, sysinfo);
639,826✔
2711
      numOfRows++;
638,841✔
2712
    }
2713

2714
    sdbRelease(pSdb, pDb);
638,864✔
2715
  }
2716

2717
  pShow->numOfRows += numOfRows;
27,635✔
2718
  mndReleaseUser(pMnode, pUser);
27,635✔
2719
  return numOfRows;
27,622✔
2720
}
2721

2722
static void mndCancelGetNextDb(SMnode *pMnode, void *pIter) {
×
2723
  SSdb *pSdb = pMnode->pSdb;
×
2724
  sdbCancelFetchByType(pSdb, pIter, SDB_DB);
×
2725
}
×
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