-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add getCompressedMintTokenHolders endpoint and fix pagination for sig…
…natures and compressed token accounts (#241) * Increase poller timeout * Fix poller bug * Intermediate commit * Intermediate commit * Intermediate commit * Intermediate commit * Intermediate commit * Update docs
- Loading branch information
1 parent
15495a9
commit cf6c6eb
Showing
34 changed files
with
517 additions
and
97 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,110 @@ | ||
| use byteorder::{ByteOrder, LittleEndian}; | ||
| use sea_orm::{ColumnTrait, DatabaseConnection, EntityTrait, QueryFilter, QueryOrder, QuerySelect}; | ||
| use serde::{Deserialize, Serialize}; | ||
| use utoipa::ToSchema; | ||
|
|
||
| use crate::common::typedefs::bs58_string::Base58String; | ||
| use crate::common::typedefs::serializable_pubkey::SerializablePubkey; | ||
| use crate::common::typedefs::unsigned_integer::UnsignedInteger; | ||
| use crate::dao::generated::token_owner_balances; | ||
|
|
||
| use super::super::error::PhotonApiError; | ||
| use super::utils::{parse_decimal, Context, Limit, PAGE_LIMIT}; | ||
|
|
||
| #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, ToSchema)] | ||
| pub struct OwnerBalance { | ||
| pub owner: SerializablePubkey, | ||
| pub balance: UnsignedInteger, | ||
| } | ||
|
|
||
| #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, ToSchema)] | ||
| pub struct OwnerBalanceList { | ||
| pub items: Vec<OwnerBalance>, | ||
| pub cursor: Option<Base58String>, | ||
| } | ||
|
|
||
| // We do not use generics to simplify documentation generation. | ||
| #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, ToSchema)] | ||
| #[serde(deny_unknown_fields, rename_all = "camelCase")] | ||
| pub struct OwnerBalancesResponse { | ||
| pub context: Context, | ||
| pub value: OwnerBalanceList, | ||
| } | ||
|
|
||
| #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, ToSchema, Default)] | ||
| #[serde(deny_unknown_fields, rename_all = "camelCase")] | ||
| pub struct GetCompressedMintTokenHoldersRequest { | ||
| pub mint: SerializablePubkey, | ||
| pub cursor: Option<Base58String>, | ||
| pub limit: Option<Limit>, | ||
| } | ||
|
|
||
| pub async fn get_compressed_mint_token_holders( | ||
| conn: &DatabaseConnection, | ||
| request: GetCompressedMintTokenHoldersRequest, | ||
| ) -> Result<OwnerBalancesResponse, PhotonApiError> { | ||
| let context = Context::extract(conn).await?; | ||
| let GetCompressedMintTokenHoldersRequest { | ||
| mint, | ||
| cursor, | ||
| limit, | ||
| } = request; | ||
| let mut filter = token_owner_balances::Column::Mint.eq::<Vec<u8>>(mint.into()); | ||
| if let Some(cursor) = cursor { | ||
| let bytes = cursor.0; | ||
| let expected_cursor_length = 40; | ||
| let (balance, owner) = if bytes.len() == expected_cursor_length { | ||
| let (balance, owner) = bytes.split_at(expected_cursor_length); | ||
| (balance, owner) | ||
| } else { | ||
| return Err(PhotonApiError::ValidationError(format!( | ||
| "Invalid cursor length. Expected {}. Received {}.", | ||
| expected_cursor_length, | ||
| bytes.len() | ||
| ))); | ||
| }; | ||
| let balance = LittleEndian::read_u64(&balance); | ||
| filter = filter.and( | ||
| token_owner_balances::Column::Amount.lt(balance).or( | ||
| token_owner_balances::Column::Amount | ||
| .eq(balance) | ||
| .and(token_owner_balances::Column::Owner.gt::<Vec<u8>>(owner.into())), | ||
| ), | ||
| ); | ||
| } | ||
| let limit = limit.map(|l| l.value()).unwrap_or(PAGE_LIMIT); | ||
|
|
||
| let items = token_owner_balances::Entity::find() | ||
| .filter(filter) | ||
| .order_by_desc(token_owner_balances::Column::Amount) | ||
| .order_by_asc(token_owner_balances::Column::Mint) | ||
| .limit(limit) | ||
| .all(conn) | ||
| .await? | ||
| .drain(..) | ||
| .map(|token_owner_balance| { | ||
| Ok(OwnerBalance { | ||
| owner: token_owner_balance.owner.try_into()?, | ||
| balance: UnsignedInteger(parse_decimal(token_owner_balance.amount)?), | ||
| }) | ||
| }) | ||
| .collect::<Result<Vec<OwnerBalance>, PhotonApiError>>()?; | ||
|
|
||
| let mut cursor = items.last().map(|item| { | ||
| Base58String({ | ||
| let item = item.clone(); | ||
| let mut bytes: Vec<u8> = Vec::new(); | ||
| bytes.extend_from_slice(&item.balance.0.to_le_bytes()); | ||
| bytes.extend_from_slice(&item.owner.0.to_bytes()); | ||
| bytes | ||
| }) | ||
| }); | ||
| if items.len() < limit as usize { | ||
| cursor = None; | ||
| } | ||
|
|
||
| Ok(OwnerBalancesResponse { | ||
| value: OwnerBalanceList { items, cursor }, | ||
| context, | ||
| }) | ||
| } |
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,54 @@ | ||
| use sea_orm_migration::prelude::*; | ||
| use sea_orm_migration::sea_orm::{ConnectionTrait, DatabaseBackend, Statement}; | ||
|
|
||
| use crate::migration::model::table::TokenOwnerBalances; | ||
|
|
||
| #[derive(DeriveMigrationName)] | ||
| pub struct Migration; | ||
|
|
||
| async fn execute_sql<'a>(manager: &SchemaManager<'_>, sql: &str) -> Result<(), DbErr> { | ||
| manager | ||
| .get_connection() | ||
| .execute(Statement::from_string( | ||
| manager.get_database_backend(), | ||
| sql.to_string(), | ||
| )) | ||
| .await?; | ||
| Ok(()) | ||
| } | ||
|
|
||
| #[async_trait::async_trait] | ||
| impl MigrationTrait for Migration { | ||
| async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> { | ||
| if manager.get_database_backend() == DatabaseBackend::Postgres { | ||
| // Create index concurrently for Postgres | ||
| execute_sql( | ||
| manager, | ||
| "CREATE INDEX CONCURRENTLY IF NOT EXISTS token_holder_mint_balance_owner_idx ON token_owner_balances (mint, amount, owner);", | ||
| ) | ||
| .await?; | ||
| } else { | ||
| // For other databases, create index normally | ||
| execute_sql( | ||
| manager, | ||
| "CREATE INDEX IF NOT EXISTS token_holder_mint_balance_owner_idx ON token_owner_balances (mint, amount, owner);", | ||
| ) | ||
| .await?; | ||
| } | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
||
| async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> { | ||
| manager | ||
| .drop_index( | ||
| Index::drop() | ||
| .name("token_holder_mint_balance_owner_idx") | ||
| .table(TokenOwnerBalances::Table) | ||
| .to_owned(), | ||
| ) | ||
| .await?; | ||
|
|
||
| Ok(()) | ||
| } | ||
| } |
Oops, something went wrong.