From 3ff56696aebcd4512c8fb5cec8a7e4e85ab2ac53 Mon Sep 17 00:00:00 2001
From: chenhaoran <haoran920c@163.com>
Date: Fri, 7 Feb 2025 11:40:56 +0800
Subject: [PATCH] fix: exit if branch checkout fails

---
 cmake/taosws_CMakeLists.txt.in | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/cmake/taosws_CMakeLists.txt.in b/cmake/taosws_CMakeLists.txt.in
index b013d45911..dc32dead52 100644
--- a/cmake/taosws_CMakeLists.txt.in
+++ b/cmake/taosws_CMakeLists.txt.in
@@ -2,7 +2,7 @@
 # taosws-rs
 ExternalProject_Add(taosws-rs
         GIT_REPOSITORY https://github.com/taosdata/taos-connector-rust.git
-        GIT_TAG main
+        GIT_TAG 3.0
         SOURCE_DIR "${TD_SOURCE_DIR}/tools/taosws-rs"
         BINARY_DIR ""
         #BUILD_IN_SOURCE TRUE
@@ -10,4 +10,13 @@ ExternalProject_Add(taosws-rs
         BUILD_COMMAND ""
         INSTALL_COMMAND ""
         TEST_COMMAND ""
+        # Fix: Exit if branch checkout fails
+        UPDATE_COMMAND "
+                cd ${TD_SOURCE_DIR}/tools/taosws-rs &&
+                git checkout -f 3.0 
+                if [ $(git rev-parse --abbrev-ref HEAD) != '3.0' ]; then
+                echo 'Failed to switch to branch 3.0 in ${TD_SOURCE_DIR}/tools/taosws-rs';
+                exit 1;
+                fi
+        "
 )