适配引擎变更

main.go

@@ -5,38 +5,22 @@ import (
 	"log"
 
 	. "github.com/Monibuca/engine/v4"
+	"github.com/Monibuca/engine/v4/config"
 	"github.com/Monibuca/engine/v4/util"
 	. "github.com/logrusorgru/aurora"
 )
 
 type RTMPConfig struct {
-	Publish   PublishConfig
-	Subscribe SubscribeConfig
-	TCPConfig
+	config.Publish
+	config.Subscribe
+	config.TCP
 	ChunkSize int
-	context.Context
-	cancel context.CancelFunc
 }
 
-var config = &RTMPConfig{
-	Publish:   DefaultPublishConfig,
-	Subscribe: DefaultSubscribeConfig,
-	ChunkSize: 4096,
-	TCPConfig: TCPConfig{ListenAddr: ":1935"},
-}
-
-func (cfg *RTMPConfig) Update(override Config) {
-	override.Unmarshal(cfg)
-	if config.cancel == nil {
-		util.Print(Green("server rtmp start at"), BrightBlue(config.ListenAddr))
-	} else if override.Has("ListenAddr") {
-		config.cancel()
-		util.Print(Green("server rtmp restart at"), BrightBlue(config.ListenAddr))
-	} else {
-		return
-	}
-	config.Context, config.cancel = context.WithCancel(Ctx)
-	err := cfg.Listen(cfg)
+func (config *RTMPConfig) Update(override config.Config) {
+	override.Unmarshal(config)
+	util.Print(Green("server rtmp start at"), BrightBlue(config.ListenAddr))
+	err := config.Listen(plugin, config)
 	if err == context.Canceled {
 		log.Println(err)
 	} else {
@@ -44,6 +28,7 @@ func (cfg *RTMPConfig) Update(override Config) {
 	}
 }
 
-func init() {
-	InstallPlugin(config)
-}
+var plugin = InstallPlugin(&RTMPConfig{
+	ChunkSize: 4096,
+	TCP:       config.TCP{ListenAddr: ":1935"},
+})