Opt: init hls

zijiren233 · Nov 3, 2023 · 58de2ae · 58de2ae
1 parent 5b7c93d
commit 58de2ae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/cmd/server.go b/cmd/server.go
@@ -44,8 +44,7 @@ func Server(cmd *cobra.Command, args []string) {
 	channels := rwmap.RWMap[string, *server.Channel]{}
 	s := server.NewRtmpServer(func(ReqAppName, ReqChannelName string, IsPublisher bool) (*server.Channel, error) {
 		c, _ := channels.LoadOrStore(ReqAppName, server.NewChannel())
-		c.InitHlsPlayer()
-		return c, nil
+		return c, c.InitHlsPlayer()
 	})
 	go s.Serve(tcp)
 	if flags.Dev {

diff --git a/server/channel.go b/server/channel.go
@@ -170,22 +170,21 @@ func (c *Channel) GetPlayers() ([]av.WriteCloser, error) {
 	return players, nil
 }
 
-func (c *Channel) InitHlsPlayer() error {
+func (c *Channel) InitHlsPlayer(conf ...hls.SourceConf) error {
 	c.wg.Add(1)
 	defer c.wg.Done()
 	if c.Closed() {
 		return ErrClosed
 	}
 	c.hlsOnce.Do(func() {
-		p := hls.NewSource()
+		p := hls.NewSource(conf...)
 		c.hlsWriter.Store(p)
 		go func() {
 			for {
 				if c.Closed() {
 					return
 				}
 				if err := c.AddPlayer(p); err != nil {
-					p.Close()
 					continue
 				}
 				p.SendPacket()