Я пытаюсь написать прокси-сервер, на котором я буду конвертировать видеофайл в прямую трансляцию. VOD-файл состоит из нескольких дочерних файлов манифеста с разной скоростью передачи данных. каждый дочерний манифест состоит из нескольких сегментов ts, скажем, 4 секунды каждый. Для простоты я создал фиктивную карту из 2 дочерних манифестов, содержащую 4 сегмента ts каждый. Моя задача состоит в том, чтобы создать бесконечный прямой эфир для цикла for. поэтому я отправляю каждый сегмент ts через 4 секунды в процедуре go и записываю в канал, и у меня есть еще одна процедура go, где я читаю из этой процедуры go и записываю в выходную строку. как только все сегменты закончились, я обновляюсь до loopEnd канала и начинаю снова. Есть 2 API для разных битрейтов, которые дают мне новые значения каждые 4 секунды.
package main
import (
"fmt"
"github.com/gin-contrib/gzip"
"github.com/gin-gonic/gin"
"sync"
"time"
)
func main() {
vodToLiveMap := convertVodToLive()
engine := gin.Default()
engine.Use(gzip.Gzip(gzip.DefaultCompression))
engine.GET("/134200", func(gctx *gin.Context) {
prepareResponse(gctx, "134200", vodToLiveMap)
})
engine.GET("/290400", func(gctx *gin.Context) {
prepareResponse(gctx, "290400", vodToLiveMap)
})
engine.Run() // listen and serve on 0.0.0.0:8080
}
// this represents one ulr for http live stream
type hlsEntry struct {
timeCode int // timecode in microseconds
body string // string including transport segment URL, timecode
}
type VodToLive struct {
SequenceNumber int
Out chan string
CurrentStream string
}
func convertVodToLive() sync.Map {
var hlsEntriesBitrate1 = []hlsEntry{
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer1_001.ts\n",
},
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer1_002.ts\n",
},
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer1_003.ts\n",
},
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer1_004.ts\n",
},
}
var hlsEntriesBitrate2 = []hlsEntry{
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer2_001.ts\n",
},
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer2_002.ts\n",
},
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer2_003.ts\n",
},
{
timeCode: 4000000,
body: "#EXTINF:4.00000,\n/master_Layer2_004.ts\n",
},
}
fmt.Println(hlsEntriesBitrate2)
var hlsEntriesMap = make(map[string][]hlsEntry)
hlsEntriesMap["134200"] = hlsEntriesBitrate1
hlsEntriesMap["290400"] = hlsEntriesBitrate2
outputString := ""
vodToLiveMap := sync.Map{}
i := 0
//Iterate for all child manifest files of different bit rates
for childURLSlug := range hlsEntriesMap {
vodToLive := &VodToLive{}
vodToLive.SequenceNumber = 1
vodToLive.Out = make(chan string)
vodToLiveMap.Store(childURLSlug, vodToLive)
loopEnd := make(chan bool)
//Writing to vodToLive for a childURLSlug
go func(childURLSlug string) {
for {
for _, entry := range hlsEntriesMap[childURLSlug] {
select {
case <-time.After(time.Duration(entry.timeCode) * time.Microsecond):
vodToLive.SequenceNumber++
vodToLive.Out <- entry.body
}
}
loopEnd <- true
fmt.Println(childURLSlug)
}
}(childURLSlug)
//Reading from vodToLive's Out channel for a childURLSlug to update CurrentStream
go func(childURLSlug string) {
for {
for {
select {
//If video reaches end of stream then restart stream from Sequence #1
case <-loopEnd:
vodToLive.SequenceNumber = 1
//time.Sleep(10 * time.Second)
vodToLive.CurrentStream = ""
break
default:
outputString = <-vodToLive.Out
fmt.Println(outputString)
vodToLive.CurrentStream = vodToLive.CurrentStream + outputString
}
}
}
}(childURLSlug)
i++
}
return vodToLiveMap
}
func prepareResponse(gctx *gin.Context, childURLSlug string, vodToLiveMap sync.Map) {
vodToLive, _ := vodToLiveMap.Load(childURLSlug)
gctx.Header("Content-Type", "application/x-mpegURL")
response := `#EXTM3U
#EXT-X-VERSION:3
#EXT-X-TARGETDURATION:4
#EXT-X-MEDIA-SEQUENCE:` + "1" + "\n" + vodToLive.(*VodToLive).CurrentStream
size := len(response)
if size > 0 && response[size-1] == '\n' {
response = response[:size-1]
}
gctx.String(200, response)
}
//curl localhost:8080/134200
//curl localhost:8080/290400
Но проблема здесь в том, что я не получаю все сегменты ts в запросе curl, и иногда даже ts не обновляются после окончания цикла. Я показываю все 4 сегмента TS в каждом вызове, но они должны быть сброшены после каждого endLoop.