optimize get w/ zero-copy ref handle

Signed-off-by: Jason Volk <jason@zemos.net>

src/service/globals/data.rs

@@ -226,7 +226,7 @@ lasttimelinecount_cache: {lasttimelinecount_cache} / {max_lasttimelinecount_cach
 				self.global.insert(b"keypair", &keypair)?;
 				Ok::<_, Error>(keypair)
 			},
-			Ok,
+			|val| Ok(val.to_vec()),
 		)?;
 
 		let mut parts = keypair_bytes.splitn(2, |&b| b == 0xFF);

src/service/globals/migrations.rs

@@ -479,7 +479,7 @@ async fn db_lt_8(db: &Arc<Database>, _config: &Config) -> Result<()> {
 			.unwrap()
 			.expect("shortroomid should exist");
 
-		let mut new_key = short_room_id;
+		let mut new_key = short_room_id.to_vec();
 		new_key.extend_from_slice(count);
 
 		Some((new_key, v))
@@ -500,7 +500,7 @@ async fn db_lt_8(db: &Arc<Database>, _config: &Config) -> Result<()> {
 			.unwrap()
 			.expect("shortroomid should exist");
 
-		let mut new_value = short_room_id;
+		let mut new_value = short_room_id.to_vec();
 		new_value.extend_from_slice(count);
 
 		Some((k, new_value))
@@ -534,7 +534,7 @@ async fn db_lt_9(db: &Arc<Database>, _config: &Config) -> Result<()> {
 				.get(room_id)
 				.unwrap()
 				.expect("shortroomid should exist");
-			let mut new_key = short_room_id;
+			let mut new_key = short_room_id.to_vec();
 			new_key.extend_from_slice(word);
 			new_key.push(0xFF);
 			new_key.extend_from_slice(pdu_id_count);

src/service/rooms/alias/data.rs

@@ -39,7 +39,7 @@ impl Data {
 
 	pub(super) fn remove_alias(&self, alias: &RoomAliasId) -> Result<()> {
 		if let Some(room_id) = self.alias_roomid.get(alias.alias().as_bytes())? {
-			let mut prefix = room_id;
+			let mut prefix = room_id.to_vec();
 			prefix.push(0xFF);
 
 			for (key, _) in self.aliasid_alias.scan_prefix(prefix) {

src/service/rooms/event_handler/mod.rs

@@ -83,7 +83,7 @@ impl Service {
 	) -> Result<Option<Vec<u8>>> {
 		// 1. Skip the PDU if we already have it as a timeline event
 		if let Some(pdu_id) = services().rooms.timeline.get_pdu_id(event_id)? {
-			return Ok(Some(pdu_id));
+			return Ok(Some(pdu_id.to_vec()));
 		}
 
 		// 1.1 Check the server is in the room
@@ -449,7 +449,7 @@ impl Service {
 	) -> Result<Option<Vec<u8>>> {
 		// Skip the PDU if we already have it as a timeline event
 		if let Ok(Some(pduid)) = services().rooms.timeline.get_pdu_id(&incoming_pdu.event_id) {
-			return Ok(Some(pduid));
+			return Ok(Some(pduid.to_vec()));
 		}
 
 		if services()

src/service/rooms/timeline/data.rs

@@ -93,7 +93,7 @@ impl Data {
 	}
 
 	/// Returns the pdu's id.
-	pub(super) fn get_pdu_id(&self, event_id: &EventId) -> Result<Option<Vec<u8>>> {
+	pub(super) fn get_pdu_id(&self, event_id: &EventId) -> Result<Option<database::Handle<'_>>> {
 		self.eventid_pduid.get(event_id.as_bytes())
 	}
 

src/service/rooms/timeline/mod.rs

@@ -159,7 +159,9 @@ impl Service {
 	}
 
 	/// Returns the pdu's id.
-	pub fn get_pdu_id(&self, event_id: &EventId) -> Result<Option<Vec<u8>>> { self.db.get_pdu_id(event_id) }
+	pub fn get_pdu_id(&self, event_id: &EventId) -> Result<Option<database::Handle<'_>>> {
+		self.db.get_pdu_id(event_id)
+	}
 
 	/// Returns the pdu.
 	///
@@ -1185,7 +1187,8 @@ impl Service {
 
 		// Skip the PDU if we already have it as a timeline event
 		if let Some(pdu_id) = self.get_pdu_id(&event_id)? {
-			info!("We already know {event_id} at {pdu_id:?}");
+			let pdu_id = pdu_id.to_vec();
+			debug!("We already know {event_id} at {pdu_id:?}");
 			return Ok(());
 		}
 

src/service/transaction_ids/data.rs

@@ -31,7 +31,7 @@ impl Data {
 
 	pub(super) fn existing_txnid(
 		&self, user_id: &UserId, device_id: Option<&DeviceId>, txn_id: &TransactionId,
-	) -> Result<Option<Vec<u8>>> {
+	) -> Result<Option<database::Handle<'_>>> {
 		let mut key = user_id.as_bytes().to_vec();
 		key.push(0xFF);
 		key.extend_from_slice(device_id.map(DeviceId::as_bytes).unwrap_or_default());

src/service/transaction_ids/mod.rs

@@ -26,7 +26,7 @@ impl Service {
 
 	pub fn existing_txnid(
 		&self, user_id: &UserId, device_id: Option<&DeviceId>, txn_id: &TransactionId,
-	) -> Result<Option<Vec<u8>>> {
+	) -> Result<Option<database::Handle<'_>>> {
 		self.db.existing_txnid(user_id, device_id, txn_id)
 	}
 }