Merge pull request #23 from speakeasy-api/feat/add-sse-flat-response-endpoints

feat: add sse flat response endpoints

Author: jagarnica

cmd/server/main.go

@@ -58,8 +58,10 @@ func main() {
 	r.HandleFunc("/eventstreams/multiline", eventstreams.HandleEventStreamMultiLine).Methods(http.MethodPost)
 	r.HandleFunc("/eventstreams/rich", eventstreams.HandleEventStreamRich).Methods(http.MethodPost)
 	r.HandleFunc("/eventstreams/chat", eventstreams.HandleEventStreamChat).Methods(http.MethodPost)
+	r.HandleFunc("/eventstreams/chat-flat", eventstreams.HandleEventStreamChatFlatten).Methods(http.MethodPost)
 	r.HandleFunc("/eventstreams/chat-chunked", eventstreams.HandleEventStreamChat).Methods(http.MethodPost)
 	r.HandleFunc("/eventstreams/differentdataschemas", eventstreams.HandleEventStreamDifferentDataSchemas).Methods(http.MethodPost)
+	r.HandleFunc("/eventstreams/differentdataschemas-flat", eventstreams.HandleEventStreamDifferentDataSchemasFlatten).Methods(http.MethodPost)
 	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)

cmd/server/server

@@ -139,6 +139,32 @@ func HandleEventStreamChat(rw http.ResponseWriter, _ *http.Request) {
 	})
 }
 
+func HandleEventStreamChatFlatten(rw http.ResponseWriter, _ *http.Request) {
+	rw.Header().Add("Content-Type", "text/event-stream")
+
+	pushEvents(rw, [][]string{
+		{
+			`data: {"content": "Hello"}`,
+		},
+
+		{
+			`data: {"content": " "}`,
+		},
+
+		{
+			`data: {"content": "world"}`,
+		},
+
+		{
+			`data: {"content": "!"}`,
+		},
+
+		{
+			`data: [DONE]`,
+		},
+	})
+}
+
 func HandleEventStreamChatChunked(rw http.ResponseWriter, _ *http.Request) {
 	rw.Header().Add("Content-Type", "text/event-stream")
 
@@ -174,3 +200,27 @@ func HandleEventStreamDifferentDataSchemas(rw http.ResponseWriter, _ *http.Reque
 		},
 	})
 }
+
+func HandleEventStreamDifferentDataSchemasFlatten(rw http.ResponseWriter, _ *http.Request) {
+	rw.Header().Add("Content-Type", "text/event-stream")
+
+	pushEvents(rw, [][]string{
+		{
+			`id: event-1`,
+			`event: message`,
+			`data: {"content": "Here is your url"}`,
+		},
+
+		{
+			`id: event-2`,
+			`event: url`,
+			`data: {"url": "https://example.com"}`,
+		},
+
+		{
+			`id: event-3`,
+			`event: message`,
+			`data: {"content": "Have a great day!"}`,
+		},
+	})
+}