actions/runner
1
0
Fork 0
mirror of https://github.com/actions/runner.git synced 2025-12-11 12:57:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP

Browse Source
This commit is contained in:
Patrick Ellis
2023-05-18 13:21:32 -07:00
committed by GitHub
parent 3b149bd8db
commit 1558d47a31
4 changed files with 29 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/Runner.Common/BrokerServer.cs
View File

@@ -5,6 +5,7 @@ using System.Threading.Tasks;
using GitHub.Actions.RunService.WebApi;
using GitHub.DistributedTask.Pipelines;
using GitHub.DistributedTask.WebApi;
using GitHub.Runner.Common.Util;
using GitHub.Runner.Sdk;
using GitHub.Services.Common;
using Sdk.RSWebApi.Contracts;
@@ -18,6 +19,7 @@ namespace GitHub.Runner.Common
Task ConnectAsync(Uri serverUrl, VssCredentials credentials);
Task<TaskAgentMessage> GetRunnerMessageAsync(CancellationToken token, TaskAgentStatus status, string version);
Task<bool> DeleteRunnerMessageAsync(string messageID, CancellationToken token);
}
public sealed class BrokerServer : RunnerService, IBrokerServer
@@ -48,9 +50,20 @@ namespace GitHub.Runner.Common
{
CheckConnection();
var jobMessage = RetryRequest<TaskAgentMessage>(
async () => await _brokerHttpClient.GetRunnerMessageAsync(version, status, cancellationToken), cancellationToken);
async () => await _brokerHttpClient.GetRunnerMessageAsync(version, status, cancellationToken), cancellationToken
);
return jobMessage;
}
public Task<bool> DeleteRunnerMessageAsync(string messageID, CancellationToken cancellationToken)
{
CheckConnection();
return RetryRequest(
async () => await _brokerHttpClient.DeleteRunnerMessageAsync(messageID, cancellationToken),
cancellationToken
);
}
}
}

13
src/Runner.Listener/BrokerMessageListener.cs
View File

@@ -168,10 +168,19 @@ namespace GitHub.Runner.Listener
}
}
public async Task DeleteMessageAsync(string messageID, CancellationToken token)
public async Task DeleteMessageAsync(TaskAgentMessage message)
{
Trace.Entering();
await _brokerServer.DeleteRunnerMessageAsync(messageID, token);
if (MessageUtil.IsRunServiceJob(message.MessageType))
{
var messageRef = StringUtil.ConvertFromJson<RunnerJobRequestRef>(message.Body);
using (var cs = new CancellationTokenSource(TimeSpan.FromSeconds(30)))
{
await _brokerServer.DeleteRunnerMessageAsync(messageRef.RunnerRequestId, cs.Token);
}
}
}
private bool IsGetNextMessageExceptionRetriable(Exception ex)

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

@@ -393,8 +393,6 @@ namespace GitHub.Runner.Listener
TaskAgentMessage message = null;
bool skipMessageDeletion = false;
bool useBrokerDeletion = false;
string brokerDeletionParamsMessageID = null;
CancellationToken brokerDeletionParamsToken = null;
try
{
Task<TaskAgentMessage> getNextMessage = _listener.GetNextMessageAsync(messageQueueLoopTokenSource.Token);
@@ -557,9 +555,6 @@ namespace GitHub.Runner.Listener
jobRequestMessage =
await runServer.GetJobMessageAsync(messageRef.RunnerRequestId,
messageQueueLoopTokenSource.Token);
useBrokerDeletion = true;
brokerDeletionParamsMessageID = messageRef.RunnerRequestId;
brokerDeletionParamsToken = messageQueueLoopTokenSource.Token;
}
catch (TaskOrchestrationJobAlreadyAcquiredException)
{
@@ -608,7 +603,8 @@ namespace GitHub.Runner.Listener
{
if (useBrokerDeletion)
{
await _listener.DeleteMessageAsync(brokerDeletionParamsMessageID, brokerDeletionParamsToken);
// await _listener.DeleteMessageAsync(brokerDeletionParamsMessageID, brokerDeletionParamsToken);
await _listener.DeleteMessageAsync(message);
}
else
{

9
src/Sdk/WebApi/WebApi/BrokerHttpClient.cs
View File

@@ -89,7 +89,7 @@ namespace GitHub.Actions.RunService.WebApi
throw new Exception($"Failed to get job message: {result.Error}");
}
public async Task<TaskAgentMessage> DeleteRunnerMessageAsync(
public async Task<bool> DeleteRunnerMessageAsync(
string messageID,
CancellationToken cancellationToken = default
)
@@ -109,12 +109,7 @@ namespace GitHub.Actions.RunService.WebApi
queryParameters: queryParams,
cancellationToken: cancellationToken);
if (result.IsSuccess)
{
return result.Value;
}
throw new Exception($"Failed to delete job message: {result.Error}");
return (result.IsSuccess);
}
}
}