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

#ifdef USE_UV
#include "transComm.h"

dengyihao's avatar
dengyihao 已提交
18
typedef struct SConnList {
dengyihao's avatar
dengyihao 已提交
19
  queue   conns;
dengyihao's avatar
dengyihao 已提交
20
  int32_t size;
dengyihao's avatar
dengyihao 已提交
21 22
} SConnList;

dengyihao's avatar
dengyihao 已提交
23
typedef struct {
dengyihao's avatar
dengyihao 已提交
24 25 26 27 28 29 30
  queue   wq;
  int32_t len;

  int connMax;
  int connCnt;
  int batchLenLimit;

dengyihao's avatar
dengyihao 已提交
31 32 33
  char*    dst;
  char*    ip;
  uint16_t port;
dengyihao's avatar
dengyihao 已提交
34 35 36 37 38 39 40 41 42 43

} SCliBatchList;

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

dengyihao's avatar
dengyihao 已提交
46
typedef struct SCliConn {
dengyihao's avatar
dengyihao 已提交
47
  T_REF_DECLARE()
dengyihao's avatar
dengyihao 已提交
48 49
  uv_connect_t connReq;
  uv_stream_t* stream;
dengyihao's avatar
dengyihao 已提交
50
  queue        wreqQueue;
dengyihao's avatar
dengyihao 已提交
51 52

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

dengyihao's avatar
dengyihao 已提交
54 55 56 57
  void* hostThrd;

  SConnBuffer readBuf;
  STransQueue cliMsgs;
dengyihao's avatar
dengyihao 已提交
58 59 60

  queue      q;
  SConnList* list;
dengyihao's avatar
dengyihao 已提交
61 62

  STransCtx  ctx;
dengyihao's avatar
dengyihao 已提交
63 64
  bool       broken;  // link broken or not
  ConnStatus status;  //
dengyihao's avatar
dengyihao 已提交
65

dengyihao's avatar
dengyihao 已提交
66 67
  SCliBatch* pBatch;

dengyihao's avatar
dengyihao 已提交
68 69
  int64_t refId;
  char*   ip;
dengyihao's avatar
dengyihao 已提交
70

dengyihao's avatar
dengyihao 已提交
71
  SDelayTask* task;
dengyihao's avatar
dengyihao 已提交
72

dengyihao's avatar
dengyihao 已提交
73
  // debug and log info
dengyihao's avatar
dengyihao 已提交
74 75 76
  char src[32];
  char dst[32];

dengyihao's avatar
dengyihao 已提交
77
} SCliConn;
dengyihao's avatar
dengyihao 已提交
78

dengyihao's avatar
dengyihao 已提交
79
typedef struct SCliMsg {
dengyihao's avatar
dengyihao 已提交
80
  STransConnCtx* ctx;
dengyihao's avatar
formate  
dengyihao 已提交
81
  STransMsg      msg;
dengyihao's avatar
dengyihao 已提交
82
  queue          q;
dengyihao's avatar
dengyihao 已提交
83
  STransMsgType  type;
dengyihao's avatar
dengyihao 已提交
84

dengyihao's avatar
dengyihao 已提交
85
  int64_t  refId;
dengyihao's avatar
dengyihao 已提交
86 87
  uint64_t st;
  int      sent;  //(0: no send, 1: alread sent)
dengyihao's avatar
dengyihao 已提交
88 89
} SCliMsg;

dengyihao's avatar
dengyihao 已提交
90
typedef struct SCliThrd {
dengyihao's avatar
dengyihao 已提交
91 92 93 94 95 96
  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 已提交
97
  // timer handles
dengyihao's avatar
dengyihao 已提交
98
  SArray* timerList;
dengyihao's avatar
dengyihao 已提交
99
  // msg queue
dengyihao's avatar
dengyihao 已提交
100
  queue         msg;
wafwerar's avatar
wafwerar 已提交
101
  TdThreadMutex msgMtx;
dengyihao's avatar
dengyihao 已提交
102
  SDelayQueue*  delayQueue;
dengyihao's avatar
dengyihao 已提交
103
  SDelayQueue*  timeoutQueue;
dengyihao's avatar
dengyihao 已提交
104 105
  uint64_t      nextTimeout;  // next timeout
  void*         pTransInst;   //
dengyihao's avatar
dengyihao 已提交
106

dengyihao's avatar
dengyihao 已提交
107
  int connCount;
dengyihao's avatar
dengyihao 已提交
108
  void (*destroyAhandleFp)(void* ahandle);
dengyihao's avatar
dengyihao 已提交
109 110
  SHashObj* fqdn2ipCache;
  SCvtAddr  cvtAddr;
dengyihao's avatar
dengyihao 已提交
111

dengyihao's avatar
dengyihao 已提交
112
  SHashObj* failFastCache;
dengyihao's avatar
dengyihao 已提交
113
  SHashObj* connLimitCache;
dengyihao's avatar
dengyihao 已提交
114
  SHashObj* batchCache;
dengyihao's avatar
dengyihao 已提交
115

dengyihao's avatar
dengyihao 已提交
116 117
  SCliMsg* stopMsg;

dengyihao's avatar
dengyihao 已提交
118
  bool quit;
dengyihao's avatar
dengyihao 已提交
119
} SCliThrd;
dengyihao's avatar
dengyihao 已提交
120

U
ubuntu 已提交
121
typedef struct SCliObj {
dengyihao's avatar
dengyihao 已提交
122 123 124 125
  char       label[TSDB_LABEL_LEN];
  int32_t    index;
  int        numOfThreads;
  SCliThrd** pThreadObj;
U
ubuntu 已提交
126
} SCliObj;
dengyihao's avatar
dengyihao 已提交
127

dengyihao's avatar
dengyihao 已提交
128 129 130 131 132 133 134
typedef struct {
  int32_t reinit;
  int64_t timestamp;
  int32_t count;
  int32_t threshold;
  int64_t interval;
} SFailFastItem;
dengyihao's avatar
dengyihao 已提交
135
// conn pool
dengyihao's avatar
dengyihao 已提交
136
// add expire timeout and capacity limit
dengyihao's avatar
dengyihao 已提交
137
static void*     createConnPool(int size);
138
static void*     destroyConnPool(void* pool);
dengyihao's avatar
dengyihao 已提交
139
static SCliConn* getConnFromPool(void* pool, char* ip, uint32_t port);
dengyihao's avatar
dengyihao 已提交
140
static void      addConnToPool(void* pool, SCliConn* conn);
dengyihao's avatar
dengyihao 已提交
141
static void      doCloseIdleConn(void* param);
dengyihao's avatar
dengyihao 已提交
142

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

dengyihao's avatar
dengyihao 已提交
161 162
static void cliHandleBatchReq(SCliBatch* pBatch, SCliThrd* pThrd);
// static void cliConnBatchCb(uv_connect_t* req, int status);
dengyihao's avatar
dengyihao 已提交
163
static void cliSendBatchCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
164 165 166
// static void cliConnBatchCb(uv_connect_t* req, int status);
//  callback after conn to server
// static void cliConnBatchCb(uv_connect_t* req, int status);
dengyihao's avatar
dengyihao 已提交
167

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

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

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

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

dengyihao's avatar
dengyihao 已提交
181 182
static int32_t cliPreCheckSessionLimit(SCliThrd* pThrd, char* ip, uint16_t port);

dengyihao's avatar
dengyihao 已提交
183
// cli util func
dengyihao's avatar
dengyihao 已提交
184 185
static FORCE_INLINE bool cliIsEpsetUpdated(int32_t code, STransConnCtx* pCtx);
static FORCE_INLINE void cliMayCvtFqdnToIp(SEpSet* pEpSet, SCvtAddr* pCvtAddr);
dengyihao's avatar
dengyihao 已提交
186

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

dengyihao's avatar
dengyihao 已提交
189 190 191
static FORCE_INLINE uint32_t cliGetIpFromFqdnCache(SHashObj* cache, char* fqdn);
static FORCE_INLINE void     cliUpdateFqdnCache(SHashObj* cache, char* fqdn);

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

199
// handle req from app
dengyihao's avatar
dengyihao 已提交
200 201 202 203 204 205
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 已提交
206 207
/// static void (*cliAsyncHandle[])(SCliMsg* pMsg, SCliThrd* pThrd) = {cliHandleReq, cliHandleQuit, cliHandleRelease,
/// NULL,cliHandleUpdate};
dengyihao's avatar
dengyihao 已提交
208

dengyihao's avatar
dengyihao 已提交
209 210
static FORCE_INLINE void destroyUserdata(STransMsg* userdata);
static FORCE_INLINE void destroyCmsg(void* cmsg);
dengyihao's avatar
dengyihao 已提交
211
static FORCE_INLINE void destroyCmsgAndAhandle(void* cmsg);
dengyihao's avatar
dengyihao 已提交
212
static FORCE_INLINE int  cliRBChoseIdx(STrans* pTransInst);
dengyihao's avatar
dengyihao 已提交
213
static FORCE_INLINE void transDestroyConnCtx(STransConnCtx* ctx);
dengyihao's avatar
dengyihao 已提交
214

dengyihao's avatar
dengyihao 已提交
215
// thread obj
dengyihao's avatar
dengyihao 已提交
216
static SCliThrd* createThrdObj(void* trans);
dengyihao's avatar
dengyihao 已提交
217
static void      destroyThrdObj(SCliThrd* pThrd);
dengyihao's avatar
dengyihao 已提交
218

dengyihao's avatar
dengyihao 已提交
219 220 221 222 223 224 225 226 227
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 已提交
228 229 230 231 232 233
// snprintf may cause performance problem
#define CONN_CONSTRUCT_HASH_KEY(key, ip, port)          \
  do {                                                  \
    snprintf(key, sizeof(key), "%s:%d", ip, (int)port); \
  } while (0)

dengyihao's avatar
dengyihao 已提交
234 235
#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 已提交
236

dengyihao's avatar
dengyihao 已提交
237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252
#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 已提交
253
  } while (0)
dengyihao's avatar
dengyihao 已提交
254

U
ubuntu 已提交
255 256 257 258 259 260 261 262 263 264 265 266 267
#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 已提交
268

dengyihao's avatar
formate  
dengyihao 已提交
269 270
#define CONN_SET_PERSIST_BY_APP(conn) \
  do {                                \
dengyihao's avatar
dengyihao 已提交
271 272
    if (conn->status == ConnNormal) { \
      conn->status = ConnAcquire;     \
dengyihao's avatar
formate  
dengyihao 已提交
273 274 275
      transRefCliHandle(conn);        \
    }                                 \
  } while (0)
276

dengyihao's avatar
dengyihao 已提交
277 278 279 280
#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)
281

S
Shengliang Guan 已提交
282 283
#define REQUEST_NO_RESP(msg)         ((msg)->info.noResp == 1)
#define REQUEST_PERSIS_HANDLE(msg)   ((msg)->info.persistHandle == 1)
dengyihao's avatar
dengyihao 已提交
284
#define REQUEST_RELEASE_HANDLE(cmsg) ((cmsg)->type == Release)
dengyihao's avatar
dengyihao 已提交
285

dengyihao's avatar
dengyihao 已提交
286
#define EPSET_IS_VALID(epSet)       ((epSet) != NULL && (epSet)->numOfEps >= 0 && (epSet)->inUse >= 0)
dengyihao's avatar
dengyihao 已提交
287
#define EPSET_GET_SIZE(epSet)       (epSet)->numOfEps
dengyihao's avatar
dengyihao 已提交
288
#define EPSET_GET_INUSE_IP(epSet)   ((epSet)->eps[(epSet)->inUse].fqdn)
dengyihao's avatar
dengyihao 已提交
289
#define EPSET_GET_INUSE_PORT(epSet) ((epSet)->eps[(epSet)->inUse].port)
dengyihao's avatar
dengyihao 已提交
290 291 292 293 294 295
#define EPSET_FORWARD_INUSE(epSet)                             \
  do {                                                         \
    if ((epSet)->numOfEps != 0) {                              \
      ++((epSet)->inUse);                                      \
      (epSet)->inUse = ((epSet)->inUse) % ((epSet)->numOfEps); \
    }                                                          \
dengyihao's avatar
dengyihao 已提交
296
  } while (0)
dengyihao's avatar
dengyihao 已提交
297

dengyihao's avatar
dengyihao 已提交
298 299 300 301 302 303 304 305 306 307 308
#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 已提交
309
  } while (0);
dengyihao's avatar
dengyihao 已提交
310

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

dengyihao's avatar
dengyihao 已提交
313 314 315 316 317 318 319 320
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 已提交
321
      } else if (msg->ctx->ahandle != NULL && pThrd->destroyAhandleFp != NULL) {
dengyihao's avatar
dengyihao 已提交
322
        tDebug("%s conn %p destroy unfinished ahandle %p", CONN_GET_INST_LABEL(conn), conn, msg->ctx->ahandle);
dengyihao's avatar
dengyihao 已提交
323
        pThrd->destroyAhandleFp(msg->ctx->ahandle);
dengyihao's avatar
dengyihao 已提交
324 325 326 327
      }
    }
    destroyCmsg(msg);
  }
