mirror of
https://github.com/lkmio/lkm.git
synced 2025-09-27 03:26:01 +08:00
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/yangjiechina/live-server/flv"
|
|
"github.com/yangjiechina/live-server/hls"
|
|
"github.com/yangjiechina/live-server/log"
|
|
"github.com/yangjiechina/live-server/rtc"
|
|
"github.com/yangjiechina/live-server/rtsp"
|
|
"go.uber.org/zap/zapcore"
|
|
"net"
|
|
"net/http"
|
|
|
|
_ "net/http/pprof"
|
|
|
|
"github.com/yangjiechina/avformat/librtmp"
|
|
"github.com/yangjiechina/avformat/utils"
|
|
"github.com/yangjiechina/live-server/rtmp"
|
|
"github.com/yangjiechina/live-server/stream"
|
|
)
|
|
|
|
var rtspAddr *net.TCPAddr
|
|
|
|
func CreateTransStream(source stream.ISource, protocol stream.Protocol, streams []utils.AVStream) stream.ITransStream {
|
|
if stream.ProtocolRtmp == protocol {
|
|
return rtmp.NewTransStream(librtmp.ChunkSize)
|
|
} else if stream.ProtocolHls == protocol {
|
|
id := source.Id()
|
|
|
|
transStream, err := hls.NewTransStream("", stream.AppConfig.Hls.M3U8Format(id), stream.AppConfig.Hls.TSFormat(id, "%d"), stream.AppConfig.Hls.Dir, stream.AppConfig.Hls.Duration, stream.AppConfig.Hls.PlaylistLength)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return transStream
|
|
} else if stream.ProtocolFlv == protocol {
|
|
return flv.NewHttpTransStream()
|
|
} else if stream.ProtocolRtsp == protocol {
|
|
trackFormat := source.Id() + "?track=%d"
|
|
return rtsp.NewTransStream(net.IPAddr{
|
|
IP: rtspAddr.IP,
|
|
Zone: rtspAddr.Zone,
|
|
}, trackFormat)
|
|
} else if stream.ProtocolRtc == protocol {
|
|
return rtc.NewTransStream()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
stream.TransStreamFactory = CreateTransStream
|
|
}
|
|
|
|
func main() {
|
|
//初始化日志
|
|
log.InitLogger(zapcore.DebugLevel, "./logs/lkm.log", 10, 100, 7, false)
|
|
|
|
stream.AppConfig.GOPCache = true
|
|
stream.AppConfig.MergeWriteLatency = 350
|
|
|
|
stream.AppConfig.Hls.Enable = true
|
|
stream.AppConfig.Hls.Dir = "../tmp"
|
|
stream.AppConfig.Hls.Duration = 2
|
|
stream.AppConfig.Hls.PlaylistLength = 10
|
|
|
|
rtmpAddr, err := net.ResolveTCPAddr("tcp", "0.0.0.0:1935")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
impl := rtmp.NewServer()
|
|
err = impl.Start(rtmpAddr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
println("启动rtmp服务成功:" + rtmpAddr.String())
|
|
|
|
rtspAddr, err = net.ResolveTCPAddr("tcp", "0.0.0.0:554")
|
|
if err != nil {
|
|
panic(rtspAddr)
|
|
}
|
|
|
|
rtspServer := rtsp.NewServer()
|
|
err = rtspServer.Start(rtspAddr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
println("启动rtsp服务成功:" + rtspAddr.String())
|
|
|
|
apiAddr := "0.0.0.0:8080"
|
|
go startApiServer(apiAddr)
|
|
|
|
loadConfigError := http.ListenAndServe(":19999", nil)
|
|
if loadConfigError != nil {
|
|
panic(loadConfigError)
|
|
}
|
|
select {}
|
|
}
|