PYTHON-5513 Use the perfcomp tool on pull requests

Author: blink1073

.evergreen/generated_configs/functions.yml

@@ -85,6 +85,22 @@ functions:
       params:
         directory: src
 
+  # Perf comment
+  create perf comment:
+    - command: subprocess.exec
+      params:
+        binary: bash
+        args:
+          - .evergreen/scripts/perf-pr-comment.sh
+        working_dir: src
+        include_expansions_in_env:
+          - version_id
+          - revision
+          - github_commit
+          - project
+          - TASK_NAME
+      type: test
+
   # Run server
   run server:
     - command: subprocess.exec

.evergreen/generated_configs/tasks.yml

@@ -2334,7 +2334,10 @@ tasks:
           SUB_TEST_NAME: sync
       - func: attach benchmark test results
       - func: send dashboard data
-    tags: [perf]
+      - func: create perf comment
+        vars:
+          TASK_NAME: perf-8.0-standalone-ssl
+    tags: [perf, pr]
   - name: perf-8.0-standalone-ssl-async
     commands:
       - func: run server

.evergreen/scripts/configure-env.sh

@@ -76,6 +76,8 @@ EOT
 rm -rf $DRIVERS_TOOLS
 BRANCH=master
 ORG=mongodb-labs
+BRANCH=PYTHON-5513
+ORG=blink1073
 git clone --branch $BRANCH https://github.com/$ORG/drivers-evergreen-tools.git $DRIVERS_TOOLS
 
 cat <<EOT > ${DRIVERS_TOOLS}/.env

.evergreen/scripts/generate_config.py

@@ -934,6 +934,11 @@ def create_perf_tasks():
             task_name += "-async"
         tags = ["perf"]
         commands = [server_func, test_func, attach_func, send_func]
+        if ssl == "ssl" and sync == "sync":
+            tags.append("pr")
+            comment_vars = dict(TASK_NAME=task_name)
+            comment_func = FunctionCall(func="create perf comment", vars=comment_vars)
+            commands.append(comment_func)
         tasks.append(EvgTask(name=task_name, tags=tags, commands=commands))
     return tasks
 
@@ -1225,6 +1230,22 @@ def create_send_dashboard_data_func():
     return "send dashboard data", cmds
 
 
+def create_perf_comment_func():
+    includes = [
+        "version_id",
+        "revision",
+        "github_commit",
+        "project",
+        "TASK_NAME",
+    ]
+    cmds = [
+        get_subprocess_exec(
+            include_expansions_in_env=includes, args=[".evergreen/scripts/perf-pr-comment.sh"]
+        )
+    ]
+    return "create perf comment", cmds
+
+
 mod = sys.modules[__name__]
 write_variants_to_file(mod)
 write_tasks_to_file(mod)

.evergreen/scripts/perf-pr-comment.sh

@@ -0,0 +1,30 @@
+#!/bin/bash
+# Generate a PR comment with the perf report
+
+set -eu
+
+VERSION_ID=${version_id}
+BASE_SHA=${revision}
+HEAD_SHA=${github_commit}
+PROJECT=${project}
+
+if [ "${PROJECT}" != "mongo-python-driver" ]; then
+  echo "Skipping PR Perf comment"
+  exit 0
+fi
+
+
+pushd $DRIVERS_TOOLS/.evergreen >/dev/null
+CONTEXT="PyMongo triage context" TASK="${TASK_NAME}" VARIANT="performance-benchmarks" sh run-perf-comp.sh
+
+if [[ -n "${BASE_SHA+set}" && -n "${HEAD_SHA+set}" && "$BASE_SHA" != "$HEAD_SHA" ]]; then
+    # Make the PR comment.
+    target=github_app/create_or_modify_comment.sh
+    bash $target -m "## 🧪 Performance Results" -c "$(pwd)/perfcomp/perf-report.md" -h $HEAD_SHA -o "mongodb" -n $PROJECT
+    rm ./perfcomp/perf-report.md
+else
+    # Skip comment if it isn't a PR run.
+    echo "Skipping Perf PR comment"
+fi
+
+popd >/dev/null