feat: add websocket handler

pkg/msgbus/service.go

@@ -12,6 +12,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/gorilla/websocket"
 	"github.com/sour-is/ev/pkg/es"
 	"github.com/sour-is/ev/pkg/es/event"
 )
@@ -27,10 +28,32 @@ func New(ctx context.Context, es *es.EventStore) (*service, error) {
 	return &service{es}, nil
 }
 
+var upgrader = websocket.Upgrader{
+	WriteBufferSize: 4096,
+	CheckOrigin: func(r *http.Request) bool {
+		return true
+	},
+}
+
 func (s *service) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	switch r.Method {
+	case http.MethodGet:
+		if r.Header.Get("Upgrade") == "websocket" {
+			s.websocket(w, r)
+			return
+		}
+		s.get(w, r)
+	case http.MethodPost, http.MethodPut:
+		s.post(w, r)
+	default:
+		w.WriteHeader(http.StatusMethodNotAllowed)
+	}
+}
+
+func (s *service) get(w http.ResponseWriter, r *http.Request) {
 	ctx := r.Context()
 
-	name, tags, _ := strings.Cut(r.URL.Path, "/")
+	name, _, _ := strings.Cut(r.URL.Path, "/")
 	if name == "" {
 		w.WriteHeader(http.StatusNotFound)
 		return
@@ -41,8 +64,6 @@ func (s *service) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		first = lis[0]
 	}
 
-	switch r.Method {
-	case http.MethodGet:
 	var pos, count int64 = -1, -99
 	qry := r.URL.Query()
 
@@ -76,9 +97,21 @@ func (s *service) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	for i := range events {
 		fmt.Fprintln(w, events[i])
 	}
+}
+func (s *service) post(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
 
+	name, tags, _ := strings.Cut(r.URL.Path, "/")
+	if name == "" {
+		w.WriteHeader(http.StatusNotFound)
 		return
-	case http.MethodPost, http.MethodPut:
+	}
+
+	var first event.Event = event.NilEvent
+	if lis, err := s.es.Read(ctx, "post-"+name, 0, 1); err == nil && len(lis) > 0 {
+		first = lis[0]
+	}
+
 	b, err := io.ReadAll(io.LimitReader(r.Body, 64*1024))
 	if err != nil {
 		log.Print(err)
@@ -126,9 +159,100 @@ func (s *service) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 	w.Header().Add("Content-Type", "text/plain")
 	fmt.Fprintf(w, "OK %d %s", m.Position, m.EventID)
+}
+func (s *service) websocket(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+
+	name, _, _ := strings.Cut(r.URL.Path, "/")
+	if name == "" {
+		w.WriteHeader(http.StatusNotFound)
 		return
-	default:
+	}
-		w.WriteHeader(http.StatusMethodNotAllowed)
+
+	var first event.Event = event.NilEvent
+	if lis, err := s.es.Read(ctx, "post-"+name, 0, 1); err == nil && len(lis) > 0 {
+		first = lis[0]
+	}
+
+	var pos int64 = -1
+	qry := r.URL.Query()
+
+	if i, err := strconv.ParseInt(qry.Get("idx"), 10, 64); err == nil {
+		pos = i
+	}
+
+	log.Print("WS topic=", name, " idx=", pos)
+
+	c, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		log.Print("upgrade:", err)
+		return
+	}
+	defer c.Close()
+
+	ctx, cancel := context.WithCancel(ctx)
+	c.SetCloseHandler(func(code int, text string) error {
+		cancel()
+		return nil
+	})
+	go func() {
+		for {
+			if err := ctx.Err(); err != nil {
+				return
+			}
+			mt, message, err := c.ReadMessage()
+			if err != nil {
+				log.Println("read:", err)
+				return
+			}
+			log.Printf("recv: %d %s", mt, message)
+		}
+	}()
+
+	es := s.es.EventStream()
+	if es == nil {
+		log.Println("EventStore does not implement streaming")
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	sub, err := es.Subscribe(ctx, "post-"+name, pos)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer func() {
+		ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
+		defer cancel()
+		log.Println("stop ws")
+		sub.Close(ctx)
+	}()
+
+	log.Println("start ws")
+	for sub.Recv(ctx) {
+		events, err := sub.Events(ctx)
+		if err != nil {
+			break
+		}
+		log.Println("got events ", len(events))
+		for _, e := range events {
+			e, ok := e.(*PostEvent)
+			if !ok {
+				continue
+			}
+			log.Println("send", e.String())
+
+			var b bytes.Buffer
+			if err = encodeJSON(&b, first, events); err != nil {
+				log.Print(err)
+			}
+
+			err = c.WriteMessage(websocket.TextMessage, b.Bytes())
+			if err != nil {
+				log.Println("write:", err)
+				break
+			}
+		}
 	}
 }
 
@@ -199,7 +323,7 @@ func encodeJSON(w io.Writer, first event.Event, events event.Events) error {
 		out[i].Created = e.EventMeta().Created().Format(time.RFC3339Nano)
 		out[i].Payload = e.Payload
 		out[i].Tags = e.Tags
-		out[i].Topic.Name = e.EventMeta().StreamID
+		out[i].Topic.Name = strings.TrimPrefix(e.EventMeta().StreamID, "post-")
 		out[i].Topic.Created = first.EventMeta().Created().Format(time.RFC3339Nano)
 		out[i].Topic.Seq = e.EventMeta().Position
 	}