From 7316fb89796120fc274700f109612eb1eade2cbd Mon Sep 17 00:00:00 2001
From: lichuang <lichuang1982@gmail.com>
Date: Thu, 20 May 2021 11:06:28 +0800
Subject: [PATCH] [TD-1920]add mnode compact wal functions

---
 src/common/inc/tglobal.h    |  1 +
 src/common/src/tglobal.c    |  1 +
 src/dnode/src/dnodeMain.c   |  1 +
 src/dnode/src/dnodeSystem.c |  2 ++
 src/inc/mnode.h             |  2 ++
 src/mnode/inc/mnodeSdb.h    |  1 +
 src/mnode/src/mnodeMain.c   |  6 ++++++
 src/mnode/src/mnodeSdb.c    | 12 ++++++++++++
 8 files changed, 26 insertions(+)

diff --git a/src/common/inc/tglobal.h b/src/common/inc/tglobal.h
index 2f4aa4c2b2..abb4d39ee1 100644
--- a/src/common/inc/tglobal.h
+++ b/src/common/inc/tglobal.h
@@ -142,6 +142,7 @@ extern int32_t tsMonitorInterval;
 extern int8_t tsEnableStream;
 
 // internal
+extern int8_t  tsCompactMnodeWal;
 extern int8_t  tsPrintAuth;
 extern int8_t  tscEmbedded;
 extern char    configDir[];
diff --git a/src/common/src/tglobal.c b/src/common/src/tglobal.c
index db97c3a5af..ab036b021c 100644
--- a/src/common/src/tglobal.c
+++ b/src/common/src/tglobal.c
@@ -176,6 +176,7 @@ int32_t tsMonitorInterval = 30;  // seconds
 int8_t  tsEnableStream = 1;
 
 // internal
+int8_t tsCompactMnodeWal = 0;
 int8_t tsPrintAuth = 0;
 int8_t tscEmbedded = 0;
 char   configDir[TSDB_FILENAME_LEN] = {0};
diff --git a/src/dnode/src/dnodeMain.c b/src/dnode/src/dnodeMain.c
index 410e6bb188..48680e20d9 100644
--- a/src/dnode/src/dnodeMain.c
+++ b/src/dnode/src/dnodeMain.c
@@ -40,6 +40,7 @@
 #include "dnodeShell.h"
 #include "dnodeTelemetry.h"
 #include "module.h"
+#include "mnode.h"
 
 #if !defined(_MODULE) || !defined(_TD_LINUX)
 int32_t moduleStart() { return 0; }
diff --git a/src/dnode/src/dnodeSystem.c b/src/dnode/src/dnodeSystem.c
index e49b3eba99..ee37ffdcbb 100644
--- a/src/dnode/src/dnodeSystem.c
+++ b/src/dnode/src/dnodeSystem.c
@@ -42,6 +42,8 @@ int32_t main(int32_t argc, char *argv[]) {
       }
     } else if (strcmp(argv[i], "-C") == 0) {
       dump_config = 1;
+    } else if (strcmp(argv[i], "--compact-mnode-wal") == 0) {
+      tsCompactMnodeWal = 1;
     } else if (strcmp(argv[i], "-V") == 0) {
 #ifdef _ACCT
       char *versionStr = "enterprise";
diff --git a/src/inc/mnode.h b/src/inc/mnode.h
index 2495a42ba2..c66dc60a1b 100644
--- a/src/inc/mnode.h
+++ b/src/inc/mnode.h
@@ -73,6 +73,8 @@ int32_t mnodeProcessPeerReq(SMnodeMsg *pMsg);
 void    mnodeProcessPeerRsp(SRpcMsg *pMsg);
 int32_t mnodeRetriveAuth(char *user, char *spi, char *encrypt, char *secret, char *ckey);
 
+int32_t mnodeCompactWal();
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/src/mnode/inc/mnodeSdb.h b/src/mnode/inc/mnodeSdb.h
index e4e4a7a054..13dd8bd06b 100644
--- a/src/mnode/inc/mnodeSdb.h
+++ b/src/mnode/inc/mnodeSdb.h
@@ -106,6 +106,7 @@ int32_t  sdbGetId(void *pTable);
 uint64_t sdbGetVersion();
 bool     sdbCheckRowDeleted(void *pTable, void *pRow);
 
+int32_t mnodeCompactWal();
 #ifdef __cplusplus
 }
 #endif
diff --git a/src/mnode/src/mnodeMain.c b/src/mnode/src/mnodeMain.c
index 7ef0488c42..76eebe0d01 100644
--- a/src/mnode/src/mnodeMain.c
+++ b/src/mnode/src/mnodeMain.c
@@ -57,6 +57,12 @@ static SStep tsMnodeSteps[] = {
   {"show",    mnodeInitShow,    mnodeCleanUpShow}
 };
 
+/*
+static SStep tsMnodeCompactSteps[] = {
+  {"cluster", mnodeInitCluster, NULL},
+};
+*/
+
 static void mnodeInitTimer();
 static void mnodeCleanupTimer();
 static bool mnodeNeedStart() ;
diff --git a/src/mnode/src/mnodeSdb.c b/src/mnode/src/mnodeSdb.c
index 505d3c519c..64c7713164 100644
--- a/src/mnode/src/mnodeSdb.c
+++ b/src/mnode/src/mnodeSdb.c
@@ -450,6 +450,12 @@ int32_t sdbInit() {
   }
 
   tsSdbMgmt.status = SDB_STATUS_SERVING;
+
+  if (tsCompactMnodeWal) {
+    mnodeCompactWal();
+    exit(EXIT_SUCCESS);
+  }
+
   return TSDB_CODE_SUCCESS;
 }
 
@@ -1138,3 +1144,9 @@ static void *sdbWorkerFp(void *pWorker) {
 int32_t sdbGetReplicaNum() {
   return tsSdbMgmt.cfg.replica;
 }
+
+int32_t mnodeCompactWal() {
+  sdbInfo("vgId:1, start compact mnode wal...");
+
+  return 0;
+}
\ No newline at end of file
-- 
GitLab