Skip to content

Commit

Permalink
feat: generalize transfer history for aex141 (#882)
Browse files Browse the repository at this point in the history
  • Loading branch information
@jyeshe
jyeshe committed Aug 31, 2022
1 parent a6da18c commit c6cb13c
Show file tree
Hide file tree
Showing 12 changed files with 1,208 additions and 310 deletions.
111 changes: 0 additions & 111 deletions lib/ae_mdw/aex9.ex
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,38 +78,6 @@ defmodule AeMdw.Aex9 do
end
end

@spec fetch_sender_transfers(State.t(), pubkey(), pagination(), cursor() | nil) ::
account_paginated_transfers()
def fetch_sender_transfers(state, sender_pk, pagination, cursor) do
paginate_account_transfers(state, pagination, Model.Aex9Transfer, cursor, sender_pk)
end

@spec fetch_recipient_transfers(State.t(), pubkey(), pagination(), cursor() | nil) ::
account_paginated_transfers()
def fetch_recipient_transfers(state, recipient_pk, pagination, cursor) do
paginate_account_transfers(state, pagination, Model.RevAex9Transfer, cursor, recipient_pk)
end

@spec fetch_pair_transfers(State.t(), pubkey(), pubkey(), pagination(), cursor() | nil) ::
pair_paginated_transfers()
def fetch_pair_transfers(
state,
sender_pk,
recipient_pk,
pagination,
cursor
) do
cursor_key = deserialize_cursor(cursor)

paginate_transfers(
state,
pagination,
Model.Aex9PairTransfer,
cursor_key,
{sender_pk, recipient_pk}
)
end

@spec fetch_chain_balances(pubkey(), pagination(), balances_cursor() | nil) ::
{:ok, balances_cursor() | nil, [aex9_balance()], balances_cursor() | nil}
| {:error, Error.t()}
Expand Down Expand Up @@ -305,51 +273,6 @@ defmodule AeMdw.Aex9 do
end
end

defp paginate_account_transfers(
state,
pagination,
table,
cursor,
account_pk
) do
cursor_key = deserialize_cursor(cursor)

paginate_transfers(
state,
pagination,
table,
cursor_key,
account_pk
)
end

defp paginate_transfers(
state,
pagination,
table,
cursor_key,
params
) do
key_boundary = key_boundary(params)

{prev_cursor_key, transfer_keys, next_cursor_key} =
state
|> build_streamer(table, cursor_key, key_boundary)
|> Collection.paginate(pagination)

{
serialize_cursor(prev_cursor_key),
transfer_keys,
serialize_cursor(next_cursor_key)
}
end

defp build_streamer(state, table, cursor_key, key_boundary) do
fn direction ->
Collection.stream(state, table, direction, key_boundary, cursor_key)
end
end

defp render_balance(contract_pk, {:address, account_pk}, amount) do
%{
contract: :aeser_api_encoder.encode(:contract_pubkey, contract_pk),
Expand Down Expand Up @@ -393,40 +316,6 @@ defmodule AeMdw.Aex9 do
end
end

defp serialize_cursor(nil), do: nil

defp serialize_cursor({cursor, is_reversed?}),
do: {cursor |> :erlang.term_to_binary() |> Base.encode64(), is_reversed?}

defp deserialize_cursor(nil), do: nil

defp deserialize_cursor(<<cursor_bin64::binary>>) do
with {:ok, cursor_bin} <- Base.decode64(cursor_bin64),
cursor_term <- :erlang.binary_to_term(cursor_bin),
true <-
match?({<<_pk1::256>>, _txi, <<_pk2::256>>, _amount, _idx}, cursor_term) or
match?({<<_pk1::256>>, <<_pk2::256>>, _txi, _amount, _idx}, cursor_term) do
cursor_term
else
_invalid ->
raise ErrInput.Cursor, value: cursor_bin64
end
end

defp key_boundary({sender_pk, recipient_pk}) do
{
{sender_pk, recipient_pk, 0, 0, 0},
{sender_pk, recipient_pk, nil, 0, 0}
}
end

defp key_boundary(account_pk) do
{
{account_pk, 0, nil, 0, 0},
{account_pk, nil, nil, 0, 0}
}
end

defp validate_aex9(contract_pk) do
if AexnContracts.is_aex9?(contract_pk) do
:ok
Expand Down
174 changes: 174 additions & 0 deletions lib/ae_mdw/aexn_transfers.ex
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,174 @@
defmodule AeMdw.AexnTransfers do
@moduledoc """
Fetches indexed AEX-N transfers (from Transfer AEX-9 and AEX-141 events).
"""

alias AeMdw.Collection
alias AeMdw.Db.Model
alias AeMdw.Db.State
alias AeMdw.Error.Input, as: ErrInput

require Model

@type aexn_type :: :aex9 | :aex141
@type amount :: non_neg_integer()
@type amounts :: map()
@type token_id :: non_neg_integer()

@typep txi :: AeMdw.Txs.txi()

@type transfer_key ::
{:aex9 | :aex141, pubkey(), txi(), pubkey(), pos_integer(), non_neg_integer()}
@type pair_transfer_key ::
{:aex9 | :aex141, pubkey(), pubkey(), txi(), pos_integer(), non_neg_integer()}

@type cursor :: binary()
@type account_paginated_transfers ::
{cursor() | nil, [transfer_key()], {cursor() | nil, boolean()}}
@type pair_paginated_transfers ::
{cursor() | nil, [pair_transfer_key()], {cursor() | nil, boolean()}}

@typep pagination :: Collection.direction_limit()
@typep pubkey :: AeMdw.Node.Db.pubkey()

@spec fetch_sender_transfers(State.t(), aexn_type(), pubkey(), pagination(), cursor() | nil) ::
account_paginated_transfers()
def fetch_sender_transfers(state, aexn_type, sender_pk, pagination, cursor) do
paginate_account_transfers(
state,
aexn_type,
pagination,
Model.AexnTransfer,
cursor,
sender_pk
)
end

@spec fetch_recipient_transfers(State.t(), aexn_type(), pubkey(), pagination(), cursor() | nil) ::
account_paginated_transfers()
def fetch_recipient_transfers(state, aexn_type, recipient_pk, pagination, cursor) do
paginate_account_transfers(
state,
aexn_type,
pagination,
Model.RevAexnTransfer,
cursor,
recipient_pk
)
end

@spec fetch_pair_transfers(
State.t(),
aexn_type(),
pubkey(),
pubkey(),
pagination(),
cursor() | nil
) ::
pair_paginated_transfers()
def fetch_pair_transfers(
state,
aexn_type,
sender_pk,
recipient_pk,
pagination,
cursor
) do
cursor_key = deserialize_cursor(cursor)

paginate_transfers(
state,
aexn_type,
pagination,
Model.AexnPairTransfer,
cursor_key,
{sender_pk, recipient_pk}
)
end

#
# Private functions
#
defp paginate_account_transfers(
state,
aexn_type,
pagination,
table,
cursor,
account_pk
) do
cursor_key = deserialize_cursor(cursor)

paginate_transfers(
state,
aexn_type,
pagination,
table,
cursor_key,
account_pk
)
end

defp paginate_transfers(
state,
aexn_type,
pagination,
table,
cursor_key,
params
) do
key_boundary = key_boundary(aexn_type, params)

{prev_cursor_key, transfer_keys, next_cursor_key} =
state
|> build_streamer(table, cursor_key, key_boundary)
|> Collection.paginate(pagination)

{
serialize_cursor(prev_cursor_key),
transfer_keys,
serialize_cursor(next_cursor_key)
}
end

defp build_streamer(state, table, cursor_key, key_boundary) do
fn direction ->
Collection.stream(state, table, direction, key_boundary, cursor_key)
end
end

defp serialize_cursor(nil), do: nil

defp serialize_cursor({cursor, is_reversed?}),
do: {cursor |> :erlang.term_to_binary() |> Base.encode64(), is_reversed?}

defp deserialize_cursor(nil), do: nil

defp deserialize_cursor(<<cursor_bin64::binary>>) do
with {:ok, cursor_bin} <- Base.decode64(cursor_bin64),
cursor_term <- :erlang.binary_to_term(cursor_bin),
true <-
elem(cursor_term, 0) in [:aex9, :aex141] and
(match?({_type, <<_pk1::256>>, _txi, <<_pk2::256>>, _amount, _idx}, cursor_term) or
match?({_type, <<_pk1::256>>, <<_pk2::256>>, _txi, _amount, _idx}, cursor_term)) do
cursor_term
else
_invalid ->
raise ErrInput.Cursor, value: cursor_bin64
end
end

defp key_boundary(aexn_type, {sender_pk, recipient_pk}) do
{
{aexn_type, sender_pk, recipient_pk, 0, 0, 0},
{aexn_type, sender_pk, recipient_pk, nil, 0, 0}
}
end

defp key_boundary(aexn_type, account_pk) do
{
{aexn_type, account_pk, 0, nil, 0, 0},
{aexn_type, account_pk, nil, nil, 0, 0}
}
end
end
Loading

0 comments on commit c6cb13c

Please sign in to comment.