project structure refactoring
This commit is contained in:
@@ -3,9 +3,10 @@ package main
|
||||
import (
|
||||
"fmt"
|
||||
"rkkvm/config"
|
||||
"rkkvm/http/hw/rtc"
|
||||
"rkkvm/http/hw/stream"
|
||||
"rkkvm/external/ffmpeg"
|
||||
"rkkvm/external/process"
|
||||
"rkkvm/http/route"
|
||||
"rkkvm/http/rtc"
|
||||
"strings"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -30,18 +31,7 @@ func main() {
|
||||
log.Println("Failed to parse log level, use default level: info")
|
||||
}
|
||||
|
||||
if cfg.Stream.Source == config.StreamSourceMjpeg {
|
||||
ustreamer := stream.Init(cfg.UStreamer.Path, strings.Split(cfg.UStreamer.Args, " "))
|
||||
ustreamer.Start()
|
||||
//go ustreamer.Watch()
|
||||
} else if cfg.Stream.Source == config.StreamSourceH264 {
|
||||
v := stream.InitFFmpeg()
|
||||
v.Start()
|
||||
} else {
|
||||
log.Fatalf("unsupported stream source type: %v", cfg.Stream.Source)
|
||||
}
|
||||
|
||||
webrtc, err := rtc.InitListener(cfg.WebRtc.Host, cfg.WebRtc.Port, 5006)
|
||||
webrtc, err := rtc.InitListener(cfg.WebRtc.Host, cfg.WebRtc.VideoPort, 5006)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
@@ -49,6 +39,17 @@ func main() {
|
||||
go webrtc.VideoListenerRead()
|
||||
go webrtc.AudioListenerRead()
|
||||
|
||||
if cfg.Stream.Source == config.StreamSourceMjpeg {
|
||||
ustreamer := process.Init(cfg.UStreamer.Path, strings.Split(cfg.UStreamer.Args, " "))
|
||||
ustreamer.Start()
|
||||
//go ustreamer.Watch()
|
||||
} else if cfg.Stream.Source == config.StreamSourceH264 {
|
||||
v := ffmpeg.InitFFmpeg()
|
||||
v.Start()
|
||||
} else {
|
||||
log.Fatalf("unsupported stream source type: %v", cfg.Stream.Source)
|
||||
}
|
||||
|
||||
r := gin.Default()
|
||||
r.Use(gin.Recovery())
|
||||
route.Api(r)
|
||||
|
||||
Reference in New Issue
Block a user