Last active
June 27, 2024 02:15
-
-
Save fiorix/f58c6f8b2a8b23521a47 to your computer and use it in GitHub Desktop.
Go and Server Sent Events for HTTP/1.1 and HTTP/2.0
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| // Go and Server Sent Events for HTTP/1.1 and HTTP/2.0 | |
| //go:generate go run $GOROOT/src/crypto/tls/generate_cert.go -host localhost | |
| package main | |
| import ( | |
| "fmt" | |
| "io" | |
| "log" | |
| "net/http" | |
| "time" | |
| "github.com/bradfitz/http2" | |
| ) | |
| func main() { | |
| http.HandleFunc("/", mainHandler) | |
| http.HandleFunc("/sse", sseHandler) | |
| s := http.Server{Addr: ":8080"} | |
| http2.ConfigureServer(&s, nil) | |
| err := s.ListenAndServeTLS("cert.pem", "key.pem") | |
| if err != nil { | |
| log.Fatal(err) | |
| } | |
| } | |
| var html = `<html> | |
| <head> | |
| <script> | |
| var ev = new EventSource("/sse"); | |
| ev.onmessage = function(event) { | |
| document.getElementById("ev").innerHTML = event.data; | |
| } | |
| </script> | |
| </head> | |
| <body> | |
| <p id="ev"></p> | |
| </body> | |
| </html>` | |
| func mainHandler(w http.ResponseWriter, r *http.Request) { | |
| io.WriteString(w, html) | |
| } | |
| func sseHandler(w http.ResponseWriter, r *http.Request) { | |
| conn, ok := w.(http.Flusher) | |
| if !ok { | |
| http.Error(w, "Oops", http.StatusInternalServerError) | |
| return | |
| } | |
| w.Header().Set("Content-Type", "text/event-stream") | |
| w.WriteHeader(http.StatusOK) | |
| i := 0 | |
| for { | |
| _, err := fmt.Fprintf(w, "data: hello world %d\n\n", i) | |
| if err != nil { | |
| break | |
| } | |
| i++ | |
| conn.Flush() | |
| time.Sleep(time.Second) | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment