feat: add epoch_time_distribution

to distribution data
shaojunda · Apr 30, 2020 · 2c40104 · 2c40104
1 parent 3a175b9
commit 2c40104
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/app/models/distribution_data.rb b/app/models/distribution_data.rb
@@ -1,5 +1,5 @@
 class DistributionData
-  VALID_INDICATORS = %w(address_balance_distribution block_time_distribution)
+  VALID_INDICATORS = %w(address_balance_distribution block_time_distribution epoch_time_distribution)
 
   def id
     Time.current.to_i
@@ -12,4 +12,8 @@ def address_balance_distribution
   def block_time_distribution
     EpochStatistic.order(epoch_number: :desc).first&.block_time_distribution || []
   end
+
+  def epoch_time_distribution
+    EpochStatistic.order(epoch_number: :desc).first&.epoch_time_distribution || []
+  end
 end
diff --git a/app/serializers/distribution_data_serializer.rb b/app/serializers/distribution_data_serializer.rb
@@ -12,4 +12,10 @@ class DistributionDataSerializer
   } do |object|
     object.block_time_distribution.map { |distribution| distribution.map(&:to_s) }
   end
+
+  attribute :epoch_time_distribution, if: Proc.new { |_record, params|
+    params && params[:indicator].include?("epoch_time_distribution")
+  } do |object|
+    object.epoch_time_distribution.map { |distribution| distribution.map(&:to_s) }
+  end
 end