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

#ifdef USE_UV

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

dengyihao's avatar
dengyihao 已提交
19 20
static TdThreadOnce transModuleInit = PTHREAD_ONCE_INIT;

dengyihao's avatar
dengyihao 已提交
21
static char* notify = "a";
dengyihao's avatar
dengyihao 已提交
22

dengyihao's avatar
dengyihao 已提交
23 24 25 26
typedef struct {
  int       notifyCount;  //
  int       init;         // init or not
  STransMsg msg;
dengyihao's avatar
dengyihao 已提交
27
} SSvrRegArg;
dengyihao's avatar
dengyihao 已提交
28

dengyihao's avatar
dengyihao 已提交
29
typedef struct SSvrConn {
dengyihao's avatar
dengyihao 已提交
30
  T_REF_DECLARE()
dengyihao's avatar
dengyihao 已提交
31
  uv_tcp_t*  pTcp;
dengyihao's avatar
dengyihao 已提交
32
  queue      wreqQueue;
dengyihao's avatar
dengyihao 已提交
33
  uv_timer_t pTimer;
dengyihao's avatar
dengyihao 已提交
34 35

  queue       queue;
dengyihao's avatar
dengyihao 已提交
36
  SConnBuffer readBuf;  // read buf,
dengyihao's avatar
dengyihao 已提交
37 38 39
  int         inType;
  void*       pTransInst;  // rpc init
  void*       ahandle;     //
dengyihao's avatar
dengyihao 已提交
40
  void*       hostThrd;
dengyihao's avatar
dengyihao 已提交
41
  STransQueue srvMsgs;
dengyihao's avatar
dengyihao 已提交
42

dengyihao's avatar
dengyihao 已提交
43
  SSvrRegArg regArg;
dengyihao's avatar
dengyihao 已提交
44
  bool       broken;  // conn broken;
dengyihao's avatar
dengyihao 已提交
45

dengyihao's avatar
dengyihao 已提交
46 47 48 49 50 51 52
  ConnStatus status;

  uint32_t clientIp;
  uint16_t port;

  char src[32];
  char dst[32];
U
ubuntu 已提交
53

dengyihao's avatar
dengyihao 已提交
54 55 56 57 58 59
  int64_t refId;
  int     spi;
  char    info[64];
  char    user[TSDB_UNI_LEN];  // user ID for the link
  char    secret[TSDB_PASSWORD_LEN];
  char    ckey[TSDB_PASSWORD_LEN];  // ciphering key
dengyihao's avatar
dengyihao 已提交
60
} SSvrConn;
dengyihao's avatar
dengyihao 已提交
61

dengyihao's avatar
dengyihao 已提交
62 63
typedef struct SSvrMsg {
  SSvrConn*     pConn;
dengyihao's avatar
dengyihao 已提交
64 65 66
  STransMsg     msg;
  queue         q;
  STransMsgType type;
dengyihao's avatar
dengyihao 已提交
67
} SSvrMsg;
dengyihao's avatar
dengyihao 已提交
68

dengyihao's avatar
dengyihao 已提交
69
typedef struct SWorkThrd {
dengyihao's avatar
dengyihao 已提交
70
  TdThread      thread;
wafwerar's avatar
wafwerar 已提交
71
  uv_connect_t  connect_req;
dengyihao's avatar
dengyihao 已提交
72 73 74 75
  uv_pipe_t*    pipe;
  uv_os_fd_t    fd;
  uv_loop_t*    loop;
  SAsyncPool*   asyncPool;
dengyihao's avatar
dengyihao 已提交
76
  uv_prepare_t* prepare;
dengyihao's avatar
dengyihao 已提交
77
  queue         msg;
dengyihao's avatar
dengyihao 已提交
78 79 80 81

  queue conn;
  void* pTransInst;
  bool  quit;
dengyihao's avatar
dengyihao 已提交
82
} SWorkThrd;
dengyihao's avatar
dengyihao 已提交
83 84

typedef struct SServerObj {
dengyihao's avatar
dengyihao 已提交
85
  TdThread   thread;
dengyihao's avatar
dengyihao 已提交
86 87 88 89
  uv_tcp_t   server;
  uv_loop_t* loop;

  // work thread info
dengyihao's avatar
dengyihao 已提交
90 91 92 93
  int         workerIdx;
  int         numOfThreads;
  int         numOfWorkerReady;
  SWorkThrd** pThreadObj;
dengyihao's avatar
dengyihao 已提交
94

wafwerar's avatar
wafwerar 已提交
95
  uv_pipe_t   pipeListen;
dengyihao's avatar
dengyihao 已提交
96 97 98 99
  uv_pipe_t** pipe;
  uint32_t    ip;
  uint32_t    port;
  uv_async_t* pAcceptAsync;  // just to quit from from accept thread
dengyihao's avatar
dengyihao 已提交
100 101

  bool inited;
dengyihao's avatar
dengyihao 已提交
102 103 104
} SServerObj;

static void uvAllocConnBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf);
dengyihao's avatar
dengyihao 已提交
105
static void uvAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf);
dengyihao's avatar
dengyihao 已提交
106
static void uvOnRecvCb(uv_stream_t* cli, ssize_t nread, const uv_buf_t* buf);
dengyihao's avatar
dengyihao 已提交
107
static void uvOnTimeoutCb(uv_timer_t* handle);
dengyihao's avatar
dengyihao 已提交
108
static void uvOnSendCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
109
static void uvOnPipeWriteCb(uv_write_t* req, int status);
dengyihao's avatar
dengyihao 已提交
110 111 112
static void uvOnAcceptCb(uv_stream_t* stream, int status);
static void uvOnConnectionCb(uv_stream_t* q, ssize_t nread, const uv_buf_t* buf);
static void uvWorkerAsyncCb(uv_async_t* handle);
dengyihao's avatar
dengyihao 已提交
113
static void uvAcceptAsyncCb(uv_async_t* handle);
dengyihao's avatar
dengyihao 已提交
114
static void uvShutDownCb(uv_shutdown_t* req, int status);
dengyihao's avatar
dengyihao 已提交
115
static void uvPrepareCb(uv_prepare_t* handle);
dengyihao's avatar
dengyihao 已提交
116

dengyihao's avatar
dengyihao 已提交
117 118
static bool uvRecvReleaseReq(SSvrConn* conn, STransMsgHead* pHead);

dengyihao's avatar
dengyihao 已提交
119 120 121 122 123 124
/*
 * time-consuming task throwed into BG work thread
 */
static void uvWorkDoTask(uv_work_t* req);
static void uvWorkAfterTask(uv_work_t* req, int status);

dengyihao's avatar
dengyihao 已提交
125 126
static void uvWalkCb(uv_handle_t* handle, void* arg);
static void uvFreeCb(uv_handle_t* handle);
dengyihao's avatar
dengyihao 已提交
127

dengyihao's avatar
dengyihao 已提交
128 129
static FORCE_INLINE void uvStartSendRespImpl(SSvrMsg* smsg);

dengyihao's avatar
dengyihao 已提交
130
static int  uvPrepareSendData(SSvrMsg* msg, uv_buf_t* wb);
dengyihao's avatar
dengyihao 已提交
131
static void uvStartSendResp(SSvrMsg* msg);
dengyihao's avatar
dengyihao 已提交
132

dengyihao's avatar
dengyihao 已提交
133
static void uvNotifyLinkBrokenToApp(SSvrConn* conn);
dengyihao's avatar
dengyihao 已提交
134

dengyihao's avatar
dengyihao 已提交
135 136 137 138
static FORCE_INLINE void      destroySmsg(SSvrMsg* smsg);
static FORCE_INLINE SSvrConn* createConn(void* hThrd);
static FORCE_INLINE void      destroyConn(SSvrConn* conn, bool clear /*clear handle or not*/);
static FORCE_INLINE void      destroyConnRegArg(SSvrConn* conn);
dengyihao's avatar
dengyihao 已提交
139

dengyihao's avatar
dengyihao 已提交
140
static int reallocConnRef(SSvrConn* conn);
dengyihao's avatar
dengyihao 已提交
141

dengyihao's avatar
dengyihao 已提交
142 143 144 145 146 147
static void uvHandleQuit(SSvrMsg* msg, SWorkThrd* thrd);
static void uvHandleRelease(SSvrMsg* msg, SWorkThrd* thrd);
static void uvHandleResp(SSvrMsg* msg, SWorkThrd* thrd);
static void uvHandleRegister(SSvrMsg* msg, SWorkThrd* thrd);
static void (*transAsyncHandle[])(SSvrMsg* msg, SWorkThrd* thrd) = {uvHandleResp, uvHandleQuit, uvHandleRelease,
                                                                    uvHandleRegister, NULL};
dengyihao's avatar
dengyihao 已提交
148

149
static void uvDestroyConn(uv_handle_t* handle);
dengyihao's avatar
dengyihao 已提交
150

dengyihao's avatar
dengyihao 已提交
151
// server and worker thread
152 153
static void* transWorkerThread(void* arg);
static void* transAcceptThread(void* arg);
dengyihao's avatar
dengyihao 已提交
154

dengyihao's avatar
dengyihao 已提交
155
// add handle loop
dengyihao's avatar
dengyihao 已提交
156
static bool addHandleToWorkloop(SWorkThrd* pThrd, char* pipeName);
dengyihao's avatar
dengyihao 已提交
157 158
static bool addHandleToAcceptloop(void* arg);

dengyihao's avatar
dengyihao 已提交
159 160 161 162 163 164 165
#define SRV_RELEASE_UV(loop)       \
  do {                             \
    uv_walk(loop, uvWalkCb, NULL); \
    uv_run(loop, UV_RUN_DEFAULT);  \
    uv_loop_close(loop);           \
  } while (0);

dengyihao's avatar
dengyihao 已提交
166 167 168 169
#define ASYNC_ERR_JRET(thrd)                            \
  do {                                                  \
    if (thrd->quit) {                                   \
      tTrace("worker thread already quit, ignore msg"); \
dengyihao's avatar
dengyihao 已提交
170
      goto _return1;                                    \
dengyihao's avatar
dengyihao 已提交
171 172 173
    }                                                   \
  } while (0)

dengyihao's avatar
dengyihao 已提交
174
void uvAllocRecvBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
175
  SSvrConn*    conn = handle->data;
dengyihao's avatar
dengyihao 已提交
176 177
  SConnBuffer* pBuf = &conn->readBuf;
  transAllocBuffer(pBuf, buf);
dengyihao's avatar
dengyihao 已提交
178 179 180 181
}

// refers specifically to query or insert timeout
static void uvHandleActivityTimeout(uv_timer_t* handle) {
dengyihao's avatar
dengyihao 已提交
182
  SSvrConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
183
  tDebug("%p timeout since no activity", conn);
dengyihao's avatar
dengyihao 已提交
184 185
}