dengyihao's avatar
dengyihao 已提交
328
  transQueueClear(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
329
  memset(&conn->ctx, 0, sizeof(conn->ctx));
dengyihao's avatar
dengyihao 已提交
330
}
dengyihao's avatar
dengyihao 已提交
331
bool cliMaySendCachedMsg(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
332
  if (!transQueueEmpty(&conn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
333
    SCliMsg* pCliMsg = NULL;
U
ubuntu 已提交
334
    CONN_GET_NEXT_SENDMSG(conn);
dengyihao's avatar
dengyihao 已提交
335 336 337 338 339 340
    if (pCliMsg == NULL)
      return false;
    else {
      cliSend(conn);
      return true;
    }
dengyihao's avatar
dengyihao 已提交
341
  }
dengyihao's avatar
dengyihao 已提交
342
  return false;
U
ubuntu 已提交
343 344
_RETURN:
  return false;
dengyihao's avatar
dengyihao 已提交
345
}
dengyihao's avatar
formate  
dengyihao 已提交
346
void cliHandleResp(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
347 348
  SCliThrd* pThrd = conn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
349

dengyihao's avatar
dengyihao 已提交
350 351 352 353 354 355
  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 已提交
356
    conn->timer->data = NULL;
dengyihao's avatar
dengyihao 已提交
357
    conn->timer = NULL;
dengyihao's avatar
dengyihao 已提交
358 359
  }

dengyihao's avatar
opt rpc  
dengyihao 已提交
360
  STransMsgHead* pHead = NULL;
dengyihao's avatar
dengyihao 已提交
361 362 363

  int32_t msgLen = transDumpFromBuffer(&conn->readBuf, (char**)&pHead);
  if (msgLen <= 0) {
dengyihao's avatar
dengyihao 已提交
364 365 366
    tDebug("%s conn %p recv invalid packet ", CONN_GET_INST_LABEL(conn), conn);
    return;
  }
dengyihao's avatar
dengyihao 已提交
367 368 369 370

  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 已提交
371 372
  pHead->code = htonl(pHead->code);
  pHead->msgLen = htonl(pHead->msgLen);
dengyihao's avatar
dengyihao 已提交
373 374 375 376
  if (cliRecvReleaseReq(conn, pHead)) {
    return;
  }

dengyihao's avatar
dengyihao 已提交
377 378 379 380 381
  STransMsg transMsg = {0};
  transMsg.contLen = transContLenFromMsg(pHead->msgLen);
  transMsg.pCont = transContFromHead((char*)pHead);
  transMsg.code = pHead->code;
  transMsg.msgType = pHead->msgType;
S
Shengliang Guan 已提交
382
  transMsg.info.ahandle = NULL;
dengyihao's avatar
dengyihao 已提交
383
  transMsg.info.traceId = pHead->traceId;
dengyihao's avatar
dengyihao 已提交
384
  transMsg.info.hasEpSet = pHead->hasEpSet;
dengyihao's avatar
dengyihao 已提交
385

dengyihao's avatar
dengyihao 已提交
386 387 388 389
  SCliMsg*       pMsg = NULL;
  STransConnCtx* pCtx = NULL;
  if (CONN_NO_PERSIST_BY_APP(conn)) {
    pMsg = transQueuePop(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
390 391 392

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

dengyihao's avatar
dengyihao 已提交
419
  STraceId* trace = &transMsg.info.traceId;
dengyihao's avatar
dengyihao 已提交
420
  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 已提交
421
          TMSG_INFO(pHead->msgType), conn->dst, conn->src, pHead->msgLen, tstrerror(transMsg.code));
dengyihao's avatar
dengyihao 已提交
422

dengyihao's avatar
dengyihao 已提交
423
  if (pCtx == NULL && CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
424
    tDebug("%s except, conn %p read while cli ignore it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
425
    transFreeMsg(transMsg.pCont);
dengyihao's avatar
dengyihao 已提交
426 427
    return;
  }
S
Shengliang Guan 已提交
428
  if (CONN_RELEASE_BY_SERVER(conn) && transMsg.info.ahandle == NULL) {
dengyihao's avatar
dengyihao 已提交
429
    tDebug("%s except, conn %p read while cli ignore it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
430
    transFreeMsg(transMsg.pCont);
dengyihao's avatar
dengyihao 已提交
431 432
    return;
  }
dengyihao's avatar
dengyihao 已提交
433

dengyihao's avatar
dengyihao 已提交
434
  if (pMsg == NULL || (pMsg && pMsg->type != Release)) {
dengyihao's avatar
dengyihao 已提交
435 436 437
    if (cliAppCb(conn, &transMsg, pMsg) != 0) {
      return;
    }
dengyihao's avatar
dengyihao 已提交
438
  }
dengyihao's avatar
dengyihao 已提交
439 440
  destroyCmsg(pMsg);

dengyihao's avatar
dengyihao 已提交
441
  if (cliMaySendCachedMsg(conn) == true) {
dengyihao's avatar
dengyihao 已提交
442 443
    return;
  }
dengyihao's avatar
dengyihao 已提交
444

U
ubuntu 已提交
445
  if (CONN_NO_PERSIST_BY_APP(conn)) {
dengyihao's avatar
dengyihao 已提交
446
    return addConnToPool(pThrd->pool, conn);
dengyihao's avatar
dengyihao 已提交
447
  }
dengyihao's avatar
test  
dengyihao 已提交
448

dengyihao's avatar
dengyihao 已提交
449
  uv_read_start((uv_stream_t*)conn->stream, cliAllocRecvBufferCb, cliRecvCb);
dengyihao's avatar
dengyihao 已提交
450
}
U
ubuntu 已提交
451

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

D
dapan1121 已提交
466 467
    if (pMsg == NULL && once) {
      break;
dengyihao's avatar
dengyihao 已提交
468
    }
dengyihao's avatar
enh log  
dengyihao 已提交
469 470 471 472 473 474

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

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

dengyihao's avatar
dengyihao 已提交
477
    STransMsg transMsg = {0};
dengyihao's avatar
dengyihao 已提交
478
    transMsg.code = code == -1 ? (pConn->broken ? TSDB_CODE_RPC_BROKEN_LINK : TSDB_CODE_RPC_NETWORK_UNAVAIL) : code;
dengyihao's avatar
dengyihao 已提交
479
    transMsg.msgType = pMsg ? pMsg->msg.msgType + 1 : 0;
S
Shengliang Guan 已提交
480
    transMsg.info.ahandle = NULL;
dengyihao's avatar
dengyihao 已提交
481

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

    if (pCtx == NULL || pCtx->pSem == NULL) {
S
Shengliang Guan 已提交
498
      if (transMsg.info.ahandle == NULL) {
dengyihao's avatar
dengyihao 已提交
499 500 501 502 503
        if (pMsg == NULL || REQUEST_NO_RESP(&pMsg->msg) || pMsg->type == Release) {
          destroyCmsg(pMsg);
          once = true;
          continue;
        }
U
ubuntu 已提交
504
      }
dengyihao's avatar
dengyihao 已提交
505
    }
dengyihao's avatar
enh log  
dengyihao 已提交
506

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

dengyihao's avatar
dengyihao 已提交
522 523
void cliConnTimeout(uv_timer_t* handle) {
  SCliConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
524 525
  SCliThrd* pThrd = conn->hostThrd;

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

dengyihao's avatar
dengyihao 已提交
528
  uv_timer_stop(handle);
dengyihao's avatar
dengyihao 已提交
529 530 531
  handle->data = NULL;
  taosArrayPush(pThrd->timerList, &conn->timer);
  conn->timer = NULL;
dengyihao's avatar
dengyihao 已提交
532 533

  cliHandleFastFail(conn, UV_ECANCELED);
dengyihao's avatar
dengyihao 已提交
534
}
dengyihao's avatar
dengyihao 已提交
535 536 537 538
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 已提交
539
  uv_read_stop(conn->stream);
dengyihao's avatar
dengyihao 已提交
540 541
  cliHandleExceptImpl(conn, TSDB_CODE_RPC_TIMEOUT);
}
U
ubuntu 已提交
542 543

void* createConnPool(int size) {
544 545
  // thread local, no lock
  return taosHashInit(size, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
546
}
U
ubuntu 已提交
547
void* destroyConnPool(void* pool) {
dengyihao's avatar
dengyihao 已提交
548
  SConnList* connList = taosHashIterate((SHashObj*)pool, NULL);
dengyihao's avatar
dengyihao 已提交
549
  while (connList != NULL) {
dengyihao's avatar
dengyihao 已提交
550 551
    while (!QUEUE_IS_EMPTY(&connList->conns)) {
      queue*    h = QUEUE_HEAD(&connList->conns);
dengyihao's avatar
dengyihao 已提交
552
      SCliConn* c = QUEUE_DATA(h, SCliConn, q);
U
ubuntu 已提交
553
      cliDestroyConn(c, true);
dengyihao's avatar
dengyihao 已提交
554
    }
dengyihao's avatar
dengyihao 已提交
555
    connList = taosHashIterate((SHashObj*)pool, connList);
dengyihao's avatar
dengyihao 已提交
556
  }
dengyihao's avatar
dengyihao 已提交
557
  taosHashCleanup(pool);
558
  return NULL;
dengyihao's avatar
dengyihao 已提交
559 560
}

dengyihao's avatar
dengyihao 已提交
561
static SCliConn* getConnFromPool(void* pool, char* ip, uint32_t port) {
dengyihao's avatar
TS-2015  
dengyihao 已提交
562
  char key[TSDB_FQDN_LEN + 64] = {0};
dengyihao's avatar
dengyihao 已提交
563
  CONN_CONSTRUCT_HASH_KEY(key, ip, port);
dengyihao's avatar
dengyihao 已提交
564 565

  SConnList* plist = taosHashGet((SHashObj*)pool, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
566
  if (plist == NULL) {
dengyihao's avatar
dengyihao 已提交
567
    SConnList list = {0};
dengyihao's avatar
dengyihao 已提交
568 569
    taosHashPut((SHashObj*)pool, key, strlen(key), (void*)&list, sizeof(list));
    plist = taosHashGet((SHashObj*)pool, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
570
    if (plist == NULL) return NULL;
dengyihao's avatar
dengyihao 已提交
571
    QUEUE_INIT(&plist->conns);
dengyihao's avatar
dengyihao 已提交
572 573
  }

dengyihao's avatar
dengyihao 已提交
574
  if (QUEUE_IS_EMPTY(&plist->conns)) {
dengyihao's avatar
dengyihao 已提交
575 576
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
577 578

  plist->size -= 1;
dengyihao's avatar
dengyihao 已提交
579
  queue*    h = QUEUE_HEAD(&plist->conns);
dengyihao's avatar
dengyihao 已提交
580
  SCliConn* conn = QUEUE_DATA(h, SCliConn, q);
dengyihao's avatar
dengyihao 已提交
581
  conn->status = ConnNormal;
dengyihao's avatar
dengyihao 已提交
582 583
  QUEUE_REMOVE(&conn->q);
  QUEUE_INIT(&conn->q);
dengyihao's avatar
dengyihao 已提交
584

dengyihao's avatar
dengyihao 已提交
585 586 587 588
  if (conn->task != NULL) {
    transDQCancel(((SCliThrd*)conn->hostThrd)->timeoutQueue, conn->task);
    conn->task = NULL;
  }
dengyihao's avatar
dengyihao 已提交
589
  return conn;
dengyihao's avatar
dengyihao 已提交
590
}
dengyihao's avatar
dengyihao 已提交
591 592 593 594 595 596
static void addConnToPool(void* pool, SCliConn* conn) {
  if (conn->status == ConnInPool) {
    return;
  }
  allocConnRef(conn, true);

dengyihao's avatar
dengyihao 已提交
597
  SCliThrd* thrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
598 599 600 601 602 603
  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 已提交
604 605 606 607 608
  if (T_REF_VAL_GET(conn) > 1) {
    transUnrefCliHandle(conn);
  }

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

dengyihao's avatar
dengyihao 已提交
610 611 612 613
  conn->status = ConnInPool;

  if (conn->list == NULL) {
    tTrace("%s conn %p added to conn pool, read buf cap:%d", CONN_GET_INST_LABEL(conn), conn, conn->readBuf.cap);
dengyihao's avatar
dengyihao 已提交
614
    conn->list = taosHashGet((SHashObj*)pool, conn->ip, strlen(conn->ip));
dengyihao's avatar
dengyihao 已提交
615 616
  } else {
    tTrace("%s conn %p added to conn pool, read buf cap:%d", CONN_GET_INST_LABEL(conn), conn, conn->readBuf.cap);
dengyihao's avatar
dengyihao 已提交
617
  }
dengyihao's avatar
dengyihao 已提交
618
  QUEUE_PUSH(&conn->list->conns, &conn->q);
dengyihao's avatar
dengyihao 已提交
619
  conn->list->size += 1;
dengyihao's avatar
dengyihao 已提交
620

dengyihao's avatar
dengyihao 已提交
621
  if (conn->list->size >= 250) {
dengyihao's avatar
dengyihao 已提交
622 623
    STaskArg* arg = taosMemoryCalloc(1, sizeof(STaskArg));
    arg->param1 = conn;
dengyihao's avatar
dengyihao 已提交
624
    arg->param2 = thrd;
dengyihao's avatar
dengyihao 已提交
625 626

    STrans* pTransInst = thrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
627 628
    conn->task = transDQSched(thrd->timeoutQueue, doCloseIdleConn, arg, CONN_PERSIST_TIME(pTransInst->idleTime));
  }
dengyihao's avatar
dengyihao 已提交
629
}
dengyihao's avatar
dengyihao 已提交
630
static int32_t allocConnRef(SCliConn* conn, bool update) {
dengyihao's avatar
dengyihao 已提交
631
  if (update) {
dengyihao's avatar
dengyihao 已提交
632
    transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
633
    transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
634
    conn->refId = -1;
dengyihao's avatar
dengyihao 已提交
635 636 637 638
  }
  SExHandle* exh = taosMemoryCalloc(1, sizeof(SExHandle));
  exh->handle = conn;
  exh->pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
639
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
dengyihao's avatar
dengyihao 已提交
640
  conn->refId = exh->refId;
641 642 643 644

  if (conn->refId == -1) {
    taosMemoryFree(exh);
  }
dengyihao's avatar
dengyihao 已提交
645 646 647 648 649
  return 0;
}

static int32_t specifyConnRef(SCliConn* conn, bool update, int64_t handle) {
  if (update) {
dengyihao's avatar
dengyihao 已提交
650
    transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
651 652 653 654 655 656 657 658 659 660 661 662 663
    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 已提交
664
}
dengyihao's avatar
dengyihao 已提交
665

dengyihao's avatar
dengyihao 已提交
666
static void cliAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
667
  SCliConn*    conn = handle->data;
dengyihao's avatar
dengyihao 已提交
668
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
669
  tDebug("%s conn %p alloc read buf", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
670
  transAllocBuffer(pBuf, buf);
dengyihao's avatar
dengyihao 已提交
671
}
U
ubuntu 已提交
672
static void cliRecvCb(uv_stream_t* handle, ssize_t nread, const uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
673
  // impl later
dengyihao's avatar
dengyihao 已提交
674 675 676
  if (handle->data == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
677 678
  SCliConn*    conn = handle->data;
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
679
  if (nread > 0) {
dengyihao's avatar
dengyihao 已提交
680
    pBuf->len += nread;
dengyihao's avatar
dengyihao 已提交
681
    while (transReadComplete(pBuf)) {
dengyihao's avatar
dengyihao 已提交
682
      tDebug("%s conn %p read complete", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
683 684 685 686 687 688
      if (pBuf->invalid) {
        cliHandleExcept(conn);
        break;
      } else {
        cliHandleResp(conn);
      }
dengyihao's avatar
dengyihao 已提交
689
    }
dengyihao's avatar
dengyihao 已提交
690 691
    return;
  }
dengyihao's avatar
dengyihao 已提交
692

693
  if (nread == 0) {
dengyihao's avatar
dengyihao 已提交
694 695 696
    // 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 已提交
697
    tTrace("%s conn %p read empty", CONN_GET_INST_LABEL(conn), conn);
698 699
    return;
  }
dengyihao's avatar
fix bug  
dengyihao 已提交
700
  if (nread < 0) {
dengyihao's avatar
dengyihao 已提交
701
    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 已提交
702
    conn->broken = true;
U
ubuntu 已提交
703
    cliHandleExcept(conn);
dengyihao's avatar
dengyihao 已提交
704
  }
dengyihao's avatar
dengyihao 已提交
705
}
dengyihao's avatar
dengyihao 已提交
706

dengyihao's avatar
dengyihao 已提交
707
static SCliConn* cliCreateConn(SCliThrd* pThrd) {
wafwerar's avatar
wafwerar 已提交
708
  SCliConn* conn = taosMemoryCalloc(1, sizeof(SCliConn));
dengyihao's avatar
dengyihao 已提交
709
  // read/write stream handle
G
gccgdb1234 已提交
710
  conn->stream = (uv_stream_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
711 712 713
  uv_tcp_init(pThrd->loop, (uv_tcp_t*)(conn->stream));
  conn->stream->data = conn;

dengyihao's avatar
dengyihao 已提交
714 715 716 717 718 719 720 721
  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 已提交
722

dengyihao's avatar
dengyihao 已提交
723
  conn->connReq.data = conn;
dengyihao's avatar
dengyihao 已提交
724 725
  transReqQueueInit(&conn->wreqQueue);

dengyihao's avatar
dengyihao 已提交
726
  transQueueInit(&conn->cliMsgs, NULL);
dengyihao's avatar
opt rpc  
dengyihao 已提交
727 728

  transInitBuffer(&conn->readBuf);
dengyihao's avatar
dengyihao 已提交
729
  QUEUE_INIT(&conn->q);
dengyihao's avatar
dengyihao 已提交
730
  conn->hostThrd = pThrd;
dengyihao's avatar
dengyihao 已提交
731 732
  conn->status = ConnNormal;
  conn->broken = 0;
dengyihao's avatar
dengyihao 已提交
733
  transRefCliHandle(conn);
dengyihao's avatar
dengyihao 已提交
734

dengyihao's avatar
dengyihao 已提交
735
  atomic_add_fetch_32(&pThrd->connCount, 1);
dengyihao's avatar
dengyihao 已提交
736
  allocConnRef(conn, false);
dengyihao's avatar
dengyihao 已提交
737

dengyihao's avatar
dengyihao 已提交
738 739
  return conn;
}
U
ubuntu 已提交
740
static void cliDestroyConn(SCliConn* conn, bool clear) {
dengyihao's avatar
dengyihao 已提交
741
  SCliThrd* pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
742
  tTrace("%s conn %p remove from conn pool", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
743 744
  QUEUE_REMOVE(&conn->q);
  QUEUE_INIT(&conn->q);
dengyihao's avatar
dengyihao 已提交
745
  transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
746
  transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
opt rpc  
dengyihao 已提交
747
  conn->refId = -1;
dengyihao's avatar
dengyihao 已提交
748

dengyihao's avatar
dengyihao 已提交
749 750 751 752 753 754 755
  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 已提交
756
    taosArrayPush(pThrd->timerList, &conn->timer);
dengyihao's avatar
dengyihao 已提交
757 758
    conn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
759

dengyihao's avatar
dengyihao 已提交
760
  if (clear) {
dengyihao's avatar
dengyihao 已提交
761
    if (!uv_is_closing((uv_handle_t*)conn->stream)) {
dengyihao's avatar
dengyihao 已提交
762
      uv_read_stop(conn->stream);
dengyihao's avatar
dengyihao 已提交
763 764
      uv_close((uv_handle_t*)conn->stream, cliDestroy);
    }
765
  }
dengyihao's avatar
dengyihao 已提交
766
}
U
ubuntu 已提交
767
static void cliDestroy(uv_handle_t* handle) {
dengyihao's avatar
dengyihao 已提交
768 769 770
  if (uv_handle_get_type(handle) != UV_TCP || handle->data == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
771
  SCliConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
772
  SCliThrd* pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
773 774
  if (conn->timer != NULL) {
    uv_timer_stop(conn->timer);
dengyihao's avatar
dengyihao 已提交
775 776
    taosArrayPush(pThrd->timerList, &conn->timer);
    conn->timer->data = NULL;
dengyihao's avatar
dengyihao 已提交
777 778
    conn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
779 780 781
  int32_t* oVal = taosHashGet(pThrd->connLimitCache, conn->ip, strlen(conn->ip));
  int32_t  nVal = oVal == NULL ? 0 : (*oVal) - 1;
  taosHashPut(pThrd->connLimitCache, conn->ip, strlen(conn->ip), &nVal, sizeof(nVal));
dengyihao's avatar
dengyihao 已提交
782

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

dengyihao's avatar
dengyihao 已提交
785
  transReleaseExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
786
  transRemoveExHandle(transGetRefMgt(), conn->refId);
wafwerar's avatar
wafwerar 已提交
787 788
  taosMemoryFree(conn->ip);
  taosMemoryFree(conn->stream);
dengyihao's avatar
dengyihao 已提交
789 790 791

  cliDestroyConnMsgs(conn, true);

dengyihao's avatar
dengyihao 已提交
792
  tTrace("%s conn %p destroy successfully", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
793
  transReqQueueClear(&conn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
794
  transDestroyBuffer(&conn->readBuf);
dengyihao's avatar
dengyihao 已提交
795

wafwerar's avatar
wafwerar 已提交
796
  taosMemoryFree(conn);
dengyihao's avatar
dengyihao 已提交
797
}
dengyihao's avatar
dengyihao 已提交
798
static bool cliHandleNoResp(SCliConn* conn) {
dengyihao's avatar
dengyihao 已提交
799 800
  bool res = false;
  if (!transQueueEmpty(&conn->cliMsgs)) {
dengyihao's avatar
dengyihao 已提交
801
    SCliMsg* pMsg = transQueueGet(&conn->cliMsgs, 0);
dengyihao's avatar
dengyihao 已提交
802
    if (REQUEST_NO_RESP(&pMsg->msg)) {
dengyihao's avatar
dengyihao 已提交
803
      transQueuePop(&conn->cliMsgs);
dengyihao's avatar
dengyihao 已提交
804 805 806 807 808
      destroyCmsg(pMsg);
      res = true;
    }
    if (res == true) {
      if (cliMaySendCachedMsg(conn) == false) {
dengyihao's avatar
dengyihao 已提交
809
        SCliThrd* thrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
810
        addConnToPool(thrd->pool, conn);
dengyihao's avatar
dengyihao 已提交
811 812 813
        res = false;
      } else {
        res = true;
dengyihao's avatar
dengyihao 已提交
814 815 816 817 818
      }
    }
  }
  return res;
}
U
ubuntu 已提交
819
static void cliSendCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
820 821
  SCliConn* pConn = transReqQueueRemove(req);
  if (pConn == NULL) return;
dengyihao's avatar
dengyihao 已提交
822

dengyihao's avatar
dengyihao 已提交
823 824 825
  SCliMsg* pMsg = !transQueueEmpty(&pConn->cliMsgs) ? transQueueGet(&pConn->cliMsgs, 0) : NULL;
  if (pMsg != NULL) {
    int64_t cost = taosGetTimestampUs() - pMsg->st;
dengyihao's avatar
dengyihao 已提交
826
    if (cost > 1000 * 20) {
dengyihao's avatar
dengyihao 已提交
827 828 829 830
      tWarn("%s conn %p send cost:%dus, send exception", CONN_GET_INST_LABEL(pConn), pConn, (int)cost);
    }
  }

dengyihao's avatar
dengyihao 已提交
831
  if (status == 0) {
dengyihao's avatar
dengyihao 已提交
832
    tTrace("%s conn %p data already was written out", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
833
  } else {
dengyihao's avatar
dengyihao 已提交
834 835 836 837
    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 已提交
838 839
    return;
  }
dengyihao's avatar
dengyihao 已提交
840
  if (cliHandleNoResp(pConn) == true) {
dengyihao's avatar
dengyihao 已提交
841
    tTrace("%s conn %p no resp required", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
842 843
    return;
  }
dengyihao's avatar
dengyihao 已提交
844
  uv_read_start((uv_stream_t*)pConn->stream, cliAllocRecvBufferCb, cliRecvCb);
dengyihao's avatar
dengyihao 已提交
845
}
dengyihao's avatar
dengyihao 已提交
846 847 848 849 850 851 852 853 854 855
void cliSendBatch(SCliConn* pConn) {
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;

  SCliBatch* pBatch = pConn->pBatch;
  int32_t    wLen = pBatch->wLen;

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

dengyihao's avatar
dengyihao 已提交
856 857
  queue* h = NULL;
  QUEUE_FOREACH(h, &pBatch->wq) {
dengyihao's avatar
dengyihao 已提交
858 859 860 861 862 863 864 865 866
    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 已提交
867

dengyihao's avatar
dengyihao 已提交
868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883
    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 已提交
884 885 886 887 888 889 890 891
    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 已提交
892 893 894 895 896
    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 已提交
897
  tDebug("%s conn %p start to send batch msg, batch size:%d, msgLen:%d", CONN_GET_INST_LABEL(pConn), pConn,
dengyihao's avatar
dengyihao 已提交
898
         pBatch->wLen, pBatch->batchSize);
dengyihao's avatar
dengyihao 已提交
899 900 901
  uv_write(req, (uv_stream_t*)pConn->stream, wb, wLen, cliSendBatchCb);
  taosMemoryFree(wb);
}
U
ubuntu 已提交
902
void cliSend(SCliConn* pConn) {
dengyihao's avatar
dengyihao 已提交
903 904 905 906 907
  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 已提交
908
    cliHandleExcept(pConn);
dengyihao's avatar
dengyihao 已提交
909 910
    return;
  }
dengyihao's avatar
dengyihao 已提交
911 912

  SCliMsg* pCliMsg = NULL;
U
ubuntu 已提交
913
  CONN_GET_NEXT_SENDMSG(pConn);
dengyihao's avatar
dengyihao 已提交
914 915
  pCliMsg->sent = 1;

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

U
ubuntu 已提交
918
  STransMsg* pMsg = (STransMsg*)(&pCliMsg->msg);
dengyihao's avatar
dengyihao 已提交
919 920 921 922
  if (pMsg->pCont == 0) {
    pMsg->pCont = (void*)rpcMallocCont(0);
    pMsg->contLen = 0;
  }
dengyihao's avatar
dengyihao 已提交
923

dengyihao's avatar
dengyihao 已提交
924
  int            msgLen = transMsgLenFromCont(pMsg->contLen);
dengyihao's avatar
dengyihao 已提交
925
  STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
dengyihao's avatar
dengyihao 已提交
926

dengyihao's avatar
dengyihao 已提交
927 928 929 930 931 932 933 934 935 936 937
  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 已提交
938
  pHead->timestamp = taosHton64(taosGetTimestampUs());
dengyihao's avatar
dengyihao 已提交
939

dengyihao's avatar
dengyihao 已提交
940 941 942
  if (pHead->persist == 1) {
    CONN_SET_PERSIST_BY_APP(pConn);
  }
dengyihao's avatar
dengyihao 已提交
943

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

dengyihao's avatar
dengyihao 已提交
946
  if (pTransInst->startTimer != NULL && pTransInst->startTimer(0, pMsg->msgType)) {
dengyihao's avatar
dengyihao 已提交
947
    uv_timer_t* timer = taosArrayGetSize(pThrd->timerList) > 0 ? *(uv_timer_t**)taosArrayPop(pThrd->timerList) : NULL;
dengyihao's avatar
dengyihao 已提交
948 949
    if (timer == NULL) {
      timer = taosMemoryCalloc(1, sizeof(uv_timer_t));
dengyihao's avatar
dengyihao 已提交
950
      tDebug("no available timer, create a timer %p", timer);
dengyihao's avatar
dengyihao 已提交
951 952 953 954 955
      uv_timer_init(pThrd->loop, timer);
    }
    timer->data = pConn;
    pConn->timer = timer;

dengyihao's avatar
dengyihao 已提交
956 957 958
    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 已提交
959

dengyihao's avatar
dengyihao 已提交
960 961 962 963 964 965 966
  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 已提交
967
  }
dengyihao's avatar
dengyihao 已提交
968

dengyihao's avatar
dengyihao 已提交
969 970
  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 已提交
971

dengyihao's avatar
dengyihao 已提交
972
  uv_buf_t    wb = uv_buf_init((char*)pHead, msgLen);
dengyihao's avatar
dengyihao 已提交
973
  uv_write_t* req = transReqQueuePush(&pConn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
974 975 976

  int status = uv_write(req, (uv_stream_t*)pConn->stream, &wb, 1, cliSendCb);
  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
977
    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 已提交
978 979 980
            uv_err_name(status));
    cliHandleExcept(pConn);
  }
U
ubuntu 已提交
981 982
  return;
_RETURN:
dengyihao's avatar
dengyihao 已提交
983
  return;
dengyihao's avatar
dengyihao 已提交
984
}
dengyihao's avatar
dengyihao 已提交
985 986

static void cliDestroyBatch(SCliBatch* pBatch) {
dengyihao's avatar
dengyihao 已提交
987
  while (!QUEUE_IS_EMPTY(&pBatch->wq)) {
dengyihao's avatar
dengyihao 已提交
988 989
    queue* h = QUEUE_HEAD(&pBatch->wq);
    QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
990

dengyihao's avatar
dengyihao 已提交
991
    SCliMsg* p = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
992 993 994 995
    destroyCmsg(p);
  }
  taosMemoryFree(pBatch);
}
dengyihao's avatar
dengyihao 已提交
996 997
static void cliHandleBatchReq(SCliBatch* pBatch, SCliThrd* pThrd) {
  if (pBatch->wLen == 0 || QUEUE_IS_EMPTY(&pBatch->wq)) {
dengyihao's avatar
dengyihao 已提交
998 999
    return;
  }
dengyihao's avatar
dengyihao 已提交
1000 1001
  STrans*        pTransInst = pThrd->pTransInst;
  SCliBatchList* pList = pBatch->pList;
dengyihao's avatar
dengyihao 已提交
1002

dengyihao's avatar
dengyihao 已提交
1003
  SCliConn* conn = getConnFromPool(pThrd->pool, pList->ip, pList->port);
dengyihao's avatar
dengyihao 已提交
1004

dengyihao's avatar
dengyihao 已提交
1005 1006 1007 1008
  if (conn == NULL && 0 != cliPreCheckSessionLimit(pThrd, pList->ip, pList->port)) {
    tError("%s failed to send batch msg, batch size:%d, msgLen: %d", pTransInst->label, pBatch->wLen,
           pBatch->batchSize);
    cliDestroyBatch(pBatch);
dengyihao's avatar
dengyihao 已提交
1009 1010
    return;
  }
dengyihao's avatar
dengyihao 已提交
1011 1012
  if (conn == NULL) {
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
1013 1014
    conn->pBatch = pBatch;
    conn->ip = strdup(pList->dst);
dengyihao's avatar
dengyihao 已提交
1015

dengyihao's avatar
dengyihao 已提交
1016
    uint32_t ipaddr = cliGetIpFromFqdnCache(pThrd->fqdn2ipCache, pList->ip);
dengyihao's avatar
dengyihao 已提交
1017 1018 1019 1020 1021 1022
    if (ipaddr == 0xffffffff) {
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

dengyihao's avatar
dengyihao 已提交
1023
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1024 1025 1026 1027 1028
      return;
    }
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = ipaddr;
dengyihao's avatar
dengyihao 已提交
1029
    addr.sin_port = (uint16_t)htons(pList->port);
dengyihao's avatar
dengyihao 已提交
1030

dengyihao's avatar
dengyihao 已提交
1031
    tTrace("%s conn %p try to connect to %s", pTransInst->label, conn, pList->dst);
dengyihao's avatar
dengyihao 已提交
1032 1033
    int32_t fd = taosCreateSocketWithTimeout(TRANS_CONN_TIMEOUT * 4);
    if (fd == -1) {
dengyihao's avatar
dengyihao 已提交
1034 1035 1036
      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 已提交
1037 1038 1039 1040
      return;
    }
    int ret = uv_tcp_open((uv_tcp_t*)conn->stream, fd);
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1041 1042
      tError("%s conn %p failed to set stream, reason:%s", transLabel(pTransInst), conn, uv_err_name(ret));
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1043 1044 1045 1046
      return;
    }
    ret = transSetConnOption((uv_tcp_t*)conn->stream);
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1047 1048
      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 已提交
1049 1050 1051 1052 1053 1054 1055 1056 1057
      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 已提交
1058
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1059 1060 1061 1062 1063 1064
      return;
    }
    uv_timer_start(conn->timer, cliConnTimeout, TRANS_CONN_TIMEOUT, 0);
    return;
  }

dengyihao's avatar
dengyihao 已提交
1065
  conn->pBatch = pBatch;
dengyihao's avatar
dengyihao 已提交
1066
  cliSendBatch(conn);
dengyihao's avatar
dengyihao 已提交
1067 1068
}
static void cliSendBatchCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
1069 1070 1071
  SCliConn*  conn = req->data;
  SCliThrd*  thrd = conn->hostThrd;
  SCliBatch* p = conn->pBatch;
dengyihao's avatar
dengyihao 已提交
1072

dengyihao's avatar
dengyihao 已提交
1073 1074 1075 1076 1077
  SCliBatchList* pBatchList = p->pList;

  int32_t empty = QUEUE_IS_EMPTY(&pBatchList->wq);
  pBatchList->connCnt -= 1;

dengyihao's avatar
dengyihao 已提交
1078 1079 1080
  conn->pBatch = NULL;

  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1081
    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 已提交
1082
           p->wLen, p->batchSize, uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
1083
    cliHandleExcept(conn);
dengyihao's avatar
dengyihao 已提交
1084

dengyihao's avatar
dengyihao 已提交
1085
  } else {
dengyihao's avatar
dengyihao 已提交
1086
    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 已提交
1087
           p->batchSize);
dengyihao's avatar
dengyihao 已提交
1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099

    if (empty == false) {
      queue* h = QUEUE_HEAD(&pBatchList->wq);
      QUEUE_REMOVE(h);
      conn->pBatch = QUEUE_DATA(h, SCliBatch, listq);

      pBatchList->connCnt += 1;
      pBatchList->len -= 1;
      cliSendBatch(conn);
      return;
    }

dengyihao's avatar
dengyihao 已提交
1100 1101
    addConnToPool(thrd->pool, conn);
  }
dengyihao's avatar
dengyihao 已提交
1102 1103 1104

  cliDestroyBatch(p);
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
1105
}
dengyihao's avatar
dengyihao 已提交
1106
static void cliHandleFastFail(SCliConn* pConn, int status) {
dengyihao's avatar
dengyihao 已提交
1107
  SCliThrd* pThrd = pConn->hostThrd;
dengyihao's avatar
dengyihao 已提交
1108
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1109 1110 1111

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

dengyihao's avatar
dengyihao 已提交
1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130
  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 已提交
1131
      } else {
dengyihao's avatar
dengyihao 已提交
1132 1133
        SFailFastItem item = {.count = 1, .timestamp = cTimestamp};
        taosHashPut(pThrd->failFastCache, pConn->ip, strlen(pConn->ip), &item, sizeof(SFailFastItem));
dengyihao's avatar
dengyihao 已提交
1134 1135
      }
    }
dengyihao's avatar
dengyihao 已提交
1136
  } else {
dengyihao's avatar
dengyihao 已提交
1137 1138
    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 已提交
1139 1140
    cliDestroyBatch(pConn->pBatch);
    pConn->pBatch = NULL;
dengyihao's avatar
dengyihao 已提交
1141 1142 1143
  }
  cliHandleExcept(pConn);
}
dengyihao's avatar
dengyihao 已提交
1144

dengyihao's avatar
dengyihao 已提交
1145 1146 1147
void cliConnCb(uv_connect_t* req, int status) {
  SCliConn* pConn = req->data;
  SCliThrd* pThrd = pConn->hostThrd;
dengyihao's avatar
dengyihao 已提交
1148 1149 1150 1151 1152 1153 1154 1155 1156 1157
  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 已提交
1158 1159

  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1160 1161 1162 1163 1164
    if (timeout == false) {
      cliHandleFastFail(pConn, status);
    } else if (timeout == true) {
      // already deal by timeout
    }
1165
    return;
dengyihao's avatar
dengyihao 已提交
1166
  }
dengyihao's avatar
dengyihao 已提交
1167

dengyihao's avatar
dengyihao 已提交
1168
  int32_t* oVal = taosHashGet(pThrd->connLimitCache, pConn->ip, strlen(pConn->ip));
dengyihao's avatar
dengyihao 已提交
1169
  int32_t  nVal = oVal == NULL ? 0 : (*oVal) + 1;
dengyihao's avatar
dengyihao 已提交
1170
  taosHashPut(pThrd->connLimitCache, pConn->ip, strlen(pConn->ip), &nVal, sizeof(nVal));
dengyihao's avatar
dengyihao 已提交
1171 1172 1173

  struct sockaddr peername, sockname;
  int             addrlen = sizeof(peername);
dengyihao's avatar
dengyihao 已提交
1174
  uv_tcp_getpeername((uv_tcp_t*)pConn->stream, &peername, &addrlen);
dengyihao's avatar
dengyihao 已提交
1175
  transSockInfo2Str(&peername, pConn->dst);
dengyihao's avatar
dengyihao 已提交
1176

dengyihao's avatar
dengyihao 已提交
1177 1178
  addrlen = sizeof(sockname);
  uv_tcp_getsockname((uv_tcp_t*)pConn->stream, &sockname, &addrlen);
dengyihao's avatar
dengyihao 已提交
1179
  transSockInfo2Str(&sockname, pConn->src);
dengyihao's avatar
dengyihao 已提交
1180

dengyihao's avatar
dengyihao 已提交
1181
  tTrace("%s conn %p connect to server successfully", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
1182 1183 1184 1185 1186
  if (pConn->pBatch != NULL) {
    cliSendBatch(pConn);
  } else {
    cliSend(pConn);
  }
dengyihao's avatar
dengyihao 已提交
1187 1188
}

dengyihao's avatar
dengyihao 已提交
1189
static void cliHandleQuit(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1190 1191 1192 1193 1194
  if (!transAsyncPoolIsEmpty(pThrd->asyncPool)) {
    pThrd->stopMsg = pMsg;
    return;
  }
  pThrd->stopMsg = NULL;
dengyihao's avatar
dengyihao 已提交
1195
  pThrd->quit = true;
U
ubuntu 已提交
1196
  tDebug("cli work thread %p start to quit", pThrd);
dengyihao's avatar
dengyihao 已提交
1197
  destroyCmsg(pMsg);
dengyihao's avatar
fix bug  
dengyihao 已提交
1198
  destroyConnPool(pThrd->pool);
dengyihao's avatar
dengyihao 已提交
1199
  uv_walk(pThrd->loop, cliWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
1200
}
dengyihao's avatar
dengyihao 已提交
1201
static void cliHandleRelease(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1202
  int64_t    refId = (int64_t)(pMsg->msg.info.handle);
dengyihao's avatar
dengyihao 已提交
1203
  SExHandle* exh = transAcquireExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1204
  if (exh == NULL) {
dengyihao's avatar
dengyihao 已提交
1205
    tDebug("%" PRId64 " already released", refId);
dengyihao's avatar
dengyihao 已提交
1206 1207
    destroyCmsg(pMsg);
    return;
dengyihao's avatar
dengyihao 已提交
1208 1209 1210
  }

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

dengyihao's avatar
dengyihao 已提交
1214 1215
  if (T_REF_VAL_GET(conn) == 2) {
    transUnrefCliHandle(conn);
dengyihao's avatar
dengyihao 已提交
1216 1217
    if (!transQueuePush(&conn->cliMsgs, pMsg)) {
      return;
dengyihao's avatar
dengyihao 已提交
1218 1219
    }
    cliSend(conn);
dengyihao's avatar
dengyihao 已提交
1220 1221 1222
  } else {
    tError("%s conn %p already released", CONN_GET_INST_LABEL(conn), conn);
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1223 1224
  }
}
dengyihao's avatar
dengyihao 已提交
1225
static void cliHandleUpdate(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1226
  STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
dengyihao 已提交
1227
  pThrd->cvtAddr = pCtx->cvtAddr;
dengyihao's avatar
dengyihao 已提交
1228 1229
  destroyCmsg(pMsg);
}
U
ubuntu 已提交
1230

dengyihao's avatar
dengyihao 已提交
1231
SCliConn* cliGetConn(SCliMsg* pMsg, SCliThrd* pThrd, bool* ignore) {
dengyihao's avatar
dengyihao 已提交
1232 1233 1234 1235
  STransConnCtx* pCtx = pMsg->ctx;
  SCliConn*      conn = NULL;

  int64_t refId = (int64_t)(pMsg->msg.info.handle);
dengyihao's avatar
dengyihao 已提交
1236
  if (refId != 0) {
dengyihao's avatar
dengyihao 已提交
1237
    SExHandle* exh = transAcquireExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1238
    if (exh == NULL) {
dengyihao's avatar
dengyihao 已提交
1239
      tError("failed to get conn, refId: %" PRId64 "", refId);
dengyihao's avatar
dengyihao 已提交
1240 1241
      *ignore = true;
      return NULL;
dengyihao's avatar
dengyihao 已提交
1242 1243
    } else {
      conn = exh->handle;
dengyihao's avatar
dengyihao 已提交
1244 1245
      if (conn == NULL) {
        conn = getConnFromPool(pThrd->pool, EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet));
dengyihao's avatar
dengyihao 已提交
1246
        if (conn != NULL) specifyConnRef(conn, true, refId);
dengyihao's avatar
dengyihao 已提交
1247
      }
dengyihao's avatar
dengyihao 已提交
1248
      transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1249 1250 1251
    }
    return conn;
  };
dengyihao's avatar
dengyihao 已提交
1252 1253 1254

  conn = getConnFromPool(pThrd->pool, EPSET_GET_INUSE_IP(&pCtx->epSet), EPSET_GET_INUSE_PORT(&pCtx->epSet));
  if (conn != NULL) {
dengyihao's avatar
dengyihao 已提交
1255
    tTrace("%s conn %p get from conn pool:%p", CONN_GET_INST_LABEL(conn), conn, pThrd->pool);
dengyihao's avatar
dengyihao 已提交
1256
  } else {
dengyihao's avatar
dengyihao 已提交
1257
    tTrace("%s not found conn in conn pool:%p", ((STrans*)pThrd->pTransInst)->label, pThrd->pool);
dengyihao's avatar
dengyihao 已提交
1258
  }
dengyihao's avatar
dengyihao 已提交
1259 1260
  return conn;
}
dengyihao's avatar
dengyihao 已提交
1261
FORCE_INLINE void cliMayCvtFqdnToIp(SEpSet* pEpSet, SCvtAddr* pCvtAddr) {
dengyihao's avatar
dengyihao 已提交
1262 1263 1264
  if (pCvtAddr->cvt == false) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1265 1266 1267
  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 已提交
1268 1269
  }
}
dengyihao's avatar
dengyihao 已提交
1270

dengyihao's avatar
dengyihao 已提交
1271
FORCE_INLINE bool cliIsEpsetUpdated(int32_t code, STransConnCtx* pCtx) {
dengyihao's avatar
dengyihao 已提交
1272
  if (code != 0) return false;
dengyihao's avatar
dengyihao 已提交
1273
  // if (pCtx->retryCnt == 0) return false;
dengyihao's avatar
dengyihao 已提交
1274 1275 1276
  if (transEpSetIsEqual(&pCtx->epSet, &pCtx->origEpSet)) return false;
  return true;
}
dengyihao's avatar
dengyihao 已提交
1277
FORCE_INLINE int32_t cliBuildExceptResp(SCliMsg* pMsg, STransMsg* pResp) {
dengyihao's avatar
dengyihao 已提交
1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288
  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 已提交
1289 1290 1291 1292 1293
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);
1294 1295 1296 1297
    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 已提交
1298 1299
    }

dengyihao's avatar
dengyihao 已提交
1300 1301 1302 1303 1304 1305 1306 1307 1308 1309
    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 已提交
1310

dengyihao's avatar
dengyihao 已提交
1311
static int32_t cliPreCheckSessionLimit(SCliThrd* pThrd, char* ip, uint16_t port) {
dengyihao's avatar
dengyihao 已提交
1312 1313
  STrans* pTransInst = pThrd->pTransInst;

dengyihao's avatar
dengyihao 已提交
1314 1315 1316
  // STransConnCtx* pCtx = pMsg->ctx;
  // char*          ip = EPSET_GET_INUSE_IP(&pCtx->epSet);
  // int32_t        port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1317 1318 1319 1320

  char key[TSDB_FQDN_LEN + 64] = {0};
  CONN_CONSTRUCT_HASH_KEY(key, ip, port);

dengyihao's avatar
dengyihao 已提交
1321
  int32_t* val = taosHashGet(pThrd->connLimitCache, key, strlen(key));
dengyihao's avatar
dengyihao 已提交
1322 1323
  if (val == NULL) return 0;

dengyihao's avatar
dengyihao 已提交
1324
  if (*val >= pTransInst->connLimitNum) {
dengyihao's avatar
dengyihao 已提交
1325 1326 1327 1328
    return -1;
  }
  return 0;
}
dengyihao's avatar
dengyihao 已提交
1329
void cliHandleReq(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
formate  
dengyihao 已提交
1330
  STrans*        pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1331
  STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
dengyihao 已提交
1332

dengyihao's avatar
dengyihao 已提交
1333
  cliMayCvtFqdnToIp(&pCtx->epSet, &pThrd->cvtAddr);
1334
  STraceId* trace = &pMsg->msg.info.traceId;
dengyihao's avatar
dengyihao 已提交
1335 1336
  char*     ip = EPSET_GET_INUSE_IP(&pCtx->epSet);
  uint16_t  port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1337

dengyihao's avatar
dengyihao 已提交
1338
  if (!EPSET_IS_VALID(&pCtx->epSet)) {
1339
    tGError("%s, msg %s sent with invalid epset", pTransInst->label, TMSG_INFO(pMsg->msg.msgType));
dengyihao's avatar
dengyihao 已提交
1340
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1341 1342
    return;
  }
dengyihao's avatar
dengyihao 已提交
1343

dengyihao's avatar
dengyihao 已提交
1344
  if (REQUEST_NO_RESP(&pMsg->msg) && (pTransInst->failFastFp != NULL && pTransInst->failFastFp(pMsg->msg.msgType))) {
dengyihao's avatar
dengyihao 已提交
1345
    char key[TSDB_FQDN_LEN + 64] = {0};
dengyihao's avatar
dengyihao 已提交
1346 1347 1348 1349
    CONN_CONSTRUCT_HASH_KEY(key, ip, port);

    SFailFastItem* item = taosHashGet(pThrd->failFastCache, key, strlen(key));
    if (item != NULL) {
dengyihao's avatar
dengyihao 已提交
1350 1351
      int32_t elapse = (int32_t)(taosGetTimestampMs() - item->timestamp);
      if (item->count >= pTransInst->failFastThreshold && (elapse >= 0 && elapse <= pTransInst->failFastInterval)) {
dengyihao's avatar
dengyihao 已提交
1352 1353
        tGTrace("%s, msg %s cancel to send, reason: failed to connect %s:%d: count: %d, at %d", pTransInst->label,
                TMSG_INFO(pMsg->msg.msgType), ip, port, item->count, elapse);
dengyihao's avatar
dengyihao 已提交
1354 1355 1356 1357 1358 1359
        destroyCmsg(pMsg);
        return;
      }
    }
  }

dengyihao's avatar
dengyihao 已提交
1360 1361 1362
  bool      ignore = false;
  SCliConn* conn = cliGetConn(pMsg, pThrd, &ignore);
  if (ignore == true) {
dengyihao's avatar
dengyihao 已提交
1363
    // persist conn already release by server
dengyihao's avatar
dengyihao 已提交
1364 1365
    STransMsg resp;
    cliBuildExceptResp(pMsg, &resp);
dengyihao's avatar
dengyihao 已提交
1366 1367 1368
    if (pMsg->type != Release) {
      pTransInst->cfp(pTransInst->parent, &resp, NULL);
    }
dengyihao's avatar
dengyihao 已提交
1369
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1370 1371
    return;
  }
dengyihao's avatar
dengyihao 已提交
1372

dengyihao's avatar
dengyihao 已提交
1373
  if (conn == NULL && REQUEST_NO_RESP(&pMsg->msg) && 0 != cliPreCheckSessionLimit(pThrd, ip, port)) {
dengyihao's avatar
dengyihao 已提交
1374 1375 1376 1377 1378 1379
    tGTrace("%s, msg %s cancel to send, reason: %s", pTransInst->label, TMSG_INFO(pMsg->msg.msgType),
            tstrerror(TSDB_CODE_RPC_MAX_SESSIONS));
    destroyCmsg(pMsg);
    return;
  }

dengyihao's avatar
dengyihao 已提交
1380
  if (conn != NULL) {
dengyihao's avatar
dengyihao 已提交
1381
    transCtxMerge(&conn->ctx, &pCtx->appCtx);
dengyihao's avatar
dengyihao 已提交
1382
    transQueuePush(&conn->cliMsgs, pMsg);
U
ubuntu 已提交
1383
    cliSend(conn);
dengyihao's avatar
dengyihao 已提交
1384
  } else {
U
ubuntu 已提交
1385
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
1386 1387 1388 1389

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

dengyihao's avatar
dengyihao 已提交
1390
    transCtxMerge(&conn->ctx, &pCtx->appCtx);
dengyihao's avatar
dengyihao 已提交
1391
    transQueuePush(&conn->cliMsgs, pMsg);
dengyihao's avatar
dengyihao 已提交
1392

dengyihao's avatar
dengyihao 已提交
1393
    char     key[TSDB_FQDN_LEN + 64] = {0};
dengyihao's avatar
dengyihao 已提交
1394
    char*    fqdn = EPSET_GET_INUSE_IP(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1395
    uint16_t port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1396
    CONN_CONSTRUCT_HASH_KEY(key, fqdn, port);
dengyihao's avatar
dengyihao 已提交
1397 1398

    conn->ip = strdup(key);
dengyihao's avatar
dengyihao 已提交
1399

dengyihao's avatar
dengyihao 已提交
1400
    uint32_t ipaddr = cliGetIpFromFqdnCache(pThrd->fqdn2ipCache, fqdn);
1401 1402 1403 1404 1405 1406 1407 1408 1409
    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 已提交
1410

dengyihao's avatar
dengyihao 已提交
1411
    struct sockaddr_in addr;
1412
    addr.sin_family = AF_INET;
1413
    addr.sin_addr.s_addr = ipaddr;
dengyihao's avatar
dengyihao 已提交
1414
    addr.sin_port = (uint16_t)htons(port);
dengyihao's avatar
dengyihao 已提交
1415

dengyihao's avatar
dengyihao 已提交
1416
    tGTrace("%s conn %p try to connect to %s", pTransInst->label, conn, conn->ip);
1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436
    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 已提交
1437

1438
    ret = uv_tcp_connect(&conn->connReq, (uv_tcp_t*)(conn->stream), (const struct sockaddr*)&addr, cliConnCb);
dengyihao's avatar
dengyihao 已提交
1439
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1440 1441 1442 1443 1444
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

dengyihao's avatar
dengyihao 已提交
1445
      cliHandleFastFail(conn, ret);
dengyihao's avatar
dengyihao 已提交
1446 1447
      return;
    }
dengyihao's avatar
dengyihao 已提交
1448
    uv_timer_start(conn->timer, cliConnTimeout, TRANS_CONN_TIMEOUT, 0);
dengyihao's avatar
dengyihao 已提交
1449
  }
dengyihao's avatar
dengyihao 已提交
1450
  tGTrace("%s conn %p ready", pTransInst->label, conn);
dengyihao's avatar
dengyihao 已提交
1451
}
dengyihao's avatar
dengyihao 已提交
1452

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

dengyihao's avatar
dengyihao 已提交
1456 1457
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
dengyihao's avatar
dengyihao 已提交
1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469
    QUEUE_REMOVE(h);

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);

    count++;
  }
  if (count >= 2) {
    tTrace("cli process batch size:%d", count);
  }
}

dengyihao's avatar
dengyihao 已提交
1470
static void cliBatchDealReq(queue* wq, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1471
  int count = 0;
dengyihao's avatar
dengyihao 已提交
1472 1473
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
dengyihao's avatar
dengyihao 已提交
1474
    QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
1475 1476

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
1477
    if (pMsg->type == Normal && REQUEST_NO_RESP(&pMsg->msg)) {
dengyihao's avatar
dengyihao 已提交
1478 1479 1480 1481 1482 1483 1484
      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 已提交
1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496
      // 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);
        pBatchList->connMax = 200;
        pBatchList->connCnt = 0;
        pBatchList->batchLenLimit = 16 * 1024;
        pBatchList->ip = strdup(ip);
        pBatchList->dst = strdup(key);
        pBatchList->port = port;

dengyihao's avatar
dengyihao 已提交
1497 1498
        SCliBatch* pBatch = taosMemoryCalloc(1, sizeof(SCliBatch));
        QUEUE_INIT(&pBatch->wq);
dengyihao's avatar
dengyihao 已提交
1499 1500
        QUEUE_INIT(&pBatch->listq);

dengyihao's avatar
dengyihao 已提交
1501 1502 1503
        QUEUE_PUSH(&pBatch->wq, h);
        pBatch->wLen += 1;
        pBatch->batchSize += pMsg->msg.contLen;
dengyihao's avatar
dengyihao 已提交
1504
        pBatch->pList = pBatchList;
dengyihao's avatar
dengyihao 已提交
1505

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

dengyihao's avatar
dengyihao 已提交
1508
        taosHashPut(pThrd->batchCache, key, sizeof(key), &pBatchList, sizeof(void*));
dengyihao's avatar
dengyihao 已提交
1509
      } else {
dengyihao's avatar
dengyihao 已提交
1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543
        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;
        } 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 已提交
1544
      }
dengyihao's avatar
dengyihao 已提交
1545
      continue;
dengyihao's avatar
dengyihao 已提交
1546
    }
dengyihao's avatar
dengyihao 已提交
1547
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
1548
    count++;
dengyihao's avatar
dengyihao 已提交
1549
  }
