From 15a113a66761445137d1d1c715f32b061e938cc9 Mon Sep 17 00:00:00 2001
From: Damien Filliat <d.filliat@criteo.com>
Date: Mon, 25 Mar 2024 10:45:22 +0100
Subject: [PATCH] Add consul_service_weight metric.

- This metrics will give total weight for a service,
- It will have the same labels as existing consul_service_count metric.
---
 samples/metrics.erb | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/samples/metrics.erb b/samples/metrics.erb
index 63525b0..c6631bc 100644
--- a/samples/metrics.erb
+++ b/samples/metrics.erb
@@ -58,15 +58,31 @@ consul_members_count{serf="lan",status="<%= k %>"} <%= v %><%
             metas: metas,
             service_name: service_name,
             state: {
-              'passing'  => 0,
-              'warning'  => 0,
-              'critical' => 0,
+              'passing' =>  {
+                instances: 0,
+                weight: 0
+              },
+              'warning' =>  {
+                instances: 0,
+                weight: 0
+              },
+              'critical' =>  {
+                instances: 0,
+                weight: 0
+              }
             }
           }
           backends[key] = back
         end
         state = snode.status
-        back[:state][state] += 1
+        # Increment instance number.
+        back[:state][state][:instances] += 1
+        # Increment total weight regarding current node's state.
+        if state == 'passing'
+          back[:state][state][:weight] += snode['Service']['Weights']['Passing']
+        elsif state == 'warning'
+          back[:state][state][:weight] += snode['Service']['Weights']['Warning']
+        end
       end
     end
   end
@@ -118,7 +134,8 @@ end
       meta_string+=",#{k}=\"#{escape_meta(v)}\""
     end
     service_info[:state].each_pair do |state_name, state_count|
-%>consul_service_count{service="<%= service_name %>",state="<%= state_name %>"<%= meta_string %>} <%= state_count %>
+%>consul_service_count{service="<%= service_name %>",state="<%= state_name %>"<%= meta_string %>} <%= state_count[:instances] %>
+consul_service_weight{service="<%= service_name %>",state="<%= state_name %>"<%= meta_string %>} <%= state_count[:weight] %>
 <%
     end
   end