Skip to content

Commit 8a855f3

Browse files
committed
feat: add sentinel and flat endpoint together
1 parent d003b56 commit 8a855f3

File tree

2 files changed

+30
-3
lines changed

2 files changed

+30
-3
lines changed

cmd/server/main.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,10 @@ func main() {
5858
r.HandleFunc("/eventstreams/multiline", eventstreams.HandleEventStreamMultiLine).Methods(http.MethodPost)
5959
r.HandleFunc("/eventstreams/rich", eventstreams.HandleEventStreamRich).Methods(http.MethodPost)
6060
r.HandleFunc("/eventstreams/chat", eventstreams.HandleEventStreamChat).Methods(http.MethodPost)
61+
r.HandleFunc("/eventstreams/chat-flat", eventstreams.HandleEventStreamChatFlatten).Methods(http.MethodPost)
6162
r.HandleFunc("/eventstreams/chat-chunked", eventstreams.HandleEventStreamChat).Methods(http.MethodPost)
6263
r.HandleFunc("/eventstreams/differentdataschemas", eventstreams.HandleEventStreamDifferentDataSchemas).Methods(http.MethodPost)
63-
r.HandleFunc("/eventstreams/differentdataschemasflatten", eventstreams.HandleEventStreamDifferentDataSchemasFlatten).Methods(http.MethodPost)
64+
r.HandleFunc("/eventstreams/differentdataschemas-flat", eventstreams.HandleEventStreamDifferentDataSchemasFlatten).Methods(http.MethodPost)
6465
r.HandleFunc("/clientcredentials/token", clientcredentials.HandleTokenRequest).Methods(http.MethodPost)
6566
r.HandleFunc("/clientcredentials/authenticatedrequest", clientcredentials.HandleAuthenticatedRequest).Methods(http.MethodPost)
6667
r.HandleFunc("/clientcredentials/alt/token", clientcredentials.HandleTokenRequest).Methods(http.MethodPost)
@@ -74,7 +75,7 @@ func main() {
7475
r.HandleFunc("/method/post", method.HandlePost).Methods(http.MethodPost)
7576
r.HandleFunc("/method/put", method.HandlePut).Methods(http.MethodPut)
7677
r.HandleFunc("/method/trace", method.HandleTrace).Methods(http.MethodTrace)
77-
78+
7879
oauth2router := r.NewRoute().Subrouter()
7980
oauth2router.Use(middleware.OAuth2)
8081
oauth2router.HandleFunc("/ecommerce/products", ecommerce.HandleListProducts).Methods(http.MethodGet)

internal/eventstreams/service.go

Lines changed: 27 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,32 @@ func HandleEventStreamChat(rw http.ResponseWriter, _ *http.Request) {
139139
})
140140
}
141141

142+
func HandleEventStreamChatFlatten(rw http.ResponseWriter, _ *http.Request) {
143+
rw.Header().Add("Content-Type", "text/event-stream")
144+
145+
pushEvents(rw, [][]string{
146+
{
147+
`data: {"content": "Hello"}`,
148+
},
149+
150+
{
151+
`data: {"content": " "}`,
152+
},
153+
154+
{
155+
`data: {"content": "world"}`,
156+
},
157+
158+
{
159+
`data: {"content": "!"}`,
160+
},
161+
162+
{
163+
`data: [DONE]`,
164+
},
165+
})
166+
}
167+
142168
func HandleEventStreamChatChunked(rw http.ResponseWriter, _ *http.Request) {
143169
rw.Header().Add("Content-Type", "text/event-stream")
144170

@@ -197,4 +223,4 @@ func HandleEventStreamDifferentDataSchemasFlatten(rw http.ResponseWriter, _ *htt
197223
`data: {"content": "Have a great day!"}`,
198224
},
199225
})
200-
}
226+
}

0 commit comments

Comments
 (0)