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

taosdata / TDengine / #4997

20 Mar 2026 06:10AM UTC coverage: 71.739% (-0.3%) from 72.069%
#4997

push

travis-ci

web-flow
feat: add query phase tracking for SHOW QUERIES (#34706)

148 of 183 new or added lines in 10 files covered. (80.87%)

9273 existing lines in 172 files now uncovered.

244572 of 340921 relevant lines covered (71.74%)

133392941.95 hits per line

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

50.76
/source/dnode/mnode/impl/src/mndXnode.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 <stdio.h>
18
#include "mndDef.h"
19
#include "tdatablock.h"
20
#include "types.h"
21
#include <curl/curl.h>
22
#ifndef WINDOWS
23
#include <openssl/bio.h>
24
#include <openssl/buffer.h>
25
#include <openssl/err.h>
26
#include <openssl/evp.h>
27
#include <openssl/hmac.h>
28
#else
29
#include "tsha.h"
30
#include "tbase64.h"
31
#endif
32
#include "audit.h"
33
#include "mndDnode.h"
34
#include "mndPrivilege.h"
35
#include "mndShow.h"
36
#include "mndTrans.h"
37
#include "mndUser.h"
38
#include "mndXnode.h"
39
#include "sdb.h"
40
#include "taoserror.h"
41
#include "tjson.h"
42
#include "xnode.h"
43

44
#define TSDB_XNODE_RESERVE_SIZE 64
45
#ifdef WINDOWS
46
#define XNODED_PIPE_SOCKET_URL "http://localhost:6051"
47
#else
48
#define XNODED_PIPE_SOCKET_URL "http://localhost"
49
#endif
50
typedef enum {
51
  HTTP_TYPE_GET = 0,
52
  HTTP_TYPE_POST,
53
  HTTP_TYPE_DELETE,
54
} EHttpType;
55
typedef struct {
56
  char   *data;
57
  int64_t dataLen;
58
} SCurlResp;
59

60
const int32_t defaultTimeout = 1000;
61

62
/** xnodes systable actions */
63
SSdbRaw *mndXnodeActionEncode(SXnodeObj *pObj);
64
SSdbRow *mndXnodeActionDecode(SSdbRaw *pRaw);
65
int32_t  mndXnodeActionInsert(SSdb *pSdb, SXnodeObj *pObj);
66
int32_t  mndXnodeActionUpdate(SSdb *pSdb, SXnodeObj *pOld, SXnodeObj *pNew);
67
int32_t  mndXnodeActionDelete(SSdb *pSdb, SXnodeObj *pObj);
68

69
/** @section xnodes request handlers */
70
static int32_t mndProcessCreateXnodeReq(SRpcMsg *pReq);
71
static int32_t mndProcessUpdateXnodeReq(SRpcMsg *pReq);
72
static int32_t mndProcessDropXnodeReq(SRpcMsg *pReq);
73
static int32_t mndProcessDrainXnodeReq(SRpcMsg *pReq);
74
static int32_t mndRetrieveXnodes(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows);
75
static void    mndCancelGetNextXnode(SMnode *pMnode, void *pIter);
76

77
/** @section xnode task handlers */
78
SSdbRaw *mndXnodeTaskActionEncode(SXnodeTaskObj *pObj);
79
SSdbRow *mndXnodeTaskActionDecode(SSdbRaw *pRaw);
80
int32_t  mndXnodeTaskActionInsert(SSdb *pSdb, SXnodeTaskObj *pObj);
81
int32_t  mndXnodeTaskActionUpdate(SSdb *pSdb, SXnodeTaskObj *pOld, SXnodeTaskObj *pNew);
82
int32_t  mndXnodeTaskActionDelete(SSdb *pSdb, SXnodeTaskObj *pObj);
83

84
static int32_t mndProcessCreateXnodeTaskReq(SRpcMsg *pReq);
85
static int32_t mndProcessStartXnodeTaskReq(SRpcMsg *pReq);
86
static int32_t mndProcessStopXnodeTaskReq(SRpcMsg *pReq);
87
static int32_t mndProcessUpdateXnodeTaskReq(SRpcMsg *pReq);
88
static int32_t mndProcessDropXnodeTaskReq(SRpcMsg *pReq);
89
static int32_t mndRetrieveXnodeTasks(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows);
90
static void    mndCancelGetNextXnodeTask(SMnode *pMnode, void *pIter);
91

92
/** @section xnode task job handlers */
93
SSdbRaw *mndXnodeJobActionEncode(SXnodeJobObj *pObj);
94
SSdbRow *mndXnodeJobActionDecode(SSdbRaw *pRaw);
95
int32_t  mndXnodeJobActionInsert(SSdb *pSdb, SXnodeJobObj *pObj);
96
int32_t  mndXnodeJobActionUpdate(SSdb *pSdb, SXnodeJobObj *pOld, SXnodeJobObj *pNew);
97
int32_t  mndXnodeJobActionDelete(SSdb *pSdb, SXnodeJobObj *pObj);
98

99
static int32_t mndProcessCreateXnodeJobReq(SRpcMsg *pReq);
100
static int32_t mndProcessUpdateXnodeJobReq(SRpcMsg *pReq);
101
static int32_t mndProcessRebalanceXnodeJobReq(SRpcMsg *pReq);
102
static int32_t mndProcessRebalanceXnodeJobsWhereReq(SRpcMsg *pReq);
103
static int32_t mndProcessDropXnodeJobReq(SRpcMsg *pReq);
104
static int32_t mndRetrieveXnodeJobs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows);
105
static void    mndCancelGetNextXnodeJob(SMnode *pMnode, void *pIter);
106

107
/** @section xnode user pass handlers */
108
SSdbRaw *mndXnodeUserPassActionEncode(SXnodeUserPassObj *pObj);
109
SSdbRow *mndXnodeUserPassActionDecode(SSdbRaw *pRaw);
110
int32_t  mndXnodeUserPassActionInsert(SSdb *pSdb, SXnodeUserPassObj *pObj);
111
int32_t  mndXnodeUserPassActionUpdate(SSdb *pSdb, SXnodeUserPassObj *pOld, SXnodeUserPassObj *pNew);
112
int32_t  mndXnodeUserPassActionDelete(SSdb *pSdb, SXnodeUserPassObj *pObj);
113

114
/** @section xnode agent handlers */
115
SSdbRaw *mndXnodeAgentActionEncode(SXnodeAgentObj *pObj);
116
SSdbRow *mndXnodeAgentActionDecode(SSdbRaw *pRaw);
117
int32_t  mndXnodeAgentActionInsert(SSdb *pSdb, SXnodeAgentObj *pObj);
118
int32_t  mndXnodeAgentActionUpdate(SSdb *pSdb, SXnodeAgentObj *pOld, SXnodeAgentObj *pNew);
119
int32_t  mndXnodeAgentActionDelete(SSdb *pSdb, SXnodeAgentObj *pObj);
120

121
static int32_t mndProcessCreateXnodeAgentReq(SRpcMsg *pReq);
122
static int32_t mndProcessUpdateXnodeAgentReq(SRpcMsg *pReq);
123
static int32_t mndProcessDropXnodeAgentReq(SRpcMsg *pReq);
124
static int32_t mndRetrieveXnodeAgents(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows);
125
static void    mndCancelGetNextXnodeAgent(SMnode *pMnode, void *pIter);
126

127
/** @section xnoded mgmt */
128
void mndStartXnoded(SMnode *pMnode, const char *user, const char *pass, const char *token);
129
void mndRestartXnoded(SMnode *pMnode);
130

131
/** @section others */
132
static int32_t mndGetXnodeStatus(SXnodeObj *pObj, char *status, int32_t statusLen);
133
SXnodeTaskObj *mndAcquireXnodeTask(SMnode *pMnode, int32_t tid);
134
SJson         *mndSendReqRetJson(const char *url, EHttpType type, int64_t timeout, const char *buf, int64_t bufLen);
135
static int32_t mndSetDropXnodeJobInfoToTrans(STrans *pTrans, SXnodeJobObj *pObj, bool force);
136
void           mndReleaseXnodeJob(SMnode *pMnode, SXnodeJobObj *pObj);
137
static int32_t mndValidateXnodePermissions(SMnode *pMnode, SRpcMsg *pReq, EOperType oper);
138
static int32_t mndXnodeCheckSuperUser(SMnode *pMnode, const char *user, bool *isSuperUser);
139
static int32_t mndCheckXnodeTaskOwner(SMnode *pMnode, const char *user, SXnodeTaskObj *pTask);
140
static int32_t mndCheckXnodeJobOwner(SMnode *pMnode, const char *user, SXnodeJobObj *pJob);
141

142
int32_t mndInitXnode(SMnode *pMnode) {
441,539✔
143
  SSdbTable table = {
441,539✔
144
      .sdbType = SDB_XNODE,
145
      .keyType = SDB_KEY_INT32,
146
      .encodeFp = (SdbEncodeFp)mndXnodeActionEncode,
147
      .decodeFp = (SdbDecodeFp)mndXnodeActionDecode,
148
      .insertFp = (SdbInsertFp)mndXnodeActionInsert,
149
      .updateFp = (SdbUpdateFp)mndXnodeActionUpdate,
150
      .deleteFp = (SdbDeleteFp)mndXnodeActionDelete,
151
  };
152

153
  int32_t code = sdbSetTable(pMnode->pSdb, table);
441,539✔
154
  if (code != 0) {
441,539✔
UNCOV
155
    return code;
×
156
  }
157

158
  SSdbTable tasks = {
441,539✔
159
      .sdbType = SDB_XNODE_TASK,
160
      .keyType = SDB_KEY_INT32,
161
      .encodeFp = (SdbEncodeFp)mndXnodeTaskActionEncode,
162
      .decodeFp = (SdbDecodeFp)mndXnodeTaskActionDecode,
163
      .insertFp = (SdbInsertFp)mndXnodeTaskActionInsert,
164
      .updateFp = (SdbUpdateFp)mndXnodeTaskActionUpdate,
165
      .deleteFp = (SdbDeleteFp)mndXnodeTaskActionDelete,
166
  };
167

168
  code = sdbSetTable(pMnode->pSdb, tasks);
441,539✔
169
  if (code != 0) {
441,539✔
UNCOV
170
    return code;
×
171
  }
172

173
  SSdbTable jobs = {
441,539✔
174
      .sdbType = SDB_XNODE_JOB,
175
      .keyType = SDB_KEY_INT32,
176
      .encodeFp = (SdbEncodeFp)mndXnodeJobActionEncode,
177
      .decodeFp = (SdbDecodeFp)mndXnodeJobActionDecode,
178
      .insertFp = (SdbInsertFp)mndXnodeJobActionInsert,
179
      .updateFp = (SdbUpdateFp)mndXnodeJobActionUpdate,
180
      .deleteFp = (SdbDeleteFp)mndXnodeJobActionDelete,
181
  };
182

183
  code = sdbSetTable(pMnode->pSdb, jobs);
441,539✔
184
  if (code != 0) {
441,539✔
UNCOV
185
    return code;
×
186
  }
187

188
  SSdbTable agents = {
441,539✔
189
      .sdbType = SDB_XNODE_AGENT,
190
      .keyType = SDB_KEY_INT32,
191
      .encodeFp = (SdbEncodeFp)mndXnodeAgentActionEncode,
192
      .decodeFp = (SdbDecodeFp)mndXnodeAgentActionDecode,
193
      .insertFp = (SdbInsertFp)mndXnodeAgentActionInsert,
194
      .updateFp = (SdbUpdateFp)mndXnodeAgentActionUpdate,
195
      .deleteFp = (SdbDeleteFp)mndXnodeAgentActionDelete,
196
  };
197

198
  code = sdbSetTable(pMnode->pSdb, agents);
441,539✔
199
  if (code != 0) {
441,539✔
UNCOV
200
    return code;
×
201
  }
202

203
  SSdbTable userPass = {
441,539✔
204
      .sdbType = SDB_XNODE_USER_PASS,
205
      .keyType = SDB_KEY_INT32,
206
      .encodeFp = (SdbEncodeFp)mndXnodeUserPassActionEncode,
207
      .decodeFp = (SdbDecodeFp)mndXnodeUserPassActionDecode,
208
      .insertFp = (SdbInsertFp)mndXnodeUserPassActionInsert,
209
      .updateFp = (SdbUpdateFp)mndXnodeUserPassActionUpdate,
210
      .deleteFp = (SdbDeleteFp)mndXnodeUserPassActionDelete,
211
  };
212

213
  code = sdbSetTable(pMnode->pSdb, userPass);
441,539✔
214
  if (code != 0) {
441,539✔
UNCOV
215
    return code;
×
216
  }
217

218
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_XNODE, mndProcessCreateXnodeReq);
441,539✔
219
  mndSetMsgHandle(pMnode, TDMT_MND_UPDATE_XNODE, mndProcessUpdateXnodeReq);
441,539✔
220
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_XNODE, mndProcessDropXnodeReq);
441,539✔
221
  mndSetMsgHandle(pMnode, TDMT_MND_DRAIN_XNODE, mndProcessDrainXnodeReq);
441,539✔
222
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_XNODES, mndRetrieveXnodes);
441,539✔
223
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_XNODES, mndCancelGetNextXnode);
441,539✔
224

225
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_XNODE_TASK, mndProcessCreateXnodeTaskReq);
441,539✔
226
  mndSetMsgHandle(pMnode, TDMT_MND_START_XNODE_TASK, mndProcessStartXnodeTaskReq);
441,539✔
227
  mndSetMsgHandle(pMnode, TDMT_MND_STOP_XNODE_TASK, mndProcessStopXnodeTaskReq);
441,539✔
228
  mndSetMsgHandle(pMnode, TDMT_MND_UPDATE_XNODE_TASK, mndProcessUpdateXnodeTaskReq);
441,539✔
229
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_XNODE_TASK, mndProcessDropXnodeTaskReq);
441,539✔
230
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_XNODE_TASKS, mndRetrieveXnodeTasks);
441,539✔
231
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_XNODE_TASKS, mndCancelGetNextXnodeTask);
441,539✔
232

233
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_XNODE_JOB, mndProcessCreateXnodeJobReq);
441,539✔
234
  mndSetMsgHandle(pMnode, TDMT_MND_UPDATE_XNODE_JOB, mndProcessUpdateXnodeJobReq);
441,539✔
235
  mndSetMsgHandle(pMnode, TDMT_MND_REBALANCE_XNODE_JOB, mndProcessRebalanceXnodeJobReq);
441,539✔
236
  mndSetMsgHandle(pMnode, TDMT_MND_REBALANCE_XNODE_JOBS_WHERE, mndProcessRebalanceXnodeJobsWhereReq);
441,539✔
237
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_XNODE_JOB, mndProcessDropXnodeJobReq);
441,539✔
238
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_XNODE_JOBS, mndRetrieveXnodeJobs);
441,539✔
239
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_XNODE_JOBS, mndCancelGetNextXnodeJob);
441,539✔
240

241
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_XNODE_AGENT, mndProcessCreateXnodeAgentReq);
441,539✔
242
  mndSetMsgHandle(pMnode, TDMT_MND_UPDATE_XNODE_AGENT, mndProcessUpdateXnodeAgentReq);
441,539✔
243
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_XNODE_AGENT, mndProcessDropXnodeAgentReq);
441,539✔
244
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_XNODE_AGENTS, mndRetrieveXnodeAgents);
441,539✔
245
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_XNODE_AGENTS, mndCancelGetNextXnodeAgent);
441,539✔
246

247
  return 0;
441,539✔
248
}
249

250
/** tools section **/
251

252
int32_t xnodeCheckPasswordFmt(const char *pwd) {
119✔
253
  if (tsEnableAdvancedSecurity == 0 && strcmp(pwd, "taosdata") == 0) {
119✔
UNCOV
254
    return 0;
×
255
  }
256

257
  if (tsEnableStrongPassword == 0) {
119✔
UNCOV
258
    for (char c = *pwd; c != 0; c = *(++pwd)) {
×
UNCOV
259
      if (c == ' ' || c == '\'' || c == '\"' || c == '`' || c == '\\') {
×
260
        return TSDB_CODE_MND_INVALID_PASS_FORMAT;
×
261
      }
262
    }
UNCOV
263
    return 0;
×
264
  }
265

266
  int32_t len = strlen(pwd);
119✔
267
  if (len < TSDB_PASSWORD_MIN_LEN) {
119✔
UNCOV
268
    return TSDB_CODE_PAR_PASSWD_TOO_SHORT_OR_EMPTY;
×
269
  }
270

271
  if (len > TSDB_PASSWORD_MAX_LEN) {
119✔
UNCOV
272
    return TSDB_CODE_PAR_NAME_OR_PASSWD_TOO_LONG;
×
273
  }
274

275
  if (taosIsComplexString(pwd)) {
119✔
276
    return 0;
119✔
277
  }
278

UNCOV
279
  return TSDB_CODE_MND_INVALID_PASS_FORMAT;
×
280
}
281

282
static void swapFields(int32_t *newLen, char **ppNewStr, int32_t *oldLen, char **ppOldStr) {
18,207✔
283
  if (*newLen > 0) {
18,207✔
284
    int32_t tempLen = *newLen;
14,518✔
285
    *newLen = *oldLen;
14,518✔
286
    *oldLen = tempLen;
14,518✔
287

288
    char *tempStr = *ppNewStr;
14,518✔
289
    *ppNewStr = *ppOldStr;
14,518✔
290
    *ppOldStr = tempStr;
14,518✔
291
  }
292
}
18,207✔
293

294
/** xnode section **/
295

296
void mndCleanupXnode(SMnode *pMnode) {}
441,475✔
297

298
SXnodeObj *mndAcquireXnode(SMnode *pMnode, int32_t xnodeId) {
833✔
299
  SXnodeObj *pObj = sdbAcquire(pMnode->pSdb, SDB_XNODE, &xnodeId);
833✔
300
  if (pObj == NULL && terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
833✔
301
    terrno = TSDB_CODE_MND_XNODE_NOT_EXIST;
595✔
302
  }
303
  return pObj;
833✔
304
}
305

306
void mndReleaseXnode(SMnode *pMnode, SXnodeObj *pObj) {
4,879✔
307
  SSdb *pSdb = pMnode->pSdb;
4,879✔
308
  sdbRelease(pSdb, pObj);
4,879✔
309
}
4,879✔
310

311
SSdbRaw *mndXnodeActionEncode(SXnodeObj *pObj) {
11,998✔
312
  int32_t code = 0;
11,998✔
313
  int32_t lino = 0;
11,998✔
314
  terrno = TSDB_CODE_OUT_OF_MEMORY;
11,998✔
315

316
  if (NULL == pObj) {
11,998✔
317
    terrno = TSDB_CODE_INVALID_PARA;
194✔
318
    return NULL;
194✔
319
  }
320

321
  int32_t rawDataLen = sizeof(SXnodeObj) + TSDB_XNODE_RESERVE_SIZE + pObj->urlLen + pObj->statusLen;
11,804✔
322

323
  SSdbRaw *pRaw = sdbAllocRaw(SDB_XNODE, TSDB_XNODE_VER_NUMBER, rawDataLen);
11,804✔
324
  if (pRaw == NULL) goto _OVER;
11,804✔
325

326
  int32_t dataPos = 0;
11,804✔
327
  SDB_SET_INT32(pRaw, dataPos, pObj->id, _OVER)
11,804✔
328
  SDB_SET_INT32(pRaw, dataPos, pObj->urlLen, _OVER)
11,804✔
329
  SDB_SET_BINARY(pRaw, dataPos, pObj->url, pObj->urlLen, _OVER)
11,804✔
330
  SDB_SET_INT32(pRaw, dataPos, pObj->statusLen, _OVER)
11,804✔
331
  SDB_SET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
11,804✔
332
  SDB_SET_INT64(pRaw, dataPos, pObj->createTime, _OVER)
11,804✔
333
  SDB_SET_INT64(pRaw, dataPos, pObj->updateTime, _OVER)
11,804✔
334

335
  SDB_SET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
11,804✔
336

337
  terrno = 0;
11,804✔
338

339
_OVER:
11,804✔
340
  if (terrno != 0) {
11,804✔
UNCOV
341
    mError("xnode:%d, failed to encode to raw:%p since %s", pObj->id, pRaw, terrstr());
×
UNCOV
342
    sdbFreeRaw(pRaw);
×
UNCOV
343
    return NULL;
×
344
  }
345

346
  mTrace("xnode:%d, encode to raw:%p, row:%p", pObj->id, pRaw, pObj);
11,804✔
347
  return pRaw;
11,804✔
348
}
349

350
SSdbRow *mndXnodeActionDecode(SSdbRaw *pRaw) {
9,618✔
351
  int32_t code = 0;
9,618✔
352
  int32_t lino = 0;
9,618✔
353
  terrno = TSDB_CODE_OUT_OF_MEMORY;
9,618✔
354
  SSdbRow   *pRow = NULL;
9,618✔
355
  SXnodeObj *pObj = NULL;
9,618✔
356

357
  if (NULL == pRaw) {
9,618✔
358
    terrno = TSDB_CODE_INVALID_PARA;
194✔
359
    return NULL;
194✔
360
  }
361

362
  int8_t sver = 0;
9,424✔
363
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
9,424✔
364

365
  if (sver != TSDB_XNODE_VER_NUMBER) {
9,424✔
UNCOV
366
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
UNCOV
367
    goto _OVER;
×
368
  }
369

370
  pRow = sdbAllocRow(sizeof(SXnodeObj));
9,424✔
371
  if (pRow == NULL) goto _OVER;
9,424✔
372

373
  pObj = sdbGetRowObj(pRow);
9,424✔
374
  if (pObj == NULL) goto _OVER;
9,424✔
375

376
  int32_t dataPos = 0;
9,424✔
377
  SDB_GET_INT32(pRaw, dataPos, &pObj->id, _OVER)
9,424✔
378
  SDB_GET_INT32(pRaw, dataPos, &pObj->urlLen, _OVER)
9,424✔
379
  if (pObj->urlLen > 0) {
9,424✔
380
    pObj->url = taosMemoryCalloc(1, pObj->urlLen + 1);
8,992✔
381
    if (pObj->url == NULL) goto _OVER;
8,992✔
382
    SDB_GET_BINARY(pRaw, dataPos, pObj->url, pObj->urlLen, _OVER)
8,992✔
383
  } else {
384
    pObj->url = NULL;
432✔
385
  }
386
  SDB_GET_INT32(pRaw, dataPos, &pObj->statusLen, _OVER)
9,424✔
387
  if (pObj->statusLen > 0) {
9,424✔
388
    pObj->status = taosMemoryCalloc(1, pObj->statusLen + 1);
3,042✔
389
    if (pObj->status == NULL) goto _OVER;
3,042✔
390
    SDB_GET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
3,042✔
391
  } else {
392
    pObj->status = NULL;
6,382✔
393
  }
394
  SDB_GET_INT64(pRaw, dataPos, &pObj->createTime, _OVER)
9,424✔
395
  SDB_GET_INT64(pRaw, dataPos, &pObj->updateTime, _OVER)
9,424✔
396

397
  SDB_GET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
9,424✔
398

399
  terrno = 0;
9,424✔
400

401
_OVER:
9,424✔
402
  if (terrno != 0) {
9,424✔
UNCOV
403
    mError("xnode:%d, failed to decode from raw:%p since %s", pObj == NULL ? 0 : pObj->id, pRaw, terrstr());
×
404
    if (pObj != NULL) {
×
405
      taosMemoryFreeClear(pObj->url);
×
406
    }
UNCOV
407
    taosMemoryFreeClear(pRow);
×
UNCOV
408
    return NULL;
×
409
  }
410

411
  mTrace("xnode:%d, decode from raw:%p, row:%p", pObj->id, pRaw, pObj);
9,424✔
412
  return pRow;
9,424✔
413
}
414

415
static void mndFreeXnode(SXnodeObj *pObj) {
8,806✔
416
  if (pObj == NULL) return;
8,806✔
417
  if (pObj->url != NULL) {
8,806✔
418
    taosMemoryFreeClear(pObj->url);
8,568✔
419
  }
420
  if (pObj->status != NULL) {
8,806✔
421
    taosMemoryFreeClear(pObj->status);
714✔
422
  }
423
}
424

425
int32_t mndXnodeActionInsert(SSdb *pSdb, SXnodeObj *pObj) {
1,904✔
426
  mDebug("xnode:%d, perform insert action, row:%p", pObj->id, pObj);
1,904✔
427
  return 0;
1,904✔
428
}
429

430
int32_t mndXnodeActionDelete(SSdb *pSdb, SXnodeObj *pObj) {
6,902✔
431
  mDebug("xnode:%d, perform delete action, row:%p", pObj->id, pObj);
6,902✔
432
  mndFreeXnode(pObj);
6,902✔
433
  return 0;
6,902✔
434
}
435

436
int32_t mndXnodeActionUpdate(SSdb *pSdb, SXnodeObj *pOld, SXnodeObj *pNew) {
3,570✔
437
  mDebug("xnode:%d, perform update action, old row:%p new row:%p", pOld->id, pOld, pNew);
3,570✔
438

439
  taosWLockLatch(&pOld->lock);
3,570✔
440
  swapFields(&pNew->statusLen, &pNew->status, &pOld->statusLen, &pOld->status);
3,570✔
441
  if (pNew->updateTime > pOld->updateTime) {
3,570✔
442
    pOld->updateTime = pNew->updateTime;
238✔
443
  }
444
  taosWUnLockLatch(&pOld->lock);
3,570✔
445
  return 0;
3,570✔
446
}
447

448
void mndReleaseXnodeUserPass(SMnode *pMnode, SXnodeUserPassObj *pObj) {
833✔
449
  SSdb *pSdb = pMnode->pSdb;
833✔
450
  sdbRelease(pSdb, pObj);
833✔
451
}
833✔
452

453
SXnodeUserPassObj *mndAcquireFirstXnodeUserPass(SMnode *pMnode) {
408,743✔
454
  SSdb *pSdb = pMnode->pSdb;
408,743✔
455

456
  void *pIter = NULL;
408,743✔
UNCOV
457
  while (1) {
×
458
    SXnodeUserPassObj *pObj = NULL;
408,743✔
459
    pIter = sdbFetch(pSdb, SDB_XNODE_USER_PASS, pIter, (void **)&pObj);
408,743✔
460
    if (pIter == NULL) break;
408,743✔
461

462
    if (pObj != NULL) {
833✔
463
      sdbCancelFetch(pSdb, pIter);
833✔
464
      return pObj;
833✔
465
    }
466

UNCOV
467
    sdbRelease(pSdb, pObj);
×
468
  }
469
  terrno = TSDB_CODE_MND_XNODE_USER_PASS_NOT_EXIST;
407,910✔
470
  return NULL;
407,910✔
471
}
472

473
static int32_t mndSetCreateXnodeUserPassRedoLogs(STrans *pTrans, SXnodeUserPassObj *pObj) {
238✔
474
  int32_t  code = 0;
238✔
475
  SSdbRaw *pRedoRaw = mndXnodeUserPassActionEncode(pObj);
238✔
476
  if (pRedoRaw == NULL) {
238✔
UNCOV
477
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
478
    if (terrno != 0) code = terrno;
×
UNCOV
479
    TAOS_RETURN(code);
×
480
  }
481
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
238✔
482
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_CREATING));
238✔
483
  TAOS_RETURN(code);
238✔
484
}
485

486
static int32_t mndSetCreateXnodeUserPassCommitLogs(STrans *pTrans, SXnodeUserPassObj *pObj) {
1,071✔
487
  int32_t  code = 0;
1,071✔
488
  SSdbRaw *pCommitRaw = mndXnodeUserPassActionEncode(pObj);
1,071✔
489
  if (pCommitRaw == NULL) {
1,071✔
UNCOV
490
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
491
    if (terrno != 0) code = terrno;
×
UNCOV
492
    TAOS_RETURN(code);
×
493
  }
494
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
1,071✔
495
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY));
1,071✔
496
  TAOS_RETURN(code);
1,071✔
497
}
498

499
static int32_t mndSetCreateXnodeRedoLogs(STrans *pTrans, SXnodeObj *pObj) {
1,904✔
500
  int32_t  code = 0;
1,904✔
501
  SSdbRaw *pRedoRaw = mndXnodeActionEncode(pObj);
1,904✔
502
  if (pRedoRaw == NULL) {
1,904✔
UNCOV
503
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
504
    if (terrno != 0) code = terrno;
×
UNCOV
505
    TAOS_RETURN(code);
×
506
  }
507
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
1,904✔
508
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_CREATING));
1,904✔
509
  TAOS_RETURN(code);
1,904✔
510
}
511

512
static int32_t mndSetCreateXnodeUndoLogs(STrans *pTrans, SXnodeObj *pObj) {
1,904✔
513
  int32_t  code = 0;
1,904✔
514
  SSdbRaw *pUndoRaw = mndXnodeActionEncode(pObj);
1,904✔
515
  if (pUndoRaw == NULL) {
1,904✔
UNCOV
516
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
517
    if (terrno != 0) code = terrno;
×
UNCOV
518
    TAOS_RETURN(code);
×
519
  }
520
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pUndoRaw));
1,904✔
521
  TAOS_CHECK_RETURN(sdbSetRawStatus(pUndoRaw, SDB_STATUS_DROPPED));
1,904✔
522
  TAOS_RETURN(code);
1,904✔
523
}
524

525
static int32_t mndSetCreateXnodeCommitLogs(STrans *pTrans, SXnodeObj *pObj) {
2,142✔
526
  int32_t  code = 0;
2,142✔
527
  SSdbRaw *pCommitRaw = mndXnodeActionEncode(pObj);
2,142✔
528
  if (pCommitRaw == NULL) {
2,142✔
UNCOV
529
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
530
    if (terrno != 0) code = terrno;
×
UNCOV
531
    TAOS_RETURN(code);
×
532
  }
533
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
2,142✔
534
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY));
2,142✔
535
  TAOS_RETURN(code);
2,142✔
536
}
537

538
static int32_t mndCreateXnode(SMnode *pMnode, SRpcMsg *pReq, SMCreateXnodeReq *pCreate) {
1,904✔
539
  int32_t code = 0, lino = 0;
1,904✔
540
  STrans *pTrans = NULL;
1,904✔
541

542
  SXnodeObj xnodeObj = {0};
1,904✔
543
  xnodeObj.id = sdbGetMaxId(pMnode->pSdb, SDB_XNODE);
1,904✔
544

545
  xnodeObj.urlLen = pCreate->urlLen;
1,904✔
546
  if (xnodeObj.urlLen > TSDB_XNODE_URL_LEN) {
1,904✔
UNCOV
547
    code = TSDB_CODE_MND_XNODE_TOO_LONG_URL;
×
UNCOV
548
    lino = __LINE__;
×
UNCOV
549
    goto _OVER;
×
550
  }
551
  xnodeObj.url = taosMemoryCalloc(1, pCreate->urlLen);
1,904✔
552
  if (xnodeObj.url == NULL) goto _OVER;
1,904✔
553
  (void)memcpy(xnodeObj.url, pCreate->url, pCreate->urlLen);
1,904✔
554

555
  xnodeObj.createTime = taosGetTimestampMs();
1,904✔
556
  xnodeObj.updateTime = xnodeObj.createTime;
1,904✔
557
  mInfo("create xnode, xnode.id:%d, xnode.url: %s, xnode.time:%" PRId64, xnodeObj.id, xnodeObj.url,
1,904✔
558
        xnodeObj.createTime);
559

560
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-xnode");
1,904✔
561
  if (pTrans == NULL) {
1,904✔
562
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
563
    if (terrno != 0) code = terrno;
×
UNCOV
564
    mInfo("failed to create transaction for xnode:%s, code:0x%x:%s", pCreate->url, code, tstrerror(code));
×
UNCOV
565
    lino = __LINE__;
×
UNCOV
566
    goto _OVER;
×
567
  }
568
  mndTransSetSerial(pTrans);
1,904✔
569

570
  mInfo("trans:%d, used to create xnode:%s as xnode:%d", pTrans->id, pCreate->url, xnodeObj.id);
1,904✔
571

572
  TAOS_CHECK_GOTO(mndSetCreateXnodeRedoLogs(pTrans, &xnodeObj), NULL, _OVER);
1,904✔
573
  TAOS_CHECK_GOTO(mndSetCreateXnodeUndoLogs(pTrans, &xnodeObj), NULL, _OVER);
1,904✔
574
  TAOS_CHECK_GOTO(mndSetCreateXnodeCommitLogs(pTrans, &xnodeObj), NULL, _OVER);
1,904✔
575
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
1,904✔
576
  code = TSDB_CODE_ACTION_IN_PROGRESS;
1,904✔
577

578
_OVER:
1,904✔
579
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,904✔
UNCOV
580
    mError("xnode:failed create xnode since %s, line:%d", tstrerror(code), lino);
×
581
  }
582
  mndFreeXnode(&xnodeObj);
1,904✔
583
  mndTransDrop(pTrans);
1,904✔
584
  TAOS_RETURN(code);
1,904✔
585
}
586

587
static SXnodeObj *mndAcquireXnodeByURL(SMnode *pMnode, char *url) {
5,950✔
588
  SSdb *pSdb = pMnode->pSdb;
5,950✔
589

590
  void *pIter = NULL;
5,950✔
591
  while (1) {
8,568✔
592
    SXnodeObj *pXnode = NULL;
14,518✔
593
    pIter = sdbFetch(pSdb, SDB_XNODE, pIter, (void **)&pXnode);
14,518✔
594
    if (pIter == NULL) break;
14,518✔
595

596
    if (strcasecmp(url, pXnode->url) == 0) {
12,257✔
597
      sdbCancelFetch(pSdb, pIter);
3,689✔
598
      return pXnode;
3,689✔
599
    }
600

601
    sdbRelease(pSdb, pXnode);
8,568✔
602
  }
603

604
  mError("xnode:%s, not found", url);
2,261✔
605
  terrno = TSDB_CODE_MND_XNODE_NOT_EXIST;
2,261✔
606
  return NULL;
2,261✔
607
}
608

609
static int32_t mndStoreXnodeUserPassToken(SMnode *pMnode, SRpcMsg *pReq, SMCreateXnodeReq *pCreate) {
238✔
610
  int32_t code = 0, lino = 0;
238✔
611
  STrans *pTrans = NULL;
238✔
612

613
  SXnodeUserPassObj upObj = {0};
238✔
614
  upObj.id = sdbGetMaxId(pMnode->pSdb, SDB_XNODE_USER_PASS);
238✔
615

616
  if (pCreate->user) {
238✔
617
    upObj.userLen = pCreate->userLen;
119✔
618
    if (upObj.userLen > TSDB_USER_LEN) {
119✔
UNCOV
619
      code = TSDB_CODE_MND_USER_NOT_AVAILABLE;
×
UNCOV
620
      lino = __LINE__;
×
UNCOV
621
      goto _OVER;
×
622
    }
623
    upObj.user = taosMemoryCalloc(1, pCreate->userLen);
119✔
624
    if (upObj.user == NULL) goto _OVER;
119✔
625
    (void)memcpy(upObj.user, pCreate->user, pCreate->userLen);
119✔
626
  }
627
  if (pCreate->pass) {
238✔
628
    upObj.passLen = pCreate->passLen;
119✔
629
    if (upObj.passLen > TSDB_USER_PASSWORD_LONGLEN) {
119✔
UNCOV
630
      code = TSDB_CODE_MND_INVALID_PASS_FORMAT;
×
UNCOV
631
      lino = __LINE__;
×
UNCOV
632
      goto _OVER;
×
633
    }
634
    upObj.pass = taosMemoryCalloc(1, pCreate->passLen);
119✔
635
    if (upObj.pass == NULL) goto _OVER;
119✔
636
    (void)memcpy(upObj.pass, pCreate->pass, pCreate->passLen);
119✔
637
  }
638

639
  if (pCreate->token.ptr) {
238✔
640
    upObj.tokenLen = pCreate->token.len + 1;
119✔
641
    upObj.token = taosMemoryCalloc(1, upObj.tokenLen);
119✔
642
    if (upObj.token == NULL) goto _OVER;
119✔
643
    (void)memcpy(upObj.token, pCreate->token.ptr, pCreate->token.len);
119✔
644
  }
645

646
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-userpass");
238✔
647
  if (pTrans == NULL) {
238✔
648
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
649
    if (terrno != 0) code = terrno;
×
UNCOV
650
    mInfo("failed to create transaction for xnode:%s, code:0x%x:%s", pCreate->url, code, tstrerror(code));
×
UNCOV
651
    lino = __LINE__;
×
UNCOV
652
    goto _OVER;
×
653
  }
654
  mndTransSetSerial(pTrans);
238✔
655

656
  mInfo("trans:%d, used to create xnode:%s as xnode:%d", pTrans->id, pCreate->url, upObj.id);
238✔
657

658
  TAOS_CHECK_GOTO(mndSetCreateXnodeUserPassRedoLogs(pTrans, &upObj), NULL, _OVER);
238✔
659
  TAOS_CHECK_GOTO(mndSetCreateXnodeUserPassCommitLogs(pTrans, &upObj), NULL, _OVER);
238✔
660
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
238✔
661
  code = TSDB_CODE_ACTION_IN_PROGRESS;
238✔
662

663
_OVER:
238✔
664
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
238✔
UNCOV
665
    mError("xnode failed to store userpass since %s, line:%d", tstrerror(code), lino);
×
666
  }
667
  taosMemoryFreeClear(upObj.user);
238✔
668
  taosMemoryFreeClear(upObj.pass);
238✔
669
  taosMemoryFreeClear(upObj.token);
238✔
670
  mndTransDrop(pTrans);
238✔
671
  TAOS_RETURN(code);
238✔
672
}
673

674
#ifndef TD_ENTERPRISE
675

676
int32_t mndXnodeCreateDefaultToken(SRpcMsg* pReq, char** ppToken) {
677
  return TSDB_CODE_OPS_NOT_SUPPORT;
678
}
679

680
#endif
681

682
static int32_t httpCreateXnode(SXnodeObj *pObj) {
1,904✔
683
  int32_t code = 0;
1,904✔
684
  SJson  *pJson = NULL;
1,904✔
685
  SJson  *postContent = NULL;
1,904✔
686
  char   *pContStr = NULL;
1,904✔
687

688
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
1,904✔
689
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/xnode", XNODED_PIPE_SOCKET_URL);
1,904✔
690
  postContent = tjsonCreateObject();
1,904✔
691
  if (postContent == NULL) {
1,904✔
UNCOV
692
    code = terrno;
×
UNCOV
693
    goto _OVER;
×
694
  }
695
  TAOS_CHECK_GOTO(tjsonAddDoubleToObject(postContent, "id", (double)pObj->id), NULL, _OVER);
1,904✔
696
  TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "url", pObj->url), NULL, _OVER);
1,904✔
697
  pContStr = tjsonToUnformattedString(postContent);
1,904✔
698
  if (pContStr == NULL) {
1,904✔
UNCOV
699
    code = terrno;
×
UNCOV
700
    goto _OVER;
×
701
  }
702
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, defaultTimeout, pContStr, strlen(pContStr));
1,904✔
703

704
_OVER:
1,904✔
705
  if (postContent != NULL) {
1,904✔
706
    tjsonDelete(postContent);
1,904✔
707
  }
708
  if (pContStr != NULL) {
1,904✔
709
    taosMemFree(pContStr);
1,904✔
710
  }
711
  if (pJson != NULL) {
1,904✔
UNCOV
712
    tjsonDelete(pJson);
×
713
  }
714
  TAOS_RETURN(code);
1,904✔
715
}
716

717
static int32_t mndProcessCreateXnodeReq(SRpcMsg *pReq) {
2,261✔
718
  SMnode          *pMnode = pReq->info.node;
2,261✔
719
  int32_t          code = 0, lino = 0;
2,261✔
720
  SXnodeObj       *pObj = NULL;
2,261✔
721
  SMCreateXnodeReq createReq = {0};
2,261✔
722
  char            *pToken = NULL;
2,261✔
723
  bool             isSuperUser = false;
2,261✔
724

725
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_CREATE_XNODE);
2,261✔
726
  if (code != TSDB_CODE_SUCCESS) {
2,261✔
UNCOV
727
    mError("failed check permission for create xnode, code:%s", tstrerror(code));
×
UNCOV
728
    goto _OVER;
×
729
  }
730
  TAOS_CHECK_GOTO(tDeserializeSMCreateXnodeReq(pReq->pCont, pReq->contLen, &createReq), &lino, _OVER);
2,261✔
731
  mDebug("xnode:%s, start to create", createReq.url);
2,261✔
732

733
  code = mndXnodeCheckSuperUser(pMnode, pReq->info.conn.user, &isSuperUser);
2,261✔
734
  if (code != 0) return code;
2,261✔
735
  if (!isSuperUser) {
2,261✔
UNCOV
736
    return TSDB_CODE_MND_XNODE_NO_PRIV;
×
737
  }
738

739
  pObj = mndAcquireXnodeByURL(pMnode, createReq.url);
2,261✔
740
  if (pObj != NULL) {
2,261✔
741
    code = TSDB_CODE_MND_XNODE_ALREADY_EXIST;
357✔
742
    goto _OVER;
357✔
743
  }
744

745
  int32_t numOfRows = sdbGetSize(pMnode->pSdb, SDB_XNODE_USER_PASS);
1,904✔
746
  if (numOfRows <= 0) {
1,904✔
747
    if (createReq.token.ptr != NULL) {
238✔
UNCOV
748
      (void)mndCreateXnode(pMnode, NULL, &createReq);
×
UNCOV
749
      code = mndStoreXnodeUserPassToken(pMnode, pReq, &createReq);
×
750
      if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
×
751
        lino = __LINE__;
×
UNCOV
752
        goto _OVER;
×
753
      }
UNCOV
754
      mndStartXnoded(pMnode, NULL, NULL, createReq.token.ptr);
×
755
    } else if (createReq.user != NULL) {
238✔
756
      TAOS_CHECK_GOTO(xnodeCheckPasswordFmt(createReq.pass), &lino, _OVER);
119✔
757
      (void)mndCreateXnode(pMnode, NULL, &createReq);
119✔
758
      code = mndStoreXnodeUserPassToken(pMnode, pReq, &createReq);
119✔
759
      if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
119✔
760
        lino = __LINE__;
×
761
        goto _OVER;
×
762
      }
763
      mndStartXnoded(pMnode, createReq.user, createReq.pass, NULL);
119✔
764
    } else {
765
      code = mndXnodeCreateDefaultToken(pReq, &pToken);
119✔
766
      if (code == TSDB_CODE_MND_TOKEN_ALREADY_EXIST) {
119✔
UNCOV
767
        code = 0;
×
768
      }
769
      if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
119✔
UNCOV
770
        lino = __LINE__;
×
771
        goto _OVER;
×
772
      }
773
      (void)mndCreateXnode(pMnode, NULL, &createReq);
119✔
774
      createReq.token = xCreateCowStr(strlen(pToken), pToken, false);
119✔
775
      (void)mndStoreXnodeUserPassToken(pMnode, NULL, &createReq);
119✔
776
      mndStartXnoded(pMnode, NULL, NULL, createReq.token.ptr);
119✔
777
    }
778
  } else {
779
    code = mndCreateXnode(pMnode, pReq, &createReq);
1,666✔
780
    if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
1,666✔
UNCOV
781
      lino = __LINE__;
×
UNCOV
782
      goto _OVER;
×
783
    }
784
  }
785

786
  taosMsleep(defaultTimeout);
1,904✔
787
  pObj = mndAcquireXnodeByURL(pMnode, createReq.url);
1,904✔
788
  if (pObj == NULL) {
1,904✔
UNCOV
789
    code = TSDB_CODE_MND_XNODE_NOT_EXIST;
×
UNCOV
790
    goto _OVER;
×
791
  }
792
  // send request
793
  (void)httpCreateXnode(pObj);
1,904✔
794

795
_OVER:
2,261✔
796
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
2,261✔
797
    mError("xnode:%s, failed to create since %s, line:%d", createReq.url, tstrerror(code), lino);
357✔
798
  }
799
  taosMemoryFreeClear(pToken);
2,261✔
800
  mndReleaseXnode(pMnode, pObj);
2,261✔
801
  tFreeSMCreateXnodeReq(&createReq);
2,261✔
802
  TAOS_RETURN(code);
2,261✔
803
}
804

805
static int32_t mndUpdateXnode(SMnode *pMnode, SXnodeObj *pXnode, SRpcMsg *pReq) {
×
806
  mInfo("xnode:%d, start to update", pXnode->id);
×
807
  int32_t   code = -1;
×
UNCOV
808
  STrans   *pTrans = NULL;
×
809
  SXnodeObj xnodeObj = {0};
×
UNCOV
810
  xnodeObj.id = pXnode->id;
×
811
  xnodeObj.updateTime = taosGetTimestampMs();
×
812

813
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "update-xnode");
×
UNCOV
814
  if (pTrans == NULL) {
×
815
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
816
    if (terrno != 0) code = terrno;
×
817
    goto _OVER;
×
818
  }
UNCOV
819
  mInfo("trans:%d, used to update xnode:%d", pTrans->id, xnodeObj.id);
×
820

UNCOV
821
  TAOS_CHECK_GOTO(mndSetCreateXnodeCommitLogs(pTrans, &xnodeObj), NULL, _OVER);
×
UNCOV
822
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
×
UNCOV
823
  code = 0;
×
824

UNCOV
825
_OVER:
×
UNCOV
826
  mndFreeXnode(&xnodeObj);
×
UNCOV
827
  mndTransDrop(pTrans);
×
UNCOV
828
  TAOS_RETURN(code);
×
829
}
830

831
static int32_t mndUpdateXnodeUserPassToken(SMnode *pMnode, SRpcMsg *pReq, SMUpdateXnodeReq *pUpdate) {
833✔
832
  int32_t           code = 0, lino = 0;
833✔
833
  STrans           *pTrans = NULL;
833✔
834
  SXnodeUserPassObj upObj = {0};
833✔
835
  upObj.id = pUpdate->id;
833✔
836
  upObj.updateTime = taosGetTimestampMs();
833✔
837

838
  if (pUpdate->user.ptr != NULL) {
833✔
839
    upObj.userLen = pUpdate->user.len + 1;
595✔
840
    upObj.user = taosMemoryCalloc(1, upObj.userLen);
595✔
841
    if (upObj.user == NULL) goto _OVER;
595✔
842
    (void)memcpy(upObj.user, pUpdate->user.ptr, pUpdate->user.len);
595✔
843
  }
844
  if (pUpdate->pass.ptr != NULL) {
833✔
845
    upObj.passLen = pUpdate->pass.len + 1;
595✔
846
    upObj.pass = taosMemoryCalloc(1, upObj.passLen);
595✔
847
    if (upObj.pass == NULL) goto _OVER;
595✔
848
    (void)memcpy(upObj.pass, pUpdate->pass.ptr, pUpdate->pass.len);
595✔
849
  }
850
  if (pUpdate->token.ptr != NULL) {
833✔
851
    upObj.tokenLen = pUpdate->token.len + 1;
238✔
852
    upObj.token = taosMemoryCalloc(1, upObj.tokenLen);
238✔
853
    if (upObj.token == NULL) goto _OVER;
238✔
854
    (void)memcpy(upObj.token, pUpdate->token.ptr, pUpdate->token.len);
238✔
855
  }
856

857
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "update-xnode-userpass");
833✔
858
  if (pTrans == NULL) {
833✔
UNCOV
859
    code = terrno;
×
UNCOV
860
    lino = __LINE__;
×
861
    goto _OVER;
×
862
  }
863
  mDebug("trans:%d, used to update xnode userpass or token:%d", pTrans->id, upObj.id);
833✔
864

865
  TAOS_CHECK_GOTO(mndSetCreateXnodeUserPassCommitLogs(pTrans, &upObj), NULL, _OVER);
833✔
866
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
833✔
867
  code = TSDB_CODE_ACTION_IN_PROGRESS;
833✔
868

869
_OVER:
833✔
870
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
833✔
UNCOV
871
    mError("xnode:failed to update xnode %d since %s, line:%d", upObj.id, tstrerror(code), lino);
×
872
  }
873
  taosMemoryFreeClear(upObj.user);
833✔
874
  taosMemoryFreeClear(upObj.pass);
833✔
875
  taosMemoryFreeClear(upObj.token);
833✔
876
  mndTransDrop(pTrans);
833✔
877
  TAOS_RETURN(code);
833✔
878
}
879

880
static int32_t mndProcessUpdateXnodeReq(SRpcMsg *pReq) {
833✔
881
  SMnode          *pMnode = pReq->info.node;
833✔
882
  int32_t          code = 0;
833✔
883
  SXnodeObj       *pObj = NULL;
833✔
884
  SMUpdateXnodeReq updateReq = {0};
833✔
885
  bool             isSuperUser = false;
833✔
886

887
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_UPDATE_XNODE);
833✔
888
  if (code != TSDB_CODE_SUCCESS) {
833✔
UNCOV
889
    mError("failed check permission for update xnode, code:%s", tstrerror(code));
×
UNCOV
890
    goto _OVER;
×
891
  }
892
  TAOS_CHECK_GOTO(tDeserializeSMUpdateXnodeReq(pReq->pCont, pReq->contLen, &updateReq), NULL, _OVER);
833✔
893

894
  code = mndXnodeCheckSuperUser(pMnode, pReq->info.conn.user, &isSuperUser);
833✔
895
  if (code != 0) return code;
833✔
896
  if (!isSuperUser) {
833✔
UNCOV
897
    return TSDB_CODE_MND_XNODE_NO_PRIV;
×
898
  }
899

900
  if (updateReq.token.ptr != NULL || (updateReq.user.ptr != NULL && updateReq.pass.ptr != NULL)) {
833✔
901
    code = mndUpdateXnodeUserPassToken(pMnode, pReq, &updateReq);
833✔
902
    if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
833✔
UNCOV
903
      goto _OVER;
×
904
    }
905
    mndRestartXnoded(pMnode);
833✔
906
  }
907

908
_OVER:
833✔
909
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
833✔
UNCOV
910
    mError("xnode:%d, failed to update since %s", updateReq.id, tstrerror(code));
×
911
  }
912
  mndReleaseXnode(pMnode, pObj);
833✔
913
  tFreeSMUpdateXnodeReq(&updateReq);
833✔
914
  TAOS_RETURN(code);
833✔
915
}
916

917
static int32_t mndSetDropXnodeRedoLogs(STrans *pTrans, SXnodeObj *pObj) {
1,428✔
918
  int32_t  code = 0;
1,428✔
919
  SSdbRaw *pRedoRaw = mndXnodeActionEncode(pObj);
1,428✔
920
  if (pRedoRaw == NULL) {
1,428✔
UNCOV
921
    code = terrno;
×
UNCOV
922
    TAOS_RETURN(code);
×
923
  }
924

925
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
1,428✔
926
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
1,428✔
927
  TAOS_RETURN(code);
1,428✔
928
}
929

930
static int32_t mndSetDropXnodeCommitLogs(STrans *pTrans, SXnodeObj *pObj) {
1,428✔
931
  int32_t  code = 0;
1,428✔
932
  SSdbRaw *pCommitRaw = mndXnodeActionEncode(pObj);
1,428✔
933
  if (pCommitRaw == NULL) {
1,428✔
UNCOV
934
    code = terrno;
×
UNCOV
935
    TAOS_RETURN(code);
×
936
  }
937

938
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
1,428✔
939
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
1,428✔
940
  TAOS_RETURN(code);
1,428✔
941
}
942

943
static int32_t mndSetDropXnodeInfoToTrans(SMnode *pMnode, STrans *pTrans, SXnodeObj *pObj, bool force) {
1,428✔
944
  if (pObj == NULL) return 0;
1,428✔
945
  TAOS_CHECK_RETURN(mndSetDropXnodeRedoLogs(pTrans, pObj));
1,428✔
946
  TAOS_CHECK_RETURN(mndSetDropXnodeCommitLogs(pTrans, pObj));
1,428✔
947
  return 0;
1,428✔
948
}
949

950
static int32_t mndDropXnode(SMnode *pMnode, SRpcMsg *pReq, SXnodeObj *pObj) {
1,428✔
951
  int32_t code = 0;
1,428✔
952
  int32_t lino = 0;
1,428✔
953

954
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_NOTHING, pReq, "drop-xnode");
1,428✔
955
  TSDB_CHECK_NULL(pTrans, code, lino, _OVER, terrno);
1,428✔
956

957
  mndTransSetSerial(pTrans);
1,428✔
958
  mDebug("trans:%d, to drop xnode:%d", pTrans->id, pObj->id);
1,428✔
959

960
  code = mndSetDropXnodeInfoToTrans(pMnode, pTrans, pObj, false);
1,428✔
961
  TSDB_CHECK_CODE(code, lino, _OVER);
1,428✔
962

963
  code = mndTransPrepare(pMnode, pTrans);
1,428✔
964

965
_OVER:
1,428✔
966
  mndTransDrop(pTrans);
1,428✔
967
  TAOS_RETURN(code);
1,428✔
968
}
969

970
static int32_t mndDrainXnode(SMnode *pMnode, SRpcMsg *pReq, SXnodeObj *pObj) {
238✔
971
  int32_t code = 0;
238✔
972
  int32_t lino = 0;
238✔
973

974
  SXnodeObj xnodeObj = {0};
238✔
975
  xnodeObj.id = pObj->id;
238✔
976
  xnodeObj.status = "drain";
238✔
977
  xnodeObj.statusLen = strlen(xnodeObj.status) + 1;
238✔
978
  xnodeObj.updateTime = taosGetTimestampMs();
238✔
979

980
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_NOTHING, pReq, "drain-xnode");
238✔
981
  TSDB_CHECK_NULL(pTrans, code, lino, _OVER, terrno);
238✔
982

983
  mndTransSetSerial(pTrans);
238✔
984
  mDebug("trans:%d, to drain xnode:%d", pTrans->id, xnodeObj.id);
238✔
985

986
  TAOS_CHECK_GOTO(mndSetCreateXnodeCommitLogs(pTrans, &xnodeObj), NULL, _OVER);
238✔
987
  code = mndTransPrepare(pMnode, pTrans);
238✔
988

989
_OVER:
238✔
990
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
238✔
UNCOV
991
    mError("xnode:%d, failed to drain since %s", xnodeObj.id, tstrerror(code));
×
992
  }
993
  mndTransDrop(pTrans);
238✔
994
  TAOS_RETURN(code);
238✔
995
}
996

997
static int32_t mndProcessDropXnodeReq(SRpcMsg *pReq) {
2,261✔
998
  SMnode        *pMnode = pReq->info.node;
2,261✔
999
  int32_t        code = -1;
2,261✔
1000
  SXnodeObj     *pObj = NULL;
2,261✔
1001
  SMDropXnodeReq dropReq = {0};
2,261✔
1002
  SJson         *pJson = NULL;
2,261✔
1003
  bool           isSuperUser = false;
2,261✔
1004

1005
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_DROP_XNODE);
2,261✔
1006
  if (code != TSDB_CODE_SUCCESS) {
2,261✔
UNCOV
1007
    mError("failed check permission for drop xnode, code:%s", tstrerror(code));
×
UNCOV
1008
    goto _OVER;
×
1009
  }
1010
  TAOS_CHECK_GOTO(tDeserializeSMDropXnodeReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
2,261✔
1011
  mDebug("xnode:%d, start to drop", dropReq.xnodeId);
2,261✔
1012

1013
  code = mndXnodeCheckSuperUser(pMnode, pReq->info.conn.user, &isSuperUser);
2,261✔
1014
  if (code != 0) return code;
2,261✔
1015
  if (!isSuperUser) {
2,261✔
UNCOV
1016
    return TSDB_CODE_MND_XNODE_NO_PRIV;
×
1017
  }
1018

1019
  if (dropReq.xnodeId <= 0 && (dropReq.url == NULL || strlen(dropReq.url) <= 0)) {
2,261✔
UNCOV
1020
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
1021
    goto _OVER;
×
1022
  }
1023

1024
  if (dropReq.url != NULL && strlen(dropReq.url) > 0) {
2,261✔
1025
    pObj = mndAcquireXnodeByURL(pMnode, dropReq.url);
1,785✔
1026
    if (pObj == NULL) {
1,785✔
1027
      code = terrno;
357✔
1028
      goto _OVER;
357✔
1029
    }
1030
  } else {
1031
    pObj = mndAcquireXnode(pMnode, dropReq.xnodeId);
476✔
1032
    if (pObj == NULL) {
476✔
1033
      code = terrno;
476✔
1034
      goto _OVER;
476✔
1035
    }
1036
  }
1037

1038
  // send request
1039
  char xnodeUrl[TSDB_XNODE_URL_LEN] = {0};
1,428✔
1040
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN, "%s/xnode/%d?force=%s", XNODED_PIPE_SOCKET_URL, pObj->id,
1,428✔
1041
           dropReq.force ? "true" : "false");
1,428✔
1042
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_DELETE, defaultTimeout, NULL, 0);
1,428✔
1043

1044
  code = mndDropXnode(pMnode, pReq, pObj);
1,428✔
1045
  if (code == 0) {
1,428✔
1046
    code = TSDB_CODE_ACTION_IN_PROGRESS;
1,428✔
1047
  }
1048

1049
_OVER:
2,261✔
1050
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
2,261✔
1051
    mError("xnode:%d, failed to drop since %s", dropReq.xnodeId, tstrerror(code));
833✔
1052
  }
1053
  if (pJson != NULL) {
2,261✔
UNCOV
1054
    tjsonDelete(pJson);
×
1055
  }
1056
  if (pObj != NULL) {
2,261✔
1057
    mndReleaseXnode(pMnode, pObj);
1,428✔
1058
  }
1059
  tFreeSMDropXnodeReq(&dropReq);
2,261✔
1060
  TAOS_RETURN(code);
2,261✔
1061
}
1062

1063
static int32_t mndProcessDrainXnodeReq(SRpcMsg *pReq) {
357✔
1064
  SMnode         *pMnode = pReq->info.node;
357✔
1065
  int32_t         code = 0;
357✔
1066
  SXnodeObj      *pObj = NULL;
357✔
1067
  SMDrainXnodeReq drainReq = {0};
357✔
1068
  SJson          *pJson = NULL;
357✔
1069
  SJson          *postContent = NULL;
357✔
1070
  char           *pContStr = NULL;
357✔
1071
  bool            isSuperUser = false;
357✔
1072

1073
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_DRAIN_XNODE);
357✔
1074
  if (code != TSDB_CODE_SUCCESS) {
357✔
UNCOV
1075
    mError("failed check permission for drain xnode, code:%s", tstrerror(code));
×
UNCOV
1076
    goto _OVER;
×
1077
  }
1078
  TAOS_CHECK_GOTO(tDeserializeSMDrainXnodeReq(pReq->pCont, pReq->contLen, &drainReq), NULL, _OVER);
357✔
1079
  mDebug("xnode:%d, start to drain", drainReq.xnodeId);
357✔
1080

1081
  code = mndXnodeCheckSuperUser(pMnode, pReq->info.conn.user, &isSuperUser);
357✔
1082
  if (code != 0) return code;
357✔
1083
  if (!isSuperUser) {
357✔
UNCOV
1084
    return TSDB_CODE_MND_XNODE_NO_PRIV;
×
1085
  }
1086

1087
  if (drainReq.xnodeId <= 0) {
357✔
UNCOV
1088
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
1089
    goto _OVER;
×
1090
  }
1091

1092
  pObj = mndAcquireXnode(pMnode, drainReq.xnodeId);
357✔
1093
  if (pObj == NULL) {
357✔
1094
    code = TSDB_CODE_MND_XNODE_NOT_EXIST;
119✔
1095
    goto _OVER;
119✔
1096
  }
1097

1098
  // send request
1099
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
238✔
1100
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/xnode/drain/%d", XNODED_PIPE_SOCKET_URL, pObj->id);
238✔
1101
  postContent = tjsonCreateObject();
238✔
1102
  if (postContent == NULL) {
238✔
UNCOV
1103
    code = terrno;
×
UNCOV
1104
    goto _OVER;
×
1105
  }
1106
  TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "xnode", pObj->url), NULL, _OVER);
238✔
1107
  pContStr = tjsonToString(postContent);
238✔
1108
  if (pContStr == NULL) {
238✔
UNCOV
1109
    code = terrno;
×
UNCOV
1110
    goto _OVER;
×
1111
  }
1112
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, defaultTimeout, pContStr, strlen(pContStr));
238✔
1113

1114
  code = mndDrainXnode(pMnode, pReq, pObj);
238✔
1115
  if (code == 0) {
238✔
1116
    code = TSDB_CODE_ACTION_IN_PROGRESS;
238✔
1117
    goto _OVER;
238✔
1118
  }
1119

1120
_OVER:
357✔
1121
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
357✔
1122
    mError("xnode:%d, failed to drain since %s", drainReq.xnodeId, tstrerror(code));
119✔
1123
  }
1124

1125
  if (postContent != NULL) {
357✔
1126
    tjsonDelete(postContent);
238✔
1127
  }
1128
  if (pContStr != NULL) {
357✔
1129
    taosMemoryFree(pContStr);
238✔
1130
  }
1131
  if (pJson != NULL) {
357✔
UNCOV
1132
    tjsonDelete(pJson);
×
1133
  }
1134
  mndReleaseXnode(pMnode, pObj);
357✔
1135
  tFreeSMDrainXnodeReq(&drainReq);
357✔
1136
  TAOS_RETURN(code);
357✔
1137
}
1138

1139
static int32_t mndRetrieveXnodes(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
2,380✔
1140
  SMnode    *pMnode = pReq->info.node;
2,380✔
1141
  SSdb      *pSdb = pMnode->pSdb;
2,380✔
1142
  int32_t    numOfRows = 0;
2,380✔
1143
  int32_t    cols = 0;
2,380✔
1144
  SXnodeObj *pObj = NULL;
2,380✔
1145
  char       buf[TSDB_XNODE_URL_LEN + VARSTR_HEADER_SIZE] = {0};
2,380✔
1146
  char       status[TSDB_XNODE_STATUS_LEN] = {0};
2,380✔
1147
  int32_t    code = 0;
2,380✔
1148
  mDebug("show.type:%d, %s:%d: retrieve xnodes with rows: %d", pShow->type, __FILE__, __LINE__, rows);
2,380✔
1149

1150
  while (numOfRows < rows) {
5,593✔
1151
    pShow->pIter = sdbFetch(pSdb, SDB_XNODE, pShow->pIter, (void **)&pObj);
5,593✔
1152
    if (pShow->pIter == NULL) break;
5,593✔
1153

1154
    cols = 0;
3,213✔
1155
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
3,213✔
1156
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->id, false);
3,213✔
1157
    if (code != 0) goto _end;
3,213✔
1158

1159
    STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->url, pShow->pMeta->pSchemas[cols].bytes);
3,213✔
1160
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
3,213✔
1161
    code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
3,213✔
1162
    if (code != 0) goto _end;
3,213✔
1163

1164
    if (mndGetXnodeStatus(pObj, status, TSDB_XNODE_STATUS_LEN) == 0) {
3,213✔
1165
      STR_TO_VARSTR(buf, status);
×
1166
    } else {
1167
      mDebug("xnode:%d, status request err: %s", pObj->id, tstrerror(terrno));
3,213✔
1168
      STR_TO_VARSTR(buf, "offline");
3,213✔
1169
    }
1170
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
3,213✔
1171
    code = colDataSetVal(pColInfo, numOfRows, buf, false);
3,213✔
1172
    if (code != 0) goto _end;
3,213✔
1173

1174
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
3,213✔
1175
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->createTime, false);
3,213✔
1176
    if (code != 0) goto _end;
3,213✔
1177

1178
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
3,213✔
1179
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->updateTime, false);
3,213✔
1180
    if (code != 0) goto _end;
3,213✔
1181

1182
    numOfRows++;
3,213✔
1183
    sdbRelease(pSdb, pObj);
3,213✔
1184
  }
1185

1186
_end:
2,380✔
1187
  if (code != 0) sdbRelease(pSdb, pObj);
2,380✔
1188

1189
  pShow->numOfRows += numOfRows;
2,380✔
1190
  return numOfRows;
2,380✔
1191
}
1192

UNCOV
1193
static void mndCancelGetNextXnode(SMnode *pMnode, void *pIter) {
×
1194
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
1195
  sdbCancelFetchByType(pSdb, pIter, SDB_XNODE);
×
UNCOV
1196
}
×
1197

1198
/** xnode task section **/
1199

1200
static SXnodeTaskObj *mndAcquireXnodeTaskById(SMnode *pMnode, int32_t tid) {
10,472✔
1201
  SSdb *pSdb = pMnode->pSdb;
10,472✔
1202

1203
  void *pIter = NULL;
10,472✔
1204
  while (1) {
×
1205
    SXnodeTaskObj *pTask = NULL;
10,472✔
1206
    pIter = sdbFetch(pSdb, SDB_XNODE_TASK, pIter, (void **)&pTask);
10,472✔
1207
    if (pIter == NULL) break;
10,472✔
1208

UNCOV
1209
    if (pTask->id == tid) {
×
UNCOV
1210
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
1211
      return pTask;
×
1212
    }
1213

UNCOV
1214
    sdbRelease(pSdb, pTask);
×
1215
  }
1216

1217
  mDebug("xnode task:%d, not found", tid);
10,472✔
1218
  terrno = TSDB_CODE_MND_XNODE_TASK_NOT_EXIST;
10,472✔
1219
  return NULL;
10,472✔
1220
}
1221
static SXnodeTaskObj *mndAcquireXnodeTaskByName(SMnode *pMnode, const char *name) {
7,735✔
1222
  SSdb *pSdb = pMnode->pSdb;
7,735✔
1223

1224
  void *pIter = NULL;
7,735✔
UNCOV
1225
  while (1) {
×
1226
    SXnodeTaskObj *pTask = NULL;
7,735✔
1227
    pIter = sdbFetch(pSdb, SDB_XNODE_TASK, pIter, (void **)&pTask);
7,735✔
1228
    if (pIter == NULL) break;
7,735✔
UNCOV
1229
    if (pTask->name == NULL) {
×
UNCOV
1230
      continue;
×
1231
    }
1232

UNCOV
1233
    if (strcasecmp(name, pTask->name) == 0) {
×
UNCOV
1234
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
1235
      return pTask;
×
1236
    }
1237

UNCOV
1238
    sdbRelease(pSdb, pTask);
×
1239
  }
1240

1241
  mDebug("xnode task:%s, not found", name);
7,735✔
1242
  terrno = TSDB_CODE_MND_XNODE_TASK_NOT_EXIST;
7,735✔
1243
  return NULL;
7,735✔
1244
}
1245

UNCOV
1246
static void mndFreeXnodeTask(SXnodeTaskObj *pObj) {
×
UNCOV
1247
  taosMemoryFreeClear(pObj->name);
×
UNCOV
1248
  taosMemoryFreeClear(pObj->sourceDsn);
×
UNCOV
1249
  taosMemoryFreeClear(pObj->sinkDsn);
×
UNCOV
1250
  taosMemoryFreeClear(pObj->parser);
×
UNCOV
1251
  taosMemoryFreeClear(pObj->reason);
×
UNCOV
1252
  taosMemoryFreeClear(pObj->status);
×
UNCOV
1253
  taosMemoryFreeClear(pObj->createdBy);
×
UNCOV
1254
  taosMemoryFreeClear(pObj->labels);
×
UNCOV
1255
}
×
1256

1257
SSdbRaw *mndXnodeTaskActionEncode(SXnodeTaskObj *pObj) {
582✔
1258
  int32_t code = 0;
582✔
1259
  int32_t lino = 0;
582✔
1260
  terrno = TSDB_CODE_OUT_OF_MEMORY;
582✔
1261
  if (NULL == pObj) {
582✔
1262
    terrno = TSDB_CODE_INVALID_PARA;
194✔
1263
    return NULL;
194✔
1264
  }
1265

1266
  int32_t totalStrLen = pObj->nameLen + pObj->sourceDsnLen + pObj->sinkDsnLen + pObj->parserLen + pObj->reasonLen +
388✔
1267
                        pObj->statusLen + pObj->createdByLen + pObj->labelsLen;
388✔
1268
  int32_t rawDataLen = sizeof(SXnodeTaskObj) + TSDB_XNODE_RESERVE_SIZE + totalStrLen;
388✔
1269

1270
  SSdbRaw *pRaw = sdbAllocRaw(SDB_XNODE_TASK, TSDB_XNODE_VER_NUMBER, rawDataLen);
388✔
1271
  if (pRaw == NULL) goto _OVER;
388✔
1272

1273
  int32_t dataPos = 0;
388✔
1274
  SDB_SET_INT32(pRaw, dataPos, pObj->id, _OVER)
388✔
1275
  SDB_SET_INT64(pRaw, dataPos, pObj->createTime, _OVER)
388✔
1276
  SDB_SET_INT64(pRaw, dataPos, pObj->updateTime, _OVER)
388✔
1277
  SDB_SET_INT32(pRaw, dataPos, pObj->statusLen, _OVER)
388✔
1278
  SDB_SET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
388✔
1279
  SDB_SET_INT32(pRaw, dataPos, pObj->via, _OVER)
388✔
1280
  SDB_SET_INT32(pRaw, dataPos, pObj->xnodeId, _OVER)
388✔
1281
  SDB_SET_INT32(pRaw, dataPos, pObj->nameLen, _OVER)
388✔
1282
  SDB_SET_BINARY(pRaw, dataPos, pObj->name, pObj->nameLen, _OVER)
388✔
1283
  SDB_SET_INT32(pRaw, dataPos, pObj->sourceType, _OVER)
388✔
1284
  SDB_SET_INT32(pRaw, dataPos, pObj->sourceDsnLen, _OVER)
388✔
1285
  SDB_SET_BINARY(pRaw, dataPos, pObj->sourceDsn, pObj->sourceDsnLen, _OVER)
388✔
1286
  SDB_SET_INT32(pRaw, dataPos, pObj->sinkType, _OVER)
388✔
1287
  SDB_SET_INT32(pRaw, dataPos, pObj->sinkDsnLen, _OVER)
388✔
1288
  SDB_SET_BINARY(pRaw, dataPos, pObj->sinkDsn, pObj->sinkDsnLen, _OVER)
388✔
1289
  SDB_SET_INT32(pRaw, dataPos, pObj->parserLen, _OVER)
388✔
1290
  SDB_SET_BINARY(pRaw, dataPos, pObj->parser, pObj->parserLen, _OVER)
388✔
1291
  SDB_SET_INT32(pRaw, dataPos, pObj->reasonLen, _OVER)
388✔
1292
  SDB_SET_BINARY(pRaw, dataPos, pObj->reason, pObj->reasonLen, _OVER)
388✔
1293
  SDB_SET_INT32(pRaw, dataPos, pObj->createdByLen, _OVER)
388✔
1294
  SDB_SET_BINARY(pRaw, dataPos, pObj->createdBy, pObj->createdByLen, _OVER)
388✔
1295
  SDB_SET_INT32(pRaw, dataPos, pObj->labelsLen, _OVER)
388✔
1296
  SDB_SET_BINARY(pRaw, dataPos, pObj->labels, pObj->labelsLen, _OVER)
388✔
1297

1298
  SDB_SET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
388✔
1299

1300
  terrno = 0;
388✔
1301

1302
_OVER:
388✔
1303
  if (terrno != 0) {
388✔
UNCOV
1304
    mError("xnode task:%d, failed to encode to raw:%p since %s", pObj->id, pRaw, terrstr());
×
UNCOV
1305
    sdbFreeRaw(pRaw);
×
UNCOV
1306
    return NULL;
×
1307
  }
1308

1309
  mTrace("xnode task:%d, encode to raw:%p, row:%p", pObj->id, pRaw, pObj);
388✔
1310
  return pRaw;
388✔
1311
}
1312

1313
SSdbRow *mndXnodeTaskActionDecode(SSdbRaw *pRaw) {
582✔
1314
  int32_t code = 0;
582✔
1315
  int32_t lino = 0;
582✔
1316
  terrno = TSDB_CODE_OUT_OF_MEMORY;
582✔
1317
  SSdbRow       *pRow = NULL;
582✔
1318
  SXnodeTaskObj *pObj = NULL;
582✔
1319

1320
  if (NULL == pRaw) {
582✔
1321
    terrno = TSDB_CODE_INVALID_PARA;
194✔
1322
    return NULL;
194✔
1323
  }
1324

1325
  int8_t sver = 0;
388✔
1326
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
388✔
1327

1328
  if (sver != TSDB_XNODE_VER_NUMBER) {
388✔
UNCOV
1329
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
UNCOV
1330
    goto _OVER;
×
1331
  }
1332

1333
  pRow = sdbAllocRow(sizeof(SXnodeTaskObj));
388✔
1334
  if (pRow == NULL) goto _OVER;
388✔
1335

1336
  pObj = sdbGetRowObj(pRow);
388✔
1337
  if (pObj == NULL) goto _OVER;
388✔
1338

1339
  int32_t dataPos = 0;
388✔
1340
  SDB_GET_INT32(pRaw, dataPos, &pObj->id, _OVER)
388✔
1341
  SDB_GET_INT64(pRaw, dataPos, &pObj->createTime, _OVER)
388✔
1342
  SDB_GET_INT64(pRaw, dataPos, &pObj->updateTime, _OVER)
388✔
1343
  SDB_GET_INT32(pRaw, dataPos, &pObj->statusLen, _OVER)
388✔
1344
  if (pObj->statusLen > 0) {
388✔
1345
    pObj->status = taosMemoryCalloc(pObj->statusLen + 1, 1);
388✔
1346
    if (pObj->status == NULL) goto _OVER;
388✔
1347
    SDB_GET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
388✔
1348
  }
1349

1350
  SDB_GET_INT32(pRaw, dataPos, &pObj->via, _OVER)
388✔
1351
  SDB_GET_INT32(pRaw, dataPos, &pObj->xnodeId, _OVER)
388✔
1352

1353
  SDB_GET_INT32(pRaw, dataPos, &pObj->nameLen, _OVER)
388✔
1354
  if (pObj->nameLen > 0) {
388✔
1355
    pObj->name = taosMemoryCalloc(pObj->nameLen + 1, 1);
388✔
1356
    if (pObj->name == NULL) goto _OVER;
388✔
1357
    SDB_GET_BINARY(pRaw, dataPos, pObj->name, pObj->nameLen, _OVER)
388✔
1358
  }
1359

1360
  SDB_GET_INT32(pRaw, dataPos, &pObj->sourceType, _OVER)
388✔
1361
  SDB_GET_INT32(pRaw, dataPos, &pObj->sourceDsnLen, _OVER)
388✔
1362
  if (pObj->sourceDsnLen > 0) {
388✔
1363
    pObj->sourceDsn = taosMemoryCalloc(pObj->sourceDsnLen + 1, 1);
388✔
1364
    if (pObj->sourceDsn == NULL) goto _OVER;
388✔
1365
    SDB_GET_BINARY(pRaw, dataPos, pObj->sourceDsn, pObj->sourceDsnLen, _OVER)
388✔
1366
  }
1367

1368
  SDB_GET_INT32(pRaw, dataPos, &pObj->sinkType, _OVER)
388✔
1369
  SDB_GET_INT32(pRaw, dataPos, &pObj->sinkDsnLen, _OVER)
388✔
1370
  if (pObj->sinkDsnLen > 0) {
388✔
1371
    pObj->sinkDsn = taosMemoryCalloc(pObj->sinkDsnLen + 1, 1);
388✔
1372
    if (pObj->sinkDsn == NULL) goto _OVER;
388✔
1373
    SDB_GET_BINARY(pRaw, dataPos, pObj->sinkDsn, pObj->sinkDsnLen, _OVER)
388✔
1374
  }
1375

1376
  SDB_GET_INT32(pRaw, dataPos, &pObj->parserLen, _OVER)
388✔
1377
  if (pObj->parserLen > 0) {
388✔
1378
    pObj->parser = taosMemoryCalloc(pObj->parserLen + 1, 1);
194✔
1379
    if (pObj->parser == NULL) goto _OVER;
194✔
1380
    SDB_GET_BINARY(pRaw, dataPos, pObj->parser, pObj->parserLen, _OVER)
194✔
1381
  }
1382

1383
  SDB_GET_INT32(pRaw, dataPos, &pObj->reasonLen, _OVER)
388✔
1384
  if (pObj->reasonLen > 0) {
388✔
1385
    pObj->reason = taosMemoryCalloc(pObj->reasonLen + 1, 1);
194✔
1386
    if (pObj->reason == NULL) goto _OVER;
194✔
1387
    SDB_GET_BINARY(pRaw, dataPos, pObj->reason, pObj->reasonLen, _OVER)
194✔
1388
  }
1389

1390
  SDB_GET_INT32(pRaw, dataPos, &pObj->createdByLen, _OVER)
388✔
1391
  if (pObj->createdByLen > 0) {
388✔
UNCOV
1392
    pObj->createdBy = taosMemoryCalloc(pObj->createdByLen + 1, 1);
×
UNCOV
1393
    if (pObj->createdBy == NULL) goto _OVER;
×
UNCOV
1394
    SDB_GET_BINARY(pRaw, dataPos, pObj->createdBy, pObj->createdByLen, _OVER)
×
1395
  }
1396

1397
  SDB_GET_INT32(pRaw, dataPos, &pObj->labelsLen, _OVER)
388✔
1398
  if (pObj->labelsLen > 0) {
388✔
UNCOV
1399
    pObj->labels = taosMemoryCalloc(pObj->labelsLen + 1, 1);
×
UNCOV
1400
    if (pObj->labels == NULL) goto _OVER;
×
1401
    SDB_GET_BINARY(pRaw, dataPos, pObj->labels, pObj->labelsLen, _OVER)
×
1402
  }
1403

1404
  SDB_GET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
388✔
1405

1406
  terrno = 0;
388✔
1407

1408
_OVER:
388✔
1409
  if (terrno != 0) {
388✔
1410
    mError("xnode task:%d, failed to decode from raw:%p since %s", pObj == NULL ? 0 : pObj->id, pRaw, terrstr());
×
1411
    if (pObj != NULL) {
×
1412
      taosMemoryFreeClear(pObj->name);
×
1413
      taosMemoryFreeClear(pObj->sourceDsn);
×
1414
      taosMemoryFreeClear(pObj->sinkDsn);
×
1415
      taosMemoryFreeClear(pObj->parser);
×
1416
      taosMemoryFreeClear(pObj->reason);
×
1417
      taosMemoryFreeClear(pObj->status);
×
1418
    }
1419
    taosMemoryFreeClear(pRow);
×
1420
    return NULL;
×
1421
  }
1422

1423
  mTrace("xnode:%d, decode from raw:%p, row:%p", pObj->id, pRaw, pObj);
388✔
1424
  return pRow;
388✔
1425
}
1426

