X-Git-Url: https://review.fuel-infra.org/gitweb?a=blobdiff_plain;f=plugins%2Fmcollective%2Faggregate%2Fsummary.rb;fp=plugins%2Fmcollective%2Faggregate%2Fsummary.rb;h=10bec513a387f4328135435cfff5c6e278e0b1ab;hb=b87d2f4e68281062df1913440ca5753ae63314a9;hp=0000000000000000000000000000000000000000;hpb=ab0ea530b8ac956091f17b104ab2311336cfc250;p=packages%2Fprecise%2Fmcollective.git

diff --git a/plugins/mcollective/aggregate/summary.rb b/plugins/mcollective/aggregate/summary.rb
new file mode 100644
index 0000000..10bec51
--- /dev/null
+++ b/plugins/mcollective/aggregate/summary.rb
@@ -0,0 +1,53 @@
+module MCollective
+  class Aggregate
+    class Summary<Base
+      # Before function is run processing
+      def startup_hook
+        @result[:value] = {}
+        @result[:type] = :collection
+
+        # set default aggregate_format if it is undefined
+        @aggregate_format = :calculate unless @aggregate_format
+      end
+
+      # Increments the value field if value has been seen before
+      # Else create a new value field
+      def process_result(value, reply)
+        unless value.nil?
+          if value.is_a? Array
+            value.map{|val| add_value(val)}
+          else
+            add_value(value)
+          end
+        end
+      end
+
+      def add_value(value)
+        if @result[:value].keys.include?(value)
+          @result[:value][value] += 1
+        else
+          @result[:value][value] = 1
+        end
+      end
+
+      def summarize
+        if @aggregate_format == :calculate
+          max_key_length = @result[:value].keys.map do |k|
+
+            # Response values retain their types. Here we check
+            # if the response is a string and turn it into a string
+            # if it isn't one.
+            if k.respond_to?(:length)
+              k.length
+            elsif k.respond_to?(:to_s)
+              k.to_s.length
+            end
+          end.max
+          @aggregate_format = "%#{max_key_length}s = %s"
+        end
+
+        super
+      end
+    end
+  end
+end