chore: yggdrasil (#224)

Swaps out the core logic for evaluating features with the Yggdrasil engine
Unleash · Oct 28, 2024 · 07e45bc · 07e45bc
1 parent 2288065
commit 07e45bc
Show file tree

Hide file tree

Showing 37 changed files with 314 additions and 741 deletions.
diff --git a/src/Unleash/ClientFactory/IUnleashClientFactory.cs b/src/Unleash/ClientFactory/IUnleashClientFactory.cs
@@ -1,6 +1,5 @@
-using System;
-using System.Threading.Tasks;
-using Unleash.Strategies;
+using System.Threading.Tasks;
+using Yggdrasil;
 
 namespace Unleash.ClientFactory
 {

diff --git a/src/Unleash/ClientFactory/UnleashClientFactory.cs b/src/Unleash/ClientFactory/UnleashClientFactory.cs
@@ -1,6 +1,6 @@
 using System.Threading;
 using System.Threading.Tasks;
-using Unleash.Strategies;
+using Yggdrasil;
 
 namespace Unleash.ClientFactory
 {

diff --git a/src/Unleash/Communication/FetchTogglesResult.cs b/src/Unleash/Communication/FetchTogglesResult.cs
@@ -1,11 +1,9 @@
-using Unleash.Internal;
-
-namespace Unleash.Communication
+namespace Unleash.Communication
 {
     internal class FetchTogglesResult
     {
-        public ToggleCollection ToggleCollection { get; set; }
         public bool HasChanged { get; set; }
         public string Etag { get; set; }
+        public string State { get; set; }
     }
 }
diff --git a/src/Unleash/Communication/IUnleashApiClient.cs b/src/Unleash/Communication/IUnleashApiClient.cs
@@ -8,6 +8,7 @@ internal interface IUnleashApiClient
     {
         Task<FetchTogglesResult> FetchToggles(string etag, CancellationToken cancellationToken, bool throwOnFail = false);
         Task<bool> RegisterClient(ClientRegistration registration, CancellationToken cancellationToken);
-        Task<bool> SendMetrics(ThreadSafeMetricsBucket metrics, CancellationToken cancellationToken);
+        // TODO: Can be simplified to `using Yggdrasil;` once MetricsBucket is dropped from Unleash.Metrics
+        Task<bool> SendMetrics(Yggdrasil.MetricsBucket metrics, CancellationToken cancellationToken);
     }
 }
diff --git a/src/Unleash/Communication/UnleashApiClient.cs b/src/Unleash/Communication/UnleashApiClient.cs
@@ -157,14 +157,12 @@ private async Task<FetchTogglesResult> HandleSuccessResponse(HttpResponseMessage
                 {
                     HasChanged = false,
                     Etag = newEtag,
-                    ToggleCollection = null,
                 };
             }
 
-            var stream = await response.Content.ReadAsStreamAsync().ConfigureAwait(false);
-            var toggleCollection = jsonSerializer.Deserialize<ToggleCollection>(stream);
+            var content = await response.Content.ReadAsStringAsync().ConfigureAwait(false);
 
-            if (toggleCollection == null)
+            if (string.IsNullOrEmpty(content))
             {
                 return new FetchTogglesResult
                 {
@@ -177,7 +175,7 @@ private async Task<FetchTogglesResult> HandleSuccessResponse(HttpResponseMessage
             {
                 HasChanged = true,
                 Etag = newEtag,
-                ToggleCollection = toggleCollection
+                State = content
             };
         }
 
@@ -212,7 +210,7 @@ public async Task<bool> RegisterClient(ClientRegistration registration, Cancella
             }
         }
 
-        public async Task<bool> SendMetrics(ThreadSafeMetricsBucket metrics, CancellationToken cancellationToken)
+        public async Task<bool> SendMetrics(Yggdrasil.MetricsBucket metrics, CancellationToken cancellationToken)
         {
             if (metricsRequestsToSkip > metricsRequestsSkipped)
             {
@@ -226,16 +224,13 @@ public async Task<bool> SendMetrics(ThreadSafeMetricsBucket metrics, Cancellatio
 
             var memoryStream = new MemoryStream();
 
-            using (metrics.StopCollectingMetrics(out var bucket))
+            jsonSerializer.Serialize(memoryStream, new ClientMetrics
             {
-                jsonSerializer.Serialize(memoryStream, new ClientMetrics
-                {
-                    AppName = clientRequestHeaders.AppName,
-                    InstanceId = clientRequestHeaders.InstanceTag,
-                    Bucket = bucket
-                });
-                memoryStream.Position = 0;
-            }
+                AppName = clientRequestHeaders.AppName,
+                InstanceId = clientRequestHeaders.InstanceTag,
+                Bucket = metrics
+            });
+            memoryStream.Position = 0;
 
             const int bufferSize = 1024 * 4;