transCli.c 35.6 KB
Newer Older
dengyihao's avatar
dengyihao 已提交
1
/* * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
dengyihao's avatar
dengyihao 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
 *
 * 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/>.
 */

#ifdef USE_UV
#include "transComm.h"

typedef struct SCliConn {
dengyihao's avatar
dengyihao 已提交
19
  T_REF_DECLARE()
dengyihao's avatar
dengyihao 已提交
20 21
  uv_connect_t connReq;
  uv_stream_t* stream;
dengyihao's avatar
dengyihao 已提交
22
  uv_write_t   writeReq;
dengyihao's avatar
dengyihao 已提交
23

dengyihao's avatar
dengyihao 已提交
24 25 26 27 28 29 30 31 32
  void* hostThrd;
  int   hThrdIdx;

  SConnBuffer readBuf;
  STransQueue cliMsgs;
  queue       conn;
  uint64_t    expireTime;

  STransCtx  ctx;
dengyihao's avatar
dengyihao 已提交
33 34
  bool       broken;  // link broken or not
  ConnStatus status;  //
dengyihao's avatar
dengyihao 已提交
35 36 37 38

  char*    ip;
  uint32_t port;

dengyihao's avatar
dengyihao 已提交
39
  // debug and log info
dengyihao's avatar
dengyihao 已提交
40
  struct sockaddr_in addr;
dengyihao's avatar
dengyihao 已提交
41
  struct sockaddr_in locaddr;
dengyihao's avatar
dengyihao 已提交
42
} SCliConn;
dengyihao's avatar
dengyihao 已提交
43

dengyihao's avatar
dengyihao 已提交
44
typedef struct SCliMsg {
dengyihao's avatar
dengyihao 已提交
45
  STransConnCtx* ctx;
dengyihao's avatar
formate  
dengyihao 已提交
46
  STransMsg      msg;
dengyihao's avatar
dengyihao 已提交
47 48
  queue          q;
  uint64_t       st;
dengyihao's avatar
dengyihao 已提交
49
  STransMsgType  type;
dengyihao's avatar
dengyihao 已提交
50
  int            sent;  //(0: no send, 1: alread sent)
dengyihao's avatar
dengyihao 已提交
51 52 53
} SCliMsg;

typedef struct SCliThrdObj {
dengyihao's avatar
dengyihao 已提交
54
  TdThread    thread;
dengyihao's avatar
dengyihao 已提交
55
  uv_loop_t*  loop;
dengyihao's avatar
dengyihao 已提交
56
  SAsyncPool* asyncPool;
dengyihao's avatar
dengyihao 已提交
57
  uv_timer_t  timer;
dengyihao's avatar
dengyihao 已提交
58 59 60
  void*       pool;  // conn pool

  // msg queue
dengyihao's avatar
dengyihao 已提交
61
  queue         msg;
wafwerar's avatar
wafwerar 已提交
62
  TdThreadMutex msgMtx;
dengyihao's avatar
dengyihao 已提交
63 64 65 66
  SDelayQueue*  delayQueue;
  uint64_t      nextTimeout;  // next timeout
  void*         pTransInst;   //
  bool          quit;
dengyihao's avatar
dengyihao 已提交
67 68
} SCliThrdObj;

U
ubuntu 已提交
69
typedef struct SCliObj {
dengyihao's avatar
dengyihao 已提交
70 71 72 73
  char          label[TSDB_LABEL_LEN];
  int32_t       index;
  int           numOfThreads;
  SCliThrdObj** pThreadObj;
U
ubuntu 已提交
74
} SCliObj;
dengyihao's avatar
dengyihao 已提交
75

dengyihao's avatar
dengyihao 已提交
76 77 78 79
typedef struct SConnList {
  queue conn;
} SConnList;

dengyihao's avatar
dengyihao 已提交
80
// conn pool
dengyihao's avatar
dengyihao 已提交
81
// add expire timeout and capacity limit
dengyihao's avatar
dengyihao 已提交
82
static void*     createConnPool(int size);
83
static void*     destroyConnPool(void* pool);
dengyihao's avatar
dengyihao 已提交
84
static SCliConn* getConnFromPool(void* pool, char* ip, uint32_t port);
dengyihao's avatar
dengyihao 已提交
85
static void      addConnToPool(void* pool, SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
86

dengyihao's avatar
dengyihao 已提交
87
// register timer in each thread to clear expire conn
U
ubuntu 已提交
88 89
static void cliTimeoutCb(uv_timer_t* handle);
// alloc buf for recv
dengyihao's avatar
dengyihao 已提交
90
static void cliAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf);
U
ubuntu 已提交
91 92
// callback after  read nbytes from socket
static void cliRecvCb(uv_stream_t* cli, ssize_t nread, const uv_buf_t* buf);
dengyihao's avatar
dengyihao 已提交
93
// callback after write data to socket
U
ubuntu 已提交
94
static void cliSendCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
95
// callback after conn  to server
U
ubuntu 已提交
96 97
static void cliConnCb(uv_connect_t* req, int status);
static void cliAsyncCb(uv_async_t* handle);
dengyihao's avatar
dengyihao 已提交
98

dengyihao's avatar
dengyihao 已提交
99
static int cliAppCb(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg);
dengyihao's avatar
dengyihao 已提交
100

U
ubuntu 已提交
101 102 103
static SCliConn* cliCreateConn(SCliThrdObj* thrd);
static void      cliDestroyConn(SCliConn* pConn, bool clear /*clear tcp handle or not*/);
static void      cliDestroy(uv_handle_t* handle);
dengyihao's avatar
dengyihao 已提交
104
static void      cliSend(SCliConn* pConn);
dengyihao's avatar
dengyihao 已提交
105

dengyihao's avatar
dengyihao 已提交
106 107 108 109
/*
 * set TCP connection timeout per-socket level
 */
