Implement auto-closing of build failure issues after three consecutive successes.

PiperOrigin-RevId: 808018965

Author: abhinavclemson

.github/workflows/RunTests.yml

@@ -157,7 +157,73 @@ jobs:
       issues: write
     steps:
     - name: Check whether one of the jobs failed
+      id: report_failure
       if: ${{ contains(needs.*.result, 'failure') && github.event.pull_request == null && github.event_name != 'workflow_dispatch' }}
       uses: jayqi/failed-build-issue-action@1a893bbf43ef1c2a8705e2b115cd4f0fe3c5649b  # v1.2.0
       with:
         github-token: ${{ secrets.GITHUB_TOKEN }}
+        labels: "failed-build"
+
+    - name: Reset consecutive success counter on failure
+      # This runs only if the previous step actually found or created an issue
+      if: ${{ steps.report_failure.outputs.issue-number != '' }}
+      env:
+        GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        GH_REPO: ${{ github.repository }}
+        # Use the issue number from the previous step's output
+        ISSUE_NUMBER: ${{ steps.report_failure.outputs.issue-number }}
+      run: |
+        echo "A failure occurred. Resetting success counter on issue #${ISSUE_NUMBER}."
+        # This command will attempt to remove both labels. 
+        # It will not fail if the labels don't exist.
+        gh issue remove-label $ISSUE_NUMBER "success-run-1" "success-run-2" --repo $GH_REPO || echo "No success labels to remove."
+
+  notify_success_and_close:
+      name: Close issue after 3 successful builds
+      # This job runs only if all the preceding test jobs succeeded
+      if: ${{ success() && github.event.pull_request == null && github.event_name != 'workflow_dispatch' }}
+      needs: [cpu_unit_tests, gpu_unit_tests, gpu_integration_tests, tpu_unit_tests, tpu_integration_tests]
+      runs-on: ubuntu-latest
+      permissions:
+        issues: write
+      steps:
+        - name: Find existing failure issue
+          id: find_issue
+          env:
+            GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+            GH_REPO: ${{ github.repository }}
+          run: |
+            ISSUE_NUMBER=$(gh issue list --label "failed-build" --state open --limit 1 --json number -q '.[0].number')
+            if [[ -z "$ISSUE_NUMBER" ]]; then
+              echo "No open build failure issue found. Nothing to do."
+              echo "issue_number=" >> $GITHUB_OUTPUT
+            else
+              echo "Found open build failure issue: #${ISSUE_NUMBER}"
+              echo "issue_number=${ISSUE_NUMBER}" >> $GITHUB_OUTPUT
+            fi
+
+        - name: Add success label or close issue
+          if: steps.find_issue.outputs.issue_number != ''
+          env:
+            GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+            GH_REPO: ${{ github.repository }}
+          run: |
+            ISSUE_NUMBER=${{ steps.find_issue.outputs.issue_number }}
+            LABELS=$(gh issue view $ISSUE_NUMBER --json labels -q '.labels[].name')
+
+            if echo "$LABELS" | grep -q "success-run-2"; then
+              echo "Third consecutive success. Closing issue #${ISSUE_NUMBER}."
+              gh issue comment $ISSUE_NUMBER --body "Build succeeded for the third consecutive time. Closing this issue automatically."
+              gh issue close $ISSUE_NUMBER
+              # Clean up all tracking labels
+              gh issue remove-label $ISSUE_NUMBER "failed-build" "success-run-2" --repo $GH_REPO
+            elif echo "$LABELS" | grep -q "success-run-1"; then
+              echo "Second consecutive success. Updating label on issue #${ISSUE_NUMBER}."
+              gh issue comment $ISSUE_NUMBER --body "Build succeeded for the second time. One more successful run will close this issue."
+              gh issue remove-label $ISSUE_NUMBER "success-run-1" --repo $GH_REPO
+              gh issue add-label $ISSUE_NUMBER "success-run-2" --repo $GH_REPO
+            else
+              echo "First consecutive success since failure. Adding label to issue #${ISSUE_NUMBER}."
+              gh issue comment $ISSUE_NUMBER --body "Build succeeded. This issue will be auto-closed after two more consecutive successful runs."
+              gh issue add-label $ISSUE_NUMBER "success-run-1" --repo $GH_REPO
+            fi