test: loadtest with varying reqs and batch_size

.github/workflows/main.yml

@@ -55,6 +55,9 @@ jobs:
 
   loadtest:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        params: [ {reqs: 10000, batch: 200}, {reqs: 20000, batch: 400}, {reqs: 40000, batch: 800} ]
     steps:
       - uses: actions/checkout@v4
 
@@ -72,7 +75,8 @@ jobs:
 
       - name: Run load test
         run: |
-          nix-shell --run "net-loadtest"
+          echo "${{ matrix.params.reqs }} requests with a batch size of ${{ matrix.params.batch }}" >> "$GITHUB_STEP_SUMMARY"
+          nix-shell --run "net-loadtest ${{ matrix.params.reqs }} ${{ matrix.params.batch }}"
           cat psrecord.md >> "$GITHUB_STEP_SUMMARY"
 
   coverage:

nix/loadtest.nix

@@ -40,7 +40,18 @@ in
 writeShellScriptBin "net-loadtest" ''
   set -euo pipefail
 
-  net-with-nginx xpg psql -c "call wait_for_many_gets()" > /dev/null &
+  reqs=""
+  batch_size="200"
+
+  if [ -n "''${1:-}" ]; then
+    reqs="$1"
+  fi
+
+  if [ -n "''${2:-}" ]; then
+    batch_size="$2"
+  fi
+
+  net-with-nginx xpg --options "-c log_min_messages=WARNING -c pg_net.batch_size=$batch_size" psql -c "call wait_for_many_gets($reqs)" > /dev/null &
 
   # wait for process to start so we can capture it with psrecord
   sleep 2

nix/xpg.nix

@@ -3,8 +3,8 @@ let
   dep = fetchFromGitHub {
     owner  = "steve-chavez";
     repo   = "xpg";
-    rev    = "v1.4.1";
-    sha256 = "sha256-OI9g78KbguLh+ynOnRmnMM4lVOgNRAWkiI/YMmcMs+k=";
+    rev    = "v1.6.0";
+    sha256 = "sha256-NsdAmsYIRH/DWIZp93AHGYdPiJOztUIUSYcPikeebvw=";
   };
   xpg = import dep;
 in

test/utils/loadtest.sql

@@ -24,7 +24,7 @@ begin
 
   commit;
 
-  raise notice 'Waiting until % requests complete', number_of_requests;
+  raise notice 'Waiting until % requests complete, using a pg_net.batch_size of %', number_of_requests, current_setting('pg_net.batch_size')::text;
 
   perform net._await_response(last_id);