#5011

Committed 03 Apr 2026 03:59PM UTC coverage: 72.3% (+0.008%) from 72.292%

Build # #5011

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

merge: from main to 3.0 branch #35067

Coverage Stats

4053 of 5985 new or added lines in 68 files covered. (67.72%)

732 existing lines in 143 files now uncovered.

257430 of 356056 relevant lines covered (72.3%)

131834103.52 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

71.9

/source/libs/transport/src/transComm.c

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

#include "transComm.h"
#include "osTime.h"
#include "tchecksum.h"
#include "tqueue.h"
#include "transLog.h"
#include "transSasl.h"

#ifndef TD_ASTRA_RPC
#define BUFFER_CAP 8 * 1024

static TdThreadOnce transModuleInit = PTHREAD_ONCE_INIT;

static int32_t refMgt;
static int32_t svrRefMgt;
static int32_t instMgt;
static int32_t transSyncMsgMgt;

static void transDestroySyncMsg(void* msg);
typedef struct {
  int64_t refId;
  STrans* pTrans;
  int32_t ref;
} STransEntry;

/*
 * pArray: array of STransEntry; typically contains fewer than 8 entries.
 * lock: The lock protects concurrent access (reads/writes) to this array.
 * Access pattern is heavily read-dominated; write operations are rare.
 */

typedef struct {
  SArray*        pArray;
  TdThreadRwlock lock;
} STransCache;

static STransCache transInstCache;

static void transCacheInit() {
  transInstCache.pArray = taosArrayInit(16, sizeof(STransEntry));
  if (transInstCache.pArray == NULL) {
    tError("failed to init trans cache since %s", tstrerror(terrno));
    return;
  }

  (void)taosThreadRwlockInit(&transInstCache.lock, NULL);
}

int32_t transCachePut(int64_t refId, STrans* pTrans) {
  int32_t code = 0;
  (void)taosThreadRwlockWrlock(&transInstCache.lock);

  STransEntry entry = {.refId = refId, .pTrans = pTrans, .ref = 0};
  if (NULL == taosArrayPush(transInstCache.pArray, &entry)) {
    code = terrno;
  }
  (void)taosThreadRwlockUnlock(&transInstCache.lock);
  return code;
}

int32_t transCacheAcquireById(int64_t refId, STrans** pTrans) {
  int32_t code = TSDB_CODE_RPC_MODULE_QUIT;

  (void)taosThreadRwlockRdlock(&transInstCache.lock);

  for (int32_t i = 0; i < taosArrayGetSize(transInstCache.pArray); ++i) {
    STransEntry* p = taosArrayGet(transInstCache.pArray, i);
    if (p->refId == refId) {
      *pTrans = p->pTrans;
      (void)atomic_fetch_add_32(&p->ref, 1);
      tDebug("trans %p acquire by refId:%" PRId64 ", ref count:%d", p->pTrans, refId, atomic_load_32(&p->ref));
      code = 0;
      break;
    }
  }

  (void)taosThreadRwlockUnlock(&transInstCache.lock);
  if (code != 0) {
    tError("failed to acquire from trans cache by refId:%" PRId64 " since %s", refId, tstrerror(code));
  }
  return code;
}

void transCacheReleaseByRefId(int64_t refId) {
  int32_t code = TSDB_CODE_RPC_MODULE_QUIT;

  (void)taosThreadRwlockRdlock(&transInstCache.lock);

  for (int32_t i = 0; i < taosArrayGetSize(transInstCache.pArray); i++) {
    STransEntry* p = taosArrayGet(transInstCache.pArray, i);
    if (p->refId == refId) {
      (void)atomic_sub_fetch_32(&p->ref, 1);
      tDebug("trans %p release by refId:%" PRId64 ", ref count:%d", p->pTrans, refId, atomic_load_32(&p->ref));
      code = 0;
      break;
    }
  }

  (void)taosThreadRwlockUnlock(&transInstCache.lock);
  if (code != 0) {
    tInfo("failed to remove from trans cache by refId:%" PRId64 " since %s", refId, tstrerror(code));
  }
}

void transCacheRemoveByRefId(int64_t refId) {
  int32_t code = TSDB_CODE_RPC_MODULE_QUIT;

  (void)taosThreadRwlockWrlock(&transInstCache.lock);

  for (int32_t i = 0; i < taosArrayGetSize(transInstCache.pArray); i++) {
    STransEntry* p = taosArrayGet(transInstCache.pArray, i);
    if (p->refId == refId) {
      taosArrayRemove(transInstCache.pArray, i);
      code = 0;
      break;
    }
  }
  (void)taosThreadRwlockUnlock(&transInstCache.lock);

  if (code != 0) {
    tError("failed to remove from trans cache by refId:%" PRId64 " since %s", refId, tstrerror(code));
  }
}

void transCacheDestroy() {
  taosArrayDestroyP(transInstCache.pArray, NULL);
  (void)taosThreadRwlockDestroy(&transInstCache.lock);
}

int32_t transCompressMsg(char* msg, int32_t len) {
  int32_t        ret = 0;
  int            compHdr = sizeof(STransCompMsg);
  STransMsgHead* pHead = transHeadFromCont(msg);

  int64_t start = taosGetTimestampMs();
  char*   buf = taosMemoryMalloc(len + compHdr + 8);  // 8 extra bytes
  if (buf == NULL) {
    tWarn("failed to allocate memory for rpc msg compression, contLen:%d", len);
    ret = len;
    return ret;
  }

  int32_t clen = LZ4_compress_default(msg, buf, len, len + compHdr);
  /*
   * only the compressed size is less than the value of contLen - overhead, the compression is applied
   * The first four bytes is set to 0, the second four bytes are utilized to keep the original length of message
   */
  if (clen > 0 && clen < len - compHdr) {
    STransCompMsg* pComp = (STransCompMsg*)msg;
    pComp->reserved = 0;
    pComp->contLen = htonl(len);
    memcpy(msg + compHdr, buf, clen);

    tDebug("compress rpc msg, before:%d, after:%d", len, clen);
    ret = clen + compHdr;
    pHead->comp = 1;
  } else {
    ret = len;
    pHead->comp = 0;
  }
  taosMemoryFree(buf);

  int64_t elapse = taosGetTimestampMs() - start;
  if (elapse >= 100) {
    tWarn("compress msg cost %dms", (int)(elapse));
  }
  return ret;
}
int32_t transDecompressMsg(char** msg, int32_t* len) {
  STransMsgHead* pHead = (STransMsgHead*)(*msg);
  if (pHead->comp == 0) return 0;

  int64_t start = taosGetTimestampMs();

  char* pCont = transContFromHead(pHead);

  STransCompMsg* pComp = (STransCompMsg*)pCont;
  int32_t        oriLen = ntohl(pComp->contLen);

  int32_t tlen = *len;
  char*   buf = taosMemoryCalloc(1, oriLen + sizeof(STransMsgHead));
  if (buf == NULL) {
    return terrno;
  }

  STransMsgHead* pNewHead = (STransMsgHead*)buf;
  int32_t        decompLen = LZ4_decompress_safe(pCont + sizeof(STransCompMsg), (char*)pNewHead->content,
                                                 tlen - sizeof(STransMsgHead) - sizeof(STransCompMsg), oriLen);

  if (decompLen != oriLen) {
    taosMemoryFree(buf);
    return TSDB_CODE_INVALID_MSG;
  }
  memcpy((char*)pNewHead, (char*)pHead, sizeof(STransMsgHead));

  *len = oriLen + sizeof(STransMsgHead);
  pNewHead->msgLen = htonl(oriLen + sizeof(STransMsgHead));

  taosMemoryFree(pHead);
  *msg = buf;

  int64_t elapse = taosGetTimestampMs() - start;
  if (elapse >= 100) {
    tWarn("dcompress msg cost %dms", (int)(elapse));
  }
  return 0;
}
int32_t transDecompressMsgExt(char const* msg, int32_t len, char** out, int32_t* outLen) {
  STransMsgHead* pHead = (STransMsgHead*)msg;
  char*          pCont = transContFromHead(pHead);

  STransCompMsg* pComp = (STransCompMsg*)pCont;
  int32_t        oriLen = ntohl(pComp->contLen);

  int32_t tlen = len;
  char*   buf = taosMemoryCalloc(1, oriLen + sizeof(STransMsgHead));
  if (buf == NULL) {
    return terrno;
  }
  int64_t start = taosGetTimestampMs();

  STransMsgHead* pNewHead = (STransMsgHead*)buf;
  int32_t        decompLen = LZ4_decompress_safe(pCont + sizeof(STransCompMsg), (char*)pNewHead->content,
                                                 tlen - sizeof(STransMsgHead) - sizeof(STransCompMsg), oriLen);
  if (decompLen != oriLen) {
    tError("msgLen:%d, originLen:%d, decompLen:%d", len, oriLen, decompLen);
    taosMemoryFree(buf);
    return TSDB_CODE_INVALID_MSG;
  }
  memcpy((char*)pNewHead, (char*)pHead, sizeof(STransMsgHead));

  *out = buf;
  *outLen = oriLen + sizeof(STransMsgHead);
  pNewHead->msgLen = *outLen;
  pNewHead->comp = 0;

  int64_t elapse = taosGetTimestampMs() - start;
  if (elapse >= 100) {
    tWarn("dcompress msg cost %dms", (int)(elapse));
  }
  return 0;
}