dengyihao's avatar
dengyihao 已提交
186
static bool uvHandleReq(SSvrConn* pConn) {
dengyihao's avatar
dengyihao 已提交
187
  STrans* pTransInst = pConn->pTransInst;
dengyihao's avatar
opt rpc  
dengyihao 已提交
188

dengyihao's avatar
dengyihao 已提交
189 190 191
  STransMsgHead* pHead = NULL;

  int msgLen = transDumpFromBuffer(&pConn->readBuf, (char**)&pHead);
dengyihao's avatar
dengyihao 已提交
192
  if (msgLen <= 0) {
dengyihao's avatar
dengyihao 已提交
193 194
    tError("%s conn %p read invalid packet", transLabel(pTransInst), pConn);
    return false;
dengyihao's avatar
dengyihao 已提交
195
  }
dengyihao's avatar
dengyihao 已提交
196

dengyihao's avatar
dengyihao 已提交
197
  if (transDecompressMsg((char**)&pHead, msgLen) < 0) {
dengyihao's avatar
dengyihao 已提交
198
    tError("%s conn %p recv invalid packet, failed to decompress", transLabel(pTransInst), pConn);
dengyihao's avatar
dengyihao 已提交
199 200 201
    return false;
  }

dengyihao's avatar
dengyihao 已提交
202
  pHead->code = htonl(pHead->code);
dengyihao's avatar
dengyihao 已提交
203
  pHead->msgLen = htonl(pHead->msgLen);
dengyihao's avatar
dengyihao 已提交
204

dengyihao's avatar
dengyihao 已提交
205
  memcpy(pConn->user, pHead->user, strlen(pHead->user));
dengyihao's avatar
dengyihao 已提交
206

dengyihao's avatar
dengyihao 已提交
207 208 209 210
  if (uvRecvReleaseReq(pConn, pHead)) {
    return true;
  }

dengyihao's avatar
dengyihao 已提交
211 212 213 214 215
  // TODO(dengyihao): time-consuming task throwed into BG Thread
  //  uv_work_t* wreq = taosMemoryMalloc(sizeof(uv_work_t));
  //  wreq->data = pConn;
  //  uv_read_stop((uv_stream_t*)pConn->pTcp);
  //  transRefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
216
  //  uv_queue_work(((SWorkThrd*)pConn->hostThrd)->loop, wreq, uvWorkDoTask, uvWorkAfterTask);
dengyihao's avatar
dengyihao 已提交
217

dengyihao's avatar
dengyihao 已提交
218
  STransMsg transMsg;
dengyihao's avatar
dengyihao 已提交
219
  memset(&transMsg, 0, sizeof(transMsg));
dengyihao's avatar
dengyihao 已提交
220 221 222 223
  transMsg.contLen = transContLenFromMsg(pHead->msgLen);
  transMsg.pCont = pHead->content;
  transMsg.msgType = pHead->msgType;
  transMsg.code = pHead->code;
dengyihao's avatar
dengyihao 已提交
224

dengyihao's avatar
dengyihao 已提交
225
  pConn->inType = pHead->msgType;
dengyihao's avatar
dengyihao 已提交
226 227 228 229
  if (pConn->status == ConnNormal) {
    if (pHead->persist == 1) {
      pConn->status = ConnAcquire;
      transRefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
230
      tDebug("conn %p acquired by server app", pConn);
dengyihao's avatar
dengyihao 已提交
231 232
    }
  }
dengyihao's avatar
dengyihao 已提交
233
  STraceId* trace = &pHead->traceId;
dengyihao's avatar
dengyihao 已提交
234 235 236 237

  int64_t        cost = taosGetTimestampUs() - taosNtoh64(pHead->timestamp);
  static int64_t EXCEPTION_LIMIT_US = 100 * 1000;

dengyihao's avatar
dengyihao 已提交
238
  if (pConn->status == ConnNormal && pHead->noResp == 0) {
dengyihao's avatar
dengyihao 已提交
239
    transRefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
240 241 242 243 244 245 246
    if (cost >= EXCEPTION_LIMIT_US) {
      tGWarn("%s conn %p %s received from %s, local info:%s, len:%d, cost:%dus, recv exception", transLabel(pTransInst),
             pConn, TMSG_INFO(transMsg.msgType), pConn->dst, pConn->src, msgLen, (int)cost);
    } else {
      tGDebug("%s conn %p %s received from %s, local info:%s, len:%d, cost:%dus", transLabel(pTransInst), pConn,
              TMSG_INFO(transMsg.msgType), pConn->dst, pConn->src, msgLen, (int)cost);
    }
dengyihao's avatar
dengyihao 已提交
247
  } else {
dengyihao's avatar
dengyihao 已提交
248 249 250 251 252
    if (cost >= EXCEPTION_LIMIT_US) {
      tGWarn("%s conn %p %s received from %s, local info:%s, len:%d, resp:%d, code:%d, cost:%dus, recv exception",
             transLabel(pTransInst), pConn, TMSG_INFO(transMsg.msgType), pConn->dst, pConn->src, msgLen, pHead->noResp,
             transMsg.code, (int)(cost));
    } else {
dengyihao's avatar
dengyihao 已提交
253 254 255
      tGDebug("%s conn %p %s received from %s, local info:%s, len:%d, resp:%d, code:%d, cost:%dus",
              transLabel(pTransInst), pConn, TMSG_INFO(transMsg.msgType), pConn->dst, pConn->src, msgLen, pHead->noResp,
              transMsg.code, (int)(cost));
dengyihao's avatar
dengyihao 已提交
256
    }
dengyihao's avatar
dengyihao 已提交
257 258
  }

dengyihao's avatar
dengyihao 已提交
259
  // pHead->noResp = 1,
dengyihao's avatar
dengyihao 已提交
260 261 262
  // 1. server application should not send resp on handle
  // 2. once send out data, cli conn released to conn pool immediately
  // 3. not mixed with persist
dengyihao's avatar
dengyihao 已提交
263
  transMsg.info.ahandle = (void*)pHead->ahandle;
dengyihao's avatar
dengyihao 已提交
264
  transMsg.info.handle = (void*)transAcquireExHandle(transGetRefMgt(), pConn->refId);
S
Shengliang Guan 已提交
265
  transMsg.info.refId = pConn->refId;
dengyihao's avatar
dengyihao 已提交
266 267
  transMsg.info.traceId = pHead->traceId;

dengyihao's avatar
dengyihao 已提交
268 269
  tGTrace("%s handle %p conn:%p translated to app, refId:%" PRIu64, transLabel(pTransInst), transMsg.info.handle, pConn,
          pConn->refId);
dengyihao's avatar
dengyihao 已提交
270 271 272 273
  ASSERTS(transMsg.info.handle != NULL, "trans-svr failed to alloc handle to msg");
  if (transMsg.info.handle == NULL) {
    return false;
  }
dengyihao's avatar
dengyihao 已提交
274

dengyihao's avatar
dengyihao 已提交
275
  if (pHead->noResp == 1) {
S
Shengliang Guan 已提交
276
    transMsg.info.refId = -1;
dengyihao's avatar
dengyihao 已提交
277
  }
dengyihao's avatar
dengyihao 已提交
278 279

  // set up conn info
280
  SRpcConnInfo* pConnInfo = &(transMsg.info.conn);
dengyihao's avatar
dengyihao 已提交
281 282
  pConnInfo->clientIp = pConn->clientIp;
  pConnInfo->clientPort = pConn->port;
dengyihao's avatar
dengyihao 已提交
283 284
  tstrncpy(pConnInfo->user, pConn->user, sizeof(pConnInfo->user));

dengyihao's avatar
dengyihao 已提交
285
  transReleaseExHandle(transGetRefMgt(), pConn->refId);
dengyihao's avatar
dengyihao 已提交
286

dengyihao's avatar
dengyihao 已提交
287
  (*pTransInst->cfp)(pTransInst->parent, &transMsg, NULL);
dengyihao's avatar
dengyihao 已提交
288
  return true;
dengyihao's avatar
dengyihao 已提交
289 290
}

dengyihao's avatar
dengyihao 已提交
291
void uvOnRecvCb(uv_stream_t* cli, ssize_t nread, const uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
292 293 294
  SSvrConn* conn = cli->data;
  STrans*   pTransInst = conn->pTransInst;

dengyihao's avatar
dengyihao 已提交
295
  SConnBuffer* pBuf = &conn->readBuf;
dengyihao's avatar
dengyihao 已提交
296 297
  if (nread > 0) {
    pBuf->len += nread;
dengyihao's avatar
dengyihao 已提交
298 299
    if (pBuf->len <= TRANS_PACKET_LIMIT) {
      while (transReadComplete(pBuf)) {
dengyihao's avatar
dengyihao 已提交
300
        if (true == pBuf->invalid || false == uvHandleReq(conn)) {
dengyihao's avatar
dengyihao 已提交
301 302
          tError("%s conn %p read invalid packet, received from %s, local info:%s", transLabel(pTransInst), conn,
                 conn->dst, conn->src);
dengyihao's avatar
dengyihao 已提交
303
          destroyConn(conn, true);
dengyihao's avatar
dengyihao 已提交
304
          return;
dengyihao's avatar
dengyihao 已提交
305
        }
dengyihao's avatar
dengyihao 已提交
306 307 308
      }
      return;
    } else {
dengyihao's avatar
dengyihao 已提交
309
      tError("%s conn %p read invalid packet, exceed limit, received from %s, local info:%s", transLabel(pTransInst),
dengyihao's avatar
dengyihao 已提交
310
             conn, conn->dst, conn->src);
dengyihao's avatar
dengyihao 已提交
311 312
      destroyConn(conn, true);
      return;
dengyihao's avatar
dengyihao 已提交
313 314
    }
  }
315 316 317
  if (nread == 0) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
318

319
  tWarn("%s conn %p read error:%s", transLabel(pTransInst), conn, uv_err_name(nread));
U
ubuntu 已提交
320
  if (nread < 0) {
dengyihao's avatar
dengyihao 已提交
321
    conn->broken = true;
dengyihao's avatar
dengyihao 已提交
322 323
    if (conn->status == ConnAcquire) {
      if (conn->regArg.init) {
dengyihao's avatar
dengyihao 已提交
324
        tTrace("%s conn %p broken, notify server app", transLabel(pTransInst), conn);
dengyihao's avatar
dengyihao 已提交
325 326 327 328 329
        STrans* pTransInst = conn->pTransInst;
        (*pTransInst->cfp)(pTransInst->parent, &(conn->regArg.msg), NULL);
        memset(&conn->regArg, 0, sizeof(conn->regArg));
      }
    }
dengyihao's avatar
dengyihao 已提交
330
    destroyConn(conn, true);
dengyihao's avatar
dengyihao 已提交
331 332 333 334
  }
}
void uvAllocConnBufferCb(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf) {
  buf->len = 2;
wafwerar's avatar
wafwerar 已提交
335
  buf->base = taosMemoryCalloc(1, sizeof(char) * buf->len);
dengyihao's avatar
dengyihao 已提交
336 337 338 339
}

void uvOnTimeoutCb(uv_timer_t* handle) {
  // opt
dengyihao's avatar
dengyihao 已提交
340
  SSvrConn* pConn = handle->data;
dengyihao's avatar
dengyihao 已提交
341
  tError("conn %p time out", pConn);
dengyihao's avatar
dengyihao 已提交
342 343
}

