u
This commit is contained in:
parent
020f2eea9b
commit
8634c204f1
@ -108,7 +108,6 @@ func (s *Server) handleWebSocket(c *gin.Context) {
|
|||||||
go func() {
|
go func() {
|
||||||
ticker := time.NewTicker(1 * time.Second)
|
ticker := time.NewTicker(1 * time.Second)
|
||||||
for range ticker.C {
|
for range ticker.C {
|
||||||
streamer.GlobalStreamer.TruncateOutput()
|
|
||||||
currentVideoPath, _ := streamer.GlobalStreamer.GetCurrentVideoPath()
|
currentVideoPath, _ := streamer.GlobalStreamer.GetCurrentVideoPath()
|
||||||
s.Broadcast(mywebsocket.Date{
|
s.Broadcast(mywebsocket.Date{
|
||||||
Timestamp: time.Now().UnixMilli(),
|
Timestamp: time.Now().UnixMilli(),
|
||||||
|
@ -8,22 +8,29 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"live-streamer/config"
|
"live-streamer/config"
|
||||||
"log"
|
"log"
|
||||||
"math"
|
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Streamer struct {
|
type playState struct {
|
||||||
videoList []config.InputItem
|
|
||||||
currentVideoIndex int
|
currentVideoIndex int
|
||||||
|
manualControl bool
|
||||||
cmd *exec.Cmd
|
cmd *exec.Cmd
|
||||||
ctx context.Context
|
ctx context.Context
|
||||||
cancel context.CancelFunc
|
cancel context.CancelFunc
|
||||||
|
}
|
||||||
|
|
||||||
|
type Streamer struct {
|
||||||
|
playStateMu sync.RWMutex
|
||||||
|
playState playState
|
||||||
|
|
||||||
|
videoMu sync.RWMutex
|
||||||
|
videoList []config.InputItem
|
||||||
|
|
||||||
|
outputMu sync.RWMutex
|
||||||
output strings.Builder
|
output strings.Builder
|
||||||
manualControl bool
|
|
||||||
mu sync.Mutex
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var GlobalStreamer *Streamer
|
var GlobalStreamer *Streamer
|
||||||
@ -31,22 +38,25 @@ var GlobalStreamer *Streamer
|
|||||||
func NewStreamer(videoList []config.InputItem) *Streamer {
|
func NewStreamer(videoList []config.InputItem) *Streamer {
|
||||||
GlobalStreamer = &Streamer{
|
GlobalStreamer = &Streamer{
|
||||||
videoList: videoList,
|
videoList: videoList,
|
||||||
currentVideoIndex: 0,
|
playState: playState{},
|
||||||
cmd: nil,
|
output: strings.Builder{},
|
||||||
ctx: nil,
|
|
||||||
}
|
}
|
||||||
return GlobalStreamer
|
return GlobalStreamer
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) start() {
|
func (s *Streamer) start() {
|
||||||
s.mu.Lock()
|
s.playStateMu.Lock()
|
||||||
s.ctx, s.cancel = context.WithCancel(context.Background())
|
s.playState.ctx, s.playState.cancel = context.WithCancel(context.Background())
|
||||||
currentVideo := s.videoList[s.currentVideoIndex]
|
currentVideo := s.videoList[s.playState.currentVideoIndex]
|
||||||
videoPath := currentVideo.Path
|
videoPath := currentVideo.Path
|
||||||
s.cmd = exec.CommandContext(s.ctx, "ffmpeg", s.buildFFmpegArgs(currentVideo)...)
|
s.playState.cmd = exec.CommandContext(s.playState.ctx, "ffmpeg", s.buildFFmpegArgs(currentVideo)...)
|
||||||
s.mu.Unlock()
|
cmd := s.playState.cmd
|
||||||
|
ctx := s.playState.ctx
|
||||||
|
s.playStateMu.Unlock()
|
||||||
|
|
||||||
s.writeOutput(fmt.Sprintln("start stream: ", videoPath))
|
s.writeOutput(fmt.Sprintln("start stream: ", videoPath))
|
||||||
pipe, err := s.cmd.StderrPipe()
|
|
||||||
|
pipe, err := cmd.StderrPipe()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("failed to get pipe: %v", err)
|
log.Printf("failed to get pipe: %v", err)
|
||||||
return
|
return
|
||||||
@ -54,25 +64,29 @@ func (s *Streamer) start() {
|
|||||||
|
|
||||||
reader := bufio.NewReader(pipe)
|
reader := bufio.NewReader(pipe)
|
||||||
|
|
||||||
if err := s.cmd.Start(); err != nil {
|
if err := cmd.Start(); err != nil {
|
||||||
s.writeOutput(fmt.Sprintf("starting ffmpeg error: %v\n", err))
|
s.writeOutput(fmt.Sprintf("starting ffmpeg error: %v\n", err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
go s.log(reader)
|
go s.log(reader)
|
||||||
|
|
||||||
<-s.ctx.Done()
|
<-ctx.Done()
|
||||||
s.writeOutput(fmt.Sprintf("stop stream: %s\n", videoPath))
|
s.writeOutput(fmt.Sprintf("stop stream: %s\n", videoPath))
|
||||||
|
|
||||||
if s.manualControl {
|
s.playStateMu.Lock()
|
||||||
s.manualControl = false
|
if s.playState.manualControl {
|
||||||
|
// manualing change video, don't increase currentVideoIndex
|
||||||
|
s.playState.manualControl = false
|
||||||
} else {
|
} else {
|
||||||
// stream next video
|
s.playState.currentVideoIndex++
|
||||||
s.currentVideoIndex++
|
s.videoMu.RLock()
|
||||||
if s.currentVideoIndex >= len(s.videoList) {
|
if s.playState.currentVideoIndex >= len(s.videoList) {
|
||||||
s.currentVideoIndex = 0
|
s.playState.currentVideoIndex = 0
|
||||||
}
|
}
|
||||||
|
s.videoMu.RUnlock()
|
||||||
}
|
}
|
||||||
|
s.playStateMu.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) Stream() {
|
func (s *Streamer) Stream() {
|
||||||
@ -86,83 +100,123 @@ func (s *Streamer) Stream() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) Stop() {
|
func (s *Streamer) Stop() {
|
||||||
s.mu.Lock()
|
s.playStateMu.Lock()
|
||||||
defer s.mu.Unlock()
|
cancel := s.playState.cancel
|
||||||
if s.cancel != nil {
|
cmd := s.playState.cmd
|
||||||
stopped := make(chan error)
|
s.playState.cancel = nil
|
||||||
|
s.playState.cmd = nil
|
||||||
|
s.playStateMu.Unlock()
|
||||||
|
|
||||||
|
if cancel == nil || cmd == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
stopped := make(chan error, 1)
|
||||||
go func() {
|
go func() {
|
||||||
if s.cmd != nil {
|
if cmd.Process != nil {
|
||||||
stopped <- s.cmd.Wait()
|
stopped <- cmd.Wait()
|
||||||
|
} else {
|
||||||
|
stopped <- nil
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
s.cancel()
|
|
||||||
if s.cmd != nil && s.cmd.Process != nil {
|
cancel()
|
||||||
|
|
||||||
|
if cmd.Process != nil {
|
||||||
select {
|
select {
|
||||||
case <-stopped:
|
case <-stopped:
|
||||||
break
|
|
||||||
case <-time.After(3 * time.Second):
|
case <-time.After(3 * time.Second):
|
||||||
_ = s.cmd.Process.Kill()
|
_ = cmd.Process.Kill()
|
||||||
break
|
|
||||||
}
|
}
|
||||||
s.cmd = nil
|
close(stopped)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Streamer) writeOutput(str string) {
|
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
s.output.WriteString(str)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) Add(videoPath string) {
|
func (s *Streamer) Add(videoPath string) {
|
||||||
s.mu.Lock()
|
s.videoMu.Lock()
|
||||||
defer s.mu.Unlock()
|
defer s.videoMu.Unlock()
|
||||||
s.videoList = append(s.videoList, config.InputItem{Path: videoPath})
|
s.videoList = append(s.videoList, config.InputItem{Path: videoPath})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) Remove(videoPath string) {
|
func (s *Streamer) Remove(videoPath string) {
|
||||||
s.mu.Lock()
|
var needStop bool // removed video is current playing
|
||||||
defer s.mu.Unlock()
|
var removeIndex int = -1
|
||||||
|
|
||||||
|
s.videoMu.Lock()
|
||||||
for i, item := range s.videoList {
|
for i, item := range s.videoList {
|
||||||
if item.Path == videoPath {
|
if item.Path == videoPath {
|
||||||
s.videoList = append(s.videoList[:i], s.videoList[i+1:]...)
|
removeIndex = i
|
||||||
if s.currentVideoIndex >= len(s.videoList) {
|
|
||||||
s.currentVideoIndex = 0
|
s.playStateMu.RLock()
|
||||||
}
|
needStop = (s.playState.currentVideoIndex == i)
|
||||||
if s.currentVideoIndex == i {
|
s.playStateMu.RUnlock()
|
||||||
s.Stop()
|
|
||||||
}
|
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if removeIndex >= 0 && removeIndex < len(s.videoList) {
|
||||||
|
oldLen := len(s.videoList)
|
||||||
|
s.videoList = append(s.videoList[:removeIndex], s.videoList[removeIndex+1:]...)
|
||||||
|
|
||||||
|
s.playStateMu.Lock()
|
||||||
|
if s.playState.currentVideoIndex >= oldLen-1 {
|
||||||
|
s.playState.currentVideoIndex = 0
|
||||||
|
}
|
||||||
|
s.playStateMu.Unlock()
|
||||||
|
}
|
||||||
|
s.videoMu.Unlock()
|
||||||
|
|
||||||
|
if needStop {
|
||||||
|
s.Stop()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) Prev() {
|
func (s *Streamer) Prev() {
|
||||||
s.mu.Lock()
|
s.videoMu.RLock()
|
||||||
s.manualControl = true
|
videoLen := len(s.videoList)
|
||||||
s.currentVideoIndex--
|
if videoLen == 0 {
|
||||||
if s.currentVideoIndex < 0 {
|
return
|
||||||
s.currentVideoIndex = len(s.videoList) - 1
|
|
||||||
}
|
}
|
||||||
s.mu.Unlock()
|
s.videoMu.RUnlock()
|
||||||
|
|
||||||
|
s.playStateMu.Lock()
|
||||||
|
s.playState.manualControl = true
|
||||||
|
s.playState.currentVideoIndex--
|
||||||
|
if s.playState.currentVideoIndex < 0 {
|
||||||
|
s.playState.currentVideoIndex = videoLen - 1
|
||||||
|
}
|
||||||
|
s.playStateMu.Unlock()
|
||||||
|
|
||||||
s.Stop()
|
s.Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) Next() {
|
func (s *Streamer) Next() {
|
||||||
s.mu.Lock()
|
s.videoMu.RLock()
|
||||||
s.manualControl = true
|
videoLen := len(s.videoList)
|
||||||
s.currentVideoIndex++
|
if videoLen == 0 {
|
||||||
if s.currentVideoIndex >= len(s.videoList) {
|
return
|
||||||
s.currentVideoIndex = 0
|
|
||||||
}
|
}
|
||||||
s.mu.Unlock()
|
s.videoMu.RUnlock()
|
||||||
|
|
||||||
|
s.playStateMu.Lock()
|
||||||
|
s.playState.manualControl = true
|
||||||
|
s.playState.currentVideoIndex++
|
||||||
|
if s.playState.currentVideoIndex >= videoLen {
|
||||||
|
s.playState.currentVideoIndex = 0
|
||||||
|
}
|
||||||
|
s.playStateMu.Unlock()
|
||||||
|
|
||||||
s.Stop()
|
s.Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) log(reader *bufio.Reader) {
|
func (s *Streamer) log(reader *bufio.Reader) {
|
||||||
|
s.playStateMu.RLock()
|
||||||
|
ctx := s.playState.ctx
|
||||||
|
s.playStateMu.RUnlock()
|
||||||
|
|
||||||
select {
|
select {
|
||||||
case <-s.ctx.Done():
|
case <-ctx.Done():
|
||||||
return
|
return
|
||||||
default:
|
default:
|
||||||
if !config.GlobalConfig.Log.PlayState {
|
if !config.GlobalConfig.Log.PlayState {
|
||||||
@ -187,23 +241,23 @@ func (s *Streamer) log(reader *bufio.Reader) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) GetCurrentVideoPath() (string, error) {
|
func (s *Streamer) GetCurrentVideoPath() (string, error) {
|
||||||
s.mu.Lock()
|
s.videoMu.RLock()
|
||||||
defer s.mu.Unlock()
|
defer s.videoMu.RUnlock()
|
||||||
if len(s.videoList) == 0 {
|
if len(s.videoList) == 0 {
|
||||||
return "", errors.New("no video streaming")
|
return "", errors.New("no video streaming")
|
||||||
}
|
}
|
||||||
return s.videoList[s.currentVideoIndex].Path, nil
|
return s.videoList[s.GetCurrentIndex()].Path, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) GetVideoList() []config.InputItem {
|
func (s *Streamer) GetVideoList() []config.InputItem {
|
||||||
s.mu.Lock()
|
s.videoMu.RLock()
|
||||||
defer s.mu.Unlock()
|
defer s.videoMu.RUnlock()
|
||||||
return s.videoList
|
return s.videoList
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) GetVideoListPath() []string {
|
func (s *Streamer) GetVideoListPath() []string {
|
||||||
s.mu.Lock()
|
s.videoMu.RLock()
|
||||||
defer s.mu.Unlock()
|
defer s.videoMu.RUnlock()
|
||||||
var videoList []string
|
var videoList []string
|
||||||
for _, item := range s.videoList {
|
for _, item := range s.videoList {
|
||||||
videoList = append(videoList, item.Path)
|
videoList = append(videoList, item.Path)
|
||||||
@ -212,30 +266,23 @@ func (s *Streamer) GetVideoListPath() []string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) GetCurrentIndex() int {
|
func (s *Streamer) GetCurrentIndex() int {
|
||||||
s.mu.Lock()
|
s.playStateMu.RLock()
|
||||||
defer s.mu.Unlock()
|
defer s.playStateMu.RUnlock()
|
||||||
return s.currentVideoIndex
|
return s.playState.currentVideoIndex
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Streamer) writeOutput(str string) {
|
||||||
|
s.outputMu.Lock()
|
||||||
|
defer s.outputMu.Unlock()
|
||||||
|
s.output.WriteString(str)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) GetOutput() string {
|
func (s *Streamer) GetOutput() string {
|
||||||
s.mu.Lock()
|
s.outputMu.RLock()
|
||||||
defer s.mu.Unlock()
|
defer s.outputMu.RUnlock()
|
||||||
return s.output.String()
|
return s.output.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Streamer) TruncateOutput() int {
|
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
currentSize := s.output.Len()
|
|
||||||
if currentSize > math.MaxInt {
|
|
||||||
newStart := currentSize - math.MaxInt
|
|
||||||
trimmedOutput := s.output.String()[newStart:]
|
|
||||||
s.output.Reset()
|
|
||||||
s.output.WriteString(trimmedOutput)
|
|
||||||
}
|
|
||||||
return currentSize
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Streamer) Close() {
|
func (s *Streamer) Close() {
|
||||||
s.Stop()
|
s.Stop()
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user