sut.cpp 5.0 KB
Newer Older
S
Shengliang Guan 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

S
Shengliang Guan 已提交
16
#include "sut.h"
S
Shengliang Guan 已提交
17 18

void Testbase::InitLog(const char* path) {
S
Shengliang Guan 已提交
19
  dDebugFlag = 143;
S
Shengliang Guan 已提交
20
  vDebugFlag = 0;
S
Shengliang Guan 已提交
21
  mDebugFlag = 143;
S
Shengliang Guan 已提交
22 23 24
  cDebugFlag = 0;
  jniDebugFlag = 0;
  tmrDebugFlag = 0;
S
Shengliang Guan 已提交
25
  uDebugFlag = 0;
S
Shengliang Guan 已提交
26 27 28 29 30 31 32
  rpcDebugFlag = 0;
  qDebugFlag = 0;
  wDebugFlag = 0;
  sDebugFlag = 0;
  tsdbDebugFlag = 0;
  cqDebugFlag = 0;
  tscEmbeddedInUtil = 1;
S
Shengliang Guan 已提交
33
  tsAsyncLog = 0;
S
Shengliang Guan 已提交
34 35 36 37 38 39 40 41 42 43 44 45

  taosRemoveDir(path);
  taosMkDir(path);

  char temp[PATH_MAX];
  snprintf(temp, PATH_MAX, "%s/taosdlog", path);
  if (taosInitLog(temp, tsNumOfLogLines, 1) != 0) {
    printf("failed to init log file\n");
  }
}

void Testbase::Init(const char* path, int16_t port) {
S
Shengliang Guan 已提交
46
  SDnodeEnvCfg cfg = {0};
S
Shengliang Guan 已提交
47
  cfg.numOfCommitThreads = 1;
S
Shengliang Guan 已提交
48
  cfg.numOfCores = 1;
S
Shengliang Guan 已提交
49 50
  dndInit(&cfg);

S
Shengliang Guan 已提交
51 52 53 54
  char fqdn[] = "localhost";
  char firstEp[TSDB_EP_LEN] = {0};
  snprintf(firstEp, TSDB_EP_LEN, "%s:%u", fqdn, port);

S
Shengliang Guan 已提交
55
  InitLog("/tmp/td");
S
Shengliang Guan 已提交
56 57 58 59 60 61 62 63
  server.Start(path, fqdn, port, firstEp);
  client.Init("root", "taosdata", fqdn, port);
  taosMsleep(1100);
}

void Testbase::Cleanup() {
  server.Stop();
  client.Cleanup();
S
Shengliang Guan 已提交
64
  dndCleanup();
S
Shengliang Guan 已提交
65 66
}

S
Shengliang Guan 已提交
67 68
void Testbase::Restart() { server.Restart(); }

S
Shengliang Guan 已提交
69 70 71 72
void Testbase::ServerStop() { server.Stop(); }

void Testbase::ServerStart() { server.DoStart(); }

S
Shengliang Guan 已提交
73
SRpcMsg* Testbase::SendReq(tmsg_t msgType, void* pCont, int32_t contLen) {
S
Shengliang Guan 已提交
74 75
  SRpcMsg rpcMsg = {0};
  rpcMsg.pCont = pCont;
S
Shengliang Guan 已提交
76
  rpcMsg.contLen = contLen;
S
Shengliang Guan 已提交
77 78
  rpcMsg.msgType = msgType;

S
Shengliang Guan 已提交
79
  return client.SendReq(&rpcMsg);
S
Shengliang Guan 已提交
80 81
}

S
Shengliang Guan 已提交
82 83 84
void Testbase::SendShowMetaReq(int8_t showType, const char* db) {
  int32_t   contLen = sizeof(SShowReq);
  SShowReq* pShow = (SShowReq*)rpcMallocCont(contLen);
S
Shengliang Guan 已提交
85
  pShow->type = showType;
S
Shengliang Guan 已提交
86
  strcpy(pShow->db, db);
S
Shengliang Guan 已提交
87

S
Shengliang Guan 已提交
88 89
  SRpcMsg*  pRsp = SendReq(TDMT_MND_SHOW, pShow, contLen);
  SShowRsp* pShowRsp = (SShowRsp*)pRsp->pCont;
S
Shengliang Guan 已提交
90 91

  ASSERT(pShowRsp != nullptr);
S
Shengliang Guan 已提交
92
  pShowRsp->showId = htobe64(pShowRsp->showId);
S
Shengliang Guan 已提交
93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
  pMeta = &pShowRsp->tableMeta;
  pMeta->numOfTags = htonl(pMeta->numOfTags);
  pMeta->numOfColumns = htonl(pMeta->numOfColumns);
  pMeta->sversion = htonl(pMeta->sversion);
  pMeta->tversion = htonl(pMeta->tversion);
  pMeta->tuid = htobe64(pMeta->tuid);
  pMeta->suid = htobe64(pMeta->suid);

  showId = pShowRsp->showId;
}

