Add assign2

Author: Manikanta Nallagatla

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

@@ -1,4 +1,4 @@
-// Copyright (c) .NET Foundation. All rights reserved.
+// Copyright (c) .NET Foundation. All rights reserved.
 // Licensed under the MIT License. See License.txt in the project root for license information.
 
 using System.Threading.Tasks;
@@ -41,6 +41,37 @@ public async Task<IActionResult> Assign([FromBody] EncryptedHostAssignmentContex
 
             var assignmentContext = _startupContextProvider.SetContext(encryptedAssignmentContext);
 
+            return await assignInternal(assignmentContext);
+        }
+
+        [HttpPost]
+        [Route("admin/instance/assign2")]
+        [Authorize(Policy = PolicyNames.AdminAuthLevel)]
+        public async Task<IActionResult> Assign2([FromBody] FunctionsWorkerContainerAssignmentContext workerAssignmentContext)
+        {
+            _logger.LogDebug($"Starting container assignment for host : {Request?.Host}");
+
+            return await assignInternal(workerAssignmentContext.AssignmentContext);
+        }
+
+        [HttpGet]
+        [Route("admin/instance/info")]
+        [Authorize(Policy = PolicyNames.AdminAuthLevel)]
+        public IActionResult GetInstanceInfo()
+        {
+            return Ok(_instanceManager.GetInstanceInfo());
+        }
+
+        [HttpGet]
+        [Route("admin/instance/http-health")]
+        public IActionResult GetHttpHealthStatus()
+        {
+            // Reaching here implies that http health of the container is ok.
+            return Ok();
+        }
+
+        private async Task<IActionResult> assignInternal(HostAssignmentContext assignmentContext)
+        {
             // before starting the assignment we want to perform as much
             // up front validation on the context as possible
             string error = await _instanceManager.ValidateContext(assignmentContext);
@@ -63,21 +94,5 @@ public async Task<IActionResult> Assign([FromBody] EncryptedHostAssignmentContex
                 ? Accepted()
                 : StatusCode(StatusCodes.Status409Conflict, "Instance already assigned");
         }
-
-        [HttpGet]
-        [Route("admin/instance/info")]
-        [Authorize(Policy = PolicyNames.AdminAuthLevel)]
-        public IActionResult GetInstanceInfo()
-        {
-            return Ok(_instanceManager.GetInstanceInfo());
-        }
-
-        [HttpGet]
-        [Route("admin/instance/http-health")]
-        public IActionResult GetHttpHealthStatus()
-        {
-            // Reaching here implies that http health of the container is ok.
-            return Ok();
-        }
     }
 }

src/WebJobs.Script.WebHost/Models/FunctionsWorkerContainerAssignmentContext.cs

@@ -0,0 +1,14 @@
+// -----------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// -----------------------------------------------------------
+
+using Newtonsoft.Json;
+
+namespace Microsoft.Azure.WebJobs.Script.WebHost.Models
+{
+    public class FunctionsWorkerContainerAssignmentContext
+    {
+        [JsonProperty("assignmentContext")]
+        public HostAssignmentContext AssignmentContext { get; }
+    }
+}