feat: add transform from start

pkg/config/types.go

@@ -94,9 +94,10 @@ type (
 		Quic
 		TCP
 		UDP
-		Pull  map[string]Pull
+		Pull      map[string]Pull
-		OnSub OnSubscribe
+		Transform map[Regexp]Transform
-		OnPub OnPublish
+		OnSub     OnSubscribe
+		OnPub     OnPublish
 		DB
 	}
 	ICommonConf interface {

plugin/transcode/pkg/transform.go

@@ -3,8 +3,6 @@ package transcode
 import (
 	"bufio"
 	"fmt"
-	"m7s.live/pro/pkg"
-	"m7s.live/pro/pkg/filerotate"
 	"net"
 	"net/url"
 	"os"
@@ -12,7 +10,10 @@ import (
 	"strings"
 	"time"
 
-	"m7s.live/pro"
+	"m7s.live/pro/pkg"
+	"m7s.live/pro/pkg/filerotate"
+
+	m7s "m7s.live/pro"
 	"m7s.live/pro/pkg/config"
 	"m7s.live/pro/pkg/task"
 	"m7s.live/pro/pkg/util"
@@ -21,18 +22,20 @@ import (
 
 // / 定义传输模式的常量
 const (
-	TRANS_MODE_PIPE TransMode = "pipe"
+	TRANS_MODE_PIPE   TransMode = "pipe"
-	TRANS_MODE_RTSP TransMode = "rtsp"
+	TRANS_MODE_RTSP   TransMode = "rtsp"
-	TRANS_MODE_RTMP TransMode = "rtmp"
+	TRANS_MODE_RTMP   TransMode = "rtmp"
-	TRANS_MODE_LIB  TransMode = "lib"
+	TRANS_MODE_LIB    TransMode = "lib"
+	TRANS_MODE_REMOTE TransMode = "remote"
 )
 
 type (
 	TransMode    string
 	DecodeConfig struct {
-		Mode  TransMode `default:"pipe" json:"mode" desc:"转码模式"` //转码模式
+		Mode   TransMode `default:"pipe" json:"mode" desc:"转码模式"` //转码模式
-		Codec string    `json:"codec" desc:"解码器"`
+		Codec  string    `json:"codec" desc:"解码器"`
-		Args  string    `json:"args" desc:"解码参数"`
+		Args   string    `json:"args" desc:"解码参数"`
+		Remote string    `json:"remote" desc:"远程地址"`
 	}
 	EncodeConfig struct {
 		Codec string `json:"codec" desc:"编码器"`
@@ -81,13 +84,13 @@ func (t *Transformer) Start() (err error) {
 		args = append(args, "-c:v", t.From.Codec)
 	}
 	switch t.From.Mode {
-	case "pipe":
+	case TRANS_MODE_PIPE:
 		err = t.TransformJob.Subscribe()
 		if err != nil {
 			return
 		}
 		args = append(args, "-f", "flv", "-i", "pipe:0")
-	case "rtsp":
+	case TRANS_MODE_RTSP:
 		if rtspPlugin, ok := t.TransformJob.Plugin.Server.Plugins.Get("RTSP"); ok {
 			listenAddr := rtspPlugin.GetCommonConf().TCP.ListenAddr
 			if strings.HasPrefix(listenAddr, ":") {
@@ -95,7 +98,7 @@ func (t *Transformer) Start() (err error) {
 			}
 			args = append(args, "-i", "rtsp://"+listenAddr+"/"+t.TransformJob.StreamPath)
 		}
-	case "rtmp":
+	case TRANS_MODE_RTMP:
 		if rtmpPlugin, ok := t.TransformJob.Plugin.Server.Plugins.Get("RTMP"); ok {
 			listenAddr := rtmpPlugin.GetCommonConf().TCP.ListenAddr
 			if strings.HasPrefix(listenAddr, ":") {
@@ -103,6 +106,8 @@ func (t *Transformer) Start() (err error) {
 			}
 			args = append(args, "-i", "rtmp://"+listenAddr+"/"+t.TransformJob.StreamPath)
 		}
+	case TRANS_MODE_REMOTE:
+		args = append(args, "-i", t.From.Remote)
 	}
 	t.To = make([]EncodeConfig, len(t.TransformJob.Config.Output))
 	for i, to := range t.TransformJob.Config.Output {

server.go

@@ -142,6 +142,9 @@ func exit() {
 
 func init() {
 	Servers.Init()
+	Servers.OnBeforeDispose(func() {
+		time.AfterFunc(3*time.Second, exit)
+	})
 	Servers.OnDispose(exit)
 	for k, v := range myip.LocalAndInternalIPs() {
 		Routes[k] = v
@@ -299,6 +302,12 @@ func (s *Server) Start() (err error) {
 					plugin.handler.Pull(streamPath, conf)
 				}
 			}
+			if plugin.Meta.Transformer != nil {
+				for streamPath, conf := range plugin.config.Transform {
+					transformer := plugin.Meta.Transformer()
+					transformer.GetTransformJob().Init(transformer, plugin, streamPath.String(), conf)
+				}
+			}
 		}
 		if s.DB != nil {
 			s.DB.AutoMigrate(&Device{})
@@ -342,11 +351,12 @@ func (s *Server) Start() (err error) {
 			s.DB.Find(&devices)
 			for _, d := range devices {
 				d.server = s
+				d.Logger = s.Logger.With("device", d.ID, "type", d.Type, "name", d.Name)
 				d.ChangeStatus(DeviceStatusOffline)
 				if d.PubConf == nil {
 					d.PubConf = config.NewPublish()
 				}
-				s.Devices.Add(d, s.Logger.With("device", d.ID, "type", d.Type, "name", d.Name))
+				s.Devices.Add(d)
 			}
 		}
 		return nil