This commit is contained in:
eric sciple
2024-06-11 12:48:51 -07:00
parent d13cf5d9a8
commit a4c63ec012
2 changed files with 11 additions and 1 deletions

View File

@@ -397,11 +397,19 @@ namespace GitHub.Runner.Listener
if (message != null && _session.SessionId != Guid.Empty) if (message != null && _session.SessionId != Guid.Empty)
{ {
using (var cs = new CancellationTokenSource(TimeSpan.FromSeconds(30))) using (var cs = new CancellationTokenSource(TimeSpan.FromSeconds(30)))
{
if (MessageUtil.IsRunServiceJob(message.MessageType))
{
var messageRef = StringUtil.ConvertFromJson<RunnerJobRequestRef>(message.Body);
await _brokerServer.DeleteRunnerMessageAsync(_session.SessionId, messageRef.RunnerRequestId, cs.Token);
}
else
{ {
await _runnerServer.DeleteAgentMessageAsync(_settings.PoolId, message.MessageId, _session.SessionId, cs.Token); await _runnerServer.DeleteAgentMessageAsync(_settings.PoolId, message.MessageId, _session.SessionId, cs.Token);
} }
} }
} }
}
public async Task RefreshListenerTokenAsync(CancellationToken cancellationToken) public async Task RefreshListenerTokenAsync(CancellationToken cancellationToken)
{ {

View File

@@ -141,6 +141,8 @@ namespace GitHub.Actions.RunService.WebApi
queryParams.Add("jobMessageKey", jobMessageKey); queryParams.Add("jobMessageKey", jobMessageKey);
} }
queryParams.Add("status", TaskAgentStatus.Online.ToString());
var result = await SendAsync<object>( var result = await SendAsync<object>(
new HttpMethod("DELETE"), new HttpMethod("DELETE"),
requestUri: requestUri, requestUri: requestUri,