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

L
Liu Jicong 已提交
16
#include "catalog.h"
H
Haojun Liao 已提交
17
#include "clientInt.h"
18
#include "clientLog.h"
L
Liu Jicong 已提交
19
#include "os.h"
D
dapan1121 已提交
20
#include "query.h"
L
Liu Jicong 已提交
21 22
#include "tdef.h"
#include "tname.h"
H
Haojun Liao 已提交
23
#include "tdatablock.h"
24
#include "systable.h"
H
Haojun Liao 已提交
25

H
Haojun Liao 已提交
26 27 28 29 30
static void setErrno(SRequestObj* pRequest, int32_t code) {
  pRequest->code = code;
  terrno = code;
}

D
dapan1121 已提交
31
int32_t genericRspCallback(void* param, SDataBuf* pMsg, int32_t code) {
32
  SRequestObj* pRequest = param;
H
Haojun Liao 已提交
33 34
  setErrno(pRequest, code);

35 36 37 38
  if (NEED_CLIENT_RM_TBLMETA_REQ(pRequest->type)) {
    removeMeta(pRequest->pTscObj, pRequest->targetTableList);
  }

dengyihao's avatar
dengyihao 已提交
39
  taosMemoryFree(pMsg->pEpSet);
wafwerar's avatar
wafwerar 已提交
40
  taosMemoryFree(pMsg->pData);
41 42 43 44 45
  if (pRequest->body.queryFp != NULL) {
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
H
Haojun Liao 已提交
46
  return code;
47 48
}

D
dapan1121 已提交
49
int32_t processConnectRsp(void* param, SDataBuf* pMsg, int32_t code) {
D
dapan1121 已提交
50 51 52 53 54
  SRequestObj *pRequest = acquireRequest(*(int64_t*)param);
  if (NULL == pRequest) {
    goto End;
  }
  
H
Haojun Liao 已提交
55
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
56
    setErrno(pRequest, code);
57
    tsem_post(&pRequest->body.rspSem);
dengyihao's avatar
dengyihao 已提交
58
    goto End;
H
Haojun Liao 已提交
59
  }
60

S
Shengliang Guan 已提交
61
  STscObj* pTscObj = pRequest->pTscObj;
H
Haojun Liao 已提交
62

D
dapan1121 已提交
63 64 65 66 67 68
  if (NULL == pTscObj->pAppInfo || NULL == pTscObj->pAppInfo->pAppHbMgr) {
    setErrno(pRequest, TSDB_CODE_TSC_DISCONNECTED);
    tsem_post(&pRequest->body.rspSem);
    goto End;
  }
  
S
Shengliang Guan 已提交
69
  SConnectRsp connectRsp = {0};
dengyihao's avatar
dengyihao 已提交
70 71 72 73
  if (tDeserializeSConnectRsp(pMsg->pData, pMsg->len, &connectRsp) != 0) {
    code = TSDB_CODE_TSC_INVALID_VERSION;
    setErrno(pRequest, code);
    tsem_post(&pRequest->body.rspSem);
dengyihao's avatar
dengyihao 已提交
74
    goto End;
dengyihao's avatar
dengyihao 已提交
75
  }
dengyihao's avatar
dengyihao 已提交
76 77 78 79 80

  int32_t now = taosGetTimestampSec();
  int32_t delta = abs(now - connectRsp.svrTimestamp);
  if (delta > timestampDeltaLimit) {
    code = TSDB_CODE_TIME_UNSYNCED;
dengyihao's avatar
dengyihao 已提交
81
    tscError("time diff:%ds is too big", delta);
dengyihao's avatar
dengyihao 已提交
82 83
    setErrno(pRequest, code);
    tsem_post(&pRequest->body.rspSem);
dengyihao's avatar
dengyihao 已提交
84
    goto End;
dengyihao's avatar
dengyihao 已提交
85 86
  }

L
Liu Jicong 已提交
87 88
  /*assert(connectRsp.epSet.numOfEps > 0);*/
  if (connectRsp.epSet.numOfEps == 0) {
S
Shengliang Guan 已提交
89
    setErrno(pRequest, TSDB_CODE_APP_ERROR);
L
Liu Jicong 已提交
90
    tsem_post(&pRequest->body.rspSem);
dengyihao's avatar
dengyihao 已提交
91
    goto End;
L
Liu Jicong 已提交
92
  }
H
Haojun Liao 已提交
93

D
dapan1121 已提交
94
  if (connectRsp.dnodeNum == 1) {
dengyihao's avatar
dengyihao 已提交
95 96 97 98
    SEpSet srcEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
    SEpSet dstEpSet = connectRsp.epSet;
    rpcSetDefaultAddr(pTscObj->pAppInfo->pTransporter, srcEpSet.eps[srcEpSet.inUse].fqdn,
                      dstEpSet.eps[dstEpSet.inUse].fqdn);
D
dapan1121 已提交
99
  } else if (connectRsp.dnodeNum > 1 && !isEpsetEqual(&pTscObj->pAppInfo->mgmtEp.epSet, &connectRsp.epSet)) {
D
dapan1121 已提交
100
    SEpSet* pOrig = &pTscObj->pAppInfo->mgmtEp.epSet;
101 102 103 104 105
    SEp*    pOrigEp = &pOrig->eps[pOrig->inUse];
    SEp*    pNewEp = &connectRsp.epSet.eps[connectRsp.epSet.inUse];
    tscDebug("mnode epset updated from %d/%d=>%s:%d to %d/%d=>%s:%d in connRsp", pOrig->inUse, pOrig->numOfEps,
             pOrigEp->fqdn, pOrigEp->port, connectRsp.epSet.inUse, connectRsp.epSet.numOfEps, pNewEp->fqdn,
             pNewEp->port);
S
Shengliang Guan 已提交
106
    updateEpSet_s(&pTscObj->pAppInfo->mgmtEp, &connectRsp.epSet);
107 108
  }

S
Shengliang Guan 已提交
109
  for (int32_t i = 0; i < connectRsp.epSet.numOfEps; ++i) {
S
Shengliang Guan 已提交
110
    tscDebug("0x%" PRIx64 " epSet.fqdn[%d]:%s port:%d, connObj:0x%" PRIx64, pRequest->requestId, i,
D
dapan1121 已提交
111
             connectRsp.epSet.eps[i].fqdn, connectRsp.epSet.eps[i].port, pTscObj->id);
H
Haojun Liao 已提交
112 113
  }

114
  pTscObj->sysInfo = connectRsp.sysInfo;
S
Shengliang Guan 已提交
115 116
  pTscObj->connId = connectRsp.connId;
  pTscObj->acctId = connectRsp.acctId;
D
dapan1121 已提交
117 118
  tstrncpy(pTscObj->sVer, connectRsp.sVer, tListLen(pTscObj->sVer));
  tstrncpy(pTscObj->sDetailVer, connectRsp.sDetailVer, tListLen(pTscObj->sDetailVer));
H
Haojun Liao 已提交
119 120

  // update the appInstInfo
S
Shengliang Guan 已提交
121
  pTscObj->pAppInfo->clusterId = connectRsp.clusterId;
D
dapan1121 已提交
122
  lastClusterId = connectRsp.clusterId;
H
Haojun Liao 已提交
123

L
Liu Jicong 已提交
124
  pTscObj->connType = connectRsp.connType;
125

D
dapan1121 已提交
126
  hbRegisterConn(pTscObj->pAppInfo->pAppHbMgr, pTscObj->id, connectRsp.clusterId, connectRsp.connType);
L
Liu Jicong 已提交
127

S
Shengliang Guan 已提交
128
  tscDebug("0x%" PRIx64 " clusterId:%" PRId64 ", totalConn:%" PRId64, pRequest->requestId, connectRsp.clusterId,
129
           pTscObj->pAppInfo->numOfConns);
D
dapan1121 已提交
130
           
131
  tsem_post(&pRequest->body.rspSem);
dengyihao's avatar
dengyihao 已提交
132 133
End:

D
dapan1121 已提交
134 135 136 137 138
  if (pRequest) {
    releaseRequest(pRequest->self);
  }
  
  taosMemoryFree(param);
dengyihao's avatar
dengyihao 已提交
139 140 141
  taosMemoryFree(pMsg->pEpSet);
  taosMemoryFree(pMsg->pData);
  return code;
142
}
H
Haojun Liao 已提交
143

