generated from ElnuDev/go-project
Want to contribute? Fork me on Codeberg.org!
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.3 KiB
54 lines
1.3 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type handler = func(http.ResponseWriter, *http.Request)
|
|
|
|
func generateHandler(file string, handler func(), data func() any) handler {
|
|
tmpl := template.Must(template.ParseFiles(fmt.Sprintf("templates/%s", file)))
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
handler()
|
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
|
tmpl.Execute(w, data())
|
|
}
|
|
}
|
|
|
|
func generateSseHandler(handler func(http.ResponseWriter, *http.Request)) handler {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "text/event-stream")
|
|
w.Header().Set("Cache-Control", "no-cache")
|
|
w.Header().Set("Connection", "keep-alive")
|
|
w.WriteHeader(http.StatusOK)
|
|
handler(w, r)
|
|
}
|
|
}
|
|
|
|
func generateCounter() handler {
|
|
return generateSseHandler(func(w http.ResponseWriter, r *http.Request) {
|
|
tick := time.Tick(500 * time.Millisecond)
|
|
ctx := r.Context()
|
|
outer:
|
|
for i := 0; ; i++ {
|
|
select {
|
|
case <-ctx.Done():
|
|
break outer
|
|
case <-tick:
|
|
fmt.Fprintf(w, "event: count\ndata: <div>%d</div>\n\n", i)
|
|
w.(http.Flusher).Flush()
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
func main() {
|
|
http.Handle("/", http.FileServer(http.Dir("static")))
|
|
http.HandleFunc("/api/counter", generateCounter())
|
|
log.Fatal(http.ListenAndServe(":3333", nil))
|
|
}
|