1427
int32_t mndXnodeTaskActionInsert(SSdb *pSdb, SXnodeTaskObj *pObj) {
×
1428
  mDebug("xtask:%d, perform insert action, row:%p", pObj->id, pObj);
×
1429
  return 0;
×
1430
}
1431

1432
int32_t mndXnodeTaskActionDelete(SSdb *pSdb, SXnodeTaskObj *pObj) {
×
1433
  mDebug("xtask:%d, perform delete action, row:%p", pObj->id, pObj);
×
UNCOV
1434
  mndFreeXnodeTask(pObj);
×
1435
  return 0;
×
1436
}
1437

UNCOV
1438
int32_t mndXnodeTaskActionUpdate(SSdb *pSdb, SXnodeTaskObj *pOld, SXnodeTaskObj *pNew) {
×
UNCOV
1439
  mDebug("xtask:%d, perform update action, old row:%p new row:%p", pOld->id, pOld, pNew);
×
1440

1441
  taosWLockLatch(&pOld->lock);
×
1442
  pOld->via = pNew->via;
×
1443
  pOld->xnodeId = pNew->xnodeId;
×
1444
  swapFields(&pNew->statusLen, &pNew->status, &pOld->statusLen, &pOld->status);
×
1445
  swapFields(&pNew->nameLen, &pNew->name, &pOld->nameLen, &pOld->name);
×
1446
  swapFields(&pNew->sourceDsnLen, &pNew->sourceDsn, &pOld->sourceDsnLen, &pOld->sourceDsn);
×
UNCOV
1447
  swapFields(&pNew->sinkDsnLen, &pNew->sinkDsn, &pOld->sinkDsnLen, &pOld->sinkDsn);
×
1448
  swapFields(&pNew->parserLen, &pNew->parser, &pOld->parserLen, &pOld->parser);
×
1449
  swapFields(&pNew->reasonLen, &pNew->reason, &pOld->reasonLen, &pOld->reason);
×
1450
  swapFields(&pNew->labelsLen, &pNew->labels, &pOld->labelsLen, &pOld->labels);
×
UNCOV
1451
  if (pNew->updateTime > pOld->updateTime) {
×
UNCOV
1452
    pOld->updateTime = pNew->updateTime;
×
1453
  }
1454
  taosWUnLockLatch(&pOld->lock);
×
1455
  return 0;
×
1456
}
1457

1458
static int32_t mndSetCreateXnodeTaskRedoLogs(STrans *pTrans, SXnodeTaskObj *pObj) {
×
1459
  int32_t  code = 0;
×
UNCOV
1460
  SSdbRaw *pRedoRaw = mndXnodeTaskActionEncode(pObj);
×
1461
  if (pRedoRaw == NULL) {
×
1462
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1463
    if (terrno != 0) code = terrno;
×
UNCOV
1464
    TAOS_RETURN(code);
×
1465
  }
UNCOV
1466
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
×
UNCOV
1467
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_CREATING));
×
UNCOV
1468
  TAOS_RETURN(code);
×
1469
}
1470

UNCOV
1471
static int32_t mndSetCreateXnodeTaskUndoLogs(STrans *pTrans, SXnodeTaskObj *pObj) {
×
UNCOV
1472
  int32_t  code = 0;
×
UNCOV
1473
  SSdbRaw *pUndoRaw = mndXnodeTaskActionEncode(pObj);
×
UNCOV
1474
  if (pUndoRaw == NULL) {
×
UNCOV
1475
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1476
    if (terrno != 0) code = terrno;
×
UNCOV
1477
    TAOS_RETURN(code);
×
1478
  }
UNCOV
1479
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pUndoRaw));
×
UNCOV
1480
  TAOS_CHECK_RETURN(sdbSetRawStatus(pUndoRaw, SDB_STATUS_DROPPED));
×
1481
  TAOS_RETURN(code);
×
1482
}
1483

UNCOV
1484
static int32_t mndSetCreateXnodeTaskCommitLogs(STrans *pTrans, SXnodeTaskObj *pObj) {
×
1485
  int32_t  code = 0;
×
1486
  SSdbRaw *pCommitRaw = mndXnodeTaskActionEncode(pObj);
×
1487
  if (pCommitRaw == NULL) {
×
1488
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
1489
    if (terrno != 0) code = terrno;
×
1490
    TAOS_RETURN(code);
×
1491
  }
1492
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
×
1493
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY));
×
1494
  TAOS_RETURN(code);
×
1495
}
1496
void mndReleaseXnodeTask(SMnode *pMnode, SXnodeTaskObj *pObj) {
10,829✔
1497
  SSdb *pSdb = pMnode->pSdb;
10,829✔
1498
  sdbRelease(pSdb, pObj);
10,829✔
1499
}
10,829✔
1500

1501
static const char *getXTaskOptionByName(xTaskOptions *pOptions, const char *name) {
4,998✔
1502
  if (pOptions == NULL || name == NULL) return NULL;
4,998✔
1503
  for (int32_t i = 0; i < pOptions->optionsNum; ++i) {
10,710✔
1504
    CowStr option = pOptions->options[i];
7,616✔
1505
    if (option.ptr != NULL && strncasecmp(option.ptr, name, strlen(name)) == 0 && option.ptr[strlen(name)] == '=') {
7,616✔
1506
      return option.ptr + strlen(name) + 1;
1,904✔
1507
    }
1508
  }
1509
  return NULL;
3,094✔
1510
}
1511

1512
static int32_t mndCreateXnodeTask(SMnode *pMnode, SRpcMsg *pReq, SMCreateXnodeTaskReq *pCreate) {
×
1513
  int32_t code = -1;
×
UNCOV
1514
  STrans *pTrans = NULL;
×
1515

1516
  SXnodeTaskObj xnodeObj = {0};
×
1517
  xnodeObj.id = sdbGetMaxId(pMnode->pSdb, SDB_XNODE_TASK);
×
1518
  xnodeObj.createTime = taosGetTimestampMs();
×
1519
  xnodeObj.updateTime = xnodeObj.createTime;
×
1520
  xnodeObj.via = pCreate->options.via;
×
1521
  xnodeObj.xnodeId = pCreate->xnodeId;
×
1522

UNCOV
1523
  xnodeObj.nameLen = pCreate->name.len + 1;
×
1524
  xnodeObj.name = taosMemoryCalloc(1, xnodeObj.nameLen);
×
1525
  if (xnodeObj.name == NULL) goto _OVER;
×
1526
  (void)memcpy(xnodeObj.name, pCreate->name.ptr, pCreate->name.len);
×
1527

UNCOV
1528
  xnodeObj.sourceType = pCreate->source.type;
×
1529
  xnodeObj.sourceDsnLen = pCreate->source.cstr.len + 1;
×
1530
  xnodeObj.sourceDsn = taosMemoryCalloc(1, xnodeObj.sourceDsnLen);
×
1531
  if (xnodeObj.sourceDsn == NULL) goto _OVER;
×
1532
  (void)memcpy(xnodeObj.sourceDsn, pCreate->source.cstr.ptr, pCreate->source.cstr.len);
×
1533

1534
  xnodeObj.sinkType = pCreate->sink.type;
×
UNCOV
1535
  xnodeObj.sinkDsnLen = pCreate->sink.cstr.len + 1;
×
UNCOV
1536
  xnodeObj.sinkDsn = taosMemoryCalloc(1, xnodeObj.sinkDsnLen);
×
1537
  if (xnodeObj.sinkDsn == NULL) goto _OVER;
×
1538
  (void)memcpy(xnodeObj.sinkDsn, pCreate->sink.cstr.ptr, pCreate->sink.cstr.len);
×
1539

1540
  xnodeObj.parserLen = pCreate->options.parser.len + 1;
×
1541
  if (pCreate->options.parser.ptr != NULL) {
×
UNCOV
1542
    xnodeObj.parser = taosMemoryCalloc(1, xnodeObj.parserLen);
×
1543
    if (xnodeObj.parser == NULL) goto _OVER;
×
1544
    (void)memcpy(xnodeObj.parser, pCreate->options.parser.ptr, pCreate->options.parser.len);
×
1545
  }
1546

UNCOV
1547
  const char *status = getXTaskOptionByName(&pCreate->options, "status");
×
1548
  if (status != NULL) {
×
UNCOV
1549
    xnodeObj.statusLen = strlen(status) + 1;
×
1550
    xnodeObj.status = taosMemoryCalloc(1, xnodeObj.statusLen);
×
1551
    if (xnodeObj.status == NULL) goto _OVER;
×
1552
    (void)memcpy(xnodeObj.status, status, xnodeObj.statusLen - 1);
×
1553
  }
1554

1555
  xnodeObj.createdByLen = strlen(pReq->info.conn.user) + 1;
×
UNCOV
1556
  xnodeObj.createdBy = taosMemoryCalloc(1, xnodeObj.createdByLen);
×
1557
  if (xnodeObj.createdBy == NULL) goto _OVER;
×
1558
  (void)memcpy(xnodeObj.createdBy, pReq->info.conn.user, xnodeObj.createdByLen - 1);
×
1559

1560
  const char *labels = getXTaskOptionByName(&pCreate->options, "labels");
×
UNCOV
1561
  if (labels != NULL) {
×
UNCOV
1562
    xnodeObj.labelsLen = strlen(labels) + 1;
×
UNCOV
1563
    xnodeObj.labels = taosMemoryCalloc(1, xnodeObj.labelsLen);
×
UNCOV
1564
    if (xnodeObj.labels == NULL) goto _OVER;
×
UNCOV
1565
    (void)memcpy(xnodeObj.labels, labels, xnodeObj.labelsLen - 1);
×
1566
  }
1567

UNCOV
1568
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-xnode-task");
×
UNCOV
1569
  if (pTrans == NULL) {
×
UNCOV
1570
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
1571
    if (terrno != 0) {
×
UNCOV
1572
      code = terrno;
×
1573
    }
UNCOV
1574
    mError("failed to create transaction for xnode-task:%s, code:0x%x:%s", pCreate->name.ptr, code, tstrerror(code));
×
UNCOV
1575
    goto _OVER;
×
1576
  }
UNCOV
1577
  mndTransSetSerial(pTrans);
×
1578

1579
  mDebug("trans:%d, used to create xnode task:%s as task:%d", pTrans->id, pCreate->name.ptr, xnodeObj.id);
×
1580

UNCOV
1581
  TAOS_CHECK_GOTO(mndSetCreateXnodeTaskRedoLogs(pTrans, &xnodeObj), NULL, _OVER);
×
UNCOV
1582
  TAOS_CHECK_GOTO(mndSetCreateXnodeTaskUndoLogs(pTrans, &xnodeObj), NULL, _OVER);
×
1583
  TAOS_CHECK_GOTO(mndSetCreateXnodeTaskCommitLogs(pTrans, &xnodeObj), NULL, _OVER);
×
1584
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
×
1585

UNCOV
1586
  code = 0;
×
1587

UNCOV
1588
_OVER:
×
UNCOV
1589
  mndFreeXnodeTask(&xnodeObj);
×
1590
  mndTransDrop(pTrans);
×
1591
  TAOS_RETURN(code);
×
1592
}
1593

1594
// Helper function to parse and validate the request
1595
static int32_t mndValidateCreateXnodeTaskReq(SRpcMsg *pReq, SMCreateXnodeTaskReq *pCreateReq) {
3,927✔
1596
  int32_t code = 0;
3,927✔
1597
  SJson  *pJson = NULL;
3,927✔
1598
  SJson  *postContent = NULL;
3,927✔
1599
  char   *srcDsn = NULL;
3,927✔
1600
  char   *sinkDsn = NULL;
3,927✔
1601
  char   *parser = NULL;
3,927✔
1602
  char   *pContStr = NULL;
3,927✔
1603

1604
  // from, to, parser check
1605
  char xnodeUrl[TSDB_XNODE_URL_LEN] = {0};
3,927✔
1606
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN, "%s/task/check", XNODED_PIPE_SOCKET_URL);
3,927✔
1607
  postContent = tjsonCreateObject();
3,927✔
1608
  if (postContent == NULL) {
3,927✔
UNCOV
1609
    code = terrno;
×
UNCOV
1610
    goto _OVER;
×
1611
  }
1612
  srcDsn = taosStrndupi(pCreateReq->source.cstr.ptr, (int64_t)pCreateReq->source.cstr.len);
3,927✔
1613
  if (srcDsn == NULL) {
3,927✔
UNCOV
1614
    code = terrno;
×
UNCOV
1615
    goto _OVER;
×
1616
  }
1617
  TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "from", srcDsn), NULL, _OVER);
3,927✔
1618

1619
  sinkDsn = taosStrndupi(pCreateReq->sink.cstr.ptr, (int64_t)pCreateReq->sink.cstr.len);
3,927✔
1620
  if (sinkDsn == NULL) {
3,927✔
1621
    code = terrno;
×
1622
    goto _OVER;
×
1623
  }
1624
  TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "to", sinkDsn), NULL, _OVER);
3,927✔
1625

1626
  if (pCreateReq->options.parser.len > 0 && pCreateReq->options.parser.ptr != NULL) {
3,927✔
1627
    parser = taosStrndupi(pCreateReq->options.parser.ptr, (int64_t)pCreateReq->options.parser.len);
2,499✔
1628
    if (parser == NULL) {
2,499✔
UNCOV
1629
      code = terrno;
×
1630
      goto _OVER;
×
1631
    }
1632
    TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "parser", parser), NULL, _OVER);
2,499✔
1633
  }
1634

1635
  if (pCreateReq->xnodeId > 0) {
3,927✔
1636
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(postContent, "xnode_id", (double)pCreateReq->xnodeId), NULL, _OVER);
×
1637
  }
1638

1639
  if (pCreateReq->options.via > 0) {
3,927✔
1640
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(postContent, "via", (double)pCreateReq->options.via), NULL, _OVER);
357✔
1641
  }
1642

1643
  const char *labels = getXTaskOptionByName(&pCreateReq->options, "labels");
3,927✔
1644
  if (labels != NULL) {
3,927✔
1645
    TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "labels", labels), NULL, _OVER);
1,547✔
1646
  }
1647

1648
  TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "created_by", pReq->info.conn.user), NULL, _OVER);
3,927✔
1649

1650
  pContStr = tjsonToUnformattedString(postContent);
3,927✔
1651
  if (pContStr == NULL) {
3,927✔
UNCOV
1652
    code = terrno;
×
UNCOV
1653
    goto _OVER;
×
1654
  }
1655

1656
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, 60000, pContStr, strlen(pContStr));
3,927✔
1657
  if (pJson == NULL) {
3,927✔
1658
    code = terrno;
3,927✔
1659
    goto _OVER;
3,927✔
1660
  }
UNCOV
1661
  SJson *errorJson = tjsonGetObjectItem(pJson, "__inner_error");
×
UNCOV
1662
  if (errorJson != NULL) {
×
UNCOV
1663
    code = TSDB_CODE_MND_XNODE_HTTP_CODE_ERROR;
×
UNCOV
1664
    char* pValueString = ((cJSON*)errorJson)->valuestring;
×
UNCOV
1665
    if (NULL == pValueString) {
×
UNCOV
1666
      mError("should not failed to get __inner_error message, task name:%s", pCreateReq->name.ptr);
×
UNCOV
1667
      goto _OVER;
×
1668
    }
1669
    //handle response
1670
    int32_t contLen = strlen(pValueString) + strlen(tstrerror(code)) + 32;
×
1671
    void *pRsp = rpcMallocCont(contLen);
×
UNCOV
1672
    if (pRsp == NULL) {
×
UNCOV
1673
      TAOS_CHECK_GOTO(terrno, NULL, _OVER);
×
1674
    }
UNCOV
1675
    pReq->info.rspLen = contLen;
×
UNCOV
1676
    pReq->info.rsp = pRsp;
×
1677
    snprintf(pReq->info.rsp, contLen, "%s, since: %s", tstrerror(code), pValueString);
×
1678
    goto _OVER;
×
1679
  }
1680

1681
  // todo: only4test
1682
  // (void)mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, 60000, pContStr, strlen(pContStr));
1683
  // code = TSDB_CODE_SUCCESS;
1684

1685
_OVER:
3,927✔
1686
  if (srcDsn != NULL) taosMemoryFreeClear(srcDsn);
3,927✔
1687
  if (sinkDsn != NULL) taosMemoryFreeClear(sinkDsn);
3,927✔
1688
  if (parser != NULL) taosMemoryFreeClear(parser);
3,927✔
1689
  if (pContStr != NULL) taosMemoryFreeClear(pContStr);
3,927✔
1690
  if (postContent != NULL) tjsonDelete(postContent);
3,927✔
1691
  if (pJson != NULL) tjsonDelete(pJson);
3,927✔
1692

1693
  TAOS_RETURN(code);
3,927✔
1694
}
1695

1696
// Helper function to check if xnode task already exists
1697
static int32_t mndCheckXnodeTaskExists(SMnode *pMnode, const char *name) {
3,927✔
1698
  SXnodeTaskObj *pObj = mndAcquireXnodeTaskByName(pMnode, name);
3,927✔
1699
  if (pObj != NULL) {
3,927✔
1700
    mError("xnode task:%s already exists", name);
×
UNCOV
1701
    mndReleaseXnodeTask(pMnode, pObj);
×
UNCOV
1702
    return TSDB_CODE_MND_XNODE_TASK_ALREADY_EXIST;
×
1703
  }
1704
  return TSDB_CODE_SUCCESS;
3,927✔
1705
}
1706

1707
// Helper function to handle the creation result
UNCOV
1708
static int32_t mndHandleCreateXnodeTaskResult(int32_t createCode) {
×
UNCOV
1709
  if (createCode == 0) {
×
1710
    return TSDB_CODE_ACTION_IN_PROGRESS;
×
1711
  }
UNCOV
1712
  return createCode;
×
1713
}
1714

1715
static int32_t mndProcessCreateXnodeTaskReq(SRpcMsg *pReq) {
3,927✔
1716
  mDebug("xnode create task request received, contLen:%d\n", pReq->contLen);
3,927✔
1717
  SMnode              *pMnode = pReq->info.node;
3,927✔
1718
  int32_t              code = 0;
3,927✔
1719
  SMCreateXnodeTaskReq createReq = {0};
3,927✔
1720

1721
  // Step 1: Validate permissions
1722
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_CREATE_XNODE_TASK);
3,927✔
1723
  if (code != TSDB_CODE_SUCCESS) {
3,927✔
UNCOV
1724
    mError("failed check permission for create xnode task, code:%s", tstrerror(code));
×
UNCOV
1725
    goto _OVER;
×
1726
  }
1727

1728
  code = tDeserializeSMCreateXnodeTaskReq(pReq->pCont, pReq->contLen, &createReq);
3,927✔
1729
  if (code != 0) {
3,927✔
UNCOV
1730
    mError("failed to deserialize create xnode task request, code:%s", tstrerror(code));
×
UNCOV
1731
    TAOS_RETURN(code);
×
1732
  }
1733

1734
  // Step 2: Check if task already exists
1735
  TAOS_CHECK_GOTO(mndCheckXnodeTaskExists(pMnode, createReq.name.ptr), NULL, _OVER);
3,927✔
1736

1737
  // Step 3: Parse and validate request
1738
  TAOS_CHECK_GOTO(mndValidateCreateXnodeTaskReq(pReq, &createReq), NULL, _OVER);
3,927✔
1739

1740
  // Step 4: Create the xnode task
1741
  TAOS_CHECK_GOTO(mndCreateXnodeTask(pMnode, pReq, &createReq), NULL, _OVER);
×
1742
  TAOS_CHECK_GOTO(mndHandleCreateXnodeTaskResult(code), NULL, _OVER);
×
1743

1744
_OVER:
3,927✔
1745
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
3,927✔
1746
    mError("xnode task:%s, failed to create since %s", createReq.name.ptr ? createReq.name.ptr : "unknown",
3,927✔
1747
           tstrerror(code));
1748
  }
1749
  tFreeSMCreateXnodeTaskReq(&createReq);
3,927✔
1750
  TAOS_RETURN(code);
3,927✔
1751
}
1752

UNCOV
1753
static int32_t httpStartXnodeTask(SXnodeTaskObj *pObj) {
×
1754
  int32_t code = 0;
×
1755
  struct {
1756
    char   xnodeUrl[TSDB_XNODE_URL_LEN + 1];
1757
    SJson *postContent;
1758
    SJson *pJson;
1759
    char  *pContStr;
1760
    char  *srcDsn;
1761
    char  *sinkDsn;
1762
    char  *parser;
1763
  } req = {0};
×
1764

UNCOV
1765
  snprintf(req.xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/task/%d", XNODED_PIPE_SOCKET_URL, pObj->id);
×
UNCOV
1766
  req.postContent = tjsonCreateObject();
×
1767
  if (req.postContent == NULL) {
×
1768
    code = terrno;
×
UNCOV
1769
    goto _OVER;
×
1770
  }
1771
  req.srcDsn = taosStrndupi(pObj->sourceDsn, (int64_t)pObj->sourceDsnLen);
×
1772
  if (req.srcDsn == NULL) {
×
UNCOV
1773
    code = terrno;
×
UNCOV
1774
    goto _OVER;
×
1775
  }
1776
  TAOS_CHECK_GOTO(tjsonAddStringToObject(req.postContent, "from", req.srcDsn), NULL, _OVER);
×
1777

UNCOV
1778
  req.sinkDsn = taosStrndupi(pObj->sinkDsn, (int64_t)pObj->sinkDsnLen);
×
1779
  if (req.sinkDsn == NULL) {
×
1780
    code = terrno;
×
1781
    goto _OVER;
×
1782
  }
UNCOV
1783
  TAOS_CHECK_GOTO(tjsonAddStringToObject(req.postContent, "to", req.sinkDsn), NULL, _OVER);
×
1784

1785
  if (pObj->parserLen > 0) {
×
UNCOV
1786
    req.parser = taosStrndupi(pObj->parser, (int64_t)pObj->parserLen);
×
1787
    if (req.parser == NULL) {
×
1788
      code = terrno;
×
1789
      goto _OVER;
×
1790
    }
1791
    TAOS_CHECK_GOTO(tjsonAddStringToObject(req.postContent, "parser", req.parser), NULL, _OVER);
×
1792
  }
1793

1794
  if (pObj->xnodeId > 0) {
×
UNCOV
1795
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(req.postContent, "xnode_id", (double)pObj->xnodeId), NULL, _OVER);
×
1796
  }
1797

UNCOV
1798
  if (pObj->via > 0) {
×
UNCOV
1799
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(req.postContent, "via", (double)pObj->via), NULL, _OVER);
×
1800
  }
1801

UNCOV
1802
  if (pObj->createdBy != NULL) {
×
UNCOV
1803
    TAOS_CHECK_GOTO(tjsonAddStringToObject(req.postContent, "created_by", pObj->createdBy), NULL, _OVER);
×
1804
  }
1805

1806
  if (pObj->labels != NULL) {
×
1807
    TAOS_CHECK_GOTO(tjsonAddStringToObject(req.postContent, "labels", pObj->labels), NULL, _OVER);
×
1808
  }
1809

UNCOV
1810
  req.pContStr = tjsonToUnformattedString(req.postContent);
×
UNCOV
1811
  if (req.pContStr == NULL) {
×
UNCOV
1812
    code = terrno;
×
1813
    goto _OVER;
×
1814
  }
UNCOV
1815
  mDebug("start xnode post content:%s", req.pContStr);
×
UNCOV
1816
  req.pJson = mndSendReqRetJson(req.xnodeUrl, HTTP_TYPE_POST, defaultTimeout, req.pContStr, strlen(req.pContStr));
×
1817

1818
_OVER:
×
UNCOV
1819
  if (req.pContStr != NULL) taosMemoryFreeClear(req.pContStr);
×
UNCOV
1820
  if (req.postContent != NULL) tjsonDelete(req.postContent);
×
UNCOV
1821
  if (req.pJson != NULL) tjsonDelete(req.pJson);
×
UNCOV
1822
  if (req.srcDsn != NULL) taosMemoryFreeClear(req.srcDsn);
×
UNCOV
1823
  if (req.sinkDsn != NULL) taosMemoryFreeClear(req.sinkDsn);
×
UNCOV
1824
  if (req.parser != NULL) taosMemoryFreeClear(req.parser);
×
UNCOV
1825
  TAOS_RETURN(code);
×
1826
}
1827

1828
static int32_t mndProcessStartXnodeTaskReq(SRpcMsg *pReq) {
119✔
1829
  SMnode             *pMnode = pReq->info.node;
119✔
1830
  int32_t             code = 0;
119✔
1831
  SXnodeTaskObj      *pObj = NULL;
119✔
1832
  SMStartXnodeTaskReq startReq = {0};
119✔
1833
  SXnodeTaskObj      *pObjClone = NULL;
119✔
1834

1835
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_START_XNODE_TASK);
119✔
1836
  if (code != TSDB_CODE_SUCCESS) {
119✔
1837
    mError("failed check permission for start xnode task, code:%s", tstrerror(code));
×
1838
    goto _OVER;
×
1839
  }
1840
  TAOS_CHECK_GOTO(tDeserializeSMStartXnodeTaskReq(pReq->pCont, pReq->contLen, &startReq), NULL, _OVER);
119✔
1841
  mDebug("xnode start xnode task with tid:%d", startReq.tid);
119✔
1842

1843
  if (startReq.tid <= 0 && (startReq.name.len <= 0 || startReq.name.ptr == NULL)) {
119✔
UNCOV
1844
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
1845
    goto _OVER;
×
1846
  }
1847

1848
  if (startReq.tid > 0) {
119✔
UNCOV
1849
    pObj = mndAcquireXnodeTask(pMnode, startReq.tid);
×
1850
  } else {
1851
    pObj = mndAcquireXnodeTaskByName(pMnode, startReq.name.ptr);
119✔
1852
  }
1853
  if (pObj == NULL) {
119✔
1854
    code = terrno;
119✔
1855
    goto _OVER;
119✔
1856
  }
UNCOV
1857
  code = mndCheckXnodeTaskOwner(pMnode, pReq->info.conn.user, pObj);
×
UNCOV
1858
  if (code != TSDB_CODE_SUCCESS) {
×
1859
    mError("user:%s has no privilege on xnode task:%d", pReq->info.conn.user, pObj->id);
×
1860
    goto _OVER;
×
1861
  }
UNCOV
1862
  (void)httpStartXnodeTask(pObj);
×
1863

1864
_OVER:
119✔
1865
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
119✔
1866
    mError("xnode task:%d, failed to start since %s", startReq.tid, tstrerror(code));
119✔
1867
  }
1868
  tFreeSMStartXnodeTaskReq(&startReq);
119✔
1869
  if (pObj != NULL) {
119✔
UNCOV
1870
    mndReleaseXnodeTask(pMnode, pObj);
×
1871
  }
1872
  if (pObjClone != NULL) {
119✔
UNCOV
1873
    mndFreeXnodeTask(pObjClone);
×
1874
    taosMemFree(pObjClone);
×
1875
  }
1876
  TAOS_RETURN(code);
119✔
1877
}
1878

1879
static int32_t mndProcessStopXnodeTaskReq(SRpcMsg *pReq) {
119✔
1880
  SMnode            *pMnode = pReq->info.node;
119✔
1881
  int32_t            code = -1;
119✔
1882
  SXnodeTaskObj     *pObj = NULL;
119✔
1883
  SMStopXnodeTaskReq stopReq = {0};
119✔
1884
  SJson             *pJson = NULL;
119✔
1885

1886
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_STOP_XNODE_TASK);
119✔
1887
  if (code != TSDB_CODE_SUCCESS) {
119✔
UNCOV
1888
    mError("failed check permission for stop xnode task, code:%s", tstrerror(code));
×
1889
    goto _OVER;
×
1890
  }
1891
  TAOS_CHECK_GOTO(tDeserializeSMStopXnodeTaskReq(pReq->pCont, pReq->contLen, &stopReq), NULL, _OVER);
119✔
1892
  mDebug("Stop xnode task with tid:%d", stopReq.tid);
119✔
1893

1894
  if (stopReq.tid <= 0 && (stopReq.name.len <= 0 || stopReq.name.ptr == NULL)) {
119✔
UNCOV
1895
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
1896
    goto _OVER;
×
1897
  }
1898

1899
  if (stopReq.tid > 0) {
119✔
UNCOV
1900
    pObj = mndAcquireXnodeTask(pMnode, stopReq.tid);
×
1901
  } else {
1902
    pObj = mndAcquireXnodeTaskByName(pMnode, stopReq.name.ptr);
119✔
1903
  }
1904
  if (pObj == NULL) {
119✔
1905
    code = terrno;
119✔
1906
    goto _OVER;
119✔
1907
  }
1908
  code = mndCheckXnodeTaskOwner(pMnode, pReq->info.conn.user, pObj);
×
1909
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
1910
    mError("user:%s has no privilege on xnode task:%d", pReq->info.conn.user, pObj->id);
×
1911
    goto _OVER;
×
1912
  }
1913

1914
  // send request
1915
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
×
1916
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/task/%d", XNODED_PIPE_SOCKET_URL, pObj->id);
×
UNCOV
1917
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_DELETE, defaultTimeout, NULL, 0);
×
1918

1919
_OVER:
119✔
1920
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
119✔
1921
    mError("xnode task:%d, failed to stop since %s", stopReq.tid, tstrerror(code));
119✔
1922
  }
1923
  if (pJson != NULL) {
119✔
1924
    tjsonDelete(pJson);
×
1925
  }
1926
  tFreeSMStopXnodeTaskReq(&stopReq);
119✔
1927
  TAOS_RETURN(code);
119✔
1928
}
1929

UNCOV
1930
static int32_t mndUpdateXnodeTask(SMnode *pMnode, SRpcMsg *pReq, const SXnodeTaskObj *pOld,
×
1931
                                  SMUpdateXnodeTaskReq *pUpdate) {
1932
  mDebug("xnode task:%d, start to update", pUpdate->tid);
×
1933
  int32_t      code = -1;
×
1934
  STrans       *pTrans = NULL;
×
1935
  SXnodeTaskObj taskObj = *pOld;
×
1936
  struct {
1937
    bool status;
1938
    bool name;
1939
    bool source;
1940
    bool sink;
1941
    bool parser;
1942
    bool reason;
1943
    bool labels;
1944
  } isChange = {0};
×
1945

1946
  if (pUpdate->via > 0) {
×
1947
    taskObj.via = pUpdate->via;
×
1948
  }
UNCOV
1949
  if (pUpdate->xnodeId > 0) {
×
1950
    taskObj.xnodeId = pUpdate->xnodeId;
×
1951
  }
UNCOV
1952
  if (pUpdate->status.ptr != NULL) {
×
1953
    taskObj.statusLen = pUpdate->status.len + 1;
×
1954
    taskObj.status = taosMemoryCalloc(1, taskObj.statusLen);
×
1955
    if (taskObj.status == NULL) {
×
1956
      code = terrno;
×
1957
      goto _OVER;
×
1958
    }
UNCOV
1959
    (void)memcpy(taskObj.status, pUpdate->status.ptr, pUpdate->status.len);
×
1960
    isChange.status = true;
×
1961
  }
UNCOV
1962
  if (pUpdate->updateName.ptr != NULL) {
×
1963
    taskObj.nameLen = pUpdate->updateName.len + 1;
×
1964
    taskObj.name = taosMemoryCalloc(1, taskObj.nameLen);
×
1965
    if (taskObj.name == NULL) {
×
1966
      code = terrno;
×
1967
      goto _OVER;
×
1968
    }
UNCOV
1969
    (void)memcpy(taskObj.name, pUpdate->updateName.ptr, pUpdate->updateName.len);
×
1970
    isChange.name = true;
×
1971
  }
UNCOV
1972
  if (pUpdate->source.cstr.ptr != NULL) {
×
1973
    taskObj.sourceType = pUpdate->source.type;
×
1974
    taskObj.sourceDsnLen = pUpdate->source.cstr.len + 1;
×
1975
    taskObj.sourceDsn = taosMemoryCalloc(1, taskObj.sourceDsnLen);
×
1976
    if (taskObj.sourceDsn == NULL) {
×
1977
      code = terrno;
×
1978
      goto _OVER;
×
1979
    }
1980
    (void)memcpy(taskObj.sourceDsn, pUpdate->source.cstr.ptr, pUpdate->source.cstr.len);
×
1981
    isChange.source = true;
×
1982
  }
1983
  if (pUpdate->sink.cstr.ptr != NULL) {
×
UNCOV
1984
    taskObj.sinkType = pUpdate->sink.type;
×
1985
    taskObj.sinkDsnLen = pUpdate->sink.cstr.len + 1;
×
1986
    taskObj.sinkDsn = taosMemoryCalloc(1, taskObj.sinkDsnLen);
×
1987
    if (taskObj.sinkDsn == NULL) {
×
1988
      code = terrno;
×
1989
      goto _OVER;
×
1990
    }
1991
    (void)memcpy(taskObj.sinkDsn, pUpdate->sink.cstr.ptr, pUpdate->sink.cstr.len);
×
UNCOV
1992
    isChange.sink = true;
×
1993
  }
1994
  if (pUpdate->parser.ptr != NULL) {
×
1995
    taskObj.parserLen = pUpdate->parser.len + 1;
×
UNCOV
1996
    taskObj.parser = taosMemoryCalloc(1, taskObj.parserLen);
×
1997
    if (taskObj.parser == NULL) {
×
1998
      code = terrno;
×
1999
      goto _OVER;
×
2000
    }
2001
    (void)memcpy(taskObj.parser, pUpdate->parser.ptr, pUpdate->parser.len);
×
2002
    isChange.parser = true;
×
2003
  }
2004
  if (pUpdate->reason.ptr != NULL) {
×
2005
    taskObj.reasonLen = pUpdate->reason.len + 1;
×
UNCOV
2006
    taskObj.reason = taosMemoryCalloc(1, taskObj.reasonLen);
×
2007
    if (taskObj.reason == NULL) {
×
2008
      code = terrno;
×
UNCOV
2009
      goto _OVER;
×
2010
    }
2011
    (void)memcpy(taskObj.reason, pUpdate->reason.ptr, pUpdate->reason.len);
×
UNCOV
2012
    isChange.reason = true;
×
2013
  }
2014
  if (pUpdate->labels.ptr != NULL) {
×
UNCOV
2015
    taskObj.labelsLen = pUpdate->labels.len + 1;
×
2016
    taskObj.labels = taosMemoryCalloc(1, taskObj.labelsLen);
×
2017
    if (taskObj.labels == NULL) {
×
UNCOV
2018
      code = terrno;
×
2019
      goto _OVER;
×
2020
    }
UNCOV
2021
    (void)memcpy(taskObj.labels, pUpdate->labels.ptr, pUpdate->labels.len);
×
UNCOV
2022
    isChange.labels = true;
×
2023
  }
UNCOV
2024
  taskObj.updateTime = taosGetTimestampMs();
×
2025

UNCOV
2026
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "update-xnode-task");
×
UNCOV
2027
  if (pTrans == NULL) {
×
UNCOV
2028
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2029
    if (terrno != 0) code = terrno;
×
UNCOV
2030
    goto _OVER;
×
2031
  }
2032
  mInfo("trans:%d, used to update xnode task:%d", pTrans->id, taskObj.id);
×
2033

UNCOV
2034
  TAOS_CHECK_GOTO(mndSetCreateXnodeTaskCommitLogs(pTrans, &taskObj), NULL, _OVER);
×
UNCOV
2035
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
×
UNCOV
2036
  code = 0;
×
2037

UNCOV
2038
_OVER:
×
UNCOV
2039
  if (NULL != taskObj.name && isChange.name) {
×
UNCOV
2040
    taosMemoryFree(taskObj.name);
×
2041
  }
UNCOV
2042
  if (NULL != taskObj.status && isChange.status) {
×
UNCOV
2043
    taosMemoryFree(taskObj.status);
×
2044
  }
UNCOV
2045
  if (NULL != taskObj.sourceDsn && isChange.source) {
×
UNCOV
2046
    taosMemoryFree(taskObj.sourceDsn);
×
2047
  }