L
Liu Jicong 已提交
144
SMsgSendInfo* buildMsgInfoImpl(SRequestObj* pRequest) {
wafwerar's avatar
wafwerar 已提交
145
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
146

H
Haojun Liao 已提交
147
  pMsgSendInfo->requestObjRefId = pRequest->self;
L
Liu Jicong 已提交
148 149 150
  pMsgSendInfo->requestId = pRequest->requestId;
  pMsgSendInfo->param = pRequest;
  pMsgSendInfo->msgType = pRequest->type;
D
dapan1121 已提交
151
  pMsgSendInfo->target.type = TARGET_TYPE_MNODE;
152

H
Haojun Liao 已提交
153 154
  assert(pRequest != NULL);
  pMsgSendInfo->msgInfo = pRequest->body.requestMsg;
H
Haojun Liao 已提交
155
  pMsgSendInfo->fp = getMsgRspHandle(pRequest->type);
156
  return pMsgSendInfo;
157 158
}

D
dapan1121 已提交
159
int32_t processCreateDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
H
Haojun Liao 已提交
160
  // todo rsp with the vnode id list
161
  SRequestObj* pRequest = param;
wafwerar's avatar
wafwerar 已提交
162
  taosMemoryFree(pMsg->pData);
dengyihao's avatar
dengyihao 已提交
163
  taosMemoryFree(pMsg->pEpSet);
