From 7687198026427f628a5e581682a099666c840e40 Mon Sep 17 00:00:00 2001
From: nmreadelf <frank.q.kong@hotmail.com>
Date: Fri, 28 Jul 2023 06:41:27 +0800
Subject: [PATCH] Add arm build image

---
 .github/workflows/build_arm_whells.yml | 93 ++++++++++++++++++--------
 1 file changed, 65 insertions(+), 28 deletions(-)

diff --git a/.github/workflows/build_arm_whells.yml b/.github/workflows/build_arm_whells.yml
index 1bd6f1a5a86..2fd85eaf3d6 100644
--- a/.github/workflows/build_arm_whells.yml
+++ b/.github/workflows/build_arm_whells.yml
@@ -1,9 +1,15 @@
 name: Build arm64
 
 on:
+  # for test action
+  pull_request:
+    branches:
+      - main
+      - feat/add-arm-wheel-build
+  # end for test action
   workflow_dispatch:
     inputs:
-      TAG_NAME:
+  https://github.com/NixOS/patchelf/releases/download/0.18.0/patchelf-0.18.0-aarch64.tar.gz     TAG_NAME:
         description: 'Release Version Tag'
         required: true
   release:
@@ -24,7 +30,7 @@ jobs:
       matrix:
         os: [ ubuntu-20.04 ]
         # python-version: [ "3.8", "3.9", "3.10", "3.11"]
-        python-version: [ "3.11" ]
+        python-version: [ "3.8" ]
     env:
       RUNNER_OS: ${{ matrix.os }}
       PYTHON_VERSION: ${{ matrix.python-version }}
@@ -33,9 +39,20 @@ jobs:
         run: |
           export PYENV_ROOT="$HOME/.pyenv"
           export PATH="$PYENV_ROOT/bin:$PATH"
-          pyenv local "${{ matrix.python-version }}"
           python --version
         shell: bash -l {0}
+      - name: Install clang++ for Ubuntu
+        if: matrix.os == 'ubuntu-20.04'
+        run: |
+          pwd
+          uname -a
+          wget https://apt.llvm.org/llvm.sh
+          chmod +x llvm.sh
+          sudo ./llvm.sh 15
+          which clang++-15
+          clang++-15 --version
+          sudo apt-get install -y make cmake ccache ninja-build yasm gawk wget
+          ccache -s
       - name: Update git
         run: |
           sudo add-apt-repository ppa:git-core/ppa -y
@@ -72,49 +89,69 @@ jobs:
           key: ${{ matrix.os }}
           max-size: 5G
           append-timestamp: true
-      - name: Prepare chdb/build.sh
+      - name: remove old clang and link clang-15 to clang
+        if: matrix.os == 'ubuntu-20.04'
+        run: |
+          sudo rm /usr/bin/clang
+          sudo ln -s /usr/bin/clang-15 /usr/bin/clang
+          sudo rm /usr/bin/clang++
+          sudo ln -s /usr/bin/clang++-15 /usr/bin/clang++
+          which clang++
+          clang++ --version
+      - name: Run chdb/build.sh
         run: |
           pyenv local "${{ matrix.python-version }}"
           python3 --version
+          python3 -m pip install pybind11
           export CC=/usr/bin/clang
           export CXX=/usr/bin/clang++
-        shell: bash -l {0}
+          bash ./chdb/build.sh
+          python3 -m pip install pandas pyarrow
+          bash -x ./chdb/test_smoke.sh
         continue-on-error: false
+      - name: Check ccache statistics
+        run: |
+          ccache -s
+          ls -lh chdb
+          df -h
       - name: Install dependencies for building wheels
         run: |
-          python3 -m pip install -U pip tox pybind11 twine setuptools wheel
-          python3 -m pip install cibuildwheel==2.12.1
+          pyenv local "${{ matrix.python-version }}"
+          python3 -m pip install -U pip tox twine setuptools wheel
         shell: bash -l {0}
       - name: Build wheels
-        run: python3 -m cibuildwheel --output-dir wheelhouse
-        timeout-minutes: 300
-        env:
-          CIBW_DEBUG: 1
-          CIBW_BEFORE_BUILD_LINUX: "yum install -y wget && wget https://apt.llvm.org/llvm.sh; chmod +x llvm.sh; bash ./llvm.sh 15; which clang++-15; wget https://github.com/NixOS/patchelf/releases/download/0.18.0/patchelf-0.18.0-aarch64.tar.gz -O patchelf.tar.gz; tar -xvf patchelf.tar.gz; cp bin/patchelf /usr/bin/; chmod +x /usr/bin/patchelf; patchelf --version;"
-          CIBW_CONTAINER_ENGINE: podman
-          CIBW_ENVIRONMENT_LINUX: "CC=/usr/bin/clang CXX=/usr/bin/clang++"
-          CIBW_BEFORE_BUILD: "pip install -U pip tox pybind11 && bash -x gen_manifest.sh && bash chdb/build.sh"
-          CIBW_BUILD_VERBOSITY: 3
-          CIBW_BUILD: "cp38-manylinux_aarch64 cp39-manylinux_aarch64 cp310-manylinux_aarch64"
-          CIBW_TEST_REQUIRES: "pyarrow pandas psutil"
-          CIBW_TEST_COMMAND: "cd {project} && make test"
-      - name: Keep killall ccache and wait for ccache to finish
-        if: always()
         run: |
-          sleep 60
-          while ps -ef | grep ccache | grep -v grep; do \
-            killall ccache; \
-            sleep 10; \
-          done
+          pyenv local "${{ matrix.python-version }}"
+          export CC=/usr/bin/clang
+          export CXX=/usr/bin/clang++
+          make wheel
+      - name: Install patchelf from github
+        run: |
+          pyenv local "${{ matrix.python-version }}"
+          wget https://github.com/NixOS/patchelf/releases/download/0.18.0/patchelf-0.18.0-aarch64.tar.gz -O patchelf.tar.gz
+          tar -xvf patchelf.tar.gz
+          sudo cp bin/patchelf /usr/bin/
+          sudo chmod +x /usr/bin/patchelf
+          patchelf --version
+      - name: Audit wheels
+        run: |
+          pyenv local "${{ matrix.python-version }}"
+          python3 -m pip install auditwheel
+          auditwheel -v repair -w dist/ --plat manylinux_2_24_aarch64 dist/*.whl
+        continue-on-error: false
       - name: Show files
-        run: ls -lh wheelhouse
+        run: |
+          sudo rm -f dist/*-aarch64*.whl
+          ls -lh dist
         shell: bash
+        continue-on-error: false
       - uses: actions/upload-artifact@v3
         with:
-          path: ./wheelhouse/*.whl
+          path: ./dist/*.whl
       - name: Upload pypi
         if: startsWith(github.ref, 'refs/tags/v')
         run: |
+          pyenv local "${{ matrix.python-version }}"
           python3 -m pip install twine
           python3 -m twine upload wheelhouse/*.whl
         env: