transCli.c 77.8 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
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "transComm.h"

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

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

  int connMax;
  int connCnt;
  int batchLenLimit;
dengyihao's avatar
dengyihao 已提交
29
  int sending;
dengyihao's avatar
dengyihao 已提交
30

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
static void cliHandleBatchReq(SCliBatch* pBatch, SCliThrd* pThrd);
dengyihao's avatar
dengyihao 已提交
162
static void cliSendBatchCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
163 164

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

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

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

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

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

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

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

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

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

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

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

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

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

dengyihao's avatar
dengyihao 已提交
217 218 219 220 221 222 223 224 225
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 已提交
226 227 228 229 230 231
// 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 已提交
232 233
#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 已提交
234

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

dengyihao's avatar
dengyihao 已提交
608 609 610 611
  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 已提交
612
    conn->list = taosHashGet((SHashObj*)pool, conn->ip, strlen(conn->ip));
dengyihao's avatar
dengyihao 已提交
613 614
  } 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 已提交
615
  }
dengyihao's avatar
dengyihao 已提交
616
  QUEUE_PUSH(&conn->list->conns, &conn->q);
dengyihao's avatar
dengyihao 已提交
617
  conn->list->size += 1;
dengyihao's avatar
dengyihao 已提交
618

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

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

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

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

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

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

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

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

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

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

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

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

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

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

dengyihao's avatar
dengyihao 已提交
758
  if (clear) {
dengyihao's avatar
dengyihao 已提交
759
    if (!uv_is_closing((uv_handle_t*)conn->stream)) {
dengyihao's avatar
dengyihao 已提交
760
      uv_read_stop(conn->stream);
dengyihao's avatar
dengyihao 已提交
761 762
      uv_close((uv_handle_t*)conn->stream, cliDestroy);
    }
763
  }
dengyihao's avatar
dengyihao 已提交
764
}
U
ubuntu 已提交
765
static void cliDestroy(uv_handle_t* handle) {
dengyihao's avatar
dengyihao 已提交
766 767 768
  if (uv_handle_get_type(handle) != UV_TCP || handle->data == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
769
  SCliConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
770
  SCliThrd* pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
771 772
  if (conn->timer != NULL) {
    uv_timer_stop(conn->timer);
dengyihao's avatar
dengyihao 已提交
773 774
    taosArrayPush(pThrd->timerList, &conn->timer);
    conn->timer->data = NULL;
dengyihao's avatar
dengyihao 已提交
775 776
    conn->timer = NULL;
  }
dengyihao's avatar
dengyihao 已提交
777 778 779
  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 已提交
780

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

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

  cliDestroyConnMsgs(conn, true);

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

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

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

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

dengyihao's avatar
dengyihao 已提交
848 849 850 851 852
  SCliBatch*     pBatch = pConn->pBatch;
  SCliBatchList* pList = pBatch->pList;
  pList->connCnt += 1;

  int32_t wLen = pBatch->wLen;
dengyihao's avatar
dengyihao 已提交
853 854 855 856

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

dengyihao's avatar
dengyihao 已提交
992
    SCliMsg* p = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
993 994
    destroyCmsg(p);
  }
dengyihao's avatar
dengyihao 已提交
995 996
  SCliBatchList* p = pBatch->pList;
  p->sending -= 1;
dengyihao's avatar
dengyihao 已提交
997 998
  taosMemoryFree(pBatch);
}
dengyihao's avatar
dengyihao 已提交
999
static void cliHandleBatchReq(SCliBatch* pBatch, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1000 1001 1002 1003 1004
  if (pThrd->quit == true) {
    cliDestroyBatch(pBatch);
    return;
  }

dengyihao's avatar
dengyihao 已提交
1005
  if (pBatch == NULL || pBatch->wLen == 0 || QUEUE_IS_EMPTY(&pBatch->wq)) {
dengyihao's avatar
dengyihao 已提交
1006 1007
    return;
  }
dengyihao's avatar
dengyihao 已提交
1008 1009
  STrans*        pTransInst = pThrd->pTransInst;
  SCliBatchList* pList = pBatch->pList;
dengyihao's avatar
dengyihao 已提交
1010

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

dengyihao's avatar
dengyihao 已提交
1013 1014 1015 1016
  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 已提交
1017 1018
    return;
  }
dengyihao's avatar
dengyihao 已提交
1019 1020
  if (conn == NULL) {
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
1021 1022
    conn->pBatch = pBatch;
    conn->ip = strdup(pList->dst);
dengyihao's avatar
dengyihao 已提交
1023

dengyihao's avatar
dengyihao 已提交
1024
    uint32_t ipaddr = cliGetIpFromFqdnCache(pThrd->fqdn2ipCache, pList->ip);
dengyihao's avatar
dengyihao 已提交
1025 1026 1027 1028 1029 1030
    if (ipaddr == 0xffffffff) {
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

dengyihao's avatar
dengyihao 已提交
1031
      cliHandleFastFail(conn, -1);
dengyihao's avatar
dengyihao 已提交
1032 1033 1034 1035 1036
      return;
    }
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = ipaddr;
dengyihao's avatar
dengyihao 已提交
1037
    addr.sin_port = (uint16_t)htons(pList->port);
dengyihao's avatar
dengyihao 已提交
1038

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

dengyihao's avatar
dengyihao 已提交
1073
  conn->pBatch = pBatch;
dengyihao's avatar
dengyihao 已提交
1074
  cliSendBatch(conn);
dengyihao's avatar
dengyihao 已提交
1075 1076
}
static void cliSendBatchCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
1077 1078 1079
  SCliConn*  conn = req->data;
  SCliThrd*  thrd = conn->hostThrd;
  SCliBatch* p = conn->pBatch;
dengyihao's avatar
dengyihao 已提交
1080

dengyihao's avatar
dengyihao 已提交
1081
  SCliBatchList* pBatchList = p->pList;
dengyihao's avatar
dengyihao 已提交
1082
  SCliBatch*     nxtBatch = cliGetHeadFromList(pBatchList);
dengyihao's avatar
dengyihao 已提交
1083 1084
  pBatchList->connCnt -= 1;

dengyihao's avatar
dengyihao 已提交
1085 1086 1087
  conn->pBatch = NULL;

  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1088
    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 已提交
1089
           p->wLen, p->batchSize, uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
1090 1091 1092

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

dengyihao's avatar
dengyihao 已提交
1093
    cliHandleBatchReq(nxtBatch, thrd);
dengyihao's avatar
dengyihao 已提交
1094
  } else {
dengyihao's avatar
dengyihao 已提交
1095
    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 已提交
1096
           p->batchSize);
dengyihao's avatar
dengyihao 已提交
1097
    if (!uv_is_closing((uv_handle_t*)&conn->stream)) {
dengyihao's avatar
dengyihao 已提交
1098 1099 1100 1101 1102 1103
      if (nxtBatch != NULL) {
        conn->pBatch = nxtBatch;
        cliSendBatch(conn);
      } else {
        addConnToPool(thrd->pool, conn);
      }
dengyihao's avatar
dengyihao 已提交
1104
    } else {
dengyihao's avatar
dengyihao 已提交
1105 1106
      cliDestroyBatch(nxtBatch);
      // conn release by other callback
dengyihao's avatar
dengyihao 已提交
1107
    }
dengyihao's avatar
dengyihao 已提交
1108
  }