X
Xiaoyu Wang 已提交
164 165 166
  if (code != TSDB_CODE_SUCCESS) {
    setErrno(pRequest, code);
  }
167 168 169 170 171 172

  if (pRequest->body.queryFp) {
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
X
Xiaoyu Wang 已提交
173
  return code;
H
Haojun Liao 已提交
174
}
H
Haojun Liao 已提交
175

D
dapan1121 已提交
176
int32_t processUseDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
H
Haojun Liao 已提交
177 178
  SRequestObj* pRequest = param;

179 180
  if (TSDB_CODE_MND_DB_NOT_EXIST == code || TSDB_CODE_MND_DB_IN_CREATING == code ||
      TSDB_CODE_MND_DB_IN_DROPPING == code) {
D
dapan1121 已提交
181 182
    SUseDbRsp usedbRsp = {0};
    tDeserializeSUseDbRsp(pMsg->pData, pMsg->len, &usedbRsp);
L
Liu Jicong 已提交
183
    struct SCatalog* pCatalog = NULL;
D
dapan1121 已提交
184

D
dapan1121 已提交
185
    if (usedbRsp.vgVersion >= 0) { // cached in local
186
      uint64_t clusterId = pRequest->pTscObj->pAppInfo->clusterId;
dengyihao's avatar
dengyihao 已提交
187
      int32_t  code1 = catalogGetHandle(clusterId, &pCatalog);
188
      if (code1 != TSDB_CODE_SUCCESS) {
dengyihao's avatar
dengyihao 已提交
189 190
        tscWarn("0x%" PRIx64 "catalogGetHandle failed, clusterId:%" PRIx64 ", error:%s", pRequest->requestId, clusterId,
                tstrerror(code1));
D
dapan1121 已提交
191 192 193 194 195
      } else {
        catalogRemoveDB(pCatalog, usedbRsp.db, usedbRsp.uid);
      }
    }

L
Liu Jicong 已提交
196
    tFreeSUsedbRsp(&usedbRsp);
D
dapan1121 已提交
197 198
  }

