split name first

livekit · Sep 14, 2023 · 8d6f628 · 8d6f628
1 parent 141c5f5
commit 8d6f628
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pkg/pipeline/builder/stream.go b/pkg/pipeline/builder/stream.go
@@ -16,7 +16,6 @@ package builder
 
 import (
 	"fmt"
-	"strings"
 	"sync"
 
 	"github.com/tinyzimmer/go-gst/gst"
@@ -98,9 +97,7 @@ func BuildStreamBin(pipeline *gstreamer.Pipeline, p *config.PipelineConfig) (*St
 	return sb, b, nil
 }
 
-func (sb *StreamBin) GetStreamUrl(element string) (string, error) {
-	name := strings.Split(element, "_")[1]
-
+func (sb *StreamBin) GetStreamUrl(name string) (string, error) {
 	sb.mu.RLock()
 	url, ok := sb.urls[name]
 	sb.mu.RUnlock()

diff --git a/pkg/pipeline/watch.go b/pkg/pipeline/watch.go
@@ -164,9 +164,11 @@ func (c *Controller) handleMessageError(gErr *gst.GError) error {
 
 	switch {
 	case element == elementGstRtmp2Sink:
+		name = strings.Split(name, "_")[1]
+
 		if strings.HasPrefix(gErr.Error(), "Connection error") && !c.eos.IsBroken() {
 			// try reconnecting
-			ok, err := c.streamBin.ResetStream(name[10:], gErr)
+			ok, err := c.streamBin.ResetStream(name, gErr)
 			if err != nil {
 				logger.Errorw("failed to reset stream", err)
 			} else if ok {