-
Notifications
You must be signed in to change notification settings - Fork 36
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into checkpoint_states_db
- Loading branch information
Showing
9 changed files
with
65 additions
and
107 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,83 +1,86 @@ | ||
| defmodule LambdaEthereumConsensus.P2P.Metadata do | ||
| @moduledoc """ | ||
| This module handles Metadata's genserver to fetch and edit. | ||
| This module handles node's Metadata (fetch and edit). | ||
| """ | ||
|
|
||
| use GenServer | ||
|
|
||
| alias LambdaEthereumConsensus.Store.Db | ||
| alias LambdaEthereumConsensus.Utils.BitVector | ||
| alias Types.Metadata | ||
|
|
||
| @metadata_prefix "metadata" | ||
|
|
||
| ########################## | ||
| ### Public API | ||
| ########################## | ||
|
|
||
| def start_link(opts) do | ||
| GenServer.start_link(__MODULE__, opts, name: __MODULE__) | ||
| @doc """ | ||
| Initializes the table in the db by creating and storing empty metadata. | ||
| """ | ||
| def init() do | ||
| Metadata.empty() |> store_metadata() | ||
| end | ||
|
|
||
| @spec get_seq_number() :: Types.uint64() | ||
| def get_seq_number() do | ||
| GenServer.call(__MODULE__, :get_seq_number) | ||
| %Metadata{seq_number: seq_number} = fetch_metadata!() | ||
| seq_number | ||
| end | ||
|
|
||
| @spec get_metadata() :: Metadata.t() | ||
| def get_metadata() do | ||
| GenServer.call(__MODULE__, :get_metadata) | ||
| fetch_metadata!() | ||
| end | ||
|
|
||
| @spec set_attnet(non_neg_integer()) :: :ok | ||
| def set_attnet(i), do: GenServer.cast(__MODULE__, {:set_attestation_subnet, i, true}) | ||
| def set_attnet(i) do | ||
| update_metadata(fn metadata -> Map.update!(metadata, :attnets, &BitVector.set(&1, i)) end) | ||
| end | ||
|
|
||
| @spec clear_attnet(non_neg_integer()) :: :ok | ||
| def clear_attnet(i), do: GenServer.cast(__MODULE__, {:set_attestation_subnet, i, false}) | ||
| def clear_attnet(i) do | ||
| update_metadata(fn metadata -> Map.update!(metadata, :attnets, &BitVector.clear(&1, i)) end) | ||
| end | ||
|
|
||
| @spec set_syncnet(non_neg_integer()) :: :ok | ||
| def set_syncnet(i), do: GenServer.cast(__MODULE__, {:set_sync_committee, i, true}) | ||
| def set_syncnet(i) do | ||
| update_metadata(fn metadata -> Map.update!(metadata, :syncnets, &BitVector.set(&1, i)) end) | ||
| end | ||
|
|
||
| @spec clear_syncnet(non_neg_integer()) :: :ok | ||
| def clear_syncnet(i), do: GenServer.cast(__MODULE__, {:set_sync_committee, i, false}) | ||
| def clear_syncnet(i) do | ||
| update_metadata(fn metadata -> Map.update!(metadata, :syncnets, &BitVector.clear(&1, i)) end) | ||
| end | ||
|
|
||
| ########################## | ||
| ### GenServer Callbacks | ||
| ### Private Functions | ||
| ########################## | ||
|
|
||
| @impl true | ||
| def init(_opts) do | ||
| Metadata.empty() |> Metadata.store_metadata() | ||
| {:ok, nil} | ||
| defp update_metadata(f) when is_function(f) do | ||
| fetch_metadata!() |> f.() |> increment_seqnum() |> store_metadata() | ||
| end | ||
|
|
||
| @impl true | ||
| def handle_call(:get_seq_number, _, _state) do | ||
| %Metadata{seq_number: seq_number} = Metadata.fetch_metadata!() | ||
| {:reply, seq_number, nil} | ||
| end | ||
|
|
||
| @impl true | ||
| def handle_call(:get_metadata, _, _metadata), do: {:reply, Metadata.fetch_metadata!(), nil} | ||
| defp increment_seqnum(state), do: %{state | seq_number: state.seq_number + 1} | ||
|
|
||
| @impl true | ||
| def handle_cast({:set_attestation_subnet, i, set}, _metadata) do | ||
| metadata = Metadata.fetch_metadata!() | ||
| attnets = set_or_clear(metadata.attnets, i, set) | ||
| %{metadata | attnets: attnets} |> increment_seqnum() |> Metadata.store_metadata() | ||
| {:noreply, nil} | ||
| defp store_metadata(%Metadata{} = map) do | ||
| :telemetry.span([:db, :latency], %{}, fn -> | ||
| {Db.put( | ||
| @metadata_prefix, | ||
| :erlang.term_to_binary(map) | ||
| ), %{module: "metadata", action: "persist"}} | ||
| end) | ||
| end | ||
|
|
||
| @impl true | ||
| def handle_cast({:set_sync_committee, i, set}, _metadata) do | ||
| metadata = Metadata.fetch_metadata!() | ||
| syncnets = set_or_clear(metadata.syncnets, i, set) | ||
| %{metadata | syncnets: syncnets} |> increment_seqnum() |> Metadata.store_metadata() | ||
| {:noreply, nil} | ||
| defp fetch_metadata() do | ||
| with {:ok, binary} <- | ||
| :telemetry.span([:db, :latency], %{}, fn -> | ||
| {Db.get(@metadata_prefix), %{module: "metadata", action: "fetch"}} | ||
| end) do | ||
| {:ok, :erlang.binary_to_term(binary)} | ||
| end | ||
| end | ||
|
|
||
| ########################## | ||
| ### Private Functions | ||
| ########################## | ||
|
|
||
| @spec set_or_clear(BitVector.t(), non_neg_integer(), boolean()) :: BitVector.t() | ||
| defp set_or_clear(bitvector, i, true), do: BitVector.set(bitvector, i) | ||
| defp set_or_clear(bitvector, i, false), do: BitVector.clear(bitvector, i) | ||
|
|
||
| defp increment_seqnum(state), do: %{state | seq_number: state.seq_number + 1} | ||
| defp fetch_metadata!() do | ||
| {:ok, metadata} = fetch_metadata() | ||
| metadata | ||
| end | ||
| end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters