Voting to scale in or scale out

tpeczek · Feb 21, 2024 · 554c92b · 554c92b
1 parent 27fd99e
commit 554c92b
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/src/RethinkDb.Azure.WebJobs.Extensions/Trigger/RethinkDbScaleMonitor.cs b/src/RethinkDb.Azure.WebJobs.Extensions/Trigger/RethinkDbScaleMonitor.cs
@@ -1,6 +1,8 @@
 using System;
+using System.Linq;
 using System.Threading.Tasks;
 using Microsoft.Azure.WebJobs.Host.Scale;
+using Microsoft.Extensions.Logging;
 
 namespace RethinkDb.Azure.WebJobs.Extensions.Trigger
 {
@@ -33,18 +35,36 @@ public Task<RethinkDbTriggerMetrics> GetMetricsAsync()
 
         public ScaleStatus GetScaleStatus(ScaleStatusContext<RethinkDbTriggerMetrics> context)
         {
-            throw new NotImplementedException();
+            return GetScaleStatus(context.WorkerCount, context.Metrics?.ToArray());
         }
 
         public ScaleStatus GetScaleStatus(ScaleStatusContext context)
         {
-            throw new NotImplementedException();
+            return GetScaleStatus(context.WorkerCount, context.Metrics?.Cast<RethinkDbTriggerMetrics>().ToArray());
         }
 
         Task<ScaleMetrics> IScaleMonitor.GetMetricsAsync()
         {
             return Task.FromResult((ScaleMetrics)_rethinkDbMetricsProvider.GetMetrics());
         }
+
+        private ScaleStatus GetScaleStatus(int workerCount, RethinkDbTriggerMetrics[] metrics)
+        {
+            ScaleStatus status = new ScaleStatus
+            {
+                Vote = ScaleVote.None
+            };
+
+            // RethinkDB change feed is not meant to be processed in paraller.
+            if (workerCount > 1)
+            {
+                status.Vote = ScaleVote.ScaleIn;
+
+                return status;
+            }
+
+            return status;
+        }
         #endregion
     }
 }