Fix :bulk_metric_registry_change event emitting

santiment · Dec 20, 2024 · 4ff88b5 · 4ff88b5
1 parent 431a4e3
commit 4ff88b5
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 11 deletions.
diff --git a/lib/sanbase/event_bus/metric_registry_subscriber.ex b/lib/sanbase/event_bus/metric_registry_subscriber.ex
@@ -126,7 +126,8 @@ defmodule Sanbase.EventBus.MetricRegistrySubscriber do
     state
   end
 
-  defp handle_event(_event, event_shadow, state) do
+  defp handle_event(event, event_shadow, state) do
+    Logger.warning("Unrecognized event #{inspect(event)} received in #{__MODULE__}")
     EventBus.mark_as_completed({__MODULE__, event_shadow})
     state
   end

diff --git a/lib/sanbase/metric/registry/event_emitter.ex b/lib/sanbase/metric/registry/event_emitter.ex
@@ -5,13 +5,12 @@ defmodule Sanbase.Metric.Registry.EventEmitter do
   def topic(), do: @topic
 
   def handle_event(_, event_type, _args) when event_type in [:metrics_failed_to_load] do
-    %{
-      event_type: event_type
-    }
+    %{event_type: event_type}
     |> notify()
   end
 
-  def handle_event({:ok, map}, event_type = :bulk_metric_registry_change, args) do
+  def handle_event({:ok, map}, event_type, args)
+      when event_type in [:bulk_metric_registry_change] do
     %{event_type: event_type}
     |> Map.merge(map)
     |> Map.merge(args)

diff --git a/lib/sanbase/metric/registry/populate.ex b/lib/sanbase/metric/registry/populate.ex
@@ -148,10 +148,10 @@ defmodule Sanbase.Metric.Registry.Populate do
   end
 
   defp emit_events(list, summary) do
-    inserts = Map.get(summary, :insert, [])
-    updates = Map.get(summary, :update, [])
+    inserts = Map.get(summary, :insert, 0)
+    updates = Map.get(summary, :update, 0)
 
-    if inserts != [] or updates != [] do
+    if inserts > 0 or updates > 0 do
       map = %{inserts_count: inserts, updates_count: updates}
 
       {inserted_metrics, updated_metrics} =
@@ -178,10 +178,12 @@ defmodule Sanbase.Metric.Registry.Populate do
       # in persistent term
       Node.list()
       |> Enum.each(fn node ->
+        IO.puts("Emitting event :bulk_metric_registry_change to #{node}")
+
         Node.spawn(node, fn ->
           Sanbase.Metric.Registry.EventEmitter.emit_event(
             {:ok, map},
-            :bulk_update_metric_registry,
+            :bulk_metric_registry_change,
             %{__only_process_by__: [Sanbase.EventBus.MetricRegistrySubscriber]}
           )
         end)

diff --git a/test/test_seeds.exs b/test/test_seeds.exs
@@ -2,6 +2,8 @@ IO.puts("Running test seeds")
 
 IO.puts("Populating the Metric Registry...")
 
-{:ok, metrics, _summary} = Sanbase.Metric.Registry.Populate.run()
+{:ok, metrics, summary} = Sanbase.Metric.Registry.Populate.run()
 
-IO.puts("Finished populating the Metric Registry. Inserted #{length(metrics)} metrics")
+IO.puts(
+  "Finished populating the Metric Registry. Inserted #{length(metrics)} metrics. Summary: #{inspect(summary)}"
+)