Stardog Patches: WriteBatch + WriteBufferManager

stardog-union · Feb 4, 2021 · a2b2f7e · a2b2f7e
1 parent fbed72f
commit a2b2f7e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/include/rocksdb/write_batch.h b/include/rocksdb/write_batch.h
@@ -173,6 +173,10 @@ class WriteBatch : public WriteBatchBase {
   // Otherwise returns Status::OK().
   Status PopSavePoint() override;
 
+  void setContentFlag(uint32_t theContentFlag) {
+    content_flags_.store(theContentFlag,std::memory_order_seq_cst);
+  }
+
   // Support for iterating over the contents of a batch.
   class Handler {
    public:

diff --git a/include/rocksdb/write_buffer_manager.h b/include/rocksdb/write_buffer_manager.h
@@ -87,8 +87,12 @@ class WriteBufferManager {
     }
   }
 
+  void SetBufferSize(size_t new_size) {
+    buffer_size_ = new_size;
+  }
+
  private:
-  const size_t buffer_size_;
+  std::atomic<size_t> buffer_size_;
   const size_t mutable_limit_;
   std::atomic<size_t> memory_used_;
   // Memory that hasn't been scheduled to free.