add k6 load test example

Author: jthrilly

.env.example

@@ -1,10 +1,22 @@
-SANDBOX_MODE=false
-UPLOADTHING_SECRET=sk_live_xxxxxx
-UPLOADTHING_APP_ID=xxxxxxx
-
-POSTGRES_USER="postgres"
-POSTGRES_PASSWORD="postgres"
-POSTGRES_DATABASE="postgres"
-POSTGRES_HOST="postgres"
-POSTGRES_PRISMA_URL="postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:5432/${POSTGRES_DATABASE}?schema=public"
-POSTGRES_URL_NON_POOLING="postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:5432/${POSTGRES_DATABASE}?schema=public"
+# -------------------
+# Optional environment variables - uncomment to use
+# -------------------
+
+#SANDBOX_MODE=false # true or false - if true, the app will use the sandbox mode, which disables resetting the database and other features
+#NEXT_PUBLIC_URL="http://yourdomain.com" # When using advanced deployment, this is required. Set to the domain name of your app
+#NEXT_PUBLIC_DISABLE_ANALYTICS=true # true or false - if true, the app will not send anonymous analytics data to the server
+#INSTALLATION_ID="your-app-name" # A unique identifier for your app, used for analytics. Generated automatically if not set.
+
+# -------------------
+# Required environment variables
+# -------------------
+
+UPLOADTHING_SECRET=sk_live_xxxxxx # Your UploadThing secret key
+UPLOADTHING_APP_ID=xxxxxxx # Your UploadThing app ID
+
+POSTGRES_USER="postgres" # Your PostgreSQL username
+POSTGRES_PASSWORD="postgres"  # Your PostgreSQL password
+POSTGRES_DATABASE="postgres"  # Your PostgreSQL database name
+POSTGRES_HOST="postgres"  # Your PostgreSQL host
+POSTGRES_PRISMA_URL="postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:5432/${POSTGRES_DATABASE}?schema=public" # A pooled connection URL for Prisma. 
+POSTGRES_URL_NON_POOLING="postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:5432/${POSTGRES_DATABASE}?schema=public" # A non-pooling connection URL for Prisma

load-test.js

@@ -0,0 +1,52 @@
+import { check, group, sleep } from 'k6';
+import http, { CookieJar } from 'k6/http';
+
+const HOST = 'host.docker.internal:3000';
+
+export let options = {
+  stages: [
+    { duration: '30s', target: 100 }, // Ramp-up to 150 users over 30 seconds.
+    { duration: '1m', target: 100 },  // Stay at 150 users for 1 minute.
+    { duration: '30s', target: 0 },  // Ramp down to 0 users over 10 seconds.
+  ],
+};
+
+export default function () {
+  let loginRes = login();
+  check(loginRes, {
+    'login successful': (r) => r.status === 200,
+  });
+
+  if (loginRes.status === 200) {
+
+    let jar = new CookieJar();
+    jar.set(`http://${HOST}`, 'auth_session', loginRes.cookies['auth_session'][0].value);
+
+    group('Authenticated requests', function () {
+      let res = http.get(`http://${HOST}/dashboard`, { cookies: jar.cookiesForURL(`http://${HOST}`) });
+      check(res, {
+        'is status 200': (r) => r.status === 200,
+      });
+
+      let res2 = http.get(`http://${HOST}/dashboard/settings`, { cookies: jar.cookiesForURL(`http://${HOST}`) });
+      check(res2, {
+        'is status 200': (r) => r.status === 200,
+      });
+    });
+  }
+
+  sleep(1);
+}
+
+function login() {
+  let payload = '[{"username":"admin","password":"Administrator1!"}]'
+
+  let params = {
+    headers: {
+      'Next-Action': '36b840346a44c93083679ff0264912dba5c1443f', // No way to get this automatically, I don't think...
+      'Content-Type': 'text/plain;charset=UTF-8',
+    },
+  };
+
+  return http.post(`http://${HOST}/signin`, payload, params);
+}

middleware.ts

@@ -14,7 +14,8 @@
     "start": "next start",
     "vercel-build": "node ./handle-migrations.js && next build",
     "knip": "knip",
-    "test": "vitest"
+    "test": "vitest",
+    "load-test": "docker run -i grafana/k6 run - <load-test.js"
   },
   "pnpm": {
     "overrides": {
@@ -135,6 +136,7 @@
     "eslint-config-prettier": "^9.1.0",
     "jest": "^29.7.0",
     "jsdom": "^24.0.0",
+    "k6": "^0.0.0",
     "knip": "^5.13.0",
     "postcss": "^8.4.38",
     "prettier": "^3.2.5",

middlewares/stackMiddleware.ts

@@ -1,10 +1,6 @@
 import { env } from '~/env.mjs';
 
 export default function getBaseUrl() {
-  if (typeof window !== 'undefined')
-    // browser should use relative path
-    return `${window.location.protocol}//${window.location.host}`;
-
   if (env.VERCEL_URL)
     // reference for vercel.com
     return `https://${env.VERCEL_URL}`;
@@ -14,5 +10,6 @@ export default function getBaseUrl() {
     return env.NEXT_PUBLIC_URL;
 
   // assume localhost
+  console.log('caught');
   return `http://127.0.0.1:3000`;
 }