fix stream handlers copying for Handler.Clone

lesismal · Jun 22, 2024 · b9bb0b6 · b9bb0b6
1 parent e73dcc7
commit b9bb0b6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/handler.go b/handler.go
@@ -317,6 +317,15 @@ func (h *handler) Clone() Handler {
 		cp.routes[k] = rh
 	}
 
+	cp.streams = map[string]*streamHandler{}
+	for k, v := range h.streams {
+		sh := &streamHandler{
+			async:   v.async,
+			handler: v.handler,
+		}
+		cp.streams[k] = sh
+	}
+
 	ctx, cancel := context.WithCancel(context.Background())
 	cp.ctx = ctx
 	cp.cancel = cancel