void transFreeMsg(void* msg) {
  if (msg == NULL) {
    return;
  }
  tTrace("cont:%p, rpc free", (char*)msg - TRANS_MSG_OVERHEAD);
  taosMemoryFree((char*)msg - sizeof(STransMsgHead));
}
void transSockInfo2Str(struct sockaddr* sockname, char* dst, int32_t cap) {
  char     buf[IP_RESERVE_CAP] = {0};
  uint16_t port = 0;
  int      r = 0;
  if (sockname->sa_family == AF_INET) {
    struct sockaddr_in* addr = (struct sockaddr_in*)sockname;

    r = uv_ip4_name(addr, (char*)buf, sizeof(buf));
    if (r != 0) {
      uError("failed to get ip from sockaddr, err:%s", uv_strerror(r));
    }

    port = ntohs(addr->sin_port);
  } else if (sockname->sa_family == AF_INET6) {
    struct sockaddr_in6* addr = (struct sockaddr_in6*)sockname;

    r = uv_ip6_name(addr, buf, sizeof(buf));
    if (r != 0) {
      uError("failed to get ip from sockaddr6, err:%s", uv_strerror(r));
    }

    port = ntohs(addr->sin6_port);
  }
  snprintf(dst, cap, "%s:%d", buf, port);
}
int32_t transInitBuffer(SConnBuffer* buf) {
  buf->buf = taosMemoryCalloc(1, BUFFER_CAP);
  if (buf->buf == NULL) {
    return terrno;
  }

  buf->cap = BUFFER_CAP;
  buf->left = -1;
  buf->len = 0;
  buf->total = 0;
  buf->invalid = 0;
  return 0;
}
void transDestroyBuffer(SConnBuffer* p) {
  taosMemoryFree(p->buf);
  p->buf = NULL;
}

int32_t transClearBuffer(SConnBuffer* buf) {
  SConnBuffer* p = buf;
  if (p->cap > BUFFER_CAP) {
    p->cap = BUFFER_CAP;
    p->buf = taosMemoryRealloc(p->buf, BUFFER_CAP);
    if (p->buf == NULL) {
      return terrno;
    }
  }
  p->left = -1;
  p->len = 0;
  p->total = 0;
  p->invalid = 0;
  return 0;
}

int32_t transDumpFromBuffer(SConnBuffer* connBuf, char** buf, int8_t resetBuf, int32_t* len) {
  static const int HEADSIZE = sizeof(STransMsgHead);
  int32_t          code = 0;
  SConnBuffer*     p = connBuf;
  if (p->left != 0 || p->total <= 0) {
    return TSDB_CODE_INVALID_MSG;
  }
  int total = p->total;
  if (total >= HEADSIZE && !p->invalid) {
    *buf = taosMemoryCalloc(1, total);
    if (*buf == NULL) {
      return terrno;
    }
    memcpy(*buf, p->buf, total);
    if ((code = transResetBuffer(connBuf, resetBuf)) < 0) {
      return code;
    }

    if (connBuf->total == 0) {
      code = transDoCrcCheck(*buf, total);
      if (code != 0) {
        tError("failed to check crc for msg in buffer, total:%d since %s", total, tstrerror(code));
        taosMemoryFree(*buf);
        *buf = NULL;
        return code;
      }
    }
    *len = total;
  } else {
    *len = -1;
    code = TSDB_CODE_INVALID_MSG;
  }
  return code;
}

int32_t transResetBuffer(SConnBuffer* connBuf, int8_t resetBuf) {
  SConnBuffer* p = connBuf;
  if (p->total < p->len) {
    int left = p->len - p->total;
    memmove(p->buf, p->buf + p->total, left);
    p->left = -1;
    p->total = 0;
    p->len = left;
  } else if (p->total == p->len) {
    p->left = -1;
    p->total = 0;
    p->len = 0;
    if (p->cap > BUFFER_CAP) {
      if (resetBuf) {
        p->cap = BUFFER_CAP;
        p->buf = taosMemoryRealloc(p->buf, p->cap);
        if (p->buf == NULL) {
          return terrno;
        }
      }
    }
  } else {
    tError("failed to reset buffer, total:%d, len:%d since %s", p->total, p->len, tstrerror(TSDB_CODE_INVALID_MSG));
    return TSDB_CODE_INVALID_MSG;
  }
  return 0;
}
int32_t transAllocBuffer(SConnBuffer* connBuf, uv_buf_t* uvBuf) {
  /*
   * formate of data buffer:
   * |<--------------------------data from socket------------------------------->|
   * |<------STransMsgHead------->|<-------------------userdata--------------->|<-----auth data----->|<----user
   * info--->|
   */
  SConnBuffer* p = connBuf;
  uvBuf->base = p->buf + p->len;
  if (p->left == -1) {
    uvBuf->len = p->cap - p->len;
  } else {
    if (p->left < p->cap - p->len) {
      uvBuf->len = p->left;
    } else {
      p->cap = p->left + p->len;
      p->buf = taosMemoryRealloc(p->buf, p->cap);
      if (p->buf == NULL) {
        uvBuf->base = NULL;
        uvBuf->len = 0;
        return terrno;
      }
      uvBuf->base = p->buf + p->len;
      uvBuf->len = p->left;
    }
  }
  return 0;
}
// check whether already read complete
bool transReadComplete(SConnBuffer* connBuf) {
  SConnBuffer* p = connBuf;
  if (p->len >= sizeof(STransMsgHead)) {
    if (p->left == -1) {
      STransMsgHead head;
      memcpy((char*)&head, connBuf->buf, sizeof(head));
      int32_t msgLen = (int32_t)ntohl(head.msgLen);
      p->total = msgLen;
      p->invalid = (head.version != TRANS_VER || msgLen >= TRANS_MSG_LIMIT);
      if (p->invalid) {
        tError("recv invalid msg, version:%d, expect:%d, msg len %d, limit:%d", head.version, TRANS_VER, msgLen, (int)(TRANS_MSG_LIMIT));
      }
    }
    if (p->total >= p->len) {
      p->left = p->total - p->len;
    } else {
      p->left = 0;
    }
  }
  return (p->left == 0 || p->invalid) ? true : false;
}

int32_t transConnBufferAppend(SConnBuffer* connBuf, char* buf, int32_t len) {
  int32_t      code = 0;
  SConnBuffer* p = connBuf;
  if (p->len + len > p->cap) {
    int32_t newCap = p->len + len;
    char*   newBuf = taosMemoryRealloc(p->buf, newCap);
    if (newBuf == NULL) {
      return terrno;
    }
    p->buf = newBuf;
    p->cap = newCap;
  }

  memcpy(p->buf + p->len, buf, len);
  p->len += len;
  return code;
}

int32_t transSetConnOption(uv_tcp_t* stream, int keepalive) {
  int32_t ret = 0;
#if defined(WINDOWS) || defined(DARWIN)
#else
  ret = uv_tcp_keepalive(stream, 1, keepalive);
#endif
  ret = uv_tcp_nodelay(stream, 1);
  return ret;
  // int ret = uv_tcp_keepalive(stream, 5, 60);
}

int32_t transAsyncPoolCreate(uv_loop_t* loop, int sz, void* arg, AsyncCB cb, SAsyncPool** pPool) {
  SAsyncPool* pool = taosMemoryCalloc(1, sizeof(SAsyncPool));
  if (pool == NULL) {
    return terrno;
    // return NULL;
  }
  int32_t code = 0;

  pool->nAsync = sz;
  pool->asyncs = taosMemoryCalloc(1, sizeof(uv_async_t) * pool->nAsync);
  if (pool->asyncs == NULL) {
    taosMemoryFree(pool);
    return terrno;
  }

  int i = 0, err = 0;
  for (i = 0; i < pool->nAsync; i++) {
    uv_async_t* async = &(pool->asyncs[i]);

    SAsyncItem* item = taosMemoryCalloc(1, sizeof(SAsyncItem));
    if (item == NULL) {
      code = terrno;
      break;
    }
    item->pThrd = arg;
    QUEUE_INIT(&item->qmsg);
    code = taosThreadMutexInit(&item->mtx, NULL);
    if (code) {
      taosMemoryFree(item);
      break;
    }

    async->data = item;
    err = uv_async_init(loop, async, cb);
    if (err != 0) {
      tError("failed to init async since %s", uv_err_name(err));
      code = TSDB_CODE_THIRDPARTY_ERROR;
      break;
    }
  }

  if (i != pool->nAsync) {
    transAsyncPoolDestroy(pool);
    pool = NULL;
  }

  *pPool = pool;
  return 0;
  // return pool;
}

