1
0
Fork 0
mirror of https://forgejo.ellis.link/continuwuation/continuwuity.git synced 2025-07-31 12:18:31 +00:00
continuwuity/src/database/key_value/transaction_ids.rs

36 lines
1 KiB
Rust
Raw Normal View History

impl service::pusher::Data for KeyValueDatabase {
2020-08-25 13:24:38 +02:00
pub fn add_txnid(
&self,
user_id: &UserId,
device_id: Option<&DeviceId>,
2022-01-17 14:35:38 +01:00
txn_id: &TransactionId,
2020-08-25 13:24:38 +02:00
data: &[u8],
) -> Result<()> {
let mut key = user_id.as_bytes().to_vec();
key.push(0xff);
key.extend_from_slice(device_id.map(|d| d.as_bytes()).unwrap_or_default());
2020-08-25 13:24:38 +02:00
key.push(0xff);
key.extend_from_slice(txn_id.as_bytes());
2021-06-08 18:10:00 +02:00
self.userdevicetxnid_response.insert(&key, data)?;
2020-08-25 13:24:38 +02:00
Ok(())
}
pub fn existing_txnid(
&self,
user_id: &UserId,
device_id: Option<&DeviceId>,
2022-01-17 14:35:38 +01:00
txn_id: &TransactionId,
2021-06-08 18:10:00 +02:00
) -> Result<Option<Vec<u8>>> {
2020-08-25 13:24:38 +02:00
let mut key = user_id.as_bytes().to_vec();
key.push(0xff);
key.extend_from_slice(device_id.map(|d| d.as_bytes()).unwrap_or_default());
2020-08-25 13:24:38 +02:00
key.push(0xff);
key.extend_from_slice(txn_id.as_bytes());
// If there's no entry, this is a new transaction
2021-06-17 20:34:14 +02:00
self.userdevicetxnid_response.get(&key)
2020-08-25 13:24:38 +02:00
}
}