trans.c 2.4 KB
Newer Older
dengyihao's avatar
dengyihao 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/*
 * 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/>.
 */

#ifdef USE_UV

#include "transComm.h"

dengyihao's avatar
dengyihao 已提交
20
void* (*taosInitHandle[])(uint32_t ip, uint32_t port, char* label, int numOfThreads, void* fp, void* shandle) = {
dengyihao's avatar
dengyihao 已提交
21
    taosInitServer, taosInitClient};
dengyihao's avatar
dengyihao 已提交
22
void (*taosCloseHandle[])(void* arg) = {taosCloseServer, taosCloseClient};
dengyihao's avatar
dengyihao 已提交
23 24 25 26 27 28 29 30 31

void* rpcOpen(const SRpcInit* pInit) {
  SRpcInfo* pRpc = calloc(1, sizeof(SRpcInfo));
  if (pRpc == NULL) {
    return NULL;
  }
  if (pInit->label) {
    tstrncpy(pRpc->label, pInit->label, strlen(pInit->label));
  }
dengyihao's avatar
dengyihao 已提交
32
  pRpc->cfp = pInit->cfp;
dengyihao's avatar
dengyihao 已提交
33 34
  pRpc->numOfThreads = pInit->numOfThreads > TSDB_MAX_RPC_THREADS ? TSDB_MAX_RPC_THREADS : pInit->numOfThreads;
  pRpc->connType = pInit->connType;
35
  pRpc->idleTime = pInit->idleTime;
dengyihao's avatar
dengyihao 已提交
36
  pRpc->tcphandle = (*taosInitHandle[pRpc->connType])(0, pInit->localPort, pRpc->label, pRpc->numOfThreads, NULL, pRpc);
dengyihao's avatar
dengyihao 已提交
37 38 39

  return pRpc;
}
dengyihao's avatar
dengyihao 已提交
40 41 42
void rpcClose(void* arg) {
  SRpcInfo* pRpc = (SRpcInfo*)arg;
  (*taosCloseHandle[pRpc->connType])(pRpc->tcphandle);
dengyihao's avatar
dengyihao 已提交
43
  free(pRpc);
dengyihao's avatar
dengyihao 已提交
44 45
  return;
}
dengyihao's avatar
dengyihao 已提交
46
void* rpcMallocCont(int contLen) {
dengyihao's avatar
dengyihao 已提交
47
  int size = contLen + TRANS_MSG_OVERHEAD;
dengyihao's avatar
dengyihao 已提交
48 49 50 51 52 53 54 55

  char* start = (char*)calloc(1, (size_t)size);
  if (start == NULL) {
    tError("failed to malloc msg, size:%d", size);
    return NULL;
  } else {
    tTrace("malloc mem:%p size:%d", start, size);
  }
dengyihao's avatar
dengyihao 已提交
56
  return start + sizeof(STransMsgHead);
dengyihao's avatar
dengyihao 已提交
57
}
dengyihao's avatar
dengyihao 已提交
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
void  rpcFreeCont(void* cont) { return; }
void* rpcReallocCont(void* ptr, int contLen) { return NULL; }

void rpcSendRedirectRsp(void* pConn, const SEpSet* pEpSet) {}
int  rpcGetConnInfo(void* thandle, SRpcConnInfo* pInfo) { return -1; }
void rpcSendRecv(void* shandle, SEpSet* pEpSet, SRpcMsg* pReq, SRpcMsg* pRsp) { return; }
int  rpcReportProgress(void* pConn, char* pCont, int contLen) { return -1; }
void rpcCancelRequest(int64_t rid) { return; }

int32_t rpcInit(void) {
  // impl later
  return -1;
}

void rpcCleanup(void) {
  // impl later
dengyihao's avatar
dengyihao 已提交
74
  //
dengyihao's avatar
dengyihao 已提交
75 76 77
  return;
}
#endif