Merge branch 'develop' into dependabot/github_actions/actions/setup-node-4.3.0

Author: hjgraca

.github/workflows/build.yml

@@ -24,7 +24,7 @@ jobs:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
 
       - name: Setup .NET SDK
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: |
             6.0.x

.github/workflows/docs.yml

@@ -59,7 +59,7 @@ jobs:
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
       - name: Setup .NET 8.0
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: '8.x'
 

.github/workflows/e2e-tests.yml

@@ -41,7 +41,7 @@ jobs:
           mask-aws-account-id: true
 
       - name: Set up .NET
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: '8.x'
 
@@ -85,7 +85,7 @@ jobs:
           mask-aws-account-id: true
 
       - name: Set up .NET
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: '8.x'
 
@@ -126,7 +126,7 @@ jobs:
           mask-aws-account-id: true
 
       - name: Set up .NET
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: '8.x'
 

.github/workflows/examples-tests.yml

@@ -24,7 +24,7 @@ jobs:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
 
       - name: Setup .NET SDK
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: |
             6.0.x

.github/workflows/publish-artifacts-examples-tests.yml

@@ -30,7 +30,7 @@ jobs:
         uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
 
       - name: Setup .NET
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: |
             6.0.x
@@ -61,7 +61,7 @@ jobs:
         uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
 
       - name: Set up .NET
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: |
             6.0.x
@@ -120,7 +120,7 @@ jobs:
           path: ./packages/
 
       - name: Setup .NET
-        uses: actions/setup-dotnet@3951f0dfe7a07e2313ec93c75700083e2005cbab # 4.3.0
+        uses: actions/setup-dotnet@67a3573c9a986a3f9c594539f4ab511d57bb3ce9 # 4.3.1
         with:
           dotnet-version: |
             6.0.x

libraries/src/AWS.Lambda.Powertools.Common/Core/Constants.cs

@@ -20,6 +20,14 @@ namespace AWS.Lambda.Powertools.Common;
 /// </summary>
 internal static class Constants
 {
+    /// <summary>
+    ///     Constant for AWS_LAMBDA_INITIALIZATION_TYPE environment variable
+    ///     This is used to determine if the Lambda function is running in provisioned concurrency mode
+    ///     or not. If the value is "provisioned-concurrency", it indicates that the function is running in provisioned
+    ///     concurrency mode. Otherwise, it is running in standard mode.
+    /// </summary>
+    internal const string AWSInitializationTypeEnv = "AWS_LAMBDA_INITIALIZATION_TYPE";
+
     /// <summary>
     ///     Constant for POWERTOOLS_SERVICE_NAME environment variable
     /// </summary>

libraries/src/AWS.Lambda.Powertools.Common/Core/IPowertoolsConfigurations.cs

@@ -167,4 +167,15 @@ public interface IPowertoolsConfigurations
     /// Gets a value indicating whether Metrics are disabled.
     /// </summary>
     bool MetricsDisabled { get; }
+    
+    /// <summary>
+    /// Indicates if the current execution is a cold start.
+    /// </summary>
+    bool IsColdStart { get; }
+
+    /// <summary>
+    /// AWS Lambda initialization type.
+    /// This is set to "on-demand" for on-demand Lambda functions and "provisioned-concurrency" for provisioned concurrency.
+    /// </summary>
+    string AwsInitializationType { get; }
 }

libraries/src/AWS.Lambda.Powertools.Common/Core/ISystemWrapper.cs

@@ -59,15 +59,4 @@ public interface ISystemWrapper
     /// </summary>
     /// <param name="type"></param>
     void SetExecutionEnvironment<T>(T type);
-
-    /// <summary>
-    /// Sets console output
-    /// Useful for testing and checking the console output
-    /// <code>
-    /// var consoleOut = new StringWriter();
-    /// SystemWrapper.Instance.SetOut(consoleOut);
-    /// </code>
-    /// </summary>
-    /// <param name="writeTo">The TextWriter instance where to write to</param>
-    void SetOut(TextWriter writeTo);
 }

libraries/src/AWS.Lambda.Powertools.Common/Core/LambdaLifecycleTracker.cs

@@ -0,0 +1,66 @@
+using System;
+using System.Threading;
+
+namespace AWS.Lambda.Powertools.Common.Core;
+
+/// <summary>
+/// Tracks Lambda lifecycle state including cold starts
+/// </summary>
+internal static class LambdaLifecycleTracker
+{
+    // Static flag that's true only for the first Lambda container initialization
+    private static bool _isFirstContainer = true;
+        
+    // Store the cold start state for the current invocation
+    private static readonly AsyncLocal<bool?> CurrentInvocationColdStart = new AsyncLocal<bool?>();
+
+    private static string _lambdaInitType;
+    private static string LambdaInitType => _lambdaInitType ?? Environment.GetEnvironmentVariable(Constants.AWSInitializationTypeEnv);
+    
+    /// <summary>
+    /// Returns true if the current Lambda invocation is a cold start
+    /// </summary>
+    public static bool IsColdStart
+    {
+        get
+        {
+            if(LambdaInitType == "provisioned-concurrency")
+            {
+                // If the Lambda is provisioned concurrency, it is not a cold start
+                return false;
+            }
+            
+            // Initialize the cold start state for this invocation if not already set
+            if (!CurrentInvocationColdStart.Value.HasValue)
+            {
+                // Capture the container's cold start state for this entire invocation
+                CurrentInvocationColdStart.Value = _isFirstContainer;
+                    
+                // After detecting the first invocation, mark future ones as warm
+                if (_isFirstContainer)
+                {
+                    _isFirstContainer = false;
+                }
+            }
+                
+            // Return the cold start state for this invocation (cannot change during the invocation)
+            return CurrentInvocationColdStart.Value ?? false;
+        }
+    }
+    
+    
+
+    /// <summary>
+    /// Resets the cold start state for testing
+    /// </summary>
+    /// <param name="resetContainer">Whether to reset the container state (defaults to true)</param>
+    internal static void Reset(bool resetContainer = true)
+    {
+        if (resetContainer)
+        {
+            _isFirstContainer = true;
+        }
+        CurrentInvocationColdStart.Value = null;
+        _lambdaInitType = null;
+    }
+}

libraries/src/AWS.Lambda.Powertools.Common/Core/PowertoolsConfigurations.cs

@@ -14,6 +14,7 @@
  */
 
 using System.Globalization;
+using AWS.Lambda.Powertools.Common.Core;
 
 namespace AWS.Lambda.Powertools.Common;
 
@@ -222,4 +223,11 @@ public void SetExecutionEnvironment<T>(T type)
 
     /// <inheritdoc />
     public bool MetricsDisabled => GetEnvironmentVariableOrDefault(Constants.PowertoolsMetricsDisabledEnv, false);
+    
+    /// <inheritdoc />
+    public bool IsColdStart => LambdaLifecycleTracker.IsColdStart;
+    
+    /// <inheritdoc />
+    public string AwsInitializationType =>
+        GetEnvironmentVariable(Constants.AWSInitializationTypeEnv);
 }