dengyihao's avatar
dengyihao 已提交
1109 1110 1111

  cliDestroyBatch(p);
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
1112
}
dengyihao's avatar
dengyihao 已提交
1113
static void cliHandleFastFail(SCliConn* pConn, int status) {
dengyihao's avatar
dengyihao 已提交
1114
  SCliThrd* pThrd = pConn->hostThrd;
dengyihao's avatar
dengyihao 已提交
1115
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1116 1117 1118

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

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

dengyihao's avatar
dengyihao 已提交
1152 1153 1154
void cliConnCb(uv_connect_t* req, int status) {
  SCliConn* pConn = req->data;
  SCliThrd* pThrd = pConn->hostThrd;
dengyihao's avatar
dengyihao 已提交
1155 1156 1157 1158 1159 1160 1161 1162 1163 1164
  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 已提交
1165 1166

  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
1167 1168 1169 1170 1171
    if (timeout == false) {
      cliHandleFastFail(pConn, status);
    } else if (timeout == true) {
      // already deal by timeout
    }
1172
    return;
dengyihao's avatar
dengyihao 已提交
1173
  }
dengyihao's avatar
dengyihao 已提交
1174

dengyihao's avatar
dengyihao 已提交
1175
  int32_t* oVal = taosHashGet(pThrd->connLimitCache, pConn->ip, strlen(pConn->ip));
dengyihao's avatar
dengyihao 已提交
1176
  int32_t  nVal = oVal == NULL ? 0 : (*oVal) + 1;
dengyihao's avatar
dengyihao 已提交
1177
  taosHashPut(pThrd->connLimitCache, pConn->ip, strlen(pConn->ip), &nVal, sizeof(nVal));
dengyihao's avatar
dengyihao 已提交
1178 1179 1180

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

dengyihao's avatar
dengyihao 已提交
1184 1185
  addrlen = sizeof(sockname);
  uv_tcp_getsockname((uv_tcp_t*)pConn->stream, &sockname, &addrlen);
dengyihao's avatar
dengyihao 已提交
1186
  transSockInfo2Str(&sockname, pConn->src);
dengyihao's avatar
dengyihao 已提交
1187

dengyihao's avatar
dengyihao 已提交
1188
  tTrace("%s conn %p connect to server successfully", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
1189 1190 1191 1192 1193
  if (pConn->pBatch != NULL) {
    cliSendBatch(pConn);
  } else {
    cliSend(pConn);
  }
dengyihao's avatar
dengyihao 已提交
1194 1195
}

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

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

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

dengyihao's avatar
dengyihao 已提交
1238
SCliConn* cliGetConn(SCliMsg* pMsg, SCliThrd* pThrd, bool* ignore) {
dengyihao's avatar
dengyihao 已提交
1239 1240 1241 1242
  STransConnCtx* pCtx = pMsg->ctx;
  SCliConn*      conn = NULL;

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

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

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

dengyihao's avatar
dengyihao 已提交
1307 1308 1309 1310 1311 1312 1313 1314 1315 1316
    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 已提交
1317

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

dengyihao's avatar
dengyihao 已提交
1321 1322 1323
  // 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 已提交
1324 1325 1326 1327

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

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

dengyihao's avatar
dengyihao 已提交
1331
  if (*val >= pTransInst->connLimitNum) {
dengyihao's avatar
dengyihao 已提交
1332 1333 1334 1335
    return -1;
  }
  return 0;
}
dengyihao's avatar
dengyihao 已提交
1336
void cliHandleReq(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
formate  
dengyihao 已提交
1337
  STrans*        pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1338
  STransConnCtx* pCtx = pMsg->ctx;
dengyihao's avatar
dengyihao 已提交
1339

dengyihao's avatar
dengyihao 已提交
1340
  cliMayCvtFqdnToIp(&pCtx->epSet, &pThrd->cvtAddr);
1341
  STraceId* trace = &pMsg->msg.info.traceId;
dengyihao's avatar
dengyihao 已提交
1342 1343
  char*     ip = EPSET_GET_INUSE_IP(&pCtx->epSet);
  uint16_t  port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1344

dengyihao's avatar
dengyihao 已提交
1345
  if (!EPSET_IS_VALID(&pCtx->epSet)) {
1346
    tGError("%s, msg %s sent with invalid epset", pTransInst->label, TMSG_INFO(pMsg->msg.msgType));
dengyihao's avatar
dengyihao 已提交
1347
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1348 1349
    return;
  }
dengyihao's avatar
dengyihao 已提交
1350

dengyihao's avatar
dengyihao 已提交
1351
  if (REQUEST_NO_RESP(&pMsg->msg) && (pTransInst->failFastFp != NULL && pTransInst->failFastFp(pMsg->msg.msgType))) {
dengyihao's avatar
dengyihao 已提交
1352
    char key[TSDB_FQDN_LEN + 64] = {0};
dengyihao's avatar
dengyihao 已提交
1353 1354 1355 1356
    CONN_CONSTRUCT_HASH_KEY(key, ip, port);

    SFailFastItem* item = taosHashGet(pThrd->failFastCache, key, strlen(key));
    if (item != NULL) {
dengyihao's avatar
dengyihao 已提交
1357 1358
      int32_t elapse = (int32_t)(taosGetTimestampMs() - item->timestamp);
      if (item->count >= pTransInst->failFastThreshold && (elapse >= 0 && elapse <= pTransInst->failFastInterval)) {
dengyihao's avatar
dengyihao 已提交
1359 1360
        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 已提交
1361 1362 1363 1364 1365 1366
        destroyCmsg(pMsg);
        return;
      }
    }
  }

dengyihao's avatar
dengyihao 已提交
1367 1368 1369
  bool      ignore = false;
  SCliConn* conn = cliGetConn(pMsg, pThrd, &ignore);
  if (ignore == true) {
dengyihao's avatar
dengyihao 已提交
1370
    // persist conn already release by server
dengyihao's avatar
dengyihao 已提交
1371 1372
    STransMsg resp;
    cliBuildExceptResp(pMsg, &resp);
dengyihao's avatar
dengyihao 已提交
1373 1374 1375
    if (pMsg->type != Release) {
      pTransInst->cfp(pTransInst->parent, &resp, NULL);
    }
dengyihao's avatar
dengyihao 已提交
1376
    destroyCmsg(pMsg);
dengyihao's avatar
dengyihao 已提交
1377 1378
    return;
  }
dengyihao's avatar
dengyihao 已提交
1379

dengyihao's avatar
dengyihao 已提交
1380
  if (conn == NULL && REQUEST_NO_RESP(&pMsg->msg) && 0 != cliPreCheckSessionLimit(pThrd, ip, port)) {
dengyihao's avatar
dengyihao 已提交
1381 1382 1383 1384 1385 1386
    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 已提交
1387
  if (conn != NULL) {
dengyihao's avatar
dengyihao 已提交
1388
    transCtxMerge(&conn->ctx, &pCtx->appCtx);
dengyihao's avatar
dengyihao 已提交
1389
    transQueuePush(&conn->cliMsgs, pMsg);
U
ubuntu 已提交
1390
    cliSend(conn);
dengyihao's avatar
dengyihao 已提交
1391
  } else {
U
ubuntu 已提交
1392
    conn = cliCreateConn(pThrd);
dengyihao's avatar
dengyihao 已提交
1393 1394 1395 1396

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

dengyihao's avatar
dengyihao 已提交
1397
    transCtxMerge(&conn->ctx, &pCtx->appCtx);
dengyihao's avatar
dengyihao 已提交
1398
    transQueuePush(&conn->cliMsgs, pMsg);
dengyihao's avatar
dengyihao 已提交
1399

dengyihao's avatar
dengyihao 已提交
1400
    char     key[TSDB_FQDN_LEN + 64] = {0};
dengyihao's avatar
dengyihao 已提交
1401
    char*    fqdn = EPSET_GET_INUSE_IP(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1402
    uint16_t port = EPSET_GET_INUSE_PORT(&pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
1403
    CONN_CONSTRUCT_HASH_KEY(key, fqdn, port);
dengyihao's avatar
dengyihao 已提交
1404 1405

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

dengyihao's avatar
dengyihao 已提交
1407
    uint32_t ipaddr = cliGetIpFromFqdnCache(pThrd->fqdn2ipCache, fqdn);
1408 1409 1410 1411 1412 1413 1414 1415 1416
    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 已提交
1417

dengyihao's avatar
dengyihao 已提交
1418
    struct sockaddr_in addr;
1419
    addr.sin_family = AF_INET;
1420
    addr.sin_addr.s_addr = ipaddr;
dengyihao's avatar
dengyihao 已提交
1421
    addr.sin_port = (uint16_t)htons(port);
dengyihao's avatar
dengyihao 已提交
1422

dengyihao's avatar
dengyihao 已提交
1423
    tGTrace("%s conn %p try to connect to %s", pTransInst->label, conn, conn->ip);
1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443
    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 已提交
1444

1445
    ret = uv_tcp_connect(&conn->connReq, (uv_tcp_t*)(conn->stream), (const struct sockaddr*)&addr, cliConnCb);
dengyihao's avatar
dengyihao 已提交
1446
    if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
1447 1448 1449 1450 1451
      uv_timer_stop(conn->timer);
      conn->timer->data = NULL;
      taosArrayPush(pThrd->timerList, &conn->timer);
      conn->timer = NULL;

dengyihao's avatar
dengyihao 已提交
1452
      cliHandleFastFail(conn, ret);
dengyihao's avatar
dengyihao 已提交
1453 1454
      return;
    }
dengyihao's avatar
dengyihao 已提交
1455
    uv_timer_start(conn->timer, cliConnTimeout, TRANS_CONN_TIMEOUT, 0);
dengyihao's avatar
dengyihao 已提交
1456
  }
dengyihao's avatar
dengyihao 已提交
1457
  tGTrace("%s conn %p ready", pTransInst->label, conn);
dengyihao's avatar
dengyihao 已提交
1458
}
dengyihao's avatar
dengyihao 已提交
1459

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

dengyihao's avatar
dengyihao 已提交
1463 1464
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
dengyihao's avatar
dengyihao 已提交
1465 1466 1467
    QUEUE_REMOVE(h);

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
1468 1469 1470 1471 1472

    if (pMsg->type == Quit) {
      pThrd->stopMsg = pMsg;
      continue;
    }
dengyihao's avatar
dengyihao 已提交
1473 1474 1475 1476 1477 1478 1479 1480
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);

    count++;
  }
  if (count >= 2) {
    tTrace("cli process batch size:%d", count);
  }
}
dengyihao's avatar
dengyihao 已提交
1481
SCliBatch* cliGetHeadFromList(SCliBatchList* pList) {
dengyihao's avatar
dengyihao 已提交
1482
  if (QUEUE_IS_EMPTY(&pList->wq) || pList->connCnt > pList->connMax || pList->sending > pList->connMax) {
dengyihao's avatar
dengyihao 已提交
1483 1484 1485 1486
    return NULL;
  }
  queue* hr = QUEUE_HEAD(&pList->wq);
  QUEUE_REMOVE(hr);
dengyihao's avatar
dengyihao 已提交
1487
  pList->sending += 1;
dengyihao's avatar
dengyihao 已提交
1488 1489 1490 1491 1492 1493

  pList->len -= 1;

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

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

dengyihao's avatar
dengyihao 已提交
1498
  int count = 0;
dengyihao's avatar
dengyihao 已提交
1499 1500
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
dengyihao's avatar
dengyihao 已提交
1501
    QUEUE_REMOVE(h);
dengyihao's avatar
dengyihao 已提交
1502 1503

    SCliMsg* pMsg = QUEUE_DATA(h, SCliMsg, q);
dengyihao's avatar
dengyihao 已提交
1504 1505 1506 1507 1508 1509

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

dengyihao's avatar
dengyihao 已提交
1510
    if (pMsg->type == Normal && REQUEST_NO_RESP(&pMsg->msg)) {
dengyihao's avatar
dengyihao 已提交
1511 1512 1513 1514 1515 1516 1517
      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 已提交
1518 1519 1520 1521 1522
      // SCliBatch** ppBatch = taosHashGet(pThrd->batchCache, key, sizeof(key));
      SCliBatchList** ppBatchList = taosHashGet(pThrd->batchCache, key, sizeof(key));
      if (ppBatchList == NULL || *ppBatchList == NULL) {
        SCliBatchList* pBatchList = taosMemoryCalloc(1, sizeof(SCliBatchList));
        QUEUE_INIT(&pBatchList->wq);
dengyihao's avatar
dengyihao 已提交
1523
        pBatchList->connMax = pInst->connLimitNum;
dengyihao's avatar
dengyihao 已提交
1524
        pBatchList->connCnt = 0;
dengyihao's avatar
dengyihao 已提交
1525
        pBatchList->batchLenLimit = pInst->batchSize;
dengyihao's avatar
dengyihao 已提交
1526
        pBatchList->len += 1;
dengyihao's avatar
dengyihao 已提交
1527

dengyihao's avatar
dengyihao 已提交
1528 1529 1530 1531
        pBatchList->ip = strdup(ip);
        pBatchList->dst = strdup(key);
        pBatchList->port = port;

dengyihao's avatar
dengyihao 已提交
1532 1533
        SCliBatch* pBatch = taosMemoryCalloc(1, sizeof(SCliBatch));
        QUEUE_INIT(&pBatch->wq);
dengyihao's avatar
dengyihao 已提交
1534 1535
        QUEUE_INIT(&pBatch->listq);

dengyihao's avatar
dengyihao 已提交
1536 1537 1538
        QUEUE_PUSH(&pBatch->wq, h);
        pBatch->wLen += 1;
        pBatch->batchSize += pMsg->msg.contLen;
dengyihao's avatar
dengyihao 已提交
1539
        pBatch->pList = pBatchList;
dengyihao's avatar
dengyihao 已提交
1540

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

dengyihao's avatar
dengyihao 已提交
1543
        taosHashPut(pThrd->batchCache, key, sizeof(key), &pBatchList, sizeof(void*));
dengyihao's avatar
dengyihao 已提交
1544
      } else {
dengyihao's avatar
dengyihao 已提交
1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
        if (QUEUE_IS_EMPTY(&(*ppBatchList)->wq)) {
          SCliBatch* pBatch = taosMemoryCalloc(1, sizeof(SCliBatch));
          QUEUE_INIT(&pBatch->wq);
          QUEUE_INIT(&pBatch->listq);

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

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

          continue;
        }

        queue*     hdr = QUEUE_TAIL(&((*ppBatchList)->wq));
        SCliBatch* pBatch = QUEUE_DATA(hdr, SCliBatch, listq);
        if ((pBatch->batchSize + pMsg->msg.contLen) < (*ppBatchList)->batchLenLimit) {
          QUEUE_PUSH(&pBatch->wq, h);
          pBatch->batchSize += pMsg->msg.contLen;
dengyihao's avatar
dengyihao 已提交
1566
          pBatch->wLen += 1;
dengyihao's avatar
dengyihao 已提交
1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
        } 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 已提交
1580
      }
dengyihao's avatar
dengyihao 已提交
1581
      continue;
dengyihao's avatar
dengyihao 已提交
1582
    }
dengyihao's avatar
dengyihao 已提交
1583
    (*cliAsyncHandle[pMsg->type])(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
1584
    count++;
dengyihao's avatar
dengyihao 已提交
1585
  }
dengyihao's avatar
dengyihao 已提交
1586

dengyihao's avatar
dengyihao 已提交
1587
  void** pIter = taosHashIterate(pThrd->batchCache, NULL);
dengyihao's avatar
dengyihao 已提交
1588
  while (pIter != NULL) {
dengyihao's avatar
dengyihao 已提交
1589
    SCliBatchList* batchList = (SCliBatchList*)(*pIter);
dengyihao's avatar
dengyihao 已提交
1590 1591 1592
    SCliBatch*     batch = cliGetHeadFromList(batchList);
    if (batch != NULL) {
      cliHandleBatchReq(batch, pThrd);
dengyihao's avatar
dengyihao 已提交
1593
    }
dengyihao's avatar
dengyihao 已提交
1594
    pIter = (void**)taosHashIterate(pThrd->batchCache, pIter);
dengyihao's avatar
dengyihao 已提交
1595 1596
  }

dengyihao's avatar
dengyihao 已提交
1597
  if (count >= 2) {
S
Shengliang Guan 已提交
1598
    tTrace("cli process batch size:%d", count);
dengyihao's avatar
dengyihao 已提交
1599
  }
dengyihao's avatar
dengyihao 已提交
1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
}

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

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

