Improve ShuffleReaderExec::new #134
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
| # Licensed to the Apache Software Foundation (ASF) under one | |
| # or more contributor license agreements. See the NOTICE file | |
| # distributed with this work for additional information | |
| # regarding copyright ownership. The ASF licenses this file | |
| # to you under the Apache License, Version 2.0 (the | |
| # "License"); you may not use this file except in compliance | |
| # with the License. You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, | |
| # software distributed under the License is distributed on an | |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
| # KIND, either express or implied. See the License for the | |
| # specific language governing permissions and limitations | |
| # under the License. | |
| name: Rust | |
| on: | |
| # always trigger | |
| push: | |
| pull_request: | |
| jobs: | |
| # build the library, a compilation step used by multiple steps below | |
| linux-build-lib: | |
| name: Build Libraries on AMD64 Rust ${{ matrix.rust }} | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| arch: [amd64] | |
| rust: [stable] | |
| container: | |
| image: ${{ matrix.arch }}/rust | |
| env: | |
| # Disable full debug symbol generation to speed up CI build and keep memory down | |
| # "1" means line tables only, which is useful for panic tracebacks. | |
| RUSTFLAGS: "-C debuginfo=1" | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - name: Cache Cargo | |
| uses: actions/cache@v3 | |
| with: | |
| # these represent dependencies downloaded by cargo | |
| # and thus do not depend on the OS, arch nor rust version. | |
| path: /github/home/.cargo | |
| key: cargo-cache- | |
| - name: Cache Rust dependencies | |
| uses: actions/cache@v3 | |
| with: | |
| # these represent compiled steps of both dependencies and arrow | |
| # and thus are specific for a particular OS, arch and rust version. | |
| path: /github/home/target | |
| key: ${{ runner.os }}-${{ matrix.arch }}-target-cache-${{ matrix.rust }}- | |
| - name: Setup Rust toolchain | |
| uses: ./.github/actions/setup-builder | |
| with: | |
| rust-version: ${{ matrix.rust }} | |
| - name: Build workspace in release mode | |
| run: | | |
| cargo build --release | |
| ls -l /github/home/target/release | |
| env: | |
| CARGO_HOME: "/github/home/.cargo" | |
| CARGO_TARGET_DIR: "/github/home/target" | |
| - name: Save artifacts | |
| uses: actions/upload-artifact@v2 | |
| with: | |
| name: rust-artifacts | |
| path: | | |
| /github/home/target/release/ballista-scheduler | |
| /github/home/target/release/ballista-executor | |
| react-build: | |
| name: React build | |
| runs-on: ubuntu-latest # proprietary github image, not ubuntu:latest | |
| container: | |
| image: ubuntu:latest # actual ubuntu:latest that ubuntu publishes | |
| env: | |
| DEBIAN_FRONTEND: "noninteractive" | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - name: Install dependencies | |
| run: | | |
| cat /etc/*-release | |
| apt-get -qq update | |
| apt-get -qq upgrade | |
| apt-get -qq install -y curl | |
| curl -fsSL https://deb.nodesource.com/setup_18.x | bash | |
| apt-get -qq update | |
| apt-get -qq install -y nodejs | |
| npm install -g yarn | |
| which node | |
| which npm | |
| which yarn | |
| node --version | |
| npm --version | |
| yarn --version | |
| - name: Run yarn build | |
| run: | | |
| cd ballista/scheduler/ui | |
| pwd | |
| yarn install | |
| yarn build | |
| - name: Save artifacts | |
| uses: actions/upload-artifact@v2 | |
| with: | |
| name: react-artifacts | |
| path: | | |
| ballista/scheduler/ui/build | |
| # test the crate | |
| linux-test: | |
| name: Test Workspace on AMD64 Rust ${{ matrix.rust }} | |
| needs: [linux-build-lib] | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| arch: [amd64] | |
| rust: [stable] | |
| container: | |
| image: ${{ matrix.arch }}/rust | |
| env: | |
| # Disable full debug symbol generation to speed up CI build and keep memory down | |
| # "1" means line tables only, which is useful for panic tracebacks. | |
| RUSTFLAGS: "-C debuginfo=1" | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| submodules: true | |
| - name: Install protobuf compiler | |
| shell: bash | |
| run: | | |
| apt-get -qq update && apt-get -y -qq install protobuf-compiler | |
| protoc --version | |
| - name: Cache Cargo | |
| uses: actions/cache@v3 | |
| with: | |
| path: /github/home/.cargo | |
| # this key equals the ones on `linux-build-lib` for re-use | |
| key: cargo-cache- | |
| - name: Cache Rust dependencies | |
| uses: actions/cache@v3 | |
| with: | |
| path: /github/home/target | |
| # this key equals the ones on `linux-build-lib` for re-use | |
| key: ${{ runner.os }}-${{ matrix.arch }}-target-cache-${{ matrix.rust }} | |
| - name: Setup Rust toolchain | |
| uses: ./.github/actions/setup-builder | |
| with: | |
| rust-version: ${{ matrix.rust }} | |
| - name: Run tests | |
| run: | | |
| export PATH=$PATH:$HOME/d/protoc/bin | |
| export ARROW_TEST_DATA=$(pwd)/testing/data | |
| export PARQUET_TEST_DATA=$(pwd)/parquet-testing/data | |
| cargo test | |
| env: | |
| CARGO_HOME: "/github/home/.cargo" | |
| CARGO_TARGET_DIR: "/github/home/target" | |
| # run ballista tests | |
| ballista-test: | |
| name: Test Ballista on AMD64 Rust ${{ matrix.rust }} | |
| needs: [linux-build-lib] | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| arch: [amd64] | |
| rust: [stable] | |
| container: | |
| image: ${{ matrix.arch }}/rust | |
| env: | |
| # Disable full debug symbol generation to speed up CI build and keep memory down | |
| # "1" means line tables only, which is useful for panic tracebacks. | |
| RUSTFLAGS: "-C debuginfo=1" | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| submodules: true | |
| - name: Install protobuf compiler | |
| shell: bash | |
| run: | | |
| mkdir -p $HOME/d/protoc | |
| cd $HOME/d/protoc | |
| export PROTO_ZIP="protoc-21.4-linux-x86_64.zip" | |
| curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v21.4/$PROTO_ZIP | |
| unzip $PROTO_ZIP | |
| export PATH=$PATH:$HOME/d/protoc/bin | |
| protoc --version | |
| - name: Cache Cargo | |
| uses: actions/cache@v3 | |
| with: | |
| path: /github/home/.cargo | |
| # this key equals the ones on `linux-build-lib` for re-use | |
| key: cargo-cache- | |
| - name: Cache Rust dependencies | |
| uses: actions/cache@v3 | |
| with: | |
| path: /github/home/target | |
| # this key equals the ones on `linux-build-lib` for re-use | |
| key: ${{ runner.os }}-${{ matrix.arch }}-target-cache-${{ matrix.rust }} | |
| - name: Setup Rust toolchain | |
| uses: ./.github/actions/setup-builder | |
| with: | |
| rust-version: ${{ matrix.rust }} | |
| # Ballista is currently not part of the main workspace so requires a separate test step | |
| - name: Run Ballista tests | |
| run: | | |
| export PATH=$PATH:$HOME/d/protoc/bin | |
| export ARROW_TEST_DATA=$(pwd)/testing/data | |
| export PARQUET_TEST_DATA=$(pwd)/parquet-testing/data | |
| cd ballista | |
| # snmalloc requires cmake so build without default features | |
| cargo test | |
| env: | |
| CARGO_HOME: "/github/home/.cargo" | |
| CARGO_TARGET_DIR: "/github/home/target" | |
| windows-and-macos: | |
| name: Test on ${{ matrix.os }} Rust ${{ matrix.rust }} | |
| runs-on: ${{ matrix.os }} | |
| strategy: | |
| matrix: | |
| os: [windows-latest, macos-latest] | |
| rust: [stable] | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| submodules: true | |
| - name: Install protobuf macos compiler | |
| shell: bash | |
| run: | | |
| mkdir -p $HOME/d/protoc | |
| cd $HOME/d/protoc | |
| export PROTO_ZIP="protoc-21.4-osx-x86_64.zip" | |
| curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v21.4/$PROTO_ZIP | |
| unzip $PROTO_ZIP | |
| echo "$HOME/d/protoc/bin" >> $GITHUB_PATH | |
| export PATH=$PATH:$HOME/d/protoc/bin | |
| protoc --version | |
| if: ${{matrix.os == 'macos-latest'}} | |
| - name: Install protobuf windows compiler | |
| shell: bash | |
| run: | | |
| mkdir -p $HOME/d/protoc | |
| cd $HOME/d/protoc | |
| export PROTO_ZIP="protoc-21.4-win64.zip" | |
| curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v21.4/$PROTO_ZIP | |
| unzip $PROTO_ZIP | |
| export PATH=$PATH:$HOME/d/protoc/bin | |
| protoc.exe --version | |
| if: ${{matrix.os == 'windows-latest'}} | |
| # TODO: this won't cache anything, which is expensive. Setup this action | |
| # with a OS-dependent path. | |
| - name: Setup Rust toolchain | |
| run: | | |
| rustup toolchain install ${{ matrix.rust }} | |
| rustup default ${{ matrix.rust }} | |
| rustup component add rustfmt | |
| - name: Run tests | |
| shell: bash | |
| run: | | |
| export PATH=$PATH:$HOME/d/protoc/bin | |
| export ARROW_TEST_DATA=$(pwd)/testing/data | |
| export PARQUET_TEST_DATA=$(pwd)/parquet-testing/data | |
| cargo test | |
| env: | |
| # do not produce debug symbols to keep memory usage down | |
| RUSTFLAGS: "-C debuginfo=0" | |
| lint: | |
| name: Lint | |
| runs-on: ubuntu-latest | |
| container: | |
| image: amd64/rust | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - name: Setup toolchain | |
| run: | | |
| rustup toolchain install stable | |
| rustup default stable | |
| rustup component add rustfmt | |
| - name: Run | |
| run: cargo fmt --all -- --check | |
| env: | |
| CARGO_HOME: "/github/home/.cargo" | |
| CARGO_TARGET_DIR: "/github/home/target" | |
| clippy: | |
| name: Clippy | |
| needs: [linux-build-lib] | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| arch: [amd64] | |
| rust: [stable] | |
| container: | |
| image: ${{ matrix.arch }}/rust | |
| env: | |
| # Disable full debug symbol generation to speed up CI build and keep memory down | |
| # "1" means line tables only, which is useful for panic tracebacks. | |
| RUSTFLAGS: "-C debuginfo=1" | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| submodules: true | |
| - name: Cache Cargo | |
| uses: actions/cache@v3 | |
| with: | |
| path: /github/home/.cargo | |
| # this key equals the ones on `linux-build-lib` for re-use | |
| key: cargo-cache- | |
| - name: Cache Rust dependencies | |
| uses: actions/cache@v3 | |
| with: | |
| path: /github/home/target | |
| # this key equals the ones on `linux-build-lib` for re-use | |
| key: ${{ runner.os }}-${{ matrix.arch }}-target-cache-${{ matrix.rust }} | |
| - name: Setup Rust toolchain | |
| uses: ./.github/actions/setup-builder | |
| with: | |
| rust-version: ${{ matrix.rust }} | |
| - name: Install Clippy | |
| run: | | |
| rustup component add clippy | |
| - name: Run clippy | |
| run: cargo clippy --all-targets --workspace -- -D warnings | |
| env: | |
| CARGO_HOME: "/github/home/.cargo" | |
| CARGO_TARGET_DIR: "/github/home/target" |