Attempt the non-blocking write in bg

restatedev · May 28, 2024 · 8bc0ad1 · 8bc0ad1
1 parent 6b7c0dd
commit 8bc0ad1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/crates/rocksdb/src/lib.rs b/crates/rocksdb/src/lib.rs
@@ -164,7 +164,9 @@ impl RocksDb {
  // Auto...
  // First, attempt to write without blocking
  write_options.set_no_slowdown(true);
- let result = self.db.write_batch(&write_batch, &write_options);
+
+ let result =
+ tokio::task::block_in_place(|| self.db.write_batch(&write_batch, &write_options));
  match result {
  Ok(_) => {
  counter!(STORAGE_IO_OP,
@@ -247,7 +249,8 @@ impl RocksDb {
  // Auto...
  // First, attempt to write without blocking
  write_options.set_no_slowdown(true);
- let result = self.db.write_tx_batch(&write_batch, &write_options);
+ let result =
+ tokio::task::block_in_place(|| self.db.write_tx_batch(&write_batch, &write_options));
  match result {
  Ok(_) => {
  counter!(STORAGE_IO_OP,