From a745c8ba75f3696c82e69717c95b0ca2c46fcd10 Mon Sep 17 00:00:00 2001
From: Vera Reynolds <vreynolds@users.noreply.github.com>
Date: Fri, 16 Jul 2021 09:14:46 -0600
Subject: [PATCH] Add redis instrumentation (#66)

---
 .../Honeycomb.OpenTelemetry.csproj                        | 1 +
 src/Honeycomb.OpenTelemetry/HoneycombOptions.cs           | 8 +++++++-
 .../TracerProviderBuilderExtensions.cs                    | 5 +++++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/Honeycomb.OpenTelemetry/Honeycomb.OpenTelemetry.csproj b/src/Honeycomb.OpenTelemetry/Honeycomb.OpenTelemetry.csproj
index 68c3df3..e2de09e 100644
--- a/src/Honeycomb.OpenTelemetry/Honeycomb.OpenTelemetry.csproj
+++ b/src/Honeycomb.OpenTelemetry/Honeycomb.OpenTelemetry.csproj
@@ -35,6 +35,7 @@
     <PackageReference Include="OpenTelemetry.Instrumentation.AspNetCore" Version="1.0.0-rc6" />
     <PackageReference Include="OpenTelemetry.Instrumentation.Http" Version="1.0.0-rc6" />
     <PackageReference Include="OpenTelemetry.Instrumentation.SqlClient" Version="1.0.0-rc6" />
+    <PackageReference Include="OpenTelemetry.Instrumentation.StackExchangeRedis" Version="1.0.0-rc6" />
     <PackageReference Include="Microsoft.SourceLink.GitHub" Version="1.0.0" PrivateAssets="All"/>
   </ItemGroup>
   
diff --git a/src/Honeycomb.OpenTelemetry/HoneycombOptions.cs b/src/Honeycomb.OpenTelemetry/HoneycombOptions.cs
index 5981e27..77b0e53 100644
--- a/src/Honeycomb.OpenTelemetry/HoneycombOptions.cs
+++ b/src/Honeycomb.OpenTelemetry/HoneycombOptions.cs
@@ -1,5 +1,6 @@
 using Microsoft.Extensions.Configuration;
 using OpenTelemetry.Trace;
+using StackExchange.Redis;
 using System.Reflection;
 using System.Collections.Generic;
 
@@ -59,6 +60,11 @@ public class HoneycombOptions
         /// </summary>
         public string ServiceVersion { get; set; } = s_defaultServiceVersion;
 
+        /// <summary>
+        /// Redis connection to enable Redis instrumentation.
+        /// </summary>
+        public IConnectionMultiplexer RedisConnection { get; set; }
+
         private static Dictionary<string, string> CommandLineSwitchMap = new Dictionary<string, string>
         {
             {"--honeycomb-apikey", "apikey"},
@@ -93,4 +99,4 @@ public static HoneycombOptions FromConfiguration(IConfiguration configuration)
                 .Get<HoneycombOptions>();
         }
     }
-}
+}
\ No newline at end of file
diff --git a/src/Honeycomb.OpenTelemetry/TracerProviderBuilderExtensions.cs b/src/Honeycomb.OpenTelemetry/TracerProviderBuilderExtensions.cs
index 0a0d394..10c6c2d 100644
--- a/src/Honeycomb.OpenTelemetry/TracerProviderBuilderExtensions.cs
+++ b/src/Honeycomb.OpenTelemetry/TracerProviderBuilderExtensions.cs
@@ -65,6 +65,11 @@ public static TracerProviderBuilder UseHoneycomb(this TracerProviderBuilder buil
                 .AddHttpClientInstrumentation()
                 .AddSqlClientInstrumentation();
 
+            if (options.RedisConnection != null)
+            {
+                builder.AddRedisInstrumentation(options.RedisConnection);
+            }
+
 #if NETSTANDARD2_1
             builder.AddGrpcClientInstrumentation();
 #endif