Use WithContext in all DB calls (#5538) #3962
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Build & Push Flyte Single Binary Images | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number }} | |
| cancel-in-progress: true | |
| on: | |
| pull_request: | |
| push: | |
| branches: | |
| - master | |
| - rc/* | |
| workflow_dispatch: | |
| jobs: | |
| test-bootstrap: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-go@v4 | |
| with: | |
| go-version: "1.21" | |
| - name: golangci-lint | |
| uses: golangci/golangci-lint-action@v3 | |
| with: | |
| working-directory: docker/sandbox-bundled/bootstrap | |
| - name: Check formatting | |
| working-directory: docker/sandbox-bundled/bootstrap | |
| run: | | |
| make check-fmt | |
| - name: Test | |
| working-directory: docker/sandbox-bundled/bootstrap | |
| run: | | |
| make test | |
| build-and-push-single-binary-image: | |
| runs-on: ubuntu-latest | |
| needs: [test-bootstrap] | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - name: Setup Golang caches | |
| uses: actions/cache@v3 | |
| with: | |
| path: | | |
| /root/.cache/go-build | |
| /root/go/pkg/mod | |
| key: ${{ runner.os }}-golang-${{ hashFiles('go.sum') }} | |
| restore-keys: | | |
| ${{ runner.os }}-golang- | |
| - name: Set versions | |
| id: set_version | |
| run: | | |
| # TODO: Find a better way to determine the flyteconsole version | |
| echo "FLYTECONSOLE_VERSION=$(yq eval '.flyteconsole.image.tag' charts/flyte-core/values.yaml)" >> $GITHUB_ENV | |
| echo "FLYTE_VERSION=${{ github.sha }}" >> $GITHUB_ENV | |
| - name: Prepare Image Names | |
| id: image-names | |
| uses: docker/metadata-action@v3 | |
| with: | |
| # list of Docker images to use as base name for tags | |
| images: | | |
| ghcr.io/${{ github.repository_owner }}/flyte-binary | |
| tags: | | |
| type=raw,value=latest,enable=${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }} | |
| type=sha,format=long | |
| - name: Set up QEMU | |
| uses: docker/setup-qemu-action@v2 | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| - name: Setup destination directories for image tarballs | |
| run: | | |
| mkdir -p docker/sandbox-bundled/images/tar/{arm64,amd64} | |
| - name: Export ARM64 Image | |
| uses: docker/build-push-action@v3 | |
| with: | |
| context: . | |
| platforms: linux/arm64 | |
| tags: flyte-binary:sandbox | |
| build-args: | | |
| FLYTECONSOLE_VERSION=${{ env.FLYTECONSOLE_VERSION }} | |
| FLYTE_VERSION=${{ env.FLYTE_VERSION }} | |
| file: Dockerfile | |
| outputs: type=docker,dest=docker/sandbox-bundled/images/tar/arm64/flyte-binary.tar | |
| - name: Export AMD64 Image | |
| uses: docker/build-push-action@v3 | |
| with: | |
| context: . | |
| platforms: linux/amd64 | |
| tags: flyte-binary:sandbox | |
| build-args: | | |
| FLYTECONSOLE_VERSION=${{ env.FLYTECONSOLE_VERSION }} | |
| FLYTE_VERSION=${{ env.FLYTE_VERSION }} | |
| file: Dockerfile | |
| outputs: type=docker,dest=docker/sandbox-bundled/images/tar/amd64/flyte-binary.tar | |
| - name: Upload single binary image | |
| uses: actions/upload-artifact@v2 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Login to GitHub Container Registry | |
| if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }} | |
| uses: docker/login-action@v2 | |
| with: | |
| registry: ghcr.io | |
| username: "${{ secrets.FLYTE_BOT_USERNAME }}" | |
| password: "${{ secrets.FLYTE_BOT_PAT }}" | |
| - name: Build and push Image | |
| if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }} | |
| uses: docker/build-push-action@v3 | |
| with: | |
| context: . | |
| platforms: linux/arm64, linux/amd64 | |
| tags: ${{ steps.image-names.outputs.tags }} | |
| build-args: | | |
| FLYTECONSOLE_VERSION=${{ env.FLYTECONSOLE_VERSION }} | |
| FLYTE_VERSION=${{ env.FLYTE_VERSION }} | |
| file: Dockerfile | |
| push: true | |
| sandbox-bundled-functional-tests: | |
| runs-on: ubuntu-latest | |
| env: | |
| FLYTESNACKS_PRIORITIES: "P0" | |
| FLYTESNACKS_VERSION: "" | |
| timeout-minutes: 60 | |
| needs: [build-and-push-single-binary-image] | |
| steps: | |
| - name: Set latest Flytesnacks release | |
| if: ${{ env.FLYTESNACKS_VERSION == '' }} | |
| run: | | |
| FLYTESNACKS_VERSION="$(curl --silent https://api.github.com/repos/flyteorg/flytesnacks/releases/latest | jq -r .tag_name)" | |
| echo "FLYTESNACKS_VERSION=${FLYTESNACKS_VERSION}" >> ${GITHUB_ENV} | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - uses: actions/download-artifact@v3 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| with: | |
| driver-opts: image=moby/buildkit:master | |
| buildkitd-flags: "--allow-insecure-entitlement security.insecure" | |
| - name: Build sandbox image for functional tests | |
| uses: docker/build-push-action@v3 | |
| with: | |
| context: docker/sandbox-bundled | |
| load: true | |
| allow: "security.insecure" | |
| tags: flyte-sandbox-bundled:local | |
| # Without this, the GHA runner is under disk pressure and evicts all the pods. | |
| # Buildx cache uses roughly 50% (7gb) of the GHA runner's disk (14gb). | |
| - name: Prune Docker Buildx cache to reclaim storage | |
| run: docker buildx prune --all --force | |
| - name: Set up Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: "3.12" | |
| - uses: unionai/[email protected] | |
| - name: Setup sandbox | |
| run: | | |
| mkdir -p ~/.flyte/sandbox | |
| cat << EOF > ~/.flyte/sandbox/config.yaml | |
| task_resources: | |
| defaults: | |
| cpu: "0" | |
| memory: "0" | |
| limits: | |
| cpu: "0" | |
| memory: "0" | |
| EOF | |
| flytectl demo start --image flyte-sandbox-bundled:local --imagePullPolicy Never | |
| - name: Install Python dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install uv | |
| uv pip install --system flytekit flytekitplugins-deck-standard flytekitplugins-envd "numpy<2.0.0" | |
| uv pip freeze | |
| - name: Checkout flytesnacks | |
| uses: actions/checkout@v4 | |
| with: | |
| repository: flyteorg/flytesnacks | |
| path: flytesnacks | |
| ref: ${{ env.FLYTESNACKS_VERSION }} | |
| - name: Register specific tests | |
| run: | | |
| flytekit_version=$(pip show flytekit | grep -i version | awk '{ print $2 }') | |
| while read -r line; | |
| do | |
| pyflyte --config ./boilerplate/flyte/end2end/functional-test-config.yaml \ | |
| register \ | |
| --project flytesnacks \ | |
| --domain development \ | |
| --image cr.flyte.org/flyteorg/flytekit:py3.12-${flytekit_version} \ | |
| --version ${{ env.FLYTESNACKS_VERSION }} \ | |
| flytesnacks/$line; | |
| done < flytesnacks/flyte_tests.txt | |
| - name: End2End | |
| run: | | |
| make end2end_execute | |
| build-and-push-sandbox-bundled-image: | |
| if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }} | |
| runs-on: ubuntu-latest | |
| needs: [build-and-push-single-binary-image] | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - uses: actions/download-artifact@v3 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Set up QEMU | |
| uses: docker/setup-qemu-action@v2 | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| with: | |
| driver-opts: image=moby/buildkit:master | |
| buildkitd-flags: "--allow-insecure-entitlement security.insecure" | |
| - name: Set version | |
| id: set_version | |
| run: | | |
| echo "FLYTE_SANDBOX_VERSION=${{ github.sha }}" >> $GITHUB_ENV | |
| - name: Prepare Image Names | |
| id: image-names | |
| uses: docker/metadata-action@v3 | |
| with: | |
| # list of Docker images to use as base name for tags | |
| images: | | |
| ghcr.io/${{ github.repository_owner }}/flyte-sandbox-bundled | |
| tags: | | |
| type=raw,value=latest,enable=${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }} | |
| type=sha,format=long, | |
| - name: Login to GitHub Container Registry | |
| uses: docker/login-action@v2 | |
| with: | |
| registry: ghcr.io | |
| username: "${{ secrets.FLYTE_BOT_USERNAME }}" | |
| password: "${{ secrets.FLYTE_BOT_PAT }}" | |
| - name: Build and push multi-arch image | |
| uses: docker/build-push-action@v3 | |
| with: | |
| context: docker/sandbox-bundled | |
| allow: "security.insecure" | |
| platforms: linux/arm64, linux/amd64 | |
| tags: ${{ steps.image-names.outputs.tags }} | |
| build-args: "FLYTE_SANDBOX_VERSION=${{ env.FLYTE_SANDBOX_VERSION }}" | |
| push: true |