liveflow/main.go

package main

import (
	"context"
	"fmt"
	"net/http"
	_ "net/http/pprof" // pprof을 사용하기 위한 패키지
	"strconv"

	"liveflow/config"
	"liveflow/media/streamer/egress/hls"
	"liveflow/media/streamer/egress/record/mp4"
	"liveflow/media/streamer/egress/record/webm"
	"liveflow/media/streamer/egress/whep"
	"liveflow/media/streamer/ingress/whip"

	"github.com/labstack/echo/v4"
	"github.com/labstack/echo/v4/middleware"
	"github.com/pion/webrtc/v3"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/sirupsen/logrus"
	"github.com/spf13/viper"

	"liveflow/httpsrv"
	"liveflow/log"
	"liveflow/media/hlshub"
	"liveflow/media/hub"
	"liveflow/media/streamer/ingress/rtmp"
)

// RTMP 받으면 자동으로 Service 서비스 동작, 녹화 서비스까지~?
func main() {
	ctx := context.Background()
	viper.SetConfigName("config") // name of config file (without extension)
	viper.SetConfigType("toml")   // REQUIRED if the config file does not have the extension in the name
	viper.AddConfigPath(".")      // optionally look for config in the working directory
	viper.BindEnv("docker.mode", "DOCKER_MODE")
	err := viper.ReadInConfig() // Find and read the config file
	if err != nil {             // Handle errors reading the config file
		panic(fmt.Errorf("fatal error config file: %w", err))
	}
	var conf config.Config
	err = viper.Unmarshal(&conf)
	if err != nil {
		panic(fmt.Errorf("failed to unmarshal config: %w", err))
	}
	fmt.Printf("Config: %+v\n", conf)
	log.Init()
	//log.SetCaller(ctx, true)
	//log.SetFormatter(ctx, &logrus.JSONFormatter{
	//	TimestampFormat: "2006-01-02 15:04:05",
	//})
	ctx = log.WithFields(ctx, logrus.Fields{
		"app": "liveflow",
	})
	log.Info(ctx, "liveflow is started")
	hub := hub.NewHub()
	var tracks map[string][]*webrtc.TrackLocalStaticRTP
	tracks = make(map[string][]*webrtc.TrackLocalStaticRTP)
	// ingress
	// Egress 서비스는 streamID 알림을 구독하여 처리 시작
	go func() {
		api := echo.New()
		api.HideBanner = true
		hlsHub := hlshub.NewHLSHub()
		hlsHandler := httpsrv.NewHandler(hlsHub)
		hlsRoute := api.Group("/hls", middleware.CORSWithConfig(middleware.CORSConfig{
			AllowOrigins: []string{"*"}, // Adjust origins as necessary
			AllowMethods: []string{http.MethodGet, http.MethodHead, http.MethodOptions},
		}))
		api.GET("/prometheus", echo.WrapHandler(promhttp.Handler()))
		api.GET("/debug/pprof/*", echo.WrapHandler(http.DefaultServeMux))
		// Enable CORS only for /hls routes
		hlsRoute.GET("/:streamID/master.m3u8", hlsHandler.HandleMasterM3U8)
		hlsRoute.GET("/:streamID/:playlistName/stream.m3u8", hlsHandler.HandleM3U8)
		hlsRoute.GET("/:streamID/:playlistName/:resourceName", hlsHandler.HandleM3U8)
		whipServer := whip.NewWHIP(whip.WHIPArgs{
			Hub:        hub,
			Tracks:     tracks,
			DockerMode: conf.Docker.Mode,
			Echo:       api,
		})
		whipServer.RegisterRoute()
		go func() {
			fmt.Println("----------------", conf.Service.Port)
			api.Start("0.0.0.0:" + strconv.Itoa(conf.Service.Port))
		}()
		// ingress 의 rtmp, whip 서비스로부터 streamID를 받아 Service, ContainerMP4, WHEP 서비스 시작
		for source := range hub.SubscribeToStreamID() {
			log.Infof(ctx, "New streamID received: %s", source.StreamID())
			if conf.MP4.Record {
				mp4 := mp4.NewMP4(mp4.MP4Args{
					Hub:             hub,
					SplitIntervalMS: 3000,
				})
				err = mp4.Start(ctx, source)
				if err != nil {
					log.Errorf(ctx, "failed to start mp4: %v", err)
				}
			}
			if conf.EBML.Record {
				webmStarter := webm.NewWEBM(webm.WebMArgs{
					Hub:             hub,
					SplitIntervalMS: 6000,
					StreamID:        source.StreamID(),
				})
				err = webmStarter.Start(ctx, source)
				if err != nil {
					log.Errorf(ctx, "failed to start webm: %v", err)
				}
			}
			hls := hls.NewHLS(hls.HLSArgs{
				Hub:     hub,
				HLSHub:  hlsHub,
				Port:    conf.Service.Port,
				LLHLS:   conf.Service.LLHLS,
				DiskRam: conf.Service.DiskRam,
			})
			err := hls.Start(ctx, source)
			if err != nil {
				log.Errorf(ctx, "failed to start hls: %v", err)
			}
			whep := whep.NewWHEP(whep.WHEPArgs{
				Tracks: tracks,
				Hub:    hub,
			})
			err = whep.Start(ctx, source)
			if err != nil {
				log.Errorf(ctx, "failed to start whep: %v", err)
			}
		}
	}()

	rtmpServer := rtmp.NewRTMP(rtmp.RTMPArgs{
		Hub:  hub,
		Port: conf.RTMP.Port,
	})
	rtmpServer.Serve(ctx)
}