mirror of
https://github.com/rosenbjerg/FFMpegCore.git
synced 2024-11-10 08:34:12 +01:00
Add cancel timeout
(cherry picked from commit 6383164f267516fbd50d50b2a511c15c25a168dc)
Former-commit-id: 9672713e63
This commit is contained in:
parent
cbb6c5a055
commit
50cb183ae2
2 changed files with 57 additions and 17 deletions
|
@ -12,6 +12,7 @@
|
|||
using FFMpegCore.Arguments;
|
||||
using FFMpegCore.Exceptions;
|
||||
using FFMpegCore.Pipes;
|
||||
using System.Threading;
|
||||
|
||||
namespace FFMpegCore.Test
|
||||
{
|
||||
|
@ -598,14 +599,13 @@ public async Task Video_Cancel_Async()
|
|||
var outputFile = new TemporaryFile("out.mp4");
|
||||
|
||||
var task = FFMpegArguments
|
||||
.FromFileInput(TestResources.Mp4Video)
|
||||
.FromFileInput("testsrc2=size=320x240[out0]; sine[out1]", false, args => args
|
||||
.WithCustomArgument("-re")
|
||||
.ForceFormat("lavfi"))
|
||||
.OutputToFile(outputFile, false, opt => opt
|
||||
.Resize(new Size(1000, 1000))
|
||||
.WithAudioCodec(AudioCodec.Aac)
|
||||
.WithVideoCodec(VideoCodec.LibX264)
|
||||
.WithConstantRateFactor(14)
|
||||
.WithSpeedPreset(Speed.VerySlow)
|
||||
.Loop(3))
|
||||
.WithSpeedPreset(Speed.VeryFast))
|
||||
.CancellableThrough(out var cancel)
|
||||
.ProcessAsynchronously(false);
|
||||
|
||||
|
@ -613,7 +613,39 @@ public async Task Video_Cancel_Async()
|
|||
cancel();
|
||||
|
||||
var result = await task;
|
||||
|
||||
Assert.IsFalse(result);
|
||||
}
|
||||
|
||||
[TestMethod, Timeout(10000)]
|
||||
public async Task Video_Cancel_Async_With_Timeout()
|
||||
{
|
||||
var outputFile = new TemporaryFile("out.mp4");
|
||||
|
||||
var task = FFMpegArguments
|
||||
.FromFileInput("testsrc2=size=320x240[out0]; sine[out1]", false, args => args
|
||||
.WithCustomArgument("-re")
|
||||
.ForceFormat("lavfi"))
|
||||
.OutputToFile(outputFile, false, opt => opt
|
||||
.WithAudioCodec(AudioCodec.Aac)
|
||||
.WithVideoCodec(VideoCodec.LibX264)
|
||||
.WithSpeedPreset(Speed.VeryFast))
|
||||
.CancellableThrough(out var cancel, 10000)
|
||||
.ProcessAsynchronously(false);
|
||||
|
||||
await Task.Delay(300);
|
||||
cancel();
|
||||
|
||||
var result = await task;
|
||||
|
||||
var outputInfo = FFProbe.Analyse(outputFile);
|
||||
|
||||
Assert.IsTrue(result);
|
||||
Assert.IsNotNull(outputInfo);
|
||||
Assert.AreEqual(320, outputInfo.PrimaryVideoStream.Width);
|
||||
Assert.AreEqual(240, outputInfo.PrimaryVideoStream.Height);
|
||||
Assert.AreEqual("h264", outputInfo.PrimaryVideoStream.CodecName);
|
||||
Assert.AreEqual("aac", outputInfo.PrimaryAudioStream.CodecName);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -27,7 +27,7 @@ internal FFMpegArgumentProcessor(FFMpegArguments ffMpegArguments)
|
|||
|
||||
public string Arguments => _ffMpegArguments.Text;
|
||||
|
||||
private event EventHandler CancelEvent = null!;
|
||||
private event EventHandler<int> CancelEvent = null!;
|
||||
|
||||
public FFMpegArgumentProcessor NotifyOnProgress(Action<double> onPercentageProgress, TimeSpan totalTimeSpan)
|
||||
{
|
||||
|
@ -45,9 +45,9 @@ public FFMpegArgumentProcessor NotifyOnOutput(Action<string, DataType> onOutput)
|
|||
_onOutput = onOutput;
|
||||
return this;
|
||||
}
|
||||
public FFMpegArgumentProcessor CancellableThrough(out Action cancel)
|
||||
public FFMpegArgumentProcessor CancellableThrough(out Action cancel, int timeout = 0)
|
||||
{
|
||||
cancel = () => CancelEvent?.Invoke(this, EventArgs.Empty);
|
||||
cancel = () => CancelEvent?.Invoke(this, timeout);
|
||||
return this;
|
||||
}
|
||||
public bool ProcessSynchronously(bool throwOnError = true)
|
||||
|
@ -55,12 +55,16 @@ public bool ProcessSynchronously(bool throwOnError = true)
|
|||
using var instance = PrepareInstance(out var cancellationTokenSource);
|
||||
var errorCode = -1;
|
||||
|
||||
void OnCancelEvent(object sender, EventArgs args)
|
||||
void OnCancelEvent(object sender, int timeout)
|
||||
{
|
||||
instance.SendInput("q");
|
||||
|
||||
if (!cancellationTokenSource.Token.WaitHandle.WaitOne(timeout, true))
|
||||
{
|
||||
cancellationTokenSource.Cancel();
|
||||
instance.Started = false;
|
||||
}
|
||||
}
|
||||
CancelEvent += OnCancelEvent;
|
||||
instance.Exited += delegate { cancellationTokenSource.Cancel(); };
|
||||
|
||||
|
@ -102,12 +106,16 @@ public async Task<bool> ProcessAsynchronously(bool throwOnError = true)
|
|||
using var instance = PrepareInstance(out var cancellationTokenSource);
|
||||
var errorCode = -1;
|
||||
|
||||
void OnCancelEvent(object sender, EventArgs args)
|
||||
void OnCancelEvent(object sender, int timeout)
|
||||
{
|
||||
instance.SendInput("q");
|
||||
|
||||
if (!cancellationTokenSource.Token.WaitHandle.WaitOne(timeout, true))
|
||||
{
|
||||
cancellationTokenSource.Cancel();
|
||||
instance.Started = false;
|
||||
}
|
||||
}
|
||||
CancelEvent += OnCancelEvent;
|
||||
|
||||
try
|
||||
|
|
Loading…
Reference in a new issue