Add OpenTelemetry

Add OpenTelemetry to the skill.
martincostello · Jan 24, 2024 · cd34464 · cd34464
1 parent ec94ae3
commit cd34464
Show file tree

Hide file tree

Showing 20 changed files with 120 additions and 42 deletions.
diff --git a/.github/dependabot.yml b/.github/dependabot.yml
@@ -11,6 +11,9 @@ updates:
 - package-ecosystem: nuget
   directory: "/"
   groups:
+    opentelemetry:
+      patterns:
+        - OpenTelemetry*
     polly:
       patterns:
         - Polly*

diff --git a/.packages/OpenTelemetry.Instrumentation.AWSLambda.1.2.0-beta.1.168.nupkg b/.packages/OpenTelemetry.Instrumentation.AWSLambda.1.2.0-beta.1.168.nupkg
diff --git a/.packages/OpenTelemetry.Instrumentation.AWSLambda.1.2.0-beta.1.168.snupkg b/.packages/OpenTelemetry.Instrumentation.AWSLambda.1.2.0-beta.1.168.snupkg
diff --git a/Directory.Packages.props b/Directory.Packages.props
@@ -26,6 +26,11 @@
     <PackageVersion Include="Microsoft.Extensions.Telemetry" Version="8.1.0" />
     <PackageVersion Include="Microsoft.NET.Test.Sdk" Version="17.8.0" />
     <PackageVersion Include="Newtonsoft.Json" Version="13.0.3" />
+    <PackageVersion Include="OpenTelemetry" Version="1.7.0" />
+    <PackageVersion Include="OpenTelemetry.Exporter.OpenTelemetryProtocol" Version="1.7.0" />
+    <PackageVersion Include="OpenTelemetry.Extensions.Hosting" Version="1.7.0" />
+    <PackageVersion Include="OpenTelemetry.Instrumentation.AWSLambda" Version="1.2.0-beta.1.168" />
+    <PackageVersion Include="OpenTelemetry.Instrumentation.Http" Version="1.7.0" />
     <PackageVersion Include="Polly.Core" Version="$(PollyVersion)" />
     <PackageVersion Include="Polly.Extensions" Version="$(PollyVersion)" />
     <PackageVersion Include="Polly.RateLimiting" Version="$(PollyVersion)" />

diff --git a/NuGet.config b/NuGet.config
@@ -2,9 +2,13 @@
 <configuration>
   <packageSources>
     <clear />
+    <add key="Local" value=".packages" />
     <add key="NuGet" value="https://api.nuget.org/v3/index.json" />
   </packageSources>
   <packageSourceMapping>
+    <packageSource key="Local">
+      <package pattern="OpenTelemetry.Instrumentation.AWSLambda" />
+    </packageSource>
     <packageSource key="NuGet">
       <package pattern="*" />
     </packageSource>

diff --git a/src/LondonTravel.Skill/AlexaFunction.cs b/src/LondonTravel.Skill/AlexaFunction.cs
@@ -2,13 +2,18 @@
 // Licensed under the Apache 2.0 license. See the LICENSE file in the project root for full license information.
 
 using System.Diagnostics.CodeAnalysis;
+using Amazon.Lambda.Core;
 using MartinCostello.LondonTravel.Skill.Extensions;
 using MartinCostello.LondonTravel.Skill.Intents;
 using MartinCostello.LondonTravel.Skill.Models;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
+using OpenTelemetry.Instrumentation.AWSLambda;
+using OpenTelemetry.Metrics;
+using OpenTelemetry.Resources;
+using OpenTelemetry.Trace;
 
 namespace MartinCostello.LondonTravel.Skill;
 
@@ -55,21 +60,15 @@ public async virtual ValueTask DisposeAsync()
     /// Handles a request to the skill as an asynchronous operation.
     /// </summary>
     /// <param name="request">The skill request.</param>
+    /// <param name="context">The Lamda request context.</param>
     /// <returns>
     /// A <see cref="Task{TResult}"/> representing the asynchronous operation to get the skill's response.
     /// </returns>
-    public async Task<SkillResponse> HandlerAsync(SkillRequest request)
+    public async Task<SkillResponse> HandlerAsync(SkillRequest request, ILambdaContext context)
     {
         EnsureInitialized();
-
-        var handler = _serviceProvider.GetRequiredService<FunctionHandler>();
-        var logger = _serviceProvider.GetRequiredService<ILogger<AlexaFunction>>();
-
-        using var activity = SkillTelemetry.ActivitySource.StartActivity("Skill Request");
-
-        Log.InvokingSkillRequest(logger, request.Request.Type);
-
-        return await handler.HandleAsync(request);
+        var tracerProvider = _serviceProvider.GetRequiredService<TracerProvider>();
+        return await AWSLambdaWrapper.TraceAsync(tracerProvider, HandlerCoreAsync, request, context);
     }
 
     /// <summary>
