Mesh changes

Author: ahmelsayed

src/WebJobs.Script.WebHost/ContainerManagement/LinuxContainerInitializationHostService.cs

@@ -68,7 +68,7 @@ private async Task ApplyContextIfPresent()
                 var encryptedAssignmentContext = JsonConvert.DeserializeObject<EncryptedHostAssignmentContext>(startContext);
                 var containerKey = _environment.GetEnvironmentVariable(EnvironmentSettingNames.ContainerEncryptionKey);
                 var assignmentContext = encryptedAssignmentContext.Decrypt(containerKey);
-                if (_instanceManager.StartAssignment(assignmentContext))
+                if (_instanceManager.StartAssignment(assignmentContext, false))
                 {
                     _logger.LogInformation("Start assign HostAssignmentContext success");
                 }

src/WebJobs.Script.WebHost/Controllers/InstanceController.cs

@@ -37,27 +37,29 @@ public async Task<IActionResult> Assign([FromBody] EncryptedHostAssignmentContex
         {
             _logger.LogDebug($"Starting container assignment for host : {Request?.Host}. ContextLength is: {encryptedAssignmentContext.EncryptedContext?.Length}");
             var containerKey = _environment.GetEnvironmentVariable(EnvironmentSettingNames.ContainerEncryptionKey);
-            var assignmentContext = encryptedAssignmentContext.Decrypt(containerKey);
+            var assignmentContext = encryptedAssignmentContext.IsWarmup
+                ? null
+                : encryptedAssignmentContext.Decrypt(containerKey);
 
             // before starting the assignment we want to perform as much
             // up front validation on the context as possible
-            string error = await _instanceManager.ValidateContext(assignmentContext);
+            string error = await _instanceManager.ValidateContext(assignmentContext, encryptedAssignmentContext.IsWarmup);
             if (error != null)
             {
                 return StatusCode(StatusCodes.Status400BadRequest, error);
             }
 
             // Wait for Sidecar specialization to complete before returning ok.
             // This shouldn't take too long so ok to do this sequentially.
-            error = await _instanceManager.SpecializeMSISidecar(assignmentContext);
+            error = await _instanceManager.SpecializeMSISidecar(assignmentContext, encryptedAssignmentContext.IsWarmup);
             if (error != null)
             {
                 return StatusCode(StatusCodes.Status500InternalServerError, error);
             }
 
-            var result = _instanceManager.StartAssignment(assignmentContext);
+            var result = _instanceManager.StartAssignment(assignmentContext, encryptedAssignmentContext.IsWarmup);
 
-            return result
+            return result || encryptedAssignmentContext.IsWarmup
                 ? Accepted()
                 : StatusCode(StatusCodes.Status409Conflict, "Instance already assigned");
         }

src/WebJobs.Script.WebHost/Management/IInstanceManager.cs

@@ -11,10 +11,10 @@ public interface IInstanceManager
     {
         IDictionary<string, string> GetInstanceInfo();
 
-        Task<string> ValidateContext(HostAssignmentContext assignmentContext);
+        Task<string> ValidateContext(HostAssignmentContext assignmentContext, bool isWarmup);
 
-        bool StartAssignment(HostAssignmentContext assignmentContext);
+        bool StartAssignment(HostAssignmentContext assignmentContext, bool isWarmup);
 
-        Task<string> SpecializeMSISidecar(HostAssignmentContext assignmentContext);
+        Task<string> SpecializeMSISidecar(HostAssignmentContext assignmentContext, bool isWarmup);
     }
 }