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