From 07b9a6c08d55676e1669d7edf8aff83a0599e9e5 Mon Sep 17 00:00:00 2001
From: Jason Hyde <github@2bad.me>
Date: Wed, 14 Aug 2024 02:46:28 +0200
Subject: [PATCH] [ts] add io simulations

---
 source/server.ts | 48 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 47 insertions(+), 1 deletion(-)

diff --git a/source/server.ts b/source/server.ts
index 9c4e14d..313f818 100644
--- a/source/server.ts
+++ b/source/server.ts
@@ -1,4 +1,7 @@
-import fastify from 'fastify';
+import fastify from 'fastify'
+import fs from 'node:fs/promises'
+import os from 'node:os'
+import path from 'node:path'
 
 const PORT = 3000
 const MAX_DELAY = 5000
@@ -107,6 +110,49 @@ server.get<{ Querystring: { size: string } }>('/memory/allocate', async (request
   return `Allocated ${size} bytes of memory\n`
 })
 
+// Route 7: Simulate file read operation
+server.get<{ Querystring: { size: string } }>('/io/read', async (request) => {
+  const size = Number.parseInt(request.query.size, 10) || 1024 * 1024 // Default to 1MB
+  const filePath = path.join(os.tmpdir(), 'temp_read_file.txt')
+
+  // Create a file with random data
+  await fs.writeFile(filePath, Buffer.alloc(size).fill('A'))
+
+  const start = Date.now()
+  await fs.readFile(filePath)
+  const duration = Date.now() - start
+
+  await fs.unlink(filePath) // Clean up the file
+
+  return `Read ${size} bytes in ${duration} ms\n`
+})
+
+// Route 8: Simulate file write operation
+server.get<{ Querystring: { size: string } }>('/io/write', async (request) => {
+  const size = Number.parseInt(request.query.size, 10) || 1024 * 1024 // Default to 1MB
+  const filePath = path.join(os.tmpdir(), 'temp_write_file.txt')
+
+  const start = Date.now()
+  await fs.writeFile(filePath, Buffer.alloc(size).fill('A'))
+  const duration = Date.now() - start
+
+  await fs.unlink(filePath) // Clean up the file
+
+  return `Wrote ${size} bytes in ${duration} ms\n`
+})
+
+// Route 9: Simulate network request
+server.get<{ Querystring: { url: string } }>('/io/network', async (request) => {
+  const url = request.query.url || 'https://test.k6.io'
+
+  const start = Date.now()
+  const response = await fetch(url)
+  await response.text()
+  const duration = Date.now() - start
+
+  return `Fetched ${url} in ${duration} ms\n`
+})
+
 server.listen({ port: PORT }, (err, address) => {
   if (err) {
     console.error(err)