feat: show quorum percentage on traces

telemetry_api/lib/telemetry_api/contract_managers/stake_registry.ex

@@ -0,0 +1,48 @@
+defmodule TelemetryApi.ContractManagers.StakeRegistry do
+  alias TelemetryApi.ContractManagers.StakeRegistry
+
+  require Logger
+
+  @aligned_config_file System.get_env("ALIGNED_CONFIG_FILE")
+
+  config_file_path =
+    case @aligned_config_file do
+      nil -> raise("ALIGNED_CONFIG_FILE not set in .env")
+      file -> file
+    end
+
+  {status, config_json_string} = File.read(config_file_path)
+
+  case status do
+    :ok ->
+      Logger.debug("Aligned deployment file read successfully")
+
+    :error ->
+      raise("Config file not read successfully")
+  end
+
+  @stake_registry_address Jason.decode!(config_json_string)
+                        |> Map.get("addresses")
+                        |> Map.get("stakeRegistry")
+
+  use Ethers.Contract,
+    abi_file: "priv/abi/IStakeRegistry.json",
+    default_address: @stake_registry_address
+
+    @doc """
+    Get the current total stake
+    We only use quorum 0
+
+    ## Examples
+
+        iex> get_current_total_stake()
+        {:ok, 100}
+
+        iex> get_current_total_stake()
+        {:error, "Error message"}
+    """
+    def get_current_total_stake() do
+      StakeRegistry.get_current_total_stake(0)
+        |> Ethers.call()
+    end
+end

telemetry_api/lib/telemetry_api/traces.ex

@@ -4,6 +4,7 @@ defmodule TelemetryApi.Traces do
   """
   alias TelemetryApi.Traces.Trace
   alias TelemetryApi.Operators
+  alias TelemetryApi.ContractManagers.StakeRegistry
 
   require OpenTelemetry.Tracer
   require OpenTelemetry.Ctx
@@ -22,26 +23,32 @@ defmodule TelemetryApi.Traces do
       {:ok, "merkle_root"}
   """
   def create_task_trace(merkle_root) do
-    span_ctx =
-      Tracer.start_span(
-        "Task: #{merkle_root}",
-        %{
-          attributes: [
-            {:merkle_root, merkle_root}
-          ]
-        }
-      )
-
-    ctx = Ctx.get_current()
-
-    TraceStore.store_trace(merkle_root, %Trace{
-      parent_span: span_ctx,
-      context: ctx,
-      responses: []
-    })
-
-    IO.inspect("New task trace with merkle_root: #{IO.inspect(merkle_root)}")
-    {:ok, merkle_root}
+    # Get total stake
+    with {:ok, total_stake} <- StakeRegistry.get_current_total_stake() do
+      span_ctx =
+        Tracer.start_span(
+          "Task: #{merkle_root}",
+          %{
+            attributes: [
+              {:merkle_root, merkle_root},
+              {:total_stake, total_stake}
+            ]
+          }
+        )
+
+      ctx = Ctx.get_current()
+
+      TraceStore.store_trace(merkle_root, %Trace{
+        parent_span: span_ctx,
+        context: ctx,
+        total_stake: total_stake,
+        current_stake: 0,
+        responses: []
+      })
+
+      IO.inspect("New task trace with merkle_root: #{IO.inspect(merkle_root)}")
+      {:ok, merkle_root}
+    end
   end
 
   @doc """
@@ -64,8 +71,9 @@ defmodule TelemetryApi.Traces do
           {:operator_id, operator_id},
           {:name, operator.name},
           {:address, operator.address},
-          {:stake, operator.stake}
-        ]
+          {:operator_stake, operator.stake}
+        ],
+        String.to_integer(operator.stake)
       )
 
       trace = TraceStore.get_trace(merkle_root)
@@ -171,6 +179,31 @@ defmodule TelemetryApi.Traces do
     add_event(merkle_root, "Missing Operators", [{:operators, missing_operators}])
   end
 
+  defp add_event(merkle_root, event_name, event_attributes, operator_stake) do
+    case TraceStore.get_trace(merkle_root) do
+      nil ->
+        IO.inspect("Context not found for #{merkle_root}")
+        {:error, "Context not found for #{merkle_root}"}
+
+      trace ->
+        Ctx.attach(trace.context)
+        Tracer.set_current_span(trace.parent_span)
+
+        new_stake = trace.current_stake + operator_stake
+        new_stake_fraction = new_stake / trace.total_stake
+        operator_stake_fraction = operator_stake / trace.total_stake
+
+        TraceStore.store_trace(merkle_root, %Trace{
+          parent_span: trace.parent_span,
+          context: trace.context,
+          total_stake: trace.total_stake,
+          current_stake: new_stake
+          })
+
+        Tracer.add_event(event_name, event_attributes ++ [{:current_stake, new_stake}, {:current_stake_fraction, new_stake_fraction}, {:operator_stake_fraction, operator_stake_fraction}])
+      end
+  end
+
   defp add_event(merkle_root, event_name, event_attributes) do
     case TraceStore.get_trace(merkle_root) do
       nil ->

telemetry_api/lib/telemetry_api/traces/trace.ex

@@ -1,4 +1,4 @@
 defmodule TelemetryApi.Traces.Trace do
-  @enforce_keys [:parent_span, :context, :responses]
-  defstruct [:parent_span, :context, :responses]
+  @enforce_keys [:parent_span, :context, :total_stake, :current_stake, :responses]
+  defstruct [:parent_span, :context, :total_stake, :current_stake, :responses]
 end

telemetry_api/lib/telemetry_api_web/controllers/fallback_controller.ex

@@ -21,4 +21,10 @@ defmodule TelemetryApiWeb.FallbackController do
     |> put_view(html: TelemetryApiWeb.ErrorHTML, json: TelemetryApiWeb.ErrorJSON)
     |> render(:"404")
   end
+
+  def call(conn, {:error, message}) do
+    conn
+    |> put_resp_content_type("application/json")
+    |> send_resp(:internal_server_error, Jason.encode!(%{error: message}))
+  end
 end