dengyihao's avatar
dengyihao 已提交
1550

dengyihao's avatar
dengyihao 已提交
1551
  void** pIter = taosHashIterate(pThrd->batchCache, NULL);
dengyihao's avatar
dengyihao 已提交
1552
  while (pIter != NULL) {
dengyihao's avatar
dengyihao 已提交
1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
    SCliBatchList* batchList = (SCliBatchList*)(*pIter);
    if (QUEUE_IS_EMPTY(&batchList->wq) || batchList->connCnt >= batchList->connMax) {
      continue;
    }
    queue* hr = QUEUE_HEAD(&batchList->wq);
    QUEUE_REMOVE(hr);

    batchList->len -= 1;

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

dengyihao's avatar
dengyihao 已提交
1564 1565
    cliHandleBatchReq(batch, pThrd);
    pIter = (void**)taosHashIterate(pThrd->batchCache, pIter);
dengyihao's avatar
dengyihao 已提交
1566 1567
  }

dengyihao's avatar
dengyihao 已提交
1568
  if (count >= 2) {
S
Shengliang Guan 已提交
1569
    tTrace("cli process batch size:%d", count);
dengyihao's avatar
dengyihao 已提交
1570
  }
dengyihao's avatar
dengyihao 已提交
1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584
}

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

  SCliMsg* pMsg = NULL;
  // 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 已提交
