Mirrors/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2025-12-17 04:14:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix race condition in running encoders (#199)

Browse Source
This commit is contained in:
WithoutPants
2019-11-15 05:27:13 +11:00
committed by StashAppDev
parent 06fa88cc0a
commit 49c9b1aa9a
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/ffmpeg/encoder.go
View File

@@ -7,6 +7,7 @@ import (
"os"
"os/exec"
"strings"
"sync"
"time"
"github.com/stashapp/stash/pkg/logger"
@@ -16,7 +17,10 @@ type Encoder struct {
Path string
}
var runningEncoders map[string][]*os.Process = make(map[string][]*os.Process)
var (
runningEncoders map[string][]*os.Process = make(map[string][]*os.Process)
runningEncodersMutex = sync.RWMutex{}
)
func NewEncoder(ffmpegPath string) Encoder {
return Encoder{
@@ -25,12 +29,16 @@ func NewEncoder(ffmpegPath string) Encoder {
}
func registerRunningEncoder(path string, process *os.Process) {
runningEncodersMutex.Lock()
processes := runningEncoders[path]
runningEncoders[path] = append(processes, process)
runningEncodersMutex.Unlock()
}
func deregisterRunningEncoder(path string, process *os.Process) {
runningEncodersMutex.Lock()
defer runningEncodersMutex.Unlock()
processes := runningEncoders[path]
for i, v := range processes {
@@ -49,7 +57,9 @@ func waitAndDeregister(path string, cmd *exec.Cmd) error {
}
func KillRunningEncoders(path string) {
runningEncodersMutex.RLock()
processes := runningEncoders[path]
runningEncodersMutex.RUnlock()
for _, process := range processes {
// assume it worked, don't check for error