Skip to content

Commit

Permalink
fix(converters): panic-safe stopping conversion
Browse files Browse the repository at this point in the history
  • Loading branch information
wailorman committed Oct 24, 2020
1 parent 442e62b commit 2034e9f
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 23 deletions.
21 changes: 11 additions & 10 deletions pkg/media/batch_converter.go
Original file line number Diff line number Diff line change
Expand Up @@ -123,43 +123,44 @@ func (bc *BatchConverter) convertOne(task ConverterTask, progress chan BatchProg
singleConverter := NewConverter(bc.infoGetter)
_progress, _finished, _failed := singleConverter.Convert(task)

go func() {
for {
select {
case <-bc.stopConversion:
singleConverter.StopConversion()
return
}
}
}()

for {
select {
case <-bc.stopConversion:
singleConverter.StopConversion()

case <-singleConverter.ConversionStarted:
bc.TaskConversionStarted <- task

case metadata := <-singleConverter.MetadataReceived:
bc.MetadataReceived <- MetadataReceivedBatchMessage{
Metadata: metadata,
Task: task,
}

case videoCodec := <-singleConverter.InputVideoCodecDetected:
bc.InputVideoCodecDetected <- InputVideoCodecDetectedBatchMessage{
Codec: videoCodec,
Task: task,
}

case <-singleConverter.ConversionStopping:
bc.ConversionStopping <- task

case <-singleConverter.ConversionStopped:
bc.ConversionStopped <- task

case progressMessage := <-_progress:
progress <- BatchProgressMessage{
Progress: progressMessage,
Task: task,
}

case errorMessage := <-_failed:
return errorMessage

case <-_finished:
return nil

}
}
}
21 changes: 8 additions & 13 deletions pkg/media/converter.go
Original file line number Diff line number Diff line change
Expand Up @@ -84,19 +84,6 @@ func (c *Converter) Convert(task ConverterTask) (
return
}

go func() {
for {
select {
case <-c.stopConversion:
c.ConversionStopping <- true
trans.Stop()
c.ConversionStopped <- true
finished <- true
return
}
}
}()

metadata, err := c.infoGetter.GetMediaInfo(task.InFile)

if err != nil {
Expand Down Expand Up @@ -149,6 +136,13 @@ func (c *Converter) Convert(task ConverterTask) (

for {
select {
case <-c.stopConversion:
c.ConversionStopping <- true
trans.Stop()
c.ConversionStopped <- true
finished <- true
return

case progressMessage := <-_progress:
if progressMessage.FramesProcessed != "" {
progress <- ConvertProgress{
Expand All @@ -161,6 +155,7 @@ func (c *Converter) Convert(task ConverterTask) (
File: task.InFile,
}
}

case err := <-done:
if err != nil {
failed <- err
Expand Down

0 comments on commit 2034e9f

Please sign in to comment.