transCli.c 80.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
 *
 * 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/>.
 */

#include "transComm.h"
dengyihao's avatar
dengyihao 已提交
16
#include "tutil.h"
dengyihao's avatar
dengyihao 已提交
17

dengyihao's avatar
dengyihao 已提交
18 19 20 21 22
typedef struct {
  int32_t numOfConn;
  queue   msgQ;
} SMsgList;

dengyihao's avatar
dengyihao 已提交
23
typedef struct SConnList {
dengyihao's avatar
dengyihao 已提交
24 25 26
  queue     conns;
  int32_t   size;
  SMsgList* list;
dengyihao's avatar
dengyihao 已提交
27 28
} SConnList;

dengyihao's avatar
dengyihao 已提交
29
typedef struct {
dengyihao's avatar
dengyihao 已提交
30 31 32 33 34 35
  queue   wq;
  int32_t len;

  int connMax;
  int connCnt;
  int batchLenLimit;
dengyihao's avatar
dengyihao 已提交
36
  int sending;
dengyihao's avatar
dengyihao 已提交
37

dengyihao's avatar
dengyihao 已提交
38 39 40
  char*    dst;
  char*    ip;
  uint16_t port;
dengyihao's avatar
dengyihao 已提交
41 42 43 44 45 46 47 48 49 50

} SCliBatchList;

typedef struct {
  queue          wq;
  queue          listq;
  int32_t        wLen;
  int32_t        batchSize;  //
  int32_t        batch;
  SCliBatchList* pList;
dengyihao's avatar
dengyihao 已提交
51
} SCliBatch;
dengyihao's avatar
dengyihao 已提交
52

dengyihao's avatar
dengyihao 已提交
53
typedef struct SCliConn {
dengyihao's avatar
dengyihao 已提交
54
  T_REF_DECLARE()
dengyihao's avatar
dengyihao 已提交
55 56
  uv_connect_t connReq;
  uv_stream_t* stream;
dengyihao's avatar
dengyihao 已提交
57
  queue        wreqQueue;
dengyihao's avatar
dengyihao 已提交
58 59

  uv_timer_t* timer;  // read timer, forbidden
dengyihao's avatar
dengyihao 已提交
60

dengyihao's avatar
dengyihao 已提交
61 62 63 64
  void* hostThrd;

  SConnBuffer readBuf;
  STransQueue cliMsgs;
dengyihao's avatar
dengyihao 已提交
65 66 67

  queue      q;
  SConnList* list;
dengyihao's avatar
dengyihao 已提交
68 69

  STransCtx  ctx;
dengyihao's avatar
dengyihao 已提交
70 71
  bool       broken;  // link broken or not
  ConnStatus status;  //
dengyihao's avatar
dengyihao 已提交
72

dengyihao's avatar
dengyihao 已提交
73 74
  SCliBatch* pBatch;

dengyihao's avatar
dengyihao 已提交
75
  SDelayTask* task;
dengyihao's avatar
dengyihao 已提交
76

dengyihao's avatar
dengyihao 已提交
77 78 79
  char* ip;
  char  src[32];
  char  dst[32];
dengyihao's avatar
dengyihao 已提交
80

dengyihao's avatar
dengyihao 已提交
81
  int64_t refId;
dengyihao's avatar
dengyihao 已提交
82
} SCliConn;
dengyihao's avatar
dengyihao 已提交
83

dengyihao's avatar
dengyihao 已提交
84
typedef struct SCliMsg {
dengyihao's avatar
dengyihao 已提交
85
  STransConnCtx* ctx;
dengyihao's avatar
formate  
dengyihao 已提交
86
  STransMsg      msg;
dengyihao's avatar
dengyihao 已提交
87
  queue          q;
dengyihao's avatar
dengyihao 已提交
88
  STransMsgType  type;
dengyihao's avatar
dengyihao 已提交
89

dengyihao's avatar
dengyihao 已提交
90
  int64_t  refId;
dengyihao's avatar
dengyihao 已提交
91 92
  uint64_t st;
  int      sent;  //(0: no send, 1: alread sent)
dengyihao's avatar
dengyihao 已提交
93 94
} SCliMsg;

dengyihao's avatar
dengyihao 已提交
95
typedef struct SCliThrd {
dengyihao's avatar
dengyihao 已提交
96 97 98 99 100 101
  TdThread      thread;  // tid
  int64_t       pid;     // pid
  uv_loop_t*    loop;
  SAsyncPool*   asyncPool;
  uv_prepare_t* prepare;
  void*         pool;  // conn pool
dengyihao's avatar
dengyihao 已提交
102
  // timer handles
dengyihao's avatar
dengyihao 已提交
103
  SArray* timerList;
dengyihao's avatar
dengyihao 已提交
104
  // msg queue
dengyihao's avatar
dengyihao 已提交
105
  queue         msg;
wafwerar's avatar
wafwerar 已提交
106
  TdThreadMutex msgMtx;
dengyihao's avatar
dengyihao 已提交
107
  SDelayQueue*  delayQueue;
dengyihao's avatar
dengyihao 已提交
108
  SDelayQueue*  timeoutQueue;
dengyihao's avatar
dengyihao 已提交
109
  SDelayQueue*  waitConnQueue;
dengyihao's avatar
dengyihao 已提交
110 111
  uint64_t      nextTimeout;  // next timeout
  void*         pTransInst;   //
dengyihao's avatar
dengyihao 已提交
112

dengyihao's avatar
dengyihao 已提交
113
  int connCount;
dengyihao's avatar
dengyihao 已提交
114
  void (*destroyAhandleFp)(void* ahandle);
dengyihao's avatar
dengyihao 已提交
115 116
  SHashObj* fqdn2ipCache;
  SCvtAddr  cvtAddr;
dengyihao's avatar
dengyihao 已提交
117

dengyihao's avatar
dengyihao 已提交
118
  SHashObj* failFastCache;
dengyihao's avatar
dengyihao 已提交
119
  SHashObj* batchCache;
dengyihao's avatar
dengyihao 已提交
120

dengyihao's avatar
dengyihao 已提交
121 122
  SCliMsg* stopMsg;

dengyihao's avatar
dengyihao 已提交
123
  bool quit;
dengyihao's avatar
dengyihao 已提交
124
} SCliThrd;
dengyihao's avatar
dengyihao 已提交
125

U
ubuntu 已提交
126
typedef struct SCliObj {
dengyihao's avatar
dengyihao 已提交
127 128 129 130
  char       label[TSDB_LABEL_LEN];
  int32_t    index;
  int        numOfThreads;
  SCliThrd** pThreadObj;
U
ubuntu 已提交
131
} SCliObj;
dengyihao's avatar
dengyihao 已提交
132

dengyihao's avatar
dengyihao 已提交
133 134 135 136 137 138 139
typedef struct {
  int32_t reinit;
  int64_t timestamp;
  int32_t count;
  int32_t threshold;
  int64_t interval;
} SFailFastItem;
dengyihao's avatar
dengyihao 已提交
140

dengyihao's avatar
dengyihao 已提交
141
// conn pool
dengyihao's avatar
dengyihao 已提交
142
// add expire timeout and capacity limit
dengyihao's avatar
dengyihao 已提交
143
static void*     createConnPool(int size);
dengyihao's avatar
dengyihao 已提交
144
static void*     destroyConnPool(SCliThrd* thread);
dengyihao's avatar
dengyihao 已提交
145
static SCliConn* getConnFromPool(SCliThrd* thread, char* key, bool* exceed);
dengyihao's avatar
dengyihao 已提交
146
static void      addConnToPool(void* pool, SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
147
static void      doCloseIdleConn(void* param);
dengyihao's avatar
dengyihao 已提交
148

dengyihao's avatar
dengyihao 已提交
149 150
// register conn timer
static void cliConnTimeout(uv_timer_t* handle);
dengyihao's avatar
dengyihao 已提交
151 152
// register timer for read
static void cliReadTimeoutCb(uv_timer_t* handle);
dengyihao's avatar
dengyihao 已提交
153
// register timer in each thread to clear expire conn
dengyihao's avatar
dengyihao 已提交
154
// static void cliTimeoutCb(uv_timer_t* handle);
dengyihao's avatar
dengyihao 已提交
155
// alloc buffer for recv
dengyihao's avatar
dengyihao 已提交
156
static FORCE_INLINE void cliAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf);
dengyihao's avatar
dengyihao 已提交
157
// callback after recv nbytes from socket
U
ubuntu 已提交
158
static void cliRecvCb(uv_stream_t* cli, ssize_t nread, const uv_buf_t* buf);
dengyihao's avatar
dengyihao 已提交
159
// callback after send data to socket
U
ubuntu 已提交
160
static void cliSendCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
161
// callback after conn to server
U
ubuntu 已提交
162 163
static void cliConnCb(uv_connect_t* req, int status);
static void cliAsyncCb(uv_async_t* handle);
dengyihao's avatar
dengyihao 已提交
164
static void cliIdleCb(uv_idle_t* handle);
dengyihao's avatar
dengyihao 已提交
165
static void cliPrepareCb(uv_prepare_t* handle);
dengyihao's avatar
dengyihao 已提交
166

dengyihao's avatar
dengyihao 已提交
167
static void cliHandleBatchReq(SCliBatch* pBatch, SCliThrd* pThrd);
dengyihao's avatar
dengyihao 已提交
168
static void cliSendBatchCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
169 170

SCliBatch* cliGetHeadFromList(SCliBatchList* pList);
dengyihao's avatar
dengyihao 已提交
171

dengyihao's avatar
dengyihao 已提交
172 173
static bool cliRecvReleaseReq(SCliConn* conn, STransMsgHead* pHead);

dengyihao's avatar
dengyihao 已提交
174
static int32_t allocConnRef(SCliConn* conn, bool update);
dengyihao's avatar
dengyihao 已提交
175

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

dengyihao's avatar
dengyihao 已提交
178
static SCliConn* cliCreateConn(SCliThrd* thrd);
U
ubuntu 已提交
179 180
static void      cliDestroyConn(SCliConn* pConn, bool clear /*clear tcp handle or not*/);
static void      cliDestroy(uv_handle_t* handle);
dengyihao's avatar
dengyihao 已提交
181
static void      cliSend(SCliConn* pConn);
dengyihao's avatar
dengyihao 已提交
182
static void      cliSendBatch(SCliConn* pConn);
dengyihao's avatar
dengyihao 已提交
183
static void      cliDestroyConnMsgs(SCliConn* conn, bool destroy);
dengyihao's avatar
dengyihao 已提交
184

dengyihao's avatar
dengyihao 已提交
185 186
static void    doFreeTimeoutMsg(void* param);
static int32_t cliPreCheckSessionLimitForMsg(SCliThrd* pThrd, char* addr, SCliMsg** pMsg);
dengyihao's avatar
dengyihao 已提交
187

dengyihao's avatar
dengyihao 已提交
188
// cli util func
dengyihao's avatar
dengyihao 已提交
189 190
static FORCE_INLINE bool cliIsEpsetUpdated(int32_t code, STransConnCtx* pCtx);
static FORCE_INLINE void cliMayCvtFqdnToIp(SEpSet* pEpSet, SCvtAddr* pCvtAddr);
dengyihao's avatar
dengyihao 已提交
191

dengyihao's avatar
dengyihao 已提交
192
static FORCE_INLINE int32_t cliBuildExceptResp(SCliMsg* pMsg, STransMsg* resp);
dengyihao's avatar
dengyihao 已提交
193

dengyihao's avatar
dengyihao 已提交
194 195 196
static FORCE_INLINE uint32_t cliGetIpFromFqdnCache(SHashObj* cache, char* fqdn);
static FORCE_INLINE void     cliUpdateFqdnCache(SHashObj* cache, char* fqdn);

