0.7.19
This commit is contained in:
195
kb_lib/src/db/queries/wallet_holding.rs
Normal file
195
kb_lib/src/db/queries/wallet_holding.rs
Normal file
@@ -0,0 +1,195 @@
|
||||
// file: kb_lib/src/db/queries/wallet_holding.rs
|
||||
|
||||
//! Queries for `kb_wallet_holdings`.
|
||||
|
||||
/// Inserts or updates one wallet-holding row and returns its stable internal id.
|
||||
pub async fn upsert_wallet_holding(
|
||||
database: &crate::KbDatabase,
|
||||
dto: &crate::KbWalletHoldingDto,
|
||||
) -> Result<i64, crate::KbError> {
|
||||
match database.connection() {
|
||||
crate::KbDatabaseConnection::Sqlite(pool) => {
|
||||
let query_result = sqlx::query(
|
||||
r#"
|
||||
INSERT INTO kb_wallet_holdings (
|
||||
wallet_id,
|
||||
token_id,
|
||||
first_transaction_id,
|
||||
last_transaction_id,
|
||||
last_decoded_event_id,
|
||||
last_pool_id,
|
||||
last_pair_id,
|
||||
last_role,
|
||||
balance_raw,
|
||||
last_slot_observed,
|
||||
source_kind,
|
||||
source_endpoint_name,
|
||||
created_at,
|
||||
updated_at
|
||||
)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(wallet_id, token_id) DO UPDATE SET
|
||||
last_transaction_id = excluded.last_transaction_id,
|
||||
last_decoded_event_id = excluded.last_decoded_event_id,
|
||||
last_pool_id = excluded.last_pool_id,
|
||||
last_pair_id = excluded.last_pair_id,
|
||||
last_role = excluded.last_role,
|
||||
balance_raw = COALESCE(excluded.balance_raw, kb_wallet_holdings.balance_raw),
|
||||
last_slot_observed = excluded.last_slot_observed,
|
||||
source_kind = excluded.source_kind,
|
||||
source_endpoint_name = excluded.source_endpoint_name,
|
||||
updated_at = excluded.updated_at
|
||||
"#,
|
||||
)
|
||||
.bind(dto.wallet_id)
|
||||
.bind(dto.token_id)
|
||||
.bind(dto.first_transaction_id)
|
||||
.bind(dto.last_transaction_id)
|
||||
.bind(dto.last_decoded_event_id)
|
||||
.bind(dto.last_pool_id)
|
||||
.bind(dto.last_pair_id)
|
||||
.bind(dto.last_role.clone())
|
||||
.bind(dto.balance_raw.clone())
|
||||
.bind(dto.last_slot_observed)
|
||||
.bind(dto.source_kind.to_i16())
|
||||
.bind(dto.source_endpoint_name.clone())
|
||||
.bind(dto.created_at.to_rfc3339())
|
||||
.bind(dto.updated_at.to_rfc3339())
|
||||
.execute(pool)
|
||||
.await;
|
||||
if let Err(error) = query_result {
|
||||
return Err(crate::KbError::Db(format!(
|
||||
"cannot upsert kb_wallet_holdings on sqlite: {}",
|
||||
error
|
||||
)));
|
||||
}
|
||||
let id_result = sqlx::query_scalar::<sqlx::Sqlite, i64>(
|
||||
r#"
|
||||
SELECT id
|
||||
FROM kb_wallet_holdings
|
||||
WHERE wallet_id = ? AND token_id = ?
|
||||
LIMIT 1
|
||||
"#,
|
||||
)
|
||||
.bind(dto.wallet_id)
|
||||
.bind(dto.token_id)
|
||||
.fetch_one(pool)
|
||||
.await;
|
||||
match id_result {
|
||||
Ok(id) => Ok(id),
|
||||
Err(error) => Err(crate::KbError::Db(format!(
|
||||
"cannot fetch kb_wallet_holdings id for wallet_id '{}' token_id '{}' on sqlite: {}",
|
||||
dto.wallet_id, dto.token_id, error
|
||||
))),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Returns one wallet-holding row identified by `(wallet_id, token_id)`, if it exists.
|
||||
pub async fn get_wallet_holding_by_wallet_and_token(
|
||||
database: &crate::KbDatabase,
|
||||
wallet_id: i64,
|
||||
token_id: i64,
|
||||
) -> Result<std::option::Option<crate::KbWalletHoldingDto>, crate::KbError> {
|
||||
match database.connection() {
|
||||
crate::KbDatabaseConnection::Sqlite(pool) => {
|
||||
let query_result = sqlx::query_as::<sqlx::Sqlite, crate::KbWalletHoldingEntity>(
|
||||
r#"
|
||||
SELECT
|
||||
id,
|
||||
wallet_id,
|
||||
token_id,
|
||||
first_transaction_id,
|
||||
last_transaction_id,
|
||||
last_decoded_event_id,
|
||||
last_pool_id,
|
||||
last_pair_id,
|
||||
last_role,
|
||||
balance_raw,
|
||||
last_slot_observed,
|
||||
source_kind,
|
||||
source_endpoint_name,
|
||||
created_at,
|
||||
updated_at
|
||||
FROM kb_wallet_holdings
|
||||
WHERE wallet_id = ? AND token_id = ?
|
||||
LIMIT 1
|
||||
"#,
|
||||
)
|
||||
.bind(wallet_id)
|
||||
.bind(token_id)
|
||||
.fetch_optional(pool)
|
||||
.await;
|
||||
let entity_option = match query_result {
|
||||
Ok(entity_option) => entity_option,
|
||||
Err(error) => {
|
||||
return Err(crate::KbError::Db(format!(
|
||||
"cannot read kb_wallet_holdings by wallet_id '{}' token_id '{}' on sqlite: {}",
|
||||
wallet_id, token_id, error
|
||||
)));
|
||||
}
|
||||
};
|
||||
match entity_option {
|
||||
Some(entity) => crate::KbWalletHoldingDto::try_from(entity).map(Some),
|
||||
None => Ok(None),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Lists wallet-holding rows for one wallet id.
|
||||
pub async fn list_wallet_holdings_by_wallet_id(
|
||||
database: &crate::KbDatabase,
|
||||
wallet_id: i64,
|
||||
) -> Result<std::vec::Vec<crate::KbWalletHoldingDto>, crate::KbError> {
|
||||
match database.connection() {
|
||||
crate::KbDatabaseConnection::Sqlite(pool) => {
|
||||
let query_result = sqlx::query_as::<sqlx::Sqlite, crate::KbWalletHoldingEntity>(
|
||||
r#"
|
||||
SELECT
|
||||
id,
|
||||
wallet_id,
|
||||
token_id,
|
||||
first_transaction_id,
|
||||
last_transaction_id,
|
||||
last_decoded_event_id,
|
||||
last_pool_id,
|
||||
last_pair_id,
|
||||
last_role,
|
||||
balance_raw,
|
||||
last_slot_observed,
|
||||
source_kind,
|
||||
source_endpoint_name,
|
||||
created_at,
|
||||
updated_at
|
||||
FROM kb_wallet_holdings
|
||||
WHERE wallet_id = ?
|
||||
ORDER BY token_id ASC, id ASC
|
||||
"#,
|
||||
)
|
||||
.bind(wallet_id)
|
||||
.fetch_all(pool)
|
||||
.await;
|
||||
let entities = match query_result {
|
||||
Ok(entities) => entities,
|
||||
Err(error) => {
|
||||
return Err(crate::KbError::Db(format!(
|
||||
"cannot list kb_wallet_holdings by wallet_id '{}' on sqlite: {}",
|
||||
wallet_id, error
|
||||
)));
|
||||
}
|
||||
};
|
||||
let mut dtos = std::vec::Vec::new();
|
||||
for entity in entities {
|
||||
let dto_result = crate::KbWalletHoldingDto::try_from(entity);
|
||||
let dto = match dto_result {
|
||||
Ok(dto) => dto,
|
||||
Err(error) => return Err(error),
|
||||
};
|
||||
dtos.push(dto);
|
||||
}
|
||||
Ok(dtos)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user