inline SnapstartHelperLambdaRequest

Author: ppittle

Libraries/src/Amazon.Lambda.AspNetCoreServer.Hosting/Amazon.Lambda.AspNetCoreServer.Hosting.csproj

@@ -21,6 +21,12 @@
     <None Include="README.md" Pack="true" PackagePath="\" />		
   </ItemGroup>
 
+  <PropertyGroup>
+    <!--This assembly contains some internal methods that need to be accessed by the Amazon.Lambda.Annotations.SourceGenerator assembly.
+      Both these assemblies need to be strongly signed for the InternalsVisibleTo attribute to take effect.-->
+    <AssemblyOriginatorKeyFile>..\..\..\buildtools\public.snk</AssemblyOriginatorKeyFile>
+    <SignAssembly>true</SignAssembly>
+  </PropertyGroup>
 
    <ItemGroup>
     <ProjectReference Include="..\Amazon.Lambda.Core\Amazon.Lambda.Core.csproj" />	  

Libraries/src/Amazon.Lambda.AspNetCoreServer.Hosting/Internal/LambdaRuntimeSupportServer.cs

@@ -1,10 +1,7 @@
 using System.Diagnostics.CodeAnalysis;
-using System.Text;
-using System.Text.Json;
 using Amazon.Lambda.AspNetCoreServer.Internal;
 using Amazon.Lambda.Core;
 using Amazon.Lambda.RuntimeSupport;
-using Amazon.Lambda.RuntimeSupport.Helpers;
 using Microsoft.AspNetCore.Hosting.Server;
 using Microsoft.Extensions.DependencyInjection;
 

Libraries/src/Amazon.Lambda.AspNetCoreServer.Hosting/Internal/LambdaSnapstartExecuteRequestsBeforeSnapshotHelper.cs

@@ -3,6 +3,7 @@
 
 using System.Diagnostics.CodeAnalysis;
 using System.Net;
+using System.Text;
 using System.Text.Json;
 using Amazon.Lambda.APIGatewayEvents;
 using Amazon.Lambda.RuntimeSupport;
@@ -61,7 +62,6 @@ public void RegisterInitializerRequests(HandlerWrapper handlerWrapper)
             {
                 var json = JsonSerializer.Serialize(req);
 
-                // TODO - inline
                 await SnapstartHelperLambdaRequests.ExecuteSnapstartInitRequests(json, times: 5, handlerWrapper);
             }
         });
@@ -88,6 +88,46 @@ internal async Task Execute(HttpClient client)
         }
     }
 
+    private static class SnapstartHelperLambdaRequests
+    {
+        private static InternalLogger _logger = InternalLogger.GetDefaultLogger();
+
+        private static readonly RuntimeApiHeaders _fakeRuntimeApiHeaders = new(new Dictionary<string, IEnumerable<string>>
+        {
+            { RuntimeApiHeaders.HeaderAwsRequestId, new List<string>() },
+            { RuntimeApiHeaders.HeaderTraceId, new List<string>() },
+            { RuntimeApiHeaders.HeaderClientContext, new List<string>() },
+            { RuntimeApiHeaders.HeaderCognitoIdentity, new List<string>() },
+            { RuntimeApiHeaders.HeaderDeadlineMs, new List<string>() },
+            { RuntimeApiHeaders.HeaderInvokedFunctionArn, new List<string>() },
+        });
+
+        public static async Task ExecuteSnapstartInitRequests(string jsonRequest, int times, HandlerWrapper handlerWrapper)
+        {
+            var dummyRequest = new InvocationRequest
+            {
+                InputStream = new MemoryStream(Encoding.UTF8.GetBytes(jsonRequest)),
+                LambdaContext = new LambdaContext(
+                    _fakeRuntimeApiHeaders,
+                    new LambdaEnvironment(),
+                    new SimpleLoggerWriter())
+            };
+
+            for (var i = 0; i < times; i++)
+            {
+                try
+                {
+                    _ = await handlerWrapper.Handler.Invoke(dummyRequest);
+                }
+                catch (Exception e)
+                {
+                    Console.WriteLine("StartAsync: " + e.Message + e.StackTrace);
+                    _logger.LogError(e, "StartAsync: Custom Warmup Failure: " + e.Message + e.StackTrace);
+                }
+            }
+        }
+    }
+
     private class LambdaSnapstartInitializerHttpMessageHandler : HttpMessageHandler
     {
         private LambdaEventSource _lambdaEventSource;

Libraries/src/Amazon.Lambda.RuntimeSupport/Amazon.Lambda.RuntimeSupport.csproj

@@ -16,6 +16,19 @@
 	<LangVersion>9.0</LangVersion>
   </PropertyGroup>
 
+  <PropertyGroup>
+    <!--This assembly contains some internal methods that need to be accessed by the Amazon.Lambda.Annotations.SourceGenerator assembly.
+    Both these assemblies need to be strongly signed for the InternalsVisibleTo attribute to take effect.-->
+    <AssemblyOriginatorKeyFile>..\..\..\buildtools\public.snk</AssemblyOriginatorKeyFile>
+    <SignAssembly>true</SignAssembly>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <AssemblyAttribute Include="System.Runtime.CompilerServices.InternalsVisibleToAttribute">
+      <_Parameter1>Amazon.Lambda.AspNetCoreServer.Hosting, PublicKey="0024000004800000940000000602000000240000525341310004000001000100db5f59f098d27276c7833875a6263a3cc74ab17ba9a9df0b52aedbe7252745db7274d5271fd79c1f08f668ecfa8eaab5626fa76adc811d3c8fc55859b0d09d3bc0a84eecd0ba891f2b8a2fc55141cdcc37c2053d53491e650a479967c3622762977900eddbf1252ed08a2413f00a28f3a0752a81203f03ccb7f684db373518b4"</_Parameter1>
+    </AssemblyAttribute>
+  </ItemGroup>
+
   <PropertyGroup Condition=" '$(ExecutableOutputType)'=='true' ">
     <OutputType>Exe</OutputType>
   </PropertyGroup>