1585
  int8_t supportBatch = pTransInst->supportBatch;
dengyihao's avatar
dengyihao 已提交
1586
  if (supportBatch == 0) {
dengyihao's avatar
dengyihao 已提交
1587
    cliNoBatchDealReq(&wq, pThrd);
dengyihao's avatar
dengyihao 已提交
1588
  } else if (supportBatch == 1) {
dengyihao's avatar
dengyihao 已提交
1589
    cliBatchDealReq(&wq, pThrd);
dengyihao's avatar
dengyihao 已提交
1590
  }
dengyihao's avatar
dengyihao 已提交
1591

dengyihao's avatar
dengyihao 已提交
1592
  if (pThrd->stopMsg != NULL) cliHandleQuit(pThrd->stopMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
1593
}
dengyihao's avatar
dengyihao 已提交
1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
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 已提交
1620
}
dengyihao's avatar
dengyihao 已提交
1621

dengyihao's avatar
dengyihao 已提交
1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650
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 已提交
1651 1652 1653
bool cliRecvReleaseReq(SCliConn* conn, STransMsgHead* pHead) {
  if (pHead->release == 1 && (pHead->msgLen) == sizeof(*pHead)) {
    uint64_t ahandle = pHead->ahandle;
dengyihao's avatar
dengyihao 已提交
1654
    tDebug("ahandle = %" PRIu64 "", ahandle);
dengyihao's avatar
dengyihao 已提交
1655 1656
    SCliMsg* pMsg = NULL;
    CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle);
dengyihao's avatar
dengyihao 已提交
1657

dengyihao's avatar
dengyihao 已提交
1658 1659
    transClearBuffer(&conn->readBuf);
    transFreeMsg(transContFromHead((char*)pHead));
dengyihao's avatar
dengyihao 已提交
1660 1661 1662 1663

    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 已提交
1664
        ASSERTS(pMsg == NULL, "trans-cli recv invaid release-req");
dengyihao's avatar
dengyihao 已提交
1665 1666
        return true;
      }
