simplify cork interface related

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

src/service/globals/data.rs

@@ -4,7 +4,7 @@ use std::{
 };
 
 use conduit::{trace, utils, Error, Result};
-use database::{Cork, Database, Map};
+use database::{Database, Map};
 use futures_util::{stream::FuturesUnordered, StreamExt};
 use lru_cache::LruCache;
 use ruma::{
@@ -185,10 +185,6 @@ impl Data {
 
 	pub fn cleanup(&self) -> Result<()> { self.db.db.cleanup() }
 
-	pub fn cork(&self) -> Cork { Cork::new(&self.db.db, false, false) }
-
-	pub fn cork_and_flush(&self) -> Cork { Cork::new(&self.db.db, true, false) }
-
 	pub fn memory_usage(&self) -> String {
 		let auth_chain_cache = self.db.auth_chain_cache.lock().unwrap().len();
 		let appservice_in_room_cache = self.db.appservice_in_room_cache.read().unwrap().len();

src/service/globals/migrations.rs

@@ -818,7 +818,7 @@ async fn handle_media_check(
 async fn fix_bad_double_separator_in_state_cache(db: &Arc<Database>, _config: &Config) -> Result<()> {
 	warn!("Fixing bad double separator in state_cache roomuserid_joined");
 	let roomuserid_joined = &db["roomuserid_joined"];
-	let _cork = database::Cork::new(&db.db, true, true);
+	let _cork = db.cork_and_sync();
 
 	let mut iter_count: usize = 0;
 	for (mut key, value) in roomuserid_joined.iter() {
@@ -851,7 +851,7 @@ async fn fix_bad_double_separator_in_state_cache(db: &Arc<Database>, _config: &C
 
 async fn retroactively_fix_bad_data_from_roomuserid_joined(db: &Arc<Database>, _config: &Config) -> Result<()> {
 	warn!("Retroactively fixing bad data from broken roomuserid_joined");
-	let _cork = database::Cork::new(&db.db, true, true);
+	let _cork = db.cork_and_sync();
 
 	let room_ids = services()
 		.rooms

src/service/rooms/timeline/mod.rs

@@ -207,7 +207,7 @@ impl Service {
 		state_lock: &mutex_map::Guard<()>, // Take mutex guard to make sure users get the room state mutex
 	) -> Result<Vec<u8>> {
 		// Coalesce database writes for the remainder of this scope.
-		let _cork = services().globals.db.cork_and_flush();
+		let _cork = services().db.cork_and_flush();
 
 		let shortroomid = services()
 			.rooms

src/service/sending/mod.rs

@@ -85,7 +85,7 @@ impl Service {
 	pub fn send_pdu_push(&self, pdu_id: &[u8], user: &UserId, pushkey: String) -> Result<()> {
 		let dest = Destination::Push(user.to_owned(), pushkey);
 		let event = SendingEvent::Pdu(pdu_id.to_owned());
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		let keys = self.db.queue_requests(&[(&dest, event.clone())])?;
 		self.dispatch(Msg {
 			dest,
@@ -98,7 +98,7 @@ impl Service {
 	pub fn send_pdu_appservice(&self, appservice_id: String, pdu_id: Vec<u8>) -> Result<()> {
 		let dest = Destination::Appservice(appservice_id);
 		let event = SendingEvent::Pdu(pdu_id);
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		let keys = self.db.queue_requests(&[(&dest, event.clone())])?;
 		self.dispatch(Msg {
 			dest,
@@ -125,7 +125,7 @@ impl Service {
 			.into_iter()
 			.map(|server| (Destination::Normal(server), SendingEvent::Pdu(pdu_id.to_owned())))
 			.collect::<Vec<_>>();
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		let keys = self.db.queue_requests(
 			&requests
 				.iter()
@@ -147,7 +147,7 @@ impl Service {
 	pub fn send_edu_server(&self, server: &ServerName, serialized: Vec<u8>) -> Result<()> {
 		let dest = Destination::Normal(server.to_owned());
 		let event = SendingEvent::Edu(serialized);
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		let keys = self.db.queue_requests(&[(&dest, event.clone())])?;
 		self.dispatch(Msg {
 			dest,
@@ -174,7 +174,7 @@ impl Service {
 			.into_iter()
 			.map(|server| (Destination::Normal(server), SendingEvent::Edu(serialized.clone())))
 			.collect::<Vec<_>>();
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		let keys = self.db.queue_requests(
 			&requests
 				.iter()

src/service/sending/sender.rs

@@ -100,7 +100,7 @@ impl Service {
 	fn handle_response_ok(
 		&self, dest: &Destination, futures: &SendingFutures<'_>, statuses: &mut CurTransactionStatus,
 	) {
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		self.db
 			.delete_all_active_requests_for(dest)
 			.expect("all active requests deleted");
@@ -173,7 +173,7 @@ impl Service {
 			return Ok(None);
 		}
 
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		let mut events = Vec::new();
 
 		// Must retry any previous transaction for this remote.
@@ -187,7 +187,7 @@ impl Service {
 		}
 
 		// Compose the next transaction
-		let _cork = services().globals.db.cork();
+		let _cork = services().db.cork();
 		if !new_events.is_empty() {
 			self.db.mark_as_active(&new_events)?;
 			for (e, _) in new_events {