Reorganize Rust benchmark file structure following Neo4jVSDoublets pattern #67
This file contains hidden or 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: Benchmark Rust version | |
| on: | |
| push: | |
| branches: | |
| - main | |
| - master | |
| pull_request: | |
| branches: | |
| - main | |
| - master | |
| env: | |
| toolchain: nightly-2022-08-22 | |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} | |
| jobs: | |
| benchmark: | |
| runs-on: ubuntu-latest | |
| services: | |
| postgres: | |
| image: postgres:latest | |
| env: | |
| POSTGRES_USER: postgres | |
| POSTGRES_PASSWORD: postgres | |
| POSTGRES_DB: postgres | |
| options: >- | |
| --health-cmd pg_isready | |
| --health-interval 10s | |
| --health-timeout 5s | |
| --health-retries 5 | |
| ports: | |
| - 5432:5432 | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| - name: Setup Rust | |
| uses: dtolnay/rust-toolchain@master | |
| with: | |
| toolchain: ${{env.toolchain}} | |
| components: rustfmt, clippy | |
| - name: Wait for PostgreSQL to be ready | |
| run: | | |
| echo "Waiting for PostgreSQL to be fully ready..." | |
| for i in {1..30}; do | |
| if pg_isready -h localhost -p 5432 -U postgres > /dev/null 2>&1; then | |
| echo "PostgreSQL is ready!" | |
| break | |
| fi | |
| echo "Attempt $i: PostgreSQL not ready yet..." | |
| sleep 2 | |
| done | |
| - name: Build benchmark | |
| run: cargo build --release --all-features --manifest-path rust/Cargo.toml | |
| - name: Run benchmark | |
| working-directory: rust | |
| env: | |
| POSTGRES_HOST: localhost | |
| POSTGRES_PORT: 5432 | |
| POSTGRES_USER: postgres | |
| POSTGRES_PASSWORD: postgres | |
| POSTGRES_DB: postgres | |
| # Use 100 links for main/master branch benchmarks, 10 for pull requests | |
| BENCHMARK_LINKS: ${{ (github.event_name == 'push' && (github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master')) && '100' || '10' }} | |
| # Use 1000 background links for main/master branch, 100 for pull requests | |
| BENCHMARK_BACKGROUND_LINKS: ${{ (github.event_name == 'push' && (github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master')) && '1000' || '100' }} | |
| run: | | |
| set -o pipefail | |
| cargo bench --bench bench -- --output-format bencher | tee out.txt | |
| - name: Prepare benchmark results | |
| if: github.event_name == 'push' && (github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master') | |
| run: | | |
| git config --global user.email "[email protected]" | |
| git config --global user.name "LinksPlatformBencher" | |
| cd rust | |
| pip install numpy matplotlib | |
| python3 out.py | |
| cd .. | |
| # Create Docs directory if it doesn't exist | |
| mkdir -p Docs | |
| # Copy generated images | |
| cp -f rust/bench_rust.png Docs/ | |
| cp -f rust/bench_rust_log_scale.png Docs/ | |
| # Update README with latest results | |
| if [ -f rust/results.md ]; then | |
| # Replace the results section in README.md | |
| python3 -c " | |
| import re | |
| with open('rust/results.md', 'r') as f: | |
| results = f.read() | |
| with open('README.md', 'r') as f: | |
| readme = f.read() | |
| # Pattern to find and replace the results table | |
| pattern = r'(\| Operation.*?\n\|[-|]+\n(?:\|.*?\n)*)' | |
| if re.search(pattern, readme): | |
| readme = re.sub(pattern, results.strip() + '\n', readme) | |
| with open('README.md', 'w') as f: | |
| f.write(readme) | |
| " | |
| fi | |
| # Commit changes if any | |
| git add Docs README.md | |
| if git diff --staged --quiet; then | |
| echo "No changes to commit" | |
| else | |
| git commit -m "Update benchmark results" | |
| git push origin HEAD | |
| fi | |
| - name: Save benchmark results | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: Benchmark results | |
| path: | | |
| rust/bench_rust.png | |
| rust/bench_rust_log_scale.png | |
| rust/out.txt |