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);
 }

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

@@ -27,6 +27,9 @@ namespace AWS.Lambda.Powertools.Common;
 public class SystemWrapper : ISystemWrapper
 {
     private static IPowertoolsEnvironment _powertoolsEnvironment;
+    private static bool _inTestMode = false;
+    private static TextWriter _testOutputStream;
+    private static bool _outputResetPerformed = false;
 
     /// <summary>
     ///     The instance
@@ -41,13 +44,11 @@ public SystemWrapper(IPowertoolsEnvironment powertoolsEnvironment)
         _powertoolsEnvironment = powertoolsEnvironment;
         _instance ??= this;
 
-        // Clear AWS SDK Console injected parameters StdOut and StdErr
-        var standardOutput = new StreamWriter(Console.OpenStandardOutput());
-        standardOutput.AutoFlush = true;
-        Console.SetOut(standardOutput);
-        var errordOutput = new StreamWriter(Console.OpenStandardError());
-        errordOutput.AutoFlush = true;
-        Console.SetError(errordOutput);
+        if (!_inTestMode)
+        {
+            // Clear AWS SDK Console injected parameters in production only
+            ResetConsoleOutput();
+        }
     }
 
     /// <summary>
@@ -72,7 +73,15 @@ public string GetEnvironmentVariable(string variable)
     /// <param name="value">The value.</param>
     public void Log(string value)
     {
-        Console.Write(value);
+        if (_inTestMode && _testOutputStream != null)
+        {
+            _testOutputStream.Write(value);
+        }
+        else
+        {
+            EnsureConsoleOutputOnce();
+            Console.Write(value);
+        }
     }
 
     /// <summary>
@@ -81,7 +90,15 @@ public void Log(string value)
     /// <param name="value">The value.</param>
     public void LogLine(string value)
     {
-        Console.WriteLine(value);
+        if (_inTestMode && _testOutputStream != null)
+        {
+            _testOutputStream.WriteLine(value);
+        }
+        else
+        {
+            EnsureConsoleOutputOnce();
+            Console.WriteLine(value);
+        }
     }
 
     /// <summary>
@@ -126,9 +143,20 @@ public void SetExecutionEnvironment<T>(T type)
         SetEnvironmentVariable(envName, envValue.ToString());
     }
 
-    /// <inheritdoc />
-    public void SetOut(TextWriter writeTo)
+    /// <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>
+
+    public static void SetOut(TextWriter writeTo)
     {
+        _testOutputStream = writeTo;
+        _inTestMode = true;
         Console.SetOut(writeTo);
     }
 
@@ -152,4 +180,33 @@ private string ParseAssemblyName(string assemblyName)
 
         return $"{Constants.FeatureContextIdentifier}/{assemblyName}";
     }
+    
+    private static void EnsureConsoleOutputOnce()
+    {
+        if (_outputResetPerformed) return;
+        ResetConsoleOutput();
+        _outputResetPerformed = true;
+    }
+    
+    private static void ResetConsoleOutput()
+    {
+        var standardOutput = new StreamWriter(Console.OpenStandardOutput());
+        standardOutput.AutoFlush = true;
+        Console.SetOut(standardOutput);
+        var errorOutput = new StreamWriter(Console.OpenStandardError());
+        errorOutput.AutoFlush = true;
+        Console.SetError(errorOutput);
+    }
+    
+    public static void ClearOutputResetFlag()
+    {
+        _outputResetPerformed = false;
+    }
+    
+    // For test cleanup
+    internal static void ResetTestMode()
+    {
+        _inTestMode = false;
+        _testOutputStream = null;
+    }
 }

libraries/src/AWS.Lambda.Powertools.Common/InternalsVisibleTo.cs

@@ -17,6 +17,7 @@
 
 [assembly: InternalsVisibleTo("AWS.Lambda.Powertools.Logging")]
 [assembly: InternalsVisibleTo("AWS.Lambda.Powertools.Metrics")]
+[assembly: InternalsVisibleTo("AWS.Lambda.Powertools.Tracing")]
 [assembly: InternalsVisibleTo("AWS.Lambda.Powertools.Idempotency")]
 [assembly: InternalsVisibleTo("AWS.Lambda.Powertools.Common.Tests")]
 [assembly: InternalsVisibleTo("AWS.Lambda.Powertools.Tracing.Tests")]

libraries/src/AWS.Lambda.Powertools.Logging/Internal/LoggingAspect.cs

@@ -21,6 +21,7 @@
 using System.Text.Json;
 using AspectInjector.Broker;
 using AWS.Lambda.Powertools.Common;
+using AWS.Lambda.Powertools.Common.Core;
 using AWS.Lambda.Powertools.Logging.Serializers;
 using Microsoft.Extensions.Logging;
 
@@ -34,11 +35,6 @@ namespace AWS.Lambda.Powertools.Logging.Internal;
 [Aspect(Scope.Global, Factory = typeof(LoggingAspectFactory))]
 public class LoggingAspect
 {
-    /// <summary>
-    ///     The is cold start
-    /// </summary>
-    private bool _isColdStart = true;
-
     /// <summary>
     ///     The initialize context
     /// </summary>
@@ -143,9 +139,8 @@ public void OnEntry(
             if (!_initializeContext)
                 return;
 
-            Logger.AppendKey(LoggingConstants.KeyColdStart, _isColdStart);
+            Logger.AppendKey(LoggingConstants.KeyColdStart, LambdaLifecycleTracker.IsColdStart);
 
-            _isColdStart = false;
             _initializeContext = false;
             _isContextInitialized = true;