Add custom global allocator (#30)

matthewmturner · web-flow · commit 368b50ed9662 · 2022-02-25T20:23:52.000+08:00
* Add sn-malloc

* Mimalloc
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -35,6 +35,7 @@ datafusion = { version = "^7.0.0", features = ["pyarrow"] }
 datafusion-expr = { version = "^7.0.0" }
 datafusion-common = { version = "^7.0.0", features = ["pyarrow"] }
 uuid = { version = "0.8", features = ["v4"] }
+mimalloc = { version = "*", default-features = false }
 
 [lib]
 name = "_internal"
diff --git a/src/lib.rs b/src/lib.rs
@@ -15,6 +15,7 @@
 // specific language governing permissions and limitations
 // under the License.
 
+use mimalloc::MiMalloc;
 use pyo3::prelude::*;
 
 mod catalog;
@@ -27,6 +28,9 @@ mod udaf;
 mod udf;
 mod utils;
 
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 /// Low-level DataFusion internal package.
 ///
 /// The higher-level public API is defined in pure python files under the
