-
Notifications
You must be signed in to change notification settings - Fork 84
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Loading status checks…
[parquet-sdk][token_v2] migrate CurrentTokenRoyaltyV1
Showing
12 changed files
with
332 additions
and
91 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1,2 @@ | ||
| pub mod raw_token_claims; | ||
| pub mod raw_v1_token_royalty; |
97 changes: 97 additions & 0 deletions
97
rust/processor/src/db/common/models/token_v2_models/raw_v1_token_royalty.rs
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,97 @@ | ||
| // Copyright © Aptos Foundation | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| // This is required because a diesel macro makes clippy sad | ||
| #![allow(clippy::extra_unused_lifetimes)] | ||
| #![allow(clippy::unused_unit)] | ||
|
|
||
| use crate::db::postgres::models::token_models::token_utils::TokenWriteSet; | ||
| use aptos_protos::transaction::v1::WriteTableItem; | ||
| use bigdecimal::BigDecimal; | ||
| use serde::{Deserialize, Serialize}; | ||
|
|
||
| #[derive(Clone, Debug, Deserialize, Serialize, PartialEq, Eq)] | ||
| pub struct RawCurrentTokenRoyaltyV1 { | ||
| pub token_data_id: String, | ||
| pub payee_address: String, | ||
| pub royalty_points_numerator: BigDecimal, | ||
| pub royalty_points_denominator: BigDecimal, | ||
| pub last_transaction_version: i64, | ||
| pub last_transaction_timestamp: chrono::NaiveDateTime, | ||
| } | ||
|
|
||
| impl Ord for RawCurrentTokenRoyaltyV1 { | ||
| fn cmp(&self, other: &Self) -> std::cmp::Ordering { | ||
| self.token_data_id.cmp(&other.token_data_id) | ||
| } | ||
| } | ||
| impl PartialOrd for RawCurrentTokenRoyaltyV1 { | ||
| fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> { | ||
| Some(self.cmp(other)) | ||
| } | ||
| } | ||
|
|
||
| impl RawCurrentTokenRoyaltyV1 { | ||
| pub fn pk(&self) -> String { | ||
| self.token_data_id.clone() | ||
| } | ||
|
|
||
| // Royalty for v2 token is more complicated and not supported yet. For token v2, royalty can be on the collection (default) or on | ||
| // the token (override). | ||
| pub fn get_v1_from_write_table_item( | ||
| write_table_item: &WriteTableItem, | ||
| transaction_version: i64, | ||
| transaction_timestamp: chrono::NaiveDateTime, | ||
| ) -> anyhow::Result<Option<Self>> { | ||
| let table_item_data = write_table_item.data.as_ref().unwrap(); | ||
|
|
||
| let maybe_token_data = match TokenWriteSet::from_table_item_type( | ||
| table_item_data.value_type.as_str(), | ||
| &table_item_data.value, | ||
| transaction_version, | ||
| )? { | ||
| Some(TokenWriteSet::TokenData(inner)) => Some(inner), | ||
| _ => None, | ||
| }; | ||
|
|
||
| if let Some(token_data) = maybe_token_data { | ||
| let maybe_token_data_id = match TokenWriteSet::from_table_item_type( | ||
| table_item_data.key_type.as_str(), | ||
| &table_item_data.key, | ||
| transaction_version, | ||
| )? { | ||
| Some(TokenWriteSet::TokenDataId(inner)) => Some(inner), | ||
| _ => None, | ||
| }; | ||
| if let Some(token_data_id_struct) = maybe_token_data_id { | ||
| // token data id is the 0x{hash} version of the creator, collection name, and token name | ||
| let token_data_id = token_data_id_struct.to_id(); | ||
| let payee_address = token_data.royalty.get_payee_address(); | ||
| let royalty_points_numerator = token_data.royalty.royalty_points_numerator.clone(); | ||
| let royalty_points_denominator = | ||
| token_data.royalty.royalty_points_denominator.clone(); | ||
|
|
||
| return Ok(Some(Self { | ||
| token_data_id, | ||
| payee_address, | ||
| royalty_points_numerator, | ||
| royalty_points_denominator, | ||
| last_transaction_version: transaction_version, | ||
| last_transaction_timestamp: transaction_timestamp, | ||
| })); | ||
| } else { | ||
| tracing::warn!( | ||
| transaction_version, | ||
| key_type = table_item_data.key_type, | ||
| key = table_item_data.key, | ||
| "Expecting token_data_id as key for value = token_data" | ||
| ); | ||
| } | ||
| } | ||
| Ok(None) | ||
| } | ||
| } | ||
|
|
||
| pub trait CurrentTokenRoyaltyV1Convertible { | ||
| fn from_raw(raw_item: RawCurrentTokenRoyaltyV1) -> Self; | ||
| } |
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 +1,2 @@ | ||
| pub mod token_claims; | ||
| pub mod v1_token_royalty; |
60 changes: 60 additions & 0 deletions
60
rust/processor/src/db/parquet/models/token_v2_models/v1_token_royalty.rs
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,60 @@ | ||
| // Copyright © Aptos Foundation | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| // This is required because a diesel macro makes clippy sad | ||
| #![allow(clippy::extra_unused_lifetimes)] | ||
| #![allow(clippy::unused_unit)] | ||
|
|
||
| use crate::{ | ||
| bq_analytics::generic_parquet_processor::{GetTimeStamp, HasVersion, NamedTable}, | ||
| db::common::models::token_v2_models::raw_v1_token_royalty::{ | ||
| CurrentTokenRoyaltyV1Convertible, RawCurrentTokenRoyaltyV1, | ||
| }, | ||
| }; | ||
| use allocative_derive::Allocative; | ||
| use field_count::FieldCount; | ||
| use parquet_derive::ParquetRecordWriter; | ||
| use serde::{Deserialize, Serialize}; | ||
|
|
||
| #[derive( | ||
| Allocative, Clone, Debug, Default, Deserialize, FieldCount, ParquetRecordWriter, Serialize, | ||
| )] | ||
| pub struct CurrentTokenRoyaltyV1 { | ||
| pub token_data_id: String, | ||
| pub payee_address: String, | ||
| pub royalty_points_numerator: String, // String format of BigDecimal | ||
| pub royalty_points_denominator: String, // String format of BigDecimal | ||
| pub last_transaction_version: i64, | ||
| #[allocative(skip)] | ||
| pub last_transaction_timestamp: chrono::NaiveDateTime, | ||
| } | ||
|
|
||
| impl NamedTable for CurrentTokenRoyaltyV1 { | ||
| const TABLE_NAME: &'static str = "current_token_royalty_v1"; | ||
| } | ||
|
|
||
| impl HasVersion for CurrentTokenRoyaltyV1 { | ||
| fn version(&self) -> i64 { | ||
| self.last_transaction_version | ||
| } | ||
| } | ||
|
|
||
| impl GetTimeStamp for CurrentTokenRoyaltyV1 { | ||
| fn get_timestamp(&self) -> chrono::NaiveDateTime { | ||
| self.last_transaction_timestamp | ||
| } | ||
| } | ||
|
|
||
| impl CurrentTokenRoyaltyV1Convertible for CurrentTokenRoyaltyV1 { | ||
| // TODO: consider returning a Result | ||
| fn from_raw(raw_item: RawCurrentTokenRoyaltyV1) -> Self { | ||
| Self { | ||
| token_data_id: raw_item.token_data_id, | ||
| payee_address: raw_item.payee_address, | ||
| royalty_points_numerator: raw_item.royalty_points_numerator.to_string(), | ||
| royalty_points_denominator: raw_item.royalty_points_denominator.to_string(), | ||
| last_transaction_version: raw_item.last_transaction_version, | ||
| last_transaction_timestamp: raw_item.last_transaction_timestamp, | ||
| } | ||
| } | ||
| } |
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
Oops, something went wrong.