Merge pull request #2036 from poanetwork/stakes-tables
New tables for staking pools and delegatorspull/2086/head
Victor Baranov
6 years ago
committed by
GitHub
commit
191e3a67c7
@ -0,0 +1,91 @@ |
||||
defmodule Explorer.Chain.Import.Runner.StakingPoolsDelegators do |
||||
@moduledoc """ |
||||
Bulk imports delegators to StakingPoolsDelegator tabe. |
||||
""" |
||||
|
||||
require Ecto.Query |
||||
|
||||
alias Ecto.{Changeset, Multi, Repo} |
||||
alias Explorer.Chain.{Import, StakingPoolsDelegator} |
||||
|
||||
import Ecto.Query, only: [from: 2] |
||||
|
||||
@behaviour Import.Runner |
||||
|
||||
# milliseconds |
||||
@timeout 60_000 |
||||
|
||||
@type imported :: [StakingPoolsDelegator.t()] |
||||
|
||||
@impl Import.Runner |
||||
def ecto_schema_module, do: StakingPoolsDelegator |
||||
|
||||
@impl Import.Runner |
||||
def option_key, do: :staking_pools_delegators |
||||
|
||||
@impl Import.Runner |
||||
def imported_table_row do |
||||
%{ |
||||
value_type: "[#{ecto_schema_module()}.t()]", |
||||
value_description: "List of `t:#{ecto_schema_module()}.t/0`s" |
||||
} |
||||
end |
||||
|
||||
@impl Import.Runner |
||||
def run(multi, changes_list, %{timestamps: timestamps} = options) do |
||||
insert_options = |
||||
options |
||||
|> Map.get(option_key(), %{}) |
||||
|> Map.take(~w(on_conflict timeout)a) |
||||
|> Map.put_new(:timeout, @timeout) |
||||
|> Map.put(:timestamps, timestamps) |
||||
|
||||
multi |
||||
|> Multi.run(:insert_staking_pools_delegators, fn repo, _ -> |
||||
insert(repo, changes_list, insert_options) |
||||
end) |
||||
end |
||||
|
||||
@impl Import.Runner |
||||
def timeout, do: @timeout |
||||
|
||||
@spec insert(Repo.t(), [map()], %{ |
||||
optional(:on_conflict) => Import.Runner.on_conflict(), |
||||
required(:timeout) => timeout, |
||||
required(:timestamps) => Import.timestamps() |
||||
}) :: |
||||
{:ok, [StakingPoolsDelegator.t()]} |
||||
| {:error, [Changeset.t()]} |
||||
defp insert(repo, changes_list, %{timeout: timeout, timestamps: timestamps} = options) when is_list(changes_list) do |
||||
on_conflict = Map.get_lazy(options, :on_conflict, &default_on_conflict/0) |
||||
|
||||
{:ok, _} = |
||||
Import.insert_changes_list( |
||||
repo, |
||||
changes_list, |
||||
conflict_target: [:pool_address_hash, :delegator_address_hash], |
||||
on_conflict: on_conflict, |
||||
for: StakingPoolsDelegator, |
||||
returning: [:pool_address_hash, :delegator_address_hash], |
||||
timeout: timeout, |
||||
timestamps: timestamps |
||||
) |
||||
end |
||||
|
||||
defp default_on_conflict do |
||||
from( |
||||
delegator in StakingPoolsDelegator, |
||||
update: [ |
||||
set: [ |
||||
stake_amount: fragment("EXCLUDED.stake_amount"), |
||||
ordered_withdraw: fragment("EXCLUDED.ordered_withdraw"), |
||||
max_withdraw_allowed: fragment("EXCLUDED.max_withdraw_allowed"), |
||||
max_ordered_withdraw_allowed: fragment("EXCLUDED.max_ordered_withdraw_allowed"), |
||||
ordered_withdraw_epoch: fragment("EXCLUDED.ordered_withdraw_epoch"), |
||||
inserted_at: fragment("LEAST(?, EXCLUDED.inserted_at)", delegator.inserted_at), |
||||
updated_at: fragment("GREATEST(?, EXCLUDED.updated_at)", delegator.updated_at) |
||||
] |
||||
] |
||||
) |
||||
end |
||||
end |
||||
@ -0,0 +1,97 @@ |
||||
defmodule Explorer.Chain.StakingPool do |
||||
@moduledoc """ |
||||
The representation of staking pool from POSDAO network. |
||||
Staking pools might be candidate or validator. |
||||
""" |
||||
use Ecto.Schema |
||||
import Ecto.Changeset |
||||
|
||||
alias Explorer.Chain.{ |
||||
Address, |
||||
Hash, |
||||
StakingPoolsDelegator, |
||||
Wei |
||||
} |
||||
|
||||
@type t :: %__MODULE__{ |
||||
staking_address_hash: Hash.Address.t(), |
||||
mining_address_hash: Hash.Address.t(), |
||||
banned_until: boolean, |
||||
delegators_count: integer, |
||||
is_active: boolean, |
||||
is_banned: boolean, |
||||
is_validator: boolean, |
||||
likelihood: integer, |
||||
staked_ratio: Decimal.t(), |
||||
self_staked_amount: Wei.t(), |
||||
staked_amount: Wei.t(), |
||||
was_banned_count: integer, |
||||
was_validator_count: integer, |
||||
is_deleted: boolean |
||||
} |
||||
|
||||
@attrs ~w( |
||||
is_active delegators_count staked_amount self_staked_amount is_validator |
||||
was_validator_count is_banned was_banned_count banned_until likelihood |
||||
staked_ratio staking_address_hash mining_address_hash |
||||
)a |
||||
@req_attrs ~w( |
||||
is_active delegators_count staked_amount self_staked_amount is_validator |
||||
was_validator_count is_banned was_banned_count banned_until |
||||
staking_address_hash mining_address_hash |
||||
)a |
||||
|
||||
schema "staking_pools" do |
||||
field(:banned_until, :integer) |
||||
field(:delegators_count, :integer) |
||||
field(:is_active, :boolean, default: false) |
||||
field(:is_banned, :boolean, default: false) |
||||
field(:is_validator, :boolean, default: false) |
||||
field(:likelihood, :decimal) |
||||
field(:staked_ratio, :decimal) |
||||
field(:self_staked_amount, Wei) |
||||
field(:staked_amount, Wei) |
||||
field(:was_banned_count, :integer) |
||||
field(:was_validator_count, :integer) |
||||
field(:is_deleted, :boolean, default: false) |
||||
has_many(:delegators, StakingPoolsDelegator, foreign_key: :pool_address_hash) |
||||
|
||||
belongs_to( |
||||
:staking_address, |
||||
Address, |
||||
foreign_key: :staking_address_hash, |
||||
references: :hash, |
||||
type: Hash.Address |
||||
) |
||||
|
||||
belongs_to( |
||||
:mining_address, |
||||
Address, |
||||
foreign_key: :mining_address_hash, |
||||
references: :hash, |
||||
type: Hash.Address |
||||
) |
||||
|
||||
timestamps(null: false, type: :utc_datetime_usec) |
||||
end |
||||
|
||||
@doc false |
||||
def changeset(staking_pool, attrs) do |
||||
staking_pool |
||||
|> cast(attrs, @attrs) |
||||
|> cast_assoc(:delegators) |
||||
|> validate_required(@req_attrs) |
||||
|> validate_staked_amount() |
||||
|> unique_constraint(:staking_address_hash) |
||||
end |
||||
|
||||
defp validate_staked_amount(%{valid?: false} = c), do: c |
||||
|
||||
defp validate_staked_amount(changeset) do |
||||
if get_field(changeset, :staked_amount) < get_field(changeset, :self_staked_amount) do |
||||
add_error(changeset, :staked_amount, "must be greater than self_staked_amount") |
||||
else |
||||
changeset |
||||
end |
||||
end |
||||
end |
||||
@ -0,0 +1,64 @@ |
||||
defmodule Explorer.Chain.StakingPoolsDelegator do |
||||
@moduledoc """ |
||||
The representation of delegators from POSDAO network. |
||||
Delegators make stakes on staking pools and withdraw from them. |
||||
""" |
||||
use Ecto.Schema |
||||
import Ecto.Changeset |
||||
|
||||
alias Explorer.Chain.{ |
||||
Address, |
||||
Hash, |
||||
StakingPool, |
||||
Wei |
||||
} |
||||
|
||||
@type t :: %__MODULE__{ |
||||
pool_address_hash: Hash.Address.t(), |
||||
delegator_address_hash: Hash.Address.t(), |
||||
max_ordered_withdraw_allowed: Wei.t(), |
||||
max_withdraw_allowed: Wei.t(), |
||||
ordered_withdraw: Wei.t(), |
||||
stake_amount: Wei.t(), |
||||
ordered_withdraw_epoch: integer() |
||||
} |
||||
|
||||
@attrs ~w( |
||||
pool_address_hash delegator_address_hash max_ordered_withdraw_allowed |
||||
max_withdraw_allowed ordered_withdraw stake_amount ordered_withdraw_epoch |
||||
)a |
||||
|
||||
schema "staking_pools_delegators" do |
||||
field(:max_ordered_withdraw_allowed, Wei) |
||||
field(:max_withdraw_allowed, Wei) |
||||
field(:ordered_withdraw, Wei) |
||||
field(:ordered_withdraw_epoch, :integer) |
||||
field(:stake_amount, Wei) |
||||
|
||||
belongs_to( |
||||
:staking_pool, |
||||
StakingPool, |
||||
foreign_key: :pool_address_hash, |
||||
references: :staking_address_hash, |
||||
type: Hash.Address |
||||
) |
||||
|
||||
belongs_to( |
||||
:delegator_address, |
||||
Address, |
||||
foreign_key: :delegator_address_hash, |
||||
references: :hash, |
||||
type: Hash.Address |
||||
) |
||||
|
||||
timestamps(null: false, type: :utc_datetime_usec) |
||||
end |
||||
|
||||
@doc false |
||||
def changeset(staking_pools_delegator, attrs) do |
||||
staking_pools_delegator |
||||
|> cast(attrs, @attrs) |
||||
|> validate_required(@attrs) |
||||
|> unique_constraint(:pool_address_hash, name: :pools_delegator_index) |
||||
end |
||||
end |
||||
@ -0,0 +1,27 @@ |
||||
defmodule Explorer.Repo.Migrations.CreateStakingPools do |
||||
use Ecto.Migration |
||||
|
||||
def change do |
||||
create table(:staking_pools) do |
||||
add(:is_active, :boolean, default: false, null: false) |
||||
add(:is_deleted, :boolean, default: false, null: false) |
||||
add(:delegators_count, :integer) |
||||
add(:staked_amount, :numeric, precision: 100) |
||||
add(:self_staked_amount, :numeric, precision: 100) |
||||
add(:is_validator, :boolean, default: false, null: false) |
||||
add(:was_validator_count, :integer) |
||||
add(:is_banned, :boolean, default: false, null: false) |
||||
add(:was_banned_count, :integer) |
||||
add(:banned_until, :bigint) |
||||
add(:likelihood, :decimal, precision: 5, scale: 2) |
||||
add(:staked_ratio, :decimal, precision: 5, scale: 2) |
||||
add(:staking_address_hash, :bytea) |
||||
add(:mining_address_hash, :bytea) |
||||
|
||||
timestamps(null: false, type: :utc_datetime_usec) |
||||
end |
||||
|
||||
create(index(:staking_pools, [:staking_address_hash], unique: true)) |
||||
create(index(:staking_pools, [:mining_address_hash])) |
||||
end |
||||
end |
||||
@ -0,0 +1,26 @@ |
||||
defmodule Explorer.Repo.Migrations.CreateStakingPoolsDelegator do |
||||
use Ecto.Migration |
||||
|
||||
def change do |
||||
create table(:staking_pools_delegators) do |
||||
add(:delegator_address_hash, :bytea) |
||||
add(:pool_address_hash, :bytea) |
||||
add(:stake_amount, :numeric, precision: 100) |
||||
add(:ordered_withdraw, :numeric, precision: 100) |
||||
add(:max_withdraw_allowed, :numeric, precision: 100) |
||||
add(:max_ordered_withdraw_allowed, :numeric, precision: 100) |
||||
add(:ordered_withdraw_epoch, :integer) |
||||
|
||||
timestamps(null: false, type: :utc_datetime_usec) |
||||
end |
||||
|
||||
create(index(:staking_pools_delegators, [:delegator_address_hash])) |
||||
|
||||
create( |
||||
index(:staking_pools_delegators, [:delegator_address_hash, :pool_address_hash], |
||||
unique: true, |
||||
name: :pools_delegator_index |
||||
) |
||||
) |
||||
end |
||||
end |
||||
@ -0,0 +1,32 @@ |
||||
defmodule Explorer.Chain.Import.Runner.StakingPoolsDelegatorsTest do |
||||
use Explorer.DataCase |
||||
|
||||
import Explorer.Factory |
||||
|
||||
alias Ecto.Multi |
||||
alias Explorer.Chain.Import.Runner.StakingPoolsDelegators |
||||
alias Explorer.Chain.StakingPoolsDelegator |
||||
|
||||
describe "run/1" do |
||||
test "insert new pools list" do |
||||
delegators = |
||||
[params_for(:staking_pools_delegator), params_for(:staking_pools_delegator)] |
||||
|> Enum.map(fn param -> |
||||
changeset = StakingPoolsDelegator.changeset(%StakingPoolsDelegator{}, param) |
||||
changeset.changes |
||||
end) |
||||
|
||||
assert {:ok, %{insert_staking_pools_delegators: list}} = run_changes(delegators) |
||||
assert Enum.count(list) == Enum.count(delegators) |
||||
end |
||||
end |
||||
|
||||
defp run_changes(changes) do |
||||
Multi.new() |
||||
|> StakingPoolsDelegators.run(changes, %{ |
||||
timeout: :infinity, |
||||
timestamps: %{inserted_at: DateTime.utc_now(), updated_at: DateTime.utc_now()} |
||||
}) |
||||
|> Repo.transaction() |
||||
end |
||||
end |
||||
@ -0,0 +1,18 @@ |
||||
defmodule Explorer.Chain.StakingPoolTest do |
||||
use Explorer.DataCase |
||||
|
||||
alias Explorer.Chain.StakingPool |
||||
|
||||
describe "changeset/2" do |
||||
test "with valid attributes" do |
||||
params = params_for(:staking_pool) |
||||
changeset = StakingPool.changeset(%StakingPool{}, params) |
||||
assert changeset.valid? |
||||
end |
||||
|
||||
test "with invalid attributes" do |
||||
changeset = StakingPool.changeset(%StakingPool{}, %{staking_address_hash: 0}) |
||||
refute changeset.valid? |
||||
end |
||||
end |
||||
end |
||||
@ -0,0 +1,18 @@ |
||||
defmodule Explorer.Chain.StakingPoolsDelegatorTest do |
||||
use Explorer.DataCase |
||||
|
||||
alias Explorer.Chain.StakingPoolsDelegator |
||||
|
||||
describe "changeset/2" do |
||||
test "with valid attributes" do |
||||
params = params_for(:staking_pools_delegator) |
||||
changeset = StakingPoolsDelegator.changeset(%StakingPoolsDelegator{}, params) |
||||
assert changeset.valid? |
||||
end |
||||
|
||||
test "with invalid attributes" do |
||||
changeset = StakingPoolsDelegator.changeset(%StakingPoolsDelegator{}, %{pool_address_hash: 0}) |
||||
refute changeset.valid? |
||||
end |
||||
end |
||||
end |
||||
Loading…
Reference in new issue