Resolve toolchain installation random issue in CI

Fix #644

Signed-off-by: Wei Liu <[email protected]>

.github/workflows/integration-tdx.yml

@@ -18,7 +18,7 @@ jobs:
   system_compile:
     name: Run TDX Integration Test
     runs-on: [self-hosted, tdx]
-    timeout-minutes: 30
+    timeout-minutes: 45
 
     steps:
       # Install first since it's needed to build NASM
@@ -34,12 +34,7 @@ jobs:
           submodules: recursive
 
       - name: Install toolchain
-        uses: actions-rs/toolchain@v1
-        with:
-          profile: ${{ env.TOOLCHAIN_PROFILE }}
-          toolchain: ${{ env.RUST_TOOLCHAIN }}
-          override: true
-          components: rust-src, llvm-tools-preview
+        run: bash sh_script/update_toolchain.sh ${{ env.RUST_TOOLCHAIN }}
 
       - name: Run cargo install cargo-xbuild
         uses: actions-rs/cargo@v1

sh_script/update_toolchain.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+TOOLCHAIN_VER=$1
+TRY_TIMES=5
+echo ${TOOLCHAIN_VER}
+
+while [ ${TRY_TIMES} -gt 0 ]
+do
+    exist=`rustup toolchain list | grep ${TOOLCHAIN_VER} | wc -l`
+    if [[ ${exist} == 0 ]]
+    then
+        rustup toolchain install ${TOOLCHAIN_VER} --component rust-src
+    else
+        rustup component add rust-src
+        echo "Toolchain ${TOOLCHAIN_VER} is installed."
+        break
+    fi
+    sleep 30
+    let "TRY_TIMES--"
+done
+
+if [[ ${TRY_TIMES} == 0 ]]
+then
+    echo "Install toolchian ${TOOLCHAIN_VER} failed."
+    exit 1
+fi