2048
  if (NULL != taskObj.sinkDsn && isChange.sink) {
×
2049
    taosMemoryFree(taskObj.sinkDsn);
×
2050
  }
2051
  if (NULL != taskObj.parser && isChange.parser) {
×
2052
    taosMemoryFree(taskObj.parser);
×
2053
  }
UNCOV
2054
  if (NULL != taskObj.reason && isChange.reason) {
×
UNCOV
2055
    taosMemoryFree(taskObj.reason);
×
2056
  }
2057
  if (NULL != taskObj.labels && isChange.labels) {
×
UNCOV
2058
    taosMemoryFree(taskObj.labels);
×
2059
  }
UNCOV
2060
  mndTransDrop(pTrans);
×
UNCOV
2061
  TAOS_RETURN(code);
×
2062
}
2063

2064
static int32_t mndProcessUpdateXnodeTaskReq(SRpcMsg *pReq) {
476✔
2065
  SMnode             *pMnode = pReq->info.node;
476✔
2066
  int32_t             code = -1;
476✔
2067
  SXnodeTaskObj       *pObj = NULL;
476✔
2068
  SMUpdateXnodeTaskReq updateReq = {0};
476✔
2069

2070
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_UPDATE_XNODE_TASK);
476✔
2071
  if (code != TSDB_CODE_SUCCESS) {
476✔
2072
    mError("failed check permission for update xnode task, code:%s", tstrerror(code));
×
2073
    goto _OVER;
×
2074
  }
2075

2076
  TAOS_CHECK_GOTO(tDeserializeSMUpdateXnodeTaskReq(pReq->pCont, pReq->contLen, &updateReq), NULL, _OVER);
476✔
2077

2078
  if (updateReq.tid > 0) {
476✔
2079
    pObj = mndAcquireXnodeTaskById(pMnode, updateReq.tid);
119✔
2080
  } else {
2081
    pObj = mndAcquireXnodeTaskByName(pMnode, updateReq.name.ptr);
357✔
2082
  }
2083
  if (pObj == NULL) {
476✔
2084
    code = terrno;
476✔
2085
    goto _OVER;
476✔
2086
  }
2087
  code = mndCheckXnodeTaskOwner(pMnode, pReq->info.conn.user, pObj);
×
UNCOV
2088
  if (code != TSDB_CODE_SUCCESS) {
×
2089
    mError("user:%s has no privilege on xnode task:%d", pReq->info.conn.user, pObj->id);
×
UNCOV
2090
    goto _OVER;
×
2091
  }
2092

2093
  if (updateReq.updateName.len > 0) {
×
2094
    SXnodeTaskObj *tmpObj = mndAcquireXnodeTaskByName(pMnode, updateReq.updateName.ptr);
×
2095
    if (tmpObj != NULL) {
×
2096
      mndReleaseXnodeTask(pMnode, tmpObj);
×
2097
      code = TSDB_CODE_MND_XNODE_NAME_DUPLICATE;
×
UNCOV
2098
      goto _OVER;
×
2099
    }
2100
  }
2101

2102
  code = mndUpdateXnodeTask(pMnode, pReq, pObj, &updateReq);
×
UNCOV
2103
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
×
2104

2105
_OVER:
476✔
2106
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
476✔
2107
    mError("xnode task:%d, failed to update since %s", updateReq.tid, tstrerror(code));
476✔
2108
  }
2109

2110
  mndReleaseXnodeTask(pMnode, pObj);
476✔
2111
  tFreeSMUpdateXnodeTaskReq(&updateReq);
476✔
2112
  TAOS_RETURN(code);
476✔
2113
  return 0;
2114
}
2115

2116
SXnodeTaskObj *mndAcquireXnodeTask(SMnode *pMnode, int32_t tid) {
×
UNCOV
2117
  SXnodeTaskObj *pObj = sdbAcquire(pMnode->pSdb, SDB_XNODE_TASK, &tid);
×
2118
  if (pObj == NULL && terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
×
2119
    terrno = TSDB_CODE_MND_XNODE_TASK_NOT_EXIST;
×
2120
  }
2121
  return pObj;
×
2122
}
2123

2124
static int32_t mndSetDropXnodeTaskRedoLogs(STrans *pTrans, SXnodeTaskObj *pObj) {
×
2125
  int32_t  code = 0;
×
UNCOV
2126
  SSdbRaw *pRedoRaw = mndXnodeTaskActionEncode(pObj);
×
2127
  if (pRedoRaw == NULL) {
×
UNCOV
2128
    code = terrno;
×
2129
    return code;
×
2130
  }
2131

2132
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
×
2133
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
×
2134

2135
  TAOS_RETURN(code);
×
2136
}
2137

UNCOV
2138
static int32_t mndSetDropXnodeTaskCommitLogs(STrans *pTrans, SXnodeTaskObj *pObj) {
×
2139
  int32_t  code = 0;
×
2140
  SSdbRaw *pCommitRaw = mndXnodeTaskActionEncode(pObj);
×
UNCOV
2141
  if (pCommitRaw == NULL) {
×
UNCOV
2142
    code = terrno;
×
UNCOV
2143
    return code;
×
2144
  }
2145

UNCOV
2146
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
×
UNCOV
2147
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
×
UNCOV
2148
  TAOS_RETURN(code);
×
2149
}
UNCOV
2150
static int32_t mndSetDropXnodeTaskInfoToTrans(SMnode *pMnode, STrans *pTrans, SXnodeTaskObj *pObj, bool force) {
×
UNCOV
2151
  if (pObj == NULL) {
×
2152
    return 0;
×
2153
  }
UNCOV
2154
  TAOS_CHECK_RETURN(mndSetDropXnodeTaskRedoLogs(pTrans, pObj));
×
UNCOV
2155
  TAOS_CHECK_RETURN(mndSetDropXnodeTaskCommitLogs(pTrans, pObj));
×
UNCOV
2156
  return 0;
×
2157
}
2158

2159
static int32_t mndDropXnodeTask(SMnode *pMnode, SRpcMsg *pReq, SXnodeTaskObj *pTask) {
×
2160
  int32_t code = 0;
×
UNCOV
2161
  int32_t lino = 0;
×
UNCOV
2162
  SArray *pArray = NULL;
×
2163

UNCOV
2164
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_NOTHING, pReq, "drop-xnode-task");
×
UNCOV
2165
  TSDB_CHECK_NULL(pTrans, code, lino, _OVER, terrno);
×
2166

UNCOV
2167
  mndTransSetSerial(pTrans);
×
UNCOV
2168
  mDebug("trans:%d, to drop xnode:%d", pTrans->id, pTask->id);
×
2169

UNCOV
2170
  code = mndSetDropXnodeTaskInfoToTrans(pMnode, pTrans, pTask, false);
×
UNCOV
2171
  mndReleaseXnodeTask(pMnode, pTask);
×
2172

UNCOV
2173
  TSDB_CHECK_CODE(code, lino, _OVER);
×
2174

2175
  code = mndTransPrepare(pMnode, pTrans);
×
2176

UNCOV
2177
_OVER:
×
2178
  if (pArray != NULL) {
×
2179
    for (int i = 0; i < pArray->size; i++) {
×
2180
      SXnodeJobObj *pJob = taosArrayGet(pArray, i);
×
UNCOV
2181
      if (pJob == NULL) continue;
×
UNCOV
2182
      mndReleaseXnodeJob(pMnode, pJob);
×
2183
    }
2184
  }
UNCOV
2185
  mndTransDrop(pTrans);
×
UNCOV
2186
  TAOS_RETURN(code);
×
2187
}
2188

2189
static int32_t mndProcessDropXnodeTaskReq(SRpcMsg *pReq) {
3,213✔
2190
  SMnode            *pMnode = pReq->info.node;
3,213✔
2191
  int32_t            code = -1;
3,213✔
2192
  SXnodeTaskObj     *pObj = NULL;
3,213✔
2193
  SMDropXnodeTaskReq dropReq = {0};
3,213✔
2194
  SJson             *pJson = NULL;
3,213✔
2195

2196
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_DROP_XNODE_TASK);
3,213✔
2197
  if (code != TSDB_CODE_SUCCESS) {
3,213✔
UNCOV
2198
    mError("failed check permission for drop xnode task, code:%s", tstrerror(code));
×
2199
    goto _OVER;
×
2200
  }
2201
  TAOS_CHECK_GOTO(tDeserializeSMDropXnodeTaskReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
3,213✔
2202
  mDebug("DropXnodeTask with tid:%d, start to drop", dropReq.id);
3,213✔
2203

2204
  if (dropReq.id <= 0 && (dropReq.name.len <= 0 || dropReq.name.ptr == NULL)) {
3,213✔
UNCOV
2205
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
2206
    goto _OVER;
×
2207
  }
2208

2209
  if (dropReq.name.len > 0 && dropReq.name.ptr != NULL) {
3,213✔
2210
    pObj = mndAcquireXnodeTaskByName(pMnode, dropReq.name.ptr);
3,213✔
2211
  } else {
2212
    pObj = mndAcquireXnodeTask(pMnode, dropReq.id);
×
2213
  }
2214
  if (pObj == NULL) {
3,213✔
2215
    code = terrno;
3,213✔
2216
    goto _OVER;
3,213✔
2217
  }
2218
  code = mndCheckXnodeTaskOwner(pMnode, pReq->info.conn.user, pObj);
×
2219
  if (code != TSDB_CODE_SUCCESS) {
×
2220
    mError("user:%s has no privilege on xnode task:%d", pReq->info.conn.user, pObj->id);
×
UNCOV
2221
    goto _OVER;
×
2222
  }
2223

2224
  // send request to drop xnode task
2225
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
×
2226
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/task/drop/%d", XNODED_PIPE_SOCKET_URL, pObj->id);
×
2227
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_DELETE, defaultTimeout, NULL, 0);
×
2228

UNCOV
2229
  code = mndDropXnodeTask(pMnode, pReq, pObj);
×
2230
  if (code == 0) {
×
2231
    code = TSDB_CODE_ACTION_IN_PROGRESS;
×
2232
  }
2233

2234
_OVER:
3,213✔
2235
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
3,213✔
2236
    mError("xnode task:%d, failed to drop since %s", dropReq.id, tstrerror(code));
3,213✔
2237
  }
2238
  if (pJson != NULL) {
3,213✔
2239
    tjsonDelete(pJson);
×
2240
  }
2241
  tFreeSMDropXnodeTaskReq(&dropReq);
3,213✔
2242
  TAOS_RETURN(code);
3,213✔
2243
}
2244
static int32_t mndRetrieveXnodeTasks(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
6,664✔
2245
  SMnode        *pMnode = pReq->info.node;
6,664✔
2246
  SSdb          *pSdb = pMnode->pSdb;
6,664✔
2247
  int32_t        numOfRows = 0;
6,664✔
2248
  int32_t        cols = 0;
6,664✔
2249
  SXnodeTaskObj *pObj = NULL;
6,664✔
2250
  char           buf[VARSTR_HEADER_SIZE +
×
2251
           TMAX(TSDB_XNODE_TASK_NAME_LEN,
6,664✔
2252
                          TMAX(TSDB_XNODE_TASK_SOURCE_LEN, TMAX(TSDB_XNODE_TASK_SINK_LEN, TSDB_XNODE_TASK_PARSER_LEN)))];
2253
  int32_t        code = 0;
6,664✔
2254
  SUserObj      *pOperUser = NULL;
6,664✔
2255
  mDebug("show.type:%d, %s:%d: retrieve xnode tasks with rows: %d", pShow->type, __FILE__, __LINE__, rows);
6,664✔
2256

2257
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pOperUser), NULL, _end);
6,664✔
2258

2259
  while (numOfRows < rows) {
6,664✔
2260
    pShow->pIter = sdbFetch(pSdb, SDB_XNODE_TASK, pShow->pIter, (void **)&pObj);
6,664✔
2261
    if (pShow->pIter == NULL) break;
6,664✔
2262

UNCOV
2263
    if (!pOperUser->superUser) {
×
2264
      if (pObj->createdBy == NULL || pObj->createdByLen <= 0 ||
×
2265
          strncmp(pObj->createdBy, pReq->info.conn.user, TSDB_USER_LEN) != 0) {
×
UNCOV
2266
        sdbRelease(pSdb, pObj);
×
UNCOV
2267
        continue;
×
2268
      }
2269
    }
2270

2271
    cols = 0;
×
2272
    // id
2273
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2274
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->id, false);
×
UNCOV
2275
    if (code != 0) goto _end;
×
2276

2277
    // name
UNCOV
2278
    buf[0] = 0;
×
UNCOV
2279
    STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->name, pShow->pMeta->pSchemas[cols].bytes);
×
UNCOV
2280
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2281
    code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
2282
    if (code != 0) goto _end;
×
2283

2284
    // from
2285
    buf[0] = 0;
×
2286
    STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->sourceDsn, pShow->pMeta->pSchemas[cols].bytes);
×
UNCOV
2287
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2288
    code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
2289
    if (code != 0) goto _end;
×
2290

2291
    // to
UNCOV
2292
    buf[0] = 0;
×
2293
    STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->sinkDsn, pShow->pMeta->pSchemas[cols].bytes);
×
2294
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2295
    code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
2296
    if (code != 0) goto _end;
×
2297

2298
    // parser
UNCOV
2299
    if (pObj->parserLen > 0 && pObj->parser != NULL) {
×
2300
      buf[0] = 0;
×
2301
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->parser, pShow->pMeta->pSchemas[cols].bytes);
×
UNCOV
2302
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
2303
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
UNCOV
2304
      if (code != 0) goto _end;
×
2305
    } else {
2306
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2307
      colDataSetNULL(pColInfo, numOfRows);
×
2308
    }
2309

2310
    // via
UNCOV
2311
    if (pObj->via != 0) {
×
2312
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2313
      code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->via, false);
×
UNCOV
2314
      if (code != 0) goto _end;
×
2315
    } else {
UNCOV
2316
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2317
      colDataSetNULL(pColInfo, numOfRows);
×
2318
    }
2319

2320
    // xnode_id
UNCOV
2321
    if (pObj->xnodeId != 0) {
×
2322
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2323
      code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->xnodeId, false);
×
2324
      if (code != 0) goto _end;
×
2325
    } else {
2326
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2327
      colDataSetNULL(pColInfo, numOfRows);
×
2328
    }
2329

2330
    // status
UNCOV
2331
    if (pObj->statusLen > 0) {
×
UNCOV
2332
      buf[0] = 0;
×
UNCOV
2333
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->status, pShow->pMeta->pSchemas[cols].bytes);
×
UNCOV
2334
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
2335
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
UNCOV
2336
      if (code != 0) goto _end;
×
2337
    } else {
2338
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2339
      colDataSetNULL(pColInfo, numOfRows);
×
2340
    }
2341

2342
    // reason
UNCOV
2343
    if (pObj->reasonLen > 0) {
×
2344
      buf[0] = 0;
×
2345
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->reason, pShow->pMeta->pSchemas[cols].bytes);
×
2346
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
2347
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
2348
      if (code != 0) goto _end;
×
2349
    } else {
2350
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2351
      colDataSetNULL(pColInfo, numOfRows);
×
2352
    }
2353

2354
    // create_by
2355
    if (pObj->createdByLen > 0) {
×
2356
      buf[0] = 0;
×
2357
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->createdBy, pShow->pMeta->pSchemas[cols].bytes);
×
2358
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2359
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
2360
      if (code != 0) goto _end;
×
2361
    } else {
2362
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2363
      colDataSetNULL(pColInfo, numOfRows);
×
2364
    }
2365

2366
    // labels
UNCOV
2367
    if (pObj->labelsLen > 0) {
×
UNCOV
2368
      buf[0] = 0;
×
UNCOV
2369
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->labels, pShow->pMeta->pSchemas[cols].bytes);
×
2370
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
2371
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
2372
      if (code != 0) goto _end;
×
2373
    } else {
2374
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2375
      colDataSetNULL(pColInfo, numOfRows);
×
2376
    }
2377

2378
    // create_time
UNCOV
2379
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
2380
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->createTime, false);
×
UNCOV
2381
    if (code != 0) goto _end;
×
2382

2383
    // update_time
2384
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
2385
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->updateTime, false);
×
UNCOV
2386
    if (code != 0) goto _end;
×
2387

UNCOV
2388
    numOfRows++;
×
UNCOV
2389
    sdbRelease(pSdb, pObj);
×
2390
  }
2391

2392
_end:
6,664✔
2393
  if (code != 0 && pObj != NULL) sdbRelease(pSdb, pObj);
6,664✔
2394
  mndReleaseUser(pMnode, pOperUser);
6,664✔
2395

2396
  pShow->numOfRows += numOfRows;
6,664✔
2397
  return numOfRows;
6,664✔
2398
}
2399

UNCOV
2400
static void mndCancelGetNextXnodeTask(SMnode *pMnode, void *pIter) {
×
UNCOV
2401
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
2402
  sdbCancelFetchByType(pSdb, pIter, SDB_XNODE_TASK);
×
UNCOV
2403
}
×
2404

2405
/** xnode job section **/
2406

UNCOV
2407
static int32_t mndAcquireXnodeJobsByTaskId(SMnode *pMnode, int32_t tid, SArray **ppArray) {
×
2408
  int32_t code = 0;
×
UNCOV
2409
  SSdb   *pSdb = pMnode->pSdb;
×
2410

UNCOV
2411
  *ppArray = taosArrayInit(16, sizeof(SXnodeJobObj));
×
UNCOV
2412
  if (ppArray == NULL) {
×
UNCOV
2413
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2414
    if (terrno != 0) code = terrno;
×
UNCOV
2415
    goto _exit;
×
2416
  }
2417

UNCOV
2418
  int32_t idx = 0;
×
UNCOV
2419
  void   *pIter = NULL;
×
UNCOV
2420
  while (1) {
×
UNCOV
2421
    SXnodeJobObj *pJob = NULL;
×
UNCOV
2422
    pIter = sdbFetch(pSdb, SDB_XNODE_JOB, pIter, (void **)&pJob);
×
UNCOV
2423
    if (pIter == NULL) break;
×
2424

UNCOV
2425
    if (pJob->taskId == tid) {
×
UNCOV
2426
      if (NULL == taosArrayInsert(*ppArray, idx++, pJob)) {
×
UNCOV
2427
        code = terrno;
×
UNCOV
2428
        sdbRelease(pSdb, pJob);
×
UNCOV
2429
        goto _exit;
×
2430
      }
2431
    }
2432

UNCOV
2433
    sdbRelease(pSdb, pJob);
×
2434
  }
UNCOV
2435
  sdbCancelFetch(pSdb, pIter);
×
2436

UNCOV
2437
_exit:
×
UNCOV
2438
  TAOS_RETURN(code);
×
2439
}
2440

2441
static int32_t mndAcquireXnodeJobsAll(SMnode *pMnode, SArray **ppArray) {
20,230✔
2442
  int32_t code = 0;
20,230✔
2443
  SSdb   *pSdb = pMnode->pSdb;
20,230✔
2444

2445
  *ppArray = taosArrayInit(64, sizeof(SXnodeJobObj));
20,230✔
2446
  if (ppArray == NULL) {
20,230✔
UNCOV
2447
    code = terrno;
×
UNCOV
2448
    goto _exit;
×
2449
  }
2450

2451
  int32_t idx = 0;
20,230✔
2452
  void   *pIter = NULL;
20,230✔
UNCOV
2453
  while (1) {
×
2454
    SXnodeJobObj *pJob = NULL;
20,230✔
2455
    pIter = sdbFetch(pSdb, SDB_XNODE_JOB, pIter, (void **)&pJob);
20,230✔
2456
    if (pIter == NULL) break;
20,230✔
UNCOV
2457
    if (NULL == taosArrayInsert(*ppArray, idx++, pJob)) {
×
UNCOV
2458
      code = terrno;
×
2459
      goto _exit;
×
2460
    }
2461
    sdbRelease(pSdb, pJob);
×
2462
  }
2463
  sdbCancelFetch(pSdb, pIter);
20,230✔
2464

2465
_exit:
20,230✔
2466
  TAOS_RETURN(code);
20,230✔
2467
}
2468

UNCOV
2469
static void mndFreeXnodeJob(SXnodeJobObj *pObj) {
×
UNCOV
2470
  if (NULL == pObj) {
×
UNCOV
2471
    return;
×
2472
  }
UNCOV
2473
  if (NULL != pObj->config) {
×
UNCOV
2474
    taosMemoryFreeClear(pObj->config);
×
2475
  }
UNCOV
2476
  if (NULL != pObj->reason) {
×
UNCOV
2477
    taosMemoryFreeClear(pObj->reason);
×
2478
  }
UNCOV
2479
  if (NULL != pObj->status) {
×
UNCOV
2480
    taosMemoryFreeClear(pObj->status);
×
2481
  }
2482
}
2483

2484
SSdbRaw *mndXnodeJobActionEncode(SXnodeJobObj *pObj) {
582✔
2485
  int32_t code = 0;
582✔
2486
  int32_t lino = 0;
582✔
2487
  terrno = TSDB_CODE_OUT_OF_MEMORY;
582✔
2488

2489
  if (NULL == pObj) {
582✔
2490
    terrno = TSDB_CODE_INVALID_PARA;
194✔
2491
    return NULL;
194✔
2492
  }
2493

2494
  mDebug("xnode tid:%d, jid:%d, start to encode to raw, row:%p", pObj->taskId, pObj->id, pObj);
388✔
2495

2496
  int32_t rawDataLen =
388✔
2497
      sizeof(SXnodeJobObj) + TSDB_XNODE_RESERVE_SIZE + pObj->configLen + pObj->reasonLen + pObj->statusLen;
388✔
2498

2499
  SSdbRaw *pRaw = sdbAllocRaw(SDB_XNODE_JOB, TSDB_XNODE_VER_NUMBER, rawDataLen);
388✔
2500
  if (pRaw == NULL) goto _OVER;
388✔
2501

2502
  int32_t dataPos = 0;
388✔
2503
  SDB_SET_INT32(pRaw, dataPos, pObj->id, _OVER)
388✔
2504
  SDB_SET_INT32(pRaw, dataPos, pObj->taskId, _OVER)
388✔
2505
  SDB_SET_INT32(pRaw, dataPos, pObj->configLen, _OVER)
388✔
2506
  SDB_SET_BINARY(pRaw, dataPos, pObj->config, pObj->configLen, _OVER)
388✔
2507
  SDB_SET_INT32(pRaw, dataPos, pObj->via, _OVER)
388✔
2508
  SDB_SET_INT32(pRaw, dataPos, pObj->xnodeId, _OVER)
388✔
2509
  SDB_SET_INT32(pRaw, dataPos, pObj->statusLen, _OVER)
388✔
2510
  SDB_SET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
388✔
2511
  SDB_SET_INT32(pRaw, dataPos, pObj->reasonLen, _OVER)
388✔
2512
  SDB_SET_BINARY(pRaw, dataPos, pObj->reason, pObj->reasonLen, _OVER)
388✔
2513
  SDB_SET_INT64(pRaw, dataPos, pObj->createTime, _OVER)
388✔
2514
  SDB_SET_INT64(pRaw, dataPos, pObj->updateTime, _OVER)
388✔
2515

2516
  SDB_SET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
388✔
2517

2518
  terrno = 0;
388✔
2519

2520
_OVER:
388✔
2521
  if (terrno != 0) {
388✔
UNCOV
2522
    mError("xnode tid:%d, jid:%d, failed to encode to raw:%p since %s", pObj->taskId, pObj->id, pRaw, terrstr());
×
UNCOV
2523
    sdbFreeRaw(pRaw);
×
UNCOV
2524
    return NULL;
×
2525
  }
2526

2527
  mTrace("xnode tid:%d, jid:%d, encode to raw:%p, row:%p", pObj->taskId, pObj->id, pRaw, pObj);
388✔
2528
  return pRaw;
388✔
2529
}
2530

2531
SSdbRow *mndXnodeJobActionDecode(SSdbRaw *pRaw) {
582✔
2532
  mInfo("xnode, start to decode from raw:%p", pRaw);
582✔
2533
  int32_t code = 0;
582✔
2534
  int32_t lino = 0;
582✔
2535
  terrno = TSDB_CODE_OUT_OF_MEMORY;
582✔
2536
  SSdbRow      *pRow = NULL;
582✔
2537
  SXnodeJobObj *pObj = NULL;
582✔
2538

2539
  if (NULL == pRaw) {
582✔
2540
    terrno = TSDB_CODE_INVALID_PARA;
194✔
2541
    return NULL;
194✔
2542
  }
2543

2544
  int8_t sver = 0;
388✔
2545
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
388✔
2546

2547
  if (sver != TSDB_XNODE_VER_NUMBER) {
388✔
UNCOV
2548
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
UNCOV
2549
    goto _OVER;
×
2550
  }
2551

2552
  pRow = sdbAllocRow(sizeof(SXnodeJobObj));
388✔
2553
  if (pRow == NULL) goto _OVER;
388✔
2554

2555
  pObj = sdbGetRowObj(pRow);
388✔
2556
  if (pObj == NULL) goto _OVER;
388✔
2557

2558
  int32_t dataPos = 0;
388✔
2559
  SDB_GET_INT32(pRaw, dataPos, &pObj->id, _OVER)
388✔
2560
  SDB_GET_INT32(pRaw, dataPos, &pObj->taskId, _OVER)
388✔
2561

2562
  SDB_GET_INT32(pRaw, dataPos, &pObj->configLen, _OVER)
388✔
2563
  if (pObj->configLen > 0) {
388✔
2564
    pObj->config = taosMemoryCalloc(pObj->configLen, 1);
388✔
2565
    if (pObj->config == NULL) goto _OVER;
388✔
2566
    SDB_GET_BINARY(pRaw, dataPos, pObj->config, pObj->configLen, _OVER)
388✔
2567
  }
2568

2569
  SDB_GET_INT32(pRaw, dataPos, &pObj->via, _OVER)
388✔
2570
  SDB_GET_INT32(pRaw, dataPos, &pObj->xnodeId, _OVER)
388✔
2571
  SDB_GET_INT32(pRaw, dataPos, &pObj->statusLen, _OVER)
388✔
2572
  if (pObj->statusLen > 0) {
388✔
2573
    pObj->status = taosMemoryCalloc(pObj->statusLen, 1);
388✔
2574
    if (pObj->status == NULL) goto _OVER;
388✔
2575
    SDB_GET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
388✔
2576
  }
2577

2578
  SDB_GET_INT32(pRaw, dataPos, &pObj->reasonLen, _OVER)
388✔
2579
  if (pObj->reasonLen > 0) {
388✔
2580
    pObj->reason = taosMemoryCalloc(pObj->reasonLen, 1);
194✔
2581
    if (pObj->reason == NULL) goto _OVER;
194✔
2582
    SDB_GET_BINARY(pRaw, dataPos, pObj->reason, pObj->reasonLen, _OVER)
194✔
2583
  }
2584

2585
  SDB_GET_INT64(pRaw, dataPos, &pObj->createTime, _OVER)
388✔
2586
  SDB_GET_INT64(pRaw, dataPos, &pObj->updateTime, _OVER)
388✔
2587

2588
  SDB_GET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
388✔
2589

2590
  terrno = 0;
388✔
2591

2592
_OVER:
388✔
2593
  if (terrno != 0) {
388✔
UNCOV
2594
    mError("xnode tid:%d, jid:%d, failed to decode from raw:%p since %s", pObj == NULL ? 0 : pObj->taskId,
×
2595
           pObj == NULL ? 0 : pObj->id, pRaw, terrstr());
UNCOV
2596
    if (pObj != NULL) {
×
UNCOV
2597
      taosMemoryFreeClear(pObj->config);
×
UNCOV
2598
      taosMemoryFreeClear(pObj->reason);
×
UNCOV
2599
      taosMemoryFreeClear(pObj->status);
×
2600
    }
UNCOV
2601
    taosMemoryFreeClear(pRow);
×
UNCOV
2602
    return NULL;
×
2603
  }
2604

2605
  mTrace("xnode:%d, decode from raw:%p, row:%p", pObj->id, pRaw, pObj);
388✔
2606
  return pRow;
388✔
2607
}
2608

2609
int32_t mndXnodeJobActionInsert(SSdb *pSdb, SXnodeJobObj *pObj) {
×
UNCOV
2610
  mInfo("xnode tid:%d, jid:%d, perform insert action, row:%p", pObj->taskId, pObj->id, pObj);
×
UNCOV
2611
  return 0;
×
2612
}
2613

UNCOV
2614
int32_t mndXnodeJobActionDelete(SSdb *pSdb, SXnodeJobObj *pObj) {
×
UNCOV
2615
  mDebug("xnode tid:%d, jid:%d, perform delete action, row:%p", pObj->taskId, pObj->id, pObj);
×
UNCOV
2616
  mndFreeXnodeJob(pObj);
×
UNCOV
2617
  return 0;
×
2618
}
2619

UNCOV
2620
int32_t mndXnodeJobActionUpdate(SSdb *pSdb, SXnodeJobObj *pOld, SXnodeJobObj *pNew) {
×
UNCOV
2621
  mDebug("xnode tid:%d, jid:%d, perform update action, old row:%p new row:%p", pOld->taskId, pOld->id, pOld, pNew);
×
2622

UNCOV
2623
  taosWLockLatch(&pOld->lock);
×
UNCOV
2624
  pOld->via = pNew->via;
×
UNCOV
2625
  pOld->xnodeId = pNew->xnodeId;
×
UNCOV
2626
  swapFields(&pNew->statusLen, &pNew->status, &pOld->statusLen, &pOld->status);
×
UNCOV
2627
  swapFields(&pNew->configLen, &pNew->config, &pOld->configLen, &pOld->config);
×
UNCOV
2628
  swapFields(&pNew->reasonLen, &pNew->reason, &pOld->reasonLen, &pOld->reason);
×
UNCOV
2629
  if (pNew->updateTime > pOld->updateTime) {
×
UNCOV
2630
    pOld->updateTime = pNew->updateTime;
×
2631
  }
2632
  taosWUnLockLatch(&pOld->lock);
×
UNCOV
2633
  return 0;
×
2634
}
2635

2636
/* xnode user pass actions */
2637
SSdbRaw *mndXnodeUserPassActionEncode(SXnodeUserPassObj *pObj) {
2,680✔
2638
  int32_t code = 0;
2,680✔
2639
  int32_t lino = 0;
2,680✔
2640
  terrno = TSDB_CODE_OUT_OF_MEMORY;
2,680✔
2641

2642
  if (NULL == pObj) {
2,680✔
2643
    terrno = TSDB_CODE_INVALID_PARA;
194✔
2644
    return NULL;
194✔
2645
  }
2646

2647
  int32_t rawDataLen =
2,486✔
2648
      sizeof(SXnodeUserPassObj) + TSDB_XNODE_RESERVE_SIZE + pObj->userLen + pObj->passLen + pObj->tokenLen;
2,486✔
2649

2650
  SSdbRaw *pRaw = sdbAllocRaw(SDB_XNODE_USER_PASS, TSDB_XNODE_VER_NUMBER, rawDataLen);
2,486✔
2651
  if (pRaw == NULL) goto _OVER;
2,486✔
2652

2653
  int32_t dataPos = 0;
2,486✔
2654
  SDB_SET_INT32(pRaw, dataPos, pObj->id, _OVER)
2,486✔
2655
  SDB_SET_INT32(pRaw, dataPos, pObj->userLen, _OVER)
2,486✔
2656
  SDB_SET_BINARY(pRaw, dataPos, pObj->user, pObj->userLen, _OVER)
2,486✔
2657
  SDB_SET_INT32(pRaw, dataPos, pObj->passLen, _OVER)
2,486✔
2658
  SDB_SET_BINARY(pRaw, dataPos, pObj->pass, pObj->passLen, _OVER)
2,486✔
2659
  SDB_SET_INT32(pRaw, dataPos, pObj->tokenLen, _OVER)
2,486✔
2660
  SDB_SET_BINARY(pRaw, dataPos, pObj->token, pObj->tokenLen, _OVER)
2,486✔
2661
  SDB_SET_INT64(pRaw, dataPos, pObj->createTime, _OVER)
2,486✔
2662
  SDB_SET_INT64(pRaw, dataPos, pObj->updateTime, _OVER)
2,486✔
2663

2664
  SDB_SET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
2,486✔
2665

2666
  terrno = 0;
2,486✔
2667

2668
_OVER:
2,486✔
2669
  if (terrno != 0) {
2,486✔
2670
    mError("xnode user pass:%d, failed to encode to raw:%p since %s", pObj->id, pRaw, terrstr());
×
2671
    sdbFreeRaw(pRaw);
×
2672
    return NULL;
×
2673
  }
2674

2675
  mTrace("xnode user pass:%d, encode to raw:%p, row:%p", pObj->id, pRaw, pObj);
2,486✔
2676
  return pRaw;
2,486✔
2677
}
2678
SSdbRow *mndXnodeUserPassActionDecode(SSdbRaw *pRaw) {
2,085✔
2679
  int32_t code = 0;
2,085✔
2680
  int32_t lino = 0;
2,085✔
2681
  terrno = TSDB_CODE_OUT_OF_MEMORY;
2,085✔
2682
  SSdbRow           *pRow = NULL;
2,085✔
2683
  SXnodeUserPassObj *pObj = NULL;
2,085✔
2684

2685
  if (NULL == pRaw) {
2,085✔
2686
    terrno = TSDB_CODE_INVALID_PARA;
194✔
2687
    return NULL;
194✔
2688
  }
2689

2690
  int8_t sver = 0;
1,891✔
2691
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
1,891✔
2692

2693
  if (sver != TSDB_XNODE_VER_NUMBER) {
1,891✔
UNCOV
2694
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
UNCOV
2695
    goto _OVER;
×
2696
  }
2697

2698
  pRow = sdbAllocRow(sizeof(SXnodeUserPassObj));
1,891✔
2699
  if (pRow == NULL) goto _OVER;
1,891✔
2700

2701
  pObj = sdbGetRowObj(pRow);
1,891✔
2702
  if (pObj == NULL) goto _OVER;
1,891✔
2703

2704
  int32_t dataPos = 0;
1,891✔
2705
  SDB_GET_INT32(pRaw, dataPos, &pObj->id, _OVER)
1,891✔
2706
  SDB_GET_INT32(pRaw, dataPos, &pObj->userLen, _OVER)
1,891✔
2707
  if (pObj->userLen > 0) {
1,891✔
2708
    pObj->user = taosMemoryCalloc(1, pObj->userLen + 1);
1,221✔
2709
    if (pObj->user == NULL) goto _OVER;
1,221✔
2710
    SDB_GET_BINARY(pRaw, dataPos, pObj->user, pObj->userLen, _OVER)
1,221✔
2711
  }
2712
  SDB_GET_INT32(pRaw, dataPos, &pObj->passLen, _OVER)
1,891✔
2713
  if (pObj->passLen > 0) {
1,891✔
2714
    pObj->pass = taosMemoryCalloc(1, pObj->passLen + 1);
1,221✔
2715
    if (pObj->pass == NULL) goto _OVER;
1,221✔
2716
    SDB_GET_BINARY(pRaw, dataPos, pObj->pass, pObj->passLen, _OVER)
1,221✔
2717
  }
2718
  SDB_GET_INT32(pRaw, dataPos, &pObj->tokenLen, _OVER)
1,891✔
2719
  if (pObj->tokenLen > 0) {
1,891✔
2720
    pObj->token = taosMemoryCalloc(1, pObj->tokenLen + 1);
476✔
2721
    if (pObj->token == NULL) goto _OVER;
476✔
2722
    SDB_GET_BINARY(pRaw, dataPos, pObj->token, pObj->tokenLen, _OVER)
476✔
2723
  }
2724
  SDB_GET_INT64(pRaw, dataPos, &pObj->createTime, _OVER)
1,891✔
2725
  SDB_GET_INT64(pRaw, dataPos, &pObj->updateTime, _OVER)
1,891✔
2726

2727
  SDB_GET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
1,891✔
2728

2729
  terrno = 0;
1,891✔
2730

2731
_OVER:
1,891✔
2732
  if (terrno != 0) {
1,891✔
UNCOV
2733
    mError("xnode user pass:%d, failed to decode from raw:%p since %s", pObj == NULL ? 0 : pObj->id, pRaw, terrstr());
×
2734
    if (pObj != NULL) {
×
2735
      if (pObj->user != NULL) {
×
2736
        taosMemoryFreeClear(pObj->user);
×
2737
      }
UNCOV
2738
      if (pObj->pass != NULL) {
×
UNCOV
2739
        taosMemoryFreeClear(pObj->pass);
×
2740
      }
2741
    }
UNCOV
2742
    taosMemoryFreeClear(pRow);
×
UNCOV
2743
    return NULL;
×
2744
  }
2745

2746
  mTrace("xnode user pass:%d, decode from raw:%p, row:%p", pObj->id, pRaw, pObj);
1,891✔
2747
  return pRow;
1,891✔
2748
}
2749
int32_t mndXnodeUserPassActionInsert(SSdb *pSdb, SXnodeUserPassObj *pObj) {
357✔
2750
  mDebug("xnode user pass:%d, perform insert action, row:%p", pObj->id, pObj);
357✔
2751
  return 0;
357✔
2752
}
2753
int32_t mndXnodeUserPassActionUpdate(SSdb *pSdb, SXnodeUserPassObj *pOld, SXnodeUserPassObj *pNew) {
952✔
2754
  mDebug("xnode user pass:%d, perform update action, old row:%p new row:%p", pOld->id, pOld, pNew);
952✔
2755
  taosWLockLatch(&pOld->lock);
952✔
2756
  if (pNew->updateTime > pOld->updateTime) {
952✔
2757
    pOld->updateTime = pNew->updateTime;
714✔
2758
  }
2759
  char* tmp = NULL;
952✔
2760
  pOld->userLen = pNew->userLen;
952✔
2761
  tmp = pOld->user;
952✔
2762
  pOld->user = pNew->user;
952✔
2763
  pNew->user = tmp;
952✔
2764

2765
  pOld->passLen = pNew->passLen;
952✔
2766
  tmp = pOld->pass;
952✔
2767
  pOld->pass = pNew->pass;
952✔
2768
  pNew->pass = tmp;
952✔
2769

2770
  pOld->tokenLen = pNew->tokenLen;
952✔
2771
  tmp = pOld->token;
952✔
2772
  pOld->token = pNew->token;
952✔
2773
  pNew->token = tmp;
952✔
2774

2775
  // swapFields(&pNew->userLen, &pNew->user, &pOld->userLen, &pOld->user);
2776
  // swapFields(&pNew->passLen, &pNew->pass, &pOld->passLen, &pOld->pass);
2777
  // swapFields(&pNew->tokenLen, &pNew->token, &pOld->tokenLen, &pOld->token);
2778
  // SXnodeUserPassObj* tmp = pNew;
2779
  // pNew = pOld;
2780
  // pOld = tmp;
2781

2782
  taosWUnLockLatch(&pOld->lock);
952✔
2783
  return 0;
952✔
2784
}
2785
int32_t mndXnodeUserPassActionDelete(SSdb *pSdb, SXnodeUserPassObj *pObj) {
1,309✔
2786
  mDebug("xnode:%d, perform delete action, row:%p", pObj->id, pObj);
1,309✔
2787
  taosMemoryFreeClear(pObj->user);
1,309✔
2788
  taosMemoryFreeClear(pObj->pass);
1,309✔
2789
  taosMemoryFreeClear(pObj->token);
1,309✔
2790
  return 0;
1,309✔
2791
}
2792

