From 0b8e888ba46db6a45032f2314f55bf11ca3fab72 Mon Sep 17 00:00:00 2001
From: Anatoli Beliaev <anatolib@microsoft.com>
Date: Mon, 23 Mar 2020 23:48:00 -0700
Subject: [PATCH] Handle FunctionEnvironmentReloadRequest

---
 src/RequestProcessor.cs                    | 13 +++++++++++++
 src/resources/PowerShellWorkerStrings.resx |  3 +++
 2 files changed, 16 insertions(+)

diff --git a/src/RequestProcessor.cs b/src/RequestProcessor.cs
index 296bf21b..3f4b51b2 100644
--- a/src/RequestProcessor.cs
+++ b/src/RequestProcessor.cs
@@ -344,6 +344,19 @@ internal StreamingMessage ProcessInvocationCancelRequest(StreamingMessage reques
 
         internal StreamingMessage ProcessFunctionEnvironmentReloadRequest(StreamingMessage request)
         {
+            var stopwatch = new Stopwatch();
+            stopwatch.Start();
+
+            var environmentReloadRequest = request.FunctionEnvironmentReloadRequest;
+            foreach (var (name, value) in environmentReloadRequest.EnvironmentVariables)
+            {
+                Environment.SetEnvironmentVariable(name, value);
+            }
+
+            var rpcLogger = new RpcLogger(_msgStream);
+            rpcLogger.SetContext(request.RequestId, null);
+            rpcLogger.Log(isUserOnlyLog: false, LogLevel.Trace, string.Format(PowerShellWorkerStrings.EnvironmentReloadCompleted, stopwatch.ElapsedMilliseconds));
+
             StreamingMessage response = NewStreamingMessageTemplate(
                 request.RequestId,
                 StreamingMessage.ContentOneofCase.FunctionEnvironmentReloadResponse,
diff --git a/src/resources/PowerShellWorkerStrings.resx b/src/resources/PowerShellWorkerStrings.resx
index dc6fd2a2..dcf9a376 100644
--- a/src/resources/PowerShellWorkerStrings.resx
+++ b/src/resources/PowerShellWorkerStrings.resx
@@ -313,4 +313,7 @@
   <data name="PowerShellVersion" xml:space="preserve">
     <value>PowerShell version: '{0}'.</value>
   </data>
+  <data name="EnvironmentReloadCompleted" xml:space="preserve">
+    <value>Environment reload completed in {0} ms.</value>
+  </data>
 </root>
\ No newline at end of file