From 3cfe19adcb46f930056513789e39639d74237701 Mon Sep 17 00:00:00 2001
From: zijiren233 <pyh1670605849@gmail.com>
Date: Sun, 16 Jun 2024 23:30:01 +0800
Subject: [PATCH] Fix: init live channel

---
 internal/op/movie.go | 49 ++++++++++++++++++++++++++++++--------------
 1 file changed, 34 insertions(+), 15 deletions(-)

diff --git a/internal/op/movie.go b/internal/op/movie.go
index f5cfeb3..05c7fd6 100644
--- a/internal/op/movie.go
+++ b/internal/op/movie.go
@@ -10,6 +10,7 @@ import (
 	"sync/atomic"
 	"time"
 
+	log "github.com/sirupsen/logrus"
 	"github.com/synctv-org/synctv/internal/cache"
 	"github.com/synctv-org/synctv/internal/conf"
 	"github.com/synctv-org/synctv/internal/model"
@@ -129,44 +130,55 @@ func (m *Movie) Channel() (*rtmps.Channel, error) {
 	if m.IsFolder {
 		return nil, errors.New("this is a folder")
 	}
-	err := m.initChannel()
+	c, err := m.initChannel()
 	if err != nil {
 		return nil, err
 	}
-	return m.channel.Load(), nil
+	return c, nil
 }
 
 func genTsName() string {
 	return utils.SortUUID()
 }
 
-func (m *Movie) compareAndSwapInitChannel() *rtmps.Channel {
+func (m *Movie) compareAndSwapInitChannel() (*rtmps.Channel, bool) {
 	c := m.channel.Load()
 	if c == nil {
 		c = rtmps.NewChannel()
 		if !m.channel.CompareAndSwap(nil, c) {
 			return m.compareAndSwapInitChannel()
 		}
-		c.InitHlsPlayer(hls.WithGenTsNameFunc(genTsName))
+		return c, true
 	}
-	return c
+	return c, false
 }
 
-func (m *Movie) initChannel() error {
+func (m *Movie) initChannel() (*rtmps.Channel, error) {
 	switch {
 	case m.Movie.MovieBase.Live && m.Movie.MovieBase.RtmpSource:
-		m.compareAndSwapInitChannel()
+		c, init := m.compareAndSwapInitChannel()
+		if !init {
+			return c, nil
+		}
+		err := c.InitHlsPlayer(hls.WithGenTsNameFunc(genTsName))
+		if err != nil {
+			return nil, err
+		}
+		return c, nil
 	case m.Movie.MovieBase.Live && m.Movie.MovieBase.Proxy:
 		u, err := url.Parse(m.Movie.MovieBase.Url)
 		if err != nil {
-			return err
+			return nil, err
 		}
 		switch u.Scheme {
 		case "rtmp":
-			c := m.compareAndSwapInitChannel()
+			c, init := m.compareAndSwapInitChannel()
+			if !init {
+				return c, nil
+			}
 			err = c.InitHlsPlayer(hls.WithGenTsNameFunc(genTsName))
 			if err != nil {
-				return err
+				return nil, err
 			}
 			go func() {
 				for {
@@ -185,11 +197,15 @@ func (m *Movie) initChannel() error {
 					}
 				}
 			}()
+			return c, nil
 		case "http", "https":
-			c := m.compareAndSwapInitChannel()
+			c, init := m.compareAndSwapInitChannel()
+			if !init {
+				return c, nil
+			}
 			err := c.InitHlsPlayer(hls.WithGenTsNameFunc(genTsName))
 			if err != nil {
-				return err
+				return nil, err
 			}
 			go func() {
 				for {
@@ -198,6 +214,7 @@ func (m *Movie) initChannel() error {
 					}
 					req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, m.Movie.MovieBase.Url, nil)
 					if err != nil {
+						log.Errorf("get live error: %v", err)
 						time.Sleep(time.Second)
 						continue
 					}
@@ -209,22 +226,24 @@ func (m *Movie) initChannel() error {
 					}
 					resp, err := uhc.Do(req)
 					if err != nil {
+						log.Errorf("get live error: %v", err)
 						time.Sleep(time.Second)
 						continue
 					}
 					if err := c.PushStart(flv.NewReader(resp.Body)); err != nil {
+						log.Errorf("push live error: %v", err)
 						time.Sleep(time.Second)
 					}
 					resp.Body.Close()
 				}
 			}()
+			return c, nil
 		default:
-			return errors.New("unsupported scheme")
+			return nil, errors.New("unsupported scheme")
 		}
 	default:
-		return errors.New("this movie not support channel")
+		return nil, errors.New("this movie not support channel")
 	}
-	return nil
 }
 
 func (movie *Movie) Validate() error {