UNCOV
2793
static int32_t mndSetCreateXnodeJobRedoLogs(STrans *pTrans, SXnodeJobObj *pObj) {
×
UNCOV
2794
  int32_t  code = 0;
×
UNCOV
2795
  SSdbRaw *pRedoRaw = mndXnodeJobActionEncode(pObj);
×
UNCOV
2796
  if (pRedoRaw == NULL) {
×
2797
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
2798
    if (terrno != 0) code = terrno;
×
2799
    TAOS_RETURN(code);
×
2800
  }
UNCOV
2801
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
×
2802
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_CREATING));
×
2803
  TAOS_RETURN(code);
×
2804
}
2805

UNCOV
2806
static int32_t mndSetCreateXnodeJobUndoLogs(STrans *pTrans, SXnodeJobObj *pObj) {
×
UNCOV
2807
  int32_t  code = 0;
×
UNCOV
2808
  SSdbRaw *pUndoRaw = mndXnodeJobActionEncode(pObj);
×
UNCOV
2809
  if (pUndoRaw == NULL) {
×
UNCOV
2810
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2811
    if (terrno != 0) code = terrno;
×
UNCOV
2812
    TAOS_RETURN(code);
×
2813
  }
UNCOV
2814
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pUndoRaw));
×
2815
  TAOS_CHECK_RETURN(sdbSetRawStatus(pUndoRaw, SDB_STATUS_DROPPED));
×
2816
  TAOS_RETURN(code);
×
2817
}
2818

UNCOV
2819
static int32_t mndSetCreateXnodeJobCommitLogs(STrans *pTrans, SXnodeJobObj *pObj) {
×
UNCOV
2820
  int32_t  code = 0;
×
UNCOV
2821
  SSdbRaw *pCommitRaw = mndXnodeJobActionEncode(pObj);
×
UNCOV
2822
  if (pCommitRaw == NULL) {
×
UNCOV
2823
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2824
    if (terrno != 0) code = terrno;
×
UNCOV
2825
    TAOS_RETURN(code);
×
2826
  }
UNCOV
2827
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
×
UNCOV
2828
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY));
×
UNCOV
2829
  TAOS_RETURN(code);
×
2830
}
2831

UNCOV
2832
static int32_t mndCreateXnodeJob(SMnode *pMnode, SRpcMsg *pReq, SMCreateXnodeJobReq *pCreate) {
×
UNCOV
2833
  int32_t code = -1;
×
UNCOV
2834
  STrans *pTrans = NULL;
×
2835

UNCOV
2836
  SXnodeJobObj jobObj = {0};
×
UNCOV
2837
  jobObj.id = sdbGetMaxId(pMnode->pSdb, SDB_XNODE_JOB);
×
UNCOV
2838
  jobObj.taskId = pCreate->tid;
×
2839

UNCOV
2840
  jobObj.configLen = pCreate->config.len + 1;
×
UNCOV
2841
  if (jobObj.configLen > TSDB_XNODE_TASK_JOB_CONFIG_LEN + 1) {
×
UNCOV
2842
    code = TSDB_CODE_MND_XNODE_TASK_JOB_CONFIG_TOO_LONG;
×
UNCOV
2843
    goto _OVER;
×
2844
  }
UNCOV
2845
  jobObj.config = taosMemoryCalloc(1, jobObj.configLen);
×
UNCOV
2846
  if (jobObj.config == NULL) goto _OVER;
×
UNCOV
2847
  (void)memcpy(jobObj.config, pCreate->config.ptr, pCreate->config.len);
×
2848

UNCOV
2849
  jobObj.via = pCreate->via;
×
2850
  jobObj.xnodeId = pCreate->xnodeId;
×
2851

UNCOV
2852
  if (pCreate->status.ptr != NULL) {
×
UNCOV
2853
    jobObj.statusLen = pCreate->status.len + 1;
×
UNCOV
2854
    jobObj.status = taosMemoryCalloc(1, jobObj.statusLen);
×
UNCOV
2855
    if (jobObj.status == NULL) goto _OVER;
×
UNCOV
2856
    (void)memmove(jobObj.status, pCreate->status.ptr, pCreate->status.len);
×
2857
  }
2858

UNCOV
2859
  if (jobObj.reason != NULL) {
×
UNCOV
2860
    jobObj.reasonLen = pCreate->reason.len + 1;
×
UNCOV
2861
    if (jobObj.reasonLen > TSDB_XNODE_TASK_REASON_LEN + 1) {
×
UNCOV
2862
      code = TSDB_CODE_MND_XNODE_TASK_REASON_TOO_LONG;
×
UNCOV
2863
      goto _OVER;
×
2864
    }
UNCOV
2865
    jobObj.reason = taosMemoryCalloc(1, jobObj.reasonLen);
×
UNCOV
2866
    if (jobObj.reason == NULL) goto _OVER;
×
UNCOV
2867
    (void)memcpy(jobObj.reason, pCreate->reason.ptr, pCreate->reason.len);
×
2868
  }
2869

2870
  jobObj.createTime = taosGetTimestampMs();
×
2871
  jobObj.updateTime = jobObj.createTime;
×
2872

2873
  mDebug("create xnode job, id:%d, tid:%d, config:%s, time:%" PRId64, jobObj.id, jobObj.taskId, jobObj.config,
×
2874
         jobObj.createTime);
2875

UNCOV
2876
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-xnode-job");
×
UNCOV
2877
  if (pTrans == NULL) {
×
UNCOV
2878
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
2879
    if (terrno != 0) code = terrno;
×
2880
    mInfo("failed to create transaction for xnode-job:%d, code:0x%x:%s", pCreate->tid, code, tstrerror(code));
×
2881
    goto _OVER;
×
2882
  }
UNCOV
2883
  mndTransSetSerial(pTrans);
×
2884

UNCOV
2885
  mInfo("trans:%d, used to create xnode job on %d as jid:%d", pTrans->id, pCreate->tid, jobObj.id);
×
2886

UNCOV
2887
  TAOS_CHECK_GOTO(mndSetCreateXnodeJobRedoLogs(pTrans, &jobObj), NULL, _OVER);
×
UNCOV
2888
  TAOS_CHECK_GOTO(mndSetCreateXnodeJobUndoLogs(pTrans, &jobObj), NULL, _OVER);
×
UNCOV
2889
  TAOS_CHECK_GOTO(mndSetCreateXnodeJobCommitLogs(pTrans, &jobObj), NULL, _OVER);
×
UNCOV
2890
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
×
2891

UNCOV
2892
  code = 0;
×
2893

UNCOV
2894
_OVER:
×
UNCOV
2895
  mndFreeXnodeJob(&jobObj);
×
UNCOV
2896
  mndTransDrop(pTrans);
×
2897
  TAOS_RETURN(code);
×
2898
}
2899

UNCOV
2900
static int32_t mndUpdateXnodeJob(SMnode *pMnode, SRpcMsg *pReq, SXnodeJobObj *pOld, SMUpdateXnodeJobReq *pUpdate) {
×
UNCOV
2901
  mInfo("xnode job:%d, start to update", pUpdate->jid);
×
UNCOV
2902
  int32_t      code = -1;
×
2903
  STrans      *pTrans = NULL;
×
2904
  SXnodeJobObj jobObj = *pOld;
×
2905
  struct {
2906
    bool status;
2907
    bool config;
2908
    bool reason;
UNCOV
2909
  } isChange = {0};
×
2910

UNCOV
2911
  jobObj.id = pUpdate->jid;
×
UNCOV
2912
  if (pUpdate->via > 0) {
×
UNCOV
2913
    jobObj.via = pUpdate->via;
×
2914
  }
UNCOV
2915
  if (pUpdate->xnodeId > 0) {
×
UNCOV
2916
    jobObj.xnodeId = pUpdate->xnodeId;
×
2917
  }
UNCOV
2918
  if (pUpdate->status.ptr != NULL) {
×
UNCOV
2919
    jobObj.statusLen = pUpdate->status.len + 1;
×
UNCOV
2920
    jobObj.status = taosMemoryCalloc(1, jobObj.statusLen);
×
UNCOV
2921
    if (jobObj.status == NULL) goto _OVER;
×
UNCOV
2922
    (void)memcpy(jobObj.status, pUpdate->status.ptr, pUpdate->status.len);
×
UNCOV
2923
    isChange.status = true;
×
2924
  }
2925
  if (pUpdate->config != NULL) {
×
UNCOV
2926
    jobObj.configLen = pUpdate->configLen + 1;
×
UNCOV
2927
    jobObj.config = taosMemoryCalloc(1, jobObj.configLen);
×
UNCOV
2928
    if (jobObj.config == NULL) goto _OVER;
×
UNCOV
2929
    (void)memcpy(jobObj.config, pUpdate->config, pUpdate->configLen);
×
UNCOV
2930
    isChange.config = true;
×
2931
  }
UNCOV
2932
  if (pUpdate->reason != NULL) {
×
UNCOV
2933
    jobObj.reasonLen = pUpdate->reasonLen + 1;
×
UNCOV
2934
    jobObj.reason = taosMemoryCalloc(1, jobObj.reasonLen);
×
UNCOV
2935
    if (jobObj.reason == NULL) goto _OVER;
×
UNCOV
2936
    (void)memcpy(jobObj.reason, pUpdate->reason, pUpdate->reasonLen);
×
UNCOV
2937
    isChange.reason = true;
×
2938
  }
2939
  jobObj.updateTime = taosGetTimestampMs();
×
2940

UNCOV
2941
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "update-xnode");
×
UNCOV
2942
  if (pTrans == NULL) {
×
UNCOV
2943
    code = terrno;
×
UNCOV
2944
    goto _OVER;
×
2945
  }
UNCOV
2946
  mInfo("trans:%d, used to update xnode job:%d", pTrans->id, jobObj.id);
×
2947

2948
  TAOS_CHECK_GOTO(mndSetCreateXnodeJobCommitLogs(pTrans, &jobObj), NULL, _OVER);
×
UNCOV
2949
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
×
UNCOV
2950
  code = 0;
×
2951

UNCOV
2952
_OVER:
×
UNCOV
2953
  if (NULL != jobObj.status && isChange.status) {
×
UNCOV
2954
    taosMemoryFree(jobObj.status);
×
2955
  }
UNCOV
2956
  if (NULL != jobObj.config && isChange.config) {
×
UNCOV
2957
    taosMemoryFree(jobObj.config);
×
2958
  }
UNCOV
2959
  if (NULL != jobObj.reason && isChange.reason) {
×
UNCOV
2960
    taosMemoryFree(jobObj.reason);
×
2961
  }
UNCOV
2962
  mndTransDrop(pTrans);
×
UNCOV
2963
  TAOS_RETURN(code);
×
2964
}
2965

UNCOV
2966
void mndReleaseXnodeTaskJob(SMnode *pMnode, SXnodeJobObj *pObj) {
×
UNCOV
2967
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
2968
  sdbRelease(pSdb, pObj);
×
UNCOV
2969
}
×
2970

2971
SXnodeJobObj *mndAcquireXnodeJob(SMnode *pMnode, int32_t jid) {
357✔
2972
  SXnodeJobObj *pObj = sdbAcquire(pMnode->pSdb, SDB_XNODE_JOB, &jid);
357✔
2973
  if (pObj == NULL && terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
357✔
2974
    terrno = TSDB_CODE_MND_XNODE_JOB_NOT_EXIST;
357✔
2975
  }
2976
  return pObj;
357✔
2977
}
2978
void mndReleaseXnodeJob(SMnode *pMnode, SXnodeJobObj *pObj) {
357✔
2979
  SSdb *pSdb = pMnode->pSdb;
357✔
2980
  sdbRelease(pSdb, pObj);
357✔
2981
}
357✔
2982

2983
static int32_t mndSetDropXnodeJobRedoLogs(STrans *pTrans, SXnodeJobObj *pObj) {
×
2984
  int32_t  code = 0;
×
UNCOV
2985
  SSdbRaw *pRedoRaw = mndXnodeJobActionEncode(pObj);
×
UNCOV
2986
  if (pRedoRaw == NULL) {
×
UNCOV
2987
    code = terrno;
×
UNCOV
2988
    return code;
×
2989
  }
2990

UNCOV
2991
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
×
UNCOV
2992
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
×
2993

2994
  return code;
×
2995
}
2996

UNCOV
2997
static int32_t mndSetDropXnodeJobCommitLogs(STrans *pTrans, SXnodeJobObj *pObj) {
×
UNCOV
2998
  int32_t  code = 0;
×
UNCOV
2999
  SSdbRaw *pCommitRaw = mndXnodeJobActionEncode(pObj);
×
UNCOV
3000
  if (pCommitRaw == NULL) {
×
UNCOV
3001
    code = terrno;
×
UNCOV
3002
    return code;
×
3003
  }
3004

UNCOV
3005
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
×
UNCOV
3006
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
×
UNCOV
3007
  TAOS_RETURN(code);
×
3008
}
3009
static int32_t mndSetDropXnodeJobInfoToTrans(STrans *pTrans, SXnodeJobObj *pObj, bool force) {
×
3010
  if (pObj == NULL) {
×
UNCOV
3011
    return 0;
×
3012
  }
UNCOV
3013
  TAOS_CHECK_RETURN(mndSetDropXnodeJobRedoLogs(pTrans, pObj));
×
UNCOV
3014
  TAOS_CHECK_RETURN(mndSetDropXnodeJobCommitLogs(pTrans, pObj));
×
UNCOV
3015
  return 0;
×
3016
}
3017

UNCOV
3018
static int32_t mndDropXnodeJob(SMnode *pMnode, SRpcMsg *pReq, SXnodeJobObj *pObj) {
×
UNCOV
3019
  int32_t code = 0;
×
UNCOV
3020
  int32_t lino = 0;
×
3021

UNCOV
3022
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_NOTHING, pReq, "drop-xnode-job");
×
UNCOV
3023
  TSDB_CHECK_NULL(pTrans, code, lino, _OVER, terrno);
×
3024

3025
  mndTransSetSerial(pTrans);
×
UNCOV
3026
  mInfo("trans:%d, to drop xnode:%d", pTrans->id, pObj->id);
×
3027

UNCOV
3028
  code = mndSetDropXnodeJobInfoToTrans(pTrans, pObj, false);
×
3029

UNCOV
3030
  TSDB_CHECK_CODE(code, lino, _OVER);
×
3031

UNCOV
3032
  code = mndTransPrepare(pMnode, pTrans);
×
3033

UNCOV
3034
_OVER:
×
UNCOV
3035
  mndTransDrop(pTrans);
×
UNCOV
3036
  return code;
×
3037
}
3038
static int32_t mndProcessCreateXnodeJobReq(SRpcMsg *pReq) {
10,353✔
3039
  mDebug("create xnode job req, content len:%d", pReq->contLen);
10,353✔
3040
  SMnode             *pMnode = pReq->info.node;
10,353✔
3041
  int32_t             code = -1;
10,353✔
3042
  SMCreateXnodeJobReq createReq = {0};
10,353✔
3043
  SXnodeTaskObj      *pTaskObj = NULL;
10,353✔
3044

3045
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_CREATE_XNODE_JOB);
10,353✔
3046
  if (code != TSDB_CODE_SUCCESS) {
10,353✔
UNCOV
3047
    mError("failed check permission for create xnode job, code:%s", tstrerror(code));
×
UNCOV
3048
    goto _OVER;
×
3049
  }
3050
  TAOS_CHECK_GOTO(tDeserializeSMCreateXnodeJobReq(pReq->pCont, pReq->contLen, &createReq), NULL, _OVER);
10,353✔
3051
  mDebug("xnode create job on xnode:%d", createReq.xnodeId);
10,353✔
3052

3053
  // task must exists
3054
  pTaskObj = mndAcquireXnodeTaskById(pMnode, createReq.tid);
10,353✔
3055
  if (pTaskObj == NULL) {
10,353✔
3056
    code = TSDB_CODE_MND_XNODE_TASK_NOT_EXIST;
10,353✔
3057
    goto _OVER;
10,353✔
3058
  }
3059
  // check task owner
UNCOV
3060
  code = mndCheckXnodeTaskOwner(pMnode, pReq->info.conn.user, pTaskObj);
×
UNCOV
3061
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3062
    mError("user:%s has no privilege on xnode task:%d", pReq->info.conn.user, pTaskObj->id);
×
UNCOV
3063
    goto _OVER;
×
3064
  }
3065

UNCOV
3066
  code = mndCreateXnodeJob(pMnode, pReq, &createReq);
×
UNCOV
3067
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
×
3068

3069
_OVER:
10,353✔
3070
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
10,353✔
3071
    mError("xnode task job on task id:%d, failed to create since %s", createReq.tid, tstrerror(code));
10,353✔
3072
  }
3073

3074
  mndReleaseXnodeTask(pMnode, pTaskObj);
10,353✔
3075
  tFreeSMCreateXnodeJobReq(&createReq);
10,353✔
3076
  TAOS_RETURN(code);
10,353✔
3077
}
3078

3079
static int32_t mndProcessUpdateXnodeJobReq(SRpcMsg *pReq) {
119✔
3080
  SMnode             *pMnode = pReq->info.node;
119✔
3081
  int32_t             code = -1;
119✔
3082
  SXnodeJobObj       *pObj = NULL;
119✔
3083
  SMUpdateXnodeJobReq updateReq = {0};
119✔
3084

3085
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_UPDATE_XNODE_JOB);
119✔
3086
  if (code != TSDB_CODE_SUCCESS) {
119✔
UNCOV
3087
    mError("failed check permission for update xnode job, code:%s", tstrerror(code));
×
UNCOV
3088
    goto _OVER;
×
3089
  }
3090
  TAOS_CHECK_GOTO(tDeserializeSMUpdateXnodeJobReq(pReq->pCont, pReq->contLen, &updateReq), NULL, _OVER);
119✔
3091

3092
  pObj = mndAcquireXnodeJob(pMnode, updateReq.jid);
119✔
3093
  if (pObj == NULL) {
119✔
3094
    code = terrno;
119✔
3095
    goto _OVER;
119✔
3096
  }
UNCOV
3097
  code = mndCheckXnodeJobOwner(pMnode, pReq->info.conn.user, pObj);
×
UNCOV
3098
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3099
    mError("user:%s has no privilege on xnode job:%d", pReq->info.conn.user, pObj->id);
×
UNCOV
3100
    goto _OVER;
×
3101
  }
3102

UNCOV
3103
  code = mndUpdateXnodeJob(pMnode, pReq, pObj, &updateReq);
×
UNCOV
3104
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
×
3105

3106
_OVER:
119✔
3107
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
119✔
3108
    mError("xnode task job on jid:%d, failed to update since %s", updateReq.jid, tstrerror(code));
119✔
3109
  }
3110

3111
  mndReleaseXnodeJob(pMnode, pObj);
119✔
3112
  tFreeSMUpdateXnodeJobReq(&updateReq);
119✔
3113
  TAOS_RETURN(code);
119✔
3114

3115
  return 0;
3116
}
3117

3118
static int32_t mndProcessRebalanceXnodeJobReq(SRpcMsg *pReq) {
238✔
3119
  SMnode                *pMnode = pReq->info.node;
238✔
3120
  int32_t                code = -1;
238✔
3121
  SXnodeJobObj          *pObj = NULL;
238✔
3122
  SMRebalanceXnodeJobReq rebalanceReq = {0};
238✔
3123
  SJson                 *pJson = NULL;
238✔
3124

3125
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_REBALANCE_XNODE_JOB);
238✔
3126
  if (code != TSDB_CODE_SUCCESS) {
238✔
UNCOV
3127
    mError("failed check permission for rebalance xnode job, code:%s", tstrerror(code));
×
UNCOV
3128
    goto _OVER;
×
3129
  }
3130
  TAOS_CHECK_GOTO(tDeserializeSMRebalanceXnodeJobReq(pReq->pCont, pReq->contLen, &rebalanceReq), NULL, _OVER);
238✔
3131
  mDebug("RebalanceXnodeJob with jid:%d, xnode_id:%d, start to rebalance", rebalanceReq.jid, rebalanceReq.xnodeId);
238✔
3132

3133
  if (rebalanceReq.jid <= 0) {
238✔
UNCOV
3134
    code = TSDB_CODE_INVALID_MSG;
×
UNCOV
3135
    goto _OVER;
×
3136
  }
3137

3138
  pObj = mndAcquireXnodeJob(pMnode, rebalanceReq.jid);
238✔
3139
  if (pObj == NULL) {
238✔
3140
    code = terrno;
238✔
3141
    goto _OVER;
238✔
3142
  }
UNCOV
3143
  code = mndCheckXnodeJobOwner(pMnode, pReq->info.conn.user, pObj);
×
3144
  if (code != TSDB_CODE_SUCCESS) {
×
3145
    mError("user:%s has no privilege on xnode job:%d", pReq->info.conn.user, pObj->id);
×
UNCOV
3146
    goto _OVER;
×
3147
  }
3148

3149
  // send request
UNCOV
3150
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
×
UNCOV
3151
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/rebalance/manual/%d/%d/%d", XNODED_PIPE_SOCKET_URL, pObj->taskId, pObj->id,
×
3152
           rebalanceReq.xnodeId);
UNCOV
3153
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, defaultTimeout, NULL, 0);
×
3154

3155
_OVER:
238✔
3156
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
238✔
3157
    mError("xnode:%d, failed to rebalance xnode job since %s", rebalanceReq.jid, tstrerror(code));
238✔
3158
  }
3159
  if (pJson != NULL) {
238✔
UNCOV
3160
    tjsonDelete(pJson);
×
3161
  }
3162
  mndReleaseXnodeJob(pMnode, pObj);
238✔
3163
  tFreeSMRebalanceXnodeJobReq(&rebalanceReq);
238✔
3164
  TAOS_RETURN(code);
238✔
3165
}
3166

3167
typedef struct {
3168
  SValueNode nd;
3169
  bool       shouldFree;
3170
} SXndRefValueNode;
3171

UNCOV
3172
static void freeSXndRefValueNode(void *pNode) {
×
UNCOV
3173
  if (pNode == NULL) return;
×
3174

UNCOV
3175
  SXndRefValueNode *pRefNode = (SXndRefValueNode *)pNode;
×
UNCOV
3176
  if (pRefNode->shouldFree) {
×
UNCOV
3177
    taosMemoryFreeClear(pRefNode->nd.datum.p);
×
3178
  }
3179
}
3180

3181
typedef struct {
3182
  SArray   *stack;
3183
  SHashObj *pMap;
3184
  int32_t   code;
3185
} SXndWhereContext;
3186

UNCOV
3187
void freeSXndWhereContext(SXndWhereContext *pCtx) {
×
3188
  if (pCtx == NULL) return;
×
3189

UNCOV
3190
  if (pCtx->pMap != NULL) {
×
UNCOV
3191
    taosHashCleanup(pCtx->pMap);
×
UNCOV
3192
    pCtx->pMap = NULL;
×
3193
  }
UNCOV
3194
  if (pCtx->stack != NULL) {
×
UNCOV
3195
    for (int32_t i = 0; i < pCtx->stack->size; i++) {
×
UNCOV
3196
      SXndRefValueNode *pRefNode = (SXndRefValueNode *)taosArrayGet(pCtx->stack, i);
×
UNCOV
3197
      if (pRefNode != NULL) {
×
UNCOV
3198
        freeSXndRefValueNode(pRefNode);
×
3199
      }
3200
    }
UNCOV
3201
    taosArrayDestroy(pCtx->stack);
×
UNCOV
3202
    pCtx->stack = NULL;
×
3203
  }
3204
}
3205

3206
static SHashObj *convertJob2Map(const SXnodeJobObj *pJob) {
×
3207
  SHashObj *pMap = taosHashInit(256, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
UNCOV
3208
  if (pMap == NULL) {
×
UNCOV
3209
    return NULL;
×
3210
  }
UNCOV
3211
  taosHashSetFreeFp(pMap, freeSXndRefValueNode);
×
3212
  // id
3213
  SXndRefValueNode id = {0};
×
3214
  id.nd.node.type = QUERY_NODE_VALUE;
×
3215
  id.nd.node.resType.type = TSDB_DATA_TYPE_UBIGINT;
×
3216
  id.nd.datum.u = pJob->id;
×
UNCOV
3217
  int32_t code = taosHashPut(pMap, "id", strlen("id") + 1, &id, sizeof(SXndRefValueNode));
×
UNCOV
3218
  if (code != 0) {
×
UNCOV
3219
    taosHashCleanup(pMap);
×
UNCOV
3220
    return NULL;
×
3221
  }
3222
  // task id
UNCOV
3223
  SXndRefValueNode taskId = {0};
×
3224
  taskId.nd.node.type = QUERY_NODE_VALUE;
×
3225
  taskId.nd.node.resType.type = TSDB_DATA_TYPE_UBIGINT;
×
UNCOV
3226
  taskId.nd.datum.u = pJob->taskId;
×
UNCOV
3227
  if (pJob->taskId == 0) {
×
UNCOV
3228
    taskId.nd.isNull = true;
×
3229
  }
UNCOV
3230
  code = taosHashPut(pMap, "task_id", strlen("task_id") + 1, &taskId, sizeof(SXndRefValueNode));
×
UNCOV
3231
  if (code != 0) {
×
UNCOV
3232
    taosHashCleanup(pMap);
×
UNCOV
3233
    return NULL;
×
3234
  }
3235
  // via
UNCOV
3236
  SXndRefValueNode via = {0};
×
3237
  via.nd.node.type = QUERY_NODE_VALUE;
×
3238
  via.nd.node.resType.type = TSDB_DATA_TYPE_UBIGINT;
×
UNCOV
3239
  via.nd.datum.u = pJob->via;
×
UNCOV
3240
  if (pJob->via == 0) {
×
UNCOV
3241
    via.nd.isNull = true;
×
3242
  }
UNCOV
3243
  code = taosHashPut(pMap, "via", strlen("via") + 1, &via, sizeof(SXndRefValueNode));
×
UNCOV
3244
  if (code != 0) {
×
UNCOV
3245
    taosHashCleanup(pMap);
×
UNCOV
3246
    return NULL;
×
3247
  }
3248
  // xnode id
UNCOV
3249
  SXndRefValueNode xnodeId = {0};
×
UNCOV
3250
  xnodeId.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3251
  xnodeId.nd.node.resType.type = TSDB_DATA_TYPE_UBIGINT;
×
UNCOV
3252
  xnodeId.nd.datum.u = pJob->xnodeId;
×
UNCOV
3253
  if (pJob->xnodeId == 0) {
×
UNCOV
3254
    xnodeId.nd.isNull = true;
×
3255
  }
UNCOV
3256
  code = taosHashPut(pMap, "xnode_id", strlen("xnode_id") + 1, &xnodeId, sizeof(SXndRefValueNode));
×
UNCOV
3257
  if (code != 0) {
×
UNCOV
3258
    taosHashCleanup(pMap);
×
UNCOV
3259
    return NULL;
×
3260
  }
3261
  // config
UNCOV
3262
  SXndRefValueNode config = {0};
×
UNCOV
3263
  config.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3264
  if (pJob->configLen > 0) {
×
UNCOV
3265
    config.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3266
    config.nd.datum.p = taosStrndupi(pJob->config, strlen(pJob->config) + 1);
×
UNCOV
3267
    config.shouldFree = true;
×
UNCOV
3268
    code = taosHashPut(pMap, "config", strlen("config") + 1, &config, sizeof(SXndRefValueNode));
×
3269
  } else {
UNCOV
3270
    config.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3271
    config.nd.datum.p = NULL;
×
UNCOV
3272
    config.nd.isNull = true;
×
UNCOV
3273
    code = taosHashPut(pMap, "config", strlen("config") + 1, &config, sizeof(SXndRefValueNode));
×
3274
  }
UNCOV
3275
  if (code != 0) {
×
UNCOV
3276
    taosHashCleanup(pMap);
×
UNCOV
3277
    return NULL;
×
3278
  }
3279
  // status
UNCOV
3280
  SXndRefValueNode status = {0};
×
3281
  status.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3282
  if (pJob->statusLen > 0) {
×
UNCOV
3283
    status.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3284
    status.nd.datum.p = taosStrndupi(pJob->status, strlen(pJob->status) + 1);
×
UNCOV
3285
    status.shouldFree = true;
×
UNCOV
3286
    code = taosHashPut(pMap, "status", strlen("status") + 1, &status, sizeof(SXndRefValueNode));
×
3287
  } else {
UNCOV
3288
    status.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3289
    status.nd.datum.p = NULL;
×
3290
    status.nd.isNull = true;
×
3291
    code = taosHashPut(pMap, "status", strlen("status") + 1, &status, sizeof(SXndRefValueNode));
×
3292
  }
UNCOV
3293
  if (code != 0) {
×
UNCOV
3294
    taosHashCleanup(pMap);
×
UNCOV
3295
    return NULL;
×
3296
  }
3297
  // reason
UNCOV
3298
  SXndRefValueNode reason = {0};
×
UNCOV
3299
  reason.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3300
  if (pJob->reasonLen > 0) {
×
UNCOV
3301
    reason.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3302
    reason.nd.datum.p = taosStrndupi(pJob->reason, strlen(pJob->reason) + 1);
×
UNCOV
3303
    reason.shouldFree = true;
×
3304
    code = taosHashPut(pMap, "reason", strlen("reason") + 1, &reason, sizeof(SXndRefValueNode));
×
3305
  } else {
3306
    reason.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3307
    reason.nd.datum.p = NULL;
×
UNCOV
3308
    reason.nd.isNull = true;
×
UNCOV
3309
    code = taosHashPut(pMap, "reason", strlen("reason") + 1, &reason, sizeof(SXndRefValueNode));
×
3310
  }
UNCOV
3311
  if (code != 0) {
×
UNCOV
3312
    taosHashCleanup(pMap);
×
UNCOV
3313
    return NULL;
×
3314
  }
3315
  // create time
UNCOV
3316
  SXndRefValueNode createTime = {0};
×
UNCOV
3317
  createTime.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3318
  createTime.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3319
  createTime.nd.datum.p = taosMemoryCalloc(1, TD_TIME_STR_LEN);
×
UNCOV
3320
  createTime.nd.datum.p =
×
UNCOV
3321
      formatTimestampLocal(createTime.nd.datum.p, TD_TIME_STR_LEN, pJob->createTime, TSDB_TIME_PRECISION_MILLI);
×
UNCOV
3322
  createTime.shouldFree = true;
×
UNCOV
3323
  code = taosHashPut(pMap, "create_time", strlen("create_time") + 1, &createTime, sizeof(SXndRefValueNode));
×
UNCOV
3324
  if (code != 0) {
×
UNCOV
3325
    taosHashCleanup(pMap);
×
UNCOV
3326
    return NULL;
×
3327
  }
3328
  // update time
UNCOV
3329
  SXndRefValueNode updateTime = {0};
×
UNCOV
3330
  updateTime.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3331
  updateTime.nd.node.resType.type = TSDB_DATA_TYPE_BINARY;
×
UNCOV
3332
  updateTime.nd.datum.p = taosMemoryCalloc(1, TD_TIME_STR_LEN);
×
UNCOV
3333
  updateTime.nd.datum.p =
×
UNCOV
3334
      formatTimestampLocal(updateTime.nd.datum.p, TD_TIME_STR_LEN, pJob->updateTime, TSDB_TIME_PRECISION_MILLI);
×
UNCOV
3335
  updateTime.shouldFree = true;
×
UNCOV
3336
  code = taosHashPut(pMap, "update_time", strlen("update_time") + 1, &updateTime, sizeof(SXndRefValueNode));
×
3337
  return pMap;
×
3338
}
3339

3340
typedef bool (*FOpCmp)(SValueNode *pval1, SValueNode *pval2);
3341

3342
#define XNODE_DEF_OP_FUNC(NAME, OP)                             \
3343
  static bool NAME(SValueNode *pval1, SValueNode *pval2) {      \
3344
    switch (pval1->node.resType.type) {                         \
3345
      case TSDB_DATA_TYPE_BOOL:                                 \
3346
        return pval1->datum.b OP pval2->datum.b;                \
3347
      case TSDB_DATA_TYPE_UBIGINT:                              \
3348
        return pval1->datum.u OP pval2->datum.u;                \
3349
      case TSDB_DATA_TYPE_BIGINT:                               \
3350
        return pval1->datum.i OP pval2->datum.i;                \
3351
      case TSDB_DATA_TYPE_FLOAT:                                \
3352
        return pval1->datum.d OP pval2->datum.d;                \
3353
      case TSDB_DATA_TYPE_BINARY: {                             \
3354
        if (pval1->datum.p == NULL || pval2->datum.p == NULL) { \
3355
          return pval1->datum.p OP pval2->datum.p;              \
3356
        }                                                       \
3357
        return strcmp(pval1->datum.p, pval2->datum.p) OP 0;     \
3358
      }                                                         \
3359
      default:                                                  \
3360
        return false;                                           \
3361
    }                                                           \
3362
  }
3363

3364
XNODE_DEF_OP_FUNC(op_greater_than, >)
×
3365
XNODE_DEF_OP_FUNC(op_greater_equal, >=)
×
3366
XNODE_DEF_OP_FUNC(op_lower_than, <)
×
UNCOV
3367
XNODE_DEF_OP_FUNC(op_lower_equal, <=)
×
UNCOV
3368
XNODE_DEF_OP_FUNC(op_equal, ==)
×
UNCOV
3369
XNODE_DEF_OP_FUNC(op_not_equal, !=)
×
3370

UNCOV
3371
static int32_t call_op_cmp(SXndWhereContext *pctx, FOpCmp opFn) {
×
UNCOV
3372
  int32_t     code = 0;
×
UNCOV
3373
  SValueNode *pval2 = (SValueNode *)taosArrayPop(pctx->stack);
×
UNCOV
3374
  SValueNode *pval1 = (SValueNode *)taosArrayPop(pctx->stack);
×
UNCOV
3375
  bool        ret = false;
×
3376

UNCOV
3377
  if (pval1->node.type != pval2->node.type) {
×
UNCOV
3378
    code = TSDB_CODE_MND_XNODE_WHERE_COL_TYPE_DIFF;
×
UNCOV
3379
    mError("xnode type not same, v1 type: %d, v2 type: %d", pval1->node.type, pval2->node.type);
×
UNCOV
3380
    goto _OVER;
×
3381
  } else {
UNCOV
3382
    mDebug("xnode type v1:%d, is null: %d, UB: %" PRIu64 ", v2 type: %d, is null: %d, UB: %" PRIu64, pval1->node.type,
×
3383
           pval1->isNull, pval1->datum.u, pval2->node.type, pval2->isNull, pval2->datum.u);
3384

UNCOV
3385
    ret = (*opFn)(pval1, pval2);
×
3386
  }
3387
  SXndRefValueNode pval = {0};
×
3388
  pval.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3389
  pval.nd.node.resType.type = TSDB_DATA_TYPE_BOOL;
×
UNCOV
3390
  pval.nd.datum.b = ret;
×
UNCOV
3391
  if (NULL == taosArrayPush(pctx->stack, &pval)) {
×
UNCOV
3392
    mError("xnode evaluate walker array push error: %s", tstrerror(terrno));
×
UNCOV
3393
    code = terrno;
×
3394
    goto _OVER;
×
3395
  }
3396

UNCOV
3397
_OVER:
×
3398
  if (pval1 != NULL) freeSXndRefValueNode((SXndRefValueNode *)pval1);
×
3399
  if (pval2 != NULL) freeSXndRefValueNode((SXndRefValueNode *)pval2);
×
3400
  TAOS_RETURN(code);
×
3401
}
3402

3403
#define XND_WALKER_CHECK_GOTO(CMD, LABEL)    \
3404
  do {                                       \
3405
    pctx->code = (CMD);                      \
3406
    if ((pctx->code != TSDB_CODE_SUCCESS)) { \
3407
      goto LABEL;                            \
3408
    }                                        \
3409
  } while (0);
3410

UNCOV
3411
static EDealRes evaluateWaker(SNode *pNode, void *pWhereCtx) {
×
UNCOV
3412
  int32_t           code = 0;
×
UNCOV
3413
  SXndWhereContext *pctx = (SXndWhereContext *)pWhereCtx;
×
3414

UNCOV
3415
  if (nodeType(pNode) == QUERY_NODE_COLUMN) {
×
UNCOV
3416
    SColumnNode *colNode = (SColumnNode *)pNode;
×
3417
    SXndRefValueNode *pval =
3418
        (SXndRefValueNode *)taosHashGet(pctx->pMap, colNode->colName, strlen(colNode->colName) + 1);
×
3419
    if (pval == NULL) {
×
3420
      mError("xnode evaluateWhereCond hash get error: %s", tstrerror(terrno));
×
UNCOV
3421
      pctx->code = TSDB_CODE_MND_XNODE_WHERE_COL_NOT_EXIST;
×
UNCOV
3422
      return DEAL_RES_END;
×
3423
    }
UNCOV
3424
    if (NULL == taosArrayPush(pctx->stack, pval)) {
×
UNCOV
3425
      mError("xnode evaluate walker array push error: %s", tstrerror(terrno));
×
UNCOV
3426
      pctx->code = TSDB_CODE_FAILED;
×
3427
      return DEAL_RES_END;
×
3428
    }
3429
    return DEAL_RES_CONTINUE;
×
3430
  }
3431
  if (nodeType(pNode) == QUERY_NODE_VALUE) {
×
3432
    SValueNode *pval = (SValueNode *)pNode;
×
3433
    if (pval->node.resType.type == TSDB_DATA_TYPE_UBIGINT) {
×
3434
      pval->datum.u = taosStr2Int64(pval->literal, NULL, 10);
×
3435
    }
UNCOV
3436
    if (pval->node.resType.type == TSDB_DATA_TYPE_BINARY) {
×
UNCOV
3437
      if (pval->datum.p == NULL) {
×
3438
        pval->datum.p = taosStrndupi(pval->literal, strlen(pval->literal) + 1);
×
3439
      }
3440
    }
UNCOV
3441
    if (pval->node.resType.type == TSDB_DATA_TYPE_NULL) {
×
3442
      pval->isNull = true;
×
3443
      pval->datum.p = NULL;
×
3444
    }
3445
    if (pval->node.resType.type == TSDB_DATA_TYPE_BOOL) {
×
3446
      pval->datum.b = pval->literal[0] == 't' || pval->literal[0] == 'T';
×
3447
    }
3448
    SXndRefValueNode refVal = {0};
×
3449
    refVal.nd = *pval;
×
UNCOV
3450
    refVal.shouldFree = false;
×
UNCOV
3451
    if (NULL == taosArrayPush(pctx->stack, &refVal)) {
×
3452
      mError("xnode evaluate walker array push error: %s", tstrerror(terrno));
×
3453
      pctx->code = TSDB_CODE_FAILED;
×
UNCOV
3454
      return DEAL_RES_END;
×
3455
    }
3456
    return DEAL_RES_CONTINUE;
×
3457
  }
3458

UNCOV
3459
  if (nodeType(pNode) == QUERY_NODE_OPERATOR) {
×
UNCOV
3460
    SOperatorNode *opNode = (SOperatorNode *)pNode;
×
3461
    switch (opNode->opType) {
×
UNCOV
3462
      case OP_TYPE_GREATER_THAN: {
×
3463
        XND_WALKER_CHECK_GOTO(call_op_cmp(pctx, op_greater_than), _exit);
×
3464
        break;
×
3465
      }
UNCOV
3466
      case OP_TYPE_GREATER_EQUAL: {
×
UNCOV
3467
        XND_WALKER_CHECK_GOTO(call_op_cmp(pctx, op_greater_equal), _exit);
×
UNCOV
3468
        break;
×
3469
      }
UNCOV
3470
      case OP_TYPE_LOWER_THAN: {
×
UNCOV
3471
        XND_WALKER_CHECK_GOTO(call_op_cmp(pctx, op_lower_than), _exit);
×
UNCOV
3472
        break;
×
3473
      }
3474
      case OP_TYPE_LOWER_EQUAL: {
×
3475
        XND_WALKER_CHECK_GOTO(call_op_cmp(pctx, op_lower_equal), _exit);
×
UNCOV
3476
        break;
×
3477
      }
UNCOV
3478
      case OP_TYPE_EQUAL: {
×
UNCOV
3479
        XND_WALKER_CHECK_GOTO(call_op_cmp(pctx, op_equal), _exit);
×
UNCOV
3480
        break;
×
3481
      }
UNCOV
3482
      case OP_TYPE_NOT_EQUAL: {
×
UNCOV
3483
        XND_WALKER_CHECK_GOTO(call_op_cmp(pctx, op_not_equal), _exit);
×
UNCOV
3484
        break;
×
3485
      }
UNCOV
3486
      default:
×
UNCOV
3487
        pctx->code = TSDB_CODE_MND_XNODE_WHERE_OP_NOT_SUPPORT;
×
3488
        return DEAL_RES_CONTINUE;
×
3489
    }
3490
    return DEAL_RES_CONTINUE;
×
3491
  }
3492

UNCOV
3493
  if (nodeType(pNode) == QUERY_NODE_LOGIC_CONDITION) {
×
UNCOV
3494
    SLogicConditionNode *logicNode = (SLogicConditionNode *)pNode;
×
UNCOV
3495
    SXndRefValueNode     pval = {0};
×
UNCOV
3496
    pval.nd.node.type = QUERY_NODE_VALUE;
×
UNCOV
3497
    pval.nd.node.resType.type = TSDB_DATA_TYPE_BOOL;
×
3498

UNCOV
3499
    switch (logicNode->condType) {
×
UNCOV
3500
      case LOGIC_COND_TYPE_AND: {
×
UNCOV
3501
        SValueNode *pval2 = (SValueNode *)taosArrayPop(pctx->stack);
×
UNCOV
3502
        SValueNode *pval1 = (SValueNode *)taosArrayPop(pctx->stack);
×
3503

UNCOV
3504
        pval.nd.datum.b = pval1->datum.b && pval2->datum.b;
×
3505
        if (NULL == taosArrayPush(pctx->stack, &pval)) {
×
3506
          mError("xnode walker AND array push err: %s", tstrerror(terrno));
×
UNCOV
3507
          pctx->code = TSDB_CODE_FAILED;
×
UNCOV
3508
          return DEAL_RES_END;
×
3509
        }
3510

UNCOV
3511
        freeSXndRefValueNode((SXndRefValueNode *)pval1);
×
UNCOV
3512
        freeSXndRefValueNode((SXndRefValueNode *)pval2);
×
3513
        break;
×
3514
      }
UNCOV
3515
      case LOGIC_COND_TYPE_OR: {
×
UNCOV
3516
        SValueNode *pval2 = (SValueNode *)taosArrayPop(pctx->stack);
×
UNCOV
3517
        SValueNode *pval1 = (SValueNode *)taosArrayPop(pctx->stack);
×
3518

3519
        pval.nd.datum.b = pval1->datum.b || pval2->datum.b;
×
3520
        if (NULL == taosArrayPush(pctx->stack, &pval)) {
×
UNCOV
3521
          mError("xnode walker OR array push err: %s", tstrerror(terrno));
×
UNCOV
3522
          pctx->code = TSDB_CODE_FAILED;
×
UNCOV
3523
          return DEAL_RES_END;
×
3524
        }
3525

UNCOV
3526
        freeSXndRefValueNode((SXndRefValueNode *)pval1);
×
UNCOV
3527
        freeSXndRefValueNode((SXndRefValueNode *)pval2);
×
UNCOV
3528
        break;
×
3529
      }
UNCOV
3530
      case LOGIC_COND_TYPE_NOT: {
×
UNCOV
3531
        SValueNode *pval1 = (SValueNode *)taosArrayPop(pctx->stack);
×
3532

UNCOV
3533
        pval.nd.datum.b = !pval1->datum.b;
×
UNCOV
3534
        if (NULL == taosArrayPush(pctx->stack, &pval)) {
×
UNCOV
3535
          mError("xnode walker NOT array push err: %s", tstrerror(terrno));
×
UNCOV
3536
          pctx->code = TSDB_CODE_FAILED;
×
UNCOV
3537
          return DEAL_RES_END;
×
3538
        }
3539

UNCOV
3540
        freeSXndRefValueNode((SXndRefValueNode *)pval1);
×
UNCOV
3541
        break;
×
3542
      }
UNCOV
3543
      default:
×
UNCOV
3544
        break;
×
3545
    }
UNCOV
3546
    return DEAL_RES_CONTINUE;
×
3547
  }
3548

3549
  pctx->code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
3550

UNCOV
3551
_exit:
×
UNCOV
3552
  return DEAL_RES_END;
×
3553
}
3554

3555
static bool evaluateWhereCond(SNode *pWhere, SHashObj *pDataMap, int32_t *code) {
×
3556
  bool             ret = false;
×
3557
  SXndWhereContext ctx = {0};
×
3558

UNCOV
3559
  ctx.stack = taosArrayInit(64, sizeof(SXndRefValueNode));
×
UNCOV
3560
  if (ctx.stack == NULL) {
×
UNCOV
3561
    mError("xnode evaluateWhereCond error: %s", tstrerror(terrno));
×
UNCOV
3562
    *code = terrno;
×
UNCOV
3563
    goto _exit;
×
3564
  }
UNCOV
3565
  ctx.pMap = pDataMap;
×
3566

3567
  // walkexpr pnode
UNCOV
3568
  nodesWalkExprPostOrder(pWhere, evaluateWaker, &ctx);
×
UNCOV
3569
  if (ctx.code != TSDB_CODE_SUCCESS) {
×
UNCOV
3570
    *code = ctx.code;
×
UNCOV
3571
    mError("xnode walkExpr error: %s", tstrerror(ctx.code));
×
UNCOV
3572
    goto _exit;
×
3573
  }
UNCOV
3574
  SValueNode *pval = taosArrayGetLast(ctx.stack);
×
UNCOV
3575
  if (pval == NULL) {
×
UNCOV
3576
    *code = terrno;
×
UNCOV
3577
    mError("xnode evaluateWhereCond error: %s", tstrerror(terrno));
×
UNCOV
3578
    goto _exit;
×
3579
  }
UNCOV
3580
  mDebug("xnode ctx stack size:%lu, last nd type:%d, bool:%d", ctx.stack->size, pval->node.type, pval->datum.b);
×
UNCOV
3581
  ret = pval->datum.b;
×
3582

UNCOV
3583
_exit:
×
UNCOV
3584
  freeSXndWhereContext(&ctx);
×
UNCOV
3585
  return ret;
×
3586
}
3587

3588
static int32_t filterJobsByWhereCond(SNode *pWhere, SArray *pArray, SArray **ppResult) {
20,230✔
3589
  int32_t code = 0;
20,230✔
3590

3591
  *ppResult = taosArrayInit(64, sizeof(SXnodeJobObj));
20,230✔
3592
  if (*ppResult == NULL) {
20,230✔
3593
    code = terrno;
×
3594
    goto _exit;
×
3595
  }
3596
  for (int32_t i = 0; i < pArray->size; i++) {
20,230✔
UNCOV
3597
    SXnodeJobObj *pJob = taosArrayGet(pArray, i);
×
3598

UNCOV
3599
    SHashObj *pDataMap = NULL;
×
UNCOV
3600
    if ((pDataMap = convertJob2Map(pJob)) == NULL) {
×
UNCOV
3601
      mError("xnode evaluate convertJow2Map error: %s", tstrerror(terrno));
×
UNCOV
3602
      goto _exit;
×
3603
    }
UNCOV
3604
    if (evaluateWhereCond(pWhere, pDataMap, &code)) {
×
3605
      if (NULL == taosArrayPush(*ppResult, pJob)) {
×
UNCOV
3606
        mError("xnode filterJobsByWhereCond array push err: %s", tstrerror(terrno));
×
UNCOV
3607
        code = TSDB_CODE_FAILED;
×
UNCOV
3608
        goto _exit;
×
3609
      }
3610
    }
UNCOV
3611
    if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3612
      goto _exit;
×
3613
    }
3614
  }
3615

3616
_exit:
20,230✔
3617
  TAOS_RETURN(code);
20,230✔
3618
}
3619