dengyihao's avatar
dengyihao 已提交
1613
  int8_t supportBatch = pTransInst->supportBatch;
dengyihao's avatar
dengyihao 已提交
1614
  if (supportBatch == 0) {
dengyihao's avatar
dengyihao 已提交
1615
    cliNoBatchDealReq(&wq, pThrd);
dengyihao's avatar
dengyihao 已提交
1616
  } else if (supportBatch == 1) {
dengyihao's avatar
dengyihao 已提交
1617
    cliBatchDealReq(&wq, pThrd);
dengyihao's avatar
dengyihao 已提交
1618
  }
dengyihao's avatar
dengyihao 已提交
1619

dengyihao's avatar
dengyihao 已提交
1620
  if (pThrd->stopMsg != NULL) cliHandleQuit(pThrd->stopMsg, pThrd);
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
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 已提交
1648
}
dengyihao's avatar
dengyihao 已提交
1649

dengyihao's avatar
dengyihao 已提交
1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678
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 已提交
1679 1680 1681
bool cliRecvReleaseReq(SCliConn* conn, STransMsgHead* pHead) {
  if (pHead->release == 1 && (pHead->msgLen) == sizeof(*pHead)) {
    uint64_t ahandle = pHead->ahandle;
dengyihao's avatar
dengyihao 已提交
1682
    tDebug("ahandle = %" PRIu64 "", ahandle);
dengyihao's avatar
dengyihao 已提交
1683 1684
    SCliMsg* pMsg = NULL;
    CONN_GET_MSGCTX_BY_AHANDLE(conn, ahandle);
dengyihao's avatar
dengyihao 已提交
1685

dengyihao's avatar
dengyihao 已提交
1686 1687
    transClearBuffer(&conn->readBuf);
    transFreeMsg(transContFromHead((char*)pHead));
dengyihao's avatar
dengyihao 已提交
1688 1689 1690 1691

    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 已提交
1692
        ASSERTS(pMsg == NULL, "trans-cli recv invaid release-req");
dengyihao's avatar
dengyihao 已提交
1693 1694
        return true;
      }
dengyihao's avatar
dengyihao 已提交
1695
    }
