From 1102a88ad062265d92b823a3d94b64227b70ead6 Mon Sep 17 00:00:00 2001
From: gccgdb1234 <wxzhang@taosdata.com>
Date: Mon, 7 Nov 2022 17:35:45 +0800
Subject: [PATCH] bench

---
 source/libs/transport/test/cliBench.c | 22 ++++++++++++++++++++--
 source/libs/transport/test/svrBench.c | 18 ++++++++++++++++--
 2 files changed, 36 insertions(+), 4 deletions(-)

diff --git a/source/libs/transport/test/cliBench.c b/source/libs/transport/test/cliBench.c
index a296625ace..01e88b9988 100644
--- a/source/libs/transport/test/cliBench.c
+++ b/source/libs/transport/test/cliBench.c
@@ -32,6 +32,22 @@ typedef struct {
   void    *pRpc;
 } SInfo;
 
+
+void initLogEnv() {
+  const char  *logDir =  "/tmp/trans_cli"; 
+  const char*  defaultLogFileNamePrefix = "taoslog";
+  const int32_t maxLogFileNum = 10000;
+  tsAsyncLog = 0;
+  //idxDebugFlag = 143;
+  strcpy(tsLogDir, (char *)logDir);
+  taosRemoveDir(tsLogDir);
+  taosMkDir(tsLogDir); 
+     
+  if (taosInitLog(defaultLogFileNamePrefix, maxLogFileNum) < 0) {
+   printf("failed to open log file in directory:%s\n", tsLogDir);  
+  }
+}
+  
 static void processResponse(void *parent, SRpcMsg *pMsg, SEpSet *pEpSet) {
   SInfo *pInfo = (SInfo *)pMsg->info.ahandle;
   tDebug("thread:%d, response is received, type:%d contLen:%d code:0x%x", pInfo->index, pMsg->msgType, pMsg->contLen,
@@ -98,7 +114,7 @@ int main(int argc, char *argv[]) {
   rpcInit.user = "michael";
   rpcInit.connType = TAOS_CONN_CLIENT;
 
-  rpcDebugFlag = 131;
+  rpcDebugFlag = 135;
   for (int i = 1; i < argc; ++i) {
     if (strcmp(argv[i], "-p") == 0 && i < argc - 1) {
     } else if (strcmp(argv[i], "-i") == 0 && i < argc - 1) {
@@ -132,7 +148,9 @@ int main(int argc, char *argv[]) {
       exit(0);
     }
   }
-  taosInitLog("client.log", 100000);
+   
+
+  initLogEnv();
 
   void *pRpc = rpcOpen(&rpcInit);
   if (pRpc == NULL) {
diff --git a/source/libs/transport/test/svrBench.c b/source/libs/transport/test/svrBench.c
index ce465d989a..7e9cfe0c61 100644
--- a/source/libs/transport/test/svrBench.c
+++ b/source/libs/transport/test/svrBench.c
@@ -26,6 +26,20 @@ TdFilePtr   pDataFile = NULL;
 STaosQueue *qhandle = NULL;
 STaosQset  *qset = NULL;
 
+void initLogEnv() {
+  const char  *logDir =  "/tmp/trans_svr"; 
+  const char*  defaultLogFileNamePrefix = "taoslog";
+  const int32_t maxLogFileNum = 10000;
+  tsAsyncLog = 0;
+  //idxDebugFlag = 143;
+  strcpy(tsLogDir, logDir);
+  taosRemoveDir(tsLogDir);
+  taosMkDir(tsLogDir); 
+     
+  if (taosInitLog(defaultLogFileNamePrefix, maxLogFileNum) < 0) {
+   printf("failed to open log file in directory:%s\n", tsLogDir);  
+  }
+}
 void processShellMsg() {
   static int num = 0;
   STaosQall *qall;
@@ -147,9 +161,9 @@ int main(int argc, char *argv[]) {
     }
   }
 
-  tsAsyncLog = 0;
   rpcInit.connType = TAOS_CONN_SERVER;
-  taosInitLog("server.log", 100000);
+  
+  initLogEnv();
 
   void *pRpc = rpcOpen(&rpcInit);
   if (pRpc == NULL) {
-- 
GitLab