dengyihao's avatar
dengyihao 已提交
197
// process data read from server, add decompress etc later
U
ubuntu 已提交
198
static void cliHandleResp(SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
199
// handle except about conn
U
ubuntu 已提交
200
static void cliHandleExcept(SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
201
static void cliReleaseUnfinishedMsg(SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
202
static void cliHandleFastFail(SCliConn* pConn, int status);
dengyihao's avatar
dengyihao 已提交
203

dengyihao's avatar
dengyihao 已提交
204
static void doNotifyApp(SCliMsg* pMsg, SCliThrd* pThrd);
205
// handle req from app
dengyihao's avatar
dengyihao 已提交
206 207 208 209 210 211
static void cliHandleReq(SCliMsg* pMsg, SCliThrd* pThrd);
static void cliHandleQuit(SCliMsg* pMsg, SCliThrd* pThrd);
static void cliHandleRelease(SCliMsg* pMsg, SCliThrd* pThrd);
static void cliHandleUpdate(SCliMsg* pMsg, SCliThrd* pThrd);
static void (*cliAsyncHandle[])(SCliMsg* pMsg, SCliThrd* pThrd) = {cliHandleReq, cliHandleQuit, cliHandleRelease, NULL,
                                                                   cliHandleUpdate};
dengyihao's avatar
dengyihao 已提交
212 213
/// static void (*cliAsyncHandle[])(SCliMsg* pMsg, SCliThrd* pThrd) = {cliHandleReq, cliHandleQuit, cliHandleRelease,
/// NULL,cliHandleUpdate};
dengyihao's avatar
dengyihao 已提交
214

dengyihao's avatar
dengyihao 已提交
215 216
static FORCE_INLINE void destroyUserdata(STransMsg* userdata);
static FORCE_INLINE void destroyCmsg(void* cmsg);
dengyihao's avatar
dengyihao 已提交
217
static FORCE_INLINE void destroyCmsgAndAhandle(void* cmsg);
dengyihao's avatar
dengyihao 已提交
218
static FORCE_INLINE int  cliRBChoseIdx(STrans* pTransInst);
dengyihao's avatar
dengyihao 已提交
219
static FORCE_INLINE void transDestroyConnCtx(STransConnCtx* ctx);
dengyihao's avatar
dengyihao 已提交
220

dengyihao's avatar
dengyihao 已提交
221
// thread obj
dengyihao's avatar
dengyihao 已提交
222
static SCliThrd* createThrdObj(void* trans);
dengyihao's avatar
dengyihao 已提交
223
static void      destroyThrdObj(SCliThrd* pThrd);
dengyihao's avatar
dengyihao 已提交
224

dengyihao's avatar
dengyihao 已提交
225 226 227 228 229 230 231 232 233
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 已提交
234
// snprintf may cause performance problem
dengyihao's avatar
dengyihao 已提交
235 236
#define CONN_CONSTRUCT_HASH_KEY(key, ip, port) \
  do {                                         \
dengyihao's avatar
dengyihao 已提交
237 238
    char*   t = key;                           \
    int16_t len = strlen(ip);                  \
dengyihao's avatar
dengyihao 已提交
239
    if (ip != NULL) memcpy(t, ip, len);        \
dengyihao's avatar
dengyihao 已提交
240 241
    t[len] = ':';                              \
    titoa(port, 10, &t[len + 1]);              \
dengyihao's avatar
dengyihao 已提交
242 243
  } while (0)

dengyihao's avatar
dengyihao 已提交
244 245
#define CONN_PERSIST_TIME(para)   ((para) <= 90000 ? 90000 : (para))
#define CONN_GET_INST_LABEL(conn) (((STrans*)(((SCliThrd*)(conn)->hostThrd)->pTransInst))->label)
dengyihao's avatar
dengyihao 已提交
246

dengyihao's avatar
dengyihao 已提交
247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262
#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);                            \
      if (pMsg->ctx != NULL && (uint64_t)pMsg->ctx->ahandle == ahandle) { \
        break;                                                            \
      }                                                                   \
    }                                                                     \
    if (i == sz) {                                                        \
      pMsg = NULL;                                                        \
      tDebug("msg not found, %" PRIu64 "", ahandle);                      \
    } else {                                                              \
      pMsg = transQueueRm(&conn->cliMsgs, i);                             \
      tDebug("msg found, %" PRIu64 "", ahandle);                          \
    }                                                                     \
dengyihao's avatar
dengyihao 已提交
263
  } while (0)
dengyihao's avatar
dengyihao 已提交
264

U
ubuntu 已提交
265 266 267 268 269 270 271 272 273 274 275 276 277
#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 已提交
278

dengyihao's avatar
formate  
dengyihao 已提交
279 280
#define CONN_SET_PERSIST_BY_APP(conn) \
  do {                                \
dengyihao's avatar
dengyihao 已提交
281 282
    if (conn->status == ConnNormal) { \
      conn->status = ConnAcquire;     \
dengyihao's avatar
formate  
dengyihao 已提交
283 284 285
      transRefCliHandle(conn);        \
    }                                 \
  } while (0)
286

dengyihao's avatar
dengyihao 已提交
287 288 289 290
#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)
291

S
Shengliang Guan 已提交
292 293
#define REQUEST_NO_RESP(msg)         ((msg)->info.noResp == 1)
#define REQUEST_PERSIS_HANDLE(msg)   ((msg)->info.persistHandle == 1)
dengyihao's avatar
dengyihao 已提交
294
#define REQUEST_RELEASE_HANDLE(cmsg) ((cmsg)->type == Release)
dengyihao's avatar
dengyihao 已提交
295

dengyihao's avatar
dengyihao 已提交
296
#define EPSET_IS_VALID(epSet)       ((epSet) != NULL && (epSet)->numOfEps >= 0 && (epSet)->inUse >= 0)
dengyihao's avatar
dengyihao 已提交
297
#define EPSET_GET_SIZE(epSet)       (epSet)->numOfEps
dengyihao's avatar
dengyihao 已提交
298
#define EPSET_GET_INUSE_IP(epSet)   ((epSet)->eps[(epSet)->inUse].fqdn)
dengyihao's avatar
dengyihao 已提交
299
#define EPSET_GET_INUSE_PORT(epSet) ((epSet)->eps[(epSet)->inUse].port)
dengyihao's avatar
dengyihao 已提交
300 301 302 303 304 305
#define EPSET_FORWARD_INUSE(epSet)                             \
  do {                                                         \
    if ((epSet)->numOfEps != 0) {                              \
      ++((epSet)->inUse);                                      \
      (epSet)->inUse = ((epSet)->inUse) % ((epSet)->numOfEps); \
    }                                                          \
dengyihao's avatar
dengyihao 已提交
306
  } while (0)
dengyihao's avatar
dengyihao 已提交
307

dengyihao's avatar
dengyihao 已提交
308 309 310 311 312 313 314 315 316 317 318
#define EPSET_DEBUG_STR(epSet, tbuf)                                                                                   \
  do {                                                                                                                 \
    int len = snprintf(tbuf, sizeof(tbuf), "epset:{");                                                                 \
    for (int i = 0; i < (epSet)->numOfEps; i++) {                                                                      \
      if (i == (epSet)->numOfEps - 1) {                                                                                \
        len += snprintf(tbuf + len, sizeof(tbuf) - len, "%d. %s:%d", i, (epSet)->eps[i].fqdn, (epSet)->eps[i].port);   \
      } else {                                                                                                         \
        len += snprintf(tbuf + len, sizeof(tbuf) - len, "%d. %s:%d, ", i, (epSet)->eps[i].fqdn, (epSet)->eps[i].port); \
      }                                                                                                                \
    }                                                                                                                  \
    len += snprintf(tbuf + len, sizeof(tbuf) - len, "}, inUse:%d", (epSet)->inUse);                                    \
dengyihao's avatar
dengyihao 已提交
319
  } while (0);
dengyihao's avatar
dengyihao 已提交
320

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

dengyihao's avatar
dengyihao 已提交
323 324 325 326 327 328 329 330
static void cliReleaseUnfinishedMsg(SCliConn* conn) {
  SCliThrd* pThrd = conn->hostThrd;

  for (int i = 0; i < transQueueSize(&conn->cliMsgs); i++) {
    SCliMsg* msg = transQueueGet(&conn->cliMsgs, i);
    if (msg != NULL && msg->ctx != NULL && msg->ctx->ahandle != (void*)0x9527) {
      if (conn->ctx.freeFunc != NULL && msg->ctx->ahandle != NULL) {
        conn->ctx.freeFunc(msg->ctx->ahandle);
dengyihao's avatar
dengyihao 已提交
331
      } else if (msg->ctx->ahandle != NULL && pThrd->destroyAhandleFp != NULL) {
dengyihao's avatar
dengyihao 已提交
332
        tDebug("%s conn %p destroy unfinished ahandle %p", CONN_GET_INST_LABEL(conn), conn, msg->ctx->ahandle);
dengyihao's avatar
dengyihao 已提交
333
        pThrd->destroyAhandleFp(msg->ctx->ahandle);
dengyihao's avatar
dengyihao 已提交
334 335 336 337
      }
    }
    destroyCmsg(msg);
  }
dengyihao's avatar
dengyihao 已提交
338
  transQueueClear(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
339
  memset(&conn->ctx, 0, sizeof(conn->ctx));
dengyihao's avatar
dengyihao 已提交
340
}
dengyihao's avatar
dengyihao 已提交
341
bool cliMaySendCachedMsg(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
342
  if (!transQueueEmpty(&conn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
343
    SCliMsg* pCliMsg = NULL;
U
ubuntu 已提交
344
    CONN_GET_NEXT_SENDMSG(conn);
dengyihao's avatar
dengyihao 已提交
345 346
    cliSend(conn);
    return true;
dengyihao's avatar
dengyihao 已提交
347
  }
dengyihao's avatar
dengyihao 已提交
348
  return false;
U
ubuntu 已提交
349 350
_RETURN:
  return false;
dengyihao's avatar
dengyihao 已提交
351
}
dengyihao's avatar
formate  
dengyihao 已提交
352
void cliHandleResp(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
353 354
  SCliThrd* pThrd = conn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
355

dengyihao's avatar
dengyihao 已提交
356 357 358 359 360 361
  if (conn->timer) {
    if (uv_is_active((uv_handle_t*)conn->timer)) {
      tDebug("%s conn %p stop timer", CONN_GET_INST_LABEL(conn), conn);
      uv_timer_stop(conn->timer);
    }
    taosArrayPush(pThrd->timerList, &conn->timer);
dengyihao's avatar
dengyihao 已提交
362
    conn->timer->data = NULL;
dengyihao's avatar
dengyihao 已提交
363
    conn->timer = NULL;
dengyihao's avatar
dengyihao 已提交
364 365
  }

dengyihao's avatar
opt rpc  
dengyihao 已提交
366
  STransMsgHead* pHead = NULL;
dengyihao's avatar
dengyihao 已提交
367 368 369

  int32_t msgLen = transDumpFromBuffer(&conn->readBuf, (char**)&pHead);
  if (msgLen <= 0) {
dengyihao's avatar
dengyihao 已提交
370
    taosMemoryFree(pHead);
dengyihao's avatar
dengyihao 已提交
371 372 373
    tDebug("%s conn %p recv invalid packet ", CONN_GET_INST_LABEL(conn), conn);
    return;
  }
dengyihao's avatar
dengyihao 已提交
374 375 376 377

  if (transDecompressMsg((char**)&pHead, msgLen) < 0) {
    tDebug("%s conn %p recv invalid packet, failed to decompress", CONN_GET_INST_LABEL(conn), conn);
  }
dengyihao's avatar
dengyihao 已提交
378 379
  pHead->code = htonl(pHead->code);
  pHead->msgLen = htonl(pHead->msgLen);
dengyihao's avatar
dengyihao 已提交
380 381 382 383
  if (cliRecvReleaseReq(conn, pHead)) {
    return;
  }

dengyihao's avatar
dengyihao 已提交
384 385 386 387 388
  STransMsg transMsg = {0};
  transMsg.contLen = transContLenFromMsg(pHead->msgLen);
  transMsg.pCont = transContFromHead((char*)pHead);
  transMsg.code = pHead->code;
  transMsg.msgType = pHead->msgType;
S
Shengliang Guan 已提交
389
  transMsg.info.ahandle = NULL;
dengyihao's avatar
dengyihao 已提交
390
  transMsg.info.traceId = pHead->traceId;
dengyihao's avatar
dengyihao 已提交
391
  transMsg.info.hasEpSet = pHead->hasEpSet;
dengyihao's avatar
dengyihao 已提交
392

dengyihao's avatar
dengyihao 已提交
393 394 395 396
  SCliMsg*       pMsg = NULL;
  STransConnCtx* pCtx = NULL;
  if (CONN_NO_PERSIST_BY_APP(conn)) {
    pMsg = transQueuePop(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
397 398 399

    pCtx = pMsg ? pMsg->ctx : NULL;
    transMsg.info.ahandle = pCtx ? pCtx->ahandle : NULL;
dengyihao's avatar
dengyihao 已提交
400
    tDebug("%s conn %p get ahandle %p, persist: 0", CONN_GET_INST_LABEL(conn), conn, transMsg.info.ahandle);
U
ubuntu 已提交
401
  } else {
dengyihao's avatar
dengyihao 已提交
402 403 404
    uint64_t ahandle = (uint64_t)pHead->ahandle;
    CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle);
    if (pMsg == NULL) {
S
Shengliang Guan 已提交
405
      transMsg.info.ahandle = transCtxDumpVal(&conn->ctx, transMsg.msgType);
dengyihao's avatar
dengyihao 已提交
406 407
      tDebug("%s conn %p construct ahandle %p by %s, persist: 1", CONN_GET_INST_LABEL(conn), conn,
             transMsg.info.ahandle, TMSG_INFO(transMsg.msgType));
S
Shengliang Guan 已提交
408
      if (!CONN_RELEASE_BY_SERVER(conn) && transMsg.info.ahandle == NULL) {
D
dapan1121 已提交
409
        transMsg.code = TSDB_CODE_RPC_BROKEN_LINK;
S
Shengliang Guan 已提交
410
        transMsg.info.ahandle = transCtxDumpBrokenlinkVal(&conn->ctx, (int32_t*)&(transMsg.msgType));
dengyihao's avatar
dengyihao 已提交
411 412
        tDebug("%s conn %p construct ahandle %p due brokenlink, persist: 1", CONN_GET_INST_LABEL(conn), conn,
               transMsg.info.ahandle);
dengyihao's avatar
dengyihao 已提交
413 414
      }
    } else {
dengyihao's avatar
dengyihao 已提交
415
      pCtx = pMsg->ctx;
S
Shengliang Guan 已提交
416
      transMsg.info.ahandle = pCtx ? pCtx->ahandle : NULL;
dengyihao's avatar
dengyihao 已提交
417
      tDebug("%s conn %p get ahandle %p, persist: 1", CONN_GET_INST_LABEL(conn), conn, transMsg.info.ahandle);
dengyihao's avatar
dengyihao 已提交
418
    }
U
ubuntu 已提交
419
  }
dengyihao's avatar
dengyihao 已提交
420
  // buf's mem alread translated to transMsg.pCont
dengyihao's avatar
dengyihao 已提交
421
  if (!CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
422
    transMsg.info.handle = (void*)conn->refId;
dengyihao's avatar
dengyihao 已提交
423
    tDebug("%s conn %p ref by app", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
424
  }
dengyihao's avatar
dengyihao 已提交
425

dengyihao's avatar
dengyihao 已提交
426
  STraceId* trace = &transMsg.info.traceId;
dengyihao's avatar
dengyihao 已提交
427
  tGDebug("%s conn %p %s received from %s, local info:%s, len:%d, code str:%s", CONN_GET_INST_LABEL(conn), conn,
dengyihao's avatar
dengyihao 已提交
428
          TMSG_INFO(pHead->msgType), conn->dst, conn->src, pHead->msgLen, tstrerror(transMsg.code));
dengyihao's avatar
dengyihao 已提交
429

dengyihao's avatar
dengyihao 已提交
430
  if (pCtx == NULL && CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
431
    tDebug("%s except, conn %p read while cli ignore it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
432
    transFreeMsg(transMsg.pCont);
dengyihao's avatar
dengyihao 已提交
433 434
    return;
  }
S
Shengliang Guan 已提交
435
  if (CONN_RELEASE_BY_SERVER(conn) && transMsg.info.ahandle == NULL) {
dengyihao's avatar
dengyihao 已提交
436
    tDebug("%s except, conn %p read while cli ignore it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
437
    transFreeMsg(transMsg.pCont);
dengyihao's avatar
dengyihao 已提交
438 439
    return;
  }
dengyihao's avatar
dengyihao 已提交
440

dengyihao's avatar
dengyihao 已提交
441
  if (pMsg == NULL || (pMsg && pMsg->type != Release)) {
dengyihao's avatar
dengyihao 已提交
442 443 444
    if (cliAppCb(conn, &transMsg, pMsg) != 0) {
      return;
    }
dengyihao's avatar
dengyihao 已提交
445
  }
dengyihao's avatar
dengyihao 已提交
446 447
  destroyCmsg(pMsg);

dengyihao's avatar
dengyihao 已提交
448
  if (cliMaySendCachedMsg(conn) == true) {
dengyihao's avatar
dengyihao 已提交
449 450
    return;
  }
dengyihao's avatar
dengyihao 已提交
451

U
ubuntu 已提交
452
  if (CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
453
    return addConnToPool(pThrd->pool, conn);
dengyihao's avatar
dengyihao 已提交
454
  }
dengyihao's avatar
test  
dengyihao 已提交
455

dengyihao's avatar
dengyihao 已提交
456
  uv_read_start((uv_stream_t*)conn->stream, cliAllocRecvBufferCb, cliRecvCb);
dengyihao's avatar
dengyihao 已提交
457
}
U
ubuntu 已提交
458

dengyihao's avatar
dengyihao 已提交
459
void cliHandleExceptImpl(SCliConn* pConn, int32_t code) {
dengyihao's avatar
dengyihao 已提交
460
  if (transQueueEmpty(&pConn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
461
    if (pConn->broken == true && CONN_NO_PERSIST_BY_APP(pConn)) {
dengyihao's avatar
dengyihao 已提交
462
      tTrace("%s conn %p handle except, persist:0", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
formate  
dengyihao 已提交
463 464 465
      transUnrefCliHandle(pConn);
      return;
    }
dengyihao's avatar
dengyihao 已提交
466
  }
dengyihao's avatar
dengyihao 已提交
467 468 469
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
  bool      once = false;
D
dapan1121 已提交
470
  do {
dengyihao's avatar
dengyihao 已提交
471
    SCliMsg* pMsg = transQueuePop(&pConn->cliMsgs);
dengyihao's avatar
enh log  
dengyihao 已提交
472

D
dapan1121 已提交
473 474
    if (pMsg == NULL && once) {
      break;
dengyihao's avatar
dengyihao 已提交
475
    }
dengyihao's avatar
enh log  
dengyihao 已提交
476 477 478 479 480 481

    if (pMsg != NULL && REQUEST_NO_RESP(&pMsg->msg)) {
      destroyCmsg(pMsg);
      break;
    }

dengyihao's avatar
dengyihao 已提交
482
    STransConnCtx* pCtx = pMsg ? pMsg->ctx : NULL;
U
ubuntu 已提交
483

dengyihao's avatar
dengyihao 已提交
484
    STransMsg transMsg = {0};
dengyihao's avatar
dengyihao 已提交
485
    transMsg.code = code == -1 ? (pConn->broken ? TSDB_CODE_RPC_BROKEN_LINK : TSDB_CODE_RPC_NETWORK_UNAVAIL) : code;
dengyihao's avatar
dengyihao 已提交
486
    transMsg.msgType = pMsg ? pMsg->msg.msgType + 1 : 0;
S
Shengliang Guan 已提交
487
    transMsg.info.ahandle = NULL;
dengyihao's avatar
dengyihao 已提交
488

dengyihao's avatar
dengyihao 已提交
489
    if (pMsg == NULL && !CONN_NO_PERSIST_BY_APP(pConn)) {
S
Shengliang Guan 已提交
490
      transMsg.info.ahandle = transCtxDumpVal(&pConn->ctx, transMsg.msgType);
dengyihao's avatar
dengyihao 已提交
491
      tDebug("%s conn %p construct ahandle %p by %s", CONN_GET_INST_LABEL(pConn), pConn, transMsg.info.ahandle,
dengyihao's avatar
dengyihao 已提交
492
             TMSG_INFO(transMsg.msgType));
S
Shengliang Guan 已提交
493
      if (transMsg.info.ahandle == NULL) {
dengyihao's avatar
dengyihao 已提交
494 495 496
        int32_t msgType = 0;
        transMsg.info.ahandle = transCtxDumpBrokenlinkVal(&pConn->ctx, &msgType);
        transMsg.msgType = msgType;
dengyihao's avatar
dengyihao 已提交
497
        tDebug("%s conn %p construct ahandle %p due to brokenlink", CONN_GET_INST_LABEL(pConn), pConn,
S
Shengliang Guan 已提交
498
               transMsg.info.ahandle);
dengyihao's avatar
dengyihao 已提交
499
      }
dengyihao's avatar
dengyihao 已提交
500
    } else {
dengyihao's avatar
dengyihao 已提交
501
      transMsg.info.ahandle = (pMsg != NULL && pMsg->type != Release && pCtx) ? pCtx->ahandle : NULL;
dengyihao's avatar
dengyihao 已提交
502 503 504
    }

    if (pCtx == NULL || pCtx->pSem == NULL) {
S
Shengliang Guan 已提交
505
      if (transMsg.info.ahandle == NULL) {
dengyihao's avatar
dengyihao 已提交
506 507 508 509 510
        if (pMsg == NULL || REQUEST_NO_RESP(&pMsg->msg) || pMsg->type == Release) {
          destroyCmsg(pMsg);
          once = true;
          continue;
        }
U
ubuntu 已提交
511
      }
dengyihao's avatar
dengyihao 已提交
512
    }
dengyihao's avatar
enh log  
dengyihao 已提交
513

dengyihao's avatar
dengyihao 已提交
514
    if (pMsg == NULL || (pMsg && pMsg->type != Release)) {
dengyihao's avatar
dengyihao 已提交
515 516 517
      if (cliAppCb(pConn, &transMsg, pMsg) != 0) {
        return;
      }
dengyihao's avatar
dengyihao 已提交
518 519
    }
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
520
    tTrace("%s conn %p start to destroy, ref:%d", CONN_GET_INST_LABEL(pConn), pConn, T_REF_VAL_GET(pConn));
D
dapan1121 已提交
521
  } while (!transQueueEmpty(&pConn->cliMsgs));
dengyihao's avatar
dengyihao 已提交
522
  transUnrefCliHandle(pConn);
dengyihao's avatar
dengyihao 已提交
523
}
dengyihao's avatar
dengyihao 已提交
524
void cliHandleExcept(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
525
  tTrace("%s conn %p except ref:%d", CONN_GET_INST_LABEL(conn), conn, T_REF_VAL_GET(conn));
dengyihao's avatar
dengyihao 已提交
526 527
  cliHandleExceptImpl(conn, -1);
}
dengyihao's avatar
dengyihao 已提交
528

dengyihao's avatar
dengyihao 已提交
529 530
void cliConnTimeout(uv_timer_t* handle) {
  SCliConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
531 532
  SCliThrd* pThrd = conn->hostThrd;

dengyihao's avatar
dengyihao 已提交
533
  tTrace("%s conn %p conn timeout, ref:%d", CONN_GET_INST_LABEL(conn), conn, T_REF_VAL_GET(conn));
dengyihao's avatar
dengyihao 已提交
534

dengyihao's avatar
dengyihao 已提交
535
  uv_timer_stop(handle);
dengyihao's avatar
dengyihao 已提交
536 537 538
  handle->data = NULL;
  taosArrayPush(pThrd->timerList, &conn->timer);
  conn->timer = NULL;
dengyihao's avatar
dengyihao 已提交
539 540

  cliHandleFastFail(conn, UV_ECANCELED);
dengyihao's avatar
dengyihao 已提交
541
}
dengyihao's avatar
dengyihao 已提交
542 543 544 545
void cliReadTimeoutCb(uv_timer_t* handle) {
  // set up timeout cb
  SCliConn* conn = handle->data;
  tTrace("%s conn %p timeout, ref:%d", CONN_GET_INST_LABEL(conn), conn, T_REF_VAL_GET(conn));
dengyihao's avatar
dengyihao 已提交
546
  uv_read_stop(conn->stream);
dengyihao's avatar
dengyihao 已提交
547 548
  cliHandleExceptImpl(conn, TSDB_CODE_RPC_TIMEOUT);
}
U
ubuntu 已提交
549 550

void* createConnPool(int size) {
551 552
  // thread local, no lock
  return taosHashInit(size, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
553
}
dengyihao's avatar
dengyihao 已提交
554 555
void* destroyConnPool(SCliThrd* pThrd) {
  void*      pool = pThrd->pool;
dengyihao's avatar
dengyihao 已提交
556
  SConnList* connList = taosHashIterate((SHashObj*)pool, NULL);
dengyihao's avatar
dengyihao 已提交
557
  while (connList != NULL) {
dengyihao's avatar
dengyihao 已提交
558 559
    while (!QUEUE_IS_EMPTY(&connList->conns)) {
      queue*    h = QUEUE_HEAD(&connList->conns);
dengyihao's avatar
dengyihao 已提交
560
      SCliConn* c = QUEUE_DATA(h, SCliConn, q);
U
ubuntu 已提交
561
      cliDestroyConn(c, true);
dengyihao's avatar
dengyihao 已提交
562
    }
dengyihao's avatar
dengyihao 已提交
563 564 565 566 567 568 569 570 571 572 573 574 575 576 577

    SMsgList* msglist = connList->list;
    while (!QUEUE_IS_EMPTY(&msglist->msgQ)) {
      queue* h = QUEUE_HEAD(&msglist->msgQ);
      QUEUE_REMOVE(h);

      SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);

      transDQCancel(pThrd->waitConnQueue, pMsg->ctx->task);
      pMsg->ctx->task = NULL;

      doNotifyApp(pMsg, pThrd);
    }
    taosMemoryFree(msglist);

dengyihao's avatar
dengyihao 已提交
578
    connList = taosHashIterate((SHashObj*)pool, connList);
dengyihao's avatar
dengyihao 已提交
579
  }
dengyihao's avatar
dengyihao 已提交
580
  taosHashCleanup(pool);
581
  return NULL;
dengyihao's avatar
dengyihao 已提交
582 583
}

dengyihao's avatar
dengyihao 已提交
584
static SCliConn* getConnFromPool(SCliThrd* pThrd, char* key, bool* exceed) {
dengyihao's avatar
dengyihao 已提交
585
  void*      pool = pThrd->pool;
dengyihao's avatar
dengyihao 已提交
586
  SConnList* plist = taosHashGet((SHashObj*)pool, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
587
  STrans*    pTranInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
588
  if (plist == NULL) {
dengyihao's avatar
dengyihao 已提交
589 590 591 592
    SConnList list = {0};
    taosHashPut((SHashObj*)pool, key, strlen(key), (void*)&list, sizeof(list));
    plist = taosHashGet(pool, key, strlen(key));

dengyihao's avatar
dengyihao 已提交
593 594 595 596
    SMsgList* nList = taosMemoryCalloc(1, sizeof(SMsgList));
    QUEUE_INIT(&nList->msgQ);
    nList->numOfConn++;

dengyihao's avatar
dengyihao 已提交
597 598
    QUEUE_INIT(&plist->conns);
    plist->list = nList;
dengyihao's avatar
dengyihao 已提交
599 600
  }

dengyihao's avatar
dengyihao 已提交
601 602 603 604
  if (QUEUE_IS_EMPTY(&plist->conns)) {
    if (plist->list->numOfConn >= pTranInst->connLimitNum) {
      *exceed = true;
    }
dengyihao's avatar
dengyihao 已提交
605 606 607
    return NULL;
  }

dengyihao's avatar
dengyihao 已提交
608
  queue* h = QUEUE_TAIL(&plist->conns);
dengyihao's avatar
dengyihao 已提交
609
  QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
610
  plist->size -= 1;
dengyihao's avatar
dengyihao 已提交
611

dengyihao's avatar
dengyihao 已提交
612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627
  SCliConn* conn = QUEUE_DATA(h, SCliConn, q);
  conn->status = ConnNormal;
  QUEUE_INIT(&conn->q);

  if (conn->task != NULL) {
    transDQCancel(((SCliThrd*)conn->hostThrd)->timeoutQueue, conn->task);
    conn->task = NULL;
  }
  return conn;
}

static SCliConn* getConnFromPool2(SCliThrd* pThrd, char* key, SCliMsg** pMsg) {
  void*      pool = pThrd->pool;
  STrans*    pTransInst = pThrd->pTransInst;
  SConnList* plist = taosHashGet((SHashObj*)pool, key, strlen(key));
  if (plist == NULL) {
dengyihao's avatar
dengyihao 已提交
628 629 630 631
    SConnList list = {0};
    taosHashPut((SHashObj*)pool, key, strlen(key), (void*)&list, sizeof(list));
    plist = taosHashGet(pool, key, strlen(key));

dengyihao's avatar
dengyihao 已提交
632 633
    SMsgList* nList = taosMemoryCalloc(1, sizeof(SMsgList));
    QUEUE_INIT(&nList->msgQ);
dengyihao's avatar
dengyihao 已提交
634 635
    nList->numOfConn++;

dengyihao's avatar
dengyihao 已提交
636 637
    QUEUE_INIT(&plist->conns);
    plist->list = nList;
dengyihao's avatar
dengyihao 已提交
638 639
  }

dengyihao's avatar
dengyihao 已提交
640
  STraceId* trace = &(*pMsg)->msg.info.traceId;
dengyihao's avatar
dengyihao 已提交
641
  // no avaliable conn in pool
dengyihao's avatar
dengyihao 已提交
642
  if (QUEUE_IS_EMPTY(&plist->conns)) {
dengyihao's avatar
dengyihao 已提交
643
    SMsgList* list = plist->list;
dengyihao's avatar
dengyihao 已提交
644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661
    if ((list)->numOfConn >= pTransInst->connLimitNum) {
      STraceId* trace = &(*pMsg)->msg.info.traceId;
      STaskArg* arg = taosMemoryMalloc(sizeof(STaskArg));
      arg->param1 = *pMsg;
      arg->param2 = pThrd;
      (*pMsg)->ctx->task = transDQSched(pThrd->waitConnQueue, doFreeTimeoutMsg, arg, pTransInst->timeToGetConn);

      tGTrace("%s msg %s delay to send, wait for avaiable connect", pTransInst->label, TMSG_INFO((*pMsg)->msg.msgType));

      QUEUE_PUSH(&(list)->msgQ, &(*pMsg)->q);
      *pMsg = NULL;
    } else {
      // send msg in delay queue
      if (!(QUEUE_IS_EMPTY(&(list)->msgQ))) {
        STaskArg* arg = taosMemoryMalloc(sizeof(STaskArg));
        arg->param1 = *pMsg;
        arg->param2 = pThrd;
        (*pMsg)->ctx->task = transDQSched(pThrd->waitConnQueue, doFreeTimeoutMsg, arg, pTransInst->timeToGetConn);
dengyihao's avatar
dengyihao 已提交
662 663
        tGTrace("%s msg %s delay to send, wait for avaiable connect", pTransInst->label,
                TMSG_INFO((*pMsg)->msg.msgType));
dengyihao's avatar
dengyihao 已提交
664 665 666 667 668 669 670

        QUEUE_PUSH(&(list)->msgQ, &(*pMsg)->q);
        queue* h = QUEUE_HEAD(&(list)->msgQ);
        QUEUE_REMOVE(h);
        SCliMsg* ans = QUEUE_DATA(h, SCliMsg, q);

        *pMsg = ans;
dengyihao's avatar
dengyihao 已提交
671 672 673

        trace = &(*pMsg)->msg.info.traceId;
        tGTrace("%s msg %s pop from delay queue, start to send", pTransInst->label, TMSG_INFO((*pMsg)->msg.msgType));
dengyihao's avatar
dengyihao 已提交
674 675 676 677
        transDQCancel(pThrd->waitConnQueue, ans->ctx->task);
      }
      list->numOfConn++;
    }
dengyihao's avatar
dengyihao 已提交
678 679
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
680

dengyihao's avatar
dengyihao 已提交
681
  queue* h = QUEUE_TAIL(&plist->conns);
dengyihao's avatar
dengyihao 已提交
682
  plist->size -= 1;
dengyihao's avatar
dengyihao 已提交
683 684
  QUEUE_REMOVE(h);

dengyihao's avatar
dengyihao 已提交
685
  SCliConn* conn = QUEUE_DATA(h, SCliConn, q);
dengyihao's avatar
dengyihao 已提交
686
  conn->status = ConnNormal;
dengyihao's avatar
dengyihao 已提交
687
  QUEUE_INIT(&conn->q);
dengyihao's avatar
dengyihao 已提交
688

dengyihao's avatar
dengyihao 已提交
689 690 691 692
  if (conn->task != NULL) {
    transDQCancel(((SCliThrd*)conn->hostThrd)->timeoutQueue, conn->task);
    conn->task = NULL;
  }
dengyihao's avatar
dengyihao 已提交
693
  return conn;
dengyihao's avatar
dengyihao 已提交
694
}
dengyihao's avatar
dengyihao 已提交
695 696 697 698 699 700
static void addConnToPool(void* pool, SCliConn* conn) {
  if (conn->status == ConnInPool) {
    return;
  }
  allocConnRef(conn, true);

dengyihao's avatar
dengyihao 已提交
701
  SCliThrd* thrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
702 703 704 705 706 707
  if (conn->timer != NULL) {
    uv_timer_stop(conn->timer);
    taosArrayPush(thrd->timerList, &conn->timer);
    conn->timer->data = NULL;
    conn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
708 709 710 711 712
  if (T_REF_VAL_GET(conn) > 1) {
    transUnrefCliHandle(conn);
  }

  cliDestroyConnMsgs(conn, false);
dengyihao's avatar
dengyihao 已提交
713

dengyihao's avatar
dengyihao 已提交
714
  if (conn->list == NULL) {
dengyihao's avatar
dengyihao 已提交
715
    conn->list = taosHashGet((SHashObj*)pool, conn->ip, strlen(conn->ip));
dengyihao's avatar
dengyihao 已提交
716
  }
dengyihao's avatar
dengyihao 已提交
717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737

  SConnList* pList = conn->list;
  SMsgList*  msgList = pList->list;
  if (!QUEUE_IS_EMPTY(&msgList->msgQ)) {
    queue* h = QUEUE_HEAD(&(msgList)->msgQ);
    QUEUE_REMOVE(h);

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);

    transDQCancel(thrd->waitConnQueue, pMsg->ctx->task);
    pMsg->ctx->task = NULL;

    transCtxMerge(&conn->ctx, &pMsg->ctx->appCtx);
    transQueuePush(&conn->cliMsgs, pMsg);

    conn->status = ConnNormal;
    cliSend(conn);
    return;
  }

  conn->status = ConnInPool;
dengyihao's avatar
dengyihao 已提交
738
  QUEUE_PUSH(&conn->list->conns, &conn->q);
dengyihao's avatar
dengyihao 已提交
739
  conn->list->size += 1;
dengyihao's avatar
dengyihao 已提交
740

dengyihao's avatar
dengyihao 已提交
741
  if (conn->list->size >= 20) {
dengyihao's avatar
dengyihao 已提交
742 743
    STaskArg* arg = taosMemoryCalloc(1, sizeof(STaskArg));
    arg->param1 = conn;
dengyihao's avatar
dengyihao 已提交
744
    arg->param2 = thrd;
dengyihao's avatar
dengyihao 已提交
745 746

    STrans* pTransInst = thrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
747 748
    conn->task = transDQSched(thrd->timeoutQueue, doCloseIdleConn, arg, CONN_PERSIST_TIME(pTransInst->idleTime));
  }
dengyihao's avatar
dengyihao 已提交
749
}
dengyihao's avatar
dengyihao 已提交
750
static int32_t allocConnRef(SCliConn* conn, bool update) {
dengyihao's avatar
dengyihao 已提交
751
  if (update) {
dengyihao's avatar
dengyihao 已提交
752
    transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
753
    transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
754
    conn->refId = -1;
dengyihao's avatar
dengyihao 已提交
755 756 757 758
  }
  SExHandle* exh = taosMemoryCalloc(1, sizeof(SExHandle));
  exh->handle = conn;
  exh->pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
759
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
dengyihao's avatar
dengyihao 已提交
760
  conn->refId = exh->refId;
761 762 763 764

  if (conn->refId == -1) {
    taosMemoryFree(exh);
  }
dengyihao's avatar
dengyihao 已提交
765 766 767 768 769
  return 0;
}

static int32_t specifyConnRef(SCliConn* conn, bool update, int64_t handle) {
  if (update) {
dengyihao's avatar
dengyihao 已提交
770
    transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
771 772 773 774 775 776 777 778 779 780 781 782 783
    transRemoveExHandle(transGetRefMgt(), conn->refId);
    conn->refId = -1;
  }
  SExHandle* exh = transAcquireExHandle(transGetRefMgt(), handle);
  if (exh == NULL) {
    return -1;
  }
  exh->handle = conn;
  exh->pThrd = conn->hostThrd;
  conn->refId = exh->refId;

  transReleaseExHandle(transGetRefMgt(), handle);
  return 0;
dengyihao's avatar
dengyihao 已提交
784
}
dengyihao's avatar
dengyihao 已提交
785

dengyihao's avatar
dengyihao 已提交
786
static void cliAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
787
  SCliConn*    conn = handle->data;
dengyihao's avatar
dengyihao 已提交
788
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
789
  tTrace("%s conn %p alloc read buf", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
790
  transAllocBuffer(pBuf, buf);
dengyihao's avatar
dengyihao 已提交
791
}
U
ubuntu 已提交
792
static void cliRecvCb(uv_stream_t* handle, ssize_t nread, const uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
793
  // impl later
dengyihao's avatar
dengyihao 已提交
794 795 796
  if (handle->data == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
797 798
  SCliConn*    conn = handle->data;
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
799
  if (nread > 0) {
dengyihao's avatar
dengyihao 已提交
800
    pBuf->len += nread;
dengyihao's avatar
dengyihao 已提交
801
    while (transReadComplete(pBuf)) {
dengyihao's avatar
dengyihao 已提交
802
      tTrace("%s conn %p read complete", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
803 804 805 806 807 808
      if (pBuf->invalid) {
        cliHandleExcept(conn);
        break;
      } else {
        cliHandleResp(conn);
      }
dengyihao's avatar
dengyihao 已提交
809
    }
dengyihao's avatar
dengyihao 已提交
810 811
    return;
  }
dengyihao's avatar
dengyihao 已提交
812

813
  if (nread == 0) {
dengyihao's avatar
dengyihao 已提交
814 815 816
    // 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 已提交
817
    tTrace("%s conn %p read empty", CONN_GET_INST_LABEL(conn), conn);
818 819
    return;
  }
dengyihao's avatar
fix bug  
dengyihao 已提交
820
  if (nread < 0) {
dengyihao's avatar
dengyihao 已提交
821
    tWarn("%s conn %p read error:%s, ref:%d", CONN_GET_INST_LABEL(conn), conn, uv_err_name(nread), T_REF_VAL_GET(conn));
dengyihao's avatar
dengyihao 已提交
822
    conn->broken = true;
U
ubuntu 已提交
823
    cliHandleExcept(conn);
dengyihao's avatar
dengyihao 已提交
824
  }
dengyihao's avatar
dengyihao 已提交
825
}
dengyihao's avatar
dengyihao 已提交
826

dengyihao's avatar
dengyihao 已提交
827
static SCliConn* cliCreateConn(SCliThrd* pThrd) {
wafwerar's avatar
wafwerar 已提交
828
  SCliConn* conn = taosMemoryCalloc(1, sizeof(SCliConn));
dengyihao's avatar
dengyihao 已提交
829
  // read/write stream handle
G
gccgdb1234 已提交
830
  conn->stream = (uv_stream_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
831 832 833
  uv_tcp_init(pThrd->loop, (uv_tcp_t*)(conn->stream));
  conn->stream->data = conn;

dengyihao's avatar
dengyihao 已提交
834 835 836 837 838 839 840 841
  uv_timer_t* timer = taosArrayGetSize(pThrd->timerList) > 0 ? *(uv_timer_t**)taosArrayPop(pThrd->timerList) : NULL;
  if (timer == NULL) {
    timer = taosMemoryCalloc(1, sizeof(uv_timer_t));
    tDebug("no available timer, create a timer %p", timer);
    uv_timer_init(pThrd->loop, timer);
  }
  timer->data = conn;
  conn->timer = timer;
dengyihao's avatar
dengyihao 已提交
842

dengyihao's avatar
dengyihao 已提交
843
  conn->connReq.data = conn;
dengyihao's avatar
dengyihao 已提交
844 845
  transReqQueueInit(&conn->wreqQueue);

dengyihao's avatar
dengyihao 已提交
846
  transQueueInit(&conn->cliMsgs, NULL);
dengyihao's avatar
opt rpc  
dengyihao 已提交
847 848

  transInitBuffer(&conn->readBuf);
dengyihao's avatar
dengyihao 已提交
849
  QUEUE_INIT(&conn->q);
dengyihao's avatar
dengyihao 已提交
850
  conn->hostThrd = pThrd;
dengyihao's avatar
dengyihao 已提交
851
  conn->status = ConnNormal;
dengyihao's avatar
dengyihao 已提交
852
  conn->broken = false;
dengyihao's avatar
dengyihao 已提交
853
  transRefCliHandle(conn);
dengyihao's avatar
dengyihao 已提交
854

dengyihao's avatar
dengyihao 已提交
855
  atomic_add_fetch_32(&pThrd->connCount, 1);
dengyihao's avatar
dengyihao 已提交
856
  allocConnRef(conn, false);
dengyihao's avatar
dengyihao 已提交
857

dengyihao's avatar
dengyihao 已提交
858 859
  return conn;
}
U
ubuntu 已提交
860
static void cliDestroyConn(SCliConn* conn, bool clear) {
dengyihao's avatar
dengyihao 已提交
861
  SCliThrd* pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
862
  tTrace("%s conn %p remove from conn pool", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
863

dengyihao's avatar
dengyihao 已提交
864 865
  QUEUE_REMOVE(&conn->q);
  QUEUE_INIT(&conn->q);
dengyihao's avatar
dengyihao 已提交
866 867 868 869

  if (conn->list != NULL) {
    SConnList* connList = conn->list;
    connList->list->numOfConn--;
dengyihao's avatar
dengyihao 已提交
870
    connList->size--;
dengyihao's avatar
dengyihao 已提交
871 872 873 874 875 876
  } else {
    SConnList* connList = taosHashGet((SHashObj*)pThrd->pool, conn->ip, strlen(conn->ip));
    connList->list->numOfConn--;
  }
  conn->list = NULL;

dengyihao's avatar
dengyihao 已提交
877
  transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
878
  transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
opt rpc  
dengyihao 已提交
879
  conn->refId = -1;
dengyihao's avatar
dengyihao 已提交
880

dengyihao's avatar
dengyihao 已提交
881 882 883 884 885 886 887
  if (conn->task != NULL) {
    transDQCancel(pThrd->timeoutQueue, conn->task);
    conn->task = NULL;
  }
  if (conn->timer != NULL) {
    uv_timer_stop(conn->timer);
    conn->timer->data = NULL;
dengyihao's avatar
dengyihao 已提交
888
    taosArrayPush(pThrd->timerList, &conn->timer);
dengyihao's avatar
dengyihao 已提交
889 890
    conn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
891

dengyihao's avatar
dengyihao 已提交
892
  if (clear) {
dengyihao's avatar
dengyihao 已提交
893
    if (!uv_is_closing((uv_handle_t*)conn->stream)) {
dengyihao's avatar
dengyihao 已提交
894
      uv_read_stop(conn->stream);
dengyihao's avatar
dengyihao 已提交
895 896
      uv_close((uv_handle_t*)conn->stream, cliDestroy);
    }
897
  }
dengyihao's avatar
dengyihao 已提交
898
}
U
ubuntu 已提交
899
static void cliDestroy(uv_handle_t* handle) {
dengyihao's avatar
dengyihao 已提交
900 901 902
  if (uv_handle_get_type(handle) != UV_TCP || handle->data == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
903
  SCliConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
904
  SCliThrd* pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
905 906
  if (conn->timer != NULL) {
    uv_timer_stop(conn->timer);
dengyihao's avatar
dengyihao 已提交
907 908
    taosArrayPush(pThrd->timerList, &conn->timer);
    conn->timer->data = NULL;
dengyihao's avatar
dengyihao 已提交
909 910
    conn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
911

dengyihao's avatar
dengyihao 已提交
912 913
  atomic_sub_fetch_32(&pThrd->connCount, 1);

dengyihao's avatar
dengyihao 已提交
914
  transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
915
  transRemoveExHandle(transGetRefMgt(), conn->refId);
wafwerar's avatar
wafwerar 已提交
916 917
  taosMemoryFree(conn->ip);
  taosMemoryFree(conn->stream);
dengyihao's avatar
dengyihao 已提交
918 919 920

  cliDestroyConnMsgs(conn, true);

dengyihao's avatar
dengyihao 已提交
921
  tTrace("%s conn %p destroy successfully", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
922
  transReqQueueClear(&conn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
923
  transDestroyBuffer(&conn->readBuf);
dengyihao's avatar
dengyihao 已提交
924

wafwerar's avatar
wafwerar 已提交
925
  taosMemoryFree(conn);
dengyihao's avatar
dengyihao 已提交
926
}
dengyihao's avatar
dengyihao 已提交
927
static bool cliHandleNoResp(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
928 929
  bool res = false;
  if (!transQueueEmpty(&conn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
930
    SCliMsg* pMsg = transQueueGet(&conn->cliMsgs, 0);
dengyihao's avatar
dengyihao 已提交
931
    if (REQUEST_NO_RESP(&pMsg->msg)) {
dengyihao's avatar
dengyihao 已提交
932
      transQueuePop(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
933 934 935 936 937
      destroyCmsg(pMsg);
      res = true;
    }
    if (res == true) {
      if (cliMaySendCachedMsg(conn) == false) {
dengyihao's avatar
dengyihao 已提交
938
        SCliThrd* thrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
939
        addConnToPool(thrd->pool, conn);
dengyihao's avatar
dengyihao 已提交
940 941 942
        res = false;
      } else {
        res = true;
dengyihao's avatar
dengyihao 已提交
943 944 945 946 947
      }
    }
  }
  return res;
}
U
ubuntu 已提交
948
static void cliSendCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
949 950
  SCliConn* pConn = transReqQueueRemove(req);
  if (pConn == NULL) return;
dengyihao's avatar
dengyihao 已提交
951

dengyihao's avatar
dengyihao 已提交
952 953 954
  SCliMsg* pMsg = !transQueueEmpty(&pConn->cliMsgs) ? transQueueGet(&pConn->cliMsgs, 0) : NULL;
  if (pMsg != NULL) {
    int64_t cost = taosGetTimestampUs() - pMsg->st;
dengyihao's avatar
dengyihao 已提交
955
    if (cost > 1000 * 20) {
dengyihao's avatar
dengyihao 已提交
956 957 958 959
      tWarn("%s conn %p send cost:%dus, send exception", CONN_GET_INST_LABEL(pConn), pConn, (int)cost);
    }
  }

dengyihao's avatar
dengyihao 已提交
960
  if (status == 0) {
dengyihao's avatar
dengyihao 已提交
961
    tTrace("%s conn %p data already was written out", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
962
  } else {
dengyihao's avatar
dengyihao 已提交
963 964 965 966
    if (!uv_is_closing((uv_handle_t*)&pConn->stream)) {
      tError("%s conn %p failed to write:%s", CONN_GET_INST_LABEL(pConn), pConn, uv_err_name(status));
      cliHandleExcept(pConn);
    }
dengyihao's avatar
dengyihao 已提交
967 968
    return;
  }
dengyihao's avatar
dengyihao 已提交
969
  if (cliHandleNoResp(pConn) == true) {
dengyihao's avatar
dengyihao 已提交
970
    tTrace("%s conn %p no resp required", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
971 972
    return;
  }
dengyihao's avatar
dengyihao 已提交
973
  uv_read_start((uv_stream_t*)pConn->stream, cliAllocRecvBufferCb, cliRecvCb);
dengyihao's avatar
dengyihao 已提交
974
}
dengyihao's avatar
dengyihao 已提交
975 976 977 978
void cliSendBatch(SCliConn* pConn) {
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;

dengyihao's avatar
dengyihao 已提交
979 980 981 982 983
  SCliBatch*     pBatch = pConn->pBatch;
  SCliBatchList* pList = pBatch->pList;
  pList->connCnt += 1;

  int32_t wLen = pBatch->wLen;
dengyihao's avatar
dengyihao 已提交
984 985 986 987

  uv_buf_t* wb = taosMemoryCalloc(wLen, sizeof(uv_buf_t));
  int       i = 0;

dengyihao's avatar
dengyihao 已提交
988 989
  queue* h = NULL;
  QUEUE_FOREACH(h, &pBatch->wq) {
dengyihao's avatar
dengyihao 已提交
990 991 992 993 994 995 996 997 998
    SCliMsg* pCliMsg = QUEUE_DATA(h, SCliMsg, q);

    STransConnCtx* pCtx = pCliMsg->ctx;

    STransMsg* pMsg = (STransMsg*)(&pCliMsg->msg);
    if (pMsg->pCont == 0) {
      pMsg->pCont = (void*)rpcMallocCont(0);
      pMsg->contLen = 0;
    }
dengyihao's avatar
dengyihao 已提交
999

dengyihao's avatar
dengyihao 已提交
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
    int            msgLen = transMsgLenFromCont(pMsg->contLen);
    STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);

    if (pHead->comp == 0) {
      pHead->ahandle = pCtx != NULL ? (uint64_t)pCtx->ahandle : 0;
      pHead->noResp = REQUEST_NO_RESP(pMsg) ? 1 : 0;
      pHead->persist = REQUEST_PERSIS_HANDLE(pMsg) ? 1 : 0;
      pHead->msgType = pMsg->msgType;
      pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
      pHead->release = REQUEST_RELEASE_HANDLE(pCliMsg) ? 1 : 0;
      memcpy(pHead->user, pTransInst->user, strlen(pTransInst->user));
      pHead->traceId = pMsg->info.traceId;
      pHead->magicNum = htonl(TRANS_MAGIC_NUM);
    }
    pHead->timestamp = taosHton64(taosGetTimestampUs());

dengyihao's avatar
dengyihao 已提交
1016 1017 1018 1019 1020 1021 1022 1023
    if (pHead->comp == 0) {
      if (pTransInst->compressSize != -1 && pTransInst->compressSize < pMsg->contLen) {
        msgLen = transCompressMsg(pMsg->pCont, pMsg->contLen) + sizeof(STransMsgHead);
        pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
      }
    } else {
      msgLen = (int32_t)ntohl((uint32_t)(pHead->msgLen));
    }
dengyihao's avatar
dengyihao 已提交
1024 1025 1026 1027 1028
    wb[i++] = uv_buf_init((char*)pHead, msgLen);
  }

  uv_write_t* req = taosMemoryCalloc(1, sizeof(uv_write_t));
  req->data = pConn;
dengyihao's avatar
dengyihao 已提交
1029
  tDebug("%s conn %p start to send batch msg, batch size:%d, msgLen:%d", CONN_GET_INST_LABEL(pConn), pConn,
dengyihao's avatar
dengyihao 已提交
1030
         pBatch->wLen, pBatch->batchSize);
dengyihao's avatar
dengyihao 已提交
1031 1032 1033
  uv_write(req, (uv_stream_t*)pConn->stream, wb, wLen, cliSendBatchCb);
  taosMemoryFree(wb);
}
U
ubuntu 已提交
1034
void cliSend(SCliConn* pConn) {
dengyihao's avatar
dengyihao 已提交
1035 1036 1037 1038 1039
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;

  if (transQueueEmpty(&pConn->cliMsgs)) {
    tError("%s conn %p not msg to send", pTransInst->label, pConn);
dengyihao's avatar
dengyihao 已提交
1040
    cliHandleExcept(pConn);
dengyihao's avatar
dengyihao 已提交
1041 1042
    return;
  }
dengyihao's avatar
dengyihao 已提交
1043 1044

  SCliMsg* pCliMsg = NULL;
U
ubuntu 已提交
1045
  CONN_GET_NEXT_SENDMSG(pConn);
dengyihao's avatar
dengyihao 已提交
1046 1047
  pCliMsg->sent = 1;

dengyihao's avatar
dengyihao 已提交
1048
  STransConnCtx* pCtx = pCliMsg->ctx;
dengyihao's avatar
dengyihao 已提交
1049

U
ubuntu 已提交
1050
  STransMsg* pMsg = (STransMsg*)(&pCliMsg->msg);
dengyihao's avatar
dengyihao 已提交
1051 1052 1053 1054
  if (pMsg->pCont == 0) {
    pMsg->pCont = (void*)rpcMallocCont(0);
    pMsg->contLen = 0;
  }
dengyihao's avatar
dengyihao 已提交
1055

dengyihao's avatar
dengyihao 已提交
1056
  int            msgLen = transMsgLenFromCont(pMsg->contLen);
dengyihao's avatar
dengyihao 已提交
1057
  STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
dengyihao's avatar
dengyihao 已提交
1058

dengyihao's avatar
dengyihao 已提交
1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
  if (pHead->comp == 0) {
    pHead->ahandle = pCtx != NULL ? (uint64_t)pCtx->ahandle : 0;
    pHead->noResp = REQUEST_NO_RESP(pMsg) ? 1 : 0;
    pHead->persist = REQUEST_PERSIS_HANDLE(pMsg) ? 1 : 0;
    pHead->msgType = pMsg->msgType;
    pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
    pHead->release = REQUEST_RELEASE_HANDLE(pCliMsg) ? 1 : 0;
    memcpy(pHead->user, pTransInst->user, strlen(pTransInst->user));
    pHead->traceId = pMsg->info.traceId;
    pHead->magicNum = htonl(TRANS_MAGIC_NUM);
  }
dengyihao's avatar
dengyihao 已提交
1070
  pHead->timestamp = taosHton64(taosGetTimestampUs());
dengyihao's avatar
dengyihao 已提交
1071

dengyihao's avatar
dengyihao 已提交
1072 1073 1074
  if (pHead->persist == 1) {
    CONN_SET_PERSIST_BY_APP(pConn);
  }
dengyihao's avatar
dengyihao 已提交
1075

dengyihao's avatar
dengyihao 已提交
1076
  STraceId* trace = &pMsg->info.traceId;
dengyihao's avatar
dengyihao 已提交
1077

dengyihao's avatar
dengyihao 已提交
1078
  if (pTransInst->startTimer != NULL && pTransInst->startTimer(0, pMsg->msgType)) {
dengyihao's avatar
dengyihao 已提交
1079
    uv_timer_t* timer = taosArrayGetSize(pThrd->timerList) > 0 ? *(uv_timer_t**)taosArrayPop(pThrd->timerList) : NULL;
dengyihao's avatar
dengyihao 已提交
1080 1081
    if (timer == NULL) {
      timer = taosMemoryCalloc(1, sizeof(uv_timer_t));
dengyihao's avatar
dengyihao 已提交
1082
      tDebug("no available timer, create a timer %p", timer);
dengyihao's avatar
dengyihao 已提交
1083 1084 1085 1086 1087
      uv_timer_init(pThrd->loop, timer);
    }
    timer->data = pConn;
    pConn->timer = timer;

dengyihao's avatar
dengyihao 已提交
1088 1089 1090
    tGTrace("%s conn %p start timer for msg:%s", CONN_GET_INST_LABEL(pConn), pConn, TMSG_INFO(pMsg->msgType));
    uv_timer_start((uv_timer_t*)pConn->timer, cliReadTimeoutCb, TRANS_READ_TIMEOUT, 0);
  }
dengyihao's avatar
dengyihao 已提交
1091

dengyihao's avatar
dengyihao 已提交
1092 1093 1094 1095 1096 1097 1098
  if (pHead->comp == 0) {
    if (pTransInst->compressSize != -1 && pTransInst->compressSize < pMsg->contLen) {
      msgLen = transCompressMsg(pMsg->pCont, pMsg->contLen) + sizeof(STransMsgHead);
      pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
    }
  } else {
    msgLen = (int32_t)ntohl((uint32_t)(pHead->msgLen));
dengyihao's avatar
dengyihao 已提交
1099
  }
dengyihao's avatar
dengyihao 已提交
1100

dengyihao's avatar
dengyihao 已提交
1101 1102
  tGDebug("%s conn %p %s is sent to %s, local info %s, len:%d", CONN_GET_INST_LABEL(pConn), pConn,
          TMSG_INFO(pHead->msgType), pConn->dst, pConn->src, msgLen);
dengyihao's avatar
dengyihao 已提交
1103

dengyihao's avatar
dengyihao 已提交
1104
  uv_buf_t    wb = uv_buf_init((char*)pHead, msgLen);
dengyihao's avatar
dengyihao 已提交
1105
  uv_write_t* req = transReqQueuePush(&pConn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
1106 1107 1108

  int status = uv_write(req, (uv_stream_t*)pConn->stream, &wb, 1, cliSendCb);
  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1109
    tGError("%s conn %p failed to send msg:%s, errmsg:%s", CONN_GET_INST_LABEL(pConn), pConn, TMSG_INFO(pMsg->msgType),
dengyihao's avatar
dengyihao 已提交
1110 1111 1112
            uv_err_name(status));
    cliHandleExcept(pConn);
  }
U
ubuntu 已提交
1113 1114
  return;
_RETURN:
dengyihao's avatar
dengyihao 已提交
1115
  return;
dengyihao's avatar
dengyihao 已提交
1116
}
dengyihao's avatar
dengyihao 已提交
1117 1118

static void cliDestroyBatch(SCliBatch* pBatch) {
dengyihao's avatar
dengyihao 已提交
1119
  if (pBatch == NULL) return;
dengyihao's avatar
dengyihao 已提交
1120
  while (!QUEUE_IS_EMPTY(&pBatch->wq)) {
dengyihao's avatar
dengyihao 已提交
1121 1122
    queue* h = QUEUE_HEAD(&pBatch->wq);
    QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
1123

dengyihao's avatar
dengyihao 已提交
1124
    SCliMsg* p = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
1125 1126
    destroyCmsg(p);
  }
dengyihao's avatar
dengyihao 已提交
1127 1128
  SCliBatchList* p = pBatch->pList;
  p->sending -= 1;
dengyihao's avatar
dengyihao 已提交
1129 1130
  taosMemoryFree(pBatch);
}
dengyihao's avatar
dengyihao 已提交
1131
static void cliHandleBatchReq(SCliBatch* pBatch, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1132 1133 1134 1135 1136
  if (pThrd->quit == true) {
    cliDestroyBatch(pBatch);
    return;
  }

dengyihao's avatar
dengyihao 已提交
1137
  if (pBatch == NULL || pBatch->wLen == 0 || QUEUE_IS_EMPTY(&pBatch->wq)) {
dengyihao's avatar
dengyihao 已提交
1138 1139
    return;
  }
dengyihao's avatar
dengyihao 已提交
1140 1141
  STrans*        pTransInst = pThrd->pTransInst;
  SCliBatchList* pList = pBatch->pList;
dengyihao's avatar
dengyihao 已提交
1142

dengyihao's avatar
dengyihao 已提交
1143 1144 1145
  char key[TSDB_FQDN_LEN + 64] = {0};
  CONN_CONSTRUCT_HASH_KEY(key, pList->ip, pList->port);

dengyihao's avatar
dengyihao 已提交
1146 1147
  bool      exceed = false;
  SCliConn* conn = getConnFromPool(pThrd, key, &exceed);
dengyihao's avatar
dengyihao 已提交
1148

dengyihao's avatar
dengyihao 已提交
1149 1150 1151
  if (conn == NULL && exceed) {
    tError("%s failed to send batch msg, batch size:%d, msgLen: %d, conn limit:%d", pTransInst->label, pBatch->wLen,
           pBatch->batchSize, pTransInst->connLimitNum);
dengyihao's avatar
dengyihao 已提交
1152
    cliDestroyBatch(pBatch);
dengyihao's avatar
dengyihao 已提交
1153 1154
    return;
  }
dengyihao's avatar
dengyihao 已提交
1155 1156
  if (conn == NULL) {
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
1157
    conn->pBatch = pBatch;
1158
    conn->ip = taosStrdup(pList->dst);
dengyihao's avatar
dengyihao 已提交
1159

dengyihao's avatar
dengyihao 已提交
1160
    uint32_t ipaddr = cliGetIpFromFqdnCache(pThrd->fqdn2ipCache, pList->ip);
dengyihao's avatar
dengyihao 已提交
1161 1162 1163 1164 1165 1166
    if (ipaddr == 0xffffffff) {
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

dengyihao's avatar
dengyihao 已提交
1167
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1168 1169 1170 1171 1172
      return;
    }
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = ipaddr;
dengyihao's avatar
dengyihao 已提交
1173
    addr.sin_port = (uint16_t)htons(pList->port);
dengyihao's avatar
dengyihao 已提交
1174

dengyihao's avatar
dengyihao 已提交
1175
    tTrace("%s conn %p try to connect to %s", pTransInst->label, conn, pList->dst);
dengyihao's avatar
dengyihao 已提交
1176
    int32_t fd = taosCreateSocketWithTimeout(TRANS_CONN_TIMEOUT * 10);
dengyihao's avatar
dengyihao 已提交
1177
    if (fd == -1) {
dengyihao's avatar
dengyihao 已提交
1178 1179 1180
      tError("%s conn %p failed to create socket, reason:%s", transLabel(pTransInst), conn,
             tstrerror(TAOS_SYSTEM_ERROR(errno)));
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1181 1182 1183 1184
      return;
    }
    int ret = uv_tcp_open((uv_tcp_t*)conn->stream, fd);
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1185 1186
      tError("%s conn %p failed to set stream, reason:%s", transLabel(pTransInst), conn, uv_err_name(ret));
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1187 1188 1189 1190
      return;
    }
    ret = transSetConnOption((uv_tcp_t*)conn->stream);
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1191 1192
      tError("%s conn %p failed to set socket opt, reason:%s", transLabel(pTransInst), conn, uv_err_name(ret));
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1193 1194 1195 1196 1197 1198 1199 1200 1201
      return;
    }

    ret = uv_tcp_connect(&conn->connReq, (uv_tcp_t*)(conn->stream), (const struct sockaddr*)&addr, cliConnCb);
    if (ret != 0) {
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;
dengyihao's avatar
dengyihao 已提交
1202
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1203 1204 1205 1206 1207 1208
      return;
    }
    uv_timer_start(conn->timer, cliConnTimeout, TRANS_CONN_TIMEOUT, 0);
    return;
  }

dengyihao's avatar
dengyihao 已提交
1209
  conn->pBatch = pBatch;
dengyihao's avatar
dengyihao 已提交
1210
  cliSendBatch(conn);
dengyihao's avatar
dengyihao 已提交
1211 1212
}
static void cliSendBatchCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
1213 1214 1215
  SCliConn*  conn = req->data;
  SCliThrd*  thrd = conn->hostThrd;
  SCliBatch* p = conn->pBatch;
dengyihao's avatar
dengyihao 已提交
1216

dengyihao's avatar
dengyihao 已提交
1217
  SCliBatchList* pBatchList = p->pList;
dengyihao's avatar
dengyihao 已提交
1218
  SCliBatch*     nxtBatch = cliGetHeadFromList(pBatchList);
dengyihao's avatar
dengyihao 已提交
1219 1220
  pBatchList->connCnt -= 1;

dengyihao's avatar
dengyihao 已提交
1221 1222 1223
  conn->pBatch = NULL;

  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1224
    tDebug("%s conn %p failed to send batch msg, batch size:%d, msgLen:%d, reason:%s", CONN_GET_INST_LABEL(conn), conn,
dengyihao's avatar
dengyihao 已提交
1225
           p->wLen, p->batchSize, uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
1226 1227 1228

    if (!uv_is_closing((uv_handle_t*)&conn->stream)) cliHandleExcept(conn);

dengyihao's avatar
dengyihao 已提交
1229
    cliHandleBatchReq(nxtBatch, thrd);
dengyihao's avatar
dengyihao 已提交
1230
  } else {
dengyihao's avatar
dengyihao 已提交
1231
    tDebug("%s conn %p succ to send batch msg, batch size:%d, msgLen:%d", CONN_GET_INST_LABEL(conn), conn, p->wLen,
dengyihao's avatar
dengyihao 已提交
1232
           p->batchSize);
dengyihao's avatar
dengyihao 已提交
1233
    if (!uv_is_closing((uv_handle_t*)&conn->stream)) {
dengyihao's avatar
dengyihao 已提交
1234 1235 1236 1237 1238 1239
      if (nxtBatch != NULL) {
        conn->pBatch = nxtBatch;
        cliSendBatch(conn);
      } else {
        addConnToPool(thrd->pool, conn);
      }
dengyihao's avatar
dengyihao 已提交
1240
    } else {
dengyihao's avatar
dengyihao 已提交
1241 1242
      cliDestroyBatch(nxtBatch);
      // conn release by other callback
dengyihao's avatar
dengyihao 已提交
1243
    }
dengyihao's avatar
dengyihao 已提交
1244
  }
dengyihao's avatar
dengyihao 已提交
1245 1246 1247

  cliDestroyBatch(p);
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
1248
}
dengyihao's avatar
dengyihao 已提交
1249
static void cliHandleFastFail(SCliConn* pConn, int status) {
dengyihao's avatar
dengyihao 已提交
1250
  SCliThrd* pThrd = pConn->hostThrd;
dengyihao's avatar
dengyihao 已提交
1251
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1252 1253 1254

  if (status == -1) status = ENETUNREACH;

dengyihao's avatar
dengyihao 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273
  if (pConn->pBatch == NULL) {
    SCliMsg* pMsg = transQueueGet(&pConn->cliMsgs, 0);

    STraceId* trace = &pMsg->msg.info.traceId;
    tGError("%s msg %s failed to send, conn %p failed to connect to %s, reason: %s", CONN_GET_INST_LABEL(pConn),
            TMSG_INFO(pMsg->msg.msgType), pConn, pConn->ip, uv_strerror(status));

    if (pMsg != NULL && REQUEST_NO_RESP(&pMsg->msg) &&
        (pTransInst->failFastFp != NULL && pTransInst->failFastFp(pMsg->msg.msgType))) {
      SFailFastItem* item = taosHashGet(pThrd->failFastCache, pConn->ip, strlen(pConn->ip));
      int64_t        cTimestamp = taosGetTimestampMs();
      if (item != NULL) {
        int32_t elapse = cTimestamp - item->timestamp;
        if (elapse >= 0 && elapse <= pTransInst->failFastInterval) {
          item->count++;
        } else {
          item->count = 1;
          item->timestamp = cTimestamp;
        }
dengyihao's avatar
dengyihao 已提交
1274
      } else {
dengyihao's avatar
dengyihao 已提交
1275 1276
        SFailFastItem item = {.count = 1, .timestamp = cTimestamp};
        taosHashPut(pThrd->failFastCache, pConn->ip, strlen(pConn->ip), &item, sizeof(SFailFastItem));
dengyihao's avatar
dengyihao 已提交
1277 1278
      }
    }
dengyihao's avatar
dengyihao 已提交
1279
  } else {
dengyihao's avatar
dengyihao 已提交
1280 1281
    tError("%s batch msg failed to send, conn %p failed to connect to %s, reason: %s", CONN_GET_INST_LABEL(pConn),
           pConn, pConn->ip, uv_strerror(status));
dengyihao's avatar
dengyihao 已提交
1282 1283
    cliDestroyBatch(pConn->pBatch);
    pConn->pBatch = NULL;
dengyihao's avatar
dengyihao 已提交
1284 1285 1286
  }
  cliHandleExcept(pConn);
}
dengyihao's avatar
dengyihao 已提交
1287

dengyihao's avatar
dengyihao 已提交
1288 1289 1290
void cliConnCb(uv_connect_t* req, int status) {
  SCliConn* pConn = req->data;
  SCliThrd* pThrd = pConn->hostThrd;
dengyihao's avatar
dengyihao 已提交
1291 1292 1293 1294 1295 1296 1297 1298 1299 1300
  bool      timeout = false;

  if (pConn->timer == NULL) {
    timeout = true;
  } else {
    uv_timer_stop(pConn->timer);
    pConn->timer->data = NULL;
    taosArrayPush(pThrd->timerList, &pConn->timer);
    pConn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
1301 1302

  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1303 1304 1305 1306 1307
    if (timeout == false) {
      cliHandleFastFail(pConn, status);
    } else if (timeout == true) {
      // already deal by timeout
    }
1308
    return;
dengyihao's avatar
dengyihao 已提交
1309
  }
dengyihao's avatar
dengyihao 已提交
1310

dengyihao's avatar
dengyihao 已提交
1311 1312
  struct sockaddr peername, sockname;
  int             addrlen = sizeof(peername);
dengyihao's avatar
dengyihao 已提交
1313
  uv_tcp_getpeername((uv_tcp_t*)pConn->stream, &peername, &addrlen);
dengyihao's avatar
dengyihao 已提交
1314
  transSockInfo2Str(&peername, pConn->dst);
dengyihao's avatar
dengyihao 已提交
1315

dengyihao's avatar
dengyihao 已提交
1316 1317
  addrlen = sizeof(sockname);
  uv_tcp_getsockname((uv_tcp_t*)pConn->stream, &sockname, &addrlen);
dengyihao's avatar
dengyihao 已提交
1318
  transSockInfo2Str(&sockname, pConn->src);
dengyihao's avatar
dengyihao 已提交
1319

dengyihao's avatar
dengyihao 已提交
1320
  tTrace("%s conn %p connect to server successfully", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
1321 1322 1323 1324 1325
  if (pConn->pBatch != NULL) {
    cliSendBatch(pConn);
  } else {
    cliSend(pConn);
  }
dengyihao's avatar
dengyihao 已提交
1326 1327
}

dengyihao's avatar
dengyihao 已提交
1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350
static void doNotifyApp(SCliMsg* pMsg, SCliThrd* pThrd) {
  STransConnCtx* pCtx = pMsg->ctx;
  STrans*        pTransInst = pThrd->pTransInst;

  STransMsg transMsg = {0};
  transMsg.contLen = 0;
  transMsg.pCont = NULL;
  transMsg.code = TSDB_CODE_RPC_MAX_SESSIONS;
  transMsg.msgType = pMsg->msg.msgType + 1;
  transMsg.info.ahandle = pMsg->ctx->ahandle;
  transMsg.info.traceId = pMsg->msg.info.traceId;
  transMsg.info.hasEpSet = false;
  if (pCtx->pSem != NULL) {
    if (pCtx->pRsp == NULL) {
    } else {
      memcpy((char*)pCtx->pRsp, (char*)&transMsg, sizeof(transMsg));
    }
  } else {
    pTransInst->cfp(pTransInst->parent, &transMsg, NULL);
  }

  destroyCmsg(pMsg);
}
dengyihao's avatar
dengyihao 已提交
1351
static void cliHandleQuit(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1352 1353 1354 1355 1356
  if (!transAsyncPoolIsEmpty(pThrd->asyncPool)) {
    pThrd->stopMsg = pMsg;
    return;
  }
  pThrd->stopMsg = NULL;
dengyihao's avatar
dengyihao 已提交
1357
  pThrd->quit = true;
U
ubuntu 已提交
1358
  tDebug("cli work thread %p start to quit", pThrd);
dengyihao's avatar
dengyihao 已提交
1359
  destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1360

dengyihao's avatar
dengyihao 已提交
1361
  destroyConnPool(pThrd);
dengyihao's avatar
dengyihao 已提交
1362
  uv_walk(pThrd->loop, cliWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
1363
}
dengyihao's avatar
dengyihao 已提交
1364
static void cliHandleRelease(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1365
  int64_t    refId = (int64_t)(pMsg->msg.info.handle);
dengyihao's avatar
dengyihao 已提交
1366
  SExHandle* exh = transAcquireExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1367
  if (exh == NULL) {
dengyihao's avatar
dengyihao 已提交
1368
    tDebug("%" PRId64 " already released", refId);
dengyihao's avatar
dengyihao 已提交
1369 1370
    destroyCmsg(pMsg);
    return;
dengyihao's avatar
dengyihao 已提交
1371 1372 1373
  }

  SCliConn* conn = exh->handle;
dengyihao's avatar
dengyihao 已提交
1374
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1375
  tDebug("%s conn %p start to release to inst", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
1376

dengyihao's avatar
dengyihao 已提交
1377 1378
  if (T_REF_VAL_GET(conn) == 2) {
    transUnrefCliHandle(conn);
dengyihao's avatar
dengyihao 已提交
1379 1380
    if (!transQueuePush(&conn->cliMsgs, pMsg)) {
      return;
dengyihao's avatar
dengyihao 已提交
1381 1382
    }
    cliSend(conn);
dengyihao's avatar
dengyihao 已提交
1383 1384 1385
  } else {
    tError("%s conn %p already released", CONN_GET_INST_LABEL(conn), conn);
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1386 1387
  }
}
dengyihao's avatar
dengyihao 已提交
1388
static void cliHandleUpdate(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1389
  STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
dengyihao 已提交
1390
  pThrd->cvtAddr = pCtx->cvtAddr;
dengyihao's avatar
dengyihao 已提交
1391 1392
  destroyCmsg(pMsg);
}
U
ubuntu 已提交
1393

dengyihao's avatar
dengyihao 已提交
1394 1395
SCliConn* cliGetConn(SCliMsg** pMsg, SCliThrd* pThrd, bool* ignore, char* addr) {
  STransConnCtx* pCtx = (*pMsg)->ctx;
dengyihao's avatar
dengyihao 已提交
1396 1397
  SCliConn*      conn = NULL;

dengyihao's avatar
dengyihao 已提交
1398
  int64_t refId = (int64_t)((*pMsg)->msg.info.handle);
dengyihao's avatar
dengyihao 已提交
1399
  if (refId != 0) {
dengyihao's avatar
dengyihao 已提交
1400
    SExHandle* exh = transAcquireExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1401
    if (exh == NULL) {
dengyihao's avatar
dengyihao 已提交
1402
      tError("failed to get conn, refId: %" PRId64 "", refId);
dengyihao's avatar
dengyihao 已提交
1403 1404
      *ignore = true;
      return NULL;
dengyihao's avatar
dengyihao 已提交
1405 1406
    } else {
      conn = exh->handle;
dengyihao's avatar
dengyihao 已提交
1407
      if (conn == NULL) {
dengyihao's avatar
dengyihao 已提交
1408
        conn = getConnFromPool2(pThrd, addr, pMsg);
dengyihao's avatar
dengyihao 已提交
1409
        if (conn != NULL) specifyConnRef(conn, true, refId);
dengyihao's avatar
dengyihao 已提交
1410
      }
dengyihao's avatar
dengyihao 已提交
1411
      transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1412 1413 1414
    }
    return conn;
  };
dengyihao's avatar
dengyihao 已提交
1415

dengyihao's avatar
dengyihao 已提交
1416
  conn = getConnFromPool2(pThrd, addr, pMsg);
dengyihao's avatar
dengyihao 已提交
1417
  if (conn != NULL) {
dengyihao's avatar
dengyihao 已提交
1418
    tTrace("%s conn %p get from conn pool:%p", CONN_GET_INST_LABEL(conn), conn, pThrd->pool);
dengyihao's avatar
dengyihao 已提交
1419
  } else {
dengyihao's avatar
dengyihao 已提交
1420
    tTrace("%s not found conn in conn pool:%p", ((STrans*)pThrd->pTransInst)->label, pThrd->pool);
dengyihao's avatar
dengyihao 已提交
1421
  }
dengyihao's avatar
dengyihao 已提交
1422 1423
  return conn;
}
dengyihao's avatar
dengyihao 已提交
1424
FORCE_INLINE void cliMayCvtFqdnToIp(SEpSet* pEpSet, SCvtAddr* pCvtAddr) {
dengyihao's avatar
dengyihao 已提交
1425 1426 1427
  if (pCvtAddr->cvt == false) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1428 1429 1430
  if (pEpSet->numOfEps == 1 && strncmp(pEpSet->eps[0].fqdn, pCvtAddr->fqdn, TSDB_FQDN_LEN) == 0) {
    memset(pEpSet->eps[0].fqdn, 0, TSDB_FQDN_LEN);
    memcpy(pEpSet->eps[0].fqdn, pCvtAddr->ip, TSDB_FQDN_LEN);
dengyihao's avatar
dengyihao 已提交
1431 1432
  }
}
dengyihao's avatar
dengyihao 已提交
1433

dengyihao's avatar
dengyihao 已提交
1434
FORCE_INLINE bool cliIsEpsetUpdated(int32_t code, STransConnCtx* pCtx) {
dengyihao's avatar
dengyihao 已提交
1435
  if (code != 0) return false;
dengyihao's avatar
dengyihao 已提交
1436
  // if (pCtx->retryCnt == 0) return false;
dengyihao's avatar
dengyihao 已提交
1437 1438 1439
  if (transEpSetIsEqual(&pCtx->epSet, &pCtx->origEpSet)) return false;
  return true;
}
dengyihao's avatar
dengyihao 已提交
1440
FORCE_INLINE int32_t cliBuildExceptResp(SCliMsg* pMsg, STransMsg* pResp) {
dengyihao's avatar
dengyihao 已提交
1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
  if (pMsg == NULL) return -1;

  memset(pResp, 0, sizeof(STransMsg));

  pResp->code = TSDB_CODE_RPC_BROKEN_LINK;
  pResp->msgType = pMsg->msg.msgType + 1;
  pResp->info.ahandle = pMsg->ctx ? pMsg->ctx->ahandle : NULL;
  pResp->info.traceId = pMsg->msg.info.traceId;

  return 0;
}
dengyihao's avatar
dengyihao 已提交
1452 1453 1454 1455 1456
static FORCE_INLINE uint32_t cliGetIpFromFqdnCache(SHashObj* cache, char* fqdn) {
  uint32_t  addr = 0;
  uint32_t* v = taosHashGet(cache, fqdn, strlen(fqdn));
  if (v == NULL) {
    addr = taosGetIpv4FromFqdn(fqdn);
1457 1458 1459 1460
    if (addr == 0xffffffff) {
      terrno = TAOS_SYSTEM_ERROR(errno);
      tError("failed to get ip from fqdn:%s since %s", fqdn, terrstr());
      return addr;
dengyihao's avatar
dengyihao 已提交
1461 1462
    }

dengyihao's avatar
dengyihao 已提交
1463 1464 1465 1466 1467 1468 1469 1470 1471 1472
    taosHashPut(cache, fqdn, strlen(fqdn), &addr, sizeof(addr));
  } else {
    addr = *v;
  }
  return addr;
}
static FORCE_INLINE void cliUpdateFqdnCache(SHashObj* cache, char* fqdn) {
  // impl later
  return;
}
dengyihao's avatar
dengyihao 已提交
1473

dengyihao's avatar
dengyihao 已提交
1474 1475 1476 1477 1478
static void doFreeTimeoutMsg(void* param) {
  STaskArg* arg = param;
  SCliMsg*  pMsg = arg->param1;
  SCliThrd* pThrd = arg->param2;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1479

dengyihao's avatar
dengyihao 已提交
1480 1481 1482 1483 1484 1485
  QUEUE_REMOVE(&pMsg->q);
  STraceId* trace = &pMsg->msg.info.traceId;
  tGTrace("%s msg %s cannot get available conn after timeout", pTransInst->label, TMSG_INFO(pMsg->msg.msgType));
  doNotifyApp(pMsg, pThrd);
  taosMemoryFree(arg);
}
dengyihao's avatar
dengyihao 已提交
1486
void cliHandleReq(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1487
  STrans* pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1488

dengyihao's avatar
dengyihao 已提交
1489 1490
  cliMayCvtFqdnToIp(&pMsg->ctx->epSet, &pThrd->cvtAddr);
  if (!EPSET_IS_VALID(&pMsg->ctx->epSet)) {
dengyihao's avatar
dengyihao 已提交
1491
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1492 1493
    return;
  }
dengyihao's avatar
dengyihao 已提交
1494

dengyihao's avatar
dengyihao 已提交
1495 1496
  char*    fqdn = EPSET_GET_INUSE_IP(&pMsg->ctx->epSet);
  uint16_t port = EPSET_GET_INUSE_PORT(&pMsg->ctx->epSet);
dengyihao's avatar
dengyihao 已提交
1497 1498
  char     addr[TSDB_FQDN_LEN + 64] = {0};
  CONN_CONSTRUCT_HASH_KEY(addr, fqdn, port);
dengyihao's avatar
dengyihao 已提交
1499

dengyihao's avatar
dengyihao 已提交
1500
  bool      ignore = false;
dengyihao's avatar
dengyihao 已提交
1501
  SCliConn* conn = cliGetConn(&pMsg, pThrd, &ignore, addr);
dengyihao's avatar
dengyihao 已提交
1502
  if (ignore == true) {
dengyihao's avatar
dengyihao 已提交
1503
    // persist conn already release by server
dengyihao's avatar
dengyihao 已提交
1504 1505
    STransMsg resp;
    cliBuildExceptResp(pMsg, &resp);
dengyihao's avatar
dengyihao 已提交
1506 1507 1508
    if (pMsg->type != Release) {
      pTransInst->cfp(pTransInst->parent, &resp, NULL);
    }
dengyihao's avatar
dengyihao 已提交
1509
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1510 1511
    return;
  }
dengyihao's avatar
dengyihao 已提交
1512
  if (conn == NULL && pMsg == NULL) {
dengyihao's avatar
dengyihao 已提交
1513 1514
    return;
  }
dengyihao's avatar
dengyihao 已提交
1515
  STraceId* trace = &pMsg->msg.info.traceId;
dengyihao's avatar
dengyihao 已提交
1516

dengyihao's avatar
dengyihao 已提交
1517
  if (conn != NULL) {
dengyihao's avatar
dengyihao 已提交
1518
    transCtxMerge(&conn->ctx, &pMsg->ctx->appCtx);
dengyihao's avatar
dengyihao 已提交
1519
    transQueuePush(&conn->cliMsgs, pMsg);
U
ubuntu 已提交
1520
    cliSend(conn);
dengyihao's avatar
dengyihao 已提交
1521
  } else {
U
ubuntu 已提交
1522
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
1523 1524 1525 1526

    int64_t refId = (int64_t)pMsg->msg.info.handle;
    if (refId != 0) specifyConnRef(conn, true, refId);

dengyihao's avatar
dengyihao 已提交
1527
    transCtxMerge(&conn->ctx, &pMsg->ctx->appCtx);
dengyihao's avatar
dengyihao 已提交
1528
    transQueuePush(&conn->cliMsgs, pMsg);
dengyihao's avatar
dengyihao 已提交
1529

dengyihao's avatar
dengyihao 已提交
1530
    conn->ip = taosStrdup(addr);
dengyihao's avatar
dengyihao 已提交
1531

dengyihao's avatar
dengyihao 已提交
1532
    uint32_t ipaddr = cliGetIpFromFqdnCache(pThrd->fqdn2ipCache, fqdn);
1533 1534 1535 1536 1537 1538 1539 1540 1541
    if (ipaddr == 0xffffffff) {
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

      cliHandleExcept(conn);
      return;
    }
dengyihao's avatar
dengyihao 已提交
1542

dengyihao's avatar
dengyihao 已提交
1543
    struct sockaddr_in addr;
1544
    addr.sin_family = AF_INET;
1545
    addr.sin_addr.s_addr = ipaddr;
dengyihao's avatar
dengyihao 已提交
1546
    addr.sin_port = (uint16_t)htons(port);
dengyihao's avatar
dengyihao 已提交
1547

dengyihao's avatar
dengyihao 已提交
1548
    tGTrace("%s conn %p try to connect to %s", pTransInst->label, conn, conn->ip);
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568
    int32_t fd = taosCreateSocketWithTimeout(TRANS_CONN_TIMEOUT * 4);
    if (fd == -1) {
      tGError("%s conn %p failed to create socket, reason:%s", transLabel(pTransInst), conn,
              tstrerror(TAOS_SYSTEM_ERROR(errno)));
      cliHandleExcept(conn);
      errno = 0;
      return;
    }
    int ret = uv_tcp_open((uv_tcp_t*)conn->stream, fd);
    if (ret != 0) {
      tGError("%s conn %p failed to set stream, reason:%s", transLabel(pTransInst), conn, uv_err_name(ret));
      cliHandleExcept(conn);
      return;
    }
    ret = transSetConnOption((uv_tcp_t*)conn->stream);
    if (ret != 0) {
      tGError("%s conn %p failed to set socket opt, reason:%s", transLabel(pTransInst), conn, uv_err_name(ret));
      cliHandleExcept(conn);
      return;
    }
dengyihao's avatar
dengyihao 已提交
1569

1570
    ret = uv_tcp_connect(&conn->connReq, (uv_tcp_t*)(conn->stream), (const struct sockaddr*)&addr, cliConnCb);
dengyihao's avatar
dengyihao 已提交
1571
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1572 1573 1574 1575 1576
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

dengyihao's avatar
dengyihao 已提交
1577
      cliHandleFastFail(conn, ret);
dengyihao's avatar
dengyihao 已提交
1578 1579
      return;
    }
dengyihao's avatar
dengyihao 已提交
1580
    uv_timer_start(conn->timer, cliConnTimeout, TRANS_CONN_TIMEOUT, 0);
dengyihao's avatar
dengyihao 已提交
1581
  }
dengyihao's avatar
dengyihao 已提交
1582
  tGTrace("%s conn %p ready", pTransInst->label, conn);
dengyihao's avatar
dengyihao 已提交
1583
}
dengyihao's avatar
dengyihao 已提交
1584

dengyihao's avatar
dengyihao 已提交
1585
static void cliNoBatchDealReq(queue* wq, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1586
  int count = 0;
dengyihao's avatar
dengyihao 已提交
1587

dengyihao's avatar
dengyihao 已提交
1588 1589
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
dengyihao's avatar
dengyihao 已提交
1590 1591 1592
    QUEUE_REMOVE(h);

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
1593 1594 1595 1596 1597

    if (pMsg->type == Quit) {
      pThrd->stopMsg = pMsg;
      continue;
    }
dengyihao's avatar
dengyihao 已提交
1598 1599 1600 1601 1602 1603 1604 1605
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);

    count++;
  }
  if (count >= 2) {
    tTrace("cli process batch size:%d", count);
  }
}
dengyihao's avatar
dengyihao 已提交
1606
SCliBatch* cliGetHeadFromList(SCliBatchList* pList) {
dengyihao's avatar
dengyihao 已提交
1607
  if (QUEUE_IS_EMPTY(&pList->wq) || pList->connCnt > pList->connMax || pList->sending > pList->connMax) {
dengyihao's avatar
dengyihao 已提交
1608 1609 1610 1611
    return NULL;
  }
  queue* hr = QUEUE_HEAD(&pList->wq);
  QUEUE_REMOVE(hr);
dengyihao's avatar
dengyihao 已提交
1612
  pList->sending += 1;
dengyihao's avatar
dengyihao 已提交
1613 1614 1615 1616 1617 1618

  pList->len -= 1;

  SCliBatch* batch = QUEUE_DATA(hr, SCliBatch, listq);
  return batch;
}
dengyihao's avatar
dengyihao 已提交
1619

dengyihao's avatar
dengyihao 已提交
1620
static void cliBatchDealReq(queue* wq, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1621 1622
  STrans* pInst = pThrd->pTransInst;

dengyihao's avatar
dengyihao 已提交
1623
  int count = 0;
dengyihao's avatar
dengyihao 已提交
1624 1625
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
dengyihao's avatar
dengyihao 已提交
1626
    QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
1627 1628

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
1629 1630 1631 1632 1633 1634

    if (pMsg->type == Quit) {
      pThrd->stopMsg = pMsg;
      continue;
    }

dengyihao's avatar
dengyihao 已提交
1635
    if (pMsg->type == Normal && REQUEST_NO_RESP(&pMsg->msg)) {
dengyihao's avatar
dengyihao 已提交
1636 1637 1638 1639 1640 1641 1642
      STransConnCtx* pCtx = pMsg->ctx;

      char*    ip = EPSET_GET_INUSE_IP(&pCtx->epSet);
      uint32_t port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
      char     key[TSDB_FQDN_LEN + 64] = {0};
      CONN_CONSTRUCT_HASH_KEY(key, ip, port);

dengyihao's avatar
dengyihao 已提交
1643 1644 1645 1646 1647
      // SCliBatch** ppBatch = taosHashGet(pThrd->batchCache, key, sizeof(key));
      SCliBatchList** ppBatchList = taosHashGet(pThrd->batchCache, key, sizeof(key));
      if (ppBatchList == NULL || *ppBatchList == NULL) {
        SCliBatchList* pBatchList = taosMemoryCalloc(1, sizeof(SCliBatchList));
        QUEUE_INIT(&pBatchList->wq);
dengyihao's avatar
dengyihao 已提交
1648
        pBatchList->connMax = pInst->connLimitNum;
dengyihao's avatar
dengyihao 已提交
1649
        pBatchList->connCnt = 0;
dengyihao's avatar
dengyihao 已提交
1650
        pBatchList->batchLenLimit = pInst->batchSize;
dengyihao's avatar
dengyihao 已提交
1651
        pBatchList->len += 1;
dengyihao's avatar
dengyihao 已提交
1652

1653 1654
        pBatchList->ip = taosStrdup(ip);
        pBatchList->dst = taosStrdup(key);
dengyihao's avatar
dengyihao 已提交
1655 1656
        pBatchList->port = port;

dengyihao's avatar
dengyihao 已提交
1657 1658
        SCliBatch* pBatch = taosMemoryCalloc(1, sizeof(SCliBatch));
        QUEUE_INIT(&pBatch->wq);
dengyihao's avatar
dengyihao 已提交
1659 1660
        QUEUE_INIT(&pBatch->listq);

dengyihao's avatar
dengyihao 已提交
1661 1662 1663
        QUEUE_PUSH(&pBatch->wq, h);
        pBatch->wLen += 1;
        pBatch->batchSize += pMsg->msg.contLen;
dengyihao's avatar
dengyihao 已提交
1664
        pBatch->pList = pBatchList;
dengyihao's avatar
dengyihao 已提交
1665

dengyihao's avatar
dengyihao 已提交
1666
        QUEUE_PUSH(&pBatchList->wq, &pBatch->listq);
dengyihao's avatar
dengyihao 已提交
1667

dengyihao's avatar
dengyihao 已提交
1668
        taosHashPut(pThrd->batchCache, key, sizeof(key), &pBatchList, sizeof(void*));
dengyihao's avatar
dengyihao 已提交
1669
      } else {
dengyihao's avatar
dengyihao 已提交
1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690
        if (QUEUE_IS_EMPTY(&(*ppBatchList)->wq)) {
          SCliBatch* pBatch = taosMemoryCalloc(1, sizeof(SCliBatch));
          QUEUE_INIT(&pBatch->wq);
          QUEUE_INIT(&pBatch->listq);

          QUEUE_PUSH(&pBatch->wq, h);
          pBatch->wLen += 1;
          pBatch->batchSize = pMsg->msg.contLen;
          pBatch->pList = *ppBatchList;

          QUEUE_PUSH(&((*ppBatchList)->wq), &pBatch->listq);
          (*ppBatchList)->len += 1;

          continue;
        }

        queue*     hdr = QUEUE_TAIL(&((*ppBatchList)->wq));
        SCliBatch* pBatch = QUEUE_DATA(hdr, SCliBatch, listq);
        if ((pBatch->batchSize + pMsg->msg.contLen) < (*ppBatchList)->batchLenLimit) {
          QUEUE_PUSH(&pBatch->wq, h);
          pBatch->batchSize += pMsg->msg.contLen;
dengyihao's avatar
dengyihao 已提交
1691
          pBatch->wLen += 1;
dengyihao's avatar
dengyihao 已提交
1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704
        } else {
          SCliBatch* pBatch = taosMemoryCalloc(1, sizeof(SCliBatch));
          QUEUE_INIT(&pBatch->wq);
          QUEUE_INIT(&pBatch->listq);

          QUEUE_PUSH(&pBatch->wq, h);
          pBatch->wLen += 1;
          pBatch->batchSize += pMsg->msg.contLen;
          pBatch->pList = *ppBatchList;

          QUEUE_PUSH(&((*ppBatchList)->wq), &pBatch->listq);
          (*ppBatchList)->len += 1;
        }
dengyihao's avatar
dengyihao 已提交
1705
      }
dengyihao's avatar
dengyihao 已提交
1706
      continue;
dengyihao's avatar
dengyihao 已提交
1707
    }
dengyihao's avatar
dengyihao 已提交
1708
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
1709
    count++;
dengyihao's avatar
dengyihao 已提交
1710
  }
dengyihao's avatar
dengyihao 已提交
1711

dengyihao's avatar
dengyihao 已提交
1712
  void** pIter = taosHashIterate(pThrd->batchCache, NULL);
dengyihao's avatar
dengyihao 已提交
1713
  while (pIter != NULL) {
dengyihao's avatar
dengyihao 已提交
1714
    SCliBatchList* batchList = (SCliBatchList*)(*pIter);
dengyihao's avatar
dengyihao 已提交
1715 1716 1717
    SCliBatch*     batch = cliGetHeadFromList(batchList);
    if (batch != NULL) {
      cliHandleBatchReq(batch, pThrd);
dengyihao's avatar
dengyihao 已提交
1718
    }
dengyihao's avatar
dengyihao 已提交
1719
    pIter = (void**)taosHashIterate(pThrd->batchCache, pIter);
dengyihao's avatar
dengyihao 已提交
1720 1721
  }

dengyihao's avatar
dengyihao 已提交
1722
  if (count >= 2) {
S
Shengliang Guan 已提交
1723
    tTrace("cli process batch size:%d", count);
dengyihao's avatar
dengyihao 已提交
1724
  }
dengyihao's avatar
dengyihao 已提交
1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737
}

static void cliAsyncCb(uv_async_t* handle) {
  SAsyncItem* item = handle->data;
  SCliThrd*   pThrd = item->pThrd;
  STrans*     pTransInst = pThrd->pTransInst;

  // batch process to avoid to lock/unlock frequently
  queue wq;
  taosThreadMutexLock(&item->mtx);
  QUEUE_MOVE(&item->qmsg, &wq);
  taosThreadMutexUnlock(&item->mtx);

dengyihao's avatar
dengyihao 已提交
1738
  int8_t supportBatch = pTransInst->supportBatch;
dengyihao's avatar
dengyihao 已提交
1739
  if (supportBatch == 0) {
dengyihao's avatar
dengyihao 已提交
1740
    cliNoBatchDealReq(&wq, pThrd);
dengyihao's avatar
dengyihao 已提交
1741
  } else if (supportBatch == 1) {
dengyihao's avatar
dengyihao 已提交
1742
    cliBatchDealReq(&wq, pThrd);
dengyihao's avatar
dengyihao 已提交
1743
  }
dengyihao's avatar
dengyihao 已提交
1744

dengyihao's avatar
dengyihao 已提交
1745
  if (pThrd->stopMsg != NULL) cliHandleQuit(pThrd->stopMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
1746
}
dengyihao's avatar
dengyihao 已提交
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772
static void cliPrepareCb(uv_prepare_t* handle) {
  SCliThrd* thrd = handle->data;
  tTrace("prepare work start");

  SAsyncPool* pool = thrd->asyncPool;
  for (int i = 0; i < pool->nAsync; i++) {
    uv_async_t* async = &(pool->asyncs[i]);
    SAsyncItem* item = async->data;

    queue wq;
    taosThreadMutexLock(&item->mtx);
    QUEUE_MOVE(&item->qmsg, &wq);
    taosThreadMutexUnlock(&item->mtx);

    int count = 0;
    while (!QUEUE_IS_EMPTY(&wq)) {
      queue* h = QUEUE_HEAD(&wq);
      QUEUE_REMOVE(h);

      SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
      (*cliAsyncHandle[pMsg->type])(pMsg, thrd);
      count++;
    }
  }
  tTrace("prepare work end");
  if (thrd->stopMsg != NULL) cliHandleQuit(thrd->stopMsg, thrd);
dengyihao's avatar
dengyihao 已提交
1773
}
dengyihao's avatar
dengyihao 已提交
1774

dengyihao's avatar
dengyihao 已提交
1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803
void cliDestroyConnMsgs(SCliConn* conn, bool destroy) {
  transCtxCleanup(&conn->ctx);
  cliReleaseUnfinishedMsg(conn);
  if (destroy == 1) {
    transQueueDestroy(&conn->cliMsgs);
  } else {
    transQueueClear(&conn->cliMsgs);
  }
}

void cliIteraConnMsgs(SCliConn* conn) {
  SCliThrd* pThrd = conn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;

  for (int i = 0; i < transQueueSize(&conn->cliMsgs); i++) {
    SCliMsg* cmsg = transQueueGet(&conn->cliMsgs, i);
    if (cmsg->type == Release || REQUEST_NO_RESP(&cmsg->msg) || cmsg->msg.msgType == TDMT_SCH_DROP_TASK) {
      continue;
    }

    STransMsg resp = {0};
    if (-1 == cliBuildExceptResp(cmsg, &resp)) {
      continue;
    }
    pTransInst->cfp(pTransInst->parent, &resp, NULL);

    cmsg->ctx->ahandle = NULL;
  }
}
dengyihao's avatar
dengyihao 已提交
1804 1805 1806
bool cliRecvReleaseReq(SCliConn* conn, STransMsgHead* pHead) {
  if (pHead->release == 1 && (pHead->msgLen) == sizeof(*pHead)) {
    uint64_t ahandle = pHead->ahandle;
dengyihao's avatar
dengyihao 已提交
1807
    tDebug("ahandle = %" PRIu64 "", ahandle);
dengyihao's avatar
dengyihao 已提交
1808 1809
    SCliMsg* pMsg = NULL;
    CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle);
dengyihao's avatar
dengyihao 已提交
1810

dengyihao's avatar
dengyihao 已提交
1811 1812
    transClearBuffer(&conn->readBuf);
    transFreeMsg(transContFromHead((char*)pHead));
dengyihao's avatar
dengyihao 已提交
1813 1814 1815 1816

    for (int i = 0; ahandle == 0 && i < transQueueSize(&conn->cliMsgs); i++) {
      SCliMsg* cliMsg = transQueueGet(&conn->cliMsgs, i);
      if (cliMsg->type == Release) {
dengyihao's avatar
dengyihao 已提交
1817
        ASSERTS(pMsg == NULL, "trans-cli recv invaid release-req");
dengyihao's avatar
dengyihao 已提交
1818 1819
        return true;
      }
dengyihao's avatar
dengyihao 已提交
1820
    }
dengyihao's avatar
dengyihao 已提交
1821 1822 1823

    cliIteraConnMsgs(conn);

dengyihao's avatar
dengyihao 已提交
1824 1825
    tDebug("%s conn %p receive release request, refId:%" PRId64 "", CONN_GET_INST_LABEL(conn), conn, conn->refId);
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1826

dengyihao's avatar
dengyihao 已提交
1827 1828 1829 1830 1831 1832
    addConnToPool(((SCliThrd*)conn->hostThrd)->pool, conn);
    return true;
  }
  return false;
}

U
ubuntu 已提交
1833
static void* cliWorkThread(void* arg) {
dengyihao's avatar
dengyihao 已提交
1834
  SCliThrd* pThrd = (SCliThrd*)arg;
dengyihao's avatar
dengyihao 已提交
1835
  pThrd->pid = taosGetSelfPthreadId();
G
gccgdb1234 已提交
1836
  setThreadName("trans-cli-work");
dengyihao's avatar
dengyihao 已提交
1837
  uv_run(pThrd->loop, UV_RUN_DEFAULT);
dengyihao's avatar
dengyihao 已提交
1838 1839

  tDebug("thread quit-thread:%08" PRId64, pThrd->pid);
1840
  return NULL;
dengyihao's avatar
dengyihao 已提交
1841 1842
}

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

dengyihao's avatar
dengyihao 已提交
1846
  STrans* pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1847
  memcpy(cli->label, label, TSDB_LABEL_LEN);
dengyihao's avatar
dengyihao 已提交
1848
  cli->numOfThreads = numOfThreads;
dengyihao's avatar
dengyihao 已提交
1849
  cli->pThreadObj = (SCliThrd**)taosMemoryCalloc(cli->numOfThreads, sizeof(SCliThrd*));
dengyihao's avatar
dengyihao 已提交
1850 1851

  for (int i = 0; i < cli->numOfThreads; i++) {
dengyihao's avatar
dengyihao 已提交
1852
    SCliThrd* pThrd = createThrdObj(shandle);
dengyihao's avatar
dengyihao 已提交
1853
    if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
1854
      goto _err;
dengyihao's avatar
dengyihao 已提交
1855 1856 1857
    }

    int err = taosThreadCreate(&pThrd->thread, NULL, cliWorkThread, (void*)(pThrd));
dengyihao's avatar
dengyihao 已提交
1858 1859 1860
    if (err != 0) {
      goto _err;
    } else {
S
Shengliang Guan 已提交
1861
      tDebug("success to create tranport-cli thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1862
    }
dengyihao's avatar
dengyihao 已提交
1863
    cli->pThreadObj[i] = pThrd;
dengyihao's avatar
dengyihao 已提交
1864 1865
  }
  return cli;
dengyihao's avatar
dengyihao 已提交
1866 1867 1868 1869 1870

_err:
  taosMemoryFree(cli->pThreadObj);
  taosMemoryFree(cli);
  return NULL;
dengyihao's avatar
dengyihao 已提交
1871
}
dengyihao's avatar
dengyihao 已提交
1872

dengyihao's avatar
dengyihao 已提交
1873
static FORCE_INLINE void destroyUserdata(STransMsg* userdata) {
dengyihao's avatar
dengyihao 已提交
1874 1875 1876 1877 1878 1879
  if (userdata->pCont == NULL) {
    return;
  }
  transFreeMsg(userdata->pCont);
  userdata->pCont = NULL;
}
dengyihao's avatar
dengyihao 已提交
1880

dengyihao's avatar
dengyihao 已提交
1881
static FORCE_INLINE void destroyCmsg(void* arg) {
dengyihao's avatar
dengyihao 已提交
1882
  SCliMsg* pMsg = arg;
dengyihao's avatar
dengyihao 已提交
1883 1884 1885
  if (pMsg == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1886

dengyihao's avatar
dengyihao 已提交
1887 1888
  transDestroyConnCtx(pMsg->ctx);
  destroyUserdata(&pMsg->msg);
wafwerar's avatar
wafwerar 已提交
1889
  taosMemoryFree(pMsg);
dengyihao's avatar
dengyihao 已提交
1890
}
dengyihao's avatar
dengyihao 已提交
1891

dengyihao's avatar
dengyihao 已提交
1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910
static FORCE_INLINE void destroyCmsgAndAhandle(void* param) {
  if (param == NULL) return;

  STaskArg* arg = param;
  SCliMsg*  pMsg = arg->param1;
  SCliThrd* pThrd = arg->param2;

  tDebug("destroy Ahandle A");
  if (pThrd != NULL && pThrd->destroyAhandleFp != NULL) {
    tDebug("destroy Ahandle B");
    pThrd->destroyAhandleFp(pMsg->ctx->ahandle);
  }
  tDebug("destroy Ahandle C");

  transDestroyConnCtx(pMsg->ctx);
  destroyUserdata(&pMsg->msg);
  taosMemoryFree(pMsg);
}

dengyihao's avatar
dengyihao 已提交
1911 1912 1913
static SCliThrd* createThrdObj(void* trans) {
  STrans* pTransInst = trans;

dengyihao's avatar
dengyihao 已提交
1914
  SCliThrd* pThrd = (SCliThrd*)taosMemoryCalloc(1, sizeof(SCliThrd));
U
ubuntu 已提交
1915

dengyihao's avatar
dengyihao 已提交
1916
  QUEUE_INIT(&pThrd->msg);
wafwerar's avatar
wafwerar 已提交
1917
  taosThreadMutexInit(&pThrd->msgMtx, NULL);
dengyihao's avatar
dengyihao 已提交
1918

wafwerar's avatar
wafwerar 已提交
1919
  pThrd->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
1920 1921 1922 1923 1924 1925 1926 1927
  int err = uv_loop_init(pThrd->loop);
  if (err != 0) {
    tError("failed to init uv_loop, reason:%s", uv_err_name(err));
    taosMemoryFree(pThrd->loop);
    taosThreadMutexDestroy(&pThrd->msgMtx);
    taosMemoryFree(pThrd);
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
1928 1929 1930 1931 1932
  if (pTransInst->supportBatch) {
    pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 4, pThrd, cliAsyncCb);
  } else {
    pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 8, pThrd, cliAsyncCb);
  }
dengyihao's avatar
dengyihao 已提交
1933
  if (pThrd->asyncPool == NULL) {
dengyihao's avatar
ref log  
dengyihao 已提交
1934
    tError("failed to init async pool");
dengyihao's avatar
dengyihao 已提交
1935 1936 1937 1938 1939 1940
    uv_loop_close(pThrd->loop);
    taosMemoryFree(pThrd->loop);
    taosThreadMutexDestroy(&pThrd->msgMtx);
    taosMemoryFree(pThrd);
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
1941 1942 1943 1944

  pThrd->prepare = taosMemoryCalloc(1, sizeof(uv_prepare_t));
  uv_prepare_init(pThrd->loop, pThrd->prepare);
  pThrd->prepare->data = pThrd;
dengyihao's avatar
dengyihao 已提交
1945
  // uv_prepare_start(pThrd->prepare, cliPrepareCb);
dengyihao's avatar
dengyihao 已提交
1946

dengyihao's avatar
dengyihao 已提交
1947
  int32_t timerSize = 64;
dengyihao's avatar
dengyihao 已提交
1948 1949 1950 1951 1952 1953 1954
  pThrd->timerList = taosArrayInit(timerSize, sizeof(void*));
  for (int i = 0; i < timerSize; i++) {
    uv_timer_t* timer = taosMemoryCalloc(1, sizeof(uv_timer_t));
    uv_timer_init(pThrd->loop, timer);
    taosArrayPush(pThrd->timerList, &timer);
  }

dengyihao's avatar
dengyihao 已提交
1955
  pThrd->pool = createConnPool(4);
dengyihao's avatar
dengyihao 已提交
1956
  transDQCreate(pThrd->loop, &pThrd->delayQueue);
dengyihao's avatar
dengyihao 已提交
1957

dengyihao's avatar
dengyihao 已提交
1958 1959
  transDQCreate(pThrd->loop, &pThrd->timeoutQueue);

dengyihao's avatar
dengyihao 已提交
1960 1961
  transDQCreate(pThrd->loop, &pThrd->waitConnQueue);

dengyihao's avatar
dengyihao 已提交
1962 1963 1964
  pThrd->nextTimeout = taosGetTimestampMs() + CONN_PERSIST_TIME(pTransInst->idleTime);
  pThrd->pTransInst = trans;

dengyihao's avatar
dengyihao 已提交
1965
  pThrd->destroyAhandleFp = pTransInst->destroyFp;
dengyihao's avatar
dengyihao 已提交
1966
  pThrd->fqdn2ipCache = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
1967 1968
  pThrd->failFastCache = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);

dengyihao's avatar
dengyihao 已提交
1969
  pThrd->batchCache = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
1970

dengyihao's avatar
dengyihao 已提交
1971
  pThrd->quit = false;
dengyihao's avatar
dengyihao 已提交
1972 1973
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
1974
static void destroyThrdObj(SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1975 1976 1977
  if (pThrd == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1978

wafwerar's avatar
wafwerar 已提交
1979
  taosThreadJoin(pThrd->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1980
  CLI_RELEASE_UV(pThrd->loop);
wafwerar's avatar
wafwerar 已提交
1981
  taosThreadMutexDestroy(&pThrd->msgMtx);
dengyihao's avatar
dengyihao 已提交
1982
  TRANS_DESTROY_ASYNC_POOL_MSG(pThrd->asyncPool, SCliMsg, destroyCmsg);
dengyihao's avatar
dengyihao 已提交
1983
  transAsyncPoolDestroy(pThrd->asyncPool);
dengyihao's avatar
dengyihao 已提交
1984

dengyihao's avatar
dengyihao 已提交
1985
  transDQDestroy(pThrd->delayQueue, destroyCmsgAndAhandle);
dengyihao's avatar
dengyihao 已提交
1986
  transDQDestroy(pThrd->timeoutQueue, NULL);
dengyihao's avatar
dengyihao 已提交
1987
  transDQDestroy(pThrd->waitConnQueue, NULL);
dengyihao's avatar
dengyihao 已提交
1988

dengyihao's avatar
dengyihao 已提交
1989
  tDebug("thread destroy %" PRId64, pThrd->pid);
dengyihao's avatar
dengyihao 已提交
1990 1991 1992 1993 1994
  for (int i = 0; i < taosArrayGetSize(pThrd->timerList); i++) {
    uv_timer_t* timer = taosArrayGetP(pThrd->timerList, i);
    taosMemoryFree(timer);
  }
  taosArrayDestroy(pThrd->timerList);
dengyihao's avatar
dengyihao 已提交
1995
  taosMemoryFree(pThrd->prepare);
wafwerar's avatar
wafwerar 已提交
1996
  taosMemoryFree(pThrd->loop);
dengyihao's avatar
dengyihao 已提交
1997
  taosHashCleanup(pThrd->fqdn2ipCache);
dengyihao's avatar
dengyihao 已提交
1998
  taosHashCleanup(pThrd->failFastCache);
dengyihao's avatar
dengyihao 已提交
1999 2000 2001

  void** pIter = taosHashIterate(pThrd->batchCache, NULL);
  while (pIter != NULL) {
dengyihao's avatar
dengyihao 已提交
2002 2003 2004 2005 2006 2007 2008 2009
    SCliBatchList* pBatchList = (SCliBatchList*)(*pIter);
    while (!QUEUE_IS_EMPTY(&pBatchList->wq)) {
      queue* h = QUEUE_HEAD(&pBatchList->wq);
      QUEUE_REMOVE(h);

      SCliBatch* pBatch = QUEUE_DATA(h, SCliBatch, listq);
      cliDestroyBatch(pBatch);
    }
dengyihao's avatar
dengyihao 已提交
2010 2011
    taosMemoryFree(pBatchList->ip);
    taosMemoryFree(pBatchList->dst);
dengyihao's avatar
dengyihao 已提交
2012
    taosMemoryFree(pBatchList);
dengyihao's avatar
dengyihao 已提交
2013

dengyihao's avatar
dengyihao 已提交
2014 2015
    pIter = (void**)taosHashIterate(pThrd->batchCache, pIter);
  }
dengyihao's avatar
dengyihao 已提交
2016
  taosHashCleanup(pThrd->batchCache);
wafwerar's avatar
wafwerar 已提交
2017
  taosMemoryFree(pThrd);
dengyihao's avatar
dengyihao 已提交
2018
}
dengyihao's avatar
dengyihao 已提交
2019

dengyihao's avatar
dengyihao 已提交
2020
static FORCE_INLINE void transDestroyConnCtx(STransConnCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
2021
  //
wafwerar's avatar
wafwerar 已提交
2022
  taosMemoryFree(ctx);
dengyihao's avatar
dengyihao 已提交
2023
}
dengyihao's avatar
dengyihao 已提交
2024

dengyihao's avatar
dengyihao 已提交
2025
void cliSendQuit(SCliThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
2026
  // cli can stop gracefully
wafwerar's avatar
wafwerar 已提交
2027
  SCliMsg* msg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2028
  msg->type = Quit;
dengyihao's avatar
dengyihao 已提交
2029
  transAsyncSend(thrd->asyncPool, &msg->q);
dengyihao's avatar
dengyihao 已提交
2030
  atomic_store_8(&thrd->asyncPool->stop, 1);
dengyihao's avatar
dengyihao 已提交
2031
}
dengyihao's avatar
dengyihao 已提交
2032 2033
void cliWalkCb(uv_handle_t* handle, void* arg) {
  if (!uv_is_closing(handle)) {
dengyihao's avatar
dengyihao 已提交
2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045
    if (uv_handle_get_type(handle) == UV_TIMER) {
      // SCliConn* pConn = handle->data;
      //  if (pConn != NULL && pConn->timer != NULL) {
      //    SCliThrd* pThrd = pConn->hostThrd;
      //    uv_timer_stop((uv_timer_t*)handle);
      //    handle->data = NULL;
      //    taosArrayPush(pThrd->timerList, &pConn->timer);
      //    pConn->timer = NULL;
      //  }
    } else {
      uv_read_stop((uv_stream_t*)handle);
    }
dengyihao's avatar
dengyihao 已提交
2046
    uv_close(handle, cliDestroy);
dengyihao's avatar
dengyihao 已提交
2047 2048
  }
}
dengyihao's avatar
dengyihao 已提交
2049

dengyihao's avatar
dengyihao 已提交
2050
FORCE_INLINE int cliRBChoseIdx(STrans* pTransInst) {
dengyihao's avatar
dengyihao 已提交
2051
  int32_t index = pTransInst->index;
dengyihao's avatar
dengyihao 已提交
2052 2053 2054
  if (pTransInst->numOfThreads == 0) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2055 2056 2057 2058
  /*
   * no lock, and to avoid CPU load imbalance, set limit pTransInst->numOfThreads * 2000;
   */
  if (pTransInst->index++ >= pTransInst->numOfThreads * 2000) {
U
ubuntu 已提交
2059 2060 2061 2062
    pTransInst->index = 0;
  }
  return index % pTransInst->numOfThreads;
}
dengyihao's avatar
dengyihao 已提交
2063
static FORCE_INLINE void doDelayTask(void* param) {
dengyihao's avatar
dengyihao 已提交
2064
  STaskArg* arg = param;
dengyihao's avatar
dengyihao 已提交
2065
  cliHandleReq((SCliMsg*)arg->param1, (SCliThrd*)arg->param2);
dengyihao's avatar
dengyihao 已提交
2066 2067
  taosMemoryFree(arg);
}
dengyihao's avatar
dengyihao 已提交
2068

dengyihao's avatar
dengyihao 已提交
2069 2070 2071
static void doCloseIdleConn(void* param) {
  STaskArg* arg = param;
  SCliConn* conn = arg->param1;
dengyihao's avatar
dengyihao 已提交
2072
  tDebug("%s conn %p idle, close it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
2073
  conn->task = NULL;
dengyihao's avatar
dengyihao 已提交
2074 2075
  cliDestroyConn(conn, true);
  taosMemoryFree(arg);
dengyihao's avatar
dengyihao 已提交
2076 2077
}

dengyihao's avatar
dengyihao 已提交
2078
static void cliSchedMsgToNextNode(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
2079
  STrans*        pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2080 2081
  STransConnCtx* pCtx = pMsg->ctx;

dengyihao's avatar
dengyihao 已提交
2082 2083 2084 2085 2086 2087 2088
  if (rpcDebugFlag & DEBUG_DEBUG) {
    STraceId* trace = &pMsg->msg.info.traceId;
    char      tbuf[256] = {0};
    EPSET_DEBUG_STR(&pCtx->epSet, tbuf);
    tGDebug("%s retry on next node,use:%s, step: %d,timeout:%" PRId64 "", transLabel(pThrd->pTransInst), tbuf,
            pCtx->retryStep, pCtx->retryNextInterval);
  }
dengyihao's avatar
dengyihao 已提交
2089 2090 2091 2092

  STaskArg* arg = taosMemoryMalloc(sizeof(STaskArg));
  arg->param1 = pMsg;
  arg->param2 = pThrd;
dengyihao's avatar
dengyihao 已提交
2093 2094

  transDQSched(pThrd->delayQueue, doDelayTask, arg, pCtx->retryNextInterval);
dengyihao's avatar
dengyihao 已提交
2095
}
dengyihao's avatar
dengyihao 已提交
2096

dengyihao's avatar
dengyihao 已提交
2097
FORCE_INLINE void cliCompareAndSwap(int8_t* val, int8_t exp, int8_t newVal) {
dengyihao's avatar
dengyihao 已提交
2098 2099 2100 2101
  if (*val != exp) {
    *val = newVal;
  }
}
dengyihao's avatar
dengyihao 已提交
2102

dengyihao's avatar
dengyihao 已提交
2103
FORCE_INLINE bool cliTryExtractEpSet(STransMsg* pResp, SEpSet* dst) {
dengyihao's avatar
dengyihao 已提交
2104 2105 2106 2107 2108 2109
  if ((pResp == NULL || pResp->info.hasEpSet == 0)) {
    return false;
  }
  // rebuild resp msg
  SEpSet epset;
  if (tDeserializeSEpSet(pResp->pCont, pResp->contLen, &epset) < 0) {
dengyihao's avatar
dengyihao 已提交
2110 2111 2112 2113
    return false;
  }
  int32_t tlen = tSerializeSEpSet(NULL, 0, dst);

dengyihao's avatar
dengyihao 已提交
2114 2115 2116 2117 2118 2119 2120
  char*   buf = NULL;
  int32_t len = pResp->contLen - tlen;
  if (len != 0) {
    buf = rpcMallocCont(len);
    memcpy(buf, (char*)pResp->pCont + tlen, len);
  }
  rpcFreeCont(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2121 2122

  pResp->pCont = buf;
dengyihao's avatar
dengyihao 已提交
2123 2124
  pResp->contLen = len;

dengyihao's avatar
dengyihao 已提交
2125
  epsetAssign(dst, &epset);
dengyihao's avatar
dengyihao 已提交
2126 2127
  return true;
}
dengyihao's avatar
dengyihao 已提交
2128
bool cliResetEpset(STransConnCtx* pCtx, STransMsg* pResp, bool hasEpSet) {
dengyihao's avatar
dengyihao 已提交
2129
  bool noDelay = true;
dengyihao's avatar
dengyihao 已提交
2130 2131 2132 2133 2134 2135 2136
  if (hasEpSet == false) {
    if (pResp->contLen == 0) {
      if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
        noDelay = false;
      } else {
        EPSET_FORWARD_INUSE(&pCtx->epSet);
      }
dengyihao's avatar
dengyihao 已提交
2137 2138 2139 2140 2141 2142 2143 2144 2145 2146
    } else if (pResp->contLen != 0) {
      SEpSet  epSet;
      int32_t valid = tDeserializeSEpSet(pResp->pCont, pResp->contLen, &epSet);
      if (valid < 0) {
        tDebug("get invalid epset, epset equal, continue");
        if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
          noDelay = false;
        } else {
          EPSET_FORWARD_INUSE(&pCtx->epSet);
        }
dengyihao's avatar
dengyihao 已提交
2147
      } else {
dengyihao's avatar
dengyihao 已提交
2148 2149
        if (!transEpSetIsEqual(&pCtx->epSet, &epSet)) {
          tDebug("epset not equal, retry new epset");
dengyihao's avatar
dengyihao 已提交
2150
          epsetAssign(&pCtx->epSet, &epSet);
dengyihao's avatar
dengyihao 已提交
2151 2152 2153 2154 2155 2156 2157 2158 2159
          noDelay = false;
        } else {
          if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
            noDelay = false;
          } else {
            tDebug("epset equal, continue");
            EPSET_FORWARD_INUSE(&pCtx->epSet);
          }
        }
dengyihao's avatar
dengyihao 已提交
2160
      }
dengyihao's avatar
dengyihao 已提交
2161 2162
    }
  } else {
dengyihao's avatar
dengyihao 已提交
2163
    SEpSet  epSet;
dengyihao's avatar
dengyihao 已提交
2164 2165
    int32_t valid = tDeserializeSEpSet(pResp->pCont, pResp->contLen, &epSet);
    if (valid < 0) {
dengyihao's avatar
dengyihao 已提交
2166 2167 2168 2169 2170 2171
      tDebug("get invalid epset, epset equal, continue");
      if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
        noDelay = false;
      } else {
        EPSET_FORWARD_INUSE(&pCtx->epSet);
      }
dengyihao's avatar
dengyihao 已提交
2172
    } else {
dengyihao's avatar
dengyihao 已提交
2173 2174
      if (!transEpSetIsEqual(&pCtx->epSet, &epSet)) {
        tDebug("epset not equal, retry new epset");
dengyihao's avatar
dengyihao 已提交
2175
        epsetAssign(&pCtx->epSet, &epSet);
dengyihao's avatar
dengyihao 已提交
2176
        noDelay = false;
dengyihao's avatar
dengyihao 已提交
2177
      } else {
dengyihao's avatar
dengyihao 已提交
2178 2179 2180 2181 2182 2183
        if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
          noDelay = false;
        } else {
          tDebug("epset equal, continue");
          EPSET_FORWARD_INUSE(&pCtx->epSet);
        }
dengyihao's avatar
dengyihao 已提交
2184
      }
dengyihao's avatar
dengyihao 已提交
2185 2186 2187 2188 2189
    }
  }
  return noDelay;
}
bool cliGenRetryRule(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
2190 2191
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2192

dengyihao's avatar
dengyihao 已提交
2193 2194
  STransConnCtx* pCtx = pMsg->ctx;
  int32_t        code = pResp->code;
dengyihao's avatar
dengyihao 已提交
2195

dengyihao's avatar
dengyihao 已提交
2196
  bool retry = pTransInst->retry != NULL ? pTransInst->retry(code, pResp->msgType - 1) : false;
dengyihao's avatar
dengyihao 已提交
2197 2198 2199
  if (retry == false) {
    return false;
  }
dengyihao's avatar
dengyihao 已提交
2200 2201 2202 2203 2204 2205 2206 2207

  if (!pCtx->retryInit) {
    pCtx->retryMinInterval = pTransInst->retryMinInterval;
    pCtx->retryMaxInterval = pTransInst->retryMaxInterval;
    pCtx->retryStepFactor = pTransInst->retryStepFactor;
    pCtx->retryMaxTimeout = pTransInst->retryMaxTimouet;
    pCtx->retryInitTimestamp = taosGetTimestampMs();
    pCtx->retryNextInterval = pCtx->retryMinInterval;
dengyihao's avatar
dengyihao 已提交
2208
    pCtx->retryStep = 0;
dengyihao's avatar
dengyihao 已提交
2209
    pCtx->retryInit = true;
dengyihao's avatar
dengyihao 已提交
2210
    pCtx->retryCode = TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
2211 2212 2213

    // already retry, not use handle specified by app;
    pMsg->msg.info.handle = 0;
dengyihao's avatar
dengyihao 已提交
2214
  }
dengyihao's avatar
dengyihao 已提交
2215

dengyihao's avatar
dengyihao 已提交
2216 2217
  if (-1 != pCtx->retryMaxTimeout && taosGetTimestampMs() - pCtx->retryInitTimestamp >= pCtx->retryMaxTimeout) {
    return false;
dengyihao's avatar
dengyihao 已提交
2218
  }
dengyihao's avatar
dengyihao 已提交
2219

dengyihao's avatar
dengyihao 已提交
2220 2221 2222 2223 2224 2225
  // code, msgType

  // A:  epset,   leader, not self
  // B:  epset,   not know leader
  // C:  no epset, leader but not serivce

dengyihao's avatar
dengyihao 已提交
2226 2227
  bool noDelay = false;
  if (code == TSDB_CODE_RPC_BROKEN_LINK || code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
dengyihao's avatar
dengyihao 已提交
2228
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2229
    noDelay = cliResetEpset(pCtx, pResp, false);
dengyihao's avatar
dengyihao 已提交
2230 2231
    transFreeMsg(pResp->pCont);
    transUnrefCliHandle(pConn);
dengyihao's avatar
dengyihao 已提交
2232
  } else if (code == TSDB_CODE_SYN_NOT_LEADER || code == TSDB_CODE_SYN_INTERNAL_ERROR ||
2233 2234
             code == TSDB_CODE_SYN_PROPOSE_NOT_READY || code == TSDB_CODE_VND_STOPPED ||
             code == TSDB_CODE_MNODE_NOT_FOUND || code == TSDB_CODE_APP_IS_STARTING ||
dengyihao's avatar
dengyihao 已提交
2235
             code == TSDB_CODE_APP_IS_STOPPING || code == TSDB_CODE_VND_STOPPED) {
dengyihao's avatar
dengyihao 已提交
2236
    tTrace("code str %s, contlen:%d 1", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2237
    noDelay = cliResetEpset(pCtx, pResp, true);
dengyihao's avatar
dengyihao 已提交
2238 2239
    transFreeMsg(pResp->pCont);
    addConnToPool(pThrd->pool, pConn);
dengyihao's avatar
dengyihao 已提交
2240
  } else if (code == TSDB_CODE_SYN_RESTORING) {
dengyihao's avatar
dengyihao 已提交
2241
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2242
    noDelay = cliResetEpset(pCtx, pResp, true);
dengyihao's avatar
dengyihao 已提交
2243 2244
    addConnToPool(pThrd->pool, pConn);
    transFreeMsg(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2245
  } else {
dengyihao's avatar
dengyihao 已提交
2246
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2247 2248 2249
    noDelay = cliResetEpset(pCtx, pResp, false);
    addConnToPool(pThrd->pool, pConn);
    transFreeMsg(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2250
  }
dengyihao's avatar
dengyihao 已提交
2251 2252 2253 2254
  if (code != TSDB_CODE_RPC_BROKEN_LINK && code != TSDB_CODE_RPC_NETWORK_UNAVAIL && code != TSDB_CODE_SUCCESS) {
    // save one internal code
    pCtx->retryCode = code;
  }
dengyihao's avatar
dengyihao 已提交
2255

dengyihao's avatar
dengyihao 已提交
2256 2257 2258
  if (noDelay == false) {
    pCtx->epsetRetryCnt = 1;
    pCtx->retryStep++;
dengyihao's avatar
dengyihao 已提交
2259

dengyihao's avatar
dengyihao 已提交
2260 2261 2262 2263 2264 2265 2266 2267
    int64_t factor = pow(pCtx->retryStepFactor, pCtx->retryStep - 1);
    pCtx->retryNextInterval = factor * pCtx->retryMinInterval;
    if (pCtx->retryNextInterval >= pCtx->retryMaxInterval) {
      pCtx->retryNextInterval = pCtx->retryMaxInterval;
    }
  } else {
    pCtx->retryNextInterval = 0;
    pCtx->epsetRetryCnt++;
dengyihao's avatar
dengyihao 已提交
2268
  }
dengyihao's avatar
dengyihao 已提交
2269

dengyihao's avatar
dengyihao 已提交
2270
  pMsg->sent = 0;
dengyihao's avatar
dengyihao 已提交
2271
  cliSchedMsgToNextNode(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
2272
  return true;
dengyihao's avatar
dengyihao 已提交
2273
}
dengyihao's avatar
dengyihao 已提交
2274
int cliAppCb(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
2275 2276
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2277

dengyihao's avatar
dengyihao 已提交
2278
  if (pMsg == NULL || pMsg->ctx == NULL) {
dengyihao's avatar
dengyihao 已提交
2279
    tTrace("%s conn %p handle resp", pTransInst->label, pConn);
dengyihao's avatar
dengyihao 已提交
2280 2281 2282
    pTransInst->cfp(pTransInst->parent, pResp, NULL);
    return 0;
  }
dengyihao's avatar
dengyihao 已提交
2283

dengyihao's avatar
dengyihao 已提交
2284
  STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
dengyihao 已提交
2285

dengyihao's avatar
dengyihao 已提交
2286 2287 2288
  bool retry = cliGenRetryRule(pConn, pResp, pMsg);
  if (retry == true) {
    return -1;
dengyihao's avatar
dengyihao 已提交
2289
  }
dengyihao's avatar
dengyihao 已提交
2290

dengyihao's avatar
dengyihao 已提交
2291 2292 2293
  if (pCtx->retryCode != TSDB_CODE_SUCCESS) {
    int32_t code = pResp->code;
    // return internal code app
dengyihao's avatar
dengyihao 已提交
2294 2295
    if (code == TSDB_CODE_RPC_NETWORK_UNAVAIL || code == TSDB_CODE_RPC_BROKEN_LINK ||
        code == TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED) {
dengyihao's avatar
dengyihao 已提交
2296 2297 2298 2299
      pResp->code = pCtx->retryCode;
    }
  }

2300
  // check whole vnodes is offline on this vgroup
2301 2302
  if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps || pCtx->retryStep > 0) {
    if (pResp->code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
A
Alex Duan 已提交
2303
      pResp->code = TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED;
2304
    } else if (pResp->code == TSDB_CODE_RPC_BROKEN_LINK) {
A
Alex Duan 已提交
2305
      pResp->code = TSDB_CODE_RPC_SOMENODE_BROKEN_LINK;
2306 2307 2308
    }
  }

dengyihao's avatar
dengyihao 已提交
2309 2310
  STraceId* trace = &pResp->info.traceId;
  bool      hasEpSet = cliTryExtractEpSet(pResp, &pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
2311
  if (hasEpSet) {
dengyihao's avatar
dengyihao 已提交
2312 2313 2314 2315 2316
    if (rpcDebugFlag & DEBUG_TRACE) {
      char tbuf[256] = {0};
      EPSET_DEBUG_STR(&pCtx->epSet, tbuf);
      tGTrace("%s conn %p extract epset from msg", CONN_GET_INST_LABEL(pConn), pConn);
    }
dengyihao's avatar
dengyihao 已提交
2317
  }
dengyihao's avatar
dengyihao 已提交
2318

dengyihao's avatar
dengyihao 已提交
2319
  if (pCtx->pSem != NULL) {
dengyihao's avatar
dengyihao 已提交
2320
    tGTrace("%s conn %p(sync) handle resp", CONN_GET_INST_LABEL(pConn), pConn);
2321
    if (pCtx->pRsp == NULL) {
dengyihao's avatar
dengyihao 已提交
2322
      tGTrace("%s conn %p(sync) failed to resp, ignore", CONN_GET_INST_LABEL(pConn), pConn);
2323 2324 2325
    } else {
      memcpy((char*)pCtx->pRsp, (char*)pResp, sizeof(*pResp));
    }
dengyihao's avatar
dengyihao 已提交
2326
    tsem_post(pCtx->pSem);
2327
    pCtx->pRsp = NULL;
dengyihao's avatar
dengyihao 已提交
2328
  } else {
dengyihao's avatar
dengyihao 已提交
2329
    tGTrace("%s conn %p handle resp", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
2330
    if (retry == false && hasEpSet == true) {
dengyihao's avatar
dengyihao 已提交
2331
      pTransInst->cfp(pTransInst->parent, pResp, &pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
2332
    } else {
dengyihao's avatar
dengyihao 已提交
2333
      if (!cliIsEpsetUpdated(pResp->code, pCtx)) {
dengyihao's avatar
dengyihao 已提交
2334 2335 2336 2337
        pTransInst->cfp(pTransInst->parent, pResp, NULL);
      } else {
        pTransInst->cfp(pTransInst->parent, pResp, &pCtx->epSet);
      }
dengyihao's avatar
dengyihao 已提交
2338
    }
dengyihao's avatar
dengyihao 已提交
2339
  }
dengyihao's avatar
dengyihao 已提交
2340
  return 0;
dengyihao's avatar
dengyihao 已提交
2341
}
U
ubuntu 已提交
2342 2343

void transCloseClient(void* arg) {
U
ubuntu 已提交
2344
  SCliObj* cli = arg;
dengyihao's avatar
dengyihao 已提交
2345
  for (int i = 0; i < cli->numOfThreads; i++) {
U
ubuntu 已提交
2346
    cliSendQuit(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
2347
    destroyThrdObj(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
2348
  }
wafwerar's avatar
wafwerar 已提交
2349 2350
  taosMemoryFree(cli->pThreadObj);
  taosMemoryFree(cli);
dengyihao's avatar
dengyihao 已提交
2351
}
dengyihao's avatar
dengyihao 已提交
2352 2353 2354 2355 2356
void transRefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_INC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
2357
  tTrace("%s conn %p ref %d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
2358 2359 2360 2361 2362 2363 2364
  UNUSED(ref);
}
void transUnrefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_DEC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
2365
  tTrace("%s conn %p ref:%d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
2366
  if (ref == 0) {
U
ubuntu 已提交
2367
    cliDestroyConn((SCliConn*)handle, true);
dengyihao's avatar
dengyihao 已提交
2368 2369
  }
}
dengyihao's avatar
dengyihao 已提交
2370
static FORCE_INLINE SCliThrd* transGetWorkThrdFromHandle(STrans* trans, int64_t handle) {
dengyihao's avatar
dengyihao 已提交
2371
  SCliThrd*  pThrd = NULL;
dengyihao's avatar
dengyihao 已提交
2372
  SExHandle* exh = transAcquireExHandle(transGetRefMgt(), handle);
dengyihao's avatar
dengyihao 已提交
2373 2374 2375
  if (exh == NULL) {
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
2376

dengyihao's avatar
dengyihao 已提交
2377 2378 2379 2380 2381 2382
  if (exh->pThrd == NULL && trans != NULL) {
    int idx = cliRBChoseIdx(trans);
    if (idx < 0) return NULL;
    exh->pThrd = ((SCliObj*)trans->tcphandle)->pThreadObj[idx];
  }

dengyihao's avatar
dengyihao 已提交
2383
  pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
2384
  transReleaseExHandle(transGetRefMgt(), handle);
dengyihao's avatar
dengyihao 已提交
2385 2386
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
2387
SCliThrd* transGetWorkThrd(STrans* trans, int64_t handle) {
dengyihao's avatar
dengyihao 已提交
2388
  if (handle == 0) {
dengyihao's avatar
dengyihao 已提交
2389
    int idx = cliRBChoseIdx(trans);
dengyihao's avatar
dengyihao 已提交
2390
    if (idx < 0) return NULL;
dengyihao's avatar
dengyihao 已提交
2391 2392
    return ((SCliObj*)trans->tcphandle)->pThreadObj[idx];
  }
dengyihao's avatar
dengyihao 已提交
2393
  SCliThrd* pThrd = transGetWorkThrdFromHandle(trans, handle);
D
dapan1121 已提交
2394
  return pThrd;
dengyihao's avatar
dengyihao 已提交
2395
}
dengyihao's avatar
dengyihao 已提交
2396
int transReleaseCliHandle(void* handle) {
dengyihao's avatar
dengyihao 已提交
2397 2398 2399
  int  idx = -1;
  bool valid = false;

dengyihao's avatar
dengyihao 已提交
2400
  SCliThrd* pThrd = transGetWorkThrdFromHandle(NULL, (int64_t)handle);
dengyihao's avatar
dengyihao 已提交
2401
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2402
    return -1;
dengyihao's avatar
dengyihao 已提交
2403
  }
dengyihao's avatar
dengyihao 已提交
2404

dengyihao's avatar
dengyihao 已提交
2405
  STransMsg tmsg = {.info.handle = handle, .info.ahandle = (void*)0x9527};
dengyihao's avatar
rm code  
dengyihao 已提交
2406
  TRACE_SET_MSGID(&tmsg.info.traceId, tGenIdPI64());
dengyihao's avatar
dengyihao 已提交
2407

dengyihao's avatar
dengyihao 已提交
2408 2409 2410
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
  pCtx->ahandle = tmsg.info.ahandle;

dengyihao's avatar
dengyihao 已提交
2411
  SCliMsg* cmsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2412
  cmsg->msg = tmsg;
dengyihao's avatar
dengyihao 已提交
2413
  cmsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2414
  cmsg->type = Release;
dengyihao's avatar
dengyihao 已提交
2415
  cmsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
2416

dengyihao's avatar
dengyihao 已提交
2417 2418 2419
  STraceId* trace = &tmsg.info.traceId;
  tGDebug("send release request at thread:%08" PRId64 "", pThrd->pid);

dengyihao's avatar
dengyihao 已提交
2420
  if (0 != transAsyncSend(pThrd->asyncPool, &cmsg->q)) {
dengyihao's avatar
dengyihao 已提交
2421
    destroyCmsg(cmsg);
dengyihao's avatar
dengyihao 已提交
2422 2423
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2424
  return 0;
dengyihao's avatar
dengyihao 已提交
2425
}
dengyihao's avatar
dengyihao 已提交
2426

dengyihao's avatar
dengyihao 已提交
2427
int transSendRequest(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
2428 2429 2430 2431 2432
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
  if (pTransInst == NULL) {
    transFreeMsg(pReq->pCont);
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2433

dengyihao's avatar
dengyihao 已提交
2434
  SCliThrd* pThrd = transGetWorkThrd(pTransInst, (int64_t)pReq->info.handle);
dengyihao's avatar
dengyihao 已提交
2435
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2436
    transFreeMsg(pReq->pCont);
dengyihao's avatar
dengyihao 已提交
2437
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2438
    return TSDB_CODE_RPC_BROKEN_LINK;
dengyihao's avatar
dengyihao 已提交
2439 2440
  }

dengyihao's avatar
dengyihao 已提交
2441
  TRACE_SET_MSGID(&pReq->info.traceId, tGenIdPI64());
wafwerar's avatar
wafwerar 已提交
2442
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2443 2444 2445
  epsetAssign(&pCtx->epSet, pEpSet);
  epsetAssign(&pCtx->origEpSet, pEpSet);

S
Shengliang Guan 已提交
2446
  pCtx->ahandle = pReq->info.ahandle;
dengyihao's avatar
dengyihao 已提交
2447
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
2448

H
Haojun Liao 已提交
2449
  if (ctx != NULL) pCtx->appCtx = *ctx;
dengyihao's avatar
dengyihao 已提交
2450

wafwerar's avatar
wafwerar 已提交
2451
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2452
  cliMsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
2453
  cliMsg->msg = *pReq;
dengyihao's avatar
dengyihao 已提交
2454
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2455
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
2456
  cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2457

dengyihao's avatar
dengyihao 已提交
2458
  STraceId* trace = &pReq->info.traceId;
dengyihao's avatar
dengyihao 已提交
2459 2460
  tGDebug("%s send request at thread:%08" PRId64 ", dst:%s:%d, app:%p", transLabel(pTransInst), pThrd->pid,
          EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet), pReq->info.ahandle);
dengyihao's avatar
dengyihao 已提交
2461 2462
  if (0 != transAsyncSend(pThrd->asyncPool, &(cliMsg->q))) {
    destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2463
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2464 2465
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2466
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2467
  return 0;
dengyihao's avatar
dengyihao 已提交
2468
}
dengyihao's avatar
dengyihao 已提交
2469

dengyihao's avatar
dengyihao 已提交
2470
int transSendRecv(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransMsg* pRsp) {
dengyihao's avatar
dengyihao 已提交
2471 2472 2473 2474 2475
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
  if (pTransInst == NULL) {
    transFreeMsg(pReq->pCont);
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2476

dengyihao's avatar
dengyihao 已提交
2477 2478
  SCliThrd* pThrd = transGetWorkThrd(pTransInst, (int64_t)pReq->info.handle);
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2479
    transFreeMsg(pReq->pCont);
dengyihao's avatar
dengyihao 已提交
2480
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2481
    return TSDB_CODE_RPC_BROKEN_LINK;
dengyihao's avatar
dengyihao 已提交
2482
  }
dengyihao's avatar
dengyihao 已提交
2483

dengyihao's avatar
dengyihao 已提交
2484 2485
  tsem_t* sem = taosMemoryCalloc(1, sizeof(tsem_t));
  tsem_init(sem, 0, 0);
dengyihao's avatar
dengyihao 已提交
2486

dengyihao's avatar
dengyihao 已提交
2487 2488
  TRACE_SET_MSGID(&pReq->info.traceId, tGenIdPI64());

wafwerar's avatar
wafwerar 已提交
2489
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2490 2491
  epsetAssign(&pCtx->epSet, pEpSet);
  epsetAssign(&pCtx->origEpSet, pEpSet);
S
Shengliang Guan 已提交
2492
  pCtx->ahandle = pReq->info.ahandle;
dengyihao's avatar
dengyihao 已提交
2493
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
2494
  pCtx->pSem = sem;
dengyihao's avatar
dengyihao 已提交
2495 2496
  pCtx->pRsp = pRsp;

wafwerar's avatar
wafwerar 已提交
2497
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2498 2499 2500
  cliMsg->ctx = pCtx;
  cliMsg->msg = *pReq;
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2501
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
2502
  cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2503

dengyihao's avatar
dengyihao 已提交
2504
  STraceId* trace = &pReq->info.traceId;
dengyihao's avatar
dengyihao 已提交
2505 2506
  tGDebug("%s send request at thread:%08" PRId64 ", dst:%s:%d, app:%p", transLabel(pTransInst), pThrd->pid,
          EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet), pReq->info.ahandle);
dengyihao's avatar
dengyihao 已提交
2507

dengyihao's avatar
dengyihao 已提交
2508 2509
  int ret = transAsyncSend(pThrd->asyncPool, &cliMsg->q);
  if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
2510
    destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2511
    goto _RETURN;
dengyihao's avatar
dengyihao 已提交
2512
  }
dengyihao's avatar
dengyihao 已提交
2513
  tsem_wait(sem);
dengyihao's avatar
dengyihao 已提交
2514 2515

_RETURN:
dengyihao's avatar
dengyihao 已提交
2516 2517
  tsem_destroy(sem);
  taosMemoryFree(sem);
dengyihao's avatar
dengyihao 已提交
2518
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2519
  return ret;
dengyihao's avatar
dengyihao 已提交
2520
}
dengyihao's avatar
dengyihao 已提交
2521 2522 2523
/*
 *
 **/
dengyihao's avatar
dengyihao 已提交
2524
int transSetDefaultAddr(void* shandle, const char* ip, const char* fqdn) {
dengyihao's avatar
dengyihao 已提交
2525
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2526 2527 2528
  if (pTransInst == NULL) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2529 2530 2531

  SCvtAddr cvtAddr = {0};
  if (ip != NULL && fqdn != NULL) {
dengyihao's avatar
dengyihao 已提交
2532 2533
    tstrncpy(cvtAddr.ip, ip, sizeof(cvtAddr.ip));
    tstrncpy(cvtAddr.fqdn, fqdn, sizeof(cvtAddr.fqdn));
dengyihao's avatar
dengyihao 已提交
2534 2535
    cvtAddr.cvt = true;
  }
dengyihao's avatar
dengyihao 已提交
2536 2537
  for (int i = 0; i < pTransInst->numOfThreads; i++) {
    STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2538
    pCtx->cvtAddr = cvtAddr;
dengyihao's avatar
dengyihao 已提交
2539 2540 2541 2542

    SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
    cliMsg->ctx = pCtx;
    cliMsg->type = Update;
dengyihao's avatar
dengyihao 已提交
2543
    cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2544

dengyihao's avatar
dengyihao 已提交
2545
    SCliThrd* thrd = ((SCliObj*)pTransInst->tcphandle)->pThreadObj[i];
S
Shengliang Guan 已提交
2546
    tDebug("%s update epset at thread:%08" PRId64, pTransInst->label, thrd->pid);
dengyihao's avatar
dengyihao 已提交
2547

dengyihao's avatar
dengyihao 已提交
2548 2549
    if (transAsyncSend(thrd->asyncPool, &(cliMsg->q)) != 0) {
      destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2550
      transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2551 2552
      return -1;
    }
dengyihao's avatar
dengyihao 已提交
2553
  }
dengyihao's avatar
dengyihao 已提交
2554
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2555
  return 0;
dengyihao's avatar
dengyihao 已提交
2556
}
dengyihao's avatar
dengyihao 已提交
2557 2558 2559 2560 2561

int64_t transAllocHandle() {
  SExHandle* exh = taosMemoryCalloc(1, sizeof(SExHandle));
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
  tDebug("pre alloc refId %" PRId64 "", exh->refId);
dengyihao's avatar
dengyihao 已提交
2562

dengyihao's avatar
dengyihao 已提交
2563 2564
  return exh->refId;
}