H
Haojun Liao 已提交
199
  if (code != TSDB_CODE_SUCCESS) {
wafwerar's avatar
wafwerar 已提交
200
    taosMemoryFree(pMsg->pData);
dengyihao's avatar
dengyihao 已提交
201
    taosMemoryFree(pMsg->pEpSet);
H
Haojun Liao 已提交
202
    setErrno(pRequest, code);
203 204 205 206 207 208 209

    if (pRequest->body.queryFp != NULL) {
      pRequest->body.queryFp(pRequest->body.param, pRequest, pRequest->code);
    } else {
      tsem_post(&pRequest->body.rspSem);
    }

H
Haojun Liao 已提交
210 211 212
    return code;
  }

S
Shengliang Guan 已提交
213 214 215
  SUseDbRsp usedbRsp = {0};
  tDeserializeSUseDbRsp(pMsg->pData, pMsg->len, &usedbRsp);

dengyihao's avatar
dengyihao 已提交
216
  if (strlen(usedbRsp.db) == 0) {
217 218 219 220 221
    if (usedbRsp.errCode != 0) {
      return usedbRsp.errCode;
    } else {
      return TSDB_CODE_APP_ERROR;
    }
wmmhello's avatar
wmmhello 已提交
222 223
  }

S
Shengliang Guan 已提交
224 225 226 227 228 229 230 231 232
  tscTrace("db:%s, usedbRsp received, numOfVgroups:%d", usedbRsp.db, usedbRsp.vgNum);
  for (int32_t i = 0; i < usedbRsp.vgNum; ++i) {
    SVgroupInfo* pInfo = taosArrayGet(usedbRsp.pVgroupInfos, i);
    tscTrace("vgId:%d, numOfEps:%d inUse:%d ", pInfo->vgId, pInfo->epSet.numOfEps, pInfo->epSet.inUse);
    for (int32_t j = 0; j < pInfo->epSet.numOfEps; ++j) {
      tscTrace("vgId:%d, index:%d epset:%s:%u", pInfo->vgId, j, pInfo->epSet.eps[j].fqdn, pInfo->epSet.eps[j].port);
    }
  }

233
  SName name = {0};
L
Liu Jicong 已提交
234
  tNameFromString(&name, usedbRsp.db, T_NAME_ACCT | T_NAME_DB);
S
Shengliang Guan 已提交
235

D
dapan1121 已提交
236 237 238 239 240 241 242
  SUseDbOutput output = {0};
  code = queryBuildUseDbOutput(&output, &usedbRsp);

  if (code != 0) {
    terrno = code;
    if (output.dbVgroup) taosHashCleanup(output.dbVgroup->vgHash);

dengyihao's avatar
dengyihao 已提交
243
    tscError("0x%" PRIx64 " failed to build use db output since %s", pRequest->requestId, terrstr());
D
dapan1121 已提交
244
  } else if (output.dbVgroup && output.dbVgroup->vgHash) {
L
Liu Jicong 已提交
245 246
    struct SCatalog* pCatalog = NULL;

247 248
    int32_t code1 = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
    if (code1 != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
249
      tscWarn("catalogGetHandle failed, clusterId:%" PRIx64 ", error:%s", pRequest->pTscObj->pAppInfo->clusterId,
250
              tstrerror(code1));
D
dapan1121 已提交
251 252
    } else {
      catalogUpdateDBVgInfo(pCatalog, output.db, output.dbId, output.dbVgroup);
D
dapan1121 已提交
253
      output.dbVgroup = NULL;
D
dapan1121 已提交
254 255 256
    }
  }

D
dapan1121 已提交
257 258
  taosMemoryFreeClear(output.dbVgroup);