@@ -132,6 +131,21 @@ protected virtual void ConfigureServices(IServiceCollection services)
         services.AddTransient<CommuteIntent>();
         services.AddTransient<DisruptionIntent>();
         services.AddTransient<StatusIntent>();
+
+        services.AddSingleton((_) => SkillTelemetry.ActivitySource);
+        services.AddOpenTelemetry()
+                .ConfigureResource((builder) => builder.AddService(SkillTelemetry.ServiceName, serviceVersion: SkillTelemetry.ServiceVersion))
+                .WithTracing((builder) =>
+                {
+                    builder.AddHttpClientInstrumentation()
+                           .AddSource(SkillTelemetry.ServiceName);
+
+                    if (IsRunningInAwsLambda())
+                    {
+                        builder.AddAWSLambdaConfigurations()
+                               .AddOtlpExporter();
+                    }
+                });
     }
 
     protected virtual void Dispose(bool disposing)
@@ -147,12 +161,22 @@ protected virtual void Dispose(bool disposing)
         }
     }
 
-    /// <summary>
-    /// Creates the <see cref="ServiceProvider"/> to use.
-    /// </summary>
-    /// <returns>
-    /// The <see cref="ServiceProvider"/> to use.
-    /// </returns>
+    private static bool IsRunningInAwsLambda()
+        => Environment.GetEnvironmentVariable("AWS_LAMBDA_FUNCTION_NAME") is { Length: > 0 } &&
+           Environment.GetEnvironmentVariable("AWS_REGION") is { Length: > 0 };
+
+    private async Task<SkillResponse> HandlerCoreAsync(SkillRequest request, ILambdaContext context)
+    {
+        var handler = _serviceProvider!.GetRequiredService<FunctionHandler>();
+        var logger = _serviceProvider!.GetRequiredService<ILogger<AlexaFunction>>();
+
+        using var activity = SkillTelemetry.ActivitySource.StartActivity("Skill Request");
+
+        Log.InvokingSkillRequest(logger, request.Request.Type);
+
+        return await handler.HandleAsync(request);
+    }
+
     private ServiceProvider CreateServiceProvider()
     {
         var services = new ServiceCollection();

diff --git a/src/LondonTravel.Skill/LondonTravel.Skill.csproj b/src/LondonTravel.Skill/LondonTravel.Skill.csproj
@@ -28,6 +28,11 @@
     <PackageReference Include="Microsoft.Extensions.Http.Resilience" />
     <PackageReference Include="Microsoft.Extensions.Logging.Console" />
     <PackageReference Include="Microsoft.Extensions.Telemetry" />
+    <PackageReference Include="OpenTelemetry" />
+    <PackageReference Include="OpenTelemetry.Exporter.OpenTelemetryProtocol" />
+    <PackageReference Include="OpenTelemetry.Extensions.Hosting" />
+    <PackageReference Include="OpenTelemetry.Instrumentation.AWSLambda" />
+    <PackageReference Include="OpenTelemetry.Instrumentation.Http" />
     <PackageReference Include="Polly.Core" />
     <PackageReference Include="Polly.Extensions" />
     <PackageReference Include="Polly.RateLimiting" />

diff --git a/src/LondonTravel.Skill/Models/Context.cs b/src/LondonTravel.Skill/Models/Context.cs
@@ -5,7 +5,9 @@
 
 namespace MartinCostello.LondonTravel.Skill.Models;
 
+#pragma warning disable CA1724
 public sealed class Context
+#pragma warning restore CA1724
 {
     [JsonPropertyName("System")]
     public AlexaSystem System { get; set; } = default!;

diff --git a/test/LondonTravel.Skill.Tests/AlexaFunctionTests.cs b/test/LondonTravel.Skill.Tests/AlexaFunctionTests.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Martin Costello, 2017. All rights reserved.
 // Licensed under the Apache 2.0 license. See the LICENSE file in the project root for full license information.
 
+using Amazon.Lambda.TestUtilities;
 using MartinCostello.LondonTravel.Skill.Models;
 
 namespace MartinCostello.LondonTravel.Skill;
@@ -12,13 +13,14 @@ public async Task Cannot_Invoke_Function_If_Application_Id_Incorrect()
     {
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
+        TestLambdaContext context = new();
 
         SkillRequest request = CreateIntentRequest("AMAZON.HelpIntent");
         request.Session.Application.ApplicationId = "not-my-skill-id";
 
         // Act
-        InvalidOperationException exception = await Assert.ThrowsAsync<InvalidOperationException>(
-            () => function.HandlerAsync(request));
+        var exception = await Assert.ThrowsAsync<InvalidOperationException>(
+            () => function.HandlerAsync(request, context));
 
         // Assert
         exception.Message.ShouldBe("Request application Id 'not-my-skill-id' and configured skill Id 'my-skill-id' mismatch.");
@@ -34,12 +36,13 @@ public async Task Can_Invoke_Function_If_Locale_Is_Invalid(string? locale)
     {
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
+        TestLambdaContext context = new();
 
         SkillRequest request = CreateIntentRequest("AMAZON.HelpIntent");
         request.Request.Locale = locale!;
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual, shouldEndSession: false);
@@ -53,6 +56,7 @@ public async Task Cannot_Invoke_Function_With_System_Failure()
     {
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
+        TestLambdaContext context = new();
 
         var error = new Request()
         {
@@ -70,7 +74,7 @@ public async Task Cannot_Invoke_Function_With_System_Failure()
         var request = CreateRequest("System.ExceptionEncountered", error);
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual);

diff --git a/test/LondonTravel.Skill.Tests/CancelTests.cs b/test/LondonTravel.Skill.Tests/CancelTests.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Martin Costello, 2017. All rights reserved.
 // Licensed under the Apache 2.0 license. See the LICENSE file in the project root for full license information.
 
+using Amazon.Lambda.TestUtilities;
 using MartinCostello.LondonTravel.Skill.Models;
 
 namespace MartinCostello.LondonTravel.Skill;
@@ -12,11 +13,12 @@ public async Task Can_Invoke_Function()
     {
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
+        TestLambdaContext context = new();
 
         SkillRequest request = CreateIntentRequest("AMAZON.CancelIntent");
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual);

diff --git a/test/LondonTravel.Skill.Tests/CommuteTests.cs b/test/LondonTravel.Skill.Tests/CommuteTests.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Martin Costello, 2017. All rights reserved.
 // Licensed under the Apache 2.0 license. See the LICENSE file in the project root for full license information.
 
+using Amazon.Lambda.TestUtilities;
 using JustEat.HttpClientInterception;
 using MartinCostello.LondonTravel.Skill.Models;
 
@@ -14,9 +15,10 @@ public async Task Can_Invoke_Function_When_The_Skill_Is_Not_Linked()
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequestWithToken(accessToken: null);
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual);
@@ -41,9 +43,10 @@ public async Task Can_Invoke_Function_When_The_Skill_Token_Is_Invalid()
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequestWithToken(accessToken: "invalid-access-token");
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual);
@@ -66,9 +69,10 @@ public async Task Can_Invoke_Function_When_The_Skill_Api_Fails()
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequestWithToken(accessToken: "random-access-token");
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual);
@@ -92,9 +96,10 @@ public async Task Can_Invoke_Function_When_The_Skill_Is_Linked_And_Has_No_Favori
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequestWithToken(accessToken: "token-for-no-favorites");
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         AssertResponse(
@@ -112,9 +117,10 @@ public async Task Can_Invoke_Function_When_The_Skill_Is_Linked_And_Has_One_Favor
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequestWithToken(accessToken: "token-for-one-favorite");
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         AssertResponse(
@@ -132,9 +138,10 @@ public async Task Can_Invoke_Function_When_The_Skill_Is_Linked_And_Has_Two_Favor
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequestWithToken(accessToken: "token-for-two-favorites");
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         AssertResponse(

diff --git a/test/LondonTravel.Skill.Tests/DisruptionTests.cs b/test/LondonTravel.Skill.Tests/DisruptionTests.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Martin Costello, 2017. All rights reserved.
 // Licensed under the Apache 2.0 license. See the LICENSE file in the project root for full license information.
 
+using Amazon.Lambda.TestUtilities;
 using JustEat.HttpClientInterception;
 using MartinCostello.LondonTravel.Skill.Models;
 
@@ -16,9 +17,10 @@ public async Task Can_Invoke_Function_When_There_Are_No_Disruptions()
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequest();
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         AssertResponse(
@@ -35,9 +37,10 @@ public async Task Can_Invoke_Function_When_There_Is_One_Disruption()
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequest();
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         AssertResponse(
@@ -54,9 +57,10 @@ public async Task Can_Invoke_Function_When_There_Are_Multiple_Disruptions()
 
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequest();
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         AssertResponse(
@@ -71,9 +75,10 @@ public async Task Can_Invoke_Function_When_The_Api_Fails()
         // Arrange
         AlexaFunction function = await CreateFunctionAsync();
         SkillRequest request = CreateIntentRequest();
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual);

diff --git a/test/LondonTravel.Skill.Tests/HelpTests.cs b/test/LondonTravel.Skill.Tests/HelpTests.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Martin Costello, 2017. All rights reserved.
 // Licensed under the Apache 2.0 license. See the LICENSE file in the project root for full license information.
 
+using Amazon.Lambda.TestUtilities;
 using MartinCostello.LondonTravel.Skill.Models;
 
 namespace MartinCostello.LondonTravel.Skill;
@@ -14,9 +15,10 @@ public async Task Can_Invoke_Function()
         AlexaFunction function = await CreateFunctionAsync();
 
         SkillRequest request = CreateIntentRequest("AMAZON.HelpIntent");
+        TestLambdaContext context = new();
 
         // Act
-        SkillResponse actual = await function.HandlerAsync(request);
+        SkillResponse actual = await function.HandlerAsync(request, context);
 
         // Assert
         ResponseBody response = AssertResponse(actual, shouldEndSession: false);