transSvr.c 39.3 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);
S
Shengliang Guan 已提交
270
  assert(transMsg.info.handle != NULL);
dengyihao's avatar
dengyihao 已提交
271

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

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

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

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

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

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

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

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

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

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

dengyihao's avatar
add UT  
dengyihao 已提交
352
      destroySmsg(msg);
dengyihao's avatar
dengyihao 已提交
353
      // send cached data
dengyihao's avatar
dengyihao 已提交
354
      if (!transQueueEmpty(&conn->srvMsgs)) {
dengyihao's avatar
dengyihao 已提交
355
        msg = (SSvrMsg*)transQueueGet(&conn->srvMsgs, 0);
dengyihao's avatar
dengyihao 已提交
356 357 358 359 360 361 362 363 364
        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 已提交
365
          transQueuePop(&conn->srvMsgs);
wafwerar's avatar
wafwerar 已提交
366
          taosMemoryFree(msg);
dengyihao's avatar
dengyihao 已提交
367

dengyihao's avatar
dengyihao 已提交
368
          msg = (SSvrMsg*)transQueueGet(&conn->srvMsgs, 0);
dengyihao's avatar
dengyihao 已提交
369
          if (msg != NULL) {
dengyihao's avatar
dengyihao 已提交
370
            uvStartSendRespImpl(msg);
dengyihao's avatar
dengyihao 已提交
371
          }
dengyihao's avatar
dengyihao 已提交
372
        } else {
dengyihao's avatar
dengyihao 已提交
373
          uvStartSendRespImpl(msg);
dengyihao's avatar
dengyihao 已提交
374
        }
dengyihao's avatar
add UT  
dengyihao 已提交
375
      }
dengyihao's avatar
dengyihao 已提交
376
    }
dengyihao's avatar
dengyihao 已提交
377
    transUnrefSrvHandle(conn);
dengyihao's avatar
dengyihao 已提交
378
  } else {
dengyihao's avatar
dengyihao 已提交
379 380 381 382 383
    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 已提交
384 385
  }
}
dengyihao's avatar
dengyihao 已提交
386 387
static void uvOnPipeWriteCb(uv_write_t* req, int status) {
  if (status == 0) {
dengyihao's avatar
dengyihao 已提交
388
    tTrace("success to dispatch conn to work thread");
dengyihao's avatar
dengyihao 已提交
389 390 391
  } else {
    tError("fail to dispatch conn to work thread");
  }
dengyihao's avatar
dengyihao 已提交
392 393 394 395 396
  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 已提交
397
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
398
}
dengyihao's avatar
dengyihao 已提交
399

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

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

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

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

dengyihao's avatar
dengyihao 已提交
439 440
  char*   msg = (char*)pHead;
  int32_t len = transMsgLenFromCont(pMsg->contLen);
dengyihao's avatar
dengyihao 已提交
441

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

