add k6 test scripts

docker/docker-compose.yml

@@ -4,6 +4,41 @@ networks:
                 driver: bridge
                 
 services:
+        influxdb:
+                image: influxdb:1.8
+                networks:
+                        - k6
+                        - grafana
+                ports:
+                        - "8086:8086"
+                environment:
+                        - INFLUXDB_DB=k6
+
+        grafana:
+                image: grafana/grafana:9.3.8
+                networks:
+                        - grafana
+                ports:
+                        - "3000:3000"
+                environment:
+                        - GF_AUTH_ANONYMOUS_ORG_ROLE=Admin
+                        - GF_AUTH_ANONYMOUS_ENABLED=true
+                        - GF_AUTH_BASIC_ENABLED=false
+                volumes:
+                        - ./grafana:/etc/grafana/provisioning/
+
+
+        k6:
+                image: grafana/k6:latest
+                networks:
+                        - k6
+                ports:
+                        - "6565:6565"
+                environment:
+                        - K6_OUT=influxdb=http://influxdb:8086/k6
+                volumes:
+                        - ./scripts:/scripts
+
         mysql:
                 container_name: "ai-mysql"
                 ports :

docker/scripts/script.js

+import http from 'k6/http';
+import { check, fail } from 'k6';
+
+export const options = {
+  vus: 5,
+  duration: '10s',
+};
+
+export default function() {
+  const url = 'http://43.198.252.255:8000/api/v1//txt2img/base/llm';
+  const payload = JSON.stringify({
+		      "test": 1
+		      })
+  const param = {
+    headers:{
+		'apikey': 'L6eXzteHYa2Uq0ar4kdMaBOiOIjmRI08',
+		'Content-Type': 'application/json'
+	},
+  };
+  var res = http.post(url, payload, param);
+  const checkOutput = check(
+    res,
+    {
+        'response code was 200': (res) => res.status == 200,
+        'task is succeed': (res) => res.json("task.is_success") == true,
+    },
+    { myTag: "task success check" }
+  );
+
+  if (!checkOutput) {
+    fail('task failed with error: ' + res.json('task.error'));
+  }
+}