dengyihao's avatar
dengyihao 已提交
344
void uvOnSendCb(uv_write_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
345
  SSvrConn* conn = transReqQueueRemove(req);
dengyihao's avatar
dengyihao 已提交
346 347
  if (conn == NULL) return;

dengyihao's avatar
dengyihao 已提交
348
  if (status == 0) {
dengyihao's avatar
dengyihao 已提交
349
    tTrace("conn %p data already was written on stream", conn);
dengyihao's avatar
dengyihao 已提交
350
    if (!transQueueEmpty(&conn->srvMsgs)) {
dengyihao's avatar
dengyihao 已提交
351 352 353 354
      SSvrMsg*  msg = transQueuePop(&conn->srvMsgs);
      STraceId* trace = &msg->msg.info.traceId;
      tGDebug("conn %p write data out", conn);

dengyihao's avatar
add UT  
dengyihao 已提交
355
      destroySmsg(msg);
dengyihao's avatar
dengyihao 已提交
356
      // send cached data
dengyihao's avatar
dengyihao 已提交
357
      if (!transQueueEmpty(&conn->srvMsgs)) {
dengyihao's avatar
dengyihao 已提交
358
        msg = (SSvrMsg*)transQueueGet(&conn->srvMsgs, 0);
dengyihao's avatar
dengyihao 已提交
359 360 361 362 363 364 365 366 367
        if (msg->type == Register && conn->status == ConnAcquire) {
          conn->regArg.notifyCount = 0;
          conn->regArg.init = 1;
          conn->regArg.msg = msg->msg;
          if (conn->broken) {
            STrans* pTransInst = conn->pTransInst;
            (pTransInst->cfp)(pTransInst->parent, &(conn->regArg.msg), NULL);
            memset(&conn->regArg, 0, sizeof(conn->regArg));
          }
dengyihao's avatar
dengyihao 已提交
368
          transQueuePop(&conn->srvMsgs);
wafwerar's avatar
wafwerar 已提交
369
          taosMemoryFree(msg);
dengyihao's avatar
dengyihao 已提交
370

dengyihao's avatar
dengyihao 已提交
371
          msg = (SSvrMsg*)transQueueGet(&conn->srvMsgs, 0);
dengyihao's avatar
dengyihao 已提交
372
          if (msg != NULL) {
dengyihao's avatar
dengyihao 已提交
373
            uvStartSendRespImpl(msg);
dengyihao's avatar
dengyihao 已提交
374
          }
dengyihao's avatar
dengyihao 已提交
375
        } else {
dengyihao's avatar
dengyihao 已提交
376
          uvStartSendRespImpl(msg);
dengyihao's avatar
dengyihao 已提交
377
        }
dengyihao's avatar
add UT  
dengyihao 已提交
378
      }
dengyihao's avatar
dengyihao 已提交
379
    }
dengyihao's avatar
dengyihao 已提交
380
    transUnrefSrvHandle(conn);
dengyihao's avatar
dengyihao 已提交
381
  } else {
dengyihao's avatar
dengyihao 已提交
382 383 384 385 386
    if (!uv_is_closing((uv_handle_t*)(conn->pTcp))) {
      tError("conn %p failed to write data, %s", conn, uv_err_name(status));
      conn->broken = true;
      transUnrefSrvHandle(conn);
    }
dengyihao's avatar
dengyihao 已提交
387 388
  }
}
dengyihao's avatar
dengyihao 已提交
389 390
static void uvOnPipeWriteCb(uv_write_t* req, int status) {
  if (status == 0) {
dengyihao's avatar
dengyihao 已提交
391
    tTrace("success to dispatch conn to work thread");
dengyihao's avatar
dengyihao 已提交
392 393 394
  } else {
    tError("fail to dispatch conn to work thread");
  }
dengyihao's avatar
dengyihao 已提交
395 396 397 398 399
  if (!uv_is_closing((uv_handle_t*)req->data)) {
    uv_close((uv_handle_t*)req->data, uvFreeCb);
  } else {
    taosMemoryFree(req->data);
  }
wafwerar's avatar
wafwerar 已提交
400
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
401
}
dengyihao's avatar
dengyihao 已提交
402

dengyihao's avatar
dengyihao 已提交
403
static int uvPrepareSendData(SSvrMsg* smsg, uv_buf_t* wb) {
dengyihao's avatar
dengyihao 已提交
404
  SSvrConn*  pConn = smsg->pConn;
dengyihao's avatar
formate  
dengyihao 已提交
405
  STransMsg* pMsg = &smsg->msg;
dengyihao's avatar
dengyihao 已提交
406 407 408 409 410
  if (pMsg->pCont == 0) {
    pMsg->pCont = (void*)rpcMallocCont(0);
    pMsg->contLen = 0;
  }
  STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
S
Shengliang Guan 已提交
411
  pHead->ahandle = (uint64_t)pMsg->info.ahandle;
dengyihao's avatar
dengyihao 已提交
412
  pHead->traceId = pMsg->info.traceId;
dengyihao's avatar
dengyihao 已提交
413
  pHead->hasEpSet = pMsg->info.hasEpSet;
dengyihao's avatar
dengyihao 已提交
414
  pHead->magicNum = htonl(TRANS_MAGIC_NUM);
dengyihao's avatar
dengyihao 已提交
415

dengyihao's avatar
dengyihao 已提交
416
  // handle invalid drop_task resp, TD-20098
dengyihao's avatar
dengyihao 已提交
417
  if (pConn->inType == TDMT_SCH_DROP_TASK && pMsg->code == TSDB_CODE_VND_INVALID_VGROUP_ID) {
dengyihao's avatar
dengyihao 已提交
418 419 420 421 422
    transQueuePop(&pConn->srvMsgs);
    destroySmsg(smsg);
    return -1;
  }

dengyihao's avatar
dengyihao 已提交
423
  if (pConn->status == ConnNormal) {
dengyihao's avatar
dengyihao 已提交
424
    pHead->msgType = (0 == pMsg->msgType ? pConn->inType + 1 : pMsg->msgType);
dengyihao's avatar
dengyihao 已提交
425
    if (smsg->type == Release) pHead->msgType = 0;
dengyihao's avatar
dengyihao 已提交
426
  } else {
dengyihao's avatar
dengyihao 已提交
427 428 429
    if (smsg->type == Release) {
      pHead->msgType = 0;
      pConn->status = ConnNormal;
dengyihao's avatar
dengyihao 已提交
430
      destroyConnRegArg(pConn);
dengyihao's avatar
dengyihao 已提交
431 432
      transUnrefSrvHandle(pConn);
    } else {
dengyihao's avatar
dengyihao 已提交
433
      // set up resp msg type
dengyihao's avatar
dengyihao 已提交
434
      pHead->msgType = (0 == pMsg->msgType ? pConn->inType + 1 : pMsg->msgType);
dengyihao's avatar
dengyihao 已提交
435
    }
dengyihao's avatar
dengyihao 已提交
436
  }
dengyihao's avatar
dengyihao 已提交
437

dengyihao's avatar
dengyihao 已提交
438
  pHead->release = smsg->type == Release ? 1 : 0;
dengyihao's avatar
dengyihao 已提交
439
  pHead->code = htonl(pMsg->code);
dengyihao's avatar
dengyihao 已提交
440
  pHead->msgLen = htonl(pMsg->contLen + sizeof(STransMsgHead));
dengyihao's avatar
dengyihao 已提交
441

dengyihao's avatar
dengyihao 已提交
442 443
  char*   msg = (char*)pHead;
  int32_t len = transMsgLenFromCont(pMsg->contLen);
dengyihao's avatar
dengyihao 已提交
444

dengyihao's avatar
dengyihao 已提交
445
  STrans* pTransInst = pConn->pTransInst;
dengyihao's avatar
dengyihao 已提交
446
  if (pTransInst->compressSize != -1 && pTransInst->compressSize < pMsg->contLen) {
dengyihao's avatar
dengyihao 已提交
447 448 449 450
    len = transCompressMsg(pMsg->pCont, pMsg->contLen) + sizeof(STransMsgHead);
    pHead->msgLen = (int32_t)htonl((uint32_t)len);
  }

dengyihao's avatar
dengyihao 已提交
451
  STraceId* trace = &pMsg->info.traceId;
dengyihao's avatar
dengyihao 已提交
452
  tGDebug("%s conn %p %s is sent to %s, local info:%s, len:%d", transLabel(pTransInst), pConn,
dengyihao's avatar
dengyihao 已提交
453
          TMSG_INFO(pHead->msgType), pConn->dst, pConn->src, len);
dengyihao's avatar
dengyihao 已提交
454

dengyihao's avatar
dengyihao 已提交
455
  wb->base = (char*)pHead;
dengyihao's avatar
dengyihao 已提交
456
  wb->len = len;
dengyihao's avatar
dengyihao 已提交
457
  return 0;
dengyihao's avatar
dengyihao 已提交
458
}
dengyihao's avatar
dengyihao 已提交
459