dengyihao's avatar
dengyihao 已提交
571
static void uvShutDownCb(uv_shutdown_t* req, int status) {
dengyihao's avatar
dengyihao 已提交
572
  if (status != 0) {
S
Shengliang Guan 已提交
573
    tDebug("conn failed to shut down:%s", uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
574
  }
dengyihao's avatar
dengyihao 已提交
575
  uv_close((uv_handle_t*)req->handle, uvDestroyConn);
wafwerar's avatar
wafwerar 已提交
576
  taosMemoryFree(req);
dengyihao's avatar
dengyihao 已提交
577
}
dengyihao's avatar
dengyihao 已提交
578 579 580 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
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 已提交
607 608 609 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
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 已提交
653

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

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

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

wafwerar's avatar
wafwerar 已提交
678
  uv_tcp_t* cli = (uv_tcp_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
679 680 681
  uv_tcp_init(pObj->loop, cli);

  if (uv_accept(stream, (uv_stream_t*)cli) == 0) {
dengyihao's avatar
dengyihao 已提交
682
#if defined(WINDOWS) || defined(DARWIN)
wafwerar's avatar
wafwerar 已提交
683
    if (pObj->numOfWorkerReady < pObj->numOfThreads) {
dengyihao's avatar
dengyihao 已提交
684 685
      tError("worker-threads are not ready for all, need %d instead of %d.", pObj->numOfThreads,
             pObj->numOfWorkerReady);
wafwerar's avatar
wafwerar 已提交
686 687 688
      uv_close((uv_handle_t*)cli, NULL);
      return;
    }
dengyihao's avatar
dengyihao 已提交
689
#endif
dengyihao's avatar
dengyihao 已提交
690

wafwerar's avatar
wafwerar 已提交
691
    uv_write_t* wr = (uv_write_t*)taosMemoryMalloc(sizeof(uv_write_t));
dengyihao's avatar
dengyihao 已提交
692
    wr->data = cli;
dengyihao's avatar
dengyihao 已提交
693 694 695
    uv_buf_t buf = uv_buf_init((char*)notify, strlen(notify));

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

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

dengyihao's avatar
dengyihao 已提交
699
    uv_write2(wr, (uv_stream_t*)&(pObj->pipe[pObj->workerIdx][0]), &buf, 1, (uv_stream_t*)cli, uvOnPipeWriteCb);
dengyihao's avatar
dengyihao 已提交
700
  } else {
dengyihao's avatar
dengyihao 已提交
701 702 703 704 705
    if (!uv_is_closing((uv_handle_t*)cli)) {
      uv_close((uv_handle_t*)cli, NULL);
    } else {
      taosMemoryFree(cli);
    }
dengyihao's avatar
dengyihao 已提交
706 707 708
  }
}
void uvOnConnectionCb(uv_stream_t* q, ssize_t nread, const uv_buf_t* buf) {
dengyihao's avatar
dengyihao 已提交
709
  tTrace("connection coming");
dengyihao's avatar
dengyihao 已提交
710 711 712 713 714
  if (nread < 0) {
    if (nread != UV_EOF) {
      tError("read error %s", uv_err_name(nread));
    }
    // TODO(log other failure reason)
S
Shengliang Guan 已提交
715
    tWarn("failed to create connect:%p", q);
dengyihao's avatar
dengyihao 已提交
716 717
    taosMemoryFree(buf->base);
    uv_close((uv_handle_t*)q, NULL);
dengyihao's avatar
dengyihao 已提交
718 719 720 721 722
    return;
  }
  // free memory allocated by
  assert(nread == strlen(notify));
  assert(buf->base[0] == notify[0]);
wafwerar's avatar
wafwerar 已提交
723
  taosMemoryFree(buf->base);
dengyihao's avatar
dengyihao 已提交
724

dengyihao's avatar
dengyihao 已提交
725
  SWorkThrd* pThrd = q->data;
dengyihao's avatar
dengyihao 已提交
726 727 728 729 730 731 732 733 734 735

  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);
  assert(pending == UV_TCP);

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

dengyihao's avatar
dengyihao 已提交
738
  pConn->pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
739
  /* init conn timer*/
dengyihao's avatar
dengyihao 已提交
740 741
  // uv_timer_init(pThrd->loop, &pConn->pTimer);
  // pConn->pTimer.data = pConn;
dengyihao's avatar
dengyihao 已提交
742 743 744 745

  pConn->hostThrd = pThrd;

  // init client handle
wafwerar's avatar
wafwerar 已提交
746
  pConn->pTcp = (uv_tcp_t*)taosMemoryMalloc(sizeof(uv_tcp_t));
dengyihao's avatar
dengyihao 已提交
747 748 749
  uv_tcp_init(pThrd->loop, pConn->pTcp);
  pConn->pTcp->data = pConn;

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

dengyihao's avatar
dengyihao 已提交
752 753 754
  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 已提交
755
    tTrace("conn %p created, fd:%d", pConn, fd);
dengyihao's avatar
dengyihao 已提交
756

dengyihao's avatar
dengyihao 已提交
757 758 759
    struct sockaddr peername, sockname;
    int             addrlen = sizeof(peername);
    if (0 != uv_tcp_getpeername(pConn->pTcp, (struct sockaddr*)&peername, &addrlen)) {
dengyihao's avatar
dengyihao 已提交
760
      tError("conn %p failed to get peer info", pConn);
dengyihao's avatar
dengyihao 已提交
761
      transUnrefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
762
      return;
dengyihao's avatar
dengyihao 已提交
763
    }
dengyihao's avatar
dengyihao 已提交
764
    transSockInfo2Str(&peername, pConn->dst);
dengyihao's avatar
dengyihao 已提交
765

dengyihao's avatar
dengyihao 已提交
766 767
    addrlen = sizeof(sockname);
    if (0 != uv_tcp_getsockname(pConn->pTcp, (struct sockaddr*)&sockname, &addrlen)) {
dengyihao's avatar
dengyihao 已提交
768
      tError("conn %p failed to get local info", pConn);
dengyihao's avatar
dengyihao 已提交
769
      transUnrefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
770 771
      return;
    }
dengyihao's avatar
dengyihao 已提交
772
    transSockInfo2Str(&sockname, pConn->src);
dengyihao's avatar
dengyihao 已提交
773

dengyihao's avatar
dengyihao 已提交
774
    struct sockaddr_in addr = *(struct sockaddr_in*)&peername;
dengyihao's avatar
dengyihao 已提交
775 776
    pConn->clientIp = addr.sin_addr.s_addr;
    pConn->port = ntohs(addr.sin_port);
dengyihao's avatar
dengyihao 已提交
777

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

dengyihao's avatar
dengyihao 已提交
780
  } else {
dengyihao's avatar
dengyihao 已提交
781
    tDebug("failed to create new connection");
dengyihao's avatar
dengyihao 已提交
782
    transUnrefSrvHandle(pConn);
dengyihao's avatar
dengyihao 已提交
783 784 785
  }
}