dengyihao's avatar
dengyihao 已提交
1696 1697 1698

    cliIteraConnMsgs(conn);

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

dengyihao's avatar
dengyihao 已提交
1702 1703 1704 1705 1706 1707
    addConnToPool(((SCliThrd*)conn->hostThrd)->pool, conn);
    return true;
  }
  return false;
}

U
ubuntu 已提交
1708
static void* cliWorkThread(void* arg) {
dengyihao's avatar
dengyihao 已提交
1709
  SCliThrd* pThrd = (SCliThrd*)arg;
dengyihao's avatar
dengyihao 已提交
1710
  pThrd->pid = taosGetSelfPthreadId();
G
gccgdb1234 已提交
1711
  setThreadName("trans-cli-work");
dengyihao's avatar
dengyihao 已提交
1712
  uv_run(pThrd->loop, UV_RUN_DEFAULT);
dengyihao's avatar
dengyihao 已提交
1713 1714

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

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

dengyihao's avatar
dengyihao 已提交
1721
  STrans* pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1722
  memcpy(cli->label, label, TSDB_LABEL_LEN);
dengyihao's avatar
dengyihao 已提交
1723
  cli->numOfThreads = numOfThreads;
dengyihao's avatar
dengyihao 已提交
1724
  cli->pThreadObj = (SCliThrd**)taosMemoryCalloc(cli->numOfThreads, sizeof(SCliThrd*));
dengyihao's avatar
dengyihao 已提交
1725 1726

  for (int i = 0; i < cli->numOfThreads; i++) {
dengyihao's avatar
dengyihao 已提交
1727
    SCliThrd* pThrd = createThrdObj(shandle);
dengyihao's avatar
dengyihao 已提交
1728 1729 1730 1731 1732
    if (pThrd == NULL) {
      return NULL;
    }

    int err = taosThreadCreate(&pThrd->thread, NULL, cliWorkThread, (void*)(pThrd));
dengyihao's avatar
dengyihao 已提交
1733
    if (err == 0) {
S
Shengliang Guan 已提交
1734
      tDebug("success to create tranport-cli thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1735
    }
dengyihao's avatar
dengyihao 已提交
1736
    cli->pThreadObj[i] = pThrd;
dengyihao's avatar
dengyihao 已提交
1737
  }
dengyihao's avatar
dengyihao 已提交
1738

dengyihao's avatar
dengyihao 已提交
1739 1740
  return cli;
}
dengyihao's avatar
dengyihao 已提交
1741

dengyihao's avatar
dengyihao 已提交
1742
static FORCE_INLINE void destroyUserdata(STransMsg* userdata) {
dengyihao's avatar
dengyihao 已提交
1743 1744 1745 1746 1747 1748
  if (userdata->pCont == NULL) {
    return;
  }
  transFreeMsg(userdata->pCont);
  userdata->pCont = NULL;
}
dengyihao's avatar
dengyihao 已提交
1749

dengyihao's avatar
dengyihao 已提交
1750
static FORCE_INLINE void destroyCmsg(void* arg) {
dengyihao's avatar
dengyihao 已提交
1751
  SCliMsg* pMsg = arg;
dengyihao's avatar
dengyihao 已提交
1752 1753 1754
  if (pMsg == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1755

dengyihao's avatar
dengyihao 已提交
1756 1757
  transDestroyConnCtx(pMsg->ctx);
  destroyUserdata(&pMsg->msg);
wafwerar's avatar
wafwerar 已提交
1758
  taosMemoryFree(pMsg);
dengyihao's avatar
dengyihao 已提交
1759
}
dengyihao's avatar
dengyihao 已提交
1760

dengyihao's avatar
dengyihao 已提交
1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779
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 已提交
1780 1781 1782
static SCliThrd* createThrdObj(void* trans) {
  STrans* pTransInst = trans;

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

dengyihao's avatar
dengyihao 已提交
1785
  QUEUE_INIT(&pThrd->msg);
wafwerar's avatar
wafwerar 已提交
1786
  taosThreadMutexInit(&pThrd->msgMtx, NULL);
dengyihao's avatar
dengyihao 已提交
1787

wafwerar's avatar
wafwerar 已提交
1788
  pThrd->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
1789 1790 1791 1792 1793 1794 1795 1796
  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 已提交
1797 1798 1799 1800 1801
  if (pTransInst->supportBatch) {
    pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 4, pThrd, cliAsyncCb);
  } else {
    pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 8, pThrd, cliAsyncCb);
  }
dengyihao's avatar
dengyihao 已提交
1802
  if (pThrd->asyncPool == NULL) {
dengyihao's avatar
ref log  
dengyihao 已提交
1803
    tError("failed to init async pool");
dengyihao's avatar
dengyihao 已提交
1804 1805 1806 1807 1808 1809
    uv_loop_close(pThrd->loop);
    taosMemoryFree(pThrd->loop);
    taosThreadMutexDestroy(&pThrd->msgMtx);
    taosMemoryFree(pThrd);
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
1810 1811 1812 1813

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

dengyihao's avatar
dengyihao 已提交
1816
  int32_t timerSize = 64;
dengyihao's avatar
dengyihao 已提交
1817 1818 1819 1820 1821 1822 1823
  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 已提交
1824
  pThrd->pool = createConnPool(4);
dengyihao's avatar
dengyihao 已提交
1825
  transDQCreate(pThrd->loop, &pThrd->delayQueue);
dengyihao's avatar
dengyihao 已提交
1826

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

dengyihao's avatar
dengyihao 已提交
1829 1830 1831
  pThrd->nextTimeout = taosGetTimestampMs() + CONN_PERSIST_TIME(pTransInst->idleTime);
  pThrd->pTransInst = trans;

dengyihao's avatar
dengyihao 已提交
1832
  pThrd->destroyAhandleFp = pTransInst->destroyFp;
dengyihao's avatar
dengyihao 已提交
1833
  pThrd->fqdn2ipCache = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
1834
  pThrd->failFastCache = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
dengyihao's avatar
dengyihao 已提交
1835 1836
  pThrd->connLimitCache = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true,
                                       pTransInst->connLimitLock == 0 ? HASH_NO_LOCK : HASH_ENTRY_LOCK);
dengyihao's avatar
dengyihao 已提交
1837

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

dengyihao's avatar
dengyihao 已提交
1840
  pThrd->quit = false;
dengyihao's avatar
dengyihao 已提交
1841 1842
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
1843
static void destroyThrdObj(SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1844 1845 1846
  if (pThrd == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1847

wafwerar's avatar
wafwerar 已提交
1848
  taosThreadJoin(pThrd->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1849
  CLI_RELEASE_UV(pThrd->loop);
wafwerar's avatar
wafwerar 已提交
1850
  taosThreadMutexDestroy(&pThrd->msgMtx);
dengyihao's avatar
dengyihao 已提交
1851
  TRANS_DESTROY_ASYNC_POOL_MSG(pThrd->asyncPool, SCliMsg, destroyCmsg);
dengyihao's avatar
dengyihao 已提交
1852
  transAsyncPoolDestroy(pThrd->asyncPool);
dengyihao's avatar
dengyihao 已提交
1853

dengyihao's avatar
dengyihao 已提交
1854
  transDQDestroy(pThrd->delayQueue, destroyCmsgAndAhandle);
dengyihao's avatar
dengyihao 已提交
1855
  transDQDestroy(pThrd->timeoutQueue, NULL);
dengyihao's avatar
dengyihao 已提交
1856

dengyihao's avatar
dengyihao 已提交
1857
  tDebug("thread destroy %" PRId64, pThrd->pid);
dengyihao's avatar
dengyihao 已提交
1858 1859 1860 1861 1862
  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 已提交
1863
  taosMemoryFree(pThrd->prepare);
wafwerar's avatar
wafwerar 已提交
1864
  taosMemoryFree(pThrd->loop);
dengyihao's avatar
dengyihao 已提交
1865
  taosHashCleanup(pThrd->fqdn2ipCache);
dengyihao's avatar
dengyihao 已提交
1866
  taosHashCleanup(pThrd->failFastCache);
dengyihao's avatar
dengyihao 已提交
1867
  taosHashCleanup(pThrd->connLimitCache);
dengyihao's avatar
dengyihao 已提交
1868 1869 1870

  void** pIter = taosHashIterate(pThrd->batchCache, NULL);
  while (pIter != NULL) {
dengyihao's avatar
dengyihao 已提交
1871 1872 1873 1874 1875 1876 1877 1878
    SCliBatchList* pBatchList = (SCliBatchList*)(*pIter);
    while (!QUEUE_IS_EMPTY(&pBatchList->wq)) {
      queue* h = QUEUE_HEAD(&pBatchList->wq);
      QUEUE_REMOVE(h);

      SCliBatch* pBatch = QUEUE_DATA(h, SCliBatch, listq);
      cliDestroyBatch(pBatch);
    }
dengyihao's avatar
dengyihao 已提交
1879 1880
    taosMemoryFree(pBatchList->ip);
    taosMemoryFree(pBatchList->dst);
dengyihao's avatar
dengyihao 已提交
1881
    taosMemoryFree(pBatchList);
dengyihao's avatar
dengyihao 已提交
1882

dengyihao's avatar
dengyihao 已提交
1883 1884
    pIter = (void**)taosHashIterate(pThrd->batchCache, pIter);
  }
dengyihao's avatar
dengyihao 已提交
1885
  taosHashCleanup(pThrd->batchCache);
wafwerar's avatar
wafwerar 已提交
1886
  taosMemoryFree(pThrd);
dengyihao's avatar
dengyihao 已提交
1887
}
dengyihao's avatar
dengyihao 已提交
1888

dengyihao's avatar
dengyihao 已提交
1889
static FORCE_INLINE void transDestroyConnCtx(STransConnCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
1890
  //
wafwerar's avatar
wafwerar 已提交
1891
  taosMemoryFree(ctx);
dengyihao's avatar
dengyihao 已提交
1892
}
dengyihao's avatar
dengyihao 已提交
1893

dengyihao's avatar
dengyihao 已提交
1894
void cliSendQuit(SCliThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1895
  // cli can stop gracefully
wafwerar's avatar
wafwerar 已提交
1896
  SCliMsg* msg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
1897
  msg->type = Quit;
dengyihao's avatar
dengyihao 已提交
1898
  transAsyncSend(thrd->asyncPool, &msg->q);
dengyihao's avatar
dengyihao 已提交
1899
  atomic_store_8(&thrd->asyncPool->stop, 1);
dengyihao's avatar
dengyihao 已提交
1900
}
dengyihao's avatar
dengyihao 已提交
1901 1902
void cliWalkCb(uv_handle_t* handle, void* arg) {
  if (!uv_is_closing(handle)) {
dengyihao's avatar
dengyihao 已提交
1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914
    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 已提交
1915
    uv_close(handle, cliDestroy);
dengyihao's avatar
dengyihao 已提交
1916 1917
  }
}
dengyihao's avatar
dengyihao 已提交
1918

dengyihao's avatar
dengyihao 已提交
1919
FORCE_INLINE int cliRBChoseIdx(STrans* pTransInst) {
dengyihao's avatar
dengyihao 已提交
1920
  int32_t index = pTransInst->index;
dengyihao's avatar
dengyihao 已提交
1921 1922 1923
  if (pTransInst->numOfThreads == 0) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
1924 1925 1926 1927
  /*
   * no lock, and to avoid CPU load imbalance, set limit pTransInst->numOfThreads * 2000;
   */
  if (pTransInst->index++ >= pTransInst->numOfThreads * 2000) {
U
ubuntu 已提交
1928 1929 1930 1931
    pTransInst->index = 0;
  }
  return index % pTransInst->numOfThreads;
}
dengyihao's avatar
dengyihao 已提交
1932
static FORCE_INLINE void doDelayTask(void* param) {
dengyihao's avatar
dengyihao 已提交
1933
  STaskArg* arg = param;
dengyihao's avatar
dengyihao 已提交
1934
  cliHandleReq((SCliMsg*)arg->param1, (SCliThrd*)arg->param2);
dengyihao's avatar
dengyihao 已提交
1935 1936
  taosMemoryFree(arg);
}
dengyihao's avatar
dengyihao 已提交
1937

dengyihao's avatar
dengyihao 已提交
1938 1939 1940
static void doCloseIdleConn(void* param) {
  STaskArg* arg = param;
  SCliConn* conn = arg->param1;
dengyihao's avatar
dengyihao 已提交
1941
  tDebug("%s conn %p idle, close it", CONN_GET_INST_LABEL(conn), conn);
dengyihao's avatar
dengyihao 已提交
1942
  conn->task = NULL;
dengyihao's avatar
dengyihao 已提交
1943 1944
  cliDestroyConn(conn, true);
  taosMemoryFree(arg);
dengyihao's avatar
dengyihao 已提交
1945 1946
}

dengyihao's avatar
dengyihao 已提交
1947
static void cliSchedMsgToNextNode(SCliMsg* pMsg, SCliThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1948
  STrans*        pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1949 1950
  STransConnCtx* pCtx = pMsg->ctx;

dengyihao's avatar
dengyihao 已提交
1951 1952
  STraceId* trace = &pMsg->msg.info.traceId;
  char      tbuf[256] = {0};
dengyihao's avatar
dengyihao 已提交
1953
  EPSET_DEBUG_STR(&pCtx->epSet, tbuf);
dengyihao's avatar
dengyihao 已提交
1954 1955
  tGDebug("%s retry on next node,use:%s, step: %d,timeout:%" PRId64 "", transLabel(pThrd->pTransInst), tbuf,
          pCtx->retryStep, pCtx->retryNextInterval);
dengyihao's avatar
dengyihao 已提交
1956 1957 1958 1959

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

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

dengyihao's avatar
dengyihao 已提交
1964
FORCE_INLINE void cliCompareAndSwap(int8_t* val, int8_t exp, int8_t newVal) {
dengyihao's avatar
dengyihao 已提交
1965 1966 1967 1968
  if (*val != exp) {
    *val = newVal;
  }
}
dengyihao's avatar
dengyihao 已提交
1969

dengyihao's avatar
dengyihao 已提交
1970
FORCE_INLINE bool cliTryExtractEpSet(STransMsg* pResp, SEpSet* dst) {
dengyihao's avatar
dengyihao 已提交
1971 1972 1973 1974 1975 1976
  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 已提交
1977 1978 1979 1980
    return false;
  }
  int32_t tlen = tSerializeSEpSet(NULL, 0, dst);

dengyihao's avatar
dengyihao 已提交
1981 1982 1983 1984 1985 1986 1987
  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 已提交
1988 1989

  pResp->pCont = buf;
dengyihao's avatar
dengyihao 已提交
1990 1991 1992
  pResp->contLen = len;

  *dst = epset;
dengyihao's avatar
dengyihao 已提交
1993 1994
  return true;
}
dengyihao's avatar
dengyihao 已提交
1995
bool cliResetEpset(STransConnCtx* pCtx, STransMsg* pResp, bool hasEpSet) {
dengyihao's avatar
dengyihao 已提交
1996
  bool noDelay = true;
dengyihao's avatar
dengyihao 已提交
1997 1998 1999 2000 2001 2002 2003
  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 已提交
2004 2005 2006 2007 2008 2009 2010 2011 2012 2013
    } 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 已提交
2014
      } else {
dengyihao's avatar
dengyihao 已提交
2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026
        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 已提交
2027
      }
dengyihao's avatar
dengyihao 已提交
2028 2029
    }
  } else {
dengyihao's avatar
dengyihao 已提交
2030
    SEpSet  epSet;
dengyihao's avatar
dengyihao 已提交
2031 2032
    int32_t valid = tDeserializeSEpSet(pResp->pCont, pResp->contLen, &epSet);
    if (valid < 0) {
dengyihao's avatar
dengyihao 已提交
2033 2034 2035 2036 2037 2038
      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 已提交
2039
    } else {
dengyihao's avatar
dengyihao 已提交
2040 2041 2042
      if (!transEpSetIsEqual(&pCtx->epSet, &epSet)) {
        tDebug("epset not equal, retry new epset");
        pCtx->epSet = epSet;
dengyihao's avatar
dengyihao 已提交
2043
        noDelay = false;
dengyihao's avatar
dengyihao 已提交
2044
      } else {
dengyihao's avatar
dengyihao 已提交
2045 2046 2047 2048 2049 2050
        if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps) {
          noDelay = false;
        } else {
          tDebug("epset equal, continue");
          EPSET_FORWARD_INUSE(&pCtx->epSet);
        }
dengyihao's avatar
dengyihao 已提交
2051
      }
dengyihao's avatar
dengyihao 已提交
2052 2053 2054 2055 2056
    }
  }
  return noDelay;
}
bool cliGenRetryRule(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
2057 2058
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2059

dengyihao's avatar
dengyihao 已提交
2060 2061
  STransConnCtx* pCtx = pMsg->ctx;
  int32_t        code = pResp->code;
dengyihao's avatar
dengyihao 已提交
2062

dengyihao's avatar
dengyihao 已提交
2063
  bool retry = pTransInst->retry != NULL ? pTransInst->retry(code, pResp->msgType - 1) : false;
dengyihao's avatar
dengyihao 已提交
2064 2065 2066
  if (retry == false) {
    return false;
  }
dengyihao's avatar
dengyihao 已提交
2067 2068 2069 2070 2071 2072 2073 2074

  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 已提交
2075
    pCtx->retryStep = 0;
dengyihao's avatar
dengyihao 已提交
2076
    pCtx->retryInit = true;
dengyihao's avatar
dengyihao 已提交
2077
    pCtx->retryCode = TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
2078 2079 2080

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

dengyihao's avatar
dengyihao 已提交
2083 2084
  if (-1 != pCtx->retryMaxTimeout && taosGetTimestampMs() - pCtx->retryInitTimestamp >= pCtx->retryMaxTimeout) {
    return false;
dengyihao's avatar
dengyihao 已提交
2085
  }
dengyihao's avatar
dengyihao 已提交
2086

dengyihao's avatar
dengyihao 已提交
2087 2088 2089 2090 2091 2092
  // code, msgType

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

dengyihao's avatar
dengyihao 已提交
2093 2094
  bool noDelay = false;
  if (code == TSDB_CODE_RPC_BROKEN_LINK || code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
dengyihao's avatar
dengyihao 已提交
2095
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2096
    noDelay = cliResetEpset(pCtx, pResp, false);
dengyihao's avatar
dengyihao 已提交
2097 2098
    transFreeMsg(pResp->pCont);
    transUnrefCliHandle(pConn);
dengyihao's avatar
dengyihao 已提交
2099
  } else if (code == TSDB_CODE_SYN_NOT_LEADER || code == TSDB_CODE_SYN_INTERNAL_ERROR ||
2100 2101
             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 已提交
2102
             code == TSDB_CODE_APP_IS_STOPPING || code == TSDB_CODE_VND_STOPPED) {
dengyihao's avatar
dengyihao 已提交
2103
    tTrace("code str %s, contlen:%d 1", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2104
    noDelay = cliResetEpset(pCtx, pResp, true);
dengyihao's avatar
dengyihao 已提交
2105 2106
    transFreeMsg(pResp->pCont);
    addConnToPool(pThrd->pool, pConn);
dengyihao's avatar
dengyihao 已提交
2107
  } else if (code == TSDB_CODE_SYN_RESTORING) {
dengyihao's avatar
dengyihao 已提交
2108
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2109
    noDelay = cliResetEpset(pCtx, pResp, true);
dengyihao's avatar
dengyihao 已提交
2110 2111
    addConnToPool(pThrd->pool, pConn);
    transFreeMsg(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2112
  } else {
dengyihao's avatar
dengyihao 已提交
2113
    tTrace("code str %s, contlen:%d 0", tstrerror(code), pResp->contLen);
dengyihao's avatar
dengyihao 已提交
2114 2115 2116
    noDelay = cliResetEpset(pCtx, pResp, false);
    addConnToPool(pThrd->pool, pConn);
    transFreeMsg(pResp->pCont);
dengyihao's avatar
dengyihao 已提交
2117
  }
dengyihao's avatar
dengyihao 已提交
2118 2119 2120 2121
  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 已提交
2122

dengyihao's avatar
dengyihao 已提交
2123 2124 2125
  if (noDelay == false) {
    pCtx->epsetRetryCnt = 1;
    pCtx->retryStep++;
dengyihao's avatar
dengyihao 已提交
2126

dengyihao's avatar
dengyihao 已提交
2127 2128 2129 2130 2131 2132
    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 已提交
2133 2134 2135
    // if (-1 != pCtx->retryMaxTimeout && taosGetTimestampMs() - pCtx->retryInitTimestamp >= pCtx->retryMaxTimeout) {
    //   return false;
    // }
dengyihao's avatar
dengyihao 已提交
2136 2137 2138
  } else {
    pCtx->retryNextInterval = 0;
    pCtx->epsetRetryCnt++;
dengyihao's avatar
dengyihao 已提交
2139
  }
dengyihao's avatar
dengyihao 已提交
2140

dengyihao's avatar
dengyihao 已提交
2141
  pMsg->sent = 0;
dengyihao's avatar
dengyihao 已提交
2142
  cliSchedMsgToNextNode(pMsg, pThrd);
dengyihao's avatar
dengyihao 已提交
2143
  return true;
dengyihao's avatar
dengyihao 已提交
2144
}
dengyihao's avatar
dengyihao 已提交
2145
int cliAppCb(SCliConn* pConn, STransMsg* pResp, SCliMsg* pMsg) {
dengyihao's avatar
dengyihao 已提交
2146 2147
  SCliThrd* pThrd = pConn->hostThrd;
  STrans*   pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
2148

dengyihao's avatar
dengyihao 已提交
2149
  if (pMsg == NULL || pMsg->ctx == NULL) {
dengyihao's avatar
dengyihao 已提交
2150
    tTrace("%s conn %p handle resp", pTransInst->label, pConn);
dengyihao's avatar
dengyihao 已提交
2151 2152 2153
    pTransInst->cfp(pTransInst->parent, pResp, NULL);
    return 0;
  }
dengyihao's avatar
dengyihao 已提交
2154

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

dengyihao's avatar
dengyihao 已提交
2157 2158 2159
  bool retry = cliGenRetryRule(pConn, pResp, pMsg);
  if (retry == true) {
    return -1;
dengyihao's avatar
dengyihao 已提交
2160
  }
dengyihao's avatar
dengyihao 已提交
2161

dengyihao's avatar
dengyihao 已提交
2162 2163 2164
  if (pCtx->retryCode != TSDB_CODE_SUCCESS) {
    int32_t code = pResp->code;
    // return internal code app
dengyihao's avatar
dengyihao 已提交
2165 2166
    if (code == TSDB_CODE_RPC_NETWORK_UNAVAIL || code == TSDB_CODE_RPC_BROKEN_LINK ||
        code == TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED) {
dengyihao's avatar
dengyihao 已提交
2167 2168 2169 2170
      pResp->code = pCtx->retryCode;
    }
  }

2171
  // check whole vnodes is offline on this vgroup
2172 2173
  if (pCtx->epsetRetryCnt >= pCtx->epSet.numOfEps || pCtx->retryStep > 0) {
    if (pResp->code == TSDB_CODE_RPC_NETWORK_UNAVAIL) {
A
Alex Duan 已提交
2174
      pResp->code = TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED;
2175
    } else if (pResp->code == TSDB_CODE_RPC_BROKEN_LINK) {
A
Alex Duan 已提交
2176
      pResp->code = TSDB_CODE_RPC_SOMENODE_BROKEN_LINK;
2177 2178 2179
    }
  }

dengyihao's avatar
dengyihao 已提交
2180 2181
  STraceId* trace = &pResp->info.traceId;
  bool      hasEpSet = cliTryExtractEpSet(pResp, &pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
2182
  if (hasEpSet) {
dengyihao's avatar
dengyihao 已提交
2183 2184
    char tbuf[256] = {0};
    EPSET_DEBUG_STR(&pCtx->epSet, tbuf);
dengyihao's avatar
dengyihao 已提交
2185
    tGTrace("%s conn %p extract epset from msg", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
2186
  }
dengyihao's avatar
dengyihao 已提交
2187

dengyihao's avatar
dengyihao 已提交
2188
  if (pCtx->pSem != NULL) {
dengyihao's avatar
dengyihao 已提交
2189
    tGTrace("%s conn %p(sync) handle resp", CONN_GET_INST_LABEL(pConn), pConn);
2190
    if (pCtx->pRsp == NULL) {
dengyihao's avatar
dengyihao 已提交
2191
      tGTrace("%s conn %p(sync) failed to resp, ignore", CONN_GET_INST_LABEL(pConn), pConn);
2192 2193 2194
    } else {
      memcpy((char*)pCtx->pRsp, (char*)pResp, sizeof(*pResp));
    }
dengyihao's avatar
dengyihao 已提交
2195
    tsem_post(pCtx->pSem);
2196
    pCtx->pRsp = NULL;
dengyihao's avatar
dengyihao 已提交
2197
  } else {
dengyihao's avatar
dengyihao 已提交
2198
    tGTrace("%s conn %p handle resp", CONN_GET_INST_LABEL(pConn), pConn);
dengyihao's avatar
dengyihao 已提交
2199
    if (retry == false && hasEpSet == true) {
dengyihao's avatar
dengyihao 已提交
2200
      pTransInst->cfp(pTransInst->parent, pResp, &pCtx->epSet);
dengyihao's avatar
dengyihao 已提交
2201
    } else {
dengyihao's avatar
dengyihao 已提交
2202
      if (!cliIsEpsetUpdated(pResp->code, pCtx)) {
dengyihao's avatar
dengyihao 已提交
2203 2204 2205 2206
        pTransInst->cfp(pTransInst->parent, pResp, NULL);
      } else {
        pTransInst->cfp(pTransInst->parent, pResp, &pCtx->epSet);
      }
dengyihao's avatar
dengyihao 已提交
2207
    }
dengyihao's avatar
dengyihao 已提交
2208
  }
dengyihao's avatar
dengyihao 已提交
2209
  return 0;
dengyihao's avatar
dengyihao 已提交
2210
}
U
ubuntu 已提交
2211 2212

void transCloseClient(void* arg) {
U
ubuntu 已提交
2213
  SCliObj* cli = arg;
dengyihao's avatar
dengyihao 已提交
2214
  for (int i = 0; i < cli->numOfThreads; i++) {
U
ubuntu 已提交
2215
    cliSendQuit(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
2216
    destroyThrdObj(cli->pThreadObj[i]);
dengyihao's avatar
dengyihao 已提交
2217
  }
wafwerar's avatar
wafwerar 已提交
2218 2219
  taosMemoryFree(cli->pThreadObj);
  taosMemoryFree(cli);
dengyihao's avatar
dengyihao 已提交
2220
}
dengyihao's avatar
dengyihao 已提交
2221 2222 2223 2224 2225
void transRefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_INC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
2226
  tTrace("%s conn %p ref %d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
2227 2228 2229 2230 2231 2232 2233
  UNUSED(ref);
}
void transUnrefCliHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  int ref = T_REF_DEC((SCliConn*)handle);
dengyihao's avatar
dengyihao 已提交
2234
  tTrace("%s conn %p ref:%d", CONN_GET_INST_LABEL((SCliConn*)handle), handle, ref);
dengyihao's avatar
dengyihao 已提交
2235
  if (ref == 0) {
U
ubuntu 已提交
2236
    cliDestroyConn((SCliConn*)handle, true);
dengyihao's avatar
dengyihao 已提交
2237 2238
  }
}
dengyihao's avatar
dengyihao 已提交
2239
static FORCE_INLINE SCliThrd* transGetWorkThrdFromHandle(STrans* trans, int64_t handle) {
dengyihao's avatar
dengyihao 已提交
2240
  SCliThrd*  pThrd = NULL;
dengyihao's avatar
dengyihao 已提交
2241
  SExHandle* exh = transAcquireExHandle(transGetRefMgt(), handle);
dengyihao's avatar
dengyihao 已提交
2242 2243 2244
  if (exh == NULL) {
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
2245

dengyihao's avatar
dengyihao 已提交
2246 2247 2248 2249 2250 2251
  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 已提交
2252
  pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
2253
  transReleaseExHandle(transGetRefMgt(), handle);
dengyihao's avatar
dengyihao 已提交
2254 2255
  return pThrd;
}
dengyihao's avatar
dengyihao 已提交
2256
SCliThrd* transGetWorkThrd(STrans* trans, int64_t handle) {
dengyihao's avatar
dengyihao 已提交
2257
  if (handle == 0) {
dengyihao's avatar
dengyihao 已提交
2258
    int idx = cliRBChoseIdx(trans);
dengyihao's avatar
dengyihao 已提交
2259
    if (idx < 0) return NULL;
dengyihao's avatar
dengyihao 已提交
2260 2261
    return ((SCliObj*)trans->tcphandle)->pThreadObj[idx];
  }
dengyihao's avatar
dengyihao 已提交
2262
  SCliThrd* pThrd = transGetWorkThrdFromHandle(trans, handle);
D
dapan1121 已提交
2263
  return pThrd;
dengyihao's avatar
dengyihao 已提交
2264
}
dengyihao's avatar
dengyihao 已提交
2265
int transReleaseCliHandle(void* handle) {
dengyihao's avatar
dengyihao 已提交
2266 2267 2268
  int  idx = -1;
  bool valid = false;

dengyihao's avatar
dengyihao 已提交
2269
  SCliThrd* pThrd = transGetWorkThrdFromHandle(NULL, (int64_t)handle);
dengyihao's avatar
dengyihao 已提交
2270
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2271
    return -1;
dengyihao's avatar
dengyihao 已提交
2272
  }
dengyihao's avatar
dengyihao 已提交
2273

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

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

dengyihao's avatar
dengyihao 已提交
2280
  SCliMsg* cmsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2281
  cmsg->msg = tmsg;
dengyihao's avatar
dengyihao 已提交
2282
  cmsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2283
  cmsg->type = Release;
dengyihao's avatar
dengyihao 已提交
2284
  cmsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
2285

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

dengyihao's avatar
dengyihao 已提交
2289
  if (0 != transAsyncSend(pThrd->asyncPool, &cmsg->q)) {
dengyihao's avatar
dengyihao 已提交
2290
    destroyCmsg(cmsg);
dengyihao's avatar
dengyihao 已提交
2291 2292
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2293
  return 0;
dengyihao's avatar
dengyihao 已提交
2294
}
dengyihao's avatar
dengyihao 已提交
2295

dengyihao's avatar
dengyihao 已提交
2296
int transSendRequest(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransCtx* ctx) {
dengyihao's avatar
dengyihao 已提交
2297 2298 2299 2300 2301
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
  if (pTransInst == NULL) {
    transFreeMsg(pReq->pCont);
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2302

dengyihao's avatar
dengyihao 已提交
2303
  SCliThrd* pThrd = transGetWorkThrd(pTransInst, (int64_t)pReq->info.handle);
dengyihao's avatar
dengyihao 已提交
2304
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2305
    transFreeMsg(pReq->pCont);
dengyihao's avatar
dengyihao 已提交
2306
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2307
    return TSDB_CODE_RPC_BROKEN_LINK;
dengyihao's avatar
dengyihao 已提交
2308 2309
  }

dengyihao's avatar
dengyihao 已提交
2310
  TRACE_SET_MSGID(&pReq->info.traceId, tGenIdPI64());
wafwerar's avatar
wafwerar 已提交
2311
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2312
  pCtx->epSet = *pEpSet;
S
Shengliang Guan 已提交
2313
  pCtx->ahandle = pReq->info.ahandle;
dengyihao's avatar
dengyihao 已提交
2314
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
2315

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

wafwerar's avatar
wafwerar 已提交
2318
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2319
  cliMsg->ctx = pCtx;
dengyihao's avatar
dengyihao 已提交
2320
  cliMsg->msg = *pReq;
dengyihao's avatar
dengyihao 已提交
2321
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2322
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
2323
  cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2324

dengyihao's avatar
dengyihao 已提交
2325
  STraceId* trace = &pReq->info.traceId;
dengyihao's avatar
dengyihao 已提交
2326 2327
  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 已提交
2328 2329
  if (0 != transAsyncSend(pThrd->asyncPool, &(cliMsg->q))) {
    destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2330
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2331 2332
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2333
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2334
  return 0;
dengyihao's avatar
dengyihao 已提交
2335
}
dengyihao's avatar
dengyihao 已提交
2336

dengyihao's avatar
dengyihao 已提交
2337
int transSendRecv(void* shandle, const SEpSet* pEpSet, STransMsg* pReq, STransMsg* pRsp) {
dengyihao's avatar
dengyihao 已提交
2338 2339 2340 2341 2342
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
  if (pTransInst == NULL) {
    transFreeMsg(pReq->pCont);
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2343

dengyihao's avatar
dengyihao 已提交
2344 2345
  SCliThrd* pThrd = transGetWorkThrd(pTransInst, (int64_t)pReq->info.handle);
  if (pThrd == NULL) {
dengyihao's avatar
dengyihao 已提交
2346
    transFreeMsg(pReq->pCont);
dengyihao's avatar
dengyihao 已提交
2347
    transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2348
    return TSDB_CODE_RPC_BROKEN_LINK;
dengyihao's avatar
dengyihao 已提交
2349
  }
dengyihao's avatar
dengyihao 已提交
2350

dengyihao's avatar
dengyihao 已提交
2351 2352
  tsem_t* sem = taosMemoryCalloc(1, sizeof(tsem_t));
  tsem_init(sem, 0, 0);
dengyihao's avatar
dengyihao 已提交
2353

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

wafwerar's avatar
wafwerar 已提交
2356
  STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2357
  pCtx->epSet = *pEpSet;
dengyihao's avatar
dengyihao 已提交
2358
  pCtx->origEpSet = *pEpSet;
S
Shengliang Guan 已提交
2359
  pCtx->ahandle = pReq->info.ahandle;
dengyihao's avatar
dengyihao 已提交
2360
  pCtx->msgType = pReq->msgType;
dengyihao's avatar
dengyihao 已提交
2361
  pCtx->pSem = sem;
dengyihao's avatar
dengyihao 已提交
2362 2363
  pCtx->pRsp = pRsp;

wafwerar's avatar
wafwerar 已提交
2364
  SCliMsg* cliMsg = taosMemoryCalloc(1, sizeof(SCliMsg));
dengyihao's avatar
dengyihao 已提交
2365 2366 2367
  cliMsg->ctx = pCtx;
  cliMsg->msg = *pReq;
  cliMsg->st = taosGetTimestampUs();
dengyihao's avatar
dengyihao 已提交
2368
  cliMsg->type = Normal;
dengyihao's avatar
dengyihao 已提交
2369
  cliMsg->refId = (int64_t)shandle;
dengyihao's avatar
dengyihao 已提交
2370

dengyihao's avatar
dengyihao 已提交
2371
  STraceId* trace = &pReq->info.traceId;
dengyihao's avatar
dengyihao 已提交
2372 2373
  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 已提交
2374

dengyihao's avatar
dengyihao 已提交
2375 2376
  int ret = transAsyncSend(pThrd->asyncPool, &cliMsg->q);
  if (ret != 0) {
dengyihao's avatar
dengyihao 已提交
2377
    destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2378
    goto _RETURN;
dengyihao's avatar
dengyihao 已提交
2379
  }
dengyihao's avatar
dengyihao 已提交
2380
  tsem_wait(sem);
dengyihao's avatar
dengyihao 已提交
2381 2382

_RETURN:
dengyihao's avatar
dengyihao 已提交
2383 2384
  tsem_destroy(sem);
  taosMemoryFree(sem);
dengyihao's avatar
dengyihao 已提交
2385
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2386
  return ret;
dengyihao's avatar
dengyihao 已提交
2387
}
dengyihao's avatar
dengyihao 已提交
2388 2389 2390
/*
 *
 **/
dengyihao's avatar
dengyihao 已提交
2391
int transSetDefaultAddr(void* shandle, const char* ip, const char* fqdn) {
dengyihao's avatar
dengyihao 已提交
2392
  STrans* pTransInst = (STrans*)transAcquireExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2393 2394 2395
  if (pTransInst == NULL) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
2396 2397 2398

  SCvtAddr cvtAddr = {0};
  if (ip != NULL && fqdn != NULL) {
dengyihao's avatar
dengyihao 已提交
2399 2400
    tstrncpy(cvtAddr.ip, ip, sizeof(cvtAddr.ip));
    tstrncpy(cvtAddr.fqdn, fqdn, sizeof(cvtAddr.fqdn));
dengyihao's avatar
dengyihao 已提交
2401 2402
    cvtAddr.cvt = true;
  }
dengyihao's avatar
dengyihao 已提交
2403 2404
  for (int i = 0; i < pTransInst->numOfThreads; i++) {
    STransConnCtx* pCtx = taosMemoryCalloc(1, sizeof(STransConnCtx));
dengyihao's avatar
dengyihao 已提交
2405
    pCtx->cvtAddr = cvtAddr;
dengyihao's avatar
dengyihao 已提交
2406 2407 2408 2409

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

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

dengyihao's avatar
dengyihao 已提交
2415 2416
    if (transAsyncSend(thrd->asyncPool, &(cliMsg->q)) != 0) {
      destroyCmsg(cliMsg);
dengyihao's avatar
dengyihao 已提交
2417
      transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2418 2419
      return -1;
    }
dengyihao's avatar
dengyihao 已提交
2420
  }
dengyihao's avatar
dengyihao 已提交
2421
  transReleaseExHandle(transGetInstMgt(), (int64_t)shandle);
dengyihao's avatar
dengyihao 已提交
2422
  return 0;
dengyihao's avatar
dengyihao 已提交
2423
}
dengyihao's avatar
dengyihao 已提交
2424 2425 2426 2427 2428

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

dengyihao's avatar
dengyihao 已提交
2430 2431
  return exh->refId;
}