void transAsyncPoolDestroy(SAsyncPool* pool) {
  if (pool == NULL) return;

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

    TAOS_UNUSED(taosThreadMutexDestroy(&item->mtx));
    taosMemoryFree(item);
  }
  taosMemoryFree(pool->asyncs);
  taosMemoryFree(pool);
}
bool transAsyncPoolIsEmpty(SAsyncPool* pool) {
  for (int i = 0; i < pool->nAsync; i++) {
    uv_async_t* async = &(pool->asyncs[i]);
    SAsyncItem* item = async->data;
    if (!QUEUE_IS_EMPTY(&item->qmsg)) return false;
  }
  return true;
}
int transAsyncSend(SAsyncPool* pool, queue* q) {
  if (atomic_load_8(&pool->stop) == 1) {
    return TSDB_CODE_RPC_ASYNC_MODULE_QUIT;
  }
  int idx = pool->index % pool->nAsync;

  // no need mutex here
  if (pool->index++ > pool->nAsync * 2000) {
    pool->index = 0;
  }
  uv_async_t* async = &(pool->asyncs[idx]);
  SAsyncItem* item = async->data;

  if (taosThreadMutexLock(&item->mtx) != 0) {
    tError("failed to lock mutex since %s", tstrerror(terrno));
    return terrno;
  }

  QUEUE_PUSH(&item->qmsg, q);
  TAOS_UNUSED(taosThreadMutexUnlock(&item->mtx));

  int ret = uv_async_send(async);
  if (ret != 0) {
    tError("failed to send async since %s", uv_err_name(ret));
    return TSDB_CODE_THIRDPARTY_ERROR;
  }
  return 0;
}

void transCtxInit(STransCtx* ctx) {
  // init transCtx
  ctx->args = taosHashInit(2, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_NO_LOCK);
  ctx->brokenVal.val = NULL;
  ctx->freeFunc = NULL;
}
void transCtxCleanup(STransCtx* ctx) {
  if (ctx == NULL || ctx->args == NULL) {
    return;
  }

  STransCtxVal* iter = taosHashIterate(ctx->args, NULL);
  while (iter) {
    int32_t* type = taosHashGetKey(iter, NULL);
    tDebug("free msg type %s dump func", TMSG_INFO(*type));
    ctx->freeFunc(iter->val);
    iter = taosHashIterate(ctx->args, iter);
  }
  if (ctx->freeFunc) ctx->freeFunc(ctx->brokenVal.val);
  taosHashCleanup(ctx->args);
  ctx->args = NULL;
}

void transCtxMerge(STransCtx* dst, STransCtx* src) {
  if (src->args == NULL || src->freeFunc == NULL) {
    return;
  }
  SRpcBrokenlinkVal tval = {0};
  void (*freeFunc)(const void* arg) = NULL;

  if (dst->args == NULL) {
    dst->args = src->args;
    dst->brokenVal = src->brokenVal;
    dst->freeFunc = src->freeFunc;
    src->args = NULL;
    return;
  } else {
    tval = dst->brokenVal;
    freeFunc = dst->freeFunc;

    dst->brokenVal = src->brokenVal;
    dst->freeFunc = src->freeFunc;
  }

  size_t klen = 0;
  void*  iter = taosHashIterate(src->args, NULL);
  while (iter) {
    STransCtxVal* sVal = (STransCtxVal*)iter;
    int32_t*      msgType = taosHashGetKey(sVal, &klen);

    STransCtxVal* dVal = taosHashGet(dst->args, msgType, sizeof(*msgType));
    if (dVal != NULL) {
      tDebug("free msg type %s dump func", TMSG_INFO(*(int32_t*)msgType));
      dst->freeFunc(dVal->val);
      dVal->val = NULL;

      TAOS_UNUSED(taosHashRemove(dst->args, msgType, sizeof(*msgType)));
    }

    int32_t code = taosHashPut(dst->args, msgType, sizeof(*msgType), sVal, sizeof(*sVal));
    if (code != 0) {
      tError("failed to put val to hash since %s", tstrerror(code));
      tDebug("put msg type %s dump func", TMSG_INFO(*(int32_t*)msgType));
      if (src->freeFunc) (src->freeFunc)(sVal->val);
      sVal->val = NULL;
    }
    iter = taosHashIterate(src->args, iter);
  }
  if (freeFunc != NULL && tval.val != NULL) {
    freeFunc(tval.val);
    tval.val = NULL;
  }

  taosHashCleanup(src->args);
  src->args = NULL;
  src->brokenVal.val = NULL;
}
void* transCtxDumpVal(STransCtx* ctx, int32_t key) {
  if (ctx->args == NULL) {
    return NULL;
  }
  STransCtxVal* cVal = taosHashGet(ctx->args, (const void*)&key, sizeof(key));
  if (cVal == NULL) {
    return NULL;
  }
  void* ret = NULL;
  TAOS_UNUSED((*cVal->clone)(cVal->val, &ret));
  return ret;
}
void* transCtxDumpBrokenlinkVal(STransCtx* ctx, int32_t* msgType) {
  void* ret = NULL;
  if (ctx->brokenVal.clone == NULL) {
    return ret;
  }
  TAOS_UNUSED((*ctx->brokenVal.clone)(ctx->brokenVal.val, &ret));

  *msgType = ctx->brokenVal.msgType;

  return ret;
}

int32_t transDoCrc(char* buf, int32_t len) {
  STransMsgHead* pHead = (STransMsgHead*)buf;
  pHead->magicNum = 0;
  uint32_t chechSum = taosCalcChecksum(0, (const uint8_t*)buf, len);
  pHead->magicNum = htonl(chechSum);

  return 0;
}
int32_t transDoCrcCheck(char* buf, int32_t len) {
  STransMsgHead* pHead = (STransMsgHead*)buf;
  uint32_t       checkSum = ntohl(pHead->magicNum);
  pHead->magicNum = 0;
  if (taosCheckChecksum((const uint8_t*)buf, len, checkSum)) {
    return TSDB_CODE_INVALID_MSG;
  } else {
    return 0;
  }
}

#if 0
int32_t transQueueInit(STransQueue* wq, void (*freeFunc)(void* arg)) {
  QUEUE_INIT(&wq->node);
  wq->freeFunc = (void (*)(void*))freeFunc;
  wq->size = 0;
  wq->inited = 1;
  return 0;
}
void transQueuePush(STransQueue* q, void* arg) {
  queue* node = arg;
  QUEUE_PUSH(&q->node, node);
  q->size++;
}
void* transQueuePop(STransQueue* q) {
  if (q->size == 0) return NULL;

  queue* head = QUEUE_HEAD(&q->node);
  QUEUE_REMOVE(head);
  q->size--;
  return head;
}
int32_t transQueueSize(STransQueue* q) { return q->size; }

void* transQueueGet(STransQueue* q, int idx) {
  if (q->size == 0) return NULL;

  while (idx-- > 0) {
    queue* node = QUEUE_NEXT(&q->node);
    if (node == &q->node) return NULL;
  }
  return NULL;
}

void transQueueRemoveByFilter(STransQueue* q, bool (*filter)(void* e, void* arg), void* arg, void* dst, int32_t size)
{
  queue* d = dst;
  queue* node = QUEUE_NEXT(&q->node);
  while (node != &q->node) {
    queue* next = QUEUE_NEXT(node);
    if (filter && filter(node, arg)) {
      QUEUE_REMOVE(node);
      q->size--;
      QUEUE_PUSH(d, node);
      if (--size == 0) {
        break;
      }
    }
    node = next;
  }
}

void* tranQueueHead(STransQueue* q) {
  if (q->size == 0) return NULL;

  queue* head = QUEUE_HEAD(&q->node);
  return head;
}

void* transQueueRm(STransQueue* q, int i) {
  // if (queue->q == NULL || taosArrayGetSize(queue->q) == 0) {
  //   return NULL;
  // }
  // if (i >= taosArrayGetSize(queue->q)) {
  //   return NULL;
  // }
  // void* ptr = taosArrayGetP(queue->q, i);
  // taosArrayRemove(queue->q, i);
  // return ptr;
  return NULL;
}

void transQueueRemove(STransQueue* q, void* e) {
  if (q->size == 0) return;
  queue* node = e;
  QUEUE_REMOVE(node);
  q->size--;
}

bool transQueueEmpty(STransQueue* q) { return q->size == 0 ? true : false; }

void transQueueClear(STransQueue* q) {
  if (q->inited == 0) return;
  while (!QUEUE_IS_EMPTY(&q->node)) {
    queue* h = QUEUE_HEAD(&q->node);
    QUEUE_REMOVE(h);
    if (q->freeFunc != NULL) (q->freeFunc)(h);
    q->size--;
  }
}
void transQueueDestroy(STransQueue* q) { transQueueClear(q); }
#endif

static FORCE_INLINE int32_t timeCompare(const HeapNode* a, const HeapNode* b) {
  SDelayTask* arg1 = container_of(a, SDelayTask, node);
  SDelayTask* arg2 = container_of(b, SDelayTask, node);
  if (arg1->execTime > arg2->execTime) {
    return 0;
  } else {
    return 1;
  }
}