S
Shengliang Guan 已提交
259
  tFreeSUsedbRsp(&usedbRsp);
260 261 262

  char db[TSDB_DB_NAME_LEN] = {0};
  tNameGetDbName(&name, db);
263

264
  setConnectionDB(pRequest->pTscObj, db);
wafwerar's avatar
wafwerar 已提交
265
  taosMemoryFree(pMsg->pData);
dengyihao's avatar
dengyihao 已提交
266
  taosMemoryFree(pMsg->pEpSet);
267 268 269 270 271 272

  if (pRequest->body.queryFp != NULL) {
    pRequest->body.queryFp(pRequest->body.param, pRequest, pRequest->code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
273
  return 0;
274 275
}

D
dapan1121 已提交
276
int32_t processCreateSTableRsp(void* param, SDataBuf* pMsg, int32_t code) {
H
Haojun Liao 已提交
277
  assert(pMsg != NULL && param != NULL);
278
  SRequestObj* pRequest = param;
H
Haojun Liao 已提交
279 280 281

  if (code != TSDB_CODE_SUCCESS) {
    setErrno(pRequest, code);
282 283
  } else {
    SMCreateStbRsp createRsp = {0};
dengyihao's avatar
dengyihao 已提交
284
    SDecoder       coder = {0};
285 286 287 288 289 290
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
    tDecodeSMCreateStbRsp(&coder, &createRsp);
    tDecoderClear(&coder);

    pRequest->body.resInfo.execRes.msgType = TDMT_MND_CREATE_STB;
    pRequest->body.resInfo.execRes.res = createRsp.pMeta;
H
Haojun Liao 已提交
291 292
  }

dengyihao's avatar
dengyihao 已提交
293
  taosMemoryFree(pMsg->pEpSet);
294 295
  taosMemoryFree(pMsg->pData);

296
  if (pRequest->body.queryFp != NULL) {
297 298 299 300 301 302 303 304 305 306 307 308 309
    SExecResult* pRes = &pRequest->body.resInfo.execRes;

    if (code == TSDB_CODE_SUCCESS) {
      SCatalog* pCatalog = NULL;
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
      if (pRes->res != NULL) {
        ret = handleCreateTbExecRes(pRes->res, pCatalog);
      }

      if (ret != TSDB_CODE_SUCCESS) {
        code = ret;
      }
    }
dengyihao's avatar
dengyihao 已提交
310

311 312 313 314
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
H
Haojun Liao 已提交
315
  return code;
316 317
}

D
dapan1121 已提交
318
int32_t processDropDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
319
  SRequestObj* pRequest = param;
H
Haojun Liao 已提交
320 321
  if (code != TSDB_CODE_SUCCESS) {
    setErrno(pRequest, code);
322 323 324
  } else {
    SDropDbRsp dropdbRsp = {0};
    tDeserializeSDropDbRsp(pMsg->pData, pMsg->len, &dropdbRsp);
D
dapan1121 已提交
325

326
    struct SCatalog* pCatalog = NULL;
dengyihao's avatar
dengyihao 已提交
327
    int32_t          code = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
D
dapan1121 已提交
328 329
    if (TSDB_CODE_SUCCESS == code) {
      catalogRemoveDB(pCatalog, dropdbRsp.db, dropdbRsp.uid);
330 331 332 333 334 335
      STscObj*             pTscObj = pRequest->pTscObj;

      SRequestConnInfo conn = {.pTrans = pTscObj->pAppInfo->pTransporter,
                               .requestId = pRequest->requestId,
                               .requestObjRefId = pRequest->self,
                               .mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp)};
D
dapan1121 已提交
336 337 338 339 340
      char dbFName[TSDB_DB_FNAME_LEN];
      snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_INFORMATION_SCHEMA_DB);
      catalogRefreshDBVgInfo(pCatalog, &conn, dbFName);
      snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_PERFORMANCE_SCHEMA_DB);
      catalogRefreshDBVgInfo(pCatalog, &conn, dbFName);
D
dapan1121 已提交
341
    }