3620
#define XND_LOG_END(code, lino)                                                                 \
3621
  do {                                                                                          \
3622
    if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {                    \
3623
      mError("xnode:%s failed at line %d code: %d, since %s", __func__, lino, code, tstrerror(code)); \
3624
    }                                                                                           \
3625
  } while (0)
3626

3627
void httpRebalanceAuto(SArray *pResult) {
714✔
3628
  int32_t code = 0;
714✔
3629
  int32_t lino = 0;
714✔
3630
  SJson *pJsonArr = NULL;
714✔
3631
  char  *pContStr = NULL;
714✔
3632
  // convert pResult to [(tid, jid)*]
3633
  pJsonArr = tjsonCreateArray();
714✔
3634
  if (pJsonArr == NULL) {
714✔
UNCOV
3635
    code = terrno;
×
UNCOV
3636
    mError("xnode json array error: %s", tstrerror(code));
×
UNCOV
3637
    goto _OVER;
×
3638
  }
3639
  for (int32_t i = 0; i < pResult->size; i++) {
714✔
UNCOV
3640
    SXnodeJobObj *pJob = taosArrayGet(pResult, i);
×
UNCOV
3641
    SJson        *pJsonObj = tjsonCreateObject();
×
UNCOV
3642
    if (pJsonObj == NULL) {
×
UNCOV
3643
      code = terrno;
×
UNCOV
3644
      mError("xnode json object error: %s", tstrerror(code));
×
UNCOV
3645
      goto _OVER;
×
3646
    }
UNCOV
3647
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJsonObj, "tid", pJob->taskId), &lino, _OVER);
×
UNCOV
3648
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJsonObj, "jid", pJob->id), &lino, _OVER);
×
UNCOV
3649
    TAOS_CHECK_GOTO(tjsonAddItemToArray(pJsonArr, pJsonObj), &lino, _OVER);
×
3650
  }
3651

3652
  pContStr = tjsonToUnformattedString(pJsonArr);
714✔
3653
  if (pContStr == NULL) {
714✔
UNCOV
3654
    mError("xnode to json string error: %s", tstrerror(terrno));
×
UNCOV
3655
    goto _OVER;
×
3656
  }
3657
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
714✔
3658
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/rebalance/auto", XNODED_PIPE_SOCKET_URL);
714✔
3659
  SJson* pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, defaultTimeout, pContStr, strlen(pContStr));
714✔
3660
  if (pJson) {
714✔
UNCOV
3661
    tjsonDelete(pJson);
×
3662
  }
3663

3664
_OVER:
714✔
3665
  if (pJsonArr != NULL) tjsonDelete(pJsonArr);
714✔
3666
  if (pContStr != NULL) taosMemoryFree(pContStr);
714✔
3667
  XND_LOG_END(code, lino);
714✔
3668
  return;
714✔
3669
}
3670

3671
static int32_t mndProcessRebalanceXnodeJobsWhereReq(SRpcMsg *pReq) {
714✔
3672
  int32_t                      code = 0;
714✔
3673
  SMnode                      *pMnode = pReq->info.node;
714✔
3674
  SMRebalanceXnodeJobsWhereReq rebalanceReq = {0};
714✔
3675
  SNode                       *pWhere = NULL;
714✔
3676
  SArray                      *pArray = NULL;
714✔
3677
  SArray                      *pResult = NULL;
714✔
3678

3679
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_REBALANCE_XNODE_JOB);
714✔
3680
  if (code != TSDB_CODE_SUCCESS) {
714✔
UNCOV
3681
    mError("failed check permission for rebalance xnode jobs where, code:%s", tstrerror(code));
×
3682
    goto _OVER;
×
3683
  }
3684
  TAOS_CHECK_GOTO(tDeserializeSMRebalanceXnodeJobsWhereReq(pReq->pCont, pReq->contLen, &rebalanceReq), NULL, _OVER);
714✔
3685

3686
  TAOS_CHECK_GOTO(mndAcquireXnodeJobsAll(pMnode, &pArray), NULL, _OVER);
714✔
3687
  for (int32_t i = 0; i < pArray->size; i++) {
714✔
UNCOV
3688
    SXnodeJobObj *pJob = taosArrayGet(pArray, i);
×
3689
    code = mndCheckXnodeJobOwner(pMnode, pReq->info.conn.user, pJob);
×
3690
    if (code == TSDB_CODE_SUCCESS) {
×
UNCOV
3691
      continue;
×
UNCOV
3692
    } else if (code == TSDB_CODE_MND_XNODE_TASK_NO_PRIV) {
×
UNCOV
3693
      taosArrayRemove(pArray, i);
×
UNCOV
3694
      i--;
×
3695
    } else {
UNCOV
3696
      mError("user:%s rebalance where xnode job:%d, error:%s", pReq->info.conn.user, pJob->id, tstrerror(code));
×
3697
      goto _OVER;
×
3698
    }
3699
  }
3700
  if (NULL != rebalanceReq.ast.ptr) {
714✔
3701
    TAOS_CHECK_GOTO(nodesStringToNode(rebalanceReq.ast.ptr, &pWhere), NULL, _OVER);
714✔
3702
    TAOS_CHECK_GOTO(filterJobsByWhereCond(pWhere, pArray, &pResult), NULL, _OVER);
714✔
3703
    httpRebalanceAuto(pResult);
714✔
3704
  } else {
3705
    httpRebalanceAuto(pArray);
×
3706
  }
3707

3708
_OVER:
714✔
3709
  if (pWhere != NULL) {
714✔
3710
    nodesDestroyNode(pWhere);
714✔
3711
  }
3712
  if (pArray != NULL) {
714✔
3713
    taosArrayDestroy(pArray);
714✔
3714
  }
3715
  if (pResult != NULL) {
714✔
3716
    taosArrayDestroy(pResult);
714✔
3717
  }
3718
  tFreeSMRebalanceXnodeJobsWhereReq(&rebalanceReq);
714✔
3719
  TAOS_RETURN(code);
714✔
3720
}
3721

UNCOV
3722
static int32_t dropXnodeJobById(SMnode *pMnode, SRpcMsg *pReq, int32_t jid) {
×
UNCOV
3723
  int32_t       code = 0;
×
UNCOV
3724
  int32_t       lino = 0;
×
UNCOV
3725
  SXnodeJobObj *pObj = NULL;
×
3726

UNCOV
3727
  pObj = mndAcquireXnodeJob(pMnode, jid);
×
UNCOV
3728
  if (pObj == NULL) {
×
UNCOV
3729
    code = terrno;
×
UNCOV
3730
    lino = __LINE__;
×
UNCOV
3731
    goto _OVER;
×
3732
  }
UNCOV
3733
  code = mndCheckXnodeJobOwner(pMnode, pReq->info.conn.user, pObj);
×
UNCOV
3734
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3735
    mError("user:%s has no privilege on xnode job:%d", pReq->info.conn.user, pObj->id);
×
UNCOV
3736
    goto _OVER;
×
3737
  }
UNCOV
3738
  code = mndDropXnodeJob(pMnode, pReq, pObj);
×
3739

UNCOV
3740
_OVER:
×
UNCOV
3741
  XND_LOG_END(code, lino);
×
UNCOV
3742
  mndReleaseXnodeJob(pMnode, pObj);
×
UNCOV
3743
  return code;
×
3744
}
3745

3746
static int32_t dropXnodeJobByWhereCond(SMnode *pMnode, SRpcMsg *pReq, SMDropXnodeJobReq *dropReq) {
19,516✔
3747
  int32_t       code = 0;
19,516✔
3748
  int32_t       lino = 0;
19,516✔
3749
  SXnodeJobObj *pObj = NULL;
19,516✔
3750
  SNode        *pWhere = NULL;
19,516✔
3751
  SArray       *pArray = NULL;
19,516✔
3752
  SArray       *pResult = NULL;
19,516✔
3753

3754
  if (NULL != dropReq->ast.ptr) {
19,516✔
3755
    TAOS_CHECK_GOTO(mndAcquireXnodeJobsAll(pMnode, &pArray), &lino, _OVER);
19,516✔
3756
    TAOS_CHECK_GOTO(nodesStringToNode(dropReq->ast.ptr, &pWhere), &lino, _OVER);
19,516✔
3757
    TAOS_CHECK_GOTO(filterJobsByWhereCond(pWhere, pArray, &pResult), &lino, _OVER);
19,516✔
3758

3759
    for (int32_t i = 0; i < pResult->size; i++) {
19,516✔
UNCOV
3760
      pObj = taosArrayGet(pResult, i);
×
UNCOV
3761
      code = mndCheckXnodeJobOwner(pMnode, pReq->info.conn.user, pObj);
×
UNCOV
3762
      if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3763
        mError("user:%s has no privilege on xnode job:%d", pReq->info.conn.user, pObj->id);
×
3764
        goto _OVER;
×
3765
      }
3766
      TAOS_CHECK_GOTO(mndDropXnodeJob(pMnode, NULL, pObj), &lino, _OVER);
×
3767
    }
3768
  }
3769

3770
_OVER:
19,516✔
3771
  XND_LOG_END(code, lino);
19,516✔
3772
  if (pResult != NULL) {
19,516✔
3773
    taosArrayDestroy(pResult);
19,516✔
3774
  }
3775
  if (pWhere != NULL) {
19,516✔
3776
    nodesDestroyNode(pWhere);
19,516✔
3777
  }
3778
  if (pArray != NULL) {
19,516✔
3779
    taosArrayDestroy(pArray);
19,516✔
3780
  }
3781
  return code;
19,516✔
3782
}
3783

3784
static int32_t mndProcessDropXnodeJobReq(SRpcMsg *pReq) {
19,516✔
3785
  mDebug("drop xnode job req, content len:%d", pReq->contLen);
19,516✔
3786
  SMnode           *pMnode = pReq->info.node;
19,516✔
3787
  int32_t           code = -1;
19,516✔
3788
  SMDropXnodeJobReq dropReq = {0};
19,516✔
3789

3790
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_DROP_XNODE_JOB);
19,516✔
3791
  if (code != TSDB_CODE_SUCCESS) {
19,516✔
UNCOV
3792
    mError("failed check permission for drop xnode jobs, code:%s", tstrerror(code));
×
UNCOV
3793
    goto _OVER;
×
3794
  }
3795
  TAOS_CHECK_GOTO(tDeserializeSMDropXnodeJobReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
19,516✔
3796
  mDebug("Xnode drop job with jid:%d", dropReq.jid);
19,516✔
3797

3798
  if (dropReq.jid <= 0 && dropReq.ast.ptr == NULL) {
19,516✔
3799
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
3800
    goto _OVER;
×
3801
  }
3802
  if (dropReq.jid > 0) {
19,516✔
UNCOV
3803
    code = dropXnodeJobById(pMnode, pReq, dropReq.jid);
×
UNCOV
3804
    if (code == 0) {
×
UNCOV
3805
      code = TSDB_CODE_ACTION_IN_PROGRESS;
×
3806
    } else {
UNCOV
3807
      goto _OVER;
×
3808
    }
3809
  } else {
3810
    TAOS_CHECK_GOTO(dropXnodeJobByWhereCond(pMnode, pReq, &dropReq), NULL, _OVER);
19,516✔
3811
  }
3812

3813
_OVER:
19,516✔
3814
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
19,516✔
UNCOV
3815
    mError("xnode:%d, failed to drop since %s", dropReq.jid, tstrerror(code));
×
3816
  }
3817
  tFreeSMDropXnodeJobReq(&dropReq);
19,516✔
3818
  TAOS_RETURN(code);
19,516✔
3819
}
3820

3821
/**
3822
 * @brief Mapping the columns of show xnode jobs
3823
 *
3824
 * See [xnodeTaskJobSchema] in systable.h.
3825
 *
3826
 *  {.name = "jid", .bytes = 4, .type = TSDB_DATA_TYPE_INT, .sysInfo = false},
3827
    {.name = "tid", .bytes = 4, .type = TSDB_DATA_TYPE_INT, .sysInfo = false},
3828
    {.name = "config", .bytes = TSDB_XNODE_NAME_LEN + VARSTR_HEADER_SIZE, .type = TSDB_DATA_TYPE_VARCHAR, .sysInfo =
3829
 false},
3830
    {.name = "status", .bytes = 10 + VARSTR_HEADER_SIZE, .type = TSDB_DATA_TYPE_VARCHAR, .sysInfo = false},
3831
    // {.name = "reason", .bytes = 10 + VARSTR_HEADER_SIZE, .type = TSDB_DATA_TYPE_VARCHAR, .sysInfo = false},
3832
    {.name = "create_time", .bytes = 8, .type = TSDB_DATA_TYPE_TIMESTAMP, .sysInfo = false},
3833
    {.name = "update_time", .bytes = 8, .type = TSDB_DATA_TYPE_TIMESTAMP, .sysInfo = false},
3834
 * @param pReq
3835
 * @param pShow
3836
 * @param pBlock
3837
 * @param rows
3838
 * @return int32_t
3839
 */
3840
static int32_t mndRetrieveXnodeJobs(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
4,522✔
3841
  SMnode       *pMnode = pReq->info.node;
4,522✔
3842
  SSdb         *pSdb = pMnode->pSdb;
4,522✔
3843
  int32_t       numOfRows = 0;
4,522✔
3844
  int32_t       cols = 0;
4,522✔
3845
  SXnodeJobObj *pObj = NULL;
4,522✔
3846
  char          buf[VARSTR_HEADER_SIZE + TMAX(TSDB_XNODE_TASK_JOB_CONFIG_LEN, TSDB_XNODE_TASK_REASON_LEN)];
4,522✔
3847
  char          status[64] = {0};
4,522✔
3848
  int32_t       code = 0;
4,522✔
3849
  SUserObj     *pOperUser = NULL;
4,522✔
3850
  mDebug("show.type:%d, %s:%d: retrieve xnode jobs with rows: %d", pShow->type, __FILE__, __LINE__, rows);
4,522✔
3851

3852
  TAOS_CHECK_GOTO(mndAcquireUser(pMnode, pReq->info.conn.user, &pOperUser), NULL, _end);
4,522✔
3853

3854
  while (numOfRows < rows) {
4,522✔
3855
    pShow->pIter = sdbFetch(pSdb, SDB_XNODE_JOB, pShow->pIter, (void **)&pObj);
4,522✔
3856
    if (pShow->pIter == NULL) break;
4,522✔
3857

UNCOV
3858
    if (!pOperUser->superUser) {
×
3859
      SXnodeTaskObj *pTask = mndAcquireXnodeTaskById(pMnode, pObj->taskId);
×
3860
      bool allowed = false;
×
UNCOV
3861
      if (pTask != NULL) {
×
3862
        if (pTask->createdBy != NULL && pTask->createdByLen > 0 &&
×
3863
            strncmp(pTask->createdBy, pReq->info.conn.user, TSDB_USER_LEN) == 0) {
×
UNCOV
3864
          allowed = true;
×
3865
        }
3866
        mndReleaseXnodeTask(pMnode, pTask);
×
3867
      }
3868
      if (!allowed) {
×
3869
        sdbRelease(pSdb, pObj);
×
3870
        continue;
×
3871
      }
3872
    }
3873

3874
    cols = 0;
×
3875
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3876
    // id
3877
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->id, false);
×
UNCOV
3878
    if (code != 0) goto _end;
×
3879
    // tid
3880
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3881
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->taskId, false);
×
3882
    if (code != 0) goto _end;
×
3883

3884
    // config
UNCOV
3885
    buf[0] = 0;
×
3886
    STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->config, pShow->pMeta->pSchemas[cols].bytes);
×
UNCOV
3887
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3888
    code = colDataSetVal(pColInfo, numOfRows, buf, false);
×
3889
    if (code != 0) goto _end;
×
3890

3891
    // via
UNCOV
3892
    if (pObj->via != 0) {
×
3893
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3894
      code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->via, false);
×
3895
      if (code != 0) goto _end;
×
3896
    } else {
3897
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3898
      colDataSetNULL(pColInfo, numOfRows);
×
3899
    }
3900

3901
    // xnode_id
3902
    if (pObj->xnodeId != 0) {
×
3903
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
3904
      code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->xnodeId, false);
×
UNCOV
3905
      if (code != 0) goto _end;
×
3906
    } else {
3907
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3908
      colDataSetNULL(pColInfo, numOfRows);
×
3909
    }
3910

3911
    // status
3912
    if (pObj->statusLen > 0) {
×
3913
      buf[0] = 0;
×
3914
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->status, pShow->pMeta->pSchemas[cols].bytes);
×
3915
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
3916
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
UNCOV
3917
      if (code != 0) goto _end;
×
3918
    } else {
UNCOV
3919
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3920
      colDataSetNULL(pColInfo, numOfRows);
×
3921
    }
3922

3923
    // reason
UNCOV
3924
    if (pObj->reasonLen > 0) {
×
3925
      buf[0] = 0;
×
3926
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->reason, pShow->pMeta->pSchemas[cols].bytes);
×
3927
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3928
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
×
UNCOV
3929
      if (code != 0) goto _end;
×
3930
    } else {
3931
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
3932
      colDataSetNULL(pColInfo, numOfRows);
×
3933
    }
3934

3935
    // create_time
3936
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3937
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->createTime, false);
×
3938
    if (code != 0) goto _end;
×
3939

3940
    // update_time
3941
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
3942
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->updateTime, false);
×
3943
    if (code != 0) goto _end;
×
3944

UNCOV
3945
    numOfRows++;
×
3946
    sdbRelease(pSdb, pObj);
×
3947
  }
3948

3949
_end:
4,522✔
3950
  if (code != 0) sdbRelease(pSdb, pObj);
4,522✔
3951
  mndReleaseUser(pMnode, pOperUser);
4,522✔
3952

3953
  pShow->numOfRows += numOfRows;
4,522✔
3954
  return numOfRows;
4,522✔
3955
}
3956
static void mndCancelGetNextXnodeJob(SMnode *pMnode, void *pIter) {
×
3957
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
3958
  sdbCancelFetchByType(pSdb, pIter, SDB_XNODE_JOB);
×
UNCOV
3959
}
×
3960

3961
static size_t taosCurlWriteData(char *pCont, size_t contLen, size_t nmemb, void *userdata) {
×
3962
  SCurlResp *pRsp = userdata;
×
3963
  if (contLen == 0 || nmemb == 0 || pCont == NULL) {
×
3964
    pRsp->dataLen = 0;
×
UNCOV
3965
    pRsp->data = NULL;
×
UNCOV
3966
    uError("curl response is received, len:%" PRId64, pRsp->dataLen);
×
3967
    return 0;
×
3968
  }
3969

3970
  int64_t newDataSize = (int64_t)contLen * nmemb;
×
UNCOV
3971
  int64_t size = pRsp->dataLen + newDataSize;
×
3972

3973
  if (pRsp->data == NULL) {
×
UNCOV
3974
    pRsp->data = taosMemoryMalloc(size + 1);
×
UNCOV
3975
    if (pRsp->data == NULL) {
×
3976
      uError("failed to prepare recv buffer for post rsp, len:%d, code:%s", (int32_t)size + 1, tstrerror(terrno));
×
3977
      return 0;  // return the recv length, if failed, return 0
×
3978
    }
3979
  } else {
UNCOV
3980
    char *p = taosMemoryRealloc(pRsp->data, size + 1);
×
3981
    if (p == NULL) {
×
3982
      uError("failed to prepare recv buffer for post rsp, len:%d, code:%s", (int32_t)size + 1, tstrerror(terrno));
×
3983
      return 0;  // return the recv length, if failed, return 0
×
3984
    }
3985

UNCOV
3986
    pRsp->data = p;
×
3987
  }
3988

UNCOV
3989
  if (pRsp->data != NULL) {
×
3990
    (void)memcpy(pRsp->data + pRsp->dataLen, pCont, newDataSize);
×
3991

3992
    pRsp->dataLen = size;
×
3993
    pRsp->data[size] = 0;
×
3994

UNCOV
3995
    uDebugL("curl response is received, len:%" PRId64 ", content:%s", size, pRsp->data);
×
3996
    return newDataSize;
×
3997
  } else {
3998
    pRsp->dataLen = 0;
×
3999
    uError("failed to malloc curl response");
×
4000
    return 0;
×
4001
  }
4002
}
4003

4004
static int32_t taosCurlGetRequest(const char *url, SCurlResp *pRsp, int32_t timeout, const char *socketPath) {
×
4005
  CURL   *curl = NULL;
×
UNCOV
4006
  int32_t code = 0;
×
UNCOV
4007
  int32_t lino = 0;
×
4008

4009
  curl = curl_easy_init();
×
4010
  if (curl == NULL) {
×
4011
    uError("failed to create curl handle");
×
4012
    return -1;
×
4013
  }
4014

4015
  #ifndef WINDOWS
4016
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_UNIX_SOCKET_PATH, socketPath), &lino, _OVER);
×
4017
  #endif
4018
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_URL, url), &lino, _OVER);
×
UNCOV
4019
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, taosCurlWriteData), &lino, _OVER);
×
UNCOV
4020
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_WRITEDATA, pRsp), &lino, _OVER);
×
UNCOV
4021
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, timeout), &lino, _OVER);
×
4022

UNCOV
4023
  uDebug("curl get request will sent, url:%s", url);
×
UNCOV
4024
  CURLcode curlCode = curl_easy_perform(curl);
×
UNCOV
4025
  if (curlCode != CURLE_OK) {
×
UNCOV
4026
    if (curlCode == CURLE_OPERATION_TIMEDOUT) {
×
UNCOV
4027
      mError("xnode failed to perform curl action, code:%d", curlCode);
×
UNCOV
4028
      code = TSDB_CODE_MND_XNODE_URL_RESP_TIMEOUT;
×
UNCOV
4029
      goto _OVER;
×
4030
    }
UNCOV
4031
    uError("failed to perform curl action, code:%d", curlCode);
×
UNCOV
4032
    code = TSDB_CODE_MND_XNODE_URL_CANT_ACCESS;
×
UNCOV
4033
    goto _OVER;
×
4034
  }
4035

4036
  long http_code = 0;
×
UNCOV
4037
  TAOS_CHECK_GOTO(curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &http_code), &lino, _OVER);
×
4038
  if (http_code != 200) {
×
4039
    code = TSDB_CODE_MND_XNODE_HTTP_CODE_ERROR;
×
4040
  }
4041

4042
_OVER:
×
4043
  if (curl != NULL) curl_easy_cleanup(curl);
×
4044
  XND_LOG_END(code, lino);
×
UNCOV
4045
  return code;
×
4046
}
4047

4048
static int32_t taosCurlPostRequest(const char *url, SCurlResp *pRsp, const char *buf, int32_t bufLen, int32_t timeout,
×
4049
                                   const char *socketPath) {
UNCOV
4050
  struct curl_slist *headers = NULL;
×
UNCOV
4051
  CURL              *curl = NULL;
×
4052
  int32_t            code = 0;
×
4053
  int32_t            lino = 0;
×
4054

4055
  curl = curl_easy_init();
×
4056
  if (curl == NULL) {
×
UNCOV
4057
    mError("xnode failed to create curl handle");
×
4058
    return -1;
×
4059
  }
4060

4061
  headers = curl_slist_append(headers, "Content-Type:application/json;charset=UTF-8");
×
4062
  #ifndef WINDOWS
4063
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_UNIX_SOCKET_PATH, socketPath), &lino, _OVER);
×
4064
  #endif
4065
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers), &lino, _OVER);
×
4066
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_URL, url), &lino, _OVER);
×
4067
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, taosCurlWriteData), &lino, _OVER);
×
4068
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_WRITEDATA, pRsp), &lino, _OVER);
×
4069
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, timeout), &lino, _OVER);
×
UNCOV
4070
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_POST, 1), &lino, _OVER);
×
4071
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, bufLen), &lino, _OVER);
×
UNCOV
4072
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_POSTFIELDS, buf), &lino, _OVER);
×
UNCOV
4073
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_VERBOSE, 0L), &lino, _OVER);
×
UNCOV
4074
  TAOS_CHECK_GOTO(curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1L), &lino, _OVER);
×
4075

UNCOV
4076
  mDebug("xnode curl post request will sent, url:%s len:%d content:%s", url, bufLen, buf);
×
UNCOV
4077
  CURLcode curlCode = curl_easy_perform(curl);
×
4078

UNCOV
4079
  if (curlCode != CURLE_OK) {
×
UNCOV
4080
    if (curlCode == CURLE_OPERATION_TIMEDOUT) {
×
UNCOV
4081
      mError("xnode failed to perform curl action, code:%d", curlCode);
×
UNCOV
4082
      code = TSDB_CODE_MND_XNODE_URL_RESP_TIMEOUT;
×
UNCOV
4083
      goto _OVER;
×
4084
    }
UNCOV
4085
    uError("xnode failed to perform curl action, code:%d", curlCode);
×
UNCOV
4086
    code = TSDB_CODE_MND_XNODE_URL_CANT_ACCESS;
×
UNCOV
4087
    goto _OVER;
×
4088
  }
4089

UNCOV
4090
  long http_code = 0;
×
UNCOV
4091
  TAOS_CHECK_GOTO(curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &http_code), &lino, _OVER);
×
UNCOV
4092
  if (http_code != 200) {
×
UNCOV
4093
    mError("xnode failed to perform curl action, http code:%ld", http_code);
×
4094
    code = TSDB_CODE_MND_XNODE_HTTP_CODE_ERROR;
×
4095
  }
4096

4097
_OVER:
×
UNCOV
4098
  if (curl != NULL) {
×
4099
    curl_slist_free_all(headers);
×
4100
    curl_easy_cleanup(curl);
×
4101
  }
UNCOV
4102
  XND_LOG_END(code, lino);
×
UNCOV
4103
  return code;
×
4104
}
4105

UNCOV
4106
static int32_t taosCurlDeleteRequest(const char *url, SCurlResp *pRsp, int32_t timeout, const char *socketPath) {
×
UNCOV
4107
  CURL   *curl = NULL;
×
UNCOV
4108
  int32_t code = 0;
×
UNCOV
4109
  int32_t lino = 0;
×
4110

UNCOV
4111
  curl = curl_easy_init();
×
UNCOV
4112
  if (curl == NULL) {
×
UNCOV
4113
    uError("xnode failed to create curl handle");
×
UNCOV
4114
    return -1;
×
4115
  }
4116

4117
  #ifndef WINDOWS
4118
  if (curl_easy_setopt(curl, CURLOPT_UNIX_SOCKET_PATH, socketPath)) goto _OVER;
×
4119
  #endif
UNCOV
4120
  if (curl_easy_setopt(curl, CURLOPT_URL, url) != 0) goto _OVER;
×
UNCOV
4121
  if (curl_easy_setopt(curl, CURLOPT_CUSTOMREQUEST, "DELETE") != 0) goto _OVER;
×
UNCOV
4122
  if (curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, taosCurlWriteData) != 0) goto _OVER;
×
UNCOV
4123
  if (curl_easy_setopt(curl, CURLOPT_WRITEDATA, pRsp) != 0) goto _OVER;
×
UNCOV
4124
  if (curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, timeout) != 0) goto _OVER;
×
4125

UNCOV
4126
  uDebug("xnode curl get request will sent, url:%s", url);
×
UNCOV
4127
  CURLcode curlCode = curl_easy_perform(curl);
×
UNCOV
4128
  if (curlCode != CURLE_OK) {
×
UNCOV
4129
    uError("xnode failed to perform curl action, curl code:%d", curlCode);
×
UNCOV
4130
    if (curlCode == CURLE_OPERATION_TIMEDOUT) {
×
UNCOV
4131
      code = TSDB_CODE_MND_XNODE_URL_RESP_TIMEOUT;
×
UNCOV
4132
      goto _OVER;
×
4133
    }
UNCOV
4134
    code = TSDB_CODE_MND_XNODE_URL_CANT_ACCESS;
×
UNCOV
4135
    goto _OVER;
×
4136
  }
