[RpcWorker]Add environment variable FUNCTIONS_WORKER_DIRECTORY (#6201)

Author: pragnagopa

src/WebJobs.Script/Workers/Rpc/RpcWorkerProcess.cs

@@ -19,12 +19,13 @@ internal class RpcWorkerProcess : WorkerProcess
         private readonly Uri _serverUri;
         private readonly string _scriptRootPath;
         private readonly WorkerProcessArguments _workerProcessArguments;
+        private readonly string _workerDirectory;
 
         internal RpcWorkerProcess(string runtime,
                                        string workerId,
                                        string rootScriptPath,
                                        Uri serverUri,
-                                       WorkerProcessArguments workerProcessArguments,
+                                       RpcWorkerConfig rpcWorkerConfig,
                                        IScriptEventManager eventManager,
                                        IWorkerProcessFactory processFactory,
                                        IProcessRegistry processRegistry,
@@ -40,12 +41,14 @@ internal RpcWorkerProcess(string runtime,
             _workerId = workerId;
             _serverUri = serverUri;
             _scriptRootPath = rootScriptPath;
-            _workerProcessArguments = workerProcessArguments;
+            _workerProcessArguments = rpcWorkerConfig.Arguments;
+            _workerDirectory = rpcWorkerConfig.Description.WorkerDirectory;
         }
 
         internal override Process CreateWorkerProcess()
         {
             var workerContext = new RpcWorkerContext(Guid.NewGuid().ToString(), RpcWorkerConstants.DefaultMaxMessageLengthBytes, _workerId, _workerProcessArguments, _scriptRootPath, _serverUri);
+            workerContext.EnvironmentVariables.Add(WorkerConstants.FunctionsWorkerDirectorySettingName, _workerDirectory);
             return _processFactory.CreateWorkerProcess(workerContext);
         }
 

src/WebJobs.Script/Workers/Rpc/RpcWorkerProcessFactory.cs

@@ -38,7 +38,7 @@ public RpcWorkerProcessFactory(IRpcServer rpcServer,
         public IWorkerProcess Create(string workerId, string runtime, string scriptRootPath, RpcWorkerConfig workerConfig)
         {
             ILogger workerProcessLogger = _loggerFactory.CreateLogger($"Worker.rpcWorkerProcess.{runtime}.{workerId}");
-            return new RpcWorkerProcess(runtime, workerId, scriptRootPath, _rpcServer.Uri, workerConfig.Arguments, _eventManager, _workerProcessFactory, _processRegistry, workerProcessLogger, _consoleLogSource, _metricsLogger);
+            return new RpcWorkerProcess(runtime, workerId, scriptRootPath, _rpcServer.Uri, workerConfig, _eventManager, _workerProcessFactory, _processRegistry, workerProcessLogger, _consoleLogSource, _metricsLogger);
         }
     }
 }

src/WebJobs.Script/Workers/WorkerConstants.cs

@@ -20,6 +20,9 @@ public static class WorkerConstants
         public const string WorkersDirectorySectionName = "workersDirectory";
         public const string WorkerDirectorySectionName = "workerDirectory";
 
+        // Environment variables names
+        public const string FunctionsWorkerDirectorySettingName = "FUNCTIONS_WORKER_DIRECTORY";
+
         // Worker description constants
         public const string WorkerDescriptionDefaultExecutablePath = "defaultExecutablePath";
         public const string WorkerDescriptionDefaultWorkerPath = "defaultWorkerPath";

test/WebJobs.Script.Tests/Workers/Rpc/RpcWorkerProcessTests.cs

@@ -24,11 +24,12 @@ public RpcWorkerProcessTests()
             var rpcServer = new TestRpcServer();
             var languageWorkerConsoleLogSource = new Mock<IWorkerConsoleLogSource>();
             var testEnv = new TestEnvironment();
+            var testWorkerConfigs = TestHelpers.GetTestWorkerConfigs();
             _rpcWorkerProcess = new RpcWorkerProcess("node",
                 "testworkerId",
                 "testrootPath",
                 rpcServer.Uri,
-                null,
+                testWorkerConfigs.ElementAt(0),
                 _eventManager.Object,
                 workerProcessFactory.Object,
                 processRegistry.Object,