342
  }
D
dapan1121 已提交
343

D
dapan1121 已提交
344
  taosMemoryFree(pMsg->pData);
dengyihao's avatar
dengyihao 已提交
345
  taosMemoryFree(pMsg->pEpSet);
D
dapan1121 已提交
346

347 348 349 350 351
  if (pRequest->body.queryFp != NULL) {
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
H
Haojun Liao 已提交
352
  return code;
353 354
}

D
dapan1121 已提交
355
int32_t processAlterStbRsp(void* param, SDataBuf* pMsg, int32_t code) {
D
dapan1121 已提交
356 357 358
  SRequestObj* pRequest = param;
  if (code != TSDB_CODE_SUCCESS) {
    setErrno(pRequest, code);
359 360 361 362 363 364 365 366 367
  } else {
    SMAlterStbRsp alterRsp = {0};
    SDecoder      coder = {0};
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
    tDecodeSMAlterStbRsp(&coder, &alterRsp);
    tDecoderClear(&coder);

    pRequest->body.resInfo.execRes.msgType = TDMT_MND_ALTER_STB;
    pRequest->body.resInfo.execRes.res = alterRsp.pMeta;
D
dapan1121 已提交
368 369
  }

D
dapan1121 已提交
370
  taosMemoryFree(pMsg->pData);
dengyihao's avatar
dengyihao 已提交
371
  taosMemoryFree(pMsg->pEpSet);
D
dapan1121 已提交
372

373
  if (pRequest->body.queryFp != NULL) {
D
dapan1121 已提交
374
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
375 376 377 378 379 380 381 382 383 384 385 386 387

    if (code == TSDB_CODE_SUCCESS) {
      SCatalog* pCatalog = NULL;
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
      if (pRes->res != NULL) {
        ret = handleAlterTbExecRes(pRes->res, pCatalog);
      }

      if (ret != TSDB_CODE_SUCCESS) {
        code = ret;
      }
    }

388 389 390 391
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
D
dapan1121 已提交
392 393 394
  return code;
}

D
dapan1121 已提交
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
static int32_t buildShowVariablesBlock(SArray* pVars, SSDataBlock** block) {
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
  pBlock->info.hasVarCol = true;

  pBlock->pDataBlock = taosArrayInit(SHOW_VARIABLES_RESULT_COLS, sizeof(SColumnInfoData));

  SColumnInfoData infoData = {0};
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD1_LEN;

  taosArrayPush(pBlock->pDataBlock, &infoData);

  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD2_LEN;
  taosArrayPush(pBlock->pDataBlock, &infoData);

  int32_t numOfCfg = taosArrayGetSize(pVars);
  blockDataEnsureCapacity(pBlock, numOfCfg);

  for (int32_t i = 0, c = 0; i < numOfCfg; ++i, c = 0) {
415
    SVariablesInfo* pInfo = taosArrayGet(pVars, i);
D
dapan1121 已提交
416 417 418

    char name[TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE] = {0};
    STR_WITH_MAXSIZE_TO_VARSTR(name, pInfo->name, TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE);
419
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
D
dapan1121 已提交
420
    colDataAppend(pColInfo, i, name, false);
421

D
dapan1121 已提交
422 423 424 425 426 427 428 429 430
    char value[TSDB_CONFIG_VALUE_LEN + VARSTR_HEADER_SIZE] = {0};
    STR_WITH_MAXSIZE_TO_VARSTR(value, pInfo->value, TSDB_CONFIG_VALUE_LEN + VARSTR_HEADER_SIZE);
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
    colDataAppend(pColInfo, i, value, false);
  }

  pBlock->info.rows = numOfCfg;

  *block = pBlock;
431

D
dapan1121 已提交
432 433 434 435 436
  return TSDB_CODE_SUCCESS;
}

static int32_t buildShowVariablesRsp(SArray* pVars, SRetrieveTableRsp** pRsp) {
  SSDataBlock* pBlock = NULL;
437
  int32_t      code = buildShowVariablesBlock(pVars, &pBlock);
D
dapan1121 已提交
438 439 440 441 442 443 444
  if (code) {
    return code;
  }

  size_t rspSize = sizeof(SRetrieveTableRsp) + blockGetEncodeSize(pBlock);
  *pRsp = taosMemoryCalloc(1, rspSize);
  if (NULL == *pRsp) {
D
dapan1121 已提交
445
    blockDataDestroy(pBlock);
D
dapan1121 已提交
446 447 448 449 450 451 452 453
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  (*pRsp)->useconds = 0;
  (*pRsp)->completed = 1;
  (*pRsp)->precision = 0;
  (*pRsp)->compressed = 0;
  (*pRsp)->compLen = 0;
454
  (*pRsp)->numOfRows = htobe64((int64_t)pBlock->info.rows);
D
dapan1121 已提交
455 456
  (*pRsp)->numOfCols = htonl(SHOW_VARIABLES_RESULT_COLS);

H
Haojun Liao 已提交
457
  int32_t len = blockEncode(pBlock, (*pRsp)->data, SHOW_VARIABLES_RESULT_COLS);
D
dapan1121 已提交
458 459 460 461 462 463
  ASSERT(len == rspSize - sizeof(SRetrieveTableRsp));

  blockDataDestroy(pBlock);
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
464
int32_t processShowVariablesRsp(void* param, SDataBuf* pMsg, int32_t code) {
D
dapan1121 已提交
465 466 467 468
  SRequestObj* pRequest = param;
  if (code != TSDB_CODE_SUCCESS) {
    setErrno(pRequest, code);
  } else {
469
    SShowVariablesRsp  rsp = {0};
D
dapan1121 已提交
470 471 472 473 474 475
    SRetrieveTableRsp* pRes = NULL;
    code = tDeserializeSShowVariablesRsp(pMsg->pData, pMsg->len, &rsp);
    if (TSDB_CODE_SUCCESS == code) {
      code = buildShowVariablesRsp(rsp.variables, &pRes);
    }
    if (TSDB_CODE_SUCCESS == code) {
D
dapan1121 已提交
476
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, true);
D
dapan1121 已提交
477
    }
478

D
dapan1121 已提交
479 480 481
    tFreeSShowVariablesRsp(&rsp);
  }

D
dapan1121 已提交
482
  taosMemoryFree(pMsg->pData);
dengyihao's avatar
dengyihao 已提交
483
  taosMemoryFree(pMsg->pEpSet);
D
dapan1121 已提交
484

D
dapan1121 已提交
485 486 487 488 489 490 491 492
  if (pRequest->body.queryFp != NULL) {
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  } else {
    tsem_post(&pRequest->body.rspSem);
  }
  return code;
}

H
Haojun Liao 已提交
493 494 495 496 497 498 499 500 501 502 503 504 505 506
__async_send_cb_fn_t getMsgRspHandle(int32_t msgType) {
  switch (msgType) {
    case TDMT_MND_CONNECT:
      return processConnectRsp;
    case TDMT_MND_CREATE_DB:
      return processCreateDbRsp;
    case TDMT_MND_USE_DB:
      return processUseDbRsp;
    case TDMT_MND_CREATE_STB:
      return processCreateSTableRsp;
    case TDMT_MND_DROP_DB:
      return processDropDbRsp;
    case TDMT_MND_ALTER_STB:
      return processAlterStbRsp;
D
dapan1121 已提交
507 508
    case TDMT_MND_SHOW_VARIABLES:
      return processShowVariablesRsp;
H
Haojun Liao 已提交
509 510 511
    default:
      return genericRspCallback;
  }
L
Liu Jicong 已提交
512
}