Adds k6 loadtesting script for ws server

examples/http_backend/main.go

@@ -38,7 +38,7 @@ func main() {
 
 	dispatcher := dispatch.NewPipeDispatcher(backend)
 	dispatcher.Use(ErrHandler)
-	dispatcher.Use(request.NewTrottlerMiddleware(10))
+	dispatcher.Use(request.NewTrottlerMiddleware(100))
 
 	channel := channel.NewDefaultChannel("/", dispatcher, channel.WithOriginPatterns("*"))
 

loadtesting/Dockerfile

@@ -0,0 +1,3 @@
+FROM nginx:alpine
+COPY hello.json /usr/share/nginx/html/hello.json
+COPY default.conf /etc/nginx/conf.d/default.conf

loadtesting/default.conf

@@ -0,0 +1,7 @@
+server {
+    listen 80;
+    location / {
+        root /usr/share/nginx/html;
+        index hello.json;
+    }
+}

loadtesting/hello.json

@@ -0,0 +1 @@
+{"result": "success", "message": "The server is up and running."}

loadtesting/k6.js

@@ -0,0 +1,42 @@
+import ws from 'k6/ws';
+import { check } from 'k6';
+import { Counter } from 'k6/metrics';
+
+// A counter for the number of messages sent
+const counterSent = new Counter('ws_messages_sent');
+const counterRecived = new Counter('ws_messages_recived');
+
+const expectedResponse = '{"result": "success", "message": "The server is up and running."}';
+
+export default function () {
+  const url = 'ws://localhost:8080/';
+  const params = { tags: { my_tag: 'hello' } };
+
+  let counter = 100;
+
+  const res = ws.connect(url, params, function (socket) {
+    socket.on('open', () => {
+      console.log('connected');
+      // Send as many messages as possible
+
+      const iterations = counter;
+      for (let i = 0; i < iterations; i++) {
+        socket.send('Hello, server!');
+
+        counterSent.add(1);
+      }
+    });
+
+    socket.on('message', (data) => {
+      check(data, { 'Got Expected message': (d) => d === expectedResponse });
+      counterRecived.add(1);
+      counter--; 
+      if (counter == 0) {
+        socket.close();
+      }
+    });
+    socket.on('close', () => console.log('disconnected'));
+  });
+
+  check(res, { 'status is 101': (r) => r && r.status === 101 });
+}