private async Task EnableByInstance(AzureDevopsInstance azInstance, bool operateOnAllAgents = false)
{
// Selects ALL agents
var agents = azInstance.Agents;
// Filter out agents which we can't operate on. This is the default behaviour
if (operateOnAllAgents == false)
{
agents = agents.Where(x => x.AllowOperations);
}
var agentsByPool = agents.GroupBy(x => x.PoolID);
// Iterate over each pool and enable agents in parallel
await Task.WhenAll(agentsByPool.Select(async abp => {
foreach (var agent in abp)
{
logger.LogInformation("Enabling agent {AgentName} {AgentPoolId} {AgentServer}", agent.Name, abp.Key, azInstance.Client.BaseAddress);
var realAgent = await azInstance.Client.GetAgentAsync(agent.PoolID, agent.Id);
realAgent.Enabled = true;
await azInstance.Client.UpdateAgentAsync(abp.Key, agent.Id, realAgent);
}
}));
}
private async Task DeleteByInstance(AzureDevopsInstance azInstance)
{
var agentsByPool = azInstance.Agents.GroupBy(p => p.PoolID);
await Task.WhenAll(agentsByPool.Select(async abp => {
foreach (var agent in abp)
{
logger.LogInformation("Deleting agent {AgentName} {AgentPoolId} {AgentServer}", agent.Name, abp.Key, azInstance.Client.BaseAddress);
await azInstance.Client.DeleteAgentAsync(agent.PoolID, agent.Id);
}
}));
}
private async Task DrainByInstance(AzureDevopsInstance azInstance)
{
var agentsByPool = azInstance.Agents.GroupBy(p => p.PoolID);
// Iterate over each pool and disable agents in parallel
await Task.WhenAll(agentsByPool.Select(async abp =>
{
await Task.WhenAll(abp.Select(async agent =>
{
if (agent.AllowOperations) // An agent can be operated on if it was enabled when agents were discovered.
{
logger.LogInformation("Disabling {AgentName} {AgentPoolId} {AgentServer}", agent.Name, abp.Key, azInstance.Client.BaseAddress);
var realAgent = await azInstance.Client.GetAgentAsync(agent.PoolID, agent.Id);
realAgent.Enabled = false;
await azInstance.Client.UpdateAgentAsync(agent.PoolID, agent.Id, realAgent); // Update the agent so that it disabled
}
else
{
logger.LogInformation("{AgentName} already disabled {AgentPoolId} {AgentServer}", agent.Name, abp.Key, azInstance.Client.BaseAddress);
}
// Wait 15 seconds incase something was assigned to this agent in the time it took for the network traffic to send from client to server
await Task.Delay(15000);
// Check if any jobs are running on the agent. If they, check every 30 seconds until the job ends
await Policy
.HandleResult <TaskAgent>(x => x.AssignedRequest != null)
.WaitAndRetryAsync(40, x => TimeSpan.FromSeconds(30),
(result, timespan, content) =>
{
if (result.Result.AssignedRequest != null)
{
logger.LogInformation("{agentName} waiting for {job} to finish {poolId}", agent.Name, result.Result.AssignedRequest.JobId, abp.Key);
}
})
.ExecuteAsync(async() => await azInstance.Client.GetAgentAsync(poolId: abp.Key, agentId: agent.Id, includeAssignedRequest: true));
}));
}));
}