dengyihao's avatar
dengyihao 已提交
460
static FORCE_INLINE void uvStartSendRespImpl(SSvrMsg* smsg) {
dengyihao's avatar
dengyihao 已提交
461 462 463 464 465
  SSvrConn* pConn = smsg->pConn;
  if (pConn->broken) {
    return;
  }

dengyihao's avatar
dengyihao 已提交
466
  uv_buf_t wb;
dengyihao's avatar
dengyihao 已提交
467 468 469
  if (uvPrepareSendData(smsg, &wb) < 0) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
470

dengyihao's avatar
dengyihao 已提交
471
  transRefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
472
  uv_write_t* req = transReqQueuePush(&pConn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
473
  uv_write(req, (uv_stream_t*)pConn->pTcp, &wb, 1, uvOnSendCb);
dengyihao's avatar
dengyihao 已提交
474
}
dengyihao's avatar
dengyihao 已提交
475
static void uvStartSendResp(SSvrMsg* smsg) {
dengyihao's avatar
dengyihao 已提交
476
  // impl
dengyihao's avatar
dengyihao 已提交
477
  SSvrConn* pConn = smsg->pConn;
dengyihao's avatar
dengyihao 已提交
478
  if (pConn->broken == true) {
dengyihao's avatar
dengyihao 已提交
479
    // persist by
dengyihao's avatar
dengyihao 已提交
480
    destroySmsg(smsg);
dengyihao's avatar
dengyihao 已提交
481 482 483
    transUnrefSrvHandle(pConn);
    return;
  }
dengyihao's avatar
dengyihao 已提交
484 485 486
  if (pConn->status == ConnNormal) {
    transUnrefSrvHandle(pConn);
  }
dengyihao's avatar
dengyihao 已提交
487

dengyihao's avatar
dengyihao 已提交
488
  if (!transQueuePush(&pConn->srvMsgs, smsg)) {
dengyihao's avatar
dengyihao 已提交
489 490
    return;
  }
dengyihao's avatar
dengyihao 已提交
491
  uvStartSendRespImpl(smsg);
dengyihao's avatar
dengyihao 已提交
492 493
  return;
}
dengyihao's avatar
dengyihao 已提交
494

dengyihao's avatar
dengyihao 已提交
495
static FORCE_INLINE void destroySmsg(SSvrMsg* smsg) {
dengyihao's avatar
dengyihao 已提交
496 497 498 499
  if (smsg == NULL) {
    return;
  }
  transFreeMsg(smsg->msg.pCont);
wafwerar's avatar
wafwerar 已提交
500
  taosMemoryFree(smsg);
dengyihao's avatar
dengyihao 已提交
501
}
dengyihao's avatar
dengyihao 已提交
502
static void destroyAllConn(SWorkThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
503
  tTrace("thread %p destroy all conn ", pThrd);
dengyihao's avatar
fix bug  
dengyihao 已提交
504 505 506 507 508
  while (!QUEUE_IS_EMPTY(&pThrd->conn)) {
    queue* h = QUEUE_HEAD(&pThrd->conn);
    QUEUE_REMOVE(h);
    QUEUE_INIT(h);

dengyihao's avatar
dengyihao 已提交
509
    SSvrConn* c = QUEUE_DATA(h, SSvrConn, queue);
dengyihao's avatar
dengyihao 已提交
510 511 512
    while (T_REF_VAL_GET(c) >= 2) {
      transUnrefSrvHandle(c);
    }
dengyihao's avatar
dengyihao 已提交
513
    transUnrefSrvHandle(c);
dengyihao's avatar
fix bug  
dengyihao 已提交
514 515
  }
}
dengyihao's avatar
dengyihao 已提交
516
void uvWorkerAsyncCb(uv_async_t* handle) {
dengyihao's avatar
dengyihao 已提交
517 518 519 520
  SAsyncItem* item = handle->data;
  SWorkThrd*  pThrd = item->pThrd;
  SSvrConn*   conn = NULL;
  queue       wq;
dengyihao's avatar
dengyihao 已提交
521

dengyihao's avatar
dengyihao 已提交
522
  // batch process to avoid to lock/unlock frequently
wafwerar's avatar
wafwerar 已提交
523
  taosThreadMutexLock(&item->mtx);
dengyihao's avatar
dengyihao 已提交
524
  QUEUE_MOVE(&item->qmsg, &wq);
wafwerar's avatar
wafwerar 已提交
525
  taosThreadMutexUnlock(&item->mtx);
dengyihao's avatar
dengyihao 已提交
526 527 528 529

  while (!QUEUE_IS_EMPTY(&wq)) {
    queue* head = QUEUE_HEAD(&wq);
    QUEUE_REMOVE(head);
dengyihao's avatar
dengyihao 已提交
530

dengyihao's avatar
dengyihao 已提交
531
    SSvrMsg* msg = QUEUE_DATA(head, SSvrMsg, q);
dengyihao's avatar
dengyihao 已提交
532
    if (msg == NULL) {
dengyihao's avatar
dengyihao 已提交
533
      tError("unexcept occurred, continue");
dengyihao's avatar
dengyihao 已提交
534
      continue;
dengyihao's avatar
dengyihao 已提交
535
    }
dengyihao's avatar
dengyihao 已提交
536

dengyihao's avatar
dengyihao 已提交
537
    // release handle to rpc init
dengyihao's avatar
dengyihao 已提交
538 539 540 541 542
    if (msg->type == Quit) {
      (*transAsyncHandle[msg->type])(msg, pThrd);
    } else {
      STransMsg transMsg = msg->msg;

S
Shengliang Guan 已提交
543 544
      SExHandle* exh1 = transMsg.info.handle;
      int64_t    refId = transMsg.info.refId;
dengyihao's avatar
dengyihao 已提交
545
      SExHandle* exh2 = transAcquireExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
546
      if (exh2 == NULL || exh1 != exh2) {
dengyihao's avatar
dengyihao 已提交
547
        tTrace("handle except msg %p, ignore it", exh1);
dengyihao's avatar
dengyihao 已提交
548
        transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
549 550 551 552
        destroySmsg(msg);
        continue;
      }
      msg->pConn = exh1->handle;
dengyihao's avatar
dengyihao 已提交
553
      transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
554
      (*transAsyncHandle[msg->type])(msg, pThrd);
dengyihao's avatar
dengyihao 已提交
555
    }
dengyihao's avatar
dengyihao 已提交
556 557
  }
}
dengyihao's avatar
dengyihao 已提交
558 559 560 561 562
static void uvWalkCb(uv_handle_t* handle, void* arg) {
  if (!uv_is_closing(handle)) {
    uv_close(handle, NULL);
  }
}
dengyihao's avatar
dengyihao 已提交
563 564 565 566
static void uvFreeCb(uv_handle_t* handle) {
  //
  taosMemoryFree(handle);
}
dengyihao's avatar
dengyihao 已提交
567

dengyihao's avatar
dengyihao 已提交
568 569
static void uvAcceptAsyncCb(uv_async_t* async) {
  SServerObj* srv = async->data;
dengyihao's avatar
dengyihao 已提交
570
  tDebug("close server port %d", srv->port);
dengyihao's avatar
dengyihao 已提交
571
  uv_walk(srv->loop, uvWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
572
}
dengyihao's avatar
dengyihao 已提交
573

dengyihao's avatar
dengyihao 已提交
574
static void uvShutDownCb(uv_shutdown_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
575
  if (status != 0) {
S
Shengliang Guan 已提交
576
    tDebug("conn failed to shut down:%s", uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
577
  }
dengyihao's avatar
dengyihao 已提交
578
  uv_close((uv_handle_t*)req->handle, uvDestroyConn);
wafwerar's avatar
wafwerar 已提交
579
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
580
}
dengyihao's avatar
dengyihao 已提交
581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609
static bool uvRecvReleaseReq(SSvrConn* pConn, STransMsgHead* pHead) {
  if ((pHead)->release == 1 && (pHead->msgLen) == sizeof(*pHead)) {
    reallocConnRef(pConn);
    tTrace("conn %p received release request", pConn);

    STraceId traceId = pHead->traceId;
    pConn->status = ConnRelease;
    transClearBuffer(&pConn->readBuf);
    transFreeMsg(transContFromHead((char*)pHead));

    STransMsg tmsg = {.code = 0, .info.handle = (void*)pConn, .info.traceId = traceId, .info.ahandle = (void*)0x9527};
    SSvrMsg*  srvMsg = taosMemoryCalloc(1, sizeof(SSvrMsg));
    srvMsg->msg = tmsg;
    srvMsg->type = Release;
    srvMsg->pConn = pConn;
    if (!transQueuePush(&pConn->srvMsgs, srvMsg)) {
      return true;
    }
    if (pConn->regArg.init) {
      tTrace("conn %p release, notify server app", pConn);
      STrans* pTransInst = pConn->pTransInst;
      (*pTransInst->cfp)(pTransInst->parent, &(pConn->regArg.msg), NULL);
      memset(&pConn->regArg, 0, sizeof(pConn->regArg));
    }
    uvStartSendRespImpl(srvMsg);
    return true;
  }
  return false;
}
dengyihao's avatar
dengyihao 已提交
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655
static void uvPrepareCb(uv_prepare_t* handle) {
  // prepare callback
  SWorkThrd*  pThrd = handle->data;
  SAsyncPool* pool = pThrd->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);

    while (!QUEUE_IS_EMPTY(&wq)) {
      queue* head = QUEUE_HEAD(&wq);
      QUEUE_REMOVE(head);

      SSvrMsg* msg = QUEUE_DATA(head, SSvrMsg, q);
      if (msg == NULL) {
        tError("unexcept occurred, continue");
        continue;
      }
      // release handle to rpc init
      if (msg->type == Quit) {
        (*transAsyncHandle[msg->type])(msg, pThrd);
        continue;
      } else {
        STransMsg transMsg = msg->msg;

        SExHandle* exh1 = transMsg.info.handle;
        int64_t    refId = transMsg.info.refId;
        SExHandle* exh2 = transAcquireExHandle(transGetRefMgt(), refId);
        if (exh2 == NULL || exh1 != exh2) {
          tTrace("handle except msg %p, ignore it", exh1);
          transReleaseExHandle(transGetRefMgt(), refId);
          destroySmsg(msg);
          continue;
        }
        msg->pConn = exh1->handle;
        transReleaseExHandle(transGetRefMgt(), refId);
        (*transAsyncHandle[msg->type])(msg, pThrd);
      }
    }
  }
}
dengyihao's avatar
dengyihao 已提交
656

dengyihao's avatar
dengyihao 已提交
657 658 659
static void uvWorkDoTask(uv_work_t* req) {
  // doing time-consumeing task
  // only auth conn currently, add more func later
dengyihao's avatar
dengyihao 已提交
660
  tTrace("conn %p start to be processed in BG Thread", req->data);
dengyihao's avatar
dengyihao 已提交
661 662 663 664 665
  return;
}

static void uvWorkAfterTask(uv_work_t* req, int status) {
  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
666
    tTrace("conn %p failed to processed ", req->data);
dengyihao's avatar
dengyihao 已提交
667 668 669 670
  }
  // Done time-consumeing task
  // add more func later
  // this func called in main loop
dengyihao's avatar
dengyihao 已提交
671
  tTrace("conn %p already processed ", req->data);
dengyihao's avatar
dengyihao 已提交
672 673 674
  taosMemoryFree(req);
}

dengyihao's avatar
dengyihao 已提交
675 676 677 678 679 680
void uvOnAcceptCb(uv_stream_t* stream, int status) {
  if (status == -1) {
    return;
  }
  SServerObj* pObj = container_of(stream, SServerObj, server);

wafwerar's avatar
wafwerar 已提交
681
  uv_tcp_t* cli = (uv_tcp_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
682
  if (cli == NULL) return;
dengyihao's avatar
dengyihao 已提交
683

dengyihao's avatar
dengyihao 已提交
684 685 686 687 688 689 690 691
  int err = uv_tcp_init(pObj->loop, cli);
  if (err != 0) {
    tError("failed to create tcp: %s", uv_err_name(err));
    taosMemoryFree(cli);
    return;
  }
  err = uv_accept(stream, (uv_stream_t*)cli);
  if (err == 0) {
dengyihao's avatar
dengyihao 已提交
692
#if defined(WINDOWS) || defined(DARWIN)
wafwerar's avatar
wafwerar 已提交
693
    if (pObj->numOfWorkerReady < pObj->numOfThreads) {
dengyihao's avatar
dengyihao 已提交
694 695
      tError("worker-threads are not ready for all, need %d instead of %d.", pObj->numOfThreads,
             pObj->numOfWorkerReady);
dengyihao's avatar
dengyihao 已提交
696
      uv_close((uv_handle_t*)cli, uvFreeCb);
wafwerar's avatar
wafwerar 已提交
697 698
      return;
    }
dengyihao's avatar
dengyihao 已提交
699
#endif
dengyihao's avatar
dengyihao 已提交
700

wafwerar's avatar
wafwerar 已提交
701
    uv_write_t* wr = (uv_write_t*)taosMemoryMalloc(sizeof(uv_write_t));
dengyihao's avatar
dengyihao 已提交
702
    wr->data = cli;
dengyihao's avatar
dengyihao 已提交
703 704 705
    uv_buf_t buf = uv_buf_init((char*)notify, strlen(notify));

    pObj->workerIdx = (pObj->workerIdx + 1) % pObj->numOfThreads;
dengyihao's avatar
dengyihao 已提交
706

dengyihao's avatar
dengyihao 已提交
707
    tTrace("new connection accepted by main server, dispatch to %dth worker-thread", pObj->workerIdx);
dengyihao's avatar
dengyihao 已提交
708

dengyihao's avatar
dengyihao 已提交
709
    uv_write2(wr, (uv_stream_t*)&(pObj->pipe[pObj->workerIdx][0]), &buf, 1, (uv_stream_t*)cli, uvOnPipeWriteCb);
dengyihao's avatar
dengyihao 已提交
710
  } else {
dengyihao's avatar
dengyihao 已提交
711
    if (!uv_is_closing((uv_handle_t*)cli)) {
dengyihao's avatar
dengyihao 已提交
712 713
      tError("failed to accept tcp: %s", uv_err_name(err));
      uv_close((uv_handle_t*)cli, uvFreeCb);
dengyihao's avatar
dengyihao 已提交
714
    } else {
dengyihao's avatar
dengyihao 已提交
715
      tError("failed to accept tcp: %s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
716 717
      taosMemoryFree(cli);
    }
dengyihao's avatar
dengyihao 已提交
718 719 720
  }
}
void uvOnConnectionCb(uv_stream_t* q, ssize_t nread, const uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
721
  tTrace("connection coming");
dengyihao's avatar
dengyihao 已提交
722 723 724 725 726
  if (nread < 0) {
    if (nread != UV_EOF) {
      tError("read error %s", uv_err_name(nread));
    }
    // TODO(log other failure reason)
S
Shengliang Guan 已提交
727
    tWarn("failed to create connect:%p", q);
dengyihao's avatar
dengyihao 已提交
728 729
    taosMemoryFree(buf->base);
    uv_close((uv_handle_t*)q, NULL);
dengyihao's avatar
dengyihao 已提交
730 731 732
    return;
  }
  // free memory allocated by
dengyihao's avatar
dengyihao 已提交
733 734
  ASSERTS(nread == strlen(notify), "trans-svr mem corrupted");
  ASSERTS(buf->base[0] == notify[0], "trans-svr mem corrupted");
wafwerar's avatar
wafwerar 已提交
735
  taosMemoryFree(buf->base);
dengyihao's avatar
dengyihao 已提交
736

dengyihao's avatar
dengyihao 已提交
737
  SWorkThrd* pThrd = q->data;
dengyihao's avatar
dengyihao 已提交
738 739 740 741 742 743 744 745 746

  uv_pipe_t* pipe = (uv_pipe_t*)q;
  if (!uv_pipe_pending_count(pipe)) {
    tError("No pending count");
    return;
  }

  uv_handle_type pending = uv_pipe_pending_type(pipe);

dengyihao's avatar
dengyihao 已提交
747
  SSvrConn* pConn = createConn(pThrd);
748

dengyihao's avatar
dengyihao 已提交
749
  pConn->pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
750
  /* init conn timer*/
dengyihao's avatar
dengyihao 已提交
751 752
  // uv_timer_init(pThrd->loop, &pConn->pTimer);
  // pConn->pTimer.data = pConn;
dengyihao's avatar
dengyihao 已提交
753 754 755 756

  pConn->hostThrd = pThrd;

  // init client handle
wafwerar's avatar
wafwerar 已提交
757
  pConn->pTcp = (uv_tcp_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
758 759 760
  uv_tcp_init(pThrd->loop, pConn->pTcp);
  pConn->pTcp->data = pConn;

dengyihao's avatar
dengyihao 已提交
761 762
  transSetConnOption((uv_tcp_t*)pConn->pTcp);

dengyihao's avatar
dengyihao 已提交
763 764 765
  if (uv_accept(q, (uv_stream_t*)(pConn->pTcp)) == 0) {
    uv_os_fd_t fd;
    uv_fileno((const uv_handle_t*)pConn->pTcp, &fd);
S
Shengliang Guan 已提交
766
    tTrace("conn %p created, fd:%d", pConn, fd);
dengyihao's avatar
dengyihao 已提交
767

dengyihao's avatar
dengyihao 已提交
768 769 770
    struct sockaddr peername, sockname;
    int             addrlen = sizeof(peername);
    if (0 != uv_tcp_getpeername(pConn->pTcp, (struct sockaddr*)&peername, &addrlen)) {
dengyihao's avatar
dengyihao 已提交
771
      tError("conn %p failed to get peer info", pConn);
dengyihao's avatar
dengyihao 已提交
772
      transUnrefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
773
      return;
dengyihao's avatar
dengyihao 已提交
774
    }
dengyihao's avatar
dengyihao 已提交
775
    transSockInfo2Str(&peername, pConn->dst);
dengyihao's avatar
dengyihao 已提交
776

dengyihao's avatar
dengyihao 已提交
777 778
    addrlen = sizeof(sockname);
    if (0 != uv_tcp_getsockname(pConn->pTcp, (struct sockaddr*)&sockname, &addrlen)) {
dengyihao's avatar
dengyihao 已提交
779
      tError("conn %p failed to get local info", pConn);
dengyihao's avatar
dengyihao 已提交
780
      transUnrefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
781 782
      return;
    }
dengyihao's avatar
dengyihao 已提交
783
    transSockInfo2Str(&sockname, pConn->src);
dengyihao's avatar
dengyihao 已提交
784

dengyihao's avatar
dengyihao 已提交
785
    struct sockaddr_in addr = *(struct sockaddr_in*)&peername;
dengyihao's avatar
dengyihao 已提交
786 787
    pConn->clientIp = addr.sin_addr.s_addr;
    pConn->port = ntohs(addr.sin_port);
dengyihao's avatar
dengyihao 已提交
788

dengyihao's avatar
dengyihao 已提交
789
    uv_read_start((uv_stream_t*)(pConn->pTcp), uvAllocRecvBufferCb, uvOnRecvCb);
dengyihao's avatar
dengyihao 已提交
790

dengyihao's avatar
dengyihao 已提交
791
  } else {
dengyihao's avatar
dengyihao 已提交
792
    tDebug("failed to create new connection");
dengyihao's avatar
dengyihao 已提交
793
    transUnrefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
794 795 796
  }
}

797
void* transAcceptThread(void* arg) {
dengyihao's avatar
dengyihao 已提交
798
  // opt
dengyihao's avatar
dengyihao 已提交
799
  setThreadName("trans-accept");
dengyihao's avatar
dengyihao 已提交
800 801
  SServerObj* srv = (SServerObj*)arg;
  uv_run(srv->loop, UV_RUN_DEFAULT);
802 803

  return NULL;
dengyihao's avatar
dengyihao 已提交
804
}
dengyihao's avatar
dengyihao 已提交
805
void uvOnPipeConnectionCb(uv_connect_t* connect, int status) {
wafwerar's avatar
wafwerar 已提交
806 807 808
  if (status != 0) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
809
  SWorkThrd* pThrd = container_of(connect, SWorkThrd, connect_req);
wafwerar's avatar
wafwerar 已提交
810 811
  uv_read_start((uv_stream_t*)pThrd->pipe, uvAllocConnBufferCb, uvOnConnectionCb);
}
dengyihao's avatar
dengyihao 已提交
812
static bool addHandleToWorkloop(SWorkThrd* pThrd, char* pipeName) {
wafwerar's avatar
wafwerar 已提交
813
  pThrd->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
814 815 816
  if (0 != uv_loop_init(pThrd->loop)) {
    return false;
  }
dengyihao's avatar
dengyihao 已提交
817

dengyihao's avatar
dengyihao 已提交
818
#if defined(WINDOWS) || defined(DARWIN)
dengyihao's avatar
dengyihao 已提交
819
  uv_pipe_init(pThrd->loop, pThrd->pipe, 1);
dengyihao's avatar
dengyihao 已提交
820 821 822 823
#else
  uv_pipe_init(pThrd->loop, pThrd->pipe, 1);
  uv_pipe_open(pThrd->pipe, pThrd->fd);
#endif
dengyihao's avatar
dengyihao 已提交
824 825 826

  pThrd->pipe->data = pThrd;

dengyihao's avatar
dengyihao 已提交
827
  QUEUE_INIT(&pThrd->msg);
dengyihao's avatar
dengyihao 已提交
828

dengyihao's avatar
dengyihao 已提交
829 830 831 832 833
  pThrd->prepare = taosMemoryCalloc(1, sizeof(uv_prepare_t));
  uv_prepare_init(pThrd->loop, pThrd->prepare);
  uv_prepare_start(pThrd->prepare, uvPrepareCb);
  pThrd->prepare->data = pThrd;

dengyihao's avatar
fix bug  
dengyihao 已提交
834 835 836
  // conn set
  QUEUE_INIT(&pThrd->conn);

dengyihao's avatar
dengyihao 已提交
837
  pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 8, pThrd, uvWorkerAsyncCb);
dengyihao's avatar
dengyihao 已提交
838
#if defined(WINDOWS) || defined(DARWIN)
wafwerar's avatar
wafwerar 已提交
839
  uv_pipe_connect(&pThrd->connect_req, pThrd->pipe, pipeName, uvOnPipeConnectionCb);
dengyihao's avatar
dengyihao 已提交
840 841 842
#else
  uv_read_start((uv_stream_t*)pThrd->pipe, uvAllocConnBufferCb, uvOnConnectionCb);
#endif
dengyihao's avatar
dengyihao 已提交
843 844 845 846 847 848 849 850 851
  return true;
}

static bool addHandleToAcceptloop(void* arg) {
  // impl later
  SServerObj* srv = arg;

  int err = 0;
  if ((err = uv_tcp_init(srv->loop, &srv->server)) != 0) {
S
Shengliang Guan 已提交
852
    tError("failed to init accept server:%s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
853 854 855
    return false;
  }

dengyihao's avatar
dengyihao 已提交
856
  // register an async here to quit server gracefully
wafwerar's avatar
wafwerar 已提交
857
  srv->pAcceptAsync = taosMemoryCalloc(1, sizeof(uv_async_t));
dengyihao's avatar
dengyihao 已提交
858 859
  uv_async_init(srv->loop, srv->pAcceptAsync, uvAcceptAsyncCb);
  srv->pAcceptAsync->data = srv;
dengyihao's avatar
dengyihao 已提交
860

dengyihao's avatar
dengyihao 已提交
861
  struct sockaddr_in bind_addr;
dengyihao's avatar
dengyihao 已提交
862 863
  uv_ip4_addr("0.0.0.0", srv->port, &bind_addr);
  if ((err = uv_tcp_bind(&srv->server, (const struct sockaddr*)&bind_addr, 0)) != 0) {
S
Shengliang Guan 已提交
864
    tError("failed to bind:%s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
865 866
    return false;
  }
dengyihao's avatar
dengyihao 已提交
867
  if ((err = uv_listen((uv_stream_t*)&srv->server, 4096 * 2, uvOnAcceptCb)) != 0) {
S
Shengliang Guan 已提交
868
    tError("failed to listen:%s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
869
    terrno = TSDB_CODE_RPC_PORT_EADDRINUSE;
dengyihao's avatar
dengyihao 已提交
870 871 872
    return false;
  }
  return true;
dengyihao's avatar
dengyihao 已提交
873
}
874
void* transWorkerThread(void* arg) {
G
gccgdb1234 已提交
875
  setThreadName("trans-svr-work");
dengyihao's avatar
dengyihao 已提交
876
  SWorkThrd* pThrd = (SWorkThrd*)arg;
dengyihao's avatar
dengyihao 已提交
877
  uv_run(pThrd->loop, UV_RUN_DEFAULT);
878 879

  return NULL;
dengyihao's avatar
dengyihao 已提交
880 881
}

dengyihao's avatar
dengyihao 已提交
882
static FORCE_INLINE SSvrConn* createConn(void* hThrd) {
dengyihao's avatar
dengyihao 已提交
883
  SWorkThrd* pThrd = hThrd;
dengyihao's avatar
fix bug  
dengyihao 已提交
884

dengyihao's avatar
dengyihao 已提交
885
  SSvrConn* pConn = (SSvrConn*)taosMemoryCalloc(1, sizeof(SSvrConn));
dengyihao's avatar
dengyihao 已提交
886

dengyihao's avatar
dengyihao 已提交
887
  transReqQueueInit(&pConn->wreqQueue);
dengyihao's avatar
fix bug  
dengyihao 已提交
888 889 890
  QUEUE_INIT(&pConn->queue);

  QUEUE_PUSH(&pThrd->conn, &pConn->queue);
dengyihao's avatar
dengyihao 已提交
891 892

  transQueueInit(&pConn->srvMsgs, NULL);
dengyihao's avatar
dengyihao 已提交
893

dengyihao's avatar
dengyihao 已提交
894
  memset(&pConn->regArg, 0, sizeof(pConn->regArg));
dengyihao's avatar
dengyihao 已提交
895
  pConn->broken = false;
dengyihao's avatar
dengyihao 已提交
896
  pConn->status = ConnNormal;
dengyihao's avatar
opt rpc  
dengyihao 已提交
897
  transInitBuffer(&pConn->readBuf);
dengyihao's avatar
dengyihao 已提交
898

dengyihao's avatar
dengyihao 已提交
899 900 901
  SExHandle* exh = taosMemoryMalloc(sizeof(SExHandle));
  exh->handle = pConn;
  exh->pThrd = pThrd;
dengyihao's avatar
dengyihao 已提交
902 903
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
  transAcquireExHandle(transGetRefMgt(), exh->refId);
dengyihao's avatar
dengyihao 已提交
904

dengyihao's avatar
dengyihao 已提交
905
  STrans* pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
906
  pConn->refId = exh->refId;
dengyihao's avatar
dengyihao 已提交
907
  transRefSrvHandle(pConn);
S
Shengliang Guan 已提交
908
  tTrace("%s handle %p, conn %p created, refId:%" PRId64, transLabel(pTransInst), exh, pConn, pConn->refId);
dengyihao's avatar
dengyihao 已提交
909 910
  return pConn;
}
dengyihao's avatar
dengyihao 已提交
911

dengyihao's avatar
dengyihao 已提交
912
static FORCE_INLINE void destroyConn(SSvrConn* conn, bool clear) {
dengyihao's avatar
dengyihao 已提交
913 914 915
  if (conn == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
916

917
  if (clear) {
dengyihao's avatar
dengyihao 已提交
918
    if (!uv_is_closing((uv_handle_t*)conn->pTcp)) {
dengyihao's avatar
dengyihao 已提交
919 920 921
      tTrace("conn %p to be destroyed", conn);
      uv_close((uv_handle_t*)conn->pTcp, uvDestroyConn);
    }
922
  }
dengyihao's avatar
dengyihao 已提交
923
}
dengyihao's avatar
dengyihao 已提交
924
static FORCE_INLINE void destroyConnRegArg(SSvrConn* conn) {
dengyihao's avatar
dengyihao 已提交
925 926 927 928 929
  if (conn->regArg.init == 1) {
    transFreeMsg(conn->regArg.msg.pCont);
    conn->regArg.init = 0;
  }
}
dengyihao's avatar
dengyihao 已提交
930
static int reallocConnRef(SSvrConn* conn) {
dengyihao's avatar
dengyihao 已提交
931 932
  transReleaseExHandle(transGetRefMgt(), conn->refId);
  transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
933 934 935 936
  // avoid app continue to send msg on invalid handle
  SExHandle* exh = taosMemoryMalloc(sizeof(SExHandle));
  exh->handle = conn;
  exh->pThrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
937 938
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
  transAcquireExHandle(transGetRefMgt(), exh->refId);
dengyihao's avatar
dengyihao 已提交
939 940 941 942
  conn->refId = exh->refId;

  return 0;
}
dengyihao's avatar
dengyihao 已提交
943
static void uvDestroyConn(uv_handle_t* handle) {
dengyihao's avatar
dengyihao 已提交
944
  SSvrConn* conn = handle->data;
dengyihao's avatar
dengyihao 已提交
945

dengyihao's avatar
dengyihao 已提交
946 947 948
  if (conn == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
949
  SWorkThrd* thrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
950

dengyihao's avatar
dengyihao 已提交
951 952
  transReleaseExHandle(transGetRefMgt(), conn->refId);
  transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
953

dengyihao's avatar
dengyihao 已提交
954 955
  STrans* pTransInst = thrd->pTransInst;
  tDebug("%s conn %p destroy", transLabel(pTransInst), conn);
dengyihao's avatar
dengyihao 已提交
956

dengyihao's avatar
dengyihao 已提交
957 958 959 960
  for (int i = 0; i < transQueueSize(&conn->srvMsgs); i++) {
    SSvrMsg* msg = transQueueGet(&conn->srvMsgs, i);
    destroySmsg(msg);
  }
dengyihao's avatar
dengyihao 已提交
961
  transQueueDestroy(&conn->srvMsgs);
dengyihao's avatar
dengyihao 已提交
962
  transReqQueueClear(&conn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
963

dengyihao's avatar
dengyihao 已提交
964
  QUEUE_REMOVE(&conn->queue);
wafwerar's avatar
wafwerar 已提交
965
  taosMemoryFree(conn->pTcp);
dengyihao's avatar
dengyihao 已提交
966
  destroyConnRegArg(conn);
dengyihao's avatar
dengyihao 已提交
967
  transDestroyBuffer(&conn->readBuf);
dengyihao's avatar
dengyihao 已提交
968
  taosMemoryFree(conn);
dengyihao's avatar
dengyihao 已提交
969

dengyihao's avatar
dengyihao 已提交
970
  if (thrd->quit && QUEUE_IS_EMPTY(&thrd->conn)) {
dengyihao's avatar
dengyihao 已提交
971
    tTrace("work thread quit");
dengyihao's avatar
dengyihao 已提交
972
    uv_walk(thrd->loop, uvWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
973
  }
dengyihao's avatar
dengyihao 已提交
974
}
wafwerar's avatar
wafwerar 已提交
975
static void uvPipeListenCb(uv_stream_t* handle, int status) {
dengyihao's avatar
dengyihao 已提交
976
  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
977
    tError("server failed to init pipe, errmsg: %s", uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
978 979
    return;
  }
wafwerar's avatar
wafwerar 已提交
980 981

  SServerObj* srv = container_of(handle, SServerObj, pipeListen);
dengyihao's avatar
dengyihao 已提交
982
  uv_pipe_t*  pipe = &(srv->pipe[srv->numOfWorkerReady][0]);
wafwerar's avatar
wafwerar 已提交
983

dengyihao's avatar
dengyihao 已提交
984
  int ret = uv_pipe_init(srv->loop, pipe, 1);
dengyihao's avatar
dengyihao 已提交
985 986
  ASSERTS(ret == 0, "trans-svr failed to init pipe");
  if (ret != 0) return;
wafwerar's avatar
wafwerar 已提交
987

dengyihao's avatar
dengyihao 已提交
988
  ret = uv_accept((uv_stream_t*)&srv->pipeListen, (uv_stream_t*)pipe);
dengyihao's avatar
dengyihao 已提交
989 990
  ASSERTS(ret == 0, "trans-svr failed to accept pipe msg");
  if (ret != 0) return;
dengyihao's avatar
dengyihao 已提交
991 992

  ret = uv_is_readable((uv_stream_t*)pipe);
dengyihao's avatar
dengyihao 已提交
993 994
  ASSERTS(ret == 1, "trans-svr pipe status corrupted");
  if (ret != 1) return;
wafwerar's avatar
wafwerar 已提交
995

dengyihao's avatar
dengyihao 已提交
996
  ret = uv_is_writable((uv_stream_t*)pipe);
dengyihao's avatar
dengyihao 已提交
997
  ASSERTS(ret == 1, "trans-svr pipe status corrupted");
dengyihao's avatar
dengyihao 已提交
998
  if (ret != 1) return;
wafwerar's avatar
wafwerar 已提交
999

dengyihao's avatar
dengyihao 已提交
1000
  ret = uv_is_closing((uv_handle_t*)pipe);
dengyihao's avatar
dengyihao 已提交
1001 1002
  ASSERTS(ret == 0, "trans-svr pipe status corrupted");
  if (ret != 0) return;
dengyihao's avatar
dengyihao 已提交
1003 1004

  srv->numOfWorkerReady++;
wafwerar's avatar
wafwerar 已提交
1005
}
dengyihao's avatar
dengyihao 已提交
1006

U
ubuntu 已提交
1007
void* transInitServer(uint32_t ip, uint32_t port, char* label, int numOfThreads, void* fp, void* shandle) {
wafwerar's avatar
wafwerar 已提交
1008 1009
  SServerObj* srv = taosMemoryCalloc(1, sizeof(SServerObj));
  srv->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
1010 1011
  srv->numOfThreads = numOfThreads;
  srv->workerIdx = 0;
wafwerar's avatar
wafwerar 已提交
1012
  srv->numOfWorkerReady = 0;
dengyihao's avatar
dengyihao 已提交
1013
  srv->pThreadObj = (SWorkThrd**)taosMemoryCalloc(srv->numOfThreads, sizeof(SWorkThrd*));
wafwerar's avatar
wafwerar 已提交
1014
  srv->pipe = (uv_pipe_t**)taosMemoryCalloc(srv->numOfThreads, sizeof(uv_pipe_t*));
dengyihao's avatar
dengyihao 已提交
1015 1016 1017 1018
  srv->ip = ip;
  srv->port = port;
  uv_loop_init(srv->loop);

dengyihao's avatar
dengyihao 已提交
1019
  char pipeName[PATH_MAX];
1020 1021 1022 1023 1024 1025 1026

  if (false == taosValidIpAndPort(srv->ip, srv->port)) {
    terrno = TAOS_SYSTEM_ERROR(errno);
    tError("invalid ip/port, %d:%d, reason:%s", srv->ip, srv->port, terrstr());
    goto End;
  }

dengyihao's avatar
dengyihao 已提交
1027
#if defined(WINDOWS) || defined(DARWIN)
dengyihao's avatar
dengyihao 已提交
1028
  int ret = uv_pipe_init(srv->loop, &srv->pipeListen, 0);
dengyihao's avatar
dengyihao 已提交
1029 1030 1031 1032
  if (ret != 0) {
    tError("failed to init pipe, errmsg: %s", uv_err_name(ret));
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1033
#if defined(WINDOWS)
dengyihao's avatar
dengyihao 已提交
1034
  snprintf(pipeName, sizeof(pipeName), "\\\\?\\pipe\\trans.rpc.%d-%" PRIu64, taosSafeRand(), GetCurrentProcessId());
dengyihao's avatar
dengyihao 已提交
1035 1036 1037 1038
#elif defined(DARWIN)
  snprintf(pipeName, sizeof(pipeName), "%s%spipe.trans.rpc.%08d-%" PRIu64, tsTempDir, TD_DIRSEP, taosSafeRand(),
           taosGetSelfPthreadId());
#endif
dengyihao's avatar
dengyihao 已提交
1039

dengyihao's avatar
dengyihao 已提交
1040
  ret = uv_pipe_bind(&srv->pipeListen, pipeName);
dengyihao's avatar
dengyihao 已提交
1041 1042 1043 1044
  if (ret != 0) {
    tError("failed to bind pipe, errmsg: %s", uv_err_name(ret));
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1045 1046

  ret = uv_listen((uv_stream_t*)&srv->pipeListen, SOMAXCONN, uvPipeListenCb);
dengyihao's avatar
dengyihao 已提交
1047 1048 1049 1050
  if (ret != 0) {
    tError("failed to listen pipe, errmsg: %s", uv_err_name(ret));
    goto End;
  }
wafwerar's avatar
wafwerar 已提交
1051

dengyihao's avatar
dengyihao 已提交
1052
  for (int i = 0; i < srv->numOfThreads; i++) {
dengyihao's avatar
dengyihao 已提交
1053
    SWorkThrd* thrd = (SWorkThrd*)taosMemoryCalloc(1, sizeof(SWorkThrd));
dengyihao's avatar
dengyihao 已提交
1054
    thrd->pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1055
    thrd->quit = false;
dengyihao's avatar
dengyihao 已提交
1056
    srv->pThreadObj[i] = thrd;
dengyihao's avatar
dengyihao 已提交
1057
    thrd->pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1058

wafwerar's avatar
wafwerar 已提交
1059
    srv->pipe[i] = (uv_pipe_t*)taosMemoryCalloc(2, sizeof(uv_pipe_t));
dengyihao's avatar
dengyihao 已提交
1060
    thrd->pipe = &(srv->pipe[i][1]);  // init read
dengyihao's avatar
dengyihao 已提交
1061

dengyihao's avatar
dengyihao 已提交
1062
    if (false == addHandleToWorkloop(thrd, pipeName)) {
dengyihao's avatar
dengyihao 已提交
1063 1064
      goto End;
    }
dengyihao's avatar
dengyihao 已提交
1065

dengyihao's avatar
dengyihao 已提交
1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
    int err = taosThreadCreate(&(thrd->thread), NULL, transWorkerThread, (void*)(thrd));
    if (err == 0) {
      tDebug("success to create worker-thread:%d", i);
    } else {
      // TODO: clear all other resource later
      tError("failed to create worker-thread:%d", i);
      goto End;
    }
  }
#else

  for (int i = 0; i < srv->numOfThreads; i++) {
    SWorkThrd* thrd = (SWorkThrd*)taosMemoryCalloc(1, sizeof(SWorkThrd));

    thrd->pTransInst = shandle;
    thrd->quit = false;
    thrd->pTransInst = shandle;

    srv->pipe[i] = (uv_pipe_t*)taosMemoryCalloc(2, sizeof(uv_pipe_t));
    srv->pThreadObj[i] = thrd;

    uv_os_sock_t fds[2];
dengyihao's avatar
dengyihao 已提交
1088
    if (uv_socketpair(SOCK_STREAM, 0, fds, UV_NONBLOCK_PIPE, UV_NONBLOCK_PIPE) != 0) {
dengyihao's avatar
dengyihao 已提交
1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
      goto End;
    }

    uv_pipe_init(srv->loop, &(srv->pipe[i][0]), 1);
    uv_pipe_open(&(srv->pipe[i][0]), fds[1]);

    thrd->pipe = &(srv->pipe[i][1]);  // init read
    thrd->fd = fds[0];

    if (false == addHandleToWorkloop(thrd, pipeName)) {
      goto End;
    }

1102
    int err = taosThreadCreate(&(thrd->thread), NULL, transWorkerThread, (void*)(thrd));
dengyihao's avatar
dengyihao 已提交
1103
    if (err == 0) {
S
Shengliang Guan 已提交
1104
      tDebug("success to create worker-thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1105 1106
    } else {
      // TODO: clear all other resource later
S
Shengliang Guan 已提交
1107
      tError("failed to create worker-thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1108
      goto End;
dengyihao's avatar
dengyihao 已提交
1109 1110
    }
  }
dengyihao's avatar
dengyihao 已提交
1111
#endif
dengyihao's avatar
dengyihao 已提交
1112

dengyihao's avatar
dengyihao 已提交
1113 1114 1115
  if (false == addHandleToAcceptloop(srv)) {
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1116

1117
  int err = taosThreadCreate(&srv->thread, NULL, transAcceptThread, (void*)srv);
dengyihao's avatar
dengyihao 已提交
1118 1119 1120
  if (err == 0) {
    tDebug("success to create accept-thread");
  } else {
dengyihao's avatar
dengyihao 已提交
1121 1122
    tError("failed  to create accept-thread");
    goto End;
dengyihao's avatar
dengyihao 已提交
1123 1124
    // clear all resource later
  }
dengyihao's avatar
dengyihao 已提交
1125

dengyihao's avatar
dengyihao 已提交
1126
  srv->inited = true;
dengyihao's avatar
dengyihao 已提交
1127
  return srv;
dengyihao's avatar
dengyihao 已提交
1128
End:
U
ubuntu 已提交
1129
  transCloseServer(srv);
dengyihao's avatar
dengyihao 已提交
1130 1131
  return NULL;
}
dengyihao's avatar
dengyihao 已提交
1132

dengyihao's avatar
dengyihao 已提交
1133
void uvHandleQuit(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1134
  thrd->quit = true;
dengyihao's avatar
dengyihao 已提交
1135
  if (QUEUE_IS_EMPTY(&thrd->conn)) {
dengyihao's avatar
dengyihao 已提交
1136
    uv_walk(thrd->loop, uvWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
1137 1138 1139
  } else {
    destroyAllConn(thrd);
  }
wafwerar's avatar
wafwerar 已提交
1140
  taosMemoryFree(msg);
dengyihao's avatar
dengyihao 已提交
1141
}
dengyihao's avatar
dengyihao 已提交
1142
void uvHandleRelease(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1143
  SSvrConn* conn = msg->pConn;
dengyihao's avatar
dengyihao 已提交
1144
  if (conn->status == ConnAcquire) {
dengyihao's avatar
dengyihao 已提交
1145
    reallocConnRef(conn);
dengyihao's avatar
dengyihao 已提交
1146
    if (!transQueuePush(&conn->srvMsgs, msg)) {
dengyihao's avatar
dengyihao 已提交
1147
      return;
dengyihao's avatar
dengyihao 已提交
1148
    }
dengyihao's avatar
dengyihao 已提交
1149
    uvStartSendRespImpl(msg);
dengyihao's avatar
dengyihao 已提交
1150
    return;
dengyihao's avatar
dengyihao 已提交
1151
  } else if (conn->status == ConnRelease || conn->status == ConnNormal) {
dengyihao's avatar
dengyihao 已提交
1152
    tDebug("%s conn %p already released, ignore release-msg", transLabel(thrd->pTransInst), conn);
dengyihao's avatar
dengyihao 已提交
1153
  }
dengyihao's avatar
dengyihao 已提交
1154
  destroySmsg(msg);
dengyihao's avatar
dengyihao 已提交
1155
}
dengyihao's avatar
dengyihao 已提交
1156
void uvHandleResp(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1157
  // send msg to client
dengyihao's avatar
dengyihao 已提交
1158
  tDebug("%s conn %p start to send resp (2/2)", transLabel(thrd->pTransInst), msg->pConn);
dengyihao's avatar
dengyihao 已提交
1159 1160
  uvStartSendResp(msg);
}
dengyihao's avatar
dengyihao 已提交
1161
void uvHandleRegister(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1162
  SSvrConn* conn = msg->pConn;
dengyihao's avatar
dengyihao 已提交
1163
  tDebug("%s conn %p register brokenlink callback", transLabel(thrd->pTransInst), conn);
dengyihao's avatar
dengyihao 已提交
1164
  if (conn->status == ConnAcquire) {
dengyihao's avatar
dengyihao 已提交
1165
    if (!transQueuePush(&conn->srvMsgs, msg)) {
dengyihao's avatar
dengyihao 已提交
1166 1167
      return;
    }
dengyihao's avatar
dengyihao 已提交
1168
    transQueuePop(&conn->srvMsgs);
dengyihao's avatar
dengyihao 已提交
1169 1170 1171
    conn->regArg.notifyCount = 0;
    conn->regArg.init = 1;
    conn->regArg.msg = msg->msg;
dengyihao's avatar
dengyihao 已提交
1172
    tDebug("conn %p register brokenlink callback succ", conn);
dengyihao's avatar
dengyihao 已提交
1173 1174 1175 1176 1177 1178

    if (conn->broken) {
      STrans* pTransInst = conn->pTransInst;
      (*pTransInst->cfp)(pTransInst->parent, &(conn->regArg.msg), NULL);
      memset(&conn->regArg, 0, sizeof(conn->regArg));
    }
wafwerar's avatar
wafwerar 已提交
1179
    taosMemoryFree(msg);
dengyihao's avatar
dengyihao 已提交
1180 1181
  }
}
dengyihao's avatar
dengyihao 已提交
1182
void destroyWorkThrd(SWorkThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1183 1184 1185
  if (pThrd == NULL) {
    return;
  }
wafwerar's avatar
wafwerar 已提交
1186
  taosThreadJoin(pThrd->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1187
  SRV_RELEASE_UV(pThrd->loop);
dengyihao's avatar
dengyihao 已提交
1188
  TRANS_DESTROY_ASYNC_POOL_MSG(pThrd->asyncPool, SSvrMsg, destroySmsg);
dengyihao's avatar
dengyihao 已提交
1189
  transAsyncPoolDestroy(pThrd->asyncPool);
dengyihao's avatar
dengyihao 已提交
1190
  taosMemoryFree(pThrd->prepare);
dengyihao's avatar
dengyihao 已提交
1191
  taosMemoryFree(pThrd->loop);
wafwerar's avatar
wafwerar 已提交
1192
  taosMemoryFree(pThrd);
dengyihao's avatar
dengyihao 已提交
1193
}
dengyihao's avatar
dengyihao 已提交
1194
void sendQuitToWorkThrd(SWorkThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1195
  SSvrMsg* msg = taosMemoryCalloc(1, sizeof(SSvrMsg));
dengyihao's avatar
dengyihao 已提交
1196
  msg->type = Quit;
dengyihao's avatar
dengyihao 已提交
1197
  tDebug("server send quit msg to work thread");
dengyihao's avatar
dengyihao 已提交
1198
  transAsyncSend(pThrd->asyncPool, &msg->q);
dengyihao's avatar
dengyihao 已提交
1199 1200
}

U
ubuntu 已提交
1201
void transCloseServer(void* arg) {
dengyihao's avatar
dengyihao 已提交
1202 1203
  // impl later
  SServerObj* srv = arg;
dengyihao's avatar
dengyihao 已提交
1204

dengyihao's avatar
dengyihao 已提交
1205
  if (srv->inited) {
1206
    tDebug("send quit msg to accept thread");
dengyihao's avatar
dengyihao 已提交
1207 1208
    uv_async_send(srv->pAcceptAsync);
    taosThreadJoin(srv->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1209
    SRV_RELEASE_UV(srv->loop);
dengyihao's avatar
dengyihao 已提交
1210

dengyihao's avatar
dengyihao 已提交
1211 1212 1213 1214
    for (int i = 0; i < srv->numOfThreads; i++) {
      sendQuitToWorkThrd(srv->pThreadObj[i]);
      destroyWorkThrd(srv->pThreadObj[i]);
    }
dengyihao's avatar
dengyihao 已提交
1215 1216
  } else {
    uv_loop_close(srv->loop);
dengyihao's avatar
dengyihao 已提交
1217 1218
  }

wafwerar's avatar
wafwerar 已提交
1219 1220 1221
  taosMemoryFree(srv->pThreadObj);
  taosMemoryFree(srv->pAcceptAsync);
  taosMemoryFree(srv->loop);
dengyihao's avatar
dengyihao 已提交
1222 1223

  for (int i = 0; i < srv->numOfThreads; i++) {
wafwerar's avatar
wafwerar 已提交
1224
    taosMemoryFree(srv->pipe[i]);
dengyihao's avatar
dengyihao 已提交
1225
  }
wafwerar's avatar
wafwerar 已提交
1226
  taosMemoryFree(srv->pipe);
dengyihao's avatar
dengyihao 已提交
1227

wafwerar's avatar
wafwerar 已提交
1228
  taosMemoryFree(srv);
dengyihao's avatar
dengyihao 已提交
1229
}
dengyihao's avatar
dengyihao 已提交
1230

dengyihao's avatar
dengyihao 已提交
1231 1232 1233 1234
void transRefSrvHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1235
  int ref = T_REF_INC((SSvrConn*)handle);
dengyihao's avatar
dengyihao 已提交
1236
  tTrace("conn %p ref count:%d", handle, ref);
dengyihao's avatar
dengyihao 已提交
1237 1238 1239 1240 1241 1242
}

void transUnrefSrvHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1243
  int ref = T_REF_DEC((SSvrConn*)handle);
dengyihao's avatar
dengyihao 已提交
1244
  tTrace("conn %p ref count:%d", handle, ref);
dengyihao's avatar
dengyihao 已提交
1245
  if (ref == 0) {
dengyihao's avatar
dengyihao 已提交
1246
    destroyConn((SSvrConn*)handle, true);
dengyihao's avatar
dengyihao 已提交
1247 1248
  }
}
dengyihao's avatar
dengyihao 已提交
1249

dengyihao's avatar
dengyihao 已提交
1250
int transReleaseSrvHandle(void* handle) {
dengyihao's avatar
dengyihao 已提交
1251 1252 1253
  SRpcHandleInfo* info = handle;
  SExHandle*      exh = info->handle;
  int64_t         refId = info->refId;
dengyihao's avatar
dengyihao 已提交
1254

dengyihao's avatar
dengyihao 已提交
1255
  ASYNC_CHECK_HANDLE(exh, refId);
dengyihao's avatar
dengyihao 已提交
1256

dengyihao's avatar
dengyihao 已提交
1257
  SWorkThrd* pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
1258 1259
  ASYNC_ERR_JRET(pThrd);

S
Shengliang Guan 已提交
1260
  STransMsg tmsg = {.code = 0, .info.handle = exh, .info.ahandle = NULL, .info.refId = refId};
dengyihao's avatar
dengyihao 已提交
1261

dengyihao's avatar
dengyihao 已提交
1262 1263 1264
  SSvrMsg* m = taosMemoryCalloc(1, sizeof(SSvrMsg));
  m->msg = tmsg;
  m->type = Release;
dengyihao's avatar
dengyihao 已提交
1265

dengyihao's avatar
dengyihao 已提交
1266
  tDebug("%s conn %p start to release", transLabel(pThrd->pTransInst), exh->handle);
1267 1268 1269
  if (0 != transAsyncSend(pThrd->asyncPool, &m->q)) {
    destroySmsg(m);
  }
H
Haojun Liao 已提交
1270

dengyihao's avatar
dengyihao 已提交
1271
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1272
  return 0;
dengyihao's avatar
dengyihao 已提交
1273
_return1:
dengyihao's avatar
dengyihao 已提交
1274
  tDebug("handle %p failed to send to release handle", exh);
dengyihao's avatar
dengyihao 已提交
1275
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1276
  return -1;
dengyihao's avatar
dengyihao 已提交
1277
_return2:
dengyihao's avatar
dengyihao 已提交
1278
  tDebug("handle %p failed to send to release handle", exh);
dengyihao's avatar
dengyihao 已提交
1279
  return -1;
dengyihao's avatar
dengyihao 已提交
1280
}
1281

dengyihao's avatar
dengyihao 已提交
1282
int transSendResponse(const STransMsg* msg) {
dengyihao's avatar
dengyihao 已提交
1283 1284 1285 1286 1287
  if (msg->info.noResp) {
    rpcFreeCont(msg->pCont);
    tTrace("no need send resp");
    return 0;
  }
S
Shengliang Guan 已提交
1288 1289
  SExHandle* exh = msg->info.handle;
  int64_t    refId = msg->info.refId;
dengyihao's avatar
dengyihao 已提交
1290 1291
  ASYNC_CHECK_HANDLE(exh, refId);

dengyihao's avatar
dengyihao 已提交
1292
  STransMsg tmsg = *msg;
S
Shengliang Guan 已提交
1293
  tmsg.info.refId = refId;
dengyihao's avatar
dengyihao 已提交
1294

dengyihao's avatar
dengyihao 已提交
1295
  SWorkThrd* pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
1296
  ASYNC_ERR_JRET(pThrd);
dengyihao's avatar
dengyihao 已提交
1297

dengyihao's avatar
dengyihao 已提交
1298 1299 1300
  SSvrMsg* m = taosMemoryCalloc(1, sizeof(SSvrMsg));
  m->msg = tmsg;
  m->type = Normal;
dengyihao's avatar
dengyihao 已提交
1301 1302

  STraceId* trace = (STraceId*)&msg->info.traceId;
dengyihao's avatar
dengyihao 已提交
1303
  tGDebug("conn %p start to send resp (1/2)", exh->handle);
1304 1305 1306
  if (0 != transAsyncSend(pThrd->asyncPool, &m->q)) {
    destroySmsg(m);
  }
H
Haojun Liao 已提交
1307

dengyihao's avatar
dengyihao 已提交
1308
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1309
  return 0;
H
Haojun Liao 已提交
1310

dengyihao's avatar
dengyihao 已提交
1311
_return1:
dengyihao's avatar
dengyihao 已提交
1312
  tDebug("handle %p failed to send resp", exh);
dengyihao's avatar
dengyihao 已提交
1313
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1314
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1315
  return -1;
dengyihao's avatar
dengyihao 已提交
1316
_return2:
dengyihao's avatar
dengyihao 已提交
1317
  tDebug("handle %p failed to send resp", exh);
dengyihao's avatar
dengyihao 已提交
1318
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1319
  return -1;
dengyihao's avatar
dengyihao 已提交
1320
}
dengyihao's avatar
dengyihao 已提交
1321
int transRegisterMsg(const STransMsg* msg) {
S
Shengliang Guan 已提交
1322 1323
  SExHandle* exh = msg->info.handle;
  int64_t    refId = msg->info.refId;
dengyihao's avatar
dengyihao 已提交
1324 1325
  ASYNC_CHECK_HANDLE(exh, refId);

dengyihao's avatar
dengyihao 已提交
1326
  STransMsg tmsg = *msg;
dengyihao's avatar
dengyihao 已提交
1327 1328
  tmsg.info.noResp = 1;

S
Shengliang Guan 已提交
1329
  tmsg.info.refId = refId;
dengyihao's avatar
dengyihao 已提交
1330

dengyihao's avatar
dengyihao 已提交
1331
  SWorkThrd* pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
1332
  ASYNC_ERR_JRET(pThrd);
dengyihao's avatar
dengyihao 已提交
1333

dengyihao's avatar
dengyihao 已提交
1334 1335 1336
  SSvrMsg* m = taosMemoryCalloc(1, sizeof(SSvrMsg));
  m->msg = tmsg;
  m->type = Register;
dengyihao's avatar
dengyihao 已提交
1337

dengyihao's avatar
dengyihao 已提交
1338
  STrans* pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1339
  tDebug("%s conn %p start to register brokenlink callback", transLabel(pTransInst), exh->handle);
1340 1341 1342
  if (0 != transAsyncSend(pThrd->asyncPool, &m->q)) {
    destroySmsg(m);
  }
H
Haojun Liao 已提交
1343

dengyihao's avatar
dengyihao 已提交
1344
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1345
  return 0;
dengyihao's avatar
dengyihao 已提交
1346

dengyihao's avatar
dengyihao 已提交
1347
_return1:
dengyihao's avatar
dengyihao 已提交
1348
  tDebug("handle %p failed to register brokenlink", exh);
dengyihao's avatar
dengyihao 已提交
1349
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1350
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1351
  return -1;
dengyihao's avatar
dengyihao 已提交
1352
_return2:
dengyihao's avatar
dengyihao 已提交
1353
  tDebug("handle %p failed to register brokenlink", exh);
dengyihao's avatar
dengyihao 已提交
1354
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1355
  return -1;
dengyihao's avatar
dengyihao 已提交
1356
}
dengyihao's avatar
dengyihao 已提交
1357

D
dapan1121 已提交
1358 1359
int transGetConnInfo(void* thandle, STransHandleInfo* pConnInfo) { return -1; }

dengyihao's avatar
dengyihao 已提交
1360
#endif