int32_t Testbase::GetMetaColId(int32_t index) {
  SSchema* pSchema = &pMeta->pSchema[index];
  pSchema->colId = htonl(pSchema->colId);
  return pSchema->colId;
}

int8_t Testbase::GetMetaType(int32_t index) {
  SSchema* pSchema = &pMeta->pSchema[index];
  return pSchema->type;
}

int32_t Testbase::GetMetaBytes(int32_t index) {
  SSchema* pSchema = &pMeta->pSchema[index];
  pSchema->bytes = htonl(pSchema->bytes);
  return pSchema->bytes;
}

const char* Testbase::GetMetaName(int32_t index) {
  SSchema* pSchema = &pMeta->pSchema[index];
  return pSchema->name;
}

int32_t Testbase::GetMetaNum() { return pMeta->numOfColumns; }

const char* Testbase::GetMetaTbName() { return pMeta->tbFname; }

S
Shengliang Guan 已提交
130 131
void Testbase::SendShowRetrieveReq() {
  int32_t contLen = sizeof(SRetrieveTableReq);
S
Shengliang Guan 已提交
132

S
Shengliang Guan 已提交
133
  SRetrieveTableReq* pRetrieve = (SRetrieveTableReq*)rpcMallocCont(contLen);
S
Shengliang Guan 已提交
134
  pRetrieve->showId = htobe64(showId);
S
Shengliang Guan 已提交
135 136
  pRetrieve->free = 0;

S
Shengliang Guan 已提交
137 138
  SRpcMsg* pRsp = SendReq(TDMT_MND_SHOW_RETRIEVE, pRetrieve, contLen);
  pRetrieveRsp = (SRetrieveTableRsp*)pRsp->pCont;
S
Shengliang Guan 已提交
139 140 141
  pRetrieveRsp->numOfRows = htonl(pRetrieveRsp->numOfRows);
  pRetrieveRsp->useconds = htobe64(pRetrieveRsp->useconds);
  pRetrieveRsp->compLen = htonl(pRetrieveRsp->compLen);
S
Shengliang Guan 已提交
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183

  pData = pRetrieveRsp->data;
  pos = 0;
}

const char* Testbase::GetShowName() { return pMeta->tbFname; }

int8_t Testbase::GetShowInt8() {
  int8_t data = *((int8_t*)(pData + pos));
  pos += sizeof(int8_t);
  return data;
}

int16_t Testbase::GetShowInt16() {
  int16_t data = *((int16_t*)(pData + pos));
  pos += sizeof(int16_t);
  return data;
}

int32_t Testbase::GetShowInt32() {
  int32_t data = *((int32_t*)(pData + pos));
  pos += sizeof(int32_t);
  return data;
}

int64_t Testbase::GetShowInt64() {
  int64_t data = *((int64_t*)(pData + pos));
  pos += sizeof(int64_t);
  return data;
}

int64_t Testbase::GetShowTimestamp() {
  int64_t data = *((int64_t*)(pData + pos));
  pos += sizeof(int64_t);
  return data;
}

const char* Testbase::GetShowBinary(int32_t len) {
  pos += sizeof(VarDataLenT);
  char* data = (char*)(pData + pos);
  pos += len;
  return data;
S
Shengliang Guan 已提交
184 185 186 187
}

int32_t Testbase::GetShowRows() { return pRetrieveRsp->numOfRows; }

S
Shengliang Guan 已提交
188
STableMetaRsp* Testbase::GetShowMeta() { return pMeta; }
S
Shengliang Guan 已提交
189 190

SRetrieveTableRsp* Testbase::GetRetrieveRsp() { return pRetrieveRsp; }