From aea910b8793d1ad60e8a0bd998e0102f26f87557 Mon Sep 17 00:00:00 2001
From: dmchen <cademfly@hotmail.com>
Date: Tue, 5 Nov 2024 19:16:40 +0800
Subject: [PATCH] fix/TS-5251-add-conflict-check-add-test

---
 tests/army/cluster/compactDBConflict.py | 147 ++++++++++++++++++++++++
 1 file changed, 147 insertions(+)
 create mode 100644 tests/army/cluster/compactDBConflict.py

diff --git a/tests/army/cluster/compactDBConflict.py b/tests/army/cluster/compactDBConflict.py
new file mode 100644
index 0000000000..f02272cc1f
--- /dev/null
+++ b/tests/army/cluster/compactDBConflict.py
@@ -0,0 +1,147 @@
+###################################################################
+#           Copyright (c) 2016 by TAOS Technologies, Inc.
+#                     All rights reserved.
+#
+#  This file is proprietary and confidential to TAOS Technologies.
+#  No part of this file may be reproduced, stored, transmitted,
+#  disclosed or used in any form or by any means other than as
+#  expressly provided by the written permission from Jianhui Tao
+#
+###################################################################
+
+# -*- coding: utf-8 -*-
+import frame.etool
+from frame.caseBase import *
+from frame.cases import *
+from frame import *
+import json
+import threading
+
+
+class TDTestCase(TBase):
+  def init(self, conn, logSql, replicaVar=1):
+        tdLog.debug(f"start to init {__file__}")
+        self.replicaVar = int(replicaVar)
+        tdSql.init(conn.cursor(), logSql)  # output sql.txt file
+        self.configJsonFile('splitVgroupByLearner.json', 'db', 4, 1, 'splitVgroupByLearner.json', 100000)
+
+  def configJsonFile(self, fileName, dbName, vgroups, replica, newFileName='', insert_rows=100000,
+                       timestamp_step=10000):
+    tdLog.debug(f"configJsonFile {fileName}")
+    filePath = etool.curFile(__file__, fileName)
+    with open(filePath, 'r') as f:
+        data = json.load(f)
+
+    if len(newFileName) == 0:
+        newFileName = fileName
+
+    data['databases'][0]['dbinfo']['name'] = dbName
+    data['databases'][0]['dbinfo']['vgroups'] = vgroups
+    data['databases'][0]['dbinfo']['replica'] = replica
+    data['databases'][0]['super_tables'][0]['insert_rows'] = insert_rows
+    data['databases'][0]['super_tables'][0]['timestamp_step'] = timestamp_step
+    json_data = json.dumps(data)
+    filePath = etool.curFile(__file__, newFileName)
+    with open(filePath, "w") as file:
+        file.write(json_data)
+
+    tdLog.debug(f"configJsonFile {json_data}")
+
+  def insertData(self, configFile):
+    tdLog.info(f"insert data.")
+    # taosBenchmark run
+    jfile = etool.curFile(__file__, configFile)
+    etool.benchMark(json=jfile)
+
+  def run(self):
+        tdLog.debug(f"start to excute {__file__}")
+
+        self.insertData('splitVgroupByLearner.json')
+
+        '''
+        t1 = threading.Thread(target=self.alterDBThread)
+        t1.start()
+        tdLog.debug("threading started!!!!!")
+        time.sleep(3)
+        tdSql.error('compact database db;', expectErrInfo="Conflict transaction not completed")
+
+        t1.join()
+
+
+        t2 = threading.Thread(target=self.splitVgroupThread)
+        t2.start()
+        tdLog.debug("threading started!!!!!")
+        time.sleep(2)
+        tdSql.error('compact database db;', expectErrInfo="Conflict transaction not completed")
+
+        t2.join()
+        '''
+        t3 = threading.Thread(target=self.RedistributeVGroups)
+        t3.start()
+        tdLog.debug("threading started!!!!!")
+        #time.sleep(2)
+        tdSql.error('compact database db;', expectErrInfo="Conflict transaction not completed")
+
+        t3.join()
+
+        t4 = threading.Thread(target=self.balanceVGROUPThread)
+        t4.start()
+        tdLog.debug("threading started!!!!!")
+        #time.sleep(2)
+        tdSql.error('compact database db;', expectErrInfo="Conflict transaction not completed")
+
+        t4.join()
+
+
+  def alterDBThread(self):
+    tdLog.info("alter db start")
+    tdSql.execute('ALTER DATABASE db REPLICA 3')
+    if self.waitTransactionZero() is False:
+            tdLog.info(f"transaction not finished")
+
+  def balanceVGROUPThread(self):
+    tdLog.info("balance VGROUP start")
+    tdSql.execute('BALANCE VGROUP')
+    if self.waitTransactionZero() is False:
+            tdLog.info(f"transaction not finished")
+
+  def RedistributeVGroups(self):
+        sql = f"REDISTRIBUTE VGROUP 5 DNODE 1"
+        tdSql.execute(sql, show=True)
+        if self.waitTransactionZero() is False:
+            tdLog.exit(f"{sql} transaction not finished")
+            return False
+  
+        sql = f"REDISTRIBUTE VGROUP 4 DNODE 1"
+        tdSql.execute(sql, show=True)
+        if self.waitTransactionZero() is False:
+            tdLog.exit(f"{sql} transaction not finished")
+            return False
+        
+        sql = f"REDISTRIBUTE VGROUP 3 DNODE 1"
+        tdSql.execute(sql, show=True)
+        if self.waitTransactionZero() is False:
+            tdLog.exit(f"{sql} transaction not finished")
+            return False
+        
+        return True
+
+  def splitVgroupThread(self):
+        tdSql.execute('use db')
+        rowLen = tdSql.query('show vgroups')
+        if rowLen > 0:
+            vgroupId = tdSql.getData(0, 0)
+            tdLog.debug(f"splitVgroupThread vgroupId:{vgroupId}")
+            tdSql.execute(f"split vgroup {vgroupId}")
+        else:
+            tdLog.exit("get vgroupId fail!")
+        if self.waitTransactionZero() is False:
+            tdLog.info(f"transaction not finished")
+
+  def stop(self):
+    tdSql.close()
+    tdLog.success(f"{__file__} successfully executed")
+
+
+tdCases.addLinux(__file__, TDTestCase())
+tdCases.addWindows(__file__, TDTestCase())
\ No newline at end of file