Refresh Token for BrokerServer (#3167)

src/Runner.Common/BrokerServer.cs

@@ -23,6 +23,8 @@ public interface IBrokerServer : IRunnerService
         Task<TaskAgentMessage> GetRunnerMessageAsync(Guid? sessionId, TaskAgentStatus status, string version, string os, string architecture, bool disableUpdate, CancellationToken token);
 
         Task UpdateConnectionIfNeeded(Uri serverUri, VssCredentials credentials);
+
+        Task ForceRefreshConnection(VssCredentials credentials);
     }
 
     public sealed class BrokerServer : RunnerService, IBrokerServer
@@ -83,6 +85,11 @@ public Task UpdateConnectionIfNeeded(Uri serverUri, VssCredentials credentials)
             return Task.CompletedTask;
         }
 
+        public Task ForceRefreshConnection(VssCredentials credentials)
+        {
+            return ConnectAsync(_brokerUri, credentials);
+        }
+
         public bool ShouldRetryException(Exception ex)
         {
             if (ex is AccessDeniedException ade && ade.ErrorCode == 1)

src/Runner.Listener/MessageListener.cs

@@ -383,6 +383,7 @@ public async Task DeleteMessageAsync(TaskAgentMessage message)
         public async Task RefreshListenerTokenAsync(CancellationToken cancellationToken)
         {
             await _runnerServer.RefreshConnectionAsync(RunnerConnectionType.MessageQueue, TimeSpan.FromSeconds(60));
+            await _brokerServer.ForceRefreshConnection(_creds);
         }
 
         private TaskAgentMessage DecryptMessage(TaskAgentMessage message)

src/Runner.Listener/Runner.cs

@@ -598,6 +598,7 @@ await runServer.GetJobMessageAsync(messageRef.RunnerRequestId,
                             }
                             else if (string.Equals(message.MessageType, TaskAgentMessageTypes.ForceTokenRefresh))
                             {
+                                Trace.Info("Received ForceTokenRefreshMessage");
                                 await _listener.RefreshListenerTokenAsync(messageQueueLoopTokenSource.Token);
                             }
                             else