static void transDQTimeout(uv_timer_t* timer) {
  SDelayQueue* queue = timer->data;
  tTrace("timer %p timeout", timer);
  uint64_t timeout = 0;
  int64_t  current = taosGetTimestampMs();
  do {
    HeapNode* minNode = heapMin(queue->heap);
    if (minNode == NULL) break;
    SDelayTask* task = container_of(minNode, SDelayTask, node);
    if (task->execTime <= current) {
      heapRemove(queue->heap, minNode);
      task->func(task->arg);
      taosMemoryFree(task);
      timeout = 0;
    } else {
      timeout = task->execTime - current;
      break;
    }
  } while (1);
  if (timeout != 0) {
    TAOS_UNUSED(uv_timer_start(queue->timer, transDQTimeout, timeout, 0));
  }
}
int32_t transDQCreate(uv_loop_t* loop, SDelayQueue** queue) {
  int32_t      code = 0;
  Heap*        heap = NULL;
  uv_timer_t*  timer = NULL;
  SDelayQueue* q = NULL;

  timer = taosMemoryCalloc(1, sizeof(uv_timer_t));
  if (timer == NULL) {
    return terrno;
  }

  heap = heapCreate(timeCompare);
  if (heap == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _return1);
  }

  q = taosMemoryCalloc(1, sizeof(SDelayQueue));
  if (q == NULL) {
    TAOS_CHECK_GOTO(terrno, NULL, _return1);
  }
  q->heap = heap;
  q->timer = timer;
  q->loop = loop;
  q->timer->data = q;

  int err = uv_timer_init(loop, timer);
  if (err != 0) {
    TAOS_CHECK_GOTO(TSDB_CODE_THIRDPARTY_ERROR, NULL, _return1);
  }

  *queue = q;
  return 0;

_return1:
  taosMemoryFree(timer);
  heapDestroy(heap);
  taosMemoryFree(q);
  return TSDB_CODE_OUT_OF_MEMORY;
}

void transDQDestroy(SDelayQueue* queue, void (*freeFunc)(void* arg)) {
  if (queue == NULL) {
    return;
  }
  taosMemoryFree(queue->timer);

  while (heapSize(queue->heap) > 0) {
    HeapNode* minNode = heapMin(queue->heap);
    if (minNode == NULL) {
      return;
    }
    heapRemove(queue->heap, minNode);

    SDelayTask* task = container_of(minNode, SDelayTask, node);

    STaskArg* arg = task->arg;
    if (freeFunc) freeFunc(arg);
    taosMemoryFree(arg);

    taosMemoryFree(task);
  }
  heapDestroy(queue->heap);
  taosMemoryFree(queue);
}
void transDQCancel(SDelayQueue* queue, SDelayTask* task) {
  TAOS_UNUSED(uv_timer_stop(queue->timer));

  if (heapSize(queue->heap) <= 0) {
    taosMemoryFree(task->arg);
    taosMemoryFree(task);
    return;
  }
  heapRemove(queue->heap, &task->node);

  taosMemoryFree(task->arg);
  taosMemoryFree(task);

  if (heapSize(queue->heap) != 0) {
    HeapNode* minNode = heapMin(queue->heap);
    if (minNode == NULL) return;

    uint64_t    now = taosGetTimestampMs();
    SDelayTask* task = container_of(minNode, SDelayTask, node);
    uint64_t    timeout = now > task->execTime ? now - task->execTime : 0;

    TAOS_UNUSED(uv_timer_start(queue->timer, transDQTimeout, timeout, 0));
  }
}

SDelayTask* transDQSched(SDelayQueue* queue, void (*func)(void* arg), void* arg, uint64_t timeoutMs) {
  uint64_t    now = taosGetTimestampMs();
  SDelayTask* task = taosMemoryCalloc(1, sizeof(SDelayTask));
  if (task == NULL) {
    return NULL;
  }

  task->func = func;
  task->arg = arg;
  task->execTime = now + timeoutMs;

  HeapNode* minNode = heapMin(queue->heap);
  if (minNode) {
    SDelayTask* minTask = container_of(minNode, SDelayTask, node);
    if (minTask->execTime < task->execTime) {
      timeoutMs = minTask->execTime <= now ? 0 : minTask->execTime - now;
    }
  }

  tTrace("timer %p put task into delay queue, timeoutMs:%" PRIu64, queue->timer, timeoutMs);
  heapInsert(queue->heap, &task->node);
  TAOS_UNUSED(uv_timer_start(queue->timer, transDQTimeout, timeoutMs, 0));
  return task;
}

#if 0
void transPrintEpSet(SEpSet* pEpSet) {
  if (pEpSet == NULL) {
    tTrace("NULL epset");
    return;
  }
  char buf[512] = {0};
  int  len = snprintf(buf, sizeof(buf), "epset:{");
  for (int i = 0; i < pEpSet->numOfEps; i++) {
    if (i == pEpSet->numOfEps - 1) {
      len += snprintf(buf + len, sizeof(buf) - len, "%d. %s:%d", i, pEpSet->eps[i].fqdn, pEpSet->eps[i].port);
    } else {
      len += snprintf(buf + len, sizeof(buf) - len, "%d. %s:%d, ", i, pEpSet->eps[i].fqdn, pEpSet->eps[i].port);
    }
  }
  len += snprintf(buf + len, sizeof(buf) - len, "}");
  tTrace("%s, inUse:%d", buf, pEpSet->inUse);
}
bool transReqEpsetIsEqual(SReqEpSet* a, SReqEpSet* b) {
  if (a == NULL && b == NULL) {
    return true;
  } else if (a == NULL || b == NULL) {
    return false;
  }

  if (a->numOfEps != b->numOfEps || a->inUse != b->inUse) {
    return false;
  }
  for (int i = 0; i < a->numOfEps; i++) {
    if (strncmp(a->eps[i].fqdn, b->eps[i].fqdn, TSDB_FQDN_LEN) != 0 || a->eps[i].port != b->eps[i].port) {
      return false;
    }
  }
  return true;
}
bool transCompareReqAndUserEpset(SReqEpSet* a, SEpSet* b) {
  if (a->numOfEps != b->numOfEps) {
    return false;
  }
  for (int i = 0; i < a->numOfEps; i++) {
    if (strncmp(a->eps[i].fqdn, b->eps[i].fqdn, TSDB_FQDN_LEN) != 0 || a->eps[i].port != b->eps[i].port) {
      return false;
    }
  }
  return true;
}
#endif

static void transInitEnv() {
  refMgt = transOpenRefMgt(50000, transDestroyExHandle);
  svrRefMgt = transOpenRefMgt(50000, transDestroyExHandle);
  instMgt = taosOpenRef(50, rpcCloseImpl);
  transCacheInit();

  transSyncMsgMgt = taosOpenRef(50, transDestroySyncMsg);
  TAOS_UNUSED(uv_os_setenv("UV_TCP_SINGLE_ACCEPT", "1"));

  saslLibInit();
}
static void transDestroyEnv() {
  transCloseRefMgt(refMgt);
  transCloseRefMgt(svrRefMgt);
  transCloseRefMgt(instMgt);
  transCloseRefMgt(transSyncMsgMgt);
}

int32_t transInit() {
  // init env
  int32_t code = taosThreadOnce(&transModuleInit, transInitEnv);
  if (code != 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
  }
  return code;
}

int32_t transGetRefMgt() { return refMgt; }
int32_t transGetSvrRefMgt() { return svrRefMgt; }
int32_t transGetInstMgt() { return instMgt; }
int32_t transGetSyncMsgMgt() { return transSyncMsgMgt; }

void transCleanup() {
  // clean env
  transDestroyEnv();
}
int32_t transOpenRefMgt(int size, void (*func)(void*)) {
  // added into once later
  return taosOpenRef(size, func);
}
void transCloseRefMgt(int32_t mgt) {
  // close ref
  taosCloseRef(mgt);
}
int64_t transAddExHandle(int32_t refMgt, void* p) {
  // acquire extern handle
  return taosAddRef(refMgt, p);
}
void transRemoveExHandle(int32_t refMgt, int64_t refId) {
  // acquire extern handle
  int32_t code = taosRemoveRef(refMgt, refId);
  if (code != 0) {
    tTrace("failed to remove %" PRId64 " from resetId:%d", refId, refMgt);
  }
}

void* transAcquireExHandle(int32_t refMgt, int64_t refId) {  // acquire extern handle
  return (void*)taosAcquireRef(refMgt, refId);
}

void transReleaseExHandle(int32_t refMgt, int64_t refId) {
  // release extern handle
  int32_t code = taosReleaseRef(refMgt, refId);
  if (code != 0) {
    tTrace("failed to release %" PRId64 " from resetId:%d", refId, refMgt);
  }
}
void transDestroyExHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  SExHandle* eh = handle;
  tDebug("trans destroy sid:%" PRId64 ", memory %p", eh->refId, handle);
  taosMemoryFree(handle);
}

void transDestroySyncMsg(void* msg) {
  if (msg == NULL) return;

  STransSyncMsg* pSyncMsg = msg;
  TAOS_UNUSED(tsem2_destroy(pSyncMsg->pSem));
  taosMemoryFree(pSyncMsg->pSem);
  transFreeMsg(pSyncMsg->pRsp->pCont);
  taosMemoryFree(pSyncMsg->pRsp);
  taosMemoryFree(pSyncMsg);
}

