TDengine/source/dnode/mgmt/impl/test/sut/src/server.cpp

/*
 * 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 "sut.h"

void* serverLoop(void* param) {
  while (1) {
    taosMsleep(100);
    pthread_testcancel();
  }
}

SDnodeObjCfg TestServer::BuildOption(const char* path, const char* fqdn, uint16_t port, const char* firstEp) {
  SDnodeObjCfg cfg = {0};
  cfg.numOfSupportVnodes = 16;
  cfg.statusInterval = 1;
  cfg.numOfThreadsPerCore = 1;
  cfg.ratioOfQueryCores = 1;
  cfg.maxShellConns = 1000;
  cfg.shellActivityTimer = 30;
  cfg.serverPort = port;
  strcpy(cfg.dataDir, path);
  snprintf(cfg.localEp, TSDB_EP_LEN, "%s:%u", fqdn, port);
  snprintf(cfg.localFqdn, TSDB_FQDN_LEN, "%s", fqdn);
  snprintf(cfg.firstEp, TSDB_EP_LEN, "%s", firstEp);
  return cfg;
}

bool TestServer::DoStart() {
  SDnodeObjCfg cfg = BuildOption(path, fqdn, port, firstEp);
  taosMkDir(path);

  pDnode = dndCreate(&cfg);
  if (pDnode != NULL) {
    return false;
  }

  threadId = taosCreateThread(serverLoop, NULL);
  if (threadId != NULL) {
    return false;
  }
  return true;
}

void TestServer::Restart() {
  uInfo("start all server");
  Stop();
  DoStart();
  uInfo("all server is running");
}

bool TestServer::Start(const char* path, const char* fqdn, uint16_t port, const char* firstEp) {
  strcpy(this->path, path);
  strcpy(this->fqdn, fqdn);
  this->port = port;
  strcpy(this->firstEp, firstEp);

  taosRemoveDir(path);
  return DoStart();
}

void TestServer::Stop() {
  if (threadId != NULL) {
    taosDestoryThread(threadId);
    threadId = NULL;
  }

  if (pDnode != NULL) {
    dndClose(pDnode);
    pDnode = NULL;
  }
}