786
void* transAcceptThread(void* arg) {
dengyihao's avatar
dengyihao 已提交
787
  // opt
dengyihao's avatar
dengyihao 已提交
788
  setThreadName("trans-accept");
dengyihao's avatar
dengyihao 已提交
789 790
  SServerObj* srv = (SServerObj*)arg;
  uv_run(srv->loop, UV_RUN_DEFAULT);
791 792

  return NULL;
dengyihao's avatar
dengyihao 已提交
793
}
dengyihao's avatar
dengyihao 已提交
794
void uvOnPipeConnectionCb(uv_connect_t* connect, int status) {
wafwerar's avatar
wafwerar 已提交
795 796 797
  if (status != 0) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
798
  SWorkThrd* pThrd = container_of(connect, SWorkThrd, connect_req);
wafwerar's avatar
wafwerar 已提交
799 800
  uv_read_start((uv_stream_t*)pThrd->pipe, uvAllocConnBufferCb, uvOnConnectionCb);
}
dengyihao's avatar
dengyihao 已提交
801
static bool addHandleToWorkloop(SWorkThrd* pThrd, char* pipeName) {
wafwerar's avatar
wafwerar 已提交
802
  pThrd->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
803 804 805
  if (0 != uv_loop_init(pThrd->loop)) {
    return false;
  }
dengyihao's avatar
dengyihao 已提交
806

dengyihao's avatar
dengyihao 已提交
807
#if defined(WINDOWS) || defined(DARWIN)
dengyihao's avatar
dengyihao 已提交
808
  uv_pipe_init(pThrd->loop, pThrd->pipe, 1);
dengyihao's avatar
dengyihao 已提交
809 810 811 812
#else
  uv_pipe_init(pThrd->loop, pThrd->pipe, 1);
  uv_pipe_open(pThrd->pipe, pThrd->fd);
#endif
dengyihao's avatar
dengyihao 已提交
813 814 815

  pThrd->pipe->data = pThrd;

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

dengyihao's avatar
dengyihao 已提交
818 819 820 821 822
  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 已提交
823 824 825
  // conn set
  QUEUE_INIT(&pThrd->conn);

dengyihao's avatar
dengyihao 已提交
826
  pThrd->asyncPool = transAsyncPoolCreate(pThrd->loop, 8, pThrd, uvWorkerAsyncCb);
dengyihao's avatar
dengyihao 已提交
827
#if defined(WINDOWS) || defined(DARWIN)
wafwerar's avatar
wafwerar 已提交
828
  uv_pipe_connect(&pThrd->connect_req, pThrd->pipe, pipeName, uvOnPipeConnectionCb);
dengyihao's avatar
dengyihao 已提交
829 830 831
#else
  uv_read_start((uv_stream_t*)pThrd->pipe, uvAllocConnBufferCb, uvOnConnectionCb);
#endif
dengyihao's avatar
dengyihao 已提交
832 833 834 835 836 837 838 839 840
  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 已提交
841
    tError("failed to init accept server:%s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
842 843 844
    return false;
  }

dengyihao's avatar
dengyihao 已提交
845
  // register an async here to quit server gracefully
wafwerar's avatar
wafwerar 已提交
846
  srv->pAcceptAsync = taosMemoryCalloc(1, sizeof(uv_async_t));
dengyihao's avatar
dengyihao 已提交
847 848
  uv_async_init(srv->loop, srv->pAcceptAsync, uvAcceptAsyncCb);
  srv->pAcceptAsync->data = srv;
dengyihao's avatar
dengyihao 已提交
849

dengyihao's avatar
dengyihao 已提交
850
  struct sockaddr_in bind_addr;
dengyihao's avatar
dengyihao 已提交
851 852
  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 已提交
853
    tError("failed to bind:%s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
854 855
    return false;
  }
dengyihao's avatar
dengyihao 已提交
856
  if ((err = uv_listen((uv_stream_t*)&srv->server, 4096 * 2, uvOnAcceptCb)) != 0) {
S
Shengliang Guan 已提交
857
    tError("failed to listen:%s", uv_err_name(err));
dengyihao's avatar
dengyihao 已提交
858
    terrno = TSDB_CODE_RPC_PORT_EADDRINUSE;
dengyihao's avatar
dengyihao 已提交
859 860 861
    return false;
  }
  return true;
dengyihao's avatar
dengyihao 已提交
862
}
863
void* transWorkerThread(void* arg) {
G
gccgdb1234 已提交
864
  setThreadName("trans-svr-work");
dengyihao's avatar
dengyihao 已提交
865
  SWorkThrd* pThrd = (SWorkThrd*)arg;
dengyihao's avatar
dengyihao 已提交
866
  uv_run(pThrd->loop, UV_RUN_DEFAULT);
867 868

  return NULL;
dengyihao's avatar
dengyihao 已提交
869 870
}

dengyihao's avatar
dengyihao 已提交
871
static FORCE_INLINE SSvrConn* createConn(void* hThrd) {
dengyihao's avatar
dengyihao 已提交
872
  SWorkThrd* pThrd = hThrd;
dengyihao's avatar
fix bug  
dengyihao 已提交
873

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

dengyihao's avatar
dengyihao 已提交
876
  transReqQueueInit(&pConn->wreqQueue);
dengyihao's avatar
fix bug  
dengyihao 已提交
877 878 879
  QUEUE_INIT(&pConn->queue);

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

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

dengyihao's avatar
dengyihao 已提交
883
  memset(&pConn->regArg, 0, sizeof(pConn->regArg));
dengyihao's avatar
dengyihao 已提交
884
  pConn->broken = false;
dengyihao's avatar
dengyihao 已提交
885
  pConn->status = ConnNormal;
dengyihao's avatar
opt rpc  
dengyihao 已提交
886
  transInitBuffer(&pConn->readBuf);
dengyihao's avatar
dengyihao 已提交
887

dengyihao's avatar
dengyihao 已提交
888 889 890
  SExHandle* exh = taosMemoryMalloc(sizeof(SExHandle));
  exh->handle = pConn;
  exh->pThrd = pThrd;
dengyihao's avatar
dengyihao 已提交
891 892
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
  transAcquireExHandle(transGetRefMgt(), exh->refId);
dengyihao's avatar
dengyihao 已提交
893

dengyihao's avatar
dengyihao 已提交
894
  STrans* pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
895
  pConn->refId = exh->refId;
dengyihao's avatar
dengyihao 已提交
896
  transRefSrvHandle(pConn);
S
Shengliang Guan 已提交
897
  tTrace("%s handle %p, conn %p created, refId:%" PRId64, transLabel(pTransInst), exh, pConn, pConn->refId);
dengyihao's avatar
dengyihao 已提交
898 899
  return pConn;
}
dengyihao's avatar
dengyihao 已提交
900

dengyihao's avatar
dengyihao 已提交
901
static FORCE_INLINE void destroyConn(SSvrConn* conn, bool clear) {
dengyihao's avatar
dengyihao 已提交
902 903 904
  if (conn == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
905

906
  if (clear) {
dengyihao's avatar
dengyihao 已提交
907
    if (!uv_is_closing((uv_handle_t*)conn->pTcp)) {
dengyihao's avatar
dengyihao 已提交
908 909 910
      tTrace("conn %p to be destroyed", conn);
      uv_close((uv_handle_t*)conn->pTcp, uvDestroyConn);
    }
911
  }
dengyihao's avatar
dengyihao 已提交
912
}
dengyihao's avatar
dengyihao 已提交
913
static FORCE_INLINE void destroyConnRegArg(SSvrConn* conn) {
dengyihao's avatar
dengyihao 已提交
914 915 916 917 918
  if (conn->regArg.init == 1) {
    transFreeMsg(conn->regArg.msg.pCont);
    conn->regArg.init = 0;
  }
}
dengyihao's avatar
dengyihao 已提交
919
static int reallocConnRef(SSvrConn* conn) {
dengyihao's avatar
dengyihao 已提交
920 921
  transReleaseExHandle(transGetRefMgt(), conn->refId);
  transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
922 923 924 925
  // 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 已提交
926 927
  exh->refId = transAddExHandle(transGetRefMgt(), exh);
  transAcquireExHandle(transGetRefMgt(), exh->refId);
dengyihao's avatar
dengyihao 已提交
928 929 930 931
  conn->refId = exh->refId;

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

dengyihao's avatar
dengyihao 已提交
935 936 937
  if (conn == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
938
  SWorkThrd* thrd = conn->hostThrd;
dengyihao's avatar
dengyihao 已提交
939

dengyihao's avatar
dengyihao 已提交
940 941
  transReleaseExHandle(transGetRefMgt(), conn->refId);
  transRemoveExHandle(transGetRefMgt(), conn->refId);
dengyihao's avatar
dengyihao 已提交
942

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

dengyihao's avatar
dengyihao 已提交
946 947 948 949
  for (int i = 0; i < transQueueSize(&conn->srvMsgs); i++) {
    SSvrMsg* msg = transQueueGet(&conn->srvMsgs, i);
    destroySmsg(msg);
  }
dengyihao's avatar
dengyihao 已提交
950
  transQueueDestroy(&conn->srvMsgs);
dengyihao's avatar
dengyihao 已提交
951
  transReqQueueClear(&conn->wreqQueue);
dengyihao's avatar
dengyihao 已提交
952

dengyihao's avatar
dengyihao 已提交
953
  QUEUE_REMOVE(&conn->queue);
wafwerar's avatar
wafwerar 已提交
954
  taosMemoryFree(conn->pTcp);
dengyihao's avatar
dengyihao 已提交
955
  destroyConnRegArg(conn);
dengyihao's avatar
dengyihao 已提交
956
  transDestroyBuffer(&conn->readBuf);
dengyihao's avatar
dengyihao 已提交
957
  taosMemoryFree(conn);
dengyihao's avatar
dengyihao 已提交
958

dengyihao's avatar
dengyihao 已提交
959
  if (thrd->quit && QUEUE_IS_EMPTY(&thrd->conn)) {
dengyihao's avatar
dengyihao 已提交
960
    tTrace("work thread quit");
dengyihao's avatar
dengyihao 已提交
961
    uv_walk(thrd->loop, uvWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
962
  }
dengyihao's avatar
dengyihao 已提交
963
}
wafwerar's avatar
wafwerar 已提交
964
static void uvPipeListenCb(uv_stream_t* handle, int status) {
dengyihao's avatar
dengyihao 已提交
965
  if (status != 0) {
dengyihao's avatar
dengyihao 已提交
966
    tError("server failed to init pipe, errmsg: %s", uv_err_name(status));
dengyihao's avatar
dengyihao 已提交
967 968
    return;
  }
wafwerar's avatar
wafwerar 已提交
969 970

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

dengyihao's avatar
dengyihao 已提交
973 974
  int ret = uv_pipe_init(srv->loop, pipe, 1);
  assert(ret == 0);
wafwerar's avatar
wafwerar 已提交
975

dengyihao's avatar
dengyihao 已提交
976 977 978 979 980
  ret = uv_accept((uv_stream_t*)&srv->pipeListen, (uv_stream_t*)pipe);
  assert(ret == 0);

  ret = uv_is_readable((uv_stream_t*)pipe);
  assert(ret == 1);
wafwerar's avatar
wafwerar 已提交
981

dengyihao's avatar
dengyihao 已提交
982 983
  ret = uv_is_writable((uv_stream_t*)pipe);
  assert(ret == 1);
wafwerar's avatar
wafwerar 已提交
984

dengyihao's avatar
dengyihao 已提交
985 986 987 988
  ret = uv_is_closing((uv_handle_t*)pipe);
  assert(ret == 0);

  srv->numOfWorkerReady++;
wafwerar's avatar
wafwerar 已提交
989
}
dengyihao's avatar
dengyihao 已提交
990

U
ubuntu 已提交
991
void* transInitServer(uint32_t ip, uint32_t port, char* label, int numOfThreads, void* fp, void* shandle) {
wafwerar's avatar
wafwerar 已提交
992 993
  SServerObj* srv = taosMemoryCalloc(1, sizeof(SServerObj));
  srv->loop = (uv_loop_t*)taosMemoryMalloc(sizeof(uv_loop_t));
dengyihao's avatar
dengyihao 已提交
994 995
  srv->numOfThreads = numOfThreads;
  srv->workerIdx = 0;
wafwerar's avatar
wafwerar 已提交
996
  srv->numOfWorkerReady = 0;
dengyihao's avatar
dengyihao 已提交
997
  srv->pThreadObj = (SWorkThrd**)taosMemoryCalloc(srv->numOfThreads, sizeof(SWorkThrd*));
wafwerar's avatar
wafwerar 已提交
998
  srv->pipe = (uv_pipe_t**)taosMemoryCalloc(srv->numOfThreads, sizeof(uv_pipe_t*));
dengyihao's avatar
dengyihao 已提交
999 1000 1001 1002
  srv->ip = ip;
  srv->port = port;
  uv_loop_init(srv->loop);

dengyihao's avatar
dengyihao 已提交
1003
  char pipeName[PATH_MAX];
dengyihao's avatar
dengyihao 已提交
1004
#if defined(WINDOWS) || defined(DARWIN)
dengyihao's avatar
dengyihao 已提交
1005
  int ret = uv_pipe_init(srv->loop, &srv->pipeListen, 0);
dengyihao's avatar
dengyihao 已提交
1006 1007 1008 1009
  if (ret != 0) {
    tError("failed to init pipe, errmsg: %s", uv_err_name(ret));
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1010
#if defined(WINDOWS)
dengyihao's avatar
dengyihao 已提交
1011
  snprintf(pipeName, sizeof(pipeName), "\\\\?\\pipe\\trans.rpc.%d-%" PRIu64, taosSafeRand(), GetCurrentProcessId());
dengyihao's avatar
dengyihao 已提交
1012 1013 1014 1015
#elif defined(DARWIN)
  snprintf(pipeName, sizeof(pipeName), "%s%spipe.trans.rpc.%08d-%" PRIu64, tsTempDir, TD_DIRSEP, taosSafeRand(),
           taosGetSelfPthreadId());
#endif
dengyihao's avatar
dengyihao 已提交
1016

dengyihao's avatar
dengyihao 已提交
1017
  ret = uv_pipe_bind(&srv->pipeListen, pipeName);
dengyihao's avatar
dengyihao 已提交
1018 1019 1020 1021
  if (ret != 0) {
    tError("failed to bind pipe, errmsg: %s", uv_err_name(ret));
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1022 1023

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

dengyihao's avatar
dengyihao 已提交
1029
  for (int i = 0; i < srv->numOfThreads; i++) {
dengyihao's avatar
dengyihao 已提交
1030
    SWorkThrd* thrd = (SWorkThrd*)taosMemoryCalloc(1, sizeof(SWorkThrd));
dengyihao's avatar
dengyihao 已提交
1031
    thrd->pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1032
    thrd->quit = false;
dengyihao's avatar
dengyihao 已提交
1033
    srv->pThreadObj[i] = thrd;
dengyihao's avatar
dengyihao 已提交
1034
    thrd->pTransInst = shandle;
dengyihao's avatar
dengyihao 已提交
1035

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

dengyihao's avatar
dengyihao 已提交
1039
    if (false == addHandleToWorkloop(thrd, pipeName)) {
dengyihao's avatar
dengyihao 已提交
1040 1041
      goto End;
    }
dengyihao's avatar
dengyihao 已提交
1042

dengyihao's avatar
dengyihao 已提交
1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
    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 已提交
1065
    if (uv_socketpair(SOCK_STREAM, 0, fds, UV_NONBLOCK_PIPE, UV_NONBLOCK_PIPE) != 0) {
dengyihao's avatar
dengyihao 已提交
1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
      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;
    }

1079
    int err = taosThreadCreate(&(thrd->thread), NULL, transWorkerThread, (void*)(thrd));
dengyihao's avatar
dengyihao 已提交
1080
    if (err == 0) {
S
Shengliang Guan 已提交
1081
      tDebug("success to create worker-thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1082 1083
    } else {
      // TODO: clear all other resource later
S
Shengliang Guan 已提交
1084
      tError("failed to create worker-thread:%d", i);
dengyihao's avatar
dengyihao 已提交
1085
      goto End;
dengyihao's avatar
dengyihao 已提交
1086 1087
    }
  }
dengyihao's avatar
dengyihao 已提交
1088
#endif
dengyihao's avatar
dengyihao 已提交
1089

1090
  if (false == taosValidIpAndPort(srv->ip, srv->port)) {
dengyihao's avatar
dengyihao 已提交
1091
    terrno = TAOS_SYSTEM_ERROR(errno);
S
Shengliang Guan 已提交
1092
    tError("invalid ip/port, %d:%d, reason:%s", srv->ip, srv->port, terrstr());
1093 1094
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1095

dengyihao's avatar
dengyihao 已提交
1096 1097 1098
  if (false == addHandleToAcceptloop(srv)) {
    goto End;
  }
dengyihao's avatar
dengyihao 已提交
1099

1100
  int err = taosThreadCreate(&srv->thread, NULL, transAcceptThread, (void*)srv);
dengyihao's avatar
dengyihao 已提交
1101 1102 1103
  if (err == 0) {
    tDebug("success to create accept-thread");
  } else {
dengyihao's avatar
dengyihao 已提交
1104 1105
    tError("failed  to create accept-thread");
    goto End;
dengyihao's avatar
dengyihao 已提交
1106 1107
    // clear all resource later
  }
dengyihao's avatar
dengyihao 已提交
1108

dengyihao's avatar
dengyihao 已提交
1109
  srv->inited = true;
dengyihao's avatar
dengyihao 已提交
1110
  return srv;
dengyihao's avatar
dengyihao 已提交
1111
End:
U
ubuntu 已提交
1112
  transCloseServer(srv);
dengyihao's avatar
dengyihao 已提交
1113 1114
  return NULL;
}
dengyihao's avatar
dengyihao 已提交
1115

dengyihao's avatar
dengyihao 已提交
1116
void uvHandleQuit(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1117
  thrd->quit = true;
dengyihao's avatar
dengyihao 已提交
1118
  if (QUEUE_IS_EMPTY(&thrd->conn)) {
dengyihao's avatar
dengyihao 已提交
1119
    uv_walk(thrd->loop, uvWalkCb, NULL);
dengyihao's avatar
dengyihao 已提交
1120 1121 1122
  } else {
    destroyAllConn(thrd);
  }
wafwerar's avatar
wafwerar 已提交
1123
  taosMemoryFree(msg);
dengyihao's avatar
dengyihao 已提交
1124
}
dengyihao's avatar
dengyihao 已提交
1125
void uvHandleRelease(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1126
  SSvrConn* conn = msg->pConn;
dengyihao's avatar
dengyihao 已提交
1127
  if (conn->status == ConnAcquire) {
dengyihao's avatar
dengyihao 已提交
1128
    reallocConnRef(conn);
dengyihao's avatar
dengyihao 已提交
1129
    if (!transQueuePush(&conn->srvMsgs, msg)) {
dengyihao's avatar
dengyihao 已提交
1130
      return;
dengyihao's avatar
dengyihao 已提交
1131
    }
dengyihao's avatar
dengyihao 已提交
1132
    uvStartSendRespImpl(msg);
dengyihao's avatar
dengyihao 已提交
1133
    return;
dengyihao's avatar
dengyihao 已提交
1134
  } else if (conn->status == ConnRelease || conn->status == ConnNormal) {
dengyihao's avatar
dengyihao 已提交
1135
    tDebug("%s conn %p already released, ignore release-msg", transLabel(thrd->pTransInst), conn);
dengyihao's avatar
dengyihao 已提交
1136
  }
dengyihao's avatar
dengyihao 已提交
1137
  destroySmsg(msg);
dengyihao's avatar
dengyihao 已提交
1138
}
dengyihao's avatar
dengyihao 已提交
1139
void uvHandleResp(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1140
  // send msg to client
dengyihao's avatar
dengyihao 已提交
1141
  tDebug("%s conn %p start to send resp (2/2)", transLabel(thrd->pTransInst), msg->pConn);
dengyihao's avatar
dengyihao 已提交
1142 1143
  uvStartSendResp(msg);
}
dengyihao's avatar
dengyihao 已提交
1144
void uvHandleRegister(SSvrMsg* msg, SWorkThrd* thrd) {
dengyihao's avatar
dengyihao 已提交
1145
  SSvrConn* conn = msg->pConn;
dengyihao's avatar
dengyihao 已提交
1146
  tDebug("%s conn %p register brokenlink callback", transLabel(thrd->pTransInst), conn);
dengyihao's avatar
dengyihao 已提交
1147
  if (conn->status == ConnAcquire) {
dengyihao's avatar
dengyihao 已提交
1148
    if (!transQueuePush(&conn->srvMsgs, msg)) {
dengyihao's avatar
dengyihao 已提交
1149 1150
      return;
    }
dengyihao's avatar
dengyihao 已提交
1151
    transQueuePop(&conn->srvMsgs);
dengyihao's avatar
dengyihao 已提交
1152 1153 1154
    conn->regArg.notifyCount = 0;
    conn->regArg.init = 1;
    conn->regArg.msg = msg->msg;
dengyihao's avatar
dengyihao 已提交
1155
    tDebug("conn %p register brokenlink callback succ", conn);
dengyihao's avatar
dengyihao 已提交
1156 1157 1158 1159 1160 1161

    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 已提交
1162
    taosMemoryFree(msg);
dengyihao's avatar
dengyihao 已提交
1163 1164
  }
}
dengyihao's avatar
dengyihao 已提交
1165
void destroyWorkThrd(SWorkThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1166 1167 1168
  if (pThrd == NULL) {
    return;
  }
wafwerar's avatar
wafwerar 已提交
1169
  taosThreadJoin(pThrd->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1170
  SRV_RELEASE_UV(pThrd->loop);
dengyihao's avatar
dengyihao 已提交
1171
  TRANS_DESTROY_ASYNC_POOL_MSG(pThrd->asyncPool, SSvrMsg, destroySmsg);
dengyihao's avatar
dengyihao 已提交
1172
  transAsyncPoolDestroy(pThrd->asyncPool);
dengyihao's avatar
dengyihao 已提交
1173
  taosMemoryFree(pThrd->prepare);
dengyihao's avatar
dengyihao 已提交
1174
  taosMemoryFree(pThrd->loop);
wafwerar's avatar
wafwerar 已提交
1175
  taosMemoryFree(pThrd);
dengyihao's avatar
dengyihao 已提交
1176
}
dengyihao's avatar
dengyihao 已提交
1177
void sendQuitToWorkThrd(SWorkThrd* pThrd) {
dengyihao's avatar
dengyihao 已提交
1178
  SSvrMsg* msg = taosMemoryCalloc(1, sizeof(SSvrMsg));
dengyihao's avatar
dengyihao 已提交
1179
  msg->type = Quit;
dengyihao's avatar
dengyihao 已提交
1180
  tDebug("server send quit msg to work thread");
dengyihao's avatar
dengyihao 已提交
1181
  transAsyncSend(pThrd->asyncPool, &msg->q);
dengyihao's avatar
dengyihao 已提交
1182 1183
}

U
ubuntu 已提交
1184
void transCloseServer(void* arg) {
dengyihao's avatar
dengyihao 已提交
1185 1186
  // impl later
  SServerObj* srv = arg;
dengyihao's avatar
dengyihao 已提交
1187 1188

  tDebug("send quit msg to accept thread");
dengyihao's avatar
dengyihao 已提交
1189 1190 1191
  if (srv->inited) {
    uv_async_send(srv->pAcceptAsync);
    taosThreadJoin(srv->thread, NULL);
dengyihao's avatar
dengyihao 已提交
1192
    SRV_RELEASE_UV(srv->loop);
dengyihao's avatar
dengyihao 已提交
1193

dengyihao's avatar
dengyihao 已提交
1194 1195 1196 1197
    for (int i = 0; i < srv->numOfThreads; i++) {
      sendQuitToWorkThrd(srv->pThreadObj[i]);
      destroyWorkThrd(srv->pThreadObj[i]);
    }
dengyihao's avatar
dengyihao 已提交
1198 1199
  } else {
    uv_loop_close(srv->loop);
dengyihao's avatar
dengyihao 已提交
1200 1201
  }

wafwerar's avatar
wafwerar 已提交
1202 1203 1204
  taosMemoryFree(srv->pThreadObj);
  taosMemoryFree(srv->pAcceptAsync);
  taosMemoryFree(srv->loop);
dengyihao's avatar
dengyihao 已提交
1205 1206

  for (int i = 0; i < srv->numOfThreads; i++) {
wafwerar's avatar
wafwerar 已提交
1207
    taosMemoryFree(srv->pipe[i]);
dengyihao's avatar
dengyihao 已提交
1208
  }
wafwerar's avatar
wafwerar 已提交
1209
  taosMemoryFree(srv->pipe);
dengyihao's avatar
dengyihao 已提交
1210

wafwerar's avatar
wafwerar 已提交
1211
  taosMemoryFree(srv);
dengyihao's avatar
dengyihao 已提交
1212
}
dengyihao's avatar
dengyihao 已提交
1213

dengyihao's avatar
dengyihao 已提交
1214 1215 1216 1217
void transRefSrvHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1218
  int ref = T_REF_INC((SSvrConn*)handle);
dengyihao's avatar
dengyihao 已提交
1219
  tTrace("conn %p ref count:%d", handle, ref);
dengyihao's avatar
dengyihao 已提交
1220 1221 1222 1223 1224 1225
}

void transUnrefSrvHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1226
  int ref = T_REF_DEC((SSvrConn*)handle);
dengyihao's avatar
dengyihao 已提交
1227
  tTrace("conn %p ref count:%d", handle, ref);
dengyihao's avatar
dengyihao 已提交
1228
  if (ref == 0) {
dengyihao's avatar
dengyihao 已提交
1229
    destroyConn((SSvrConn*)handle, true);
dengyihao's avatar
dengyihao 已提交
1230 1231
  }
}
dengyihao's avatar
dengyihao 已提交
1232

dengyihao's avatar
dengyihao 已提交
1233
int transReleaseSrvHandle(void* handle) {
dengyihao's avatar
dengyihao 已提交
1234 1235 1236
  SRpcHandleInfo* info = handle;
  SExHandle*      exh = info->handle;
  int64_t         refId = info->refId;
dengyihao's avatar
dengyihao 已提交
1237

dengyihao's avatar
dengyihao 已提交
1238
  ASYNC_CHECK_HANDLE(exh, refId);
dengyihao's avatar
dengyihao 已提交
1239

dengyihao's avatar
dengyihao 已提交
1240
  SWorkThrd* pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
1241 1242
  ASYNC_ERR_JRET(pThrd);

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

dengyihao's avatar
dengyihao 已提交
1245 1246 1247
  SSvrMsg* m = taosMemoryCalloc(1, sizeof(SSvrMsg));
  m->msg = tmsg;
  m->type = Release;
dengyihao's avatar
dengyihao 已提交
1248

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

dengyihao's avatar
dengyihao 已提交
1254
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1255
  return 0;
dengyihao's avatar
dengyihao 已提交
1256
_return1:
dengyihao's avatar
dengyihao 已提交
1257
  tDebug("handle %p failed to send to release handle", exh);
dengyihao's avatar
dengyihao 已提交
1258
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1259
  return -1;
dengyihao's avatar
dengyihao 已提交
1260
_return2:
dengyihao's avatar
dengyihao 已提交
1261
  tDebug("handle %p failed to send to release handle", exh);
dengyihao's avatar
dengyihao 已提交
1262
  return -1;
dengyihao's avatar
dengyihao 已提交
1263
}
1264

dengyihao's avatar
dengyihao 已提交
1265
int transSendResponse(const STransMsg* msg) {
dengyihao's avatar
dengyihao 已提交
1266 1267 1268 1269 1270
  if (msg->info.noResp) {
    rpcFreeCont(msg->pCont);
    tTrace("no need send resp");
    return 0;
  }
S
Shengliang Guan 已提交
1271 1272
  SExHandle* exh = msg->info.handle;
  int64_t    refId = msg->info.refId;
dengyihao's avatar
dengyihao 已提交
1273
  ASYNC_CHECK_HANDLE(exh, refId);
dengyihao's avatar
dengyihao 已提交
1274
  assert(refId != 0);
dengyihao's avatar
dengyihao 已提交
1275

dengyihao's avatar
dengyihao 已提交
1276
  STransMsg tmsg = *msg;
S
Shengliang Guan 已提交
1277
  tmsg.info.refId = refId;
dengyihao's avatar
dengyihao 已提交
1278

dengyihao's avatar
dengyihao 已提交
1279
  SWorkThrd* pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
1280
  ASYNC_ERR_JRET(pThrd);
dengyihao's avatar
dengyihao 已提交
1281

dengyihao's avatar
dengyihao 已提交
1282 1283 1284
  SSvrMsg* m = taosMemoryCalloc(1, sizeof(SSvrMsg));
  m->msg = tmsg;
  m->type = Normal;
dengyihao's avatar
dengyihao 已提交
1285 1286

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

dengyihao's avatar
dengyihao 已提交
1292
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1293
  return 0;
H
Haojun Liao 已提交
1294

dengyihao's avatar
dengyihao 已提交
1295
_return1:
dengyihao's avatar
dengyihao 已提交
1296
  tDebug("handle %p failed to send resp", exh);
dengyihao's avatar
dengyihao 已提交
1297
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1298
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1299
  return -1;
dengyihao's avatar
dengyihao 已提交
1300
_return2:
dengyihao's avatar
dengyihao 已提交
1301
  tDebug("handle %p failed to send resp", exh);
dengyihao's avatar
dengyihao 已提交
1302
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1303
  return -1;
dengyihao's avatar
dengyihao 已提交
1304
}
dengyihao's avatar
dengyihao 已提交
1305
int transRegisterMsg(const STransMsg* msg) {
S
Shengliang Guan 已提交
1306 1307
  SExHandle* exh = msg->info.handle;
  int64_t    refId = msg->info.refId;
dengyihao's avatar
dengyihao 已提交
1308 1309
  ASYNC_CHECK_HANDLE(exh, refId);

dengyihao's avatar
dengyihao 已提交
1310
  STransMsg tmsg = *msg;
dengyihao's avatar
dengyihao 已提交
1311 1312
  tmsg.info.noResp = 1;

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

dengyihao's avatar
dengyihao 已提交
1315
  SWorkThrd* pThrd = exh->pThrd;
dengyihao's avatar
dengyihao 已提交
1316
  ASYNC_ERR_JRET(pThrd);
dengyihao's avatar
dengyihao 已提交
1317

dengyihao's avatar
dengyihao 已提交
1318 1319 1320
  SSvrMsg* m = taosMemoryCalloc(1, sizeof(SSvrMsg));
  m->msg = tmsg;
  m->type = Register;
dengyihao's avatar
dengyihao 已提交
1321

dengyihao's avatar
dengyihao 已提交
1322
  STrans* pTransInst = pThrd->pTransInst;
dengyihao's avatar
dengyihao 已提交
1323
  tDebug("%s conn %p start to register brokenlink callback", transLabel(pTransInst), exh->handle);
1324 1325 1326
  if (0 != transAsyncSend(pThrd->asyncPool, &m->q)) {
    destroySmsg(m);
  }
H
Haojun Liao 已提交
1327

dengyihao's avatar
dengyihao 已提交
1328
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1329
  return 0;
dengyihao's avatar
dengyihao 已提交
1330

dengyihao's avatar
dengyihao 已提交
1331
_return1:
dengyihao's avatar
dengyihao 已提交
1332
  tDebug("handle %p failed to register brokenlink", exh);
dengyihao's avatar
dengyihao 已提交
1333
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1334
  transReleaseExHandle(transGetRefMgt(), refId);
dengyihao's avatar
dengyihao 已提交
1335
  return -1;
dengyihao's avatar
dengyihao 已提交
1336
_return2:
dengyihao's avatar
dengyihao 已提交
1337
  tDebug("handle %p failed to register brokenlink", exh);
dengyihao's avatar
dengyihao 已提交
1338
  rpcFreeCont(msg->pCont);
dengyihao's avatar
dengyihao 已提交
1339
  return -1;
dengyihao's avatar
dengyihao 已提交
1340
}
dengyihao's avatar
dengyihao 已提交
1341

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

dengyihao's avatar
dengyihao 已提交
1344
#endif