actions/runner
1
0
Fork 0
mirror of https://github.com/actions/runner.git synced 2025-12-10 20:36:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add common retry func

Browse Source
This commit is contained in:
Tatyana Kostromskaya
2022-06-10 09:46:18 +00:00
committed by GitHub
parent 9623a44c2f
commit fa0dd79c30
2 changed files with 52 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/Runner.Listener/Runner.cs
View File

@@ -13,6 +13,11 @@ using GitHub.Runner.Listener.Configuration;
using GitHub.Runner.Sdk;
using GitHub.Services.WebApi;
using Pipelines = GitHub.DistributedTask.Pipelines;
using System.Runtime.Serialization;
using System.Net;
using System.Net.Http;
using System.Net.Http.Headers;
using GitHub.Services.Common;
namespace GitHub.Runner.Listener
{
@@ -477,7 +482,12 @@ namespace GitHub.Runner.Listener
// todo: add retries https://github.com/github/actions-broker/issues/49
var runServer = HostContext.CreateService<IRunServer>();
await runServer.ConnectAsync(new Uri(settings.ServerUrl), creds);
var jobMessage = await runServer.GetJobMessageAsync(messageRef.RunnerRequestId);
Func<Task<AgentJobRequestMessage>> getJobRequestMessageAsync = async() => {
return await runServer.GetJobMessageAsync(messageRef.RunnerRequestId);
};
var jobMessage = await RetryHelper<AgentJobRequestMessage>.RetryWithTimeoutAsync(getJobRequestMessageAsync, TimeSpan.FromSeconds(5), TimeSpan.FromSeconds(10));
jobDispatcher.Run(jobMessage, runOnce);
if (runOnce)

41
src/Sdk/Common/Common/Utility/RetryHelper.cs Normal file
View File

@@ -0,0 +1,41 @@
using System;
using System.ComponentModel;
using System.Threading.Tasks;
namespace GitHub.Services.Common
{
[EditorBrowsable(EditorBrowsableState.Never)]
public static class RetryHelper<T>
{
public static async Task<T> RetryWithTimeoutAsync(
Func<Task<T>> retriableAction,
TimeSpan minBackoff,
TimeSpan maxBackoff,
int maxTimeout = 5
)
{
var remainingTime = TimeSpan.FromMinutes(maxTimeout);
while (true)
{
try
{
return await retriableAction();
}
catch
{
if (remainingTime > TimeSpan.Zero)
{
var backOff = BackoffTimerHelper.GetRandomBackoff(minBackoff, maxBackoff);
remainingTime -= backOff;
await Task.Delay(backOff);
}
else
{
throw;
}
}
}
}
}
}