uint32_t subnetIpRang2Int(SIpV4Range* pRange) {
  uint32_t ip = pRange->ip;
  return ((ip & 0xFF) << 24) | ((ip & 0xFF00) << 8) | ((ip & 0xFF0000) >> 8) | ((ip >> 24) & 0xFF);
}
int32_t subnetInit(SubnetUtils* pUtils, SIpV4Range* pRange) {
  if (pRange->mask == 32) {
    pUtils->type = 0;
    pUtils->address = pRange->ip;
    return 0;
  }
  pUtils->address = subnetIpRang2Int(pRange);

  for (int i = 0; i < pRange->mask; i++) {
    pUtils->netmask |= (1 << (31 - i));
  }

  pUtils->network = pUtils->address & pUtils->netmask;
  pUtils->broadcast = (pUtils->network) | (pUtils->netmask ^ 0xFFFFFFFF);
  pUtils->type = (pRange->mask == 32 ? 0 : 1);

  return 0;
}
int32_t subnetCheckIp(SubnetUtils* pUtils, uint32_t ip) {
  // impl later
  if (pUtils == NULL) return false;
  if (pUtils->type == 0) {
    return pUtils->address == ip;
  } else {
    SIpV4Range range = {.ip = ip, .mask = 32};

    uint32_t t = subnetIpRang2Int(&range);
    return t >= pUtils->network && t <= pUtils->broadcast;
  }
}

int32_t transUtilSIpRangeToStr(SIpV4Range* pRange, char* buf, int32_t cap) {
  int32_t len = 0;

  struct in_addr addr;
  addr.s_addr = pRange->ip;

  int32_t err = uv_inet_ntop(AF_INET, &addr, buf, 32);
  if (err != 0) {
    tError("failed to convert ip to string since %s", uv_strerror(err));
    return TSDB_CODE_THIRDPARTY_ERROR;
  }

  len = strlen(buf);

  if (pRange->mask != 32) {
    len += snprintf(buf + len, cap - len, "/%d", pRange->mask);
  }
  return len;
}

int32_t transUtilSWhiteListToStr(SIpWhiteListDual* pList, char** ppBuf) {
  int32_t code = 0;
  int32_t lino = 0;
  char*   pBuf = NULL;
  int32_t len = 0;
  if (pList->num == 0) {
    TSDB_CHECK_CODE(code = TSDB_CODE_INVALID_PARA, lino, _error);
  }
  int32_t cap = pList->num * IP_RESERVE_CAP;
  pBuf = taosMemoryCalloc(1, cap);
  if (pBuf == NULL) {
    TSDB_CHECK_CODE(code = terrno, lino, _error);
  }

  for (int i = 0; i < pList->num; i++) {
    SIpRange* pRange = &pList->pIpRanges[i];
    SIpAddr   addr = {0};
    code = tIpUintToStr(pRange, &addr);
    TSDB_CHECK_CODE(code, lino, _error);

    len += snprintf(pBuf + len, cap - (len), "%s,", IP_ADDR_STR(&addr));
  }
  if (len > 0) {
    pBuf[len - 1] = 0;
  }

  *ppBuf = pBuf;
_error:
  if (code != 0) {
    taosMemoryFree(pBuf);
    *ppBuf = NULL;
  }

  return len;
}

bool transUtilCheckDualIp(SIpRange* range, SIpRange* ip) {
  SIpV6Range* p6 = &range->ipV6;
  SIpV6Range* pIp = &ip->ipV6;

  if (p6->mask == 0) {
    return true;
  } else if (p6->mask == 128) {
    return p6->addr[0] == pIp->addr[0] && p6->addr[1] == pIp->addr[1];
  }

  uint64_t maskHigh = 0, maskLow = 0;
  if (p6->mask <= 64) {
    maskHigh = (0xFFFFFFFFFFFFFFFFULL << (64 - p6->mask));
    maskLow = 0;
  } else {
    maskHigh = 0xFFFFFFFFFFFFFFFFULL;
    maskLow = (0xFFFFFFFFFFFFFFFFULL << (128 - p6->mask));
  }

  return ((pIp->addr[0] & maskHigh) == (p6->addr[0] & maskHigh)) &&
         ((pIp->addr[1] & maskLow) == (p6->addr[1] & maskLow));
}

int32_t initWQ(queue* wq) {
  int32_t code = 0;
  QUEUE_INIT(wq);
  for (int i = 0; i < 4; i++) {
    SWReqsWrapper* w = taosMemoryCalloc(1, sizeof(SWReqsWrapper));
    if (w == NULL) {
      TAOS_CHECK_GOTO(terrno, NULL, _exception);
    }
    w->wreq.data = w;
    w->arg = NULL;
    QUEUE_INIT(&w->node);
    QUEUE_PUSH(wq, &w->q);
  }
  return 0;
_exception:
  destroyWQ(wq);
  return code;
}
void destroyWQ(queue* wq) {
  while (!QUEUE_IS_EMPTY(wq)) {
    queue* h = QUEUE_HEAD(wq);
    QUEUE_REMOVE(h);
    SWReqsWrapper* w = QUEUE_DATA(h, SWReqsWrapper, q);
    taosMemoryFree(w);
  }
}

uv_write_t* allocWReqFromWQ(queue* wq, void* arg) {
  if (!QUEUE_IS_EMPTY(wq)) {
    queue* node = QUEUE_HEAD(wq);
    QUEUE_REMOVE(node);
    SWReqsWrapper* w = QUEUE_DATA(node, SWReqsWrapper, q);
    w->arg = arg;
    QUEUE_INIT(&w->node);

    return &w->wreq;
  } else {
    SWReqsWrapper* w = taosMemoryCalloc(1, sizeof(SWReqsWrapper));
    if (w == NULL) {
      return NULL;
    }
    w->wreq.data = w;
    w->arg = arg;
    QUEUE_INIT(&w->node);
    return &w->wreq;
  }
}

void freeWReqToWQ(queue* wq, SWReqsWrapper* w) {
  QUEUE_INIT(&w->node);
  QUEUE_PUSH(wq, &w->q);
}

int32_t transSetReadOption(uv_handle_t* handle) {
  int32_t code = 0;
  int32_t fd;
  int     ret = uv_fileno((uv_handle_t*)handle, &fd);
  if (ret != 0) {
    tWarn("failed to get fd since %s", uv_err_name(ret));
    return TSDB_CODE_THIRDPARTY_ERROR;
  }
  code = taosSetSockOpt2(fd);
  return code;
}