4137

UNCOV
4138
  long http_code = 0;
×
UNCOV
4139
  TAOS_CHECK_GOTO(curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &http_code), &lino, _OVER);
×
UNCOV
4140
  if (http_code != 200 && http_code != 204) {
×
UNCOV
4141
    uError("xnode curl request response http code:%ld", http_code);
×
UNCOV
4142
    code = TSDB_CODE_MND_XNODE_HTTP_CODE_ERROR;
×
4143
  }
4144

4145
_OVER:
×
4146
  if (curl != NULL) curl_easy_cleanup(curl);
×
UNCOV
4147
  XND_LOG_END(code, lino);
×
UNCOV
4148
  return code;
×
4149
}
4150

4151
SJson *mndSendReqRetJson(const char *url, EHttpType type, int64_t timeout, const char *buf, int64_t bufLen) {
15,946✔
4152
  SJson    *pJson = NULL;
15,946✔
4153
  SCurlResp curlRsp = {0};
15,946✔
4154
  char      socketPath[PATH_MAX] = {0};
15,946✔
4155

4156
  getXnodedPipeName(socketPath, sizeof(socketPath));
15,946✔
4157
  #ifndef WINDOWS
4158
  if (!taosCheckExistFile(socketPath)) {
15,946✔
4159
    uError("xnode failed to send request, socket path:%s not exist", socketPath);
15,946✔
4160
    terrno = TSDB_CODE_MND_XNODE_URL_CANT_ACCESS;
15,946✔
4161
    goto _EXIT;
15,946✔
4162
  }
4163
  #endif
UNCOV
4164
  if (type == HTTP_TYPE_GET) {
×
UNCOV
4165
    if ((terrno = taosCurlGetRequest(url, &curlRsp, timeout, socketPath)) != 0) {
×
UNCOV
4166
      goto _OVER;
×
4167
    }
UNCOV
4168
  } else if (type == HTTP_TYPE_POST) {
×
4169
    if ((terrno = taosCurlPostRequest(url, &curlRsp, buf, bufLen, timeout, socketPath)) != 0) {
×
4170
      goto _OVER;
×
4171
    }
UNCOV
4172
  } else if (type == HTTP_TYPE_DELETE) {
×
UNCOV
4173
    if ((terrno = taosCurlDeleteRequest(url, &curlRsp, timeout, socketPath)) != 0) {
×
UNCOV
4174
      goto _OVER;
×
4175
    }
4176
  } else {
UNCOV
4177
    uError("xnode invalid http type:%d", type);
×
UNCOV
4178
    terrno = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
4179
    goto _EXIT;
×
4180
  }
4181

UNCOV
4182
_OVER:
×
UNCOV
4183
  if (terrno == TSDB_CODE_SUCCESS) {
×
UNCOV
4184
    if (curlRsp.data == NULL || curlRsp.dataLen == 0) {
×
UNCOV
4185
      pJson = tjsonCreateObject();
×
UNCOV
4186
      goto _EXIT;
×
4187
    }
UNCOV
4188
    pJson = tjsonParse(curlRsp.data);
×
UNCOV
4189
    if (pJson == NULL) {
×
UNCOV
4190
      terrno = TSDB_CODE_INVALID_JSON_FORMAT;
×
UNCOV
4191
      goto _EXIT;
×
4192
    }
UNCOV
4193
  } else if (terrno == TSDB_CODE_MND_XNODE_HTTP_CODE_ERROR) {
×
UNCOV
4194
    pJson = tjsonCreateObject();
×
4195
    char *buf = taosMemCalloc(1, curlRsp.dataLen + 1);
×
UNCOV
4196
    (void)memcpy(buf, curlRsp.data, curlRsp.dataLen);
×
UNCOV
4197
    if (tjsonAddStringToObject(pJson, "__inner_error", buf) != TSDB_CODE_SUCCESS) {
×
UNCOV
4198
      taosMemoryFreeClear(buf);
×
UNCOV
4199
      goto _EXIT;
×
4200
    }
UNCOV
4201
    taosMemoryFreeClear(buf);
×
4202
  }
4203

4204
_EXIT:
15,946✔
4205
  if (curlRsp.data != NULL) taosMemoryFreeClear(curlRsp.data);
15,946✔
4206
  if (terrno != TSDB_CODE_SUCCESS) {
15,946✔
4207
    mError("xnode failed to send request, url: %s, since:%s", url, tstrerror(terrno));
15,946✔
4208
  }
4209
  return pJson;
15,946✔
4210
}
4211

4212
static int32_t mndGetXnodeStatus(SXnodeObj *pObj, char *status, int32_t statusLen) {
3,213✔
4213
  int32_t code = 0;
3,213✔
4214
  SJson  *pJson = NULL;
3,213✔
4215

4216
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
3,213✔
4217
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/xnode/%d", XNODED_PIPE_SOCKET_URL, pObj->id);
3,213✔
4218
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_GET, defaultTimeout, NULL, 0);
3,213✔
4219
  if (pJson == NULL) {
3,213✔
4220
    code = terrno;
3,213✔
4221
    goto _OVER;
3,213✔
4222
  }
4223

UNCOV
4224
  code = tjsonGetStringValue2(pJson, "status", status, statusLen);
×
UNCOV
4225
  if (code < 0) {
×
UNCOV
4226
    code = TSDB_CODE_INVALID_JSON_FORMAT;
×
UNCOV
4227
    goto _OVER;
×
4228
  }
UNCOV
4229
  if (strlen(status) == 0) {
×
UNCOV
4230
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
4231
    goto _OVER;
×
4232
  }
4233

4234
_OVER:
3,213✔
4235
  if (pJson != NULL) tjsonDelete(pJson);
3,213✔
4236
  TAOS_RETURN(code);
3,213✔
4237
}
4238

4239
/** xnode agent section **/
4240

4241
SSdbRaw *mndXnodeAgentActionEncode(SXnodeAgentObj *pObj) {
13,121✔
4242
  int32_t code = 0;
13,121✔
4243
  int32_t lino = 0;
13,121✔
4244
  terrno = TSDB_CODE_OUT_OF_MEMORY;
13,121✔
4245

4246
  if (NULL == pObj) {
13,121✔
UNCOV
4247
    terrno = TSDB_CODE_INVALID_PARA;
×
UNCOV
4248
    return NULL;
×
4249
  }
4250

4251
  int32_t rawDataLen =
13,121✔
4252
      sizeof(SXnodeAgentObj) + TSDB_XNODE_RESERVE_SIZE + pObj->nameLen + pObj->tokenLen + pObj->statusLen;
13,121✔
4253

4254
  SSdbRaw *pRaw = sdbAllocRaw(SDB_XNODE_AGENT, TSDB_XNODE_VER_NUMBER, rawDataLen);
13,121✔
4255
  if (pRaw == NULL) goto _OVER;
13,121✔
4256

4257
  int32_t dataPos = 0;
13,121✔
4258
  SDB_SET_INT32(pRaw, dataPos, pObj->id, _OVER)
13,121✔
4259
  SDB_SET_INT32(pRaw, dataPos, pObj->nameLen, _OVER)
13,121✔
4260
  SDB_SET_BINARY(pRaw, dataPos, pObj->name, pObj->nameLen, _OVER)
13,121✔
4261
  SDB_SET_INT32(pRaw, dataPos, pObj->tokenLen, _OVER)
13,121✔
4262
  SDB_SET_BINARY(pRaw, dataPos, pObj->token, pObj->tokenLen, _OVER)
13,121✔
4263
  SDB_SET_INT32(pRaw, dataPos, pObj->statusLen, _OVER)
13,121✔
4264
  SDB_SET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
13,121✔
4265
  SDB_SET_INT64(pRaw, dataPos, pObj->createTime, _OVER)
13,121✔
4266
  SDB_SET_INT64(pRaw, dataPos, pObj->updateTime, _OVER)
13,121✔
4267

4268
  SDB_SET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
13,121✔
4269

4270
  terrno = 0;
13,121✔
4271

4272
_OVER:
13,121✔
4273
  if (terrno != 0) {
13,121✔
UNCOV
4274
    mError("xnode agent:%d, failed to encode to raw:%p since %s", pObj->id, pRaw, terrstr());
×
UNCOV
4275
    sdbFreeRaw(pRaw);
×
UNCOV
4276
    return NULL;
×
4277
  }
4278

4279
  mTrace("xnode agent:%d, encode to raw:%p, row:%p", pObj->id, pRaw, pObj);
13,121✔
4280
  return pRaw;
13,121✔
4281
}
4282

4283
SSdbRow *mndXnodeAgentActionDecode(SSdbRaw *pRaw) {
9,789✔
4284
  int32_t code = 0;
9,789✔
4285
  int32_t lino = 0;
9,789✔
4286
  terrno = TSDB_CODE_OUT_OF_MEMORY;
9,789✔
4287
  SSdbRow        *pRow = NULL;
9,789✔
4288
  SXnodeAgentObj *pObj = NULL;
9,789✔
4289

4290
  if (NULL == pRaw) {
9,789✔
UNCOV
4291
    terrno = TSDB_CODE_INVALID_PARA;
×
UNCOV
4292
    return NULL;
×
4293
  }
4294

4295
  int8_t sver = 0;
9,789✔
4296
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
9,789✔
4297

4298
  if (sver != TSDB_XNODE_VER_NUMBER) {
9,789✔
UNCOV
4299
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
×
UNCOV
4300
    goto _OVER;
×
4301
  }
4302

4303
  pRow = sdbAllocRow(sizeof(SXnodeAgentObj));
9,789✔
4304
  if (pRow == NULL) goto _OVER;
9,789✔
4305

4306
  pObj = sdbGetRowObj(pRow);
9,789✔
4307
  if (pObj == NULL) goto _OVER;
9,789✔
4308

4309
  int32_t dataPos = 0;
9,789✔
4310
  SDB_GET_INT32(pRaw, dataPos, &pObj->id, _OVER)
9,789✔
4311
  SDB_GET_INT32(pRaw, dataPos, &pObj->nameLen, _OVER)
9,789✔
4312
  if (pObj->nameLen > 0) {
9,789✔
4313
    pObj->name = taosMemoryCalloc(pObj->nameLen, 1);
9,789✔
4314
    if (pObj->name == NULL) goto _OVER;
9,789✔
4315
    SDB_GET_BINARY(pRaw, dataPos, pObj->name, pObj->nameLen, _OVER)
9,789✔
4316
  } else {
UNCOV
4317
    pObj->name = NULL;
×
4318
  }
4319
  SDB_GET_INT32(pRaw, dataPos, &pObj->tokenLen, _OVER)
9,789✔
4320
  if (pObj->tokenLen > 0) {
9,789✔
4321
    pObj->token = taosMemoryCalloc(pObj->tokenLen, 1);
9,789✔
4322
    if (pObj->token == NULL) goto _OVER;
9,789✔
4323
    SDB_GET_BINARY(pRaw, dataPos, pObj->token, pObj->tokenLen, _OVER)
9,789✔
4324
  } else {
4325
    pObj->token = NULL;
×
4326
  }
4327
  SDB_GET_INT32(pRaw, dataPos, &pObj->statusLen, _OVER)
9,789✔
4328
  if (pObj->statusLen > 0) {
9,789✔
4329
    pObj->status = taosMemoryCalloc(pObj->statusLen, 1);
8,405✔
4330
    if (pObj->status == NULL) goto _OVER;
8,405✔
4331
    SDB_GET_BINARY(pRaw, dataPos, pObj->status, pObj->statusLen, _OVER)
8,405✔
4332
  } else {
4333
    pObj->status = NULL;
1,384✔
4334
  }
4335
  SDB_GET_INT64(pRaw, dataPos, &pObj->createTime, _OVER)
9,789✔
4336
  SDB_GET_INT64(pRaw, dataPos, &pObj->updateTime, _OVER)
9,789✔
4337

4338
  SDB_GET_RESERVE(pRaw, dataPos, TSDB_XNODE_RESERVE_SIZE, _OVER)
9,789✔
4339

4340
  terrno = 0;
9,789✔
4341

4342
_OVER:
9,789✔
4343
  if (terrno != 0) {
9,789✔
UNCOV
4344
    mError("xnode agent:%d, failed to decode from raw:%p since %s", pObj == NULL ? 0 : pObj->id, pRaw, terrstr());
×
UNCOV
4345
    if (pObj != NULL) {
×
UNCOV
4346
      taosMemoryFreeClear(pObj->name);
×
UNCOV
4347
      taosMemoryFreeClear(pObj->token);
×
UNCOV
4348
      taosMemoryFreeClear(pObj->status);
×
4349
    }
UNCOV
4350
    taosMemoryFreeClear(pRow);
×
UNCOV
4351
    return NULL;
×
4352
  }
4353

4354
  mTrace("xnode agent:%d, decode from raw:%p, row:%p", pObj->id, pRaw, pObj);
9,789✔
4355
  return pRow;
9,789✔
4356
}
4357

4358
int32_t mndXnodeAgentActionInsert(SSdb *pSdb, SXnodeAgentObj *pObj) {
2,261✔
4359
  mDebug("xnode agent:%d, perform insert action, row:%p", pObj->id, pObj);
2,261✔
4360
  return 0;
2,261✔
4361
}
4362

4363
int32_t mndXnodeAgentActionUpdate(SSdb *pSdb, SXnodeAgentObj *pOld, SXnodeAgentObj *pNew) {
4,879✔
4364
  mDebug("xnode agent:%d, perform update action, old row:%p new row:%p", pOld->id, pOld, pNew);
4,879✔
4365

4366
  taosWLockLatch(&pOld->lock);
4,879✔
4367
  swapFields(&pNew->nameLen, &pNew->name, &pOld->nameLen, &pOld->name);
4,879✔
4368
  swapFields(&pNew->tokenLen, &pNew->token, &pOld->tokenLen, &pOld->token);
4,879✔
4369
  swapFields(&pNew->statusLen, &pNew->status, &pOld->statusLen, &pOld->status);
4,879✔
4370
  if (pNew->updateTime > pOld->updateTime) {
4,879✔
4371
    pOld->updateTime = pNew->updateTime;
357✔
4372
  }
4373
  taosWUnLockLatch(&pOld->lock);
4,879✔
4374
  return 0;
4,879✔
4375
}
4376

4377
static void mndFreeXnodeAgent(SXnodeAgentObj *pObj) {
11,662✔
4378
  if (pObj == NULL) return;
11,662✔
4379
  if (pObj->name != NULL) {
11,662✔
4380
    taosMemoryFreeClear(pObj->name);
11,662✔
4381
  }
4382
  if (pObj->token != NULL) {
11,662✔
4383
    taosMemoryFreeClear(pObj->token);
11,662✔
4384
  }
4385
  if (pObj->status != NULL) {
11,662✔
4386
    taosMemoryFreeClear(pObj->status);
10,115✔
4387
  }
4388
}
4389

4390
int32_t mndXnodeAgentActionDelete(SSdb *pSdb, SXnodeAgentObj *pObj) {
9,401✔
4391
  mDebug("xnode agent:%d, perform delete action, row:%p", pObj->id, pObj);
9,401✔
4392
  mndFreeXnodeAgent(pObj);
9,401✔
4393
  return 0;
9,401✔
4394
}
4395

4396
static int32_t mndSetCreateXnodeAgentRedoLogs(STrans *pTrans, SXnodeAgentObj *pObj) {
2,261✔
4397
  int32_t  code = 0;
2,261✔
4398
  SSdbRaw *pRedoRaw = mndXnodeAgentActionEncode(pObj);
2,261✔
4399
  if (pRedoRaw == NULL) {
2,261✔
4400
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
4401
    if (terrno != 0) code = terrno;
×
UNCOV
4402
    TAOS_RETURN(code);
×
4403
  }
4404
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
2,261✔
4405
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_CREATING));
2,261✔
4406
  TAOS_RETURN(code);
2,261✔
4407
}
4408

4409
static int32_t mndSetCreateXnodeAgentUndoLogs(STrans *pTrans, SXnodeAgentObj *pObj) {
2,261✔
4410
  int32_t  code = 0;
2,261✔
4411
  SSdbRaw *pUndoRaw = mndXnodeAgentActionEncode(pObj);
2,261✔
4412
  if (pUndoRaw == NULL) {
2,261✔
UNCOV
4413
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
4414
    if (terrno != 0) code = terrno;
×
4415
    TAOS_RETURN(code);
×
4416
  }
4417
  TAOS_CHECK_RETURN(mndTransAppendUndolog(pTrans, pUndoRaw));
2,261✔
4418
  TAOS_CHECK_RETURN(sdbSetRawStatus(pUndoRaw, SDB_STATUS_DROPPED));
2,261✔
4419
  TAOS_RETURN(code);
2,261✔
4420
}
4421

4422
static int32_t mndSetCreateXnodeAgentCommitLogs(STrans *pTrans, SXnodeAgentObj *pObj) {
2,618✔
4423
  int32_t  code = 0;
2,618✔
4424
  SSdbRaw *pCommitRaw = mndXnodeAgentActionEncode(pObj);
2,618✔
4425
  if (pCommitRaw == NULL) {
2,618✔
4426
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
4427
    if (terrno != 0) code = terrno;
×
4428
    TAOS_RETURN(code);
×
4429
  }
4430
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
2,618✔
4431
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY));
2,618✔
4432
  TAOS_RETURN(code);
2,618✔
4433
}
4434

4435
void mndReleaseXnodeAgent(SMnode *pMnode, SXnodeAgentObj *pObj) {
2,737✔
4436
  SSdb *pSdb = pMnode->pSdb;
2,737✔
4437
  sdbRelease(pSdb, pObj);
2,737✔
4438
}
2,737✔
4439

4440
static int32_t mndXnodeCheckSuperUser(SMnode *pMnode, const char *user, bool *isSuperUser) {
5,712✔
4441
  if (pMnode == NULL || user == NULL || isSuperUser == NULL) {
5,712✔
UNCOV
4442
    return TSDB_CODE_INVALID_PARA;
×
4443
  }
4444

4445
  SUserObj *pUser = NULL;
5,712✔
4446
  int32_t   code = mndAcquireUser(pMnode, user, &pUser);
5,712✔
4447
  if (code != 0) {
5,712✔
UNCOV
4448
    *isSuperUser = false;
×
UNCOV
4449
    return code;
×
4450
  }
4451

4452
  if (pUser->superUser) {
5,712✔
4453
    *isSuperUser = true;
5,712✔
4454
  } else {
UNCOV
4455
    *isSuperUser = false;
×
4456
  }
4457
  mndReleaseUser(pMnode, pUser);
5,712✔
4458
  return TSDB_CODE_SUCCESS;
5,712✔
4459
}
4460

4461
static int32_t mndCheckXnodeTaskOwner(SMnode *pMnode, const char *user, SXnodeTaskObj *pTask) {
×
UNCOV
4462
  SUserObj *pUser = NULL;
×
4463
  int32_t   code = mndAcquireUser(pMnode, user, &pUser);
×
4464
  if (code != 0) return code;
×
4465

UNCOV
4466
  if (pUser->superUser) {
×
UNCOV
4467
    mndReleaseUser(pMnode, pUser);
×
UNCOV
4468
    return TSDB_CODE_SUCCESS;
×
4469
  }
UNCOV
4470
  mndReleaseUser(pMnode, pUser);
×
4471

4472
  if (pTask->createdBy != NULL && pTask->createdByLen > 0 &&
×
4473
      strncmp(pTask->createdBy, user, TSDB_USER_LEN) == 0) {
×
4474
    return TSDB_CODE_SUCCESS;
×
4475
  }
4476

UNCOV
4477
  return TSDB_CODE_MND_XNODE_TASK_NO_PRIV;
×
4478
}
4479

UNCOV
4480
static int32_t mndCheckXnodeJobOwner(SMnode *pMnode, const char *user, SXnodeJobObj *pJob) {
×
UNCOV
4481
  SUserObj *pUser = NULL;
×
UNCOV
4482
  int32_t   code = mndAcquireUser(pMnode, user, &pUser);
×
UNCOV
4483
  if (code != 0) return code;
×
4484

UNCOV
4485
  if (pUser->superUser) {
×
UNCOV
4486
    mndReleaseUser(pMnode, pUser);
×
UNCOV
4487
    return TSDB_CODE_SUCCESS;
×
4488
  }
UNCOV
4489
  mndReleaseUser(pMnode, pUser);
×
4490

4491
  SXnodeTaskObj *pTask = mndAcquireXnodeTaskById(pMnode, pJob->taskId);
×
4492
  if (pTask == NULL) return TSDB_CODE_MND_XNODE_TASK_NOT_EXIST;
×
4493

UNCOV
4494
  code = TSDB_CODE_MND_XNODE_TASK_NO_PRIV;
×
UNCOV
4495
  if (pTask->createdBy != NULL && pTask->createdByLen > 0 &&
×
UNCOV
4496
      strncmp(pTask->createdBy, user, TSDB_USER_LEN) == 0) {
×
UNCOV
4497
    code = TSDB_CODE_SUCCESS;
×
4498
  }
UNCOV
4499
  mndReleaseXnodeTask(pMnode, pTask);
×
4500
  return code;
×
4501
}
4502

4503
static int32_t mndValidateXnodePermissions(SMnode *pMnode, SRpcMsg *pReq, EOperType oper) {
49,504✔
4504
  int32_t code = grantCheck(TSDB_GRANT_XNODE);
49,504✔
4505
  if (code != TSDB_CODE_SUCCESS) {
49,504✔
4506
    mError("failed to validate xnode permissions, code:%s, oper:%d", tstrerror(code), oper);
×
UNCOV
4507
    return code;
×
4508
  }
4509

4510
  // return mndCheckOperPrivilege(pMnode, pReq->info.conn.user, NULL, oper);
4511
  return TSDB_CODE_SUCCESS;
49,504✔
4512
}
4513

UNCOV
4514
SXnodeAgentObj *mndAcquireXnodeAgentById(SMnode *pMnode, int32_t id) {
×
UNCOV
4515
  SXnodeAgentObj *pObj = sdbAcquire(pMnode->pSdb, SDB_XNODE_AGENT, &id);
×
UNCOV
4516
  if (pObj == NULL && terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
×
UNCOV
4517
    terrno = TSDB_CODE_MND_XNODE_AGENT_NOT_EXIST;
×
4518
  }
UNCOV
4519
  return pObj;
×
4520
}
4521

4522
static SXnodeAgentObj *mndAcquireXnodeAgentByName(SMnode *pMnode, const char *name) {
4,998✔
4523
  SSdb *pSdb = pMnode->pSdb;
4,998✔
4524

4525
  void *pIter = NULL;
4,998✔
4526
  while (1) {
6,069✔
4527
    SXnodeAgentObj *pAgent = NULL;
11,067✔
4528
    pIter = sdbFetch(pSdb, SDB_XNODE_AGENT, pIter, (void **)&pAgent);
11,067✔
4529
    if (pIter == NULL) break;
11,067✔
4530
    if (pAgent->name == NULL) {
8,687✔
UNCOV
4531
      continue;
×
4532
    }
4533

4534
    if (strcasecmp(name, pAgent->name) == 0) {
8,687✔
4535
      sdbCancelFetch(pSdb, pIter);
2,618✔
4536
      return pAgent;
2,618✔
4537
    }
4538

4539
    sdbRelease(pSdb, pAgent);
6,069✔
4540
  }
4541

4542
  mDebug("xnode agent:%s, not found", name);
2,380✔
4543
  terrno = TSDB_CODE_MND_XNODE_AGENT_NOT_EXIST;
2,380✔
4544
  return NULL;
2,380✔
4545
}
4546

4547
static int32_t mndCheckXnodeAgentExists(SMnode *pMnode, const char *name) {
2,261✔
4548
  SXnodeAgentObj *pObj = mndAcquireXnodeAgentByName(pMnode, name);
2,261✔
4549
  if (pObj != NULL) {
2,261✔
4550
    mError("xnode agent:%s already exists", name);
×
UNCOV
4551
    mndReleaseXnodeAgent(pMnode, pObj);
×
UNCOV
4552
    return TSDB_CODE_MND_XNODE_AGENT_ALREADY_EXIST;
×
4553
  }
4554
  terrno = TSDB_CODE_SUCCESS;
2,261✔
4555
  return TSDB_CODE_SUCCESS;
2,261✔
4556
}
4557

4558
typedef struct {
4559
  int64_t sub;  // agent ID
4560
  int64_t iat;  // issued at time
4561
} agentTokenField;
4562

4563
const unsigned char MNDXNODE_DEFAULT_SECRET[] = {126, 222, 130, 137, 43,  122, 41,  173, 144, 146, 116,
4564
                                                 138, 153, 244, 251, 99,  50,  55,  140, 238, 218, 232,
4565
                                                 15,  161, 226, 54,  130, 40,  211, 234, 111, 171};
4566

4567
#ifndef WINDOWS
4568

UNCOV
4569
agentTokenField mndXnodeCreateAgentTokenField(long agent_id, time_t issued_at) {
×
UNCOV
4570
  agentTokenField field = {0};
×
UNCOV
4571
  field.sub = agent_id;
×
UNCOV
4572
  field.iat = issued_at;
×
UNCOV
4573
  return field;
×
4574
}
4575

4576
static char *mndXnodeBase64UrlEncodeOpenssl(const unsigned char *input, size_t input_len) {
7,947✔
4577
  __uint64_t code = 0;
7,947✔
4578
  int32_t lino = 0;
7,947✔
4579
  BIO     *bio = NULL, *b64 = NULL;
7,947✔
4580
  BUF_MEM *bufferPtr = NULL;
7,947✔
4581
  char    *base64_str = NULL;
7,947✔
4582

4583
  b64 = BIO_new(BIO_f_base64());
7,947✔
4584
  if (!b64) {
7,947✔
UNCOV
4585
    lino = __LINE__;
×
UNCOV
4586
    code = ERR_get_error();
×
UNCOV
4587
    goto _err;
×
4588
  }
4589

4590
  bio = BIO_new(BIO_s_mem());
7,947✔
4591
  if (!bio) {
7,947✔
UNCOV
4592
    lino = __LINE__;
×
UNCOV
4593
    code = ERR_get_error();
×
4594
    goto _err;
×
4595
  }
4596

4597
  // BIO chain:b64 → bio
4598
  bio = BIO_push(b64, bio);
7,947✔
4599
  if (!bio) {
7,947✔
UNCOV
4600
    lino = __LINE__;
×
UNCOV
4601
    code = ERR_get_error();
×
UNCOV
4602
    goto _err;
×
4603
  }
4604
  BIO_set_flags(bio, BIO_FLAGS_BASE64_NO_NL);
7,947✔
4605

4606
  int32_t write_ret = BIO_write(bio, input, input_len);
7,947✔
4607
  if (write_ret <= 0 || (size_t)write_ret != input_len) {
7,947✔
UNCOV
4608
    lino = __LINE__;
×
4609
    code = ERR_get_error();
×
4610
    goto _err;
×
4611
  }
4612
  int32_t flush_ret = BIO_flush(bio);
7,947✔
4613
  if (flush_ret != 1) {
7,947✔
4614
    lino = __LINE__;
×
4615
    code = ERR_get_error();
×
UNCOV
4616
    goto _err;
×
4617
  }
4618
  int64_t ret = BIO_get_mem_ptr(bio, &bufferPtr);
7,947✔
4619
  if (ret <= 0) {
7,947✔
4620
    lino = __LINE__;
×
4621
    code = ERR_get_error();
×
UNCOV
4622
    goto _err;
×
4623
}
4624
  if (!bufferPtr || !bufferPtr->data || bufferPtr->length == 0) {
7,947✔
UNCOV
4625
    lino = __LINE__;
×
4626
    code = ERR_get_error();
×
4627
    goto _err;
×
4628
  }
4629
  base64_str = taosMemoryMalloc(bufferPtr->length + 1);
7,947✔
4630
  if (!base64_str) {
7,947✔
UNCOV
4631
    lino = __LINE__;
×
UNCOV
4632
    code = ERR_get_error();
×
4633
    goto _err;
×
4634
  }
4635
  memcpy(base64_str, bufferPtr->data, bufferPtr->length);
7,947✔
4636
  base64_str[bufferPtr->length] = '\0';
7,947✔
4637
  // url safe
4638
  for (size_t i = 0; i < bufferPtr->length; i++) {
325,051✔
4639
    if (base64_str[i] == '+') {
317,104✔
4640
      base64_str[i] = '-';
820✔
4641
    } else if (base64_str[i] == '/') {
316,284✔
4642
      base64_str[i] = '_';
2,023✔
4643
    }
4644
  }
4645
  // remove padding char '='
4646
  size_t len = strlen(base64_str);
7,947✔
4647
  while (len > 0 && base64_str[len - 1] == '=') {
11,861✔
4648
    base64_str[len - 1] = '\0';
3,914✔
4649
    len--;
3,914✔
4650
  }
4651
  goto _exit;
7,947✔
4652

UNCOV
4653
_err:
×
UNCOV
4654
  if (code != TSDB_CODE_SUCCESS) {
×
4655
    mError("xnode agent: line: %d failed to encode base64 since %s", lino, ERR_error_string(code, NULL));
×
4656
  }
4657
  if (base64_str) {
×
UNCOV
4658
    taosMemoryFree(base64_str);
×
UNCOV
4659
    base64_str = NULL;
×
4660
  }
4661

4662
_exit:
6,783✔
4663
  if (bio) {
7,947✔
4664
    BIO_free_all(bio);  // will release b64 and bio
7,947✔
UNCOV
4665
  } else if (b64) {
×
UNCOV
4666
    int32_t ret = BIO_free(b64);
×
UNCOV
4667
    if (ret != 1) {
×
UNCOV
4668
      lino = __LINE__;
×
UNCOV
4669
      code = ERR_get_error();
×
UNCOV
4670
      mError("xnode agent: line: %d BIO failed to free since %s", lino, ERR_error_string(code, NULL));
×
4671
    }
4672
  }
4673

4674
  return base64_str;
7,947✔
4675
}
4676

4677
static char *mndXnodeCreateTokenHeader() {
2,649✔
4678
  int32_t code = 0, lino = 0;
2,649✔
4679
  cJSON  *headerJson = NULL;
2,649✔
4680
  char   *headerJsonStr = NULL;
2,649✔
4681
  char   *encoded = NULL;
2,649✔
4682

4683
  headerJson = tjsonCreateObject();
2,649✔
4684
  if (!headerJson) {
2,649✔
UNCOV
4685
    code = terrno;
×
UNCOV
4686
    goto _exit;
×
4687
  }
4688

4689
  TAOS_CHECK_EXIT(tjsonAddStringToObject(headerJson, "alg", "HS256"));
2,649✔
4690
  TAOS_CHECK_EXIT(tjsonAddStringToObject(headerJson, "typ", "JWT"));
2,649✔
4691

4692
  headerJsonStr = tjsonToUnformattedString(headerJson);
2,649✔
4693
  if (!headerJsonStr) {
2,649✔
UNCOV
4694
    code = terrno;
×
UNCOV
4695
    goto _exit;
×
4696
  }
4697
  encoded = mndXnodeBase64UrlEncodeOpenssl((const unsigned char *)headerJsonStr, strlen(headerJsonStr));
2,649✔
4698
  if (!encoded) {
2,649✔
UNCOV
4699
    code = terrno;
×
UNCOV
4700
    goto _exit;
×
4701
  }
4702

4703
_exit:
2,649✔
4704
  if (code != TSDB_CODE_SUCCESS) {
2,649✔
UNCOV
4705
    mError("xnode agent: line: %d failed to create header since %s", lino, tstrerror(code));
×
UNCOV
4706
    taosMemoryFree(encoded);
×
UNCOV
4707
    encoded = NULL;
×
4708
  }
4709

4710
  if (headerJsonStr) {
2,649✔
4711
    taosMemoryFree(headerJsonStr);
2,649✔
4712
  }
4713
  if (headerJson) {
2,649✔
4714
    tjsonDelete(headerJson);
2,649✔
4715
  }
4716

4717
  return encoded;
2,649✔
4718
}
4719

4720
static char *mndXnodeCreateTokenPayload(const agentTokenField *claims) {
2,649✔
4721
  int32_t code = 0, lino = 0;
2,649✔
4722
  cJSON  *payloadJson = NULL;
2,649✔
4723
  char   *payloadStr = NULL;
2,649✔
4724
  char   *encoded = NULL;
2,649✔
4725

4726
  if (!claims) {
2,649✔
UNCOV
4727
    code = TSDB_CODE_INVALID_PARA;
×
UNCOV
4728
    terrno = code;
×
UNCOV
4729
    return NULL;
×
4730
  }
4731

4732
  payloadJson = tjsonCreateObject();
2,649✔
4733
  if (!payloadJson) {
2,649✔
UNCOV
4734
    code = terrno;
×
UNCOV
4735
    goto _exit;
×
4736
  }
4737

4738
  TAOS_CHECK_EXIT(tjsonAddDoubleToObject(payloadJson, "iat", claims->iat));
2,649✔
4739
  TAOS_CHECK_EXIT(tjsonAddDoubleToObject(payloadJson, "sub", claims->sub));
2,649✔
4740

4741
  payloadStr = tjsonToUnformattedString(payloadJson);
2,649✔
4742
  if (!payloadStr) {
2,649✔
UNCOV
4743
    code = terrno;
×
UNCOV
4744
    goto _exit;
×
4745
  }
4746
  encoded = mndXnodeBase64UrlEncodeOpenssl((const unsigned char *)payloadStr, strlen(payloadStr));
2,649✔
4747
  if (!encoded) {
2,649✔
UNCOV
4748
    code = terrno;
×
UNCOV
4749
    goto _exit;
×
4750
  }
4751

4752
_exit:
2,649✔
4753
  if (code != TSDB_CODE_SUCCESS) {
2,649✔
UNCOV
4754
    mError("xnode agent line: %d failed to create payload since %s", lino, tstrerror(code));
×
UNCOV
4755
    taosMemoryFree(encoded);
×
UNCOV
4756
    encoded = NULL;
×
4757
  }
4758
  if (payloadStr) {
2,649✔
4759
    taosMemoryFree(payloadStr);
2,649✔
4760
  }
4761
  if (payloadJson) {
2,649✔
4762
    tjsonDelete(payloadJson);
2,649✔
4763
  }
4764
  return encoded;
2,649✔
4765
}
4766

4767
static char *mndXnodeCreateTokenSignature(const char *header_payload, const unsigned char *secret, size_t secret_len) {
2,649✔
4768
  int32_t       code = 0, lino = 0;
2,649✔
4769
  unsigned char hash[EVP_MAX_MD_SIZE] = {0};
2,649✔
4770
  unsigned int  hash_len = 0;
2,649✔
4771
  char         *encoded = NULL;
2,649✔
4772

4773
  // HMAC-SHA256
4774
  if (!HMAC(EVP_sha256(), secret, secret_len, (const unsigned char *)header_payload, strlen(header_payload), hash,
2,649✔
4775
            &hash_len)) {
UNCOV
4776
    code = terrno;
×
UNCOV
4777
    goto _exit;
×
4778
  }
4779

4780
  encoded = mndXnodeBase64UrlEncodeOpenssl(hash, hash_len);
2,649✔
4781
  if (!encoded) {
2,649✔
UNCOV
4782
    code = terrno;
×
UNCOV
4783
    goto _exit;
×
4784
  }
4785

4786
_exit:
2,649✔
4787
  if (code != TSDB_CODE_SUCCESS) {
2,649✔
UNCOV
4788
    mError("xnode agent line: %d failed create signature since %s", lino, tstrerror(code));
×
UNCOV
4789
    taosMemoryFree(encoded);
×
UNCOV
4790
    encoded = NULL;
×
4791
  }
4792
  return encoded;
2,649✔
4793
}
4794

4795
char *mndXnodeCreateAgentToken(const agentTokenField *claims, const unsigned char *secret, size_t secret_len) {
2,649✔
4796
  int32_t code = 0, lino = 0;
2,649✔
4797
  char   *header = NULL, *payload = NULL;
2,649✔
4798
  char   *headerPayload = NULL;
2,649✔
4799
  char   *signature = NULL;
2,649✔
4800
  char   *token = NULL;
2,649✔
4801

4802
  if (!claims) {
2,649✔
UNCOV
4803
    code = TSDB_CODE_INVALID_PARA;
×
UNCOV
4804
    goto _exit;
×
4805
  }
4806

4807
  if (!secret || secret_len == 0) {
2,649✔
UNCOV
4808
    secret = MNDXNODE_DEFAULT_SECRET;
×
UNCOV
4809
    secret_len = sizeof(MNDXNODE_DEFAULT_SECRET);
×
4810
  }
4811

4812
  header = mndXnodeCreateTokenHeader();
2,649✔
4813
  if (!header) {
2,649✔
UNCOV
4814
    code = terrno;
×
UNCOV
4815
    goto _exit;
×
4816
  }
4817

4818
  payload = mndXnodeCreateTokenPayload(claims);
2,649✔
4819
  if (!payload) {
2,649✔
UNCOV
4820
    code = terrno;
×
UNCOV
4821
    goto _exit;
×
4822
  }
4823

4824
  size_t header_payload_len = strlen(header) + strlen(payload) + 2;
2,649✔
4825
  headerPayload = taosMemoryMalloc(header_payload_len);
2,649✔
4826
  if (!headerPayload) {
2,649✔
UNCOV
4827
    code = terrno;
×
UNCOV
4828
    goto _exit;
×
4829
  }
4830
  snprintf(headerPayload, header_payload_len, "%s.%s", header, payload);
2,649✔
4831

4832
  signature = mndXnodeCreateTokenSignature(headerPayload, secret, secret_len);
2,649✔
4833
  if (!signature) {
2,649✔
UNCOV
4834
    code = terrno;
×
UNCOV
4835
    goto _exit;
×
4836
  }
4837

4838
  size_t token_len = strlen(headerPayload) + strlen(signature) + 2;
2,649✔
4839
  token = taosMemoryCalloc(1, token_len);
2,649✔
4840
  if (!token) {
2,649✔
UNCOV
4841
    code = terrno;
×
UNCOV
4842
    goto _exit;
×
4843
  }
4844

4845
  snprintf(token, token_len, "%s.%s", headerPayload, signature);
2,649✔
4846

4847
_exit:
2,649✔
4848
  if (code != TSDB_CODE_SUCCESS) {
2,649✔
UNCOV
4849
    mError("xnode agent line: %d failed create token since %s", lino, tstrerror(code));
×
UNCOV
4850
    taosMemoryFree(token);
×
UNCOV
4851
    token = NULL;
×
4852
  }
4853
  taosMemoryFree(signature);
2,649✔
4854
  taosMemoryFree(headerPayload);
2,649✔
4855
  taosMemoryFree(payload);
2,649✔
4856
  taosMemoryFree(header);
2,649✔
4857

4858
  return token;
2,649✔
4859
}
4860
#else
4861

4862
static int32_t mndXnodeHmacSha256Windows(const unsigned char *key, size_t key_len, const unsigned char *message,
4863
                                         size_t message_len, unsigned char *output) {
4864
  unsigned char k_pad[64];
4865
  unsigned char k_ipad[64];
4866
  unsigned char k_opad[64];
4867
  unsigned char inner_hash[SHA256_DIGEST_SIZE];
4868
  sha256_ctx    ctx;
4869

4870
  if (key_len > 64) {
4871
    sha256_init(&ctx);
4872
    sha256_update(&ctx, key, key_len);
4873
    sha256_final(&ctx, k_pad);
4874
    memset(k_pad + SHA256_DIGEST_SIZE, 0, 64 - SHA256_DIGEST_SIZE);
4875
  } else {
4876
    memcpy(k_pad, key, key_len);
4877
    memset(k_pad + key_len, 0, 64 - key_len);
4878
  }
4879

4880
  for (int i = 0; i < 64; i++) {
4881
    k_ipad[i] = k_pad[i] ^ 0x36;
4882
    k_opad[i] = k_pad[i] ^ 0x5c;
4883
  }
4884

4885
  sha256_init(&ctx);
4886
  sha256_update(&ctx, k_ipad, 64);
4887
  sha256_update(&ctx, message, message_len);
4888
  sha256_final(&ctx, inner_hash);
4889

4890
  sha256_init(&ctx);
4891
  sha256_update(&ctx, k_opad, 64);
4892
  sha256_update(&ctx, inner_hash, SHA256_DIGEST_SIZE);
4893
  sha256_final(&ctx, output);
4894

4895
  return 0;
4896
}
4897

4898
static char *mndXnodeBase64UrlEncodeWindows(const unsigned char *input, size_t input_len) {
4899
  char *base64_str = NULL;
4900

4901
  int32_t ret = base64_encode(input, input_len, &base64_str);
4902
  if (ret != 0 || !base64_str) {
4903
    return NULL;
4904
  }
4905

4906
  for (size_t i = 0; base64_str[i] != '\0'; i++) {
4907
    if (base64_str[i] == '+') {
4908
      base64_str[i] = '-';
4909
    } else if (base64_str[i] == '/') {
4910
      base64_str[i] = '_';
4911
    }
4912
  }
4913

4914
  size_t len = strlen(base64_str);
4915
  while (len > 0 && base64_str[len - 1] == '=') {
4916
    base64_str[len - 1] = '\0';
4917
    len--;
4918
  }
4919

4920
  return base64_str;
4921
}
4922

4923
static char *mndXnodeCreateTokenHeaderWindows() {
4924
  int32_t code = 0, lino = 0;
4925
  cJSON  *headerJson = NULL;
4926
  char   *headerJsonStr = NULL;
4927
  char   *encoded = NULL;
4928

4929
  headerJson = tjsonCreateObject();
4930
  if (!headerJson) {
4931
    code = terrno;
4932
    goto _exit;
4933
  }
4934

4935
  TAOS_CHECK_EXIT(tjsonAddStringToObject(headerJson, "alg", "HS256"));
4936
  TAOS_CHECK_EXIT(tjsonAddStringToObject(headerJson, "typ", "JWT"));
4937

4938
  headerJsonStr = tjsonToUnformattedString(headerJson);
4939
  if (!headerJsonStr) {
4940
    code = terrno;
4941
    goto _exit;
4942
  }
4943
  encoded = mndXnodeBase64UrlEncodeWindows((const unsigned char *)headerJsonStr, strlen(headerJsonStr));
4944
  if (!encoded) {
4945
    code = terrno;
4946
    goto _exit;
4947
  }
4948

4949
_exit:
4950
  if (code != TSDB_CODE_SUCCESS) {
4951
    mError("xnode agent: line: %d failed to create header since %s", lino, tstrerror(code));
4952
    taosMemoryFree(encoded);
4953
    encoded = NULL;
4954
  }
4955

4956
  if (headerJsonStr) {
4957
    taosMemoryFree(headerJsonStr);
4958
  }
4959
  if (headerJson) {
4960
    tjsonDelete(headerJson);
4961
  }
4962

4963
  return encoded;
4964
}
4965

4966
static char *mndXnodeCreateTokenPayloadWindows(const agentTokenField *claims) {
4967
  int32_t code = 0, lino = 0;
4968
  cJSON  *payloadJson = NULL;
4969
  char   *payloadStr = NULL;
4970
  char   *encoded = NULL;
4971

4972
  if (!claims) {
4973
    code = TSDB_CODE_INVALID_PARA;
4974
    terrno = code;
4975
    return NULL;
4976
  }
4977

4978
  payloadJson = tjsonCreateObject();
4979
  if (!payloadJson) {
4980
    code = terrno;
4981
    goto _exit;
4982
  }
4983

4984
  TAOS_CHECK_EXIT(tjsonAddDoubleToObject(payloadJson, "iat", claims->iat));
4985
  TAOS_CHECK_EXIT(tjsonAddDoubleToObject(payloadJson, "sub", claims->sub));
4986

4987
  payloadStr = tjsonToUnformattedString(payloadJson);
4988
  if (!payloadStr) {
4989
    code = terrno;
4990
    goto _exit;
4991
  }
4992
  encoded = mndXnodeBase64UrlEncodeWindows((const unsigned char *)payloadStr, strlen(payloadStr));
4993
  if (!encoded) {
4994
    code = terrno;
4995
    goto _exit;
4996
  }
4997

4998
_exit:
4999
  if (code != TSDB_CODE_SUCCESS) {
5000
    mError("xnode agent line: %d failed to create payload since %s", lino, tstrerror(code));
5001
    taosMemoryFree(encoded);
5002
    encoded = NULL;
5003
  }
5004
  if (payloadStr) {
5005
    taosMemoryFree(payloadStr);
5006
  }
5007
  if (payloadJson) {
5008
    tjsonDelete(payloadJson);
5009
  }
5010
  return encoded;
5011
}
5012

5013
static char *mndXnodeCreateTokenSignatureWindows(const char *header_payload, const unsigned char *secret,
5014
                                                  size_t secret_len) {
5015
  int32_t       code = 0, lino = 0;
5016
  unsigned char hash[SHA256_DIGEST_SIZE] = {0};
5017
  char         *encoded = NULL;
5018

5019
  int32_t ret = mndXnodeHmacSha256Windows(secret, secret_len, (const unsigned char *)header_payload,
5020
                                          strlen(header_payload), hash);
5021
  if (ret != 0) {
5022
    code = terrno;
5023
    goto _exit;
5024
  }
5025

5026
  encoded = mndXnodeBase64UrlEncodeWindows(hash, SHA256_DIGEST_SIZE);
5027
  if (!encoded) {
5028
    code = terrno;
5029
    goto _exit;
5030
  }
5031

5032
_exit:
5033
  if (code != TSDB_CODE_SUCCESS) {
5034
    mError("xnode agent line: %d failed create signature since %s", lino, tstrerror(code));
5035
    taosMemoryFree(encoded);
5036
    encoded = NULL;
5037
  }
5038
  return encoded;
5039
}
5040

5041
char *mndXnodeCreateAgentToken(const agentTokenField *claims, const unsigned char *secret, size_t secret_len) {
5042
  int32_t code = 0, lino = 0;
5043
  char   *header = NULL, *payload = NULL;
5044
  char   *headerPayload = NULL;
5045
  char   *signature = NULL;
5046
  char   *token = NULL;
5047

5048
  if (!claims) {
5049
    code = TSDB_CODE_INVALID_PARA;
5050
    goto _exit;
5051
  }
5052

5053
  if (!secret || secret_len == 0) {
5054
    secret = MNDXNODE_DEFAULT_SECRET;
5055
    secret_len = sizeof(MNDXNODE_DEFAULT_SECRET);
5056
  }
5057

5058
  header = mndXnodeCreateTokenHeaderWindows();
5059
  if (!header) {
5060
    code = terrno;
5061
    goto _exit;
5062
  }
5063

5064
  payload = mndXnodeCreateTokenPayloadWindows(claims);
5065
  if (!payload) {
5066
    code = terrno;
5067
    goto _exit;
5068
  }
5069

5070
  size_t header_payload_len = strlen(header) + strlen(payload) + 2;
5071
  headerPayload = taosMemoryMalloc(header_payload_len);
5072
  if (!headerPayload) {
5073
    code = terrno;
5074
    goto _exit;
5075
  }
5076
  snprintf(headerPayload, header_payload_len, "%s.%s", header, payload);
5077

5078
  signature = mndXnodeCreateTokenSignatureWindows(headerPayload, secret, secret_len);
5079
  if (!signature) {
5080
    code = terrno;
5081
    goto _exit;
5082
  }
5083

5084
  size_t token_len = strlen(headerPayload) + strlen(signature) + 2;
5085
  token = taosMemoryCalloc(1, token_len);
5086
  if (!token) {
5087
    code = terrno;
5088
    goto _exit;
5089
  }
5090

5091
  snprintf(token, token_len, "%s.%s", headerPayload, signature);
5092

5093
_exit:
5094
  if (code != TSDB_CODE_SUCCESS) {
5095
    mError("xnode agent line: %d failed create token since %s", lino, tstrerror(code));
5096
    taosMemoryFree(token);
5097
    token = NULL;
5098
  }
5099
  taosMemoryFree(signature);
5100
  taosMemoryFree(headerPayload);
5101
  taosMemoryFree(payload);
5102
  taosMemoryFree(header);
5103

5104
  return token;
5105
}
5106

5107
#endif
5108

5109
int32_t mndXnodeGenAgentToken(const SXnodeAgentObj *pAgent, char *pTokenBuf) {
2,261✔
5110
  int32_t code = 0, lino = 0;
2,261✔
5111

5112
  agentTokenField claims = {
2,261✔
5113
      .iat = pAgent->createTime,
2,261✔
5114
      .sub = pAgent->id,
2,261✔
5115
  };
5116
  // token be like:
5117
  // "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpYXQiOjE3Njc1OTc3NzIsInN1YiI6MTIzNDV9.i7HvYf_S-yWGEExDzQESPUwVX23Ok_7Fxo93aqgKrtw"
5118
  char *token = mndXnodeCreateAgentToken(&claims, MNDXNODE_DEFAULT_SECRET, sizeof(MNDXNODE_DEFAULT_SECRET));
2,261✔
5119
  if (!token) {
2,261✔
UNCOV
5120
    code = terrno;
×
UNCOV
5121
    lino = __LINE__;
×
UNCOV
5122
    goto _exit;
×
5123
  }
5124
  (void)memcpy(pTokenBuf, token, TMIN(strlen(token) + 1, TSDB_XNODE_AGENT_TOKEN_LEN));
2,261✔
5125

5126
_exit:
2,261✔
5127
  if (code != TSDB_CODE_SUCCESS) {
2,261✔
UNCOV
5128
    mError("xnode agent line: %d failed gen token since %s", lino, tstrerror(code));
×
5129
  }
5130
  taosMemoryFree(token);
2,261✔
5131
  TAOS_RETURN(code);
2,261✔
5132
}
5133

5134
static int32_t mndCreateXnodeAgent(SMnode *pMnode, SRpcMsg *pReq, SMCreateXnodeAgentReq *pCreate,
2,261✔
5135
                                   SXnodeAgentObj **ppObj) {
5136
  int32_t code = -1;
2,261✔
5137
  STrans *pTrans = NULL;
2,261✔
5138

5139
  if ((*ppObj) == NULL) {
2,261✔
5140
    *ppObj = taosMemoryCalloc(1, sizeof(SXnodeAgentObj));
2,261✔
5141
    if (*ppObj == NULL) {
2,261✔
5142
      code = terrno;
×
UNCOV
5143
      goto _OVER;
×
5144
    }
5145
  }
5146
  SXnodeAgentObj *pAgentObj = *ppObj;
2,261✔
5147

5148
  pAgentObj->id = sdbGetMaxId(pMnode->pSdb, SDB_XNODE_AGENT);
2,261✔
5149
  pAgentObj->createTime = taosGetTimestampMs();
2,261✔
5150
  pAgentObj->updateTime = pAgentObj->createTime;
2,261✔
5151

5152
  pAgentObj->nameLen = pCreate->name.len + 1;
2,261✔
5153
  pAgentObj->name = taosMemoryCalloc(1, pAgentObj->nameLen);
2,261✔
5154
  if (pAgentObj->name == NULL) goto _OVER;
2,261✔
5155
  (void)memcpy(pAgentObj->name, pCreate->name.ptr, pCreate->name.len);
2,261✔
5156

5157
  if (pCreate->status.ptr != NULL) {
2,261✔
5158
    pAgentObj->statusLen = pCreate->status.len + 1;
1,904✔
5159
    pAgentObj->status = taosMemoryCalloc(1, pAgentObj->statusLen);
1,904✔
5160
    if (pAgentObj->status == NULL) goto _OVER;
1,904✔
5161
    (void)memcpy(pAgentObj->status, pCreate->status.ptr, pCreate->status.len);
1,904✔
5162
  }
5163
  // gen token
5164
  char token[TSDB_XNODE_AGENT_TOKEN_LEN] = {0};
2,261✔
5165
  TAOS_CHECK_GOTO(mndXnodeGenAgentToken(pAgentObj, token), NULL, _OVER);
2,261✔
5166
  pAgentObj->tokenLen = strlen(token) + 1;
2,261✔
5167
  pAgentObj->token = taosMemoryCalloc(1, pAgentObj->tokenLen);
2,261✔
5168
  if (pAgentObj->token == NULL) goto _OVER;
2,261✔
5169
  (void)memcpy(pAgentObj->token, token, pAgentObj->tokenLen - 1);
2,261✔
5170

5171
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-xnode-agent");
2,261✔
5172
  if (pTrans == NULL) {
2,261✔
UNCOV
5173
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
5174
    if (terrno != 0) {
×
UNCOV
5175
      code = terrno;
×
5176
    }
UNCOV
5177
    mError("failed to create transaction for xnode-agent:%s, code:0x%x:%s", pCreate->name.ptr, code, tstrerror(code));
×
UNCOV
5178
    goto _OVER;
×
5179
  }
5180
  mndTransSetSerial(pTrans);
2,261✔
5181

5182
  mDebug("trans:%d, used to create xnode agent:%s as agent:%d", pTrans->id, pCreate->name.ptr, pAgentObj->id);
2,261✔
5183
  TAOS_CHECK_GOTO(mndSetCreateXnodeAgentRedoLogs(pTrans, pAgentObj), NULL, _OVER);
2,261✔
5184
  TAOS_CHECK_GOTO(mndSetCreateXnodeAgentUndoLogs(pTrans, pAgentObj), NULL, _OVER);
2,261✔
5185
  TAOS_CHECK_GOTO(mndSetCreateXnodeAgentCommitLogs(pTrans, pAgentObj), NULL, _OVER);
2,261✔
5186
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
2,261✔
5187

5188
  code = 0;
2,261✔
5189

5190
_OVER:
2,261✔
5191
  mndTransDrop(pTrans);
2,261✔
5192
  TAOS_RETURN(code);
2,261✔
5193
}
5194

5195
static int32_t httpCreateAgent(SXnodeAgentObj *pObj) {
2,261✔
5196
  int32_t code = 0;
2,261✔
5197
  SJson  *pJson = NULL;
2,261✔
5198
  SJson  *postContent = NULL;
2,261✔
5199
  char   *pContStr = NULL;
2,261✔
5200

5201
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
2,261✔
5202
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/agent", XNODED_PIPE_SOCKET_URL);
2,261✔
5203
  postContent = tjsonCreateObject();
2,261✔
5204
  if (postContent == NULL) {
2,261✔
UNCOV
5205
    code = terrno;
×
UNCOV
5206
    goto _OVER;
×
5207
  }
5208
  TAOS_CHECK_GOTO(tjsonAddStringToObject(postContent, "token", pObj->token), NULL, _OVER);
2,261✔
5209
  pContStr = tjsonToUnformattedString(postContent);
2,261✔
5210
  if (pContStr == NULL) {
2,261✔
5211
    code = terrno;
×
UNCOV
5212
    goto _OVER;
×
5213
  }
5214
  pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_POST, defaultTimeout, pContStr, strlen(pContStr));
2,261✔
5215

5216
_OVER:
2,261✔
5217
  if (postContent != NULL) {
2,261✔
5218
    tjsonDelete(postContent);
2,261✔
5219
  }
5220
  if (pContStr != NULL) {
2,261✔
5221
    taosMemFree(pContStr);
2,261✔
5222
  }
5223
  if (pJson != NULL) {
2,261✔
UNCOV
5224
    tjsonDelete(pJson);
×
5225
  }
5226
  TAOS_RETURN(code);
2,261✔
5227
}
5228

5229
static int32_t mndProcessCreateXnodeAgentReq(SRpcMsg *pReq) {
2,261✔
5230
  SMnode               *pMnode = pReq->info.node;
2,261✔
5231
  int32_t               code = 0;
2,261✔
5232
  SXnodeAgentObj       *pObj = NULL;
2,261✔
5233
  SMCreateXnodeAgentReq createReq = {0};
2,261✔
5234

5235
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_CREATE_XNODE_AGENT);
2,261✔
5236
  if (code != TSDB_CODE_SUCCESS) {
2,261✔
UNCOV
5237
    mError("failed check permission for create xnode agent, code:%s", tstrerror(code));
×
UNCOV
5238
    goto _OVER;
×
5239
  }
5240

5241
  code = tDeserializeSMCreateXnodeAgentReq(pReq->pCont, pReq->contLen, &createReq);
2,261✔
5242
  if (code != 0) {
2,261✔
5243
    mError("failed to deserialize create xnode agent request, code:%s", tstrerror(code));
×
UNCOV
5244
    TAOS_RETURN(code);
×
5245
  }
5246

5247
  TAOS_CHECK_GOTO(mndCheckXnodeAgentExists(pMnode, createReq.name.ptr), NULL, _OVER);
2,261✔
5248

5249
  TAOS_CHECK_GOTO(mndCreateXnodeAgent(pMnode, pReq, &createReq, &pObj), NULL, _OVER);
2,261✔
5250
  code = TSDB_CODE_ACTION_IN_PROGRESS;
2,261✔
5251

5252
  (void)httpCreateAgent(pObj);
2,261✔
5253

5254
_OVER:
2,261✔
5255
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
2,261✔
UNCOV
5256
    mError("xnode agent:%s, failed to create since %s", createReq.name.ptr ? createReq.name.ptr : "unknown",
×
5257
           tstrerror(code));
5258
  }
5259
  if (pObj != NULL) {
2,261✔
5260
    mndFreeXnodeAgent(pObj);
2,261✔
5261
    taosMemoryFree(pObj);
2,261✔
5262
  }
5263
  tFreeSMCreateXnodeAgentReq(&createReq);
2,261✔
5264
  TAOS_RETURN(code);
2,261✔
5265
}
5266

5267
static int32_t mndUpdateXnodeAgent(SMnode *pMnode, SRpcMsg *pReq, const SXnodeAgentObj *pOld,
357✔
5268
                                   SMUpdateXnodeAgentReq *pUpdate) {
5269
  mDebug("xnode agent:%d, start to update", pUpdate->id);
357✔
5270
  int32_t        code = -1;
357✔
5271
  STrans        *pTrans = NULL;
357✔
5272
  struct {
5273
    bool status;
5274
    bool name;
5275
  } isChange = {0};
357✔
5276
  SXnodeAgentObj agentObjRef = *pOld;
357✔
5277

5278
  const char *status = getXTaskOptionByName(&pUpdate->options, "status");
357✔
5279
  if (status != NULL) {
357✔
5280
    isChange.status = true;
357✔
5281
    agentObjRef.statusLen = strlen(status) + 1;
357✔
5282
    agentObjRef.status = taosMemoryCalloc(1, agentObjRef.statusLen);
357✔
5283
    if (agentObjRef.status == NULL) goto _OVER;
357✔
5284
    (void)memcpy(agentObjRef.status, status, agentObjRef.statusLen);
357✔
5285
  }
5286
  const char *name = getXTaskOptionByName(&pUpdate->options, "name");
357✔
5287
  if (name != NULL) {
357✔
UNCOV
5288
    isChange.name = true;
×
UNCOV
5289
    agentObjRef.nameLen = strlen(name) + 1;
×
UNCOV
5290
    agentObjRef.name = taosMemoryCalloc(1, agentObjRef.nameLen);
×
UNCOV
5291
    if (agentObjRef.name == NULL) goto _OVER;
×
UNCOV
5292
    (void)memcpy(agentObjRef.name, name, agentObjRef.nameLen);
×
5293
  }
5294
  agentObjRef.updateTime = taosGetTimestampMs();
357✔
5295

5296
  pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "update-xnode-agent");
357✔
5297
  if (pTrans == NULL) {
357✔
UNCOV
5298
    code = TSDB_CODE_MND_RETURN_VALUE_NULL;
×
UNCOV
5299
    if (terrno != 0) code = terrno;
×
UNCOV
5300
    goto _OVER;
×
5301
  }
5302
  mInfo("trans:%d, used to update xnode agent:%d", pTrans->id, agentObjRef.id);
357✔
5303

5304
  TAOS_CHECK_GOTO(mndSetCreateXnodeAgentCommitLogs(pTrans, &agentObjRef), NULL, _OVER);
357✔
5305
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
357✔
5306
  code = 0;
357✔
5307

5308
_OVER:
357✔
5309
  if (isChange.status) {
357✔
5310
    taosMemoryFree(agentObjRef.status);
357✔
5311
  }
5312
  if (isChange.name) {
357✔
UNCOV
5313
    taosMemoryFree(agentObjRef.name);
×
5314
  }
5315
  mndTransDrop(pTrans);
357✔
5316
  TAOS_RETURN(code);
357✔
5317
}
5318

5319
static int32_t mndProcessUpdateXnodeAgentReq(SRpcMsg *pReq) {
357✔
5320
  SMnode               *pMnode = pReq->info.node;
357✔
5321
  int32_t               code = -1;
357✔
5322
  SXnodeAgentObj       *pObj = NULL;
357✔
5323
  SMUpdateXnodeAgentReq updateReq = {0};
357✔
5324

5325
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_UPDATE_XNODE_AGENT);
357✔
5326
  if (code != TSDB_CODE_SUCCESS) {
357✔
UNCOV
5327
    mError("failed check permission for update xnode agent, code:%s", tstrerror(code));
×
UNCOV
5328
    goto _OVER;
×
5329
  }
5330

5331
  TAOS_CHECK_GOTO(tDeserializeSMUpdateXnodeAgentReq(pReq->pCont, pReq->contLen, &updateReq), NULL, _OVER);
357✔
5332
  mDebug("xnode update agent request id:%d, nameLen:%d\n", updateReq.id, updateReq.name.len);
357✔
5333

5334
  if (updateReq.id <= 0 && (updateReq.name.len <= 0 || updateReq.name.ptr == NULL)) {
357✔
UNCOV
5335
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
5336
    goto _OVER;
×
5337
  }
5338

5339
  if (updateReq.id > 0) {
357✔
UNCOV
5340
    pObj = mndAcquireXnodeAgentById(pMnode, updateReq.id);
×
5341
  } else {
5342
    pObj = mndAcquireXnodeAgentByName(pMnode, updateReq.name.ptr);
357✔
5343
  }
5344
  if (pObj == NULL) {
357✔
5345
    code = terrno;
×
UNCOV
5346
    goto _OVER;
×
5347
  }
5348
  const char *nameRef = getXTaskOptionByName(&updateReq.options, "name");
357✔
5349
  if (nameRef != NULL) {
357✔
UNCOV
5350
    SXnodeAgentObj* tmpObj = mndAcquireXnodeAgentByName(pMnode, nameRef);
×
UNCOV
5351
    if (tmpObj != NULL) {
×
UNCOV
5352
      mndReleaseXnodeAgent(pMnode, tmpObj);
×
UNCOV
5353
      code = TSDB_CODE_MND_XNODE_NAME_DUPLICATE;
×
5354
      goto _OVER;
×
5355
    }
5356
  }
5357

5358
  code = mndUpdateXnodeAgent(pMnode, pReq, pObj, &updateReq);
357✔
5359
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
357✔
5360

5361
_OVER:
357✔
5362
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
357✔
5363
    mError("xnode agent:%d, failed to update since %s", updateReq.id, tstrerror(code));
×
5364
  }
5365

5366
  mndReleaseXnodeAgent(pMnode, pObj);
357✔
5367
  tFreeSMUpdateXnodeAgentReq(&updateReq);
357✔
5368
  TAOS_RETURN(code);
357✔
5369
}
5370

5371
static int32_t mndSetDropXnodeAgentRedoLogs(STrans *pTrans, SXnodeAgentObj *pObj) {
2,261✔
5372
  int32_t  code = 0;
2,261✔
5373
  SSdbRaw *pRedoRaw = mndXnodeAgentActionEncode(pObj);
2,261✔
5374
  if (pRedoRaw == NULL) {
2,261✔
UNCOV
5375
    code = terrno;
×
UNCOV
5376
    return code;
×
5377
  }
5378

5379
  TAOS_CHECK_RETURN(mndTransAppendRedolog(pTrans, pRedoRaw));
2,261✔
5380
  TAOS_CHECK_RETURN(sdbSetRawStatus(pRedoRaw, SDB_STATUS_DROPPING));
2,261✔
5381

5382
  TAOS_RETURN(code);
2,261✔
5383
}
5384

5385
static int32_t mndSetDropXnodeAgentCommitLogs(STrans *pTrans, SXnodeAgentObj *pObj) {
2,261✔
5386
  int32_t  code = 0;
2,261✔
5387
  SSdbRaw *pCommitRaw = mndXnodeAgentActionEncode(pObj);
2,261✔
5388
  if (pCommitRaw == NULL) {
2,261✔
UNCOV
5389
    code = terrno;
×
5390
    return code;
×
5391
  }
5392

5393
  TAOS_CHECK_RETURN(mndTransAppendCommitlog(pTrans, pCommitRaw));
2,261✔
5394
  TAOS_CHECK_RETURN(sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED));
2,261✔
5395
  TAOS_RETURN(code);
2,261✔
5396
}
5397

5398
static int32_t mndDropXnodeAgent(SMnode *pMnode, SRpcMsg *pReq, SXnodeAgentObj *pAgent) {
2,261✔
5399
  int32_t code = 0;
2,261✔
5400
  int32_t lino = 0;
2,261✔
5401

5402
  if (pAgent == NULL) {
2,261✔
UNCOV
5403
    mError("xnode agent fail to drop since pAgent is NULL");
×
UNCOV
5404
    code = TSDB_CODE_INVALID_PARA;
×
UNCOV
5405
    return code;
×
5406
  }
5407
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_NOTHING, pReq, "drop-xnode-agent");
2,261✔
5408
  TSDB_CHECK_NULL(pTrans, code, lino, _OVER, terrno);
2,261✔
5409
  mndTransSetSerial(pTrans);
2,261✔
5410
  mDebug("trans:%d, to drop xnode agent:%d", pTrans->id, pAgent->id);
2,261✔
5411

5412
  TAOS_CHECK_GOTO(mndSetDropXnodeAgentRedoLogs(pTrans, pAgent), NULL, _OVER);
2,261✔
5413
  TAOS_CHECK_GOTO(mndSetDropXnodeAgentCommitLogs(pTrans, pAgent), NULL, _OVER);
2,261✔
5414
  TAOS_CHECK_GOTO(mndTransPrepare(pMnode, pTrans), NULL, _OVER);
2,261✔
5415

5416
_OVER:
2,261✔
5417
  mndTransDrop(pTrans);
2,261✔
5418
  TAOS_RETURN(code);
2,261✔
5419
}
5420

5421
static int32_t mndProcessDropXnodeAgentReq(SRpcMsg *pReq) {
2,380✔
5422
  SMnode             *pMnode = pReq->info.node;
2,380✔
5423
  int32_t             code = -1;
2,380✔
5424
  SXnodeAgentObj     *pObj = NULL;
2,380✔
5425
  SMDropXnodeAgentReq dropReq = {0};
2,380✔
5426

5427
  code = mndValidateXnodePermissions(pMnode, pReq, MND_OPER_DROP_XNODE_AGENT);
2,380✔
5428
  if (code != TSDB_CODE_SUCCESS) {
2,380✔
UNCOV
5429
    mError("failed check permission for drop xnode agent, code:%s", tstrerror(code));
×
UNCOV
5430
    goto _OVER;
×
5431
  }
5432
  TAOS_CHECK_GOTO(tDeserializeSMDropXnodeAgentReq(pReq->pCont, pReq->contLen, &dropReq), NULL, _OVER);
2,380✔
5433
  mDebug("xnode drop agent with id:%d, start to drop", dropReq.id);
2,380✔
5434

5435
  if (dropReq.id <= 0 && (dropReq.name.len <= 0 || dropReq.name.ptr == NULL)) {
2,380✔
UNCOV
5436
    code = TSDB_CODE_MND_XNODE_INVALID_MSG;
×
UNCOV
5437
    goto _OVER;
×
5438
  }
5439

5440
  if (dropReq.name.len > 0 && dropReq.name.ptr != NULL) {
2,380✔
5441
    pObj = mndAcquireXnodeAgentByName(pMnode, dropReq.name.ptr);
2,380✔
5442
  } else {
UNCOV
5443
    pObj = mndAcquireXnodeAgentById(pMnode, dropReq.id);
×
5444
  }
5445
  if (pObj == NULL) {
2,380✔
5446
    code = terrno;
119✔
5447
    goto _OVER;
119✔
5448
  }
5449

5450
  // send request to drop xnode task
5451
  char xnodeUrl[TSDB_XNODE_URL_LEN + 1] = {0};
2,261✔
5452
  snprintf(xnodeUrl, TSDB_XNODE_URL_LEN + 1, "%s/agent/%d", XNODED_PIPE_SOCKET_URL, pObj->id);
2,261✔
5453
  SJson *pJson = mndSendReqRetJson(xnodeUrl, HTTP_TYPE_DELETE, defaultTimeout, NULL, 0);
2,261✔
5454
  if (pJson) {
2,261✔
UNCOV
5455
    tjsonDelete(pJson);
×
5456
  }
5457

5458
  code = mndDropXnodeAgent(pMnode, pReq, pObj);
2,261✔
5459
  if (code == 0) {
2,261✔
5460
    code = TSDB_CODE_ACTION_IN_PROGRESS;
2,261✔
5461
  }
5462

5463
_OVER:
2,380✔
5464
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
2,380✔
5465
    mError("xnode task:%d, failed to drop since %s", dropReq.id, tstrerror(code));
119✔
5466
  }
5467
  mndReleaseXnodeAgent(pMnode, pObj);
2,380✔
5468
  tFreeSMDropXnodeAgentReq(&dropReq);
2,380✔
5469
  TAOS_RETURN(code);
2,380✔
5470
}
5471

5472
static int32_t mndRetrieveXnodeAgents(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
3,689✔
5473
  int32_t         code = 0;
3,689✔
5474
  SMnode         *pMnode = pReq->info.node;
3,689✔
5475
  SSdb           *pSdb = pMnode->pSdb;
3,689✔
5476
  int32_t         numOfRows = 0;
3,689✔
5477
  int32_t         cols = 0;
3,689✔
5478
  char            buf[VARSTR_HEADER_SIZE + TSDB_XNODE_AGENT_TOKEN_LEN];
3,689✔
5479
  SXnodeAgentObj *pObj = NULL;
3,689✔
5480

5481
  while (numOfRows < rows) {
10,591✔
5482
    pShow->pIter = sdbFetch(pSdb, SDB_XNODE_AGENT, pShow->pIter, (void **)&pObj);
10,591✔
5483
    if (pShow->pIter == NULL) break;
10,591✔
5484

5485
    cols = 0;
6,902✔
5486
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
6,902✔
5487
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->id, false);
6,902✔
5488
    if (code != 0) goto _end;
6,902✔
5489

5490
    STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->name, pShow->pMeta->pSchemas[cols].bytes);
6,902✔
5491
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
6,902✔
5492
    code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
6,902✔
5493
    if (code != 0) goto _end;
6,902✔
5494

5495
    if (pObj->tokenLen > 0) {
6,902✔
5496
      buf[0] = 0;
6,902✔
5497
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->token, pShow->pMeta->pSchemas[cols].bytes);
6,902✔
5498
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
6,902✔
5499
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
6,902✔
5500
      if (code != 0) goto _end;
6,902✔
5501
    } else {
UNCOV
5502
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
5503
      colDataSetNULL(pColInfo, numOfRows);
×
5504
    }
5505

5506
    if (pObj->statusLen > 0) {
6,902✔
5507
      buf[0] = 0;
5,355✔
5508
      STR_WITH_MAXSIZE_TO_VARSTR(buf, pObj->status, pShow->pMeta->pSchemas[cols].bytes);
5,355✔
5509
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
5,355✔
5510
      code = colDataSetVal(pColInfo, numOfRows, (const char *)buf, false);
5,355✔
5511
      if (code != 0) goto _end;
5,355✔
5512
    } else {
5513
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,547✔
5514
      colDataSetNULL(pColInfo, numOfRows);
1,547✔
5515
    }
5516

5517
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
6,902✔
5518
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->createTime, false);
6,902✔
5519
    if (code != 0) goto _end;
6,902✔
5520

5521
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
6,902✔
5522
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pObj->updateTime, false);
6,902✔
5523
    if (code != 0) goto _end;
6,902✔
5524

5525
    numOfRows++;
6,902✔
5526
    sdbRelease(pSdb, pObj);
6,902✔
5527
  }
5528

5529
_end:
3,689✔
5530
  if (code != 0) sdbRelease(pSdb, pObj);
3,689✔
5531

5532
  pShow->numOfRows += numOfRows;
3,689✔
5533
  return numOfRows;
3,689✔
5534
}
5535

UNCOV
5536
static void mndCancelGetNextXnodeAgent(SMnode *pMnode, void *pIter) {
×
UNCOV
5537
  SSdb *pSdb = pMnode->pSdb;
×
UNCOV
5538
  sdbCancelFetchByType(pSdb, pIter, SDB_XNODE_AGENT);
×
UNCOV
5539
}
×
5540

5541
/** xnoded mgmt section **/
5542

5543
void mndStartXnoded(SMnode *pMnode, const char *user, const char *pass, const char *token) {
1,071✔
5544
  int32_t   code = 0;
1,071✔
5545
  SXnodeOpt pOption = {0};
1,071✔
5546

5547
  if ((user == NULL || pass == NULL) && token == NULL) {
1,071✔
UNCOV
5548
    mError("xnode failed to start xnoded, dnode:%d", pMnode->selfDnodeId);
×
UNCOV
5549
    return;
×
5550
  }
5551

5552
  pOption.dnodeId = pMnode->selfDnodeId;
1,071✔
5553
  pOption.clusterId = pMnode->clusterId;
1,071✔
5554

5555
  SEpSet epset = mndGetDnodeEpsetById(pMnode, pMnode->selfDnodeId);
1,071✔
5556
  if (epset.numOfEps == 0) {
1,071✔
UNCOV
5557
    mError("xnode failed to start xnoded, dnode:%d", pMnode->selfDnodeId);
×
UNCOV
5558
    return;
×
5559
  }
5560
  pOption.ep = epset.eps[0];
1,071✔
5561
  // add user password
5562
  if (user != NULL && pass != NULL) {
1,071✔
5563
    pOption.upLen = strlen(user) + strlen(pass) + 1;
714✔
5564
    snprintf(pOption.userPass, XNODE_USER_PASS_LEN, "%s:%s", user, pass);
714✔
5565
  }
5566
  // add token
5567
  if (token != NULL) {
1,071✔
5568
    snprintf(pOption.token, sizeof(pOption.token), "%s", token);
357✔
5569
  }
5570
  if ((code = mndOpenXnd(&pOption)) != 0) {
1,071✔
5571
    mError("xnode failed to open xnd since %s, dnodeId:%d", tstrerror(code), pOption.dnodeId);
238✔
5572
    return;
238✔
5573
  }
5574
}
5575

5576
void mndXnodeHandleBecomeLeader(SMnode *pMnode) {
407,910✔
5577
  mInfo("mndxnode start to process mnode become leader");
407,910✔
5578
  SXnodeUserPassObj *pObj = mndAcquireFirstXnodeUserPass(pMnode);
407,910✔
5579
  if (pObj == NULL) {
407,910✔
5580
    mInfo("mndXnode become leader found no xnoded user pass");
407,910✔
5581
    return;
407,910✔
5582
  }
5583

UNCOV
5584
  mndStartXnoded(pMnode, pObj->user, pObj->pass, pObj->token);
×
UNCOV
5585
  mndReleaseXnodeUserPass(pMnode, pObj);
×
5586
}
5587

5588
void mndXnodeHandleBecomeNotLeader() {
57,106✔
5589
  mInfo("mndxnode handle mnode become not leader");
57,106✔
5590
  mndCloseXnd();
57,106✔
5591
}
57,106✔
5592

5593
void mndRestartXnoded(SMnode *pMnode) {
833✔
5594
  mInfo("mndxnode restart xnoded");
833✔
5595
  mndCloseXnd();
833✔
5596

5597
  taosMsleep(200);
833✔
5598
  SXnodeUserPassObj *pObj = mndAcquireFirstXnodeUserPass(pMnode);
833✔
5599
  if (pObj == NULL) {
833✔
UNCOV
5600
    mInfo("mndXnode restart found no xnoded user pass");
×
UNCOV
5601
    return;
×
5602
  }
5603
  mndStartXnoded(pMnode, pObj->user, pObj->pass, pObj->token);
833✔
5604
  mndReleaseXnodeUserPass(pMnode, pObj);
833✔
5605
  mInfo("mndxnode xnoded restarted");
833✔
5606
  return;
833✔
5607
}
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