feat: Make count and delay configurable in /sse endpoint

Author: sharat87

assets/index.html

@@ -248,7 +248,8 @@ <h3 id=client-tuned-responses>Client Tuned Response <a href="#client-tuned-respo
 
     <dt id=sse>/sse</dt>
     <dd>Responds with 10 <a href="https://developer.mozilla.org/en-US/docs/Web/API/Server-sent_events/Using_server-sent_events">
-        Server sent events</a>, each after 1s of delay. The count and delay aren't configurable today.
+        Server sent events</a>, each after 1s of delay. The count and delay can be set as query params. Count should be between 1
+        and 100. Delay should be between 1 and 10.
     </dd>
 
     <dt id=links>/links/<span class=var>{count}</span></dt>

routes/sse/handler.go

@@ -16,18 +16,40 @@ var RouteList = []ex.Route{
 }
 
 func handleServerSentEvents(ex *ex.Exchange) response.Response {
+	delay, err := ex.QueryParamInt("delay", 1)
+	if err != nil {
+		return response.BadRequest("Invalid delay value")
+	}
+	if delay < 1 {
+		return response.BadRequest("Delay must be greater than 0")
+	}
+	if delay > 10 {
+		return response.BadRequest("Delay must be less than 10")
+	}
+
+	count, err := ex.QueryParamInt("count", 10)
+	if err != nil {
+		return response.BadRequest("Invalid count value")
+	}
+	if count < 1 {
+		return response.BadRequest("Count must be greater than 0")
+	}
+	if count > 100 {
+		return response.BadRequest("Count must be less than 100")
+	}
+
 	return response.Response{
 		Header: map[string][]string{
 			"Cache-Control": {"no-store"},
 			c.ContentType:   {"text/event-stream"},
 		},
 		Writer: func(w response.BodyWriter) {
-			for id := range 10 {
+			for id := range count {
 				err := w.Write(strings.Join(pingMessage(id+1), "\n") + "\n\n")
 				if err != nil {
 					log.Printf("Error writing to response: %v\n", err)
 				}
-				time.Sleep(1 * time.Second)
+				time.Sleep(time.Duration(delay) * time.Second)
 			}
 		},
 	}