dengyihao's avatar
dengyihao 已提交
1667
    }
dengyihao's avatar
dengyihao 已提交
1668 1669 1670

    cliIteraConnMsgs(conn);

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

dengyihao's avatar
dengyihao 已提交
1674 1675 1676 1677 1678 1679
    addConnToPool(((SCliThrd*)conn->hostThrd)->pool, conn);
    return true;
  }
  return false;
}

U
ubuntu 已提交
1680
static void* cliWorkThread(void* arg) {
dengyihao's avatar
dengyihao 已提交
1681
  SCliThrd* pThrd = (SCliThrd*)arg;
dengyihao's avatar
dengyihao 已提交
1682
  pThrd->pid = taosGetSelfPthreadId();
G
gccgdb1234 已提交
1683
  setThreadName("trans-cli-work");
dengyihao's avatar
dengyihao 已提交
1684
  uv_run(pThrd->loop, UV_RUN_DEFAULT);
dengyihao's avatar
dengyihao 已提交
1685 1686

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

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

dengyihao's avatar
dengyihao 已提交
1693
  STrans* pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1694
  memcpy(cli->label, label, TSDB_LABEL_LEN);
dengyihao's avatar
dengyihao 已提交
1695
  cli->numOfThreads = numOfThreads;
dengyihao's avatar
dengyihao 已提交
1696
  cli->pThreadObj = (SCliThrd**)taosMemoryCalloc(cli->numOfThreads, sizeof(SCliThrd*));
dengyihao's avatar
dengyihao 已提交
1697 1698

  for (int i = 0; i < cli->numOfThreads; i++) {
dengyihao's avatar
dengyihao 已提交
1699
    SCliThrd* pThrd = createThrdObj(shandle);
dengyihao's avatar
dengyihao 已提交
1700 1701 1702 1703 1704
    if (pThrd == NULL) {
      return NULL;
    }

    int err = taosThreadCreate(&pThrd->thread, NULL, cliWorkThread, (void*)(pThrd));
dengyihao's avatar
dengyihao 已提交
1705
    if (err == 0) {
S
Shengliang Guan 已提交
1706
      tDebug("success to create tranport-cli thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1707
    }
dengyihao's avatar
dengyihao 已提交
1708
    cli->pThreadObj[i] = pThrd;
dengyihao's avatar
dengyihao 已提交
1709
  }
dengyihao's avatar
dengyihao 已提交
1710

dengyihao's avatar
dengyihao 已提交
1711 1712
  return cli;
}
dengyihao's avatar
dengyihao 已提交
1713

dengyihao's avatar
dengyihao 已提交
1714
static FORCE_INLINE void destroyUserdata(STransMsg* userdata) {
dengyihao's avatar
dengyihao 已提交
1715 1716 1717 1718 1719 1720
  if (userdata->pCont == NULL) {
    return;
  }
  transFreeMsg(userdata->pCont);
  userdata->pCont = NULL;
}
dengyihao's avatar
dengyihao 已提交
1721

dengyihao's avatar
dengyihao 已提交
1722
static FORCE_INLINE void destroyCmsg(void* arg) {
dengyihao's avatar
dengyihao 已提交
1723
  SCliMsg* pMsg = arg;
dengyihao's avatar
dengyihao 已提交
1724 1725 1726
  if (pMsg == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1727

dengyihao's avatar
dengyihao 已提交
1728 1729
  transDestroyConnCtx(pMsg->ctx);
  destroyUserdata(&pMsg->msg);
wafwerar's avatar
wafwerar 已提交
1730
  taosMemoryFree(pMsg);
dengyihao's avatar
dengyihao 已提交
1731
}
dengyihao's avatar
dengyihao 已提交
1732

dengyihao's avatar
dengyihao 已提交
1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751
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 已提交
1752 1753 1754
static SCliThrd* createThrdObj(void* trans) {
  STrans* pTransInst = trans;

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

dengyihao's avatar
dengyihao 已提交
1757
  QUEUE_INIT(&pThrd->msg);
wafwerar's avatar
wafwerar 已提交
1758
  taosThreadMutexInit(&pThrd->msgMtx, NULL);
dengyihao's avatar
dengyihao 已提交
1759

wafwerar's avatar
wafwerar 已提交
1760
  pThrd->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
1761 1762 1763 1764 1765 1766 1767 1768
  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 已提交
1769 1770 1771 1772 1773
  if (pTransInst->supportBatch) {
    pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 4, pThrd, cliAsyncCb);
  } else {
    pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 8, pThrd, cliAsyncCb);
  }
dengyihao's avatar
dengyihao 已提交
1774
  if (pThrd->asyncPool == NULL) {
dengyihao's avatar
ref log  
dengyihao 已提交
1775
    tError("failed to init async pool");
dengyihao's avatar
dengyihao 已提交
1776 1777 1778 1779 1780 1781
    uv_loop_close(pThrd->loop);
    taosMemoryFree(pThrd->loop);
    taosThreadMutexDestroy(&pThrd->msgMtx);
    taosMemoryFree(pThrd);
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
1782 1783 1784 1785

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

dengyihao's avatar
dengyihao 已提交
1788
  int32_t timerSize = 64;
dengyihao's avatar
dengyihao 已提交
1789 1790 1791 1792 1793 1794 1795
  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 已提交
1796
  pThrd->pool = createConnPool(4);
dengyihao's avatar
dengyihao 已提交
1797
  transDQCreate(pThrd->loop, &pThrd->delayQueue);
dengyihao's avatar
dengyihao 已提交
1798

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

dengyihao's avatar
dengyihao 已提交
1801 1802 1803
  pThrd->nextTimeout = taosGetTimestampMs() + CONN_PERSIST_TIME(pTransInst->idleTime);
  pThrd->pTransInst = trans;

dengyihao's avatar
dengyihao 已提交
1804
  pThrd->destroyAhandleFp = pTransInst->destroyFp;
dengyihao's avatar
dengyihao 已提交
1805
  pThrd->fqdn2ipCache = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
1806
  pThrd->failFastCache = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
1807 1808
  pThrd->connLimitCache = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true,
                                       pTransInst->connLimitLock == 0 ? HASH_NO_LOCK : HASH_ENTRY_LOCK);
dengyihao's avatar
dengyihao 已提交
1809

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

dengyihao's avatar
dengyihao 已提交
1812
  pThrd->quit = false;
dengyihao's avatar
dengyihao 已提交
1813 1814
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
1815
static void destroyThrdObj(SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1816 1817 1818
  if (pThrd == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1819

wafwerar's avatar
wafwerar 已提交
1820
  taosThreadJoin(pThrd->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1821
  CLI_RELEASE_UV(pThrd->loop);
wafwerar's avatar
wafwerar 已提交
1822
  taosThreadMutexDestroy(&pThrd->msgMtx);
dengyihao's avatar
dengyihao 已提交
1823
  TRANS_DESTROY_ASYNC_POOL_MSG(pThrd->asyncPool, SCliMsg, destroyCmsg);
dengyihao's avatar
dengyihao 已提交
1824
  transAsyncPoolDestroy(pThrd->asyncPool);
dengyihao's avatar
dengyihao 已提交
1825

dengyihao's avatar
dengyihao 已提交
1826
  transDQDestroy(pThrd->delayQueue, destroyCmsgAndAhandle);
dengyihao's avatar
dengyihao 已提交
1827
  transDQDestroy(pThrd->timeoutQueue, NULL);
dengyihao's avatar
dengyihao 已提交
1828

dengyihao's avatar
dengyihao 已提交
1829
  tDebug("thread destroy %" PRId64, pThrd->pid);
dengyihao's avatar
dengyihao 已提交
1830 1831 1832 1833 1834
  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 已提交
1835
  taosMemoryFree(pThrd->prepare);
wafwerar's avatar
wafwerar 已提交
1836
  taosMemoryFree(pThrd->loop);
dengyihao's avatar
dengyihao 已提交
1837
  taosHashCleanup(pThrd->fqdn2ipCache);
dengyihao's avatar
dengyihao 已提交
1838
  taosHashCleanup(pThrd->failFastCache);
dengyihao's avatar
dengyihao 已提交
1839
  taosHashCleanup(pThrd->connLimitCache);
dengyihao's avatar
dengyihao 已提交
1840 1841 1842

  void** pIter = taosHashIterate(pThrd->batchCache, NULL);
  while (pIter != NULL) {
dengyihao's avatar
dengyihao 已提交
1843 1844 1845 1846 1847 1848 1849 1850 1851
    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);
    }
    taosMemoryFree(pBatchList);
dengyihao's avatar
dengyihao 已提交
1852 1853
    pIter = (void**)taosHashIterate(pThrd->batchCache, pIter);
  }
dengyihao's avatar
dengyihao 已提交
1854
  taosHashCleanup(pThrd->batchCache);
wafwerar's avatar
wafwerar 已提交
1855
  taosMemoryFree(pThrd);
dengyihao's avatar
dengyihao 已提交
1856
}
dengyihao's avatar
dengyihao 已提交
1857

dengyihao's avatar
dengyihao 已提交
1858
static FORCE_INLINE void transDestroyConnCtx(STransConnCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
1859
  //
wafwerar's avatar
wafwerar 已提交
1860
  taosMemoryFree(ctx);
dengyihao's avatar
dengyihao 已提交
1861
}
dengyihao's avatar
dengyihao 已提交
1862

dengyihao's avatar
dengyihao 已提交
1863
void cliSendQuit(SCliThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1864
  // cli can stop gracefully
wafwerar's avatar
wafwerar 已提交
1865
  SCliMsg* msg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
1866
  msg->type = Quit;
dengyihao's avatar
dengyihao 已提交
1867
  transAsyncSend(thrd->asyncPool, &msg->q);
dengyihao's avatar
dengyihao 已提交
1868
  atomic_store_8(&thrd->asyncPool->stop, 1);
dengyihao's avatar
dengyihao 已提交
1869
}
dengyihao's avatar
dengyihao 已提交
1870 1871
void cliWalkCb(uv_handle_t* handle, void* arg) {
  if (!uv_is_closing(handle)) {
dengyihao's avatar
dengyihao 已提交
1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883
    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 已提交
1884
    uv_close(handle, cliDestroy);
dengyihao's avatar
dengyihao 已提交
1885 1886
  }
}
dengyihao's avatar
dengyihao 已提交
1887

dengyihao's avatar
dengyihao 已提交
1888
FORCE_INLINE int cliRBChoseIdx(STrans* pTransInst) {
dengyihao's avatar
dengyihao 已提交
1889
  int32_t index = pTransInst->index;
dengyihao's avatar
dengyihao 已提交
1890 1891 1892
  if (pTransInst->numOfThreads == 0) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
1893 1894 1895 1896
  /*
   * no lock, and to avoid CPU load imbalance, set limit pTransInst->numOfThreads * 2000;
   */
  if (pTransInst->index++ >= pTransInst->numOfThreads * 2000) {
U
ubuntu 已提交
1897 1898 1899 1900
    pTransInst->index = 0;
  }
  return index % pTransInst->numOfThreads;
}
dengyihao's avatar
dengyihao 已提交
1901
static FORCE_INLINE void doDelayTask(void* param) {
dengyihao's avatar
dengyihao 已提交
1902
  STaskArg* arg = param;
dengyihao's avatar
dengyihao 已提交
1903
  cliHandleReq((SCliMsg*)arg->param1, (SCliThrd*)arg->param2);
dengyihao's avatar
dengyihao 已提交
1904 1905
  taosMemoryFree(arg);
}
dengyihao's avatar
dengyihao 已提交
1906

dengyihao's avatar
dengyihao 已提交
1907 1908 1909
static void doCloseIdleConn(void* param) {
  STaskArg* arg = param;
  SCliConn* conn = arg->param1;
dengyihao's avatar
dengyihao 已提交
1910
  tDebug("%s conn %p idle, close it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
1911
  conn->task = NULL;
dengyihao's avatar
dengyihao 已提交
1912 1913
  cliDestroyConn(conn, true);
  taosMemoryFree(arg);
dengyihao's avatar
dengyihao 已提交
1914 1915
}

dengyihao's avatar
dengyihao 已提交
1916
static void cliSchedMsgToNextNode(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1917
  STrans*        pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1918 1919
  STransConnCtx* pCtx = pMsg->ctx;

dengyihao's avatar
dengyihao 已提交
1920 1921
  STraceId* trace = &pMsg->msg.info.traceId;
  char      tbuf[256] = {0};
dengyihao's avatar
dengyihao 已提交
1922
  EPSET_DEBUG_STR(&pCtx->epSet, tbuf);
dengyihao's avatar
dengyihao 已提交
1923 1924
  tGDebug("%s retry on next node,use:%s, step: %d,timeout:%" PRId64 "", transLabel(pThrd->pTransInst), tbuf,
          pCtx->retryStep, pCtx->retryNextInterval);
dengyihao's avatar
dengyihao 已提交
1925 1926 1927 1928

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

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

dengyihao's avatar
dengyihao 已提交
1933
FORCE_INLINE void cliCompareAndSwap(int8_t* val, int8_t exp, int8_t newVal) {
dengyihao's avatar
dengyihao 已提交
1934 1935 1936 1937
  if (*val != exp) {
    *val = newVal;
  }
}
dengyihao's avatar
dengyihao 已提交
1938

dengyihao's avatar
dengyihao 已提交
1939
FORCE_INLINE bool cliTryExtractEpSet(STransMsg* pResp, SEpSet* dst) {
dengyihao's avatar
dengyihao 已提交
1940 1941 1942 1943 1944 1945
  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 已提交
1946 1947 1948 1949
    return false;
  }
  int32_t tlen = tSerializeSEpSet(NULL, 0, dst);

dengyihao's avatar
dengyihao 已提交
1950 1951 1952 1953 1954 1955 1956
  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 已提交
1957 1958

  pResp->pCont = buf;
dengyihao's avatar
dengyihao 已提交
1959 1960 1961
  pResp->contLen = len;

  *dst = epset;
dengyihao's avatar
dengyihao 已提交
1962 1963
  return true;
}
dengyihao's avatar
dengyihao 已提交
1964
bool cliResetEpset(STransConnCtx* pCtx, STransMsg* pResp, bool hasEpSet) {
dengyihao's avatar
dengyihao 已提交
1965
  bool noDelay = true;
dengyihao's avatar
dengyihao 已提交
1966 1967 1968 1969 1970 1971 1972
  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 已提交
1973 1974 1975 1976 1977 1978 1979 1980 1981 1982
    } 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 已提交
1983
      } else {
dengyihao's avatar
dengyihao 已提交
1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995
        if (!transEpSetIsEqual(&pCtx->epSet, &epSet)) {
          tDebug("epset not equal, retry new epset");
          pCtx->epSet = epSet;
          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 已提交
1996
      }
dengyihao's avatar
dengyihao 已提交
1997 1998
    }
  } else {
dengyihao's avatar
dengyihao 已提交
1999
    SEpSet  epSet;
dengyihao's avatar
dengyihao 已提交
2000 2001
    int32_t valid = tDeserializeSEpSet(pResp->pCont, pResp->contLen, &epSet);
    if (valid < 0) {
dengyihao's avatar
dengyihao 已提交
2002 2003 2004 2005 2006 2007
      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 已提交
2008
    } else {
dengyihao's avatar
dengyihao 已提交
2009 2010 2011
      if (!transEpSetIsEqual(&pCtx->epSet, &epSet)) {
        tDebug("epset not equal, retry new epset");
        pCtx->epSet = epSet;
dengyihao's avatar
dengyihao 已提交
2012
        noDelay = false;
dengyihao's avatar
dengyihao 已提交
2013
      } else {
dengyihao's avatar
dengyihao 已提交
2014 2015 2016 2017 2018 2019
        if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
          noDelay = false;
        } else {
          tDebug("epset equal, continue");
          EPSET_FORWARD_INUSE(&pCtx->epSet);
        }
dengyihao's avatar
dengyihao 已提交
2020
      }
dengyihao's avatar
dengyihao 已提交
2021 2022 2023 2024 2025
    }
  }
  return noDelay;
}
bool cliGenRetryRule(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
2026 2027
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2028

dengyihao's avatar
dengyihao 已提交
2029 2030
  STransConnCtx* pCtx = pMsg->ctx;
  int32_t        code = pResp->code;
dengyihao's avatar
dengyihao 已提交
2031

dengyihao's avatar
dengyihao 已提交
2032
  bool retry = pTransInst->retry != NULL ? pTransInst->retry(code, pResp->msgType - 1) : false;
dengyihao's avatar
dengyihao 已提交
2033 2034 2035
  if (retry == false) {
    return false;
  }
dengyihao's avatar
dengyihao 已提交
2036 2037 2038 2039 2040 2041 2042 2043

  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 已提交
2044
    pCtx->retryStep = 0;
dengyihao's avatar
dengyihao 已提交
2045
    pCtx->retryInit = true;
dengyihao's avatar
dengyihao 已提交
2046
    pCtx->retryCode = TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
2047 2048 2049

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

dengyihao's avatar
dengyihao 已提交
2052 2053
  if (-1 != pCtx->retryMaxTimeout && taosGetTimestampMs() - pCtx->retryInitTimestamp >= pCtx->retryMaxTimeout) {
    return false;
dengyihao's avatar
dengyihao 已提交
2054
  }
dengyihao's avatar
dengyihao 已提交
2055

dengyihao's avatar
dengyihao 已提交
2056 2057 2058 2059 2060 2061
  // code, msgType

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

dengyihao's avatar
dengyihao 已提交
2062 2063
  bool noDelay = false;
  if (code == TSDB_CODE_RPC_BROKEN_LINK || code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
dengyihao's avatar
dengyihao 已提交
2064
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2065
    noDelay = cliResetEpset(pCtx, pResp, false);
dengyihao's avatar
dengyihao 已提交
2066 2067
    transFreeMsg(pResp->pCont);
    transUnrefCliHandle(pConn);
dengyihao's avatar
dengyihao 已提交
2068
  } else if (code == TSDB_CODE_SYN_NOT_LEADER || code == TSDB_CODE_SYN_INTERNAL_ERROR ||
2069 2070
             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 已提交
2071
             code == TSDB_CODE_APP_IS_STOPPING || code == TSDB_CODE_VND_STOPPED) {
dengyihao's avatar
dengyihao 已提交
2072
    tTrace("code str %s, contlen:%d 1", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2073
    noDelay = cliResetEpset(pCtx, pResp, true);
dengyihao's avatar
dengyihao 已提交
2074 2075
    transFreeMsg(pResp->pCont);
    addConnToPool(pThrd->pool, pConn);
dengyihao's avatar
dengyihao 已提交
2076
  } else if (code == TSDB_CODE_SYN_RESTORING) {
dengyihao's avatar
dengyihao 已提交
2077
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2078
    noDelay = cliResetEpset(pCtx, pResp, true);
dengyihao's avatar
dengyihao 已提交
2079 2080
    addConnToPool(pThrd->pool, pConn);
    transFreeMsg(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2081
  } else {
dengyihao's avatar
dengyihao 已提交
2082
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2083 2084 2085
    noDelay = cliResetEpset(pCtx, pResp, false);
    addConnToPool(pThrd->pool, pConn);
    transFreeMsg(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2086
  }
dengyihao's avatar
dengyihao 已提交
2087 2088 2089 2090
  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 已提交
2091

dengyihao's avatar
dengyihao 已提交
2092 2093 2094
  if (noDelay == false) {
    pCtx->epsetRetryCnt = 1;
    pCtx->retryStep++;
dengyihao's avatar
dengyihao 已提交
2095

dengyihao's avatar
dengyihao 已提交
2096 2097 2098 2099 2100 2101
    int64_t factor = pow(pCtx->retryStepFactor, pCtx->retryStep - 1);
    pCtx->retryNextInterval = factor * pCtx->retryMinInterval;
    if (pCtx->retryNextInterval >= pCtx->retryMaxInterval) {
      pCtx->retryNextInterval = pCtx->retryMaxInterval;
    }

dengyihao's avatar
dengyihao 已提交
2102 2103 2104
    // if (-1 != pCtx->retryMaxTimeout && taosGetTimestampMs() - pCtx->retryInitTimestamp >= pCtx->retryMaxTimeout) {
    //   return false;
    // }
dengyihao's avatar
dengyihao 已提交
2105 2106 2107
  } else {
    pCtx->retryNextInterval = 0;
    pCtx->epsetRetryCnt++;
dengyihao's avatar
dengyihao 已提交
2108
  }
dengyihao's avatar
dengyihao 已提交
2109

dengyihao's avatar
dengyihao 已提交
2110
  pMsg->sent = 0;
dengyihao's avatar
dengyihao 已提交
2111
  cliSchedMsgToNextNode(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
2112
  return true;
dengyihao's avatar
dengyihao 已提交
2113
}
dengyihao's avatar
dengyihao 已提交
2114
int cliAppCb(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
2115 2116
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2117

dengyihao's avatar
dengyihao 已提交
2118
  if (pMsg == NULL || pMsg->ctx == NULL) {
dengyihao's avatar
dengyihao 已提交
2119
    tTrace("%s conn %p handle resp", pTransInst->label, pConn);
dengyihao's avatar
dengyihao 已提交
2120 2121 2122
    pTransInst->cfp(pTransInst->parent, pResp, NULL);
    return 0;
  }
dengyihao's avatar
dengyihao 已提交
2123

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

dengyihao's avatar
dengyihao 已提交
2126 2127 2128
  bool retry = cliGenRetryRule(pConn, pResp, pMsg);
  if (retry == true) {
    return -1;
dengyihao's avatar
dengyihao 已提交
2129
  }
dengyihao's avatar
dengyihao 已提交
2130

dengyihao's avatar
dengyihao 已提交
2131 2132 2133
  if (pCtx->retryCode != TSDB_CODE_SUCCESS) {
    int32_t code = pResp->code;
    // return internal code app
dengyihao's avatar
dengyihao 已提交
2134 2135
    if (code == TSDB_CODE_RPC_NETWORK_UNAVAIL || code == TSDB_CODE_RPC_BROKEN_LINK ||
        code == TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED) {
dengyihao's avatar
dengyihao 已提交
2136 2137 2138 2139
      pResp->code = pCtx->retryCode;
    }
  }

2140
  // check whole vnodes is offline on this vgroup
2141 2142
  if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps || pCtx->retryStep > 0) {
    if (pResp->code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
A
Alex Duan 已提交
2143
      pResp->code = TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED;
2144
    } else if (pResp->code == TSDB_CODE_RPC_BROKEN_LINK) {
A
Alex Duan 已提交
2145
      pResp->code = TSDB_CODE_RPC_SOMENODE_BROKEN_LINK;
2146 2147 2148
    }
  }

dengyihao's avatar
dengyihao 已提交
2149 2150
  STraceId* trace = &pResp->info.traceId;
  bool      hasEpSet = cliTryExtractEpSet(pResp, &pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
2151
  if (hasEpSet) {
dengyihao's avatar
dengyihao 已提交
2152 2153
    char tbuf[256] = {0};
    EPSET_DEBUG_STR(&pCtx->epSet, tbuf);
dengyihao's avatar
dengyihao 已提交
2154
    tGTrace("%s conn %p extract epset from msg", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
2155
  }
dengyihao's avatar
dengyihao 已提交
2156

dengyihao's avatar
dengyihao 已提交
2157
  if (pCtx->pSem != NULL) {
dengyihao's avatar
dengyihao 已提交
2158
    tGTrace("%s conn %p(sync) handle resp", CONN_GET_INST_LABEL(pConn), pConn);
2159
    if (pCtx->pRsp == NULL) {
dengyihao's avatar
dengyihao 已提交
2160
      tGTrace("%s conn %p(sync) failed to resp, ignore", CONN_GET_INST_LABEL(pConn), pConn);
2161 2162 2163
    } else {
      memcpy((char*)pCtx->pRsp, (char*)pResp, sizeof(*pResp));
    }
dengyihao's avatar
dengyihao 已提交
2164
    tsem_post(pCtx->pSem);
2165
    pCtx->pRsp = NULL;
dengyihao's avatar
dengyihao 已提交
2166
  } else {
dengyihao's avatar
dengyihao 已提交
2167
    tGTrace("%s conn %p handle resp", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
2168
    if (retry == false && hasEpSet == true) {
dengyihao's avatar
dengyihao 已提交
2169
      pTransInst->cfp(pTransInst->parent, pResp, &pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
2170
    } else {
dengyihao's avatar
dengyihao 已提交
2171
      if (!cliIsEpsetUpdated(pResp->code, pCtx)) {
dengyihao's avatar
dengyihao 已提交
2172 2173 2174 2175
        pTransInst->cfp(pTransInst->parent, pResp, NULL);
      } else {
        pTransInst->cfp(pTransInst->parent, pResp, &pCtx->epSet);
      }
dengyihao's avatar
dengyihao 已提交
2176
    }
dengyihao's avatar
dengyihao 已提交
2177
  }
dengyihao's avatar
dengyihao 已提交
2178
  return 0;
dengyihao's avatar
dengyihao 已提交
2179
}
U
ubuntu 已提交
2180 2181

void transCloseClient(void* arg) {
U
ubuntu 已提交
2182
  SCliObj* cli = arg;
dengyihao's avatar
dengyihao 已提交
2183
  for (int i = 0; i < cli->numOfThreads; i++) {
U
ubuntu 已提交
2184
    cliSendQuit(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
2185
    destroyThrdObj(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
2186
  }
wafwerar's avatar
wafwerar 已提交
2187 2188
  taosMemoryFree(cli->pThreadObj);
  taosMemoryFree(cli);
dengyihao's avatar
dengyihao 已提交
2189
}
dengyihao's avatar
dengyihao 已提交
2190 2191 2192 2193 2194
void transRefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_INC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
2195
  tTrace("%s conn %p ref %d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
2196 2197 2198 2199 2200 2201 2202
  UNUSED(ref);
}
void transUnrefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_DEC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
2203
  tTrace("%s conn %p ref:%d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
2204
  if (ref == 0) {
U
ubuntu 已提交
2205
    cliDestroyConn((SCliConn*)handle, true);
dengyihao's avatar
dengyihao 已提交
2206 2207
  }
}
dengyihao's avatar
dengyihao 已提交
2208
static FORCE_INLINE SCliThrd* transGetWorkThrdFromHandle(STrans* trans, int64_t handle) {
dengyihao's avatar
dengyihao 已提交
2209
  SCliThrd*  pThrd = NULL;
dengyihao's avatar
dengyihao 已提交
2210
  SExHandle* exh = transAcquireExHandle(transGetRefMgt(), handle);
dengyihao's avatar
dengyihao 已提交
2211 2212 2213
  if (exh == NULL) {
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
2214

dengyihao's avatar
dengyihao 已提交
2215 2216 2217 2218 2219 2220
  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 已提交
2221
  pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
2222
  transReleaseExHandle(transGetRefMgt(), handle);
dengyihao's avatar
dengyihao 已提交
2223 2224
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
2225
SCliThrd* transGetWorkThrd(STrans* trans, int64_t handle) {
dengyihao's avatar
dengyihao 已提交
2226
  if (handle == 0) {
dengyihao's avatar
dengyihao 已提交
2227
    int idx = cliRBChoseIdx(trans);
dengyihao's avatar
dengyihao 已提交
2228
    if (idx < 0) return NULL;
dengyihao's avatar
dengyihao 已提交
2229 2230
    return ((SCliObj*)trans->tcphandle)->pThreadObj[idx];
  }
dengyihao's avatar
dengyihao 已提交
2231
  SCliThrd* pThrd = transGetWorkThrdFromHandle(trans, handle);
D
dapan1121 已提交
2232
  return pThrd;
dengyihao's avatar
dengyihao 已提交
2233
}
dengyihao's avatar
dengyihao 已提交
2234
int transReleaseCliHandle(void* handle) {
dengyihao's avatar
dengyihao 已提交
2235 2236 2237
  int  idx = -1;
  bool valid = false;

dengyihao's avatar
dengyihao 已提交
2238
  SCliThrd* pThrd = transGetWorkThrdFromHandle(NULL, (int64_t)handle);
dengyihao's avatar
dengyihao 已提交
2239
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2240
    return -1;
dengyihao's avatar
dengyihao 已提交
2241
  }
dengyihao's avatar
dengyihao 已提交
2242

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

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

dengyihao's avatar
dengyihao 已提交
2249
  SCliMsg* cmsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2250
  cmsg->msg = tmsg;
dengyihao's avatar
dengyihao 已提交
2251
  cmsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2252
  cmsg->type = Release;
dengyihao's avatar
dengyihao 已提交
2253
  cmsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
2254

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

dengyihao's avatar
dengyihao 已提交
2258
  if (0 != transAsyncSend(pThrd->asyncPool, &cmsg->q)) {
dengyihao's avatar
dengyihao 已提交
2259
    destroyCmsg(cmsg);
dengyihao's avatar
dengyihao 已提交
2260 2261
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2262
  return 0;
dengyihao's avatar
dengyihao 已提交
2263
}
dengyihao's avatar
dengyihao 已提交
2264

dengyihao's avatar
dengyihao 已提交
2265
int transSendRequest(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
2266 2267 2268 2269 2270
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
  if (pTransInst == NULL) {
    transFreeMsg(pReq->pCont);
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2271

dengyihao's avatar
dengyihao 已提交
2272
  SCliThrd* pThrd = transGetWorkThrd(pTransInst, (int64_t)pReq->info.handle);
dengyihao's avatar
dengyihao 已提交
2273
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2274
    transFreeMsg(pReq->pCont);
dengyihao's avatar
dengyihao 已提交
2275
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2276
    return TSDB_CODE_RPC_BROKEN_LINK;
dengyihao's avatar
dengyihao 已提交
2277
  }
dengyihao's avatar
dengyihao 已提交
2278 2279 2280 2281 2282 2283 2284 2285 2286 2287
  if (pTransInst->connLimitNum > 0 && REQUEST_NO_RESP(pReq)) {
    char     key[TSDB_FQDN_LEN + 64] = {0};
    char*    ip = EPSET_GET_INUSE_IP((SEpSet*)pEpSet);
    uint16_t port = EPSET_GET_INUSE_PORT((SEpSet*)pEpSet);
    CONN_CONSTRUCT_HASH_KEY(key, ip, port);

    int32_t* val = taosHashGet(pThrd->connLimitCache, key, strlen(key));
    if (val != NULL && *val >= pTransInst->connLimitNum) {
      transFreeMsg(pReq->pCont);
      transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2288
      return TSDB_CODE_RPC_MAX_SESSIONS;
dengyihao's avatar
dengyihao 已提交
2289 2290
    }
  }
dengyihao's avatar
dengyihao 已提交
2291

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

wafwerar's avatar
wafwerar 已提交
2294
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2295
  pCtx->epSet = *pEpSet;
S
Shengliang Guan 已提交
2296
  pCtx->ahandle = pReq->info.ahandle;
dengyihao's avatar
dengyihao 已提交
2297
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
2298

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

wafwerar's avatar
wafwerar 已提交
2301
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2302
  cliMsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
2303
  cliMsg->msg = *pReq;
dengyihao's avatar
dengyihao 已提交
2304
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2305
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
2306
  cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2307

dengyihao's avatar
dengyihao 已提交
2308
  STraceId* trace = &pReq->info.traceId;
dengyihao's avatar
dengyihao 已提交
2309 2310
  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 已提交
2311 2312
  if (0 != transAsyncSend(pThrd->asyncPool, &(cliMsg->q))) {
    destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2313
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2314 2315
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2316
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2317
  return 0;
dengyihao's avatar
dengyihao 已提交
2318
}
dengyihao's avatar
dengyihao 已提交
2319

dengyihao's avatar
dengyihao 已提交
2320
int transSendRecv(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransMsg* pRsp) {
dengyihao's avatar
dengyihao 已提交
2321 2322 2323 2324 2325
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
  if (pTransInst == NULL) {
    transFreeMsg(pReq->pCont);
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2326

dengyihao's avatar
dengyihao 已提交
2327 2328
  SCliThrd* pThrd = transGetWorkThrd(pTransInst, (int64_t)pReq->info.handle);
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2329
    transFreeMsg(pReq->pCont);
dengyihao's avatar
dengyihao 已提交
2330
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2331
    return TSDB_CODE_RPC_BROKEN_LINK;
dengyihao's avatar
dengyihao 已提交
2332
  }
dengyihao's avatar
dengyihao 已提交
2333

dengyihao's avatar
dengyihao 已提交
2334 2335
  tsem_t* sem = taosMemoryCalloc(1, sizeof(tsem_t));
  tsem_init(sem, 0, 0);
dengyihao's avatar
dengyihao 已提交
2336

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

wafwerar's avatar
wafwerar 已提交
2339
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2340
  pCtx->epSet = *pEpSet;
dengyihao's avatar
dengyihao 已提交
2341
  pCtx->origEpSet = *pEpSet;
S
Shengliang Guan 已提交
2342
  pCtx->ahandle = pReq->info.ahandle;
dengyihao's avatar
dengyihao 已提交
2343
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
2344
  pCtx->pSem = sem;
dengyihao's avatar
dengyihao 已提交
2345 2346
  pCtx->pRsp = pRsp;

wafwerar's avatar
wafwerar 已提交
2347
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2348 2349 2350
  cliMsg->ctx = pCtx;
  cliMsg->msg = *pReq;
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2351
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
2352
  cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2353

dengyihao's avatar
dengyihao 已提交
2354
  STraceId* trace = &pReq->info.traceId;
dengyihao's avatar
dengyihao 已提交
2355 2356
  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 已提交
2357

dengyihao's avatar
dengyihao 已提交
2358 2359
  int ret = transAsyncSend(pThrd->asyncPool, &cliMsg->q);
  if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
2360
    destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2361
    goto _RETURN;
dengyihao's avatar
dengyihao 已提交
2362
  }
dengyihao's avatar
dengyihao 已提交
2363
  tsem_wait(sem);
dengyihao's avatar
dengyihao 已提交
2364 2365

_RETURN:
dengyihao's avatar
dengyihao 已提交
2366 2367
  tsem_destroy(sem);
  taosMemoryFree(sem);
dengyihao's avatar
dengyihao 已提交
2368
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2369
  return ret;
dengyihao's avatar
dengyihao 已提交
2370
}
dengyihao's avatar
dengyihao 已提交
2371 2372 2373
/*
 *
 **/
dengyihao's avatar
dengyihao 已提交
2374
int transSetDefaultAddr(void* shandle, const char* ip, const char* fqdn) {
dengyihao's avatar
dengyihao 已提交
2375
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2376 2377 2378
  if (pTransInst == NULL) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2379 2380 2381

  SCvtAddr cvtAddr = {0};
  if (ip != NULL && fqdn != NULL) {
dengyihao's avatar
dengyihao 已提交
2382 2383
    tstrncpy(cvtAddr.ip, ip, sizeof(cvtAddr.ip));
    tstrncpy(cvtAddr.fqdn, fqdn, sizeof(cvtAddr.fqdn));
dengyihao's avatar
dengyihao 已提交
2384 2385
    cvtAddr.cvt = true;
  }
dengyihao's avatar
dengyihao 已提交
2386 2387
  for (int i = 0; i < pTransInst->numOfThreads; i++) {
    STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2388
    pCtx->cvtAddr = cvtAddr;
dengyihao's avatar
dengyihao 已提交
2389 2390 2391 2392

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

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

dengyihao's avatar
dengyihao 已提交
2398 2399
    if (transAsyncSend(thrd->asyncPool, &(cliMsg->q)) != 0) {
      destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2400
      transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2401 2402
      return -1;
    }
dengyihao's avatar
dengyihao 已提交
2403
  }
dengyihao's avatar
dengyihao 已提交
2404
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2405
  return 0;
dengyihao's avatar
dengyihao 已提交
2406
}
dengyihao's avatar
dengyihao 已提交
2407 2408 2409 2410 2411

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 已提交
2412

dengyihao's avatar
dengyihao 已提交
2413 2414
  return exh->refId;
}
dengyihao's avatar
dengyihao 已提交
2415
#endif