datastar/site/routes_examples_quick_prime...

85 lines
2.0 KiB
Go

package site
import (
"encoding/binary"
"encoding/hex"
"math/rand"
"net/http"
"sync"
"time"
"github.com/go-chi/chi/v5"
"github.com/goccy/go-json"
datastar "github.com/starfederation/datastar/sdk/go"
// "github.com/goccy/go-json"
)
func setupExamplesQuickPrimerGo(examplesRouter chi.Router) error {
examplesRouter.Route("/quick_primer_go/data", func(dataRouter chi.Router) {
signals := &QuickPrimerGoSignals{"initial backend data", false}
mu := &sync.RWMutex{}
dataRouter.Get("/replace", func(w http.ResponseWriter, r *http.Request) {
mu.RLock()
defer mu.RUnlock()
sse := datastar.NewSSE(w, r)
sse.MergeFragmentTempl(QuickPrimerGoView(signals))
})
dataRouter.Put("/", func(w http.ResponseWriter, r *http.Request) {
reqSignals := &QuickPrimerGoSignals{}
if err := datastar.ReadSignals(r, reqSignals); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if err := sanitizer.Sanitize(reqSignals); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
mu.Lock()
signals = reqSignals
mu.Unlock()
datastar.NewSSE(w, r).MergeFragmentTempl(QuickPrimerGoPut(signals))
})
dataRouter.Get("/", func(w http.ResponseWriter, r *http.Request) {
sse := datastar.NewSSE(w, r)
mu.RLock()
defer mu.RUnlock()
b, err := json.Marshal(signals)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
sse.MergeFragmentTempl(
QuickPrimerGoGet(string(b)),
datastar.WithSelector("main"),
datastar.WithMergePrepend(),
)
})
dataRouter.Get("/feed", func(w http.ResponseWriter, r *http.Request) {
sse := datastar.NewSSE(w, r)
ticker := time.NewTicker(2 * time.Second)
for {
select {
case <-r.Context().Done():
return
case <-ticker.C:
buf := make([]byte, 8)
binary.LittleEndian.PutUint64(buf, rand.Uint64())
sse.MergeFragmentTempl(QuickPrimerGoFeed(hex.EncodeToString(buf)))
}
}
})
})
return nil
}