feat: add endpoint for testing x-ndjson (#27)

Author: kanwardeep007

cmd/server/main.go

@@ -20,6 +20,7 @@ import (
 	"github.com/speakeasy-api/speakeasy-api-test-service/internal/reflect"
 	"github.com/speakeasy-api/speakeasy-api-test-service/internal/responseHeaders"
 	"github.com/speakeasy-api/speakeasy-api-test-service/internal/retries"
+	"github.com/speakeasy-api/speakeasy-api-test-service/internal/xNdJson"
 
 	"github.com/gorilla/mux"
 	"github.com/speakeasy-api/speakeasy-api-test-service/internal/auth"
@@ -66,6 +67,8 @@ func main() {
 	r.HandleFunc("/eventstreams/differentdataschemas-flat", eventstreams.HandleEventStreamDifferentDataSchemasFlatten).Methods(http.MethodPost)
 	r.HandleFunc("/jsonl", jsonLines.HandleJSONLinesRich).Methods(http.MethodGet)
 	r.HandleFunc("/jsonl/chunks", jsonLines.HandleJSONLinesChunksRich).Methods(http.MethodGet)
+	r.HandleFunc("/x-ndjson", xNdJson.HandleXNdJsonLinesRich).Methods(http.MethodGet)
+	r.HandleFunc("/x-ndjson/chunks", xNdJson.HandleXNdJsonLinesChunksRich).Methods(http.MethodGet)
 	r.HandleFunc("/clientcredentials/token", clientcredentials.HandleTokenRequest).Methods(http.MethodPost)
 	r.HandleFunc("/clientcredentials/authenticatedrequest", clientcredentials.HandleAuthenticatedRequest).Methods(http.MethodPost)
 	r.HandleFunc("/clientcredentials/alt/token", clientcredentials.HandleTokenRequest).Methods(http.MethodPost)

internal/xNdJson/service.go

@@ -0,0 +1,56 @@
+package xNdJson
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+)
+
+func pushEvents(rw http.ResponseWriter, events [][]string) {
+	for _, event := range events {
+		for _, line := range event {
+			fmt.Fprint(rw, line)
+		}
+
+		if f, ok := rw.(http.Flusher); ok {
+			f.Flush()
+		}
+
+		time.Sleep(100 * time.Millisecond)
+	}
+}
+
+func pushChunks(rw http.ResponseWriter, chunks []string) {
+	for _, chunk := range chunks {
+		fmt.Fprint(rw, chunk)
+
+		if f, ok := rw.(http.Flusher); ok {
+			f.Flush()
+		}
+
+		time.Sleep(100 * time.Millisecond)
+	}
+}
+
+func HandleXNdJsonLinesChunksRich(rw http.ResponseWriter, _ *http.Request) {
+	rw.Header().Add("Content-Type", "application/x-ndjson")
+
+	pushChunks(rw, []string{
+		"{\"name\": \"Peter\", \"skills\": [\"Go\"",
+		", \"Python\"]}\n{\"name\": \"John\"",
+		", \"skills\": [\"Go\", \"Rust\"]}\n",
+	})
+}
+
+func HandleXNdJsonLinesRich(rw http.ResponseWriter, _ *http.Request) {
+	rw.Header().Add("Content-Type", "application/x-ndjson")
+
+	pushEvents(rw, [][]string{
+		{
+			"{\"name\": \"Peter\", \"skills\": [\"Go\", \"Python\"]}\n",
+		},
+		{
+			"{\"name\": \"John\", \"skills\": [\"Go\", \"Rust\"]}\n",
+		},
+	})
+}