sled can create directly in a temporary file

sozu-proxy · Jul 13, 2022 · f12bacb · f12bacb
1 parent 8ffca7f
commit f12bacb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/lib/Cargo.toml b/lib/Cargo.toml
@@ -48,7 +48,6 @@ regex = "1.1"
 cookie-factory = "^0.3"
 hpack = "^0.3"
 sled = "0.34"
-tempfile = "^3"
 
 [dev-dependencies]
 tiny_http = "^0.8"

diff --git a/lib/src/metrics/local_drain.rs b/lib/src/metrics/local_drain.rs
@@ -106,7 +106,6 @@ pub struct LocalDrain {
   pub prefix:          String,
   pub created:         Instant,
   pub db:              sled::Db,
-  pub data_dir:        tempfile::TempDir,
   pub data:            BTreeMap<String, AggregatedMetric>,
   metrics:             BTreeMap<String, (MetricMeta, MetricKind)>,
   use_tagged_metrics:  bool,
@@ -115,13 +114,15 @@ pub struct LocalDrain {
 
 impl LocalDrain {
   pub fn new(prefix: String) -> Self {
-    let data_dir = tempfile::TempDir::new().unwrap();
-    let db = sled::open(data_dir.path()).unwrap();
+    let db = sled::Config::new()
+        .temporary(true)
+        .mode(sled::Mode::LowSpace)
+        .open()
+        .unwrap();
 
     LocalDrain {
       prefix,
       created:     Instant::now(),
-      data_dir,
       db,
       metrics:     BTreeMap::new(),
       data:        BTreeMap::new(),