static int cliCreateSocket();
dengyihao's avatar
dengyihao 已提交
110
// process data read from server, add decompress etc later
U
ubuntu 已提交
111
static void cliHandleResp(SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
112
// handle except about conn
U
ubuntu 已提交
113
static void cliHandleExcept(SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
114

115
// handle req from app
U
ubuntu 已提交
116 117
static void cliHandleReq(SCliMsg* pMsg, SCliThrdObj* pThrd);
static void cliHandleQuit(SCliMsg* pMsg, SCliThrdObj* pThrd);
dengyihao's avatar
dengyihao 已提交
118
static void cliHandleRelease(SCliMsg* pMsg, SCliThrdObj* pThrd);
dengyihao's avatar
dengyihao 已提交
119
static void (*cliAsyncHandle[])(SCliMsg* pMsg, SCliThrdObj* pThrd) = {cliHandleReq, cliHandleQuit, cliHandleRelease};
dengyihao's avatar
dengyihao 已提交
120

U
ubuntu 已提交
121
static void cliSendQuit(SCliThrdObj* thrd);
U
ubuntu 已提交
122
static void destroyUserdata(STransMsg* userdata);
dengyihao's avatar
dengyihao 已提交
123

U
ubuntu 已提交
124
static int cliRBChoseIdx(STrans* pTransInst);
dengyihao's avatar
dengyihao 已提交
125

dengyihao's avatar
dengyihao 已提交
126 127
static void destroyCmsg(SCliMsg* cmsg);
static void transDestroyConnCtx(STransConnCtx* ctx);
dengyihao's avatar
dengyihao 已提交
128 129 130
// thread obj
static SCliThrdObj* createThrdObj();
static void         destroyThrdObj(SCliThrdObj* pThrd);
dengyihao's avatar
dengyihao 已提交
131

dengyihao's avatar
dengyihao 已提交
132 133 134 135 136 137 138 139 140
static void cliWalkCb(uv_handle_t* handle, void* arg);

#define CLI_RELEASE_UV(loop)        \
  do {                              \
    uv_walk(loop, cliWalkCb, NULL); \
    uv_run(loop, UV_RUN_DEFAULT);   \
    uv_loop_close(loop);            \
  } while (0);

dengyihao's avatar
dengyihao 已提交
141 142 143 144 145 146
// snprintf may cause performance problem
#define CONN_CONSTRUCT_HASH_KEY(key, ip, port)          \
  do {                                                  \
    snprintf(key, sizeof(key), "%s:%d", ip, (int)port); \
  } while (0)

dengyihao's avatar
dengyihao 已提交
147
#define CONN_HOST_THREAD_INDEX(conn) (conn ? ((SCliConn*)conn)->hThrdIdx : -1)
dengyihao's avatar
dengyihao 已提交
148 149 150
#define CONN_PERSIST_TIME(para)      (para * 1000 * 10)
#define CONN_GET_HOST_THREAD(conn)   (conn ? ((SCliConn*)conn)->hostThrd : NULL)
#define CONN_GET_INST_LABEL(conn)    (((STrans*)(((SCliThrdObj*)(conn)->hostThrd)->pTransInst))->label)
dengyihao's avatar
dengyihao 已提交
151 152 153
#define CONN_SHOULD_RELEASE(conn, head)                                                  \
  do {                                                                                   \
    if ((head)->release == 1 && (head->msgLen) == sizeof(*head)) {                       \
dengyihao's avatar
dengyihao 已提交
154 155
      uint64_t ahandle = head->ahandle;                                                  \
      CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle);                                         \
dengyihao's avatar
dengyihao 已提交
156 157 158 159
      conn->status = ConnRelease;                                                        \
      transClearBuffer(&conn->readBuf);                                                  \
      transFreeMsg(transContFromHead((char*)head));                                      \
      tDebug("cli conn %p receive release request, ref: %d", conn, T_REF_VAL_GET(conn)); \
dengyihao's avatar
dengyihao 已提交
160
      if (T_REF_VAL_GET(conn) > 1) {                                                     \
161
        transUnrefCliHandle(conn);                                                       \
dengyihao's avatar
dengyihao 已提交
162
      }                                                                                  \
dengyihao's avatar
dengyihao 已提交
163
      destroyCmsg(pMsg);                                                                 \
dengyihao's avatar
dengyihao 已提交
164
      addConnToPool(((SCliThrdObj*)conn->hostThrd)->pool, conn);                         \
dengyihao's avatar
dengyihao 已提交
165 166
      return;                                                                            \
    }                                                                                    \
dengyihao's avatar
dengyihao 已提交
167 168
  } while (0)

dengyihao's avatar
dengyihao 已提交
169 170 171 172 173
#define CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle)                                         \
  do {                                                                                    \
    int i = 0, sz = transQueueSize(&conn->cliMsgs);                                       \
    for (; i < sz; i++) {                                                                 \
      pMsg = transQueueGet(&conn->cliMsgs, i);                                            \
S
Shengliang 已提交
174
      if (pMsg != NULL && pMsg->ctx != NULL && (uint64_t)pMsg->ctx->ahandle == ahandle) { \
dengyihao's avatar
dengyihao 已提交
175 176 177 178 179 180 181 182
        break;                                                                            \
      }                                                                                   \
    }                                                                                     \
    if (i == sz) {                                                                        \
      pMsg = NULL;                                                                        \
    } else {                                                                              \
      pMsg = transQueueRm(&conn->cliMsgs, i);                                             \
    }                                                                                     \
dengyihao's avatar
dengyihao 已提交
183
  } while (0)
U
ubuntu 已提交
184 185 186 187 188 189 190 191 192 193 194 195 196
#define CONN_GET_NEXT_SENDMSG(conn)                 \
  do {                                              \
    int i = 0;                                      \
    do {                                            \
      pCliMsg = transQueueGet(&conn->cliMsgs, i++); \
      if (pCliMsg && 0 == pCliMsg->sent) {          \
        break;                                      \
      }                                             \
    } while (pCliMsg != NULL);                      \
    if (pCliMsg == NULL) {                          \
      goto _RETURN;                                 \
    }                                               \
  } while (0)
dengyihao's avatar
dengyihao 已提交
197

dengyihao's avatar
dengyihao 已提交
198 199 200 201 202
#define CONN_HANDLE_THREAD_QUIT(thrd) \
  do {                                \
    if (thrd->quit) {                 \
      return;                         \
    }                                 \
dengyihao's avatar
dengyihao 已提交
203 204 205 206 207
  } while (0)

#define CONN_HANDLE_BROKEN(conn) \
  do {                           \
    if (conn->broken) {          \
dengyihao's avatar
formate  
dengyihao 已提交
208
      cliHandleExcept(conn);     \
dengyihao's avatar
dengyihao 已提交
209
      return;                    \
dengyihao's avatar
dengyihao 已提交
210
    }                            \
dengyihao's avatar
dengyihao 已提交
211
  } while (0)
dengyihao's avatar
dengyihao 已提交
212

dengyihao's avatar
formate  
dengyihao 已提交
213 214
#define CONN_SET_PERSIST_BY_APP(conn) \
  do {                                \
dengyihao's avatar
dengyihao 已提交
215 216
    if (conn->status == ConnNormal) { \
      conn->status = ConnAcquire;     \
dengyihao's avatar
formate  
dengyihao 已提交
217 218 219
      transRefCliHandle(conn);        \
    }                                 \
  } while (0)
220

dengyihao's avatar
dengyihao 已提交
221 222 223 224
#define CONN_NO_PERSIST_BY_APP(conn) \
  (((conn)->status == ConnNormal || (conn)->status == ConnInPool) && T_REF_VAL_GET(conn) == 1)
#define CONN_RELEASE_BY_SERVER(conn) \
  (((conn)->status == ConnRelease || (conn)->status == ConnInPool) && T_REF_VAL_GET(conn) == 1)
225

dengyihao's avatar
dengyihao 已提交
226 227
#define REQUEST_NO_RESP(msg)         ((msg)->noResp == 1)
#define REQUEST_PERSIS_HANDLE(msg)   ((msg)->persistHandle == 1)
dengyihao's avatar
dengyihao 已提交
228
#define REQUEST_RELEASE_HANDLE(cmsg) ((cmsg)->type == Release)
dengyihao's avatar
dengyihao 已提交
229

dengyihao's avatar
dengyihao 已提交
230 231 232
#define EPSET_GET_INUSE_IP(epSet)   ((epSet)->eps[(epSet)->inUse].fqdn)
#define EPSET_GET_INUSE_PORT(epSet) ((epSet)->eps[(epSet)->inUse].port)

U
ubuntu 已提交
233
static void* cliWorkThread(void* arg);
dengyihao's avatar
dengyihao 已提交
234

dengyihao's avatar
dengyihao 已提交
235
bool cliMaySendCachedMsg(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
236
  if (!transQueueEmpty(&conn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
237
    SCliMsg* pCliMsg = NULL;
U
ubuntu 已提交
238
    CONN_GET_NEXT_SENDMSG(conn);
dengyihao's avatar
dengyihao 已提交
239 240
    cliSend(conn);
  }
dengyihao's avatar
dengyihao 已提交
241
  return false;
U
ubuntu 已提交
242 243
_RETURN:
  return false;
dengyihao's avatar
dengyihao 已提交
244
}
dengyihao's avatar
formate  
dengyihao 已提交
245
void cliHandleResp(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
246
  SCliThrdObj* pThrd = conn->hostThrd;
dengyihao's avatar
formate  
dengyihao 已提交
247
  STrans*      pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
248

dengyihao's avatar
dengyihao 已提交
249 250 251
  STransMsgHead* pHead = (STransMsgHead*)(conn->readBuf.buf);
  pHead->code = htonl(pHead->code);
  pHead->msgLen = htonl(pHead->msgLen);
dengyihao's avatar
dengyihao 已提交
252

dengyihao's avatar
dengyihao 已提交
253 254 255 256 257 258
  STransMsg transMsg = {0};
  transMsg.contLen = transContLenFromMsg(pHead->msgLen);
  transMsg.pCont = transContFromHead((char*)pHead);
  transMsg.code = pHead->code;
  transMsg.msgType = pHead->msgType;
  transMsg.ahandle = NULL;
dengyihao's avatar
dengyihao 已提交
259

dengyihao's avatar
dengyihao 已提交
260 261
  SCliMsg*       pMsg = NULL;
  STransConnCtx* pCtx = NULL;
dengyihao's avatar
dengyihao 已提交
262
  CONN_SHOULD_RELEASE(conn, pHead);
dengyihao's avatar
dengyihao 已提交
263

dengyihao's avatar
dengyihao 已提交
264 265
  if (CONN_NO_PERSIST_BY_APP(conn)) {
    pMsg = transQueuePop(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
266 267 268 269 270 271 272 273 274 275 276
    pCtx = pMsg ? pMsg->ctx : NULL;
    if (pMsg == NULL && !CONN_NO_PERSIST_BY_APP(conn)) {
      transMsg.ahandle = transCtxDumpVal(&conn->ctx, transMsg.msgType);
      if (transMsg.ahandle == NULL) {
        transMsg.ahandle = transCtxDumpBrokenlinkVal(&conn->ctx, (int32_t*)&(transMsg.msgType));
      }
      tDebug("cli conn %p construct ahandle %p, persist: 0", conn, transMsg.ahandle);
    } else {
      transMsg.ahandle = pCtx ? pCtx->ahandle : NULL;
      tDebug("cli conn %p get ahandle %p, persist: 0", conn, transMsg.ahandle);
    }
U
ubuntu 已提交
277
  } else {
dengyihao's avatar
dengyihao 已提交
278 279 280 281
    uint64_t ahandle = (uint64_t)pHead->ahandle;
    CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle);
    if (pMsg == NULL) {
      transMsg.ahandle = transCtxDumpVal(&conn->ctx, transMsg.msgType);
dengyihao's avatar
dengyihao 已提交
282
      tDebug("cli conn %p construct ahandle %p by %s, persist: 1", conn, transMsg.ahandle, TMSG_INFO(transMsg.msgType));
dengyihao's avatar
dengyihao 已提交
283
      if (!CONN_RELEASE_BY_SERVER(conn) && transMsg.ahandle == NULL) {
dengyihao's avatar
dengyihao 已提交
284
        transMsg.code = TSDB_CODE_RPC_NETWORK_UNAVAIL;
dengyihao's avatar
dengyihao 已提交
285
        transMsg.ahandle = transCtxDumpBrokenlinkVal(&conn->ctx, (int32_t*)&(transMsg.msgType));
dengyihao's avatar
dengyihao 已提交
286
        tDebug("cli conn %p construct ahandle %p due brokenlink, persist: 1", conn, transMsg.ahandle);
dengyihao's avatar
dengyihao 已提交
287 288 289 290 291 292
      }
    } else {
      pCtx = pMsg ? pMsg->ctx : NULL;
      transMsg.ahandle = pCtx ? pCtx->ahandle : NULL;
      tDebug("cli conn %p get ahandle %p, persist: 1", conn, transMsg.ahandle);
    }
U
ubuntu 已提交
293
  }
dengyihao's avatar
dengyihao 已提交
294 295 296
  // buf's mem alread translated to transMsg.pCont
  transClearBuffer(&conn->readBuf);

dengyihao's avatar
dengyihao 已提交
297
  if (!CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
298
    transMsg.handle = conn;
U
ubuntu 已提交
299
    tDebug("%s cli conn %p ref by app", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
300 301
  }

dengyihao's avatar
dengyihao 已提交
302 303 304
  tDebug("%s cli conn %p %s received from %s:%d, local info: %s:%d, msg size: %d", pTransInst->label, conn,
         TMSG_INFO(pHead->msgType), taosInetNtoa(conn->addr.sin_addr), ntohs(conn->addr.sin_port),
         taosInetNtoa(conn->locaddr.sin_addr), ntohs(conn->locaddr.sin_port), transMsg.contLen);
dengyihao's avatar
dengyihao 已提交
305

dengyihao's avatar
dengyihao 已提交
306 307 308 309 310
  if (pCtx == NULL && CONN_NO_PERSIST_BY_APP(conn)) {
    tTrace("except, server continue send while cli ignore it");
    // transUnrefCliHandle(conn);
    return;
  }
dengyihao's avatar
dengyihao 已提交
311 312 313 314 315
  if (CONN_RELEASE_BY_SERVER(conn) && transMsg.ahandle == NULL) {
    tTrace("except, server continue send while cli ignore it");
    // transUnrefCliHandle(conn);
    return;
  }
dengyihao's avatar
dengyihao 已提交
316

dengyihao's avatar
dengyihao 已提交
317 318 319 320
  int ret = cliAppCb(conn, &transMsg, pMsg);
  if (ret != 0) {
    tTrace("try to send req to next node");
    return;
dengyihao's avatar
dengyihao 已提交
321
  }
dengyihao's avatar
dengyihao 已提交
322 323
  destroyCmsg(pMsg);

dengyihao's avatar
dengyihao 已提交
324
  if (cliMaySendCachedMsg(conn) == true) {
dengyihao's avatar
dengyihao 已提交
325 326
    return;
  }
dengyihao's avatar
dengyihao 已提交
327

U
ubuntu 已提交
328
  if (CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
329
    addConnToPool(pThrd->pool, conn);
dengyihao's avatar
dengyihao 已提交
330
  }
dengyihao's avatar
test  
dengyihao 已提交
331

dengyihao's avatar
dengyihao 已提交
332
  uv_read_start((uv_stream_t*)conn->stream, cliAllocRecvBufferCb, cliRecvCb);
dengyihao's avatar
dengyihao 已提交
333
  // start thread's timer of conn pool if not active
dengyihao's avatar
dengyihao 已提交
334
  if (!uv_is_active((uv_handle_t*)&pThrd->timer) && pTransInst->idleTime > 0) {
U
ubuntu 已提交
335
    // uv_timer_start((uv_timer_t*)&pThrd->timer, cliTimeoutCb, CONN_PERSIST_TIME(pRpc->idleTime) / 2, 0);
dengyihao's avatar
dengyihao 已提交
336
  }
dengyihao's avatar
dengyihao 已提交
337
}
U
ubuntu 已提交
338 339

void cliHandleExcept(SCliConn* pConn) {
dengyihao's avatar
dengyihao 已提交
340
  if (transQueueEmpty(&pConn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
341
    if (pConn->broken == true && CONN_NO_PERSIST_BY_APP(pConn)) {
dengyihao's avatar
dengyihao 已提交
342
      tTrace("%s cli conn %p handle except, persist:0", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
formate  
dengyihao 已提交
343 344 345
      transUnrefCliHandle(pConn);
      return;
    }
dengyihao's avatar
dengyihao 已提交
346
  }
dengyihao's avatar
dengyihao 已提交
347
  SCliThrdObj* pThrd = pConn->hostThrd;
dengyihao's avatar
formate  
dengyihao 已提交
348
  STrans*      pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
349
  bool         once = false;
D
dapan1121 已提交
350
  do {
dengyihao's avatar
dengyihao 已提交
351
    SCliMsg* pMsg = transQueuePop(&pConn->cliMsgs);
D
dapan1121 已提交
352 353
    if (pMsg == NULL && once) {
      break;
dengyihao's avatar
dengyihao 已提交
354
    }
dengyihao's avatar
dengyihao 已提交
355
    STransConnCtx* pCtx = pMsg ? pMsg->ctx : NULL;
U
ubuntu 已提交
356

dengyihao's avatar
dengyihao 已提交
357 358 359 360
    STransMsg transMsg = {0};
    transMsg.code = TSDB_CODE_RPC_NETWORK_UNAVAIL;
    transMsg.msgType = pMsg ? pMsg->msg.msgType + 1 : 0;
    transMsg.ahandle = NULL;
dengyihao's avatar
dengyihao 已提交
361
    transMsg.handle = pConn;
dengyihao's avatar
dengyihao 已提交
362

dengyihao's avatar
dengyihao 已提交
363
    if (pMsg == NULL && !CONN_NO_PERSIST_BY_APP(pConn)) {
dengyihao's avatar
dengyihao 已提交
364
      transMsg.ahandle = transCtxDumpVal(&pConn->ctx, transMsg.msgType);
dengyihao's avatar
dengyihao 已提交
365 366
      tDebug("%s cli conn %p construct ahandle %p by %s", CONN_GET_INST_LABEL(pConn), pConn, transMsg.ahandle,
             TMSG_INFO(transMsg.msgType));
dengyihao's avatar
dengyihao 已提交
367 368
      if (transMsg.ahandle == NULL) {
        transMsg.ahandle = transCtxDumpBrokenlinkVal(&pConn->ctx, (int32_t*)&(transMsg.msgType));
dengyihao's avatar
dengyihao 已提交
369 370
        tDebug("%s cli conn %p construct ahandle %p due to brokenlink", CONN_GET_INST_LABEL(pConn), pConn,
               transMsg.ahandle);
dengyihao's avatar
dengyihao 已提交
371
      }
dengyihao's avatar
dengyihao 已提交
372 373 374 375 376
    } else {
      transMsg.ahandle = pCtx ? pCtx->ahandle : NULL;
    }

    if (pCtx == NULL || pCtx->pSem == NULL) {
U
ubuntu 已提交
377
      if (transMsg.ahandle == NULL) {
D
dapan1121 已提交
378
        once = true;
U
ubuntu 已提交
379 380
        continue;
      }
dengyihao's avatar
dengyihao 已提交
381 382 383 384 385
    }
    int ret = cliAppCb(pConn, &transMsg, pMsg);
    if (ret != 0) {
      tTrace("try to send req to next node");
      return;
dengyihao's avatar
dengyihao 已提交
386 387 388
    }
    destroyCmsg(pMsg);
    tTrace("%s cli conn %p start to destroy", CONN_GET_INST_LABEL(pConn), pConn);
D
dapan1121 已提交
389
  } while (!transQueueEmpty(&pConn->cliMsgs));
dengyihao's avatar
test  
dengyihao 已提交
390

dengyihao's avatar
dengyihao 已提交
391
  transUnrefCliHandle(pConn);
dengyihao's avatar
dengyihao 已提交
392
}
dengyihao's avatar
dengyihao 已提交
393

U
ubuntu 已提交
394
void cliTimeoutCb(uv_timer_t* handle) {
dengyihao's avatar
dengyihao 已提交
395
  SCliThrdObj* pThrd = handle->data;
dengyihao's avatar
dengyihao 已提交
396
  STrans*      pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
397
  int64_t      currentTime = pThrd->nextTimeout;
dengyihao's avatar
dengyihao 已提交
398
  tTrace("%s, cli conn timeout, try to remove expire conn from conn pool", pTransInst->label);
dengyihao's avatar
dengyihao 已提交
399

dengyihao's avatar
dengyihao 已提交
400
  SConnList* p = taosHashIterate((SHashObj*)pThrd->pool, NULL);
dengyihao's avatar
dengyihao 已提交
401 402 403 404 405 406
  while (p != NULL) {
    while (!QUEUE_IS_EMPTY(&p->conn)) {
      queue*    h = QUEUE_HEAD(&p->conn);
      SCliConn* c = QUEUE_DATA(h, SCliConn, conn);
      if (c->expireTime < currentTime) {
        QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
407
        transUnrefCliHandle(c);
dengyihao's avatar
dengyihao 已提交
408 409 410 411
      } else {
        break;
      }
    }
dengyihao's avatar
dengyihao 已提交
412
    p = taosHashIterate((SHashObj*)pThrd->pool, p);
dengyihao's avatar
dengyihao 已提交
413 414
  }

dengyihao's avatar
dengyihao 已提交
415 416
  pThrd->nextTimeout = taosGetTimestampMs() + CONN_PERSIST_TIME(pTransInst->idleTime);
  uv_timer_start(handle, cliTimeoutCb, CONN_PERSIST_TIME(pTransInst->idleTime) / 2, 0);
dengyihao's avatar
dengyihao 已提交
417
}
U
ubuntu 已提交
418 419

void* createConnPool(int size) {
420 421
  // thread local, no lock
  return taosHashInit(size, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
422
}
U
ubuntu 已提交
423
void* destroyConnPool(void* pool) {
dengyihao's avatar
dengyihao 已提交
424
  SConnList* connList = taosHashIterate((SHashObj*)pool, NULL);
dengyihao's avatar
dengyihao 已提交
425 426 427 428 429
  while (connList != NULL) {
    while (!QUEUE_IS_EMPTY(&connList->conn)) {
      queue* h = QUEUE_HEAD(&connList->conn);
      QUEUE_REMOVE(h);
      SCliConn* c = QUEUE_DATA(h, SCliConn, conn);
U
ubuntu 已提交
430
      cliDestroyConn(c, true);
dengyihao's avatar
dengyihao 已提交
431
    }
dengyihao's avatar
dengyihao 已提交
432
    connList = taosHashIterate((SHashObj*)pool, connList);
dengyihao's avatar
dengyihao 已提交
433
  }
dengyihao's avatar
dengyihao 已提交
434
  taosHashCleanup(pool);
435
  return NULL;
dengyihao's avatar
dengyihao 已提交
436 437
}

dengyihao's avatar
dengyihao 已提交
438
static SCliConn* getConnFromPool(void* pool, char* ip, uint32_t port) {
dengyihao's avatar
dengyihao 已提交
439
  char key[128] = {0};
dengyihao's avatar
dengyihao 已提交
440
  CONN_CONSTRUCT_HASH_KEY(key, ip, port);
dengyihao's avatar
dengyihao 已提交
441

dengyihao's avatar
dengyihao 已提交
442 443
  SHashObj*  pPool = pool;
  SConnList* plist = taosHashGet(pPool, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
444 445
  if (plist == NULL) {
    SConnList list;
dengyihao's avatar
dengyihao 已提交
446 447
    taosHashPut(pPool, key, strlen(key), (void*)&list, sizeof(list));
    plist = taosHashGet(pPool, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
448
    QUEUE_INIT(&plist->conn);
dengyihao's avatar
dengyihao 已提交
449 450 451 452 453 454 455
  }

  if (QUEUE_IS_EMPTY(&plist->conn)) {
    return NULL;
  }
  queue* h = QUEUE_HEAD(&plist->conn);
  QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
456
  SCliConn* conn = QUEUE_DATA(h, SCliConn, conn);
U
ubuntu 已提交
457
  conn->status = ConnNormal;
dengyihao's avatar
dengyihao 已提交
458 459
  QUEUE_INIT(&conn->conn);
  return conn;
dengyihao's avatar
dengyihao 已提交
460
}
dengyihao's avatar
dengyihao 已提交
461
static void addConnToPool(void* pool, SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
462 463 464
  SCliThrdObj* thrd = conn->hostThrd;
  CONN_HANDLE_THREAD_QUIT(thrd);

dengyihao's avatar
dengyihao 已提交
465 466 467 468
  STrans* pTransInst = ((SCliThrdObj*)conn->hostThrd)->pTransInst;
  conn->expireTime = taosGetTimestampMs() + CONN_PERSIST_TIME(pTransInst->idleTime);
  transCtxCleanup(&conn->ctx);
  transQueueClear(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
469
  conn->status = ConnInPool;
dengyihao's avatar
dengyihao 已提交
470

dengyihao's avatar
dengyihao 已提交
471
  char key[128] = {0};
dengyihao's avatar
dengyihao 已提交
472
  CONN_CONSTRUCT_HASH_KEY(key, conn->ip, conn->port);
U
ubuntu 已提交
473
  tTrace("cli conn %p added to conn pool, read buf cap: %d", conn, conn->readBuf.cap);
dengyihao's avatar
dengyihao 已提交
474

dengyihao's avatar
dengyihao 已提交
475
  SConnList* plist = taosHashGet((SHashObj*)pool, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
476 477 478
  // list already create before
  assert(plist != NULL);
  QUEUE_PUSH(&plist->conn, &conn->conn);
dengyihao's avatar
dengyihao 已提交
479
  assert(!QUEUE_IS_EMPTY(&plist->conn));
dengyihao's avatar
dengyihao 已提交
480
}
dengyihao's avatar
dengyihao 已提交
481
static void cliAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
482
  SCliConn*    conn = handle->data;
dengyihao's avatar
dengyihao 已提交
483
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
484
  transAllocBuffer(pBuf, buf);
dengyihao's avatar
dengyihao 已提交
485
}
U
ubuntu 已提交
486
static void cliRecvCb(uv_stream_t* handle, ssize_t nread, const uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
487
  // impl later
dengyihao's avatar
dengyihao 已提交
488 489 490
  if (handle->data == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
491 492
  SCliConn*    conn = handle->data;
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
493
  if (nread > 0) {
dengyihao's avatar
dengyihao 已提交
494
    pBuf->len += nread;
dengyihao's avatar
dengyihao 已提交
495
    if (transReadComplete(pBuf)) {
U
ubuntu 已提交
496 497
      tTrace("%s cli conn %p read complete", CONN_GET_INST_LABEL(conn), conn);
      cliHandleResp(conn);
dengyihao's avatar
dengyihao 已提交
498
    } else {
U
ubuntu 已提交
499
      tTrace("%s cli conn %p read partial packet, continue to read", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
500
    }
dengyihao's avatar
dengyihao 已提交
501 502
    return;
  }
dengyihao's avatar
dengyihao 已提交
503

504 505
  assert(nread <= 0);
  if (nread == 0) {
dengyihao's avatar
dengyihao 已提交
506 507 508
    // ref http://docs.libuv.org/en/v1.x/stream.html?highlight=uv_read_start#c.uv_read_cb
    // nread might be 0, which does not indicate an error or EOF. This is equivalent to EAGAIN or EWOULDBLOCK under
    // read(2).
dengyihao's avatar
dengyihao 已提交
509
    tTrace("%s cli conn %p read empty", CONN_GET_INST_LABEL(conn), conn);
510 511
    return;
  }
dengyihao's avatar
fix bug  
dengyihao 已提交
512
  if (nread < 0) {
U
ubuntu 已提交
513
    tError("%s cli conn %p read error: %s", CONN_GET_INST_LABEL(conn), conn, uv_err_name(nread));
dengyihao's avatar
dengyihao 已提交
514
    conn->broken = true;
U
ubuntu 已提交
515
    cliHandleExcept(conn);
dengyihao's avatar
dengyihao 已提交
516
  }
dengyihao's avatar
dengyihao 已提交
517
}
dengyihao's avatar
dengyihao 已提交
518

U
ubuntu 已提交
519
static SCliConn* cliCreateConn(SCliThrdObj* pThrd) {
wafwerar's avatar
wafwerar 已提交
520
  SCliConn* conn = taosMemoryCalloc(1, sizeof(SCliConn));
dengyihao's avatar
dengyihao 已提交
521
  // read/write stream handle
wafwerar's avatar
wafwerar 已提交
522
  conn->stream = (uv_stream_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
523 524 525 526 527
  uv_tcp_init(pThrd->loop, (uv_tcp_t*)(conn->stream));
  conn->stream->data = conn;

  conn->writeReq.data = conn;
  conn->connReq.data = conn;
dengyihao's avatar
dengyihao 已提交
528 529

  transQueueInit(&conn->cliMsgs, NULL);
dengyihao's avatar
dengyihao 已提交
530 531
  QUEUE_INIT(&conn->conn);
  conn->hostThrd = pThrd;
dengyihao's avatar
dengyihao 已提交
532 533
  conn->status = ConnNormal;
  conn->broken = 0;
dengyihao's avatar
dengyihao 已提交
534 535 536
  transRefCliHandle(conn);
  return conn;
}
U
ubuntu 已提交
537 538
static void cliDestroyConn(SCliConn* conn, bool clear) {
  tTrace("%s cli conn %p remove from conn pool", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
539

dengyihao's avatar
dengyihao 已提交
540 541
  QUEUE_REMOVE(&conn->conn);
  if (clear) {
U
ubuntu 已提交
542
    uv_close((uv_handle_t*)conn->stream, cliDestroy);
543
  }
dengyihao's avatar
dengyihao 已提交
544
}
U
ubuntu 已提交
545
static void cliDestroy(uv_handle_t* handle) {
dengyihao's avatar
dengyihao 已提交
546
  SCliConn* conn = handle->data;
wafwerar's avatar
wafwerar 已提交
547 548
  taosMemoryFree(conn->ip);
  taosMemoryFree(conn->stream);
dengyihao's avatar
dengyihao 已提交
549 550
  transCtxCleanup(&conn->ctx);
  transQueueDestroy(&conn->cliMsgs);
U
ubuntu 已提交
551
  tTrace("%s cli conn %p destroy successfully", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
552
  transDestroyBuffer(&conn->readBuf);
wafwerar's avatar
wafwerar 已提交
553
  taosMemoryFree(conn);
dengyihao's avatar
dengyihao 已提交
554
}
dengyihao's avatar
dengyihao 已提交
555
static bool cliHandleNoResp(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
556 557
  bool res = false;
  if (!transQueueEmpty(&conn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
558
    SCliMsg* pMsg = transQueueGet(&conn->cliMsgs, 0);
dengyihao's avatar
dengyihao 已提交
559
    if (REQUEST_NO_RESP(&pMsg->msg)) {
dengyihao's avatar
dengyihao 已提交
560 561
      transQueuePop(&conn->cliMsgs);
      // taosArrayRemove(msgs, 0);
dengyihao's avatar
dengyihao 已提交
562 563 564 565 566
      destroyCmsg(pMsg);
      res = true;
    }
    if (res == true) {
      if (cliMaySendCachedMsg(conn) == false) {
dengyihao's avatar
dengyihao 已提交
567 568
        SCliThrdObj* thrd = conn->hostThrd;
        addConnToPool(thrd->pool, conn);
dengyihao's avatar
dengyihao 已提交
569 570 571 572 573
      }
    }
  }
  return res;
}
U
ubuntu 已提交
574
static void cliSendCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
575
  SCliConn* pConn = req->data;
dengyihao's avatar
dengyihao 已提交
576

dengyihao's avatar
dengyihao 已提交
577
  if (status == 0) {
U
ubuntu 已提交
578
    tTrace("%s cli conn %p data already was written out", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
579
  } else {
U
ubuntu 已提交
580 581
    tError("%s cli conn %p failed to write: %s", CONN_GET_INST_LABEL(pConn), pConn, uv_err_name(status));
    cliHandleExcept(pConn);
dengyihao's avatar
dengyihao 已提交
582 583
    return;
  }
dengyihao's avatar
dengyihao 已提交
584 585 586 587
  if (cliHandleNoResp(pConn) == true) {
    tTrace("%s cli conn %p no resp required", CONN_GET_INST_LABEL(pConn), pConn);
    return;
  }
dengyihao's avatar
dengyihao 已提交
588
  uv_read_start((uv_stream_t*)pConn->stream, cliAllocRecvBufferCb, cliRecvCb);
dengyihao's avatar
dengyihao 已提交
589
}
dengyihao's avatar
dengyihao 已提交
590

U
ubuntu 已提交
591
void cliSend(SCliConn* pConn) {
dengyihao's avatar
dengyihao 已提交
592 593
  CONN_HANDLE_BROKEN(pConn);

dengyihao's avatar
dengyihao 已提交
594 595
  // assert(taosArrayGetSize(pConn->cliMsgs) > 0);
  assert(!transQueueEmpty(&pConn->cliMsgs));
dengyihao's avatar
dengyihao 已提交
596 597

  SCliMsg* pCliMsg = NULL;
U
ubuntu 已提交
598
  CONN_GET_NEXT_SENDMSG(pConn);
dengyihao's avatar
dengyihao 已提交
599 600
  pCliMsg->sent = 1;

dengyihao's avatar
dengyihao 已提交
601
  STransConnCtx* pCtx = pCliMsg->ctx;
dengyihao's avatar
dengyihao 已提交
602 603

  SCliThrdObj* pThrd = pConn->hostThrd;
dengyihao's avatar
formate  
dengyihao 已提交
604
  STrans*      pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
605

U
ubuntu 已提交
606
  STransMsg* pMsg = (STransMsg*)(&pCliMsg->msg);
dengyihao's avatar
dengyihao 已提交
607 608 609 610
  if (pMsg->pCont == 0) {
    pMsg->pCont = (void*)rpcMallocCont(0);
    pMsg->contLen = 0;
  }
dengyihao's avatar
dengyihao 已提交
611
  int msgLen = transMsgLenFromCont(pMsg->contLen);
dengyihao's avatar
dengyihao 已提交
612

dengyihao's avatar
dengyihao 已提交
613 614
  STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
  pHead->ahandle = pCtx != NULL ? (uint64_t)pCtx->ahandle : 0;
dengyihao's avatar
dengyihao 已提交
615 616
  pHead->noResp = REQUEST_NO_RESP(pMsg) ? 1 : 0;
  pHead->persist = REQUEST_PERSIS_HANDLE(pMsg) ? 1 : 0;
dengyihao's avatar
dengyihao 已提交
617 618
  pHead->msgType = pMsg->msgType;
  pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
dengyihao's avatar
dengyihao 已提交
619
  pHead->release = REQUEST_RELEASE_HANDLE(pCliMsg) ? 1 : 0;
dengyihao's avatar
dengyihao 已提交
620
  memcpy(pHead->user, pTransInst->user, strlen(pTransInst->user));
dengyihao's avatar
dengyihao 已提交
621 622

  uv_buf_t wb = uv_buf_init((char*)pHead, msgLen);
dengyihao's avatar
dengyihao 已提交
623 624 625
  tDebug("%s cli conn %p %s is send to %s:%d, local info %s:%d", CONN_GET_INST_LABEL(pConn), pConn,
         TMSG_INFO(pHead->msgType), taosInetNtoa(pConn->addr.sin_addr), ntohs(pConn->addr.sin_port),
         taosInetNtoa(pConn->locaddr.sin_addr), ntohs(pConn->locaddr.sin_port));
dengyihao's avatar
dengyihao 已提交
626

dengyihao's avatar
dengyihao 已提交
627 628 629
  if (pHead->persist == 1) {
    CONN_SET_PERSIST_BY_APP(pConn);
  }
dengyihao's avatar
dengyihao 已提交
630

dengyihao's avatar
dengyihao 已提交
631
  pConn->writeReq.data = pConn;
U
ubuntu 已提交
632
  uv_write(&pConn->writeReq, (uv_stream_t*)pConn->stream, &wb, 1, cliSendCb);
dengyihao's avatar
dengyihao 已提交
633

U
ubuntu 已提交
634 635
  return;
_RETURN:
dengyihao's avatar
dengyihao 已提交
636
  return;
dengyihao's avatar
dengyihao 已提交
637
}
U
ubuntu 已提交
638 639

void cliConnCb(uv_connect_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
640
  // impl later
dengyihao's avatar
dengyihao 已提交
641 642
  SCliConn* pConn = req->data;
  if (status != 0) {
U
ubuntu 已提交
643 644
    tError("%s cli conn %p failed to connect server: %s", CONN_GET_INST_LABEL(pConn), pConn, uv_strerror(status));
    cliHandleExcept(pConn);
645
    return;
dengyihao's avatar
dengyihao 已提交
646
  }
dengyihao's avatar
dengyihao 已提交
647 648 649
  int addrlen = sizeof(pConn->addr);
  uv_tcp_getpeername((uv_tcp_t*)pConn->stream, (struct sockaddr*)&pConn->addr, &addrlen);

dengyihao's avatar
dengyihao 已提交
650 651 652
  addrlen = sizeof(pConn->locaddr);
  uv_tcp_getsockname((uv_tcp_t*)pConn->stream, (struct sockaddr*)&pConn->locaddr, &addrlen);

U
ubuntu 已提交
653
  tTrace("%s cli conn %p connect to server successfully", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
654
  assert(pConn->stream == req->handle);
dengyihao's avatar
dengyihao 已提交
655

U
ubuntu 已提交
656
  cliSend(pConn);
dengyihao's avatar
dengyihao 已提交
657 658
}

U
ubuntu 已提交
659 660
static void cliHandleQuit(SCliMsg* pMsg, SCliThrdObj* pThrd) {
  tDebug("cli work thread %p start to quit", pThrd);
dengyihao's avatar
dengyihao 已提交
661
  destroyCmsg(pMsg);
dengyihao's avatar
fix bug  
dengyihao 已提交
662
  destroyConnPool(pThrd->pool);
dengyihao's avatar
dengyihao 已提交
663
  uv_timer_stop(&pThrd->timer);
dengyihao's avatar
dengyihao 已提交
664 665
  uv_walk(pThrd->loop, cliWalkCb, NULL);

dengyihao's avatar
dengyihao 已提交
666
  pThrd->quit = true;
dengyihao's avatar
dengyihao 已提交
667

dengyihao's avatar
dengyihao 已提交
668
  // uv_stop(pThrd->loop);
dengyihao's avatar
dengyihao 已提交
669
}
dengyihao's avatar
dengyihao 已提交
670 671
static void cliHandleRelease(SCliMsg* pMsg, SCliThrdObj* pThrd) {
  SCliConn* conn = pMsg->msg.handle;
dengyihao's avatar
dengyihao 已提交
672
  tDebug("%s cli conn %p start to release to inst", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
673

dengyihao's avatar
dengyihao 已提交
674 675
  if (T_REF_VAL_GET(conn) == 2) {
    transUnrefCliHandle(conn);
dengyihao's avatar
dengyihao 已提交
676 677
    if (!transQueuePush(&conn->cliMsgs, pMsg)) {
      return;
dengyihao's avatar
dengyihao 已提交
678 679 680 681 682
    }
    cliSend(conn);
  } else {
    // conn already broken down
    transUnrefCliHandle(conn);
dengyihao's avatar
dengyihao 已提交
683 684
  }
}
U
ubuntu 已提交
685 686

SCliConn* cliGetConn(SCliMsg* pMsg, SCliThrdObj* pThrd) {
dengyihao's avatar
dengyihao 已提交
687
  SCliConn* conn = NULL;
dengyihao's avatar
dengyihao 已提交
688 689
  if (pMsg->msg.handle != NULL) {
    conn = (SCliConn*)(pMsg->msg.handle);
dengyihao's avatar
dengyihao 已提交
690
    if (conn != NULL) {
U
ubuntu 已提交
691
      tTrace("%s cli conn %p reused", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
692 693
    }
  } else {
dengyihao's avatar
dengyihao 已提交
694
    STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
dengyihao 已提交
695
    conn = getConnFromPool(pThrd->pool, EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet));
dengyihao's avatar
dengyihao 已提交
696 697
    if (conn != NULL) {
      tTrace("%s cli conn %p get from conn pool", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
698 699
    } else {
      tTrace("not found conn in conn pool %p", pThrd->pool);
dengyihao's avatar
dengyihao 已提交
700
    }
dengyihao's avatar
dengyihao 已提交
701
  }
dengyihao's avatar
dengyihao 已提交
702 703
  return conn;
}
U
ubuntu 已提交
704 705

void cliHandleReq(SCliMsg* pMsg, SCliThrdObj* pThrd) {
dengyihao's avatar
dengyihao 已提交
706 707
  uint64_t et = taosGetTimestampUs();
  uint64_t el = et - pMsg->st;
dengyihao's avatar
dengyihao 已提交
708 709
  tTrace("%s cli msg tran time cost: %" PRIu64 "us, threadID: %" PRId64 "", ((STrans*)pThrd->pTransInst)->label, el,
         pThrd->thread);
dengyihao's avatar
dengyihao 已提交
710 711

  STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
formate  
dengyihao 已提交
712
  STrans*        pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
713

U
ubuntu 已提交
714
  SCliConn* conn = cliGetConn(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
715
  if (conn != NULL) {
dengyihao's avatar
dengyihao 已提交
716
    conn->hThrdIdx = pCtx->hThrdIdx;
dengyihao's avatar
dengyihao 已提交
717

dengyihao's avatar
dengyihao 已提交
718
    transCtxMerge(&conn->ctx, &pCtx->appCtx);
dengyihao's avatar
dengyihao 已提交
719
    transQueuePush(&conn->cliMsgs, pMsg);
U
ubuntu 已提交
720
    cliSend(conn);
dengyihao's avatar
dengyihao 已提交
721
  } else {
U
ubuntu 已提交
722
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
723
    transCtxMerge(&conn->ctx, &pCtx->appCtx);
dengyihao's avatar
dengyihao 已提交
724
    transQueuePush(&conn->cliMsgs, pMsg);
dengyihao's avatar
dengyihao 已提交
725

dengyihao's avatar
dengyihao 已提交
726
    conn->hThrdIdx = pCtx->hThrdIdx;
dengyihao's avatar
dengyihao 已提交
727 728
    conn->ip = strdup(EPSET_GET_INUSE_IP(&pCtx->epSet));
    conn->port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
729

dengyihao's avatar
dengyihao 已提交
730 731
    int ret = transSetConnOption((uv_tcp_t*)conn->stream);
    if (ret) {
U
ubuntu 已提交
732
      tError("%s cli conn %p failed to set conn option, errmsg %s", pTransInst->label, conn, uv_err_name(ret));
dengyihao's avatar
dengyihao 已提交
733
    }
dengyihao's avatar
dengyihao 已提交
734 735 736 737 738 739 740
    int fd = taosCreateSocketWithTimeOutOpt(TRANS_CONN_TIMEOUT);
    if (fd == -1) {
      tTrace("%s cli conn %p failed to create socket", pTransInst->label, conn);
      cliHandleExcept(conn);
      return;
    }
    uv_tcp_open((uv_tcp_t*)conn->stream, fd);
741

dengyihao's avatar
dengyihao 已提交
742
    struct sockaddr_in addr;
743 744 745
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = taosGetIpv4FromFqdn(conn->ip);
    addr.sin_port = (uint16_t)htons((uint16_t)conn->port);
dengyihao's avatar
dengyihao 已提交
746 747 748 749 750 751 752 753
    tTrace("%s cli conn %p try to connect to %s:%d", pTransInst->label, conn, conn->ip, conn->port);
    ret = uv_tcp_connect(&conn->connReq, (uv_tcp_t*)(conn->stream), (const struct sockaddr*)&addr, cliConnCb);
    if (ret != 0) {
      tTrace("%s cli conn %p failed to connect to %s:%d, reason: %s", pTransInst->label, conn, conn->ip, conn->port,
             uv_err_name(ret));
      cliHandleExcept(conn);
      return;
    }
dengyihao's avatar
dengyihao 已提交
754 755
  }
}
U
ubuntu 已提交
756
static void cliAsyncCb(uv_async_t* handle) {
dengyihao's avatar
dengyihao 已提交
757 758
  SAsyncItem*  item = handle->data;
  SCliThrdObj* pThrd = item->pThrd;
dengyihao's avatar
dengyihao 已提交
759
  SCliMsg*     pMsg = NULL;
dengyihao's avatar
dengyihao 已提交
760

dengyihao's avatar
dengyihao 已提交
761
  // batch process to avoid to lock/unlock frequently
dengyihao's avatar
formate  
dengyihao 已提交
762
  queue wq;
wafwerar's avatar
wafwerar 已提交
763
  taosThreadMutexLock(&item->mtx);
dengyihao's avatar
dengyihao 已提交
764
  QUEUE_MOVE(&item->qmsg, &wq);
wafwerar's avatar
wafwerar 已提交
765
  taosThreadMutexUnlock(&item->mtx);
dengyihao's avatar
dengyihao 已提交
766

dengyihao's avatar
dengyihao 已提交
767 768 769 770
  int count = 0;
  while (!QUEUE_IS_EMPTY(&wq)) {
    queue* h = QUEUE_HEAD(&wq);
    QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
771 772

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
773 774
    if (pMsg == NULL) {
      continue;
dengyihao's avatar
dengyihao 已提交
775
    }
dengyihao's avatar
dengyihao 已提交
776
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
777
    count++;
dengyihao's avatar
dengyihao 已提交
778 779
  }
  if (count >= 2) {
U
ubuntu 已提交
780
    tTrace("cli process batch size: %d", count);
dengyihao's avatar
dengyihao 已提交
781
  }
dengyihao's avatar
dengyihao 已提交
782 783
}

U
ubuntu 已提交
784
static void* cliWorkThread(void* arg) {
dengyihao's avatar
dengyihao 已提交
785
  SCliThrdObj* pThrd = (SCliThrdObj*)arg;
U
ubuntu 已提交
786
  setThreadName("trans-cli-work");
dengyihao's avatar
dengyihao 已提交
787
  uv_run(pThrd->loop, UV_RUN_DEFAULT);
788
  return NULL;
dengyihao's avatar
dengyihao 已提交
789 790
}

U
ubuntu 已提交
791
void* transInitClient(uint32_t ip, uint32_t port, char* label, int numOfThreads, void* fp, void* shandle) {
wafwerar's avatar
wafwerar 已提交
792
  SCliObj* cli = taosMemoryCalloc(1, sizeof(SCliObj));
dengyihao's avatar
dengyihao 已提交
793

dengyihao's avatar
dengyihao 已提交
794
  STrans* pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
795 796
  memcpy(cli->label, label, strlen(label));
  cli->numOfThreads = numOfThreads;
wafwerar's avatar
wafwerar 已提交
797
  cli->pThreadObj = (SCliThrdObj**)taosMemoryCalloc(cli->numOfThreads, sizeof(SCliThrdObj*));
dengyihao's avatar
dengyihao 已提交
798 799

  for (int i = 0; i < cli->numOfThreads; i++) {
dengyihao's avatar
dengyihao 已提交
800
    SCliThrdObj* pThrd = createThrdObj();
dengyihao's avatar
dengyihao 已提交
801
    pThrd->nextTimeout = taosGetTimestampMs() + CONN_PERSIST_TIME(pTransInst->idleTime);
dengyihao's avatar
dengyihao 已提交
802
    pThrd->pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
803

wafwerar's avatar
wafwerar 已提交
804
    int err = taosThreadCreate(&pThrd->thread, NULL, cliWorkThread, (void*)(pThrd));
dengyihao's avatar
dengyihao 已提交
805
    if (err == 0) {
U
ubuntu 已提交
806
      tDebug("success to create tranport-cli thread %d", i);
dengyihao's avatar
dengyihao 已提交
807
    }
dengyihao's avatar
dengyihao 已提交
808
    cli->pThreadObj[i] = pThrd;
dengyihao's avatar
dengyihao 已提交
809 810 811
  }
  return cli;
}
dengyihao's avatar
dengyihao 已提交
812

U
ubuntu 已提交
813
static void destroyUserdata(STransMsg* userdata) {
dengyihao's avatar
dengyihao 已提交
814 815 816 817 818 819 820 821 822 823 824 825
  if (userdata->pCont == NULL) {
    return;
  }
  transFreeMsg(userdata->pCont);
  userdata->pCont = NULL;
}
static void destroyCmsg(SCliMsg* pMsg) {
  if (pMsg == NULL) {
    return;
  }
  transDestroyConnCtx(pMsg->ctx);
  destroyUserdata(&pMsg->msg);
wafwerar's avatar
wafwerar 已提交
826
  taosMemoryFree(pMsg);
dengyihao's avatar
dengyihao 已提交
827
}
dengyihao's avatar
dengyihao 已提交
828

dengyihao's avatar
dengyihao 已提交
829
static SCliThrdObj* createThrdObj() {
wafwerar's avatar
wafwerar 已提交
830
  SCliThrdObj* pThrd = (SCliThrdObj*)taosMemoryCalloc(1, sizeof(SCliThrdObj));
U
ubuntu 已提交
831

dengyihao's avatar
dengyihao 已提交
832
  QUEUE_INIT(&pThrd->msg);
wafwerar's avatar
wafwerar 已提交
833
  taosThreadMutexInit(&pThrd->msgMtx, NULL);
dengyihao's avatar
dengyihao 已提交
834

wafwerar's avatar
wafwerar 已提交
835
  pThrd->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
836 837
  uv_loop_init(pThrd->loop);

U
ubuntu 已提交
838
  pThrd->asyncPool = transCreateAsyncPool(pThrd->loop, 5, pThrd, cliAsyncCb);
dengyihao's avatar
dengyihao 已提交
839 840
  uv_timer_init(pThrd->loop, &pThrd->timer);
  pThrd->timer.data = pThrd;
dengyihao's avatar
dengyihao 已提交
841

dengyihao's avatar
dengyihao 已提交
842
  pThrd->pool = createConnPool(4);
dengyihao's avatar
dengyihao 已提交
843

dengyihao's avatar
dengyihao 已提交
844
  transDQCreate(pThrd->loop, &pThrd->delayQueue);
dengyihao's avatar
dengyihao 已提交
845

dengyihao's avatar
dengyihao 已提交
846
  pThrd->quit = false;
dengyihao's avatar
dengyihao 已提交
847 848
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
849
static void destroyThrdObj(SCliThrdObj* pThrd) {
dengyihao's avatar
dengyihao 已提交
850 851 852
  if (pThrd == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
853

wafwerar's avatar
wafwerar 已提交
854
  taosThreadJoin(pThrd->thread, NULL);
dengyihao's avatar
dengyihao 已提交
855
  CLI_RELEASE_UV(pThrd->loop);
wafwerar's avatar
wafwerar 已提交
856
  taosThreadMutexDestroy(&pThrd->msgMtx);
dengyihao's avatar
dengyihao 已提交
857
  transDestroyAsyncPool(pThrd->asyncPool);
dengyihao's avatar
dengyihao 已提交
858

dengyihao's avatar
dengyihao 已提交
859
  transDQDestroy(pThrd->delayQueue);
wafwerar's avatar
wafwerar 已提交
860 861
  taosMemoryFree(pThrd->loop);
  taosMemoryFree(pThrd);
dengyihao's avatar
dengyihao 已提交
862
}
dengyihao's avatar
dengyihao 已提交
863

dengyihao's avatar
dengyihao 已提交
864
static void transDestroyConnCtx(STransConnCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
865
  //
wafwerar's avatar
wafwerar 已提交
866
  taosMemoryFree(ctx);
dengyihao's avatar
dengyihao 已提交
867
}
dengyihao's avatar
dengyihao 已提交
868

U
ubuntu 已提交
869
void cliSendQuit(SCliThrdObj* thrd) {
dengyihao's avatar
dengyihao 已提交
870
  // cli can stop gracefully
wafwerar's avatar
wafwerar 已提交
871
  SCliMsg* msg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
872
  msg->type = Quit;
dengyihao's avatar
dengyihao 已提交
873
  transSendAsync(thrd->asyncPool, &msg->q);
dengyihao's avatar
dengyihao 已提交
874
}
dengyihao's avatar
dengyihao 已提交
875 876 877 878 879
void cliWalkCb(uv_handle_t* handle, void* arg) {
  if (!uv_is_closing(handle)) {
    uv_close(handle, NULL);
  }
}
dengyihao's avatar
dengyihao 已提交
880

U
ubuntu 已提交
881 882 883 884 885 886 887
int cliRBChoseIdx(STrans* pTransInst) {
  int64_t index = pTransInst->index;
  if (pTransInst->index++ >= pTransInst->numOfThreads) {
    pTransInst->index = 0;
  }
  return index % pTransInst->numOfThreads;
}
dengyihao's avatar
dengyihao 已提交
888 889 890 891 892 893 894 895 896
static void doDelayTask(void* param) {
  STaskArg* arg = param;

  SCliMsg*     pMsg = arg->param1;
  SCliThrdObj* pThrd = arg->param2;
  cliHandleReq(pMsg, pThrd);

  taosMemoryFree(arg);
}
dengyihao's avatar
dengyihao 已提交
897
int cliAppCb(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
898 899 900
  SCliThrdObj* pThrd = pConn->hostThrd;
  STrans*      pTransInst = pThrd->pTransInst;

dengyihao's avatar
dengyihao 已提交
901 902 903 904 905 906 907 908
  if (pMsg == NULL || pMsg->ctx == NULL) {
    tTrace("%s cli conn %p handle resp", pTransInst->label, pConn);
    pTransInst->cfp(pTransInst->parent, pResp, NULL);
    return 0;
  }

  STransConnCtx* pCtx = pMsg->ctx;
  SEpSet*        pEpSet = &pCtx->epSet;
dengyihao's avatar
dengyihao 已提交
909 910 911
  /*
   * upper layer handle retry if code equal TSDB_CODE_RPC_NETWORK_UNAVAIL
   */
dengyihao's avatar
dengyihao 已提交
912 913
  tmsg_t msgType = pCtx->msgType;
  if ((pTransInst->retry != NULL && (pTransInst->retry(pResp->code))) ||
dengyihao's avatar
dengyihao 已提交
914
      ((pResp->code == TSDB_CODE_RPC_NETWORK_UNAVAIL) && msgType == TDMT_MND_CONNECT)) {
dengyihao's avatar
dengyihao 已提交
915
    pMsg->sent = 0;
dengyihao's avatar
dengyihao 已提交
916
    pMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
917
    pCtx->retryCount += 1;
dengyihao's avatar
dengyihao 已提交
918
    if (msgType == TDMT_MND_CONNECT && pResp->code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
dengyihao's avatar
dengyihao 已提交
919 920
      if (pCtx->retryCount < pEpSet->numOfEps) {
        pEpSet->inUse = (++pEpSet->inUse) % pEpSet->numOfEps;
dengyihao's avatar
dengyihao 已提交
921 922 923 924

        STaskArg* arg = taosMemoryMalloc(sizeof(STaskArg));
        arg->param1 = pMsg;
        arg->param2 = pThrd;
dengyihao's avatar
dengyihao 已提交
925
        transDQSched(pThrd->delayQueue, doDelayTask, arg, TRANS_RETRY_INTERVAL);
dengyihao's avatar
dengyihao 已提交
926

dengyihao's avatar
dengyihao 已提交
927
        cliDestroy((uv_handle_t*)pConn->stream);
dengyihao's avatar
dengyihao 已提交
928 929 930 931
        return -1;
      }
    } else if (pCtx->retryCount < TRANS_RETRY_COUNT_LIMIT) {
      if (pResp->contLen == 0) {
dengyihao's avatar
dengyihao 已提交
932
        pEpSet->inUse = (++pEpSet->inUse) % pEpSet->numOfEps;
dengyihao's avatar
dengyihao 已提交
933 934 935 936 937
      } else {
        SMEpSet emsg = {0};
        tDeserializeSMEpSet(pResp->pCont, pResp->contLen, &emsg);
        pCtx->epSet = emsg.epSet;
      }
dengyihao's avatar
dengyihao 已提交
938
      addConnToPool(pThrd->pool, pConn);
dengyihao's avatar
dengyihao 已提交
939
      tTrace("use remote epset, current in use: %d, retry count:%d, try limit: %d", pEpSet->inUse, pCtx->retryCount + 1,
dengyihao's avatar
dengyihao 已提交
940 941
             TRANS_RETRY_COUNT_LIMIT);

dengyihao's avatar
dengyihao 已提交
942 943 944
      STaskArg* arg = taosMemoryMalloc(sizeof(STaskArg));
      arg->param1 = pMsg;
      arg->param2 = pThrd;
dengyihao's avatar
dengyihao 已提交
945
      transDQSched(pThrd->delayQueue, doDelayTask, arg, TRANS_RETRY_INTERVAL);
dengyihao's avatar
dengyihao 已提交
946
      return -1;
dengyihao's avatar
dengyihao 已提交
947 948 949 950
    }
  }

  if (pCtx->pSem != NULL) {
951 952 953 954 955 956
    tTrace("%s cli conn %p(sync) handle resp", pTransInst->label, pConn);
    if (pCtx->pRsp == NULL) {
      tTrace("%s cli conn %p(sync) failed to resp, ignore", pTransInst->label, pConn);
    } else {
      memcpy((char*)pCtx->pRsp, (char*)pResp, sizeof(*pResp));
    }
dengyihao's avatar
dengyihao 已提交
957
    tsem_post(pCtx->pSem);
958
    pCtx->pRsp = NULL;
dengyihao's avatar
dengyihao 已提交
959
  } else {
dengyihao's avatar
dengyihao 已提交
960 961
    tTrace("%s cli conn %p handle resp", pTransInst->label, pConn);
    pTransInst->cfp(pTransInst->parent, pResp, pEpSet);
dengyihao's avatar
dengyihao 已提交
962
  }
dengyihao's avatar
dengyihao 已提交
963
  return 0;
dengyihao's avatar
dengyihao 已提交
964
}
U
ubuntu 已提交
965 966

void transCloseClient(void* arg) {
U
ubuntu 已提交
967
  SCliObj* cli = arg;
dengyihao's avatar
dengyihao 已提交
968
  for (int i = 0; i < cli->numOfThreads; i++) {
U
ubuntu 已提交
969
    cliSendQuit(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
970
    destroyThrdObj(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
971
  }
wafwerar's avatar
wafwerar 已提交
972 973
  taosMemoryFree(cli->pThreadObj);
  taosMemoryFree(cli);
dengyihao's avatar
dengyihao 已提交
974
}
dengyihao's avatar
dengyihao 已提交
975 976 977 978 979 980 981 982 983 984 985 986
void transRefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_INC((SCliConn*)handle);
  UNUSED(ref);
}
void transUnrefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_DEC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
987
  tDebug("%s cli conn %p ref %d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
988
  if (ref == 0) {
U
ubuntu 已提交
989
    cliDestroyConn((SCliConn*)handle, true);
dengyihao's avatar
dengyihao 已提交
990 991
  }
}
dengyihao's avatar
dengyihao 已提交
992 993 994 995 996 997 998
void transReleaseCliHandle(void* handle) {
  SCliThrdObj* thrd = CONN_GET_HOST_THREAD(handle);
  if (thrd == NULL) {
    return;
  }

  STransMsg tmsg = {.handle = handle};
wafwerar's avatar
wafwerar 已提交
999
  SCliMsg*  cmsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
1000
  cmsg->msg = tmsg;
dengyihao's avatar
dengyihao 已提交
1001
  cmsg->type = Release;
dengyihao's avatar
dengyihao 已提交
1002 1003 1004

  transSendAsync(thrd->asyncPool, &cmsg->q);
}
dengyihao's avatar
dengyihao 已提交
1005

dengyihao's avatar
dengyihao 已提交
1006
void transSendRequest(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransCtx* ctx) {
U
ubuntu 已提交
1007
  STrans* pTransInst = (STrans*)shandle;
dengyihao's avatar
dengyihao 已提交
1008
  int     index = CONN_HOST_THREAD_INDEX((SCliConn*)pReq->handle);
dengyihao's avatar
dengyihao 已提交
1009
  if (index == -1) {
U
ubuntu 已提交
1010
    index = cliRBChoseIdx(pTransInst);
dengyihao's avatar
dengyihao 已提交
1011
  }
dengyihao's avatar
dengyihao 已提交
1012

wafwerar's avatar
wafwerar 已提交
1013
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
1014 1015 1016
  pCtx->epSet = *pEpSet;
  pCtx->ahandle = pReq->ahandle;
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
1017
  pCtx->hThrdIdx = index;
dengyihao's avatar
dengyihao 已提交
1018 1019 1020 1021

  if (ctx != NULL) {
    pCtx->appCtx = *ctx;
  }
dengyihao's avatar
dengyihao 已提交
1022
  assert(pTransInst->connType == TAOS_CONN_CLIENT);
dengyihao's avatar
dengyihao 已提交
1023

wafwerar's avatar
wafwerar 已提交
1024
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
1025
  cliMsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
1026
  cliMsg->msg = *pReq;
dengyihao's avatar
dengyihao 已提交
1027
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
1028
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
1029

U
ubuntu 已提交
1030
  SCliThrdObj* thrd = ((SCliObj*)pTransInst->tcphandle)->pThreadObj[index];
dengyihao's avatar
dengyihao 已提交
1031

dengyihao's avatar
dengyihao 已提交
1032 1033
  tDebug("send request at thread:%d, threadID: %" PRId64 ",  msg: %p, dst: %s:%d, app:%p", index, thrd->thread, pReq,
         EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet), pReq->ahandle);
L
Liu Jicong 已提交
1034
  ASSERT(transSendAsync(thrd->asyncPool, &(cliMsg->q)) == 0);
dengyihao's avatar
dengyihao 已提交
1035
}
dengyihao's avatar
dengyihao 已提交
1036

dengyihao's avatar
dengyihao 已提交
1037
void transSendRecv(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransMsg* pRsp) {
U
ubuntu 已提交
1038
  STrans* pTransInst = (STrans*)shandle;
dengyihao's avatar
formate  
dengyihao 已提交
1039
  int     index = CONN_HOST_THREAD_INDEX(pReq->handle);
dengyihao's avatar
dengyihao 已提交
1040
  if (index == -1) {
U
ubuntu 已提交
1041
    index = cliRBChoseIdx(pTransInst);
dengyihao's avatar
dengyihao 已提交
1042 1043
  }

wafwerar's avatar
wafwerar 已提交
1044
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
1045
  pCtx->epSet = *pEpSet;
dengyihao's avatar
dengyihao 已提交
1046 1047
  pCtx->ahandle = pReq->ahandle;
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
1048
  pCtx->hThrdIdx = index;
wafwerar's avatar
wafwerar 已提交
1049
  pCtx->pSem = taosMemoryCalloc(1, sizeof(tsem_t));
dengyihao's avatar
dengyihao 已提交
1050 1051 1052
  pCtx->pRsp = pRsp;
  tsem_init(pCtx->pSem, 0, 0);

wafwerar's avatar
wafwerar 已提交
1053
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
1054 1055 1056
  cliMsg->ctx = pCtx;
  cliMsg->msg = *pReq;
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
1057
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
1058

U
ubuntu 已提交
1059
  SCliThrdObj* thrd = ((SCliObj*)pTransInst->tcphandle)->pThreadObj[index];
dengyihao's avatar
dengyihao 已提交
1060 1061
  tDebug("send request at thread:%d, threadID:%" PRId64 ", msg: %p, dst: %s:%d, app:%p", index, thrd->thread, pReq,
         EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet), pReq->ahandle);
dengyihao's avatar
dengyihao 已提交
1062

dengyihao's avatar
dengyihao 已提交
1063 1064 1065 1066
  transSendAsync(thrd->asyncPool, &(cliMsg->q));
  tsem_t* pSem = pCtx->pSem;
  tsem_wait(pSem);
  tsem_destroy(pSem);
wafwerar's avatar
wafwerar 已提交
1067
  taosMemoryFree(pSem);
dengyihao's avatar
dengyihao 已提交
1068
}
U
ubuntu 已提交
1069

dengyihao's avatar
dengyihao 已提交
1070
#endif