lock journal writer during memtable insert

for good measure

src/batch/mod.rs

@@ -111,8 +111,6 @@ impl Batch {
             }
         }
 
-        drop(journal_writer);
-
         // NOTE: Fully (write) lock, so the batch can be committed atomically
         log::trace!("batch: Acquiring partitions lock");
         let partitions = self.keyspace.partitions.write().expect("lock is poisoned");
@@ -174,6 +172,8 @@ impl Batch {
             partitions_with_possible_stall.insert(partition.clone());
         }
 
+        drop(journal_writer);
+
         drop(locked_memtables);
         drop(partitions);
 

src/partition/mod.rs

@@ -893,10 +893,10 @@ impl PartitionHandle {
                 })?;
         }
 
-        drop(journal_writer);
-
         let (item_size, memtable_size) = self.tree.insert(key, value, seqno);
 
+        drop(journal_writer);
+
         let write_buffer_size = self.write_buffer_manager.allocate(u64::from(item_size));
 
         self.check_memtable_overflow(memtable_size)?;
@@ -967,10 +967,10 @@ impl PartitionHandle {
                 })?;
         }
 
-        drop(journal_writer);
-
         let (item_size, memtable_size) = self.tree.remove(key, seqno);
 
+        drop(journal_writer);
+
         let write_buffer_size = self.write_buffer_manager.allocate(u64::from(item_size));
 
         self.check_memtable_overflow(memtable_size)?;