test relay server

internal/relay/relay.go

@@ -0,0 +1,31 @@
+package relay
+
+import (
+	"sync"
+
+	"github.com/okdaichi/gomoqt/moqt"
+)
+
+type Relay struct {
+	mu    sync.RWMutex
+	rooms map[string]*moqt.TrackMux
+}
+
+func New() *Relay {
+	return &Relay{
+		rooms: make(map[string]*moqt.TrackMux),
+	}
+}
+
+func (r *Relay) GetMux(room string) *moqt.TrackMux {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+
+	if mux, ok := r.rooms[room]; ok {
+		return mux
+	}
+
+	mux := moqt.NewTrackMux()
+	r.rooms[room] = mux
+	return mux
+}