Attempting to fix apache#14331

Author: Omega359

.github/workflows/extended.yml

@@ -33,16 +33,47 @@ on:
   push:
     branches:
       - main
+  issue_comment:
+    types: [ created ]
+
+permissions:
+  pull-requests: write
 
 jobs:
+  # Check issue comment and notify that extended tests are running
+  check_issue_comment:
+    name: Check issue comment
+    runs-on: ubuntu-latest
+    if: github.event.issue.pull_request && github.event.comment.body == 'run extended tests'
+    steps:
+      - uses: actions/github-script@v7
+        with:
+          github-token: ${{secrets.GITHUB_TOKEN}}
+          script: |
+            github.rest.issues.createComment({
+              issue_number: context.issue.number,
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              body: "Running extended tests..."
+            })
+
   # Check crate compiles and base cargo check passes
   linux-build-lib:
     name: linux build test
     runs-on: ubuntu-latest
     container:
       image: amd64/rust
+    if: |
+      github.event_name == 'push' ||
+      (github.event_name == 'issue_comment' && github.event.issue.pull_request && github.event.comment.body == 'run extended tests')
     steps:
       - uses: actions/checkout@v4
+        with:
+          # Check out the pull request if triggered by a comment
+          ref: ${{ github.event_name == 'issue_comment' && github.event.issue.pull_request.head.sha || github.ref }}
+          repository: ${{ github.event.pull_request.head.repo.full_name }}
+          submodules: true
+          fetch-depth: 1
       - name: Setup Rust toolchain
         uses: ./.github/actions/setup-builder
         with:
@@ -59,9 +90,15 @@ jobs:
     runs-on: ubuntu-latest
     container:
       image: amd64/rust
+    if: |
+      github.event_name == 'push' ||
+      (github.event_name == 'issue_comment' && github.event.issue.pull_request && github.event.comment.body == 'run extended tests')
     steps:
       - uses: actions/checkout@v4
         with:
+          # Check out the pull request if triggered by a comment
+          ref: ${{ github.event_name == 'issue_comment' && github.event.issue.pull_request.head.sha || github.ref }}
+          repository: ${{ github.event.pull_request.head.repo.full_name }}
           submodules: true
           fetch-depth: 1
       - name: Setup Rust toolchain
@@ -81,9 +118,15 @@ jobs:
     runs-on: ubuntu-latest
     container:
       image: amd64/rust
+    if: |
+      github.event_name == 'push' ||
+      (github.event_name == 'issue_comment' && github.event.issue.pull_request && github.event.comment.body == 'run extended tests')
     steps:
       - uses: actions/checkout@v4
         with:
+          # Check out the pull request if triggered by a comment
+          ref: ${{ github.event_name == 'issue_comment' && github.event.issue.pull_request.head.sha || github.ref }}
+          repository: ${{ github.event.pull_request.head.repo.full_name }}
           submodules: true
           fetch-depth: 1
       - name: Setup Rust toolchain
@@ -94,25 +137,72 @@ jobs:
         run: |
           cd datafusion
           cargo test  --profile ci --exclude datafusion-examples --exclude datafusion-benchmarks --exclude datafusion-sqllogictest --workspace --lib --tests --features=force_hash_collisions,avro,extended_tests
-          cargo clean
 
   sqllogictest-sqlite:
     name: "Run sqllogictests with the sqlite test suite"
     runs-on: ubuntu-latest
     container:
       image: amd64/rust
+    if: |
+      github.event_name == 'push' ||
+      (github.event_name == 'issue_comment' && github.event.issue.pull_request && github.event.comment.body == 'run extended tests')
     steps:
       - uses: actions/checkout@v4
         with:
+          # Check out the pull request if triggered by a comment
+          ref: ${{ github.event_name == 'issue_comment' && github.event.issue.pull_request.head.sha || github.ref }}
+          repository: ${{ github.event.pull_request.head.repo.full_name }}
           submodules: true
           fetch-depth: 1
       - name: Setup Rust toolchain
         uses: ./.github/actions/setup-builder
         with:
           rust-version: stable
       - name: Run sqllogictest
-        run: |
-          cargo test --profile release-nonlto --test sqllogictests -- --include-sqlite
-          cargo clean
+        run: cargo test --profile release-nonlto --test sqllogictests -- --include-sqlite
 
+  notify_if_run_on_pr_success:
+    name: Notify
+    runs-on: ubuntu-latest
+    needs:
+      [
+        linux-test-extended,
+        hash-collisions,
+        sqllogictest-sqlite,
+        check_issue_comment,
+      ]
+    if: success()
+    steps:
+      - uses: actions/github-script@v7
+        with:
+          github-token: ${{secrets.GITHUB_TOKEN}}
+          script: |
+            github.rest.issues.createComment({
+              issue_number: context.issue.number,
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              body: "extended test suite ran successfully on this PR."
+            })
 
+  notify_if_run_on_pr_failure:
+    name: Notify
+    runs-on: ubuntu-latest
+    needs:
+      [
+        linux-test-extended,
+        hash-collisions,
+        sqllogictest-sqlite,
+        check_issue_comment,
+      ]
+    if: failure()
+    steps:
+      - uses: actions/github-script@v7
+        with:
+          github-token: ${{secrets.GITHUB_TOKEN}}
+          script: |
+            github.rest.issues.createComment({
+              issue_number: context.issue.number,
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              body: "extended test suite failed on this PR."
+            })