int32_t transCreateReqEpsetFromUserEpset(const SEpSet* pEpset, SReqEpSet** pReqEpSet) {
  if (pEpset == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  if (pReqEpSet == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  int32_t    size = sizeof(SReqEpSet) + sizeof(SEp) * pEpset->numOfEps;
  SReqEpSet* pReq = (SReqEpSet*)taosMemoryCalloc(1, size);
  if (pReq == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  memcpy((char*)pReq, (char*)pEpset, size);
  // clear previous
  taosMemoryFree(*pReqEpSet);

  if (transValidReqEpset(pReq) != TSDB_CODE_SUCCESS) {
    taosMemoryFree(pReq);
    return TSDB_CODE_INVALID_PARA;
  }

  *pReqEpSet = pReq;
  return TSDB_CODE_SUCCESS;
}

int32_t transCreateUserEpsetFromReqEpset(const SReqEpSet* pReqEpSet, SEpSet* pEpSet) {
  if (pReqEpSet == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }
  memcpy((char*)pEpSet, (char*)pReqEpSet, sizeof(SReqEpSet) + sizeof(SEp) * pReqEpSet->numOfEps);
  return TSDB_CODE_SUCCESS;
}

int32_t transValidReqEpset(SReqEpSet* pReqEpSet) {
  if (pReqEpSet == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }
  if (pReqEpSet->numOfEps == 0 || pReqEpSet->numOfEps > TSDB_MAX_EP_NUM || pReqEpSet->inUse >= TSDB_MAX_EP_NUM) {
    return TSDB_CODE_INVALID_PARA;
  }
  return TSDB_CODE_SUCCESS;
}

#else
#define BUFFER_CAP 4096

typedef struct {
  int32_t      numOfThread;
  STaosQueue** qhandle;
  STaosQset**  qset;
  int64_t      idx;

} MultiThreadQhandle;
typedef struct TThread {
  TdThread thread;
  int      idx;
} TThread;

TdThreadMutex       mutex[2];
MultiThreadQhandle* multiQ[2] = {NULL, NULL};
static TdThreadOnce transModuleInit = PTHREAD_ONCE_INIT;

static int32_t refMgt;
static int32_t svrRefMgt;
static int32_t instMgt;
static int32_t transSyncMsgMgt;
TdThreadMutex  mutex[2];

TdThreadMutex tableMutex;
SHashObj*     hashTable = NULL;

void transDestroySyncMsg(void* msg);

int32_t transCompressMsg(char* msg, int32_t len) {
  int32_t        ret = 0;
  int            compHdr = sizeof(STransCompMsg);
  STransMsgHead* pHead = transHeadFromCont(msg);

  char* buf = taosMemoryMalloc(len + compHdr + 8);  // 8 extra bytes
  if (buf == NULL) {
    tWarn("failed to allocate memory for rpc msg compression, contLen:%d", len);
    ret = len;
    return ret;
  }

  int32_t clen = LZ4_compress_default(msg, buf, len, len + compHdr);
  /*
   * only the compressed size is less than the value of contLen - overhead, the compression is applied
   * The first four bytes is set to 0, the second four bytes are utilized to keep the original length of message
   */
  if (clen > 0 && clen < len - compHdr) {
    STransCompMsg* pComp = (STransCompMsg*)msg;
    pComp->reserved = 0;
    pComp->contLen = htonl(len);
    memcpy(msg + compHdr, buf, clen);

    tDebug("compress rpc msg, before:%d, after:%d", len, clen);
    ret = clen + compHdr;
    pHead->comp = 1;
  } else {
    ret = len;
    pHead->comp = 0;
  }
  taosMemoryFree(buf);
  return ret;
}
int32_t transDecompressMsg(char** msg, int32_t* len) { return 0; }

void transFreeMsg(void* msg) {
  if (msg == NULL) {
    return;
  }
  tTrace("rpc free cont:%p", (char*)msg - TRANS_MSG_OVERHEAD);
  taosMemoryFree((char*)msg - sizeof(STransMsgHead));
}

void transCtxInit(STransCtx* ctx) {
  // init transCtx
  ctx->args = taosHashInit(2, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UINT), true, HASH_NO_LOCK);
}
void transCtxCleanup(STransCtx* ctx) {
  if (ctx == NULL || ctx->args == NULL) {
    return;
  }

  STransCtxVal* iter = taosHashIterate(ctx->args, NULL);
  while (iter) {
    ctx->freeFunc(iter->val);
    iter = taosHashIterate(ctx->args, iter);
  }
  if (ctx->freeFunc) ctx->freeFunc(ctx->brokenVal.val);
  taosHashCleanup(ctx->args);
  ctx->args = NULL;
}

void transCtxMerge(STransCtx* dst, STransCtx* src) {
  if (src->args == NULL || src->freeFunc == NULL) {
    return;
  }
  if (dst->args == NULL) {
    dst->args = src->args;
    dst->brokenVal = src->brokenVal;
    dst->freeFunc = src->freeFunc;
    src->args = NULL;
    return;
  }
  void*  key = NULL;
  size_t klen = 0;
  void*  iter = taosHashIterate(src->args, NULL);
  while (iter) {
    STransCtxVal* sVal = (STransCtxVal*)iter;
    key = taosHashGetKey(sVal, &klen);

    int32_t code = taosHashPut(dst->args, key, klen, sVal, sizeof(*sVal));
    if (code != 0) {
      tError("failed to put val to hash, reason:%s", tstrerror(code));
    }
    iter = taosHashIterate(src->args, iter);
  }
  taosHashCleanup(src->args);
}
void* transCtxDumpVal(STransCtx* ctx, int32_t key) {
  if (ctx->args == NULL) {
    return NULL;
  }
  STransCtxVal* cVal = taosHashGet(ctx->args, (const void*)&key, sizeof(key));
  if (cVal == NULL) {
    return NULL;
  }
  void* ret = NULL;
  TAOS_UNUSED((*cVal->clone)(cVal->val, &ret));
  return ret;
}
void* transCtxDumpBrokenlinkVal(STransCtx* ctx, int32_t* msgType) {
  void* ret = NULL;
  if (ctx->brokenVal.clone == NULL) {
    return ret;
  }
  TAOS_UNUSED((*ctx->brokenVal.clone)(ctx->brokenVal.val, &ret));

  *msgType = ctx->brokenVal.msgType;

  return ret;
}

bool cliMayGetAhandle(STrans* pTrans, SRpcMsg* pMsg) {
  int64_t  seq = pMsg->info.seq;
  int32_t* msgType = NULL;

  if (pMsg->msgType == TDMT_SCH_TASK_RELEASE || pMsg->msgType == TDMT_SCH_TASK_RELEASE + 1) {
    STransCtx* ctx = taosHashGet(pTrans->sidTable, &pMsg->info.qId, sizeof(pMsg->info.qId));
    transCtxCleanup(ctx);
    taosHashRemove(pTrans->sidTable, &pMsg->info.qId, sizeof(pMsg->info.qId));
    return true;
  }
  taosThreadMutexLock(&pTrans->seqMutex);
  msgType = taosHashGet(pTrans->seqTable, &seq, sizeof(seq));
  taosThreadMutexUnlock(&pTrans->seqMutex);
  if (msgType == NULL) {
    STransCtx* ctx = taosHashGet(pTrans->sidTable, &pMsg->info.qId, sizeof(pMsg->info.qId));
    if (ctx == NULL) {
      return false;
    }
    pMsg->info.ahandle = transCtxDumpVal(ctx, pMsg->msgType);
    tError("failed to find msg type for seq:%" PRId64 ", gen ahandle for type %s", seq, TMSG_INFO(pMsg->msgType));
  } else {
    taosThreadMutexLock(&pTrans->seqMutex);
    taosHashRemove(pTrans->seqTable, &seq, sizeof(seq));
    msgType = taosHashGet(pTrans->seqTable, &seq, sizeof(seq));
    taosThreadMutexUnlock(&pTrans->seqMutex);
  }
  return true;
}

void* processSvrMsg(void* arg) {
  TThread* thread = (TThread*)arg;

  int32_t    idx = thread->idx;
  static int num = 0;
  STaosQall* qall;
  SRpcMsg *  pRpcMsg, rpcMsg;
  int        type;
  SQueueInfo qinfo = {0};

  taosAllocateQall(&qall);

  while (1) {
    int numOfMsgs = taosReadAllQitemsFromQset(multiQ[1]->qset[idx], qall, &qinfo);
    if (numOfMsgs <= 0) break;
    taosResetQitems(qall);
    for (int i = 0; i < numOfMsgs; i++) {
      taosGetQitem(qall, (void**)&pRpcMsg);
      taosThreadMutexLock(&mutex[1]);
      RpcCfp    fp = NULL;
      void*     parent = NULL;
      STraceId* trace = &pRpcMsg->info.traceId;
      tGDebug("taos %s received from taosd", TMSG_INFO(pRpcMsg->msgType));
      STrans* pTrans = NULL;
      transGetCb(pRpcMsg->type, &pTrans);

      taosThreadMutexUnlock(&mutex[1]);

      if (pTrans != NULL) {
        if (cliMayGetAhandle(pTrans, pRpcMsg)) {
          if (pRpcMsg->info.reqWithSem == NULL) {
            (pTrans->cfp)(pTrans->parent, pRpcMsg, NULL);
          } else {
            STransReqWithSem* reqWithSem = pRpcMsg->info.reqWithSem;
            memcpy(&reqWithSem->pMsg, pRpcMsg, sizeof(SRpcMsg));
            tsem_post(reqWithSem->sem);
          }
        } else {
          tDebug("taosd %s received from taosd, ignore", TMSG_INFO(pRpcMsg->msgType));
        }
      }
      taosFreeQitem(pRpcMsg);
    }
    taosUpdateItemSize(qinfo.queue, numOfMsgs);
  }

  taosFreeQall(qall);
  return NULL;
}
void* procClientMsg(void* arg) {
  TThread* thread = (TThread*)arg;

  int32_t    idx = thread->idx;
  static int num = 0;
  STaosQall* qall;
  SRpcMsg *  pRpcMsg, rpcMsg;
  int        type;
  SQueueInfo qinfo = {0};

  taosAllocateQall(&qall);

  while (1) {
    int numOfMsgs = taosReadAllQitemsFromQset(multiQ[0]->qset[idx], qall, &qinfo);
    tDebug("%d msgs are received", numOfMsgs);
    if (numOfMsgs <= 0) break;
    taosResetQitems(qall);
    for (int i = 0; i < numOfMsgs; i++) {
      taosGetQitem(qall, (void**)&pRpcMsg);

      STraceId* trace = &pRpcMsg->info.traceId;
      tDebug("taosc %s received from taosc", TMSG_INFO(pRpcMsg->msgType));
      RpcCfp fp = NULL;
      // void*  parent;
      STrans* pTrans = NULL;
      taosThreadMutexLock(&mutex[1]);
      if ((pRpcMsg->type & TD_ASTRA_DSVR) != 0) {
        transGetCb(TD_ASTRA_DSVR, &pTrans);
      }
      taosThreadMutexUnlock(&mutex[1]);
      if (pTrans->cfp != NULL) {
        (pTrans->cfp)(pTrans->parent, pRpcMsg, NULL);
      } else {
        tError("taosc failed to find callback for msg type:%s", TMSG_INFO(pRpcMsg->msgType));
      }
      taosFreeQitem(pRpcMsg);
    }
    taosUpdateItemSize(qinfo.queue, numOfMsgs);
  }

  taosFreeQall(qall);
  return NULL;
}
static void transInitEnv() {
  refMgt = transOpenRefMgt(50000, transDestroyExHandle);
  svrRefMgt = transOpenRefMgt(50000, transDestroyExHandle);
  instMgt = taosOpenRef(50, rpcCloseImpl);
  transSyncMsgMgt = taosOpenRef(50, transDestroySyncMsg);

  taosThreadMutexInit(&tableMutex, NULL);
  hashTable = taosHashInit(2, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UINT), true, HASH_NO_LOCK);

  int32_t numOfAthread = 1;

  multiQ[0] = taosMemoryMalloc(sizeof(MultiThreadQhandle));
  multiQ[0]->numOfThread = numOfAthread;
  multiQ[0]->qhandle = (STaosQueue**)taosMemoryMalloc(sizeof(STaosQueue*) * numOfAthread);
  multiQ[0]->qset = (STaosQset**)taosMemoryMalloc(sizeof(STaosQset*) * numOfAthread);

  taosThreadMutexInit(&mutex[0], NULL);

  for (int i = 0; i < numOfAthread; i++) {
    taosOpenQueue(&(multiQ[0]->qhandle[i]));
    taosOpenQset(&multiQ[0]->qset[i]);
    taosAddIntoQset(multiQ[0]->qset[i], multiQ[0]->qhandle[i], NULL);
  }
  {
    TThread* threads = taosMemoryMalloc(sizeof(TThread) * numOfAthread);
    for (int i = 0; i < numOfAthread; i++) {
      threads[i].idx = i;
      taosThreadCreate(&(threads[i].thread), NULL, procClientMsg, (void*)&threads[i]);
    }
  }

  multiQ[1] = taosMemoryMalloc(sizeof(MultiThreadQhandle));
  multiQ[1]->numOfThread = numOfAthread;
  multiQ[1]->qhandle = (STaosQueue**)taosMemoryMalloc(sizeof(STaosQueue*) * numOfAthread);
  multiQ[1]->qset = (STaosQset**)taosMemoryMalloc(sizeof(STaosQset*) * numOfAthread);
  taosThreadMutexInit(&mutex[1], NULL);

  for (int i = 0; i < numOfAthread; i++) {
    taosOpenQueue(&(multiQ[1]->qhandle[i]));
    taosOpenQset(&multiQ[1]->qset[i]);
    taosAddIntoQset(multiQ[1]->qset[i], multiQ[1]->qhandle[i], NULL);
  }
  {
    TThread* threads = taosMemoryMalloc(sizeof(TThread) * numOfAthread);
    for (int i = 0; i < numOfAthread; i++) {
      threads[i].idx = i;
      taosThreadCreate(&(threads[i].thread), NULL, processSvrMsg, (void*)&threads[i]);
    }
  }
}
static void transDestroyEnv() {
  transCloseRefMgt(refMgt);
  transCloseRefMgt(svrRefMgt);
}

typedef struct {
  void (*fp)(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet);
  RPC_TYPE type;
  void*    parant;
  STrans*  pTransport;
} FP_TYPE;
int32_t transUpdateCb(RPC_TYPE type, STrans* pTransport) {
  taosThreadMutexLock(&tableMutex);

  FP_TYPE t = {.type = type, .pTransport = pTransport};
  int32_t code = taosHashPut(hashTable, &type, sizeof(type), &t, sizeof(t));
  taosThreadMutexUnlock(&tableMutex);
  return 0;
}
int32_t transGetCb(RPC_TYPE type, STrans** ppTransport) {
  taosThreadMutexLock(&tableMutex);
  void* p = taosHashGet(hashTable, &type, sizeof(type));
  if (p == NULL) {
    taosThreadMutexUnlock(&tableMutex);
    return TSDB_CODE_INVALID_MSG;
  }
  FP_TYPE* t = p;
  *ppTransport = t->pTransport;
  // *fp = t->fp;
  // *arg = t->parant;
  taosThreadMutexUnlock(&tableMutex);
  return 0;
}

int32_t transSendReq(STrans* pTransport, SRpcMsg* pMsg, void* pEpSet) {
  SRpcMsg* pTemp;

  taosAllocateQitem(sizeof(SRpcMsg), DEF_QITEM, 0, (void**)&pTemp);
  memcpy(pTemp, pMsg, sizeof(SRpcMsg));

  int64_t cidx = multiQ[0]->idx++;
  int32_t idx = cidx % (multiQ[0]->numOfThread);
  tDebug("taos request is sent , type:%s, contLen:%d, item:%p", TMSG_INFO(pMsg->msgType), pMsg->contLen, pTemp);
  taosWriteQitem(multiQ[0]->qhandle[idx], pTemp);
  return 0;
}
int32_t transSendResp(const SRpcMsg* pMsg) {
  SRpcMsg* pTemp;

  taosAllocateQitem(sizeof(SRpcMsg), DEF_QITEM, 0, (void**)&pTemp);
  memcpy(pTemp, pMsg, sizeof(SRpcMsg));

  int64_t cidx = multiQ[1]->idx++;
  int32_t idx = cidx % (multiQ[1]->numOfThread);
  tDebug("taos resp is sent to, type:%s, contLen:%d, item:%p", TMSG_INFO(pMsg->msgType), pMsg->contLen, pTemp);
  taosWriteQitem(multiQ[1]->qhandle[idx], pTemp);
  return 0;
}

int32_t transInit() {
  // init env
  int32_t code = taosThreadOnce(&transModuleInit, transInitEnv);
  if (code != 0) {
    code = TAOS_SYSTEM_ERROR(ERRNO);
  }
  return code;
}

int32_t transGetRefMgt() { return refMgt; }
int32_t transGetSvrRefMgt() { return svrRefMgt; }
int32_t transGetInstMgt() { return instMgt; }
int32_t transGetSyncMsgMgt() { return transSyncMsgMgt; }

void transCleanup() {
  // clean env
  transDestroyEnv();
  return;
}
int32_t transOpenRefMgt(int size, void (*func)(void*)) {
  /// add later
  return taosOpenRef(size, func);
}
void transCloseRefMgt(int32_t mgt) {
  // close ref
  taosCloseRef(mgt);
  return;
}
int64_t transAddExHandle(int32_t refMgt, void* p) {
  return taosAddRef(refMgt, p);
  // acquire extern handle
}
void transRemoveExHandle(int32_t refMgt, int64_t refId) {
  // acquire extern handle
  int32_t code = taosRemoveRef(refMgt, refId);
  return;
}

void* transAcquireExHandle(int32_t refMgt, int64_t refId) {
  // acquire extern handle
  return (void*)taosAcquireRef(refMgt, refId);
}

void transReleaseExHandle(int32_t refMgt, int64_t refId) {
  // release extern handle
  int32_t code = taosReleaseRef(refMgt, refId);
  return;
}
void transDestroyExHandle(void* handle) {
  if (handle == NULL) {
    return;
  }
  SExHandle* eh = handle;
  if (!QUEUE_IS_EMPTY(&eh->q)) {
    tDebug("handle %p mem leak", handle);
  }
  tDebug("free exhandle %p", handle);
  taosMemoryFree(handle);
  return;
}

void transDestroySyncMsg(void* msg) {
  if (msg == NULL) return;

  STransSyncMsg* pSyncMsg = msg;
  TAOS_UNUSED(tsem2_destroy(pSyncMsg->pSem));
  taosMemoryFree(pSyncMsg->pSem);
  transFreeMsg(pSyncMsg->pRsp->pCont);
  taosMemoryFree(pSyncMsg->pRsp);
  taosMemoryFree(pSyncMsg);
  return;
}

uint32_t subnetIpRang2Int(SIpV4Range* pRange) { return 0; }
int32_t  subnetInit(SubnetUtils* pUtils, SIpV4Range* pRange) { return 0; }
int32_t  subnetCheckIp(SubnetUtils* pUtils, uint32_t ip) { return 0; }

int32_t transUtilSIpRangeToStr(SIpV4Range* pRange, char* buf) { return 0; }

int32_t transUtilSWhiteListToStr(SIpWhiteList* pList, char** ppBuf) { return 0; }

int32_t transInitBuffer(SConnBuffer* buf) {
  buf->buf = taosMemoryCalloc(1, BUFFER_CAP);
  if (buf->buf == NULL) {
    return terrno;
  }

  buf->cap = BUFFER_CAP;
  buf->left = -1;
  buf->len = 0;
  buf->total = 0;
  buf->invalid = 0;
  return 0;
}
void transDestroyBuffer(SConnBuffer* p) {
  taosMemoryFree(p->buf);
  p->buf = NULL;
}

int32_t transClearBuffer(SConnBuffer* buf) {
  SConnBuffer* p = buf;
  if (p->cap > BUFFER_CAP) {
    p->cap = BUFFER_CAP;
    p->buf = taosMemoryRealloc(p->buf, BUFFER_CAP);
    if (p->buf == NULL) {
      return terrno;
    }
  }
  p->left = -1;
  p->len = 0;
  p->total = 0;
  p->invalid = 0;
  return 0;
}

int32_t transDumpFromBuffer(SConnBuffer* connBuf, char** buf, int8_t resetBuf, int32_t* len) {
  static const int HEADSIZE = sizeof(STransMsgHead);
  int32_t          code = 0;
  SConnBuffer*     p = connBuf;
  if (p->left != 0 || p->total <= 0) {
    return TSDB_CODE_INVALID_MSG;
  }
  int total = p->total;
  if (total >= HEADSIZE && !p->invalid) {
    *buf = taosMemoryCalloc(1, total);
    if (*buf == NULL) {
      return terrno;
    }
    memcpy(*buf, p->buf, total);
    if ((code = transResetBuffer(connBuf, resetBuf)) < 0) {
      return code;
    }
  } else {
    total = -1;
    code = TSDB_CODE_INVALID_MSG;
  }
  *len = total;
  return code;
}

int32_t transResetBuffer(SConnBuffer* connBuf, int8_t resetBuf) {
  SConnBuffer* p = connBuf;
  if (p->total < p->len) {
    int left = p->len - p->total;
    memmove(p->buf, p->buf + p->total, left);
    p->left = -1;
    p->total = 0;
    p->len = left;
  } else if (p->total == p->len) {
    p->left = -1;
    p->total = 0;
    p->len = 0;
    if (p->cap > BUFFER_CAP) {
      if (resetBuf) {
        p->cap = BUFFER_CAP;
        p->buf = taosMemoryRealloc(p->buf, p->cap);
        if (p->buf == NULL) {
          return terrno;
        }
      }
    }
  } else {
    tError("failed to reset buffer, total:%d, len:%d since %s", p->total, p->len, tstrerror(TSDB_CODE_INVALID_MSG));
    return TSDB_CODE_INVALID_MSG;
  }
  return 0;
}

int32_t transCreateReqEpsetFromUserEpset(const SEpSet* pEpset, SReqEpSet** pReqEpSet) {
  if (pEpset == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  if (pReqEpSet == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  int32_t    size = sizeof(SReqEpSet) + sizeof(SEp) * pEpset->numOfEps;
  SReqEpSet* pReq = (SReqEpSet*)taosMemoryCalloc(1, size);
  if (pReq == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  memcpy((char*)pReq, (char*)pEpset, size);
  // clear previous
  taosMemoryFree(*pReqEpSet);

  if (transValidReqEpset(pReq) != TSDB_CODE_SUCCESS) {
    taosMemoryFree(pReq);
    return TSDB_CODE_INVALID_PARA;
  }

  *pReqEpSet = pReq;
  return TSDB_CODE_SUCCESS;
}

int32_t transCreateUserEpsetFromReqEpset(const SReqEpSet* pReqEpSet, SEpSet* pEpSet) {
  if (pReqEpSet == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }
  memcpy((char*)pEpSet, (char*)pReqEpSet, sizeof(SReqEpSet) + sizeof(SEp) * pReqEpSet->numOfEps);
  return TSDB_CODE_SUCCESS;
}

int32_t transValidReqEpset(SReqEpSet* pReqEpSet) {
  if (pReqEpSet == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }
  if (pReqEpSet->numOfEps == 0 || pReqEpSet->numOfEps > TSDB_MAX_EP_NUM || pReqEpSet->inUse >= TSDB_MAX_EP_NUM) {
    return TSDB_CODE_INVALID_PARA;
  }
  return TSDB_CODE_SUCCESS;
}

#endif  // TD_ASTRA_RPC

int32_t transQueueInit(STransQueue* wq, void (*freeFunc)(void* arg)) {
  QUEUE_INIT(&wq->node);
  wq->freeFunc = (void (*)(void*))freeFunc;
  wq->size = 0;
  wq->inited = 1;
  return 0;
}
void transQueuePush(STransQueue* q, void* arg) {
  queue* node = arg;
  QUEUE_PUSH(&q->node, node);
  q->size++;
}
void* transQueuePop(STransQueue* q) {
  if (q->size == 0) return NULL;

  queue* head = QUEUE_HEAD(&q->node);
  QUEUE_REMOVE(head);
  q->size--;
  return head;
}
int32_t transQueueSize(STransQueue* q) { return q->size; }

void* transQueueGet(STransQueue* q, int idx) {
  if (q->size == 0) return NULL;

  while (idx-- > 0) {
    queue* node = QUEUE_NEXT(&q->node);
    if (node == &q->node) return NULL;
  }
  return NULL;
}

void transQueueRemoveByFilter(STransQueue* q, bool (*filter)(void* e, void* arg), void* arg, void* dst, int32_t size) {
  queue* d = dst;
  queue* node = QUEUE_NEXT(&q->node);
  while (node != &q->node) {
    queue* next = QUEUE_NEXT(node);
    if (filter && filter(node, arg)) {
      QUEUE_REMOVE(node);
      q->size--;
      QUEUE_PUSH(d, node);
      if (--size == 0) {
        break;
      }
    }
    node = next;
  }
}

void* tranQueueHead(STransQueue* q) {
  if (q->size == 0) return NULL;

  queue* head = QUEUE_HEAD(&q->node);
  return head;
}

void* transQueueRm(STransQueue* q, int i) {
  // if (queue->q == NULL || taosArrayGetSize(queue->q) == 0) {
  //   return NULL;
  // }
  // if (i >= taosArrayGetSize(queue->q)) {
  //   return NULL;
  // }
  // void* ptr = taosArrayGetP(queue->q, i);
  // taosArrayRemove(queue->q, i);
  // return ptr;
  return NULL;
}

void transQueueRemove(STransQueue* q, void* e) {
  if (q->size == 0) return;
  queue* node = e;
  QUEUE_REMOVE(node);
  q->size--;
}

bool transQueueEmpty(STransQueue* q) { return q->size == 0 ? true : false; }

void transQueueClear(STransQueue* q) {
  if (q->inited == 0) return;
  while (!QUEUE_IS_EMPTY(&q->node)) {
    queue* h = QUEUE_HEAD(&q->node);
    QUEUE_REMOVE(h);
    if (q->freeFunc != NULL) (q->freeFunc)(h);
    q->size--;
  }
}
void transQueueDestroy(STransQueue* q) { transQueueClear(q); }

void transPrintEpSet(SEpSet* pEpSet) {
  if (pEpSet == NULL) {
    tTrace("NULL epset");
    return;
  }
  char buf[512] = {0};
  int  len = snprintf(buf, sizeof(buf), "epset:{");
  for (int i = 0; i < pEpSet->numOfEps; i++) {
    if (i == pEpSet->numOfEps - 1) {
      len += snprintf(buf + len, sizeof(buf) - len, "%d. %s:%d", i, pEpSet->eps[i].fqdn, pEpSet->eps[i].port);
    } else {
      len += snprintf(buf + len, sizeof(buf) - len, "%d. %s:%d, ", i, pEpSet->eps[i].fqdn, pEpSet->eps[i].port);
    }
  }
  len += snprintf(buf + len, sizeof(buf) - len, "}");
  tTrace("%s, inUse:%d", buf, pEpSet->inUse);
}
bool transReqEpsetIsEqual(SReqEpSet* a, SReqEpSet* b) {
  if (a == NULL && b == NULL) {
    return true;
  } else if (a == NULL || b == NULL) {
    return false;
  }

  if (a->numOfEps != b->numOfEps || a->inUse != b->inUse) {
    return false;
  }
  for (int i = 0; i < a->numOfEps; i++) {
    int32_t l1 = strlen(a->eps[i].fqdn);
    int32_t l2 = strlen(b->eps[i].fqdn);
    if (l1 >= TSDB_FQDN_LEN || l2 >= TSDB_FQDN_LEN) {
      tWarn("get invalid epset, a:%s, b:%s", a->eps[i].fqdn, b->eps[i].fqdn);
      return false;
    }

    if (l1 != l2 || strncmp(a->eps[i].fqdn, b->eps[i].fqdn, l1) != 0 || a->eps[i].port != b->eps[i].port) {
      return false;
    }
  }
  return true;
}
bool transCompareReqAndUserEpset(SReqEpSet* a, SEpSet* b) {
  if (a->numOfEps != b->numOfEps) {
    return false;
  }
  for (int i = 0; i < a->numOfEps; i++) {
    if (strncmp(a->eps[i].fqdn, b->eps[i].fqdn, TSDB_FQDN_LEN) != 0 || a->eps[i].port != b->eps[i].port) {
      return false;
    }
  }
  return true;
}

int32_t transReloadTlsConfig(void* handle, int8_t type) {
  int32_t code = 0;

   
  if (type == TAOS_CONN_CLIENT) {
    code = transReloadClientTlsConfig(handle);
  } else if (type == TAOS_CONN_SERVER) {
    code = transReloadServerTlsConfig(handle);
  } else {
    code = TSDB_CODE_INVALID_PARA;
  }
  return code;
}
STrans* transInstAcquire(int64_t mgtId, int64_t instId) {
  STrans* ppInst = NULL;
  int32_t code = transCacheAcquireById(instId, &ppInst);
  if (code == TSDB_CODE_SUCCESS) {
    return ppInst;
  } else {
    return NULL;
  }
}

void transInstRelease(int64_t instId) { transCacheReleaseByRefId(instId); }

taosdata / TDengine / #5011

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous