feat(qe): use mimalloc instead of the system allocator

query-engine/query-engine-node-api/Cargo.toml

@@ -54,6 +54,8 @@ tokio.workspace = true
 futures = "0.3"
 query-engine-metrics = { path = "../metrics" }
 
+mimalloc = "0.1.43"
+
 [build-dependencies]
 napi-build = "1"
 build-utils.path = "../../libs/build-utils"

query-engine/query-engine-node-api/src/lib.rs

@@ -1,3 +1,5 @@
+use mimalloc::MiMalloc;
+
 pub mod engine;
 pub mod error;
 pub mod functions;
@@ -6,3 +8,6 @@ pub mod logger;
 mod tracer;
 
 pub(crate) type Result<T> = std::result::Result<T, error::ApiError>;
+
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;

query-engine/query-engine/Cargo.toml

@@ -17,6 +17,7 @@ connector = { path = "../connectors/query-connector", package = "query-connector
 enumflags2.workspace = true
 psl = { workspace = true, features = ["all"] }
 graphql-parser = { git = "https://github.com/prisma/graphql-parser" }
+mimalloc = "0.1.43"
 mongodb-connector = { path = "../connectors/mongodb-query-connector", optional = true, package = "mongodb-query-connector" }
 query-core = { path = "../core", features = ["metrics"] }
 request-handlers = { path = "../request-handlers", features = ["all"] }

query-engine/query-engine/src/main.rs

@@ -3,6 +3,7 @@
 #[macro_use]
 extern crate tracing;
 
+use mimalloc::MiMalloc;
 use query_engine::cli::CliCommand;
 use query_engine::context;
 use query_engine::error::PrismaError;
@@ -13,6 +14,9 @@ use std::{error::Error, process};
 use structopt::StructOpt;
 use tracing::Instrument;
 
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 type AnyError = Box<dyn Error + Send + Sync + 'static>;
 
 #[tokio::main]