WIP Simplify example script

grafana · Jul 7, 2023 · b0aa982 · b0aa982
1 parent 554d517
commit b0aa982
Showing 1 changed file with 111 additions and 84 deletions.
diff --git a/examples/loadtest.js b/examples/loadtest.js
@@ -1,132 +1,159 @@
 import { check } from "k6";
-import http from "k6/http";
+// import http from "k6/http";
 import redis from "k6/x/redis";
-import exec from "k6/execution";
-import { textSummary } from "https://jslib.k6.io/k6-summary/0.0.1/index.js";
+// import exec from "k6/execution";
+// import { textSummary } from "https://jslib.k6.io/k6-summary/0.0.1/index.js";
 
 export const options = {
-  scenarios: {
-    redisPerformance: {
-      executor: "shared-iterations",
-      vus: 10,
-      iterations: 200,
-      exec: "measureRedisPerformance",
-    },
-    usingRedisData: {
-      executor: "shared-iterations",
-      vus: 10,
-      iterations: 200,
-      exec: "measureUsingRedisData",
-    },
-  },
+  insecureSkipTLSVerify: true,
+  // scenarios: {
+  //   redisPerformance: {
+  //     executor: "shared-iterations",
+  //     vus: 10,
+  //     iterations: 200,
+  //     exec: "measureRedisPerformance",
+  //   },
+  //   usingRedisData: {
+  //     executor: "shared-iterations",
+  //     vus: 10,
+  //     iterations: 200,
+  //     exec: "measureUsingRedisData",
+  //   },
+  // },
 };
 
 // Instantiate a new redis client with a URL
-const redisClient = new redis.Client(`redis://:${__ENV.REDIS_PASSWORD}@localhost:6379`);
+// const client = new redis.Client('redis://:tjkbZ8jrwz3pGiku@localhost:6379');
 
 // A single-node client configured with an object
-// const redisClient = new redis.Client({
+// const client = new redis.Client({
 //   socket: {
 //     host: 'localhost',
 //     port: 6379,
+//     // tls: {
+//     //   ca: [open('ca.crt')],
+//     // }
 //   },
-//   password: __ENV.REDIS_PASSWORD,
+//   password: __ENV.REDIS_PASSWORD || "",
 // });
 
+// A single-node client configured with an object and mTLS
+const client = new redis.Client({
+  socket: {
+    host: 'localhost',
+    port: 6379,
+    tls: {
+      ca: [open('ca.crt')],
+      cert: open('client.crt'),
+      key: open('client.key'),
+    }
+  },
+  password: __ENV.REDIS_PASSWORD,
+});
+
 // A cluster client configured with node URLs
 // const client = new redis.Client({
 //   cluster: {
-//     nodes: [
-//       `redis://:${__ENV.REDIS_PASSWORD}@host1:6379',
-//       `redis://:${__ENV.REDIS_PASSWORD}@host2:6379',
-//     ]
+//     nodes: ['redis://host1:6379', 'redis://host2:6379']
 //   }
 // });
 
 // A cluster client configured with node objects
-// const redisClient = new redis.Client({
+// const client = new redis.Client({
 //   cluster: {
 //     nodes: [
 //       {
 //         socket: {
 //           host: 'host1',
 //           port: 6379,
 //         },
-//         password: __ENV.REDIS_PASSWORD,
+//         password: __ENV.REDIS_PASSWORD || "",
 //       },
 //       {
 //         socket: {
 //           host: 'host2',
 //           port: 6379,
 //         },
-//         password: __ENV.REDIS_PASSWORD,
+//         password: __ENV.REDIS_PASSWORD || "",
 //       }
 //     ]
 //   }
 // });
 
+export default function () {
+  client.set("a", 1).then((val) => {
+    console.log(val);
+  })
+}
+
+export function teardown() {
+  client.get("a").then((val) => {
+    console.log(val);
+  })
+}
+
 // Prepare an array of crocodile ids for later use
 // in the context of the measureUsingRedisData function.
 const crocodileIDs = new Array(0, 1, 2, 3, 4, 5, 6, 7, 8, 9);
 
-export function measureRedisPerformance() {
-  // VUs are executed in a parallel fashion,
-  // thus, to ensure that parallel VUs are not
-  // modifying the same key at the same time,
-  // we use keys indexed by the VU id.
-  const key = `foo-${exec.vu.idInTest}`;
-
-  redisClient
-    .set(`foo-${exec.vu.idInTest}`, 1)
-    .then(() => redisClient.get(`foo-${exec.vu.idInTest}`))
-    .then((value) => redisClient.incrBy(`foo-${exec.vu.idInTest}`, value))
-    .then((_) => redisClient.del(`foo-${exec.vu.idInTest}`))
-    .then((_) => redisClient.exists(`foo-${exec.vu.idInTest}`))
-    .then((exists) => {
-      if (exists !== 0) {
-        throw new Error("foo should have been deleted");
-      }
-    });
-}
+// export function measureRedisPerformance() {
+//   // VUs are executed in a parallel fashion,
+//   // thus, to ensure that parallel VUs are not
+//   // modifying the same key at the same time,
+//   // we use keys indexed by the VU id.
+//   const key = `foo-${exec.vu.idInTest}`;
 
-export function setup() {
-  redisClient.sadd("crocodile_ids", ...crocodileIDs);
-}
+//   client
+//     .set(`foo-${exec.vu.idInTest}`, 1)
+//     .then(() => client.get(`foo-${exec.vu.idInTest}`))
+//     .then((value) => client.incrBy(`foo-${exec.vu.idInTest}`, value))
+//     .then((_) => client.del(`foo-${exec.vu.idInTest}`))
+//     .then((_) => client.exists(`foo-${exec.vu.idInTest}`))
+//     .then((exists) => {
+//       if (exists !== 0) {
+//         throw new Error("foo should have been deleted");
+//       }
+//     });
+// }
 
-export function measureUsingRedisData() {
-  // Pick a random crocodile id from the dedicated redis set,
-  // we have filled in setup().
-  redisClient
-    .srandmember("crocodile_ids")
-    .then((randomID) => {
-      const url = `https://test-api.k6.io/public/crocodiles/${randomID}`;
-      const res = http.get(url);
-
-      check(res, {
-        "status is 200": (r) => r.status === 200,
-        "content-type is application/json": (r) =>
-          r.headers["content-type"] === "application/json",
-      });
-
-      return url;
-    })
-    .then((url) => redisClient.hincrby("k6_crocodile_fetched", url, 1));
-}
+// export function setup() {
+//   client.sadd("crocodile_ids", ...crocodileIDs);
+// }
 
-export function teardown() {
-  redisClient.del("crocodile_ids");
-}
+// export function measureUsingRedisData() {
+//   // Pick a random crocodile id from the dedicated redis set,
+//   // we have filled in setup().
+//   client
+//     .srandmember("crocodile_ids")
+//     .then((randomID) => {
+//       const url = `https://test-api.k6.io/public/crocodiles/${randomID}`;
+//       const res = http.get(url);
 
-export function handleSummary(data) {
-  redisClient
-    .hgetall("k6_crocodile_fetched")
-    .then((fetched) => Object.assign(data, { k6_crocodile_fetched: fetched }))
-    .then((data) =>
-      redisClient.set(`k6_report_${Date.now()}`, JSON.stringify(data))
-    )
-    .then(() => redisClient.del("k6_crocodile_fetched"));
-
-  return {
-    stdout: textSummary(data, { indent: "  ", enableColors: true }),
-  };
-}
+//       check(res, {
+//         "status is 200": (r) => r.status === 200,
+//         "content-type is application/json": (r) =>
+//           r.headers["content-type"] === "application/json",
+//       });
+
+//       return url;
+//     })
+//     .then((url) => client.hincrby("k6_crocodile_fetched", url, 1));
+// }
+
+// export function teardown() {
+//   client.del("crocodile_ids");
+// }
+
+// export function handleSummary(data) {
+//   client
+//     .hgetall("k6_crocodile_fetched")
+//     .then((fetched) => Object.assign(data, { k6_crocodile_fetched: fetched }))
+//     .then((data) =>
+//       client.set(`k6_report_${Date.now()}`, JSON.stringify(data))
+//     )
+//     .then(() => client.del("k6_crocodile_fetched"));
+
+//   return {
+//     stdout: textSummary(data, { indent: "  ", enableColors: true }),
+//   };
+// }