perf(qe): Examples useful for performance analysis of connecting to D…

…B servers (prisma#3892)

* Update benchmarking/profiling scripts

* Add a few useful examples for benchmarking / profiling

* Update Dockerfile to profile code snippets in typescript too

Makefile

@@ -12,13 +12,16 @@ LIBRARY_EXT := $(shell                            \
 
 default: build
 
-#####################
-# Boostrap commands #
-#####################
+###################
+# script wrappers #
+###################
 
 bootstrap-darwin:
 	script/bootstrap-darwin
 
+profile-shell:
+	script/profile-shell
+
 ##################
 # Build commands #
 ##################

query-engine/connectors/sql-query-connector/Cargo.toml

@@ -49,3 +49,8 @@ version = "1.2"
 [dependencies.user-facing-errors]
 features = ["sql"]
 path = "../../../libs/user-facing-errors"
+
+[dev-dependencies]
+postgres = "0.19"
+tokio-postgres = "0.7"
+mysql_async = { git="https://github.com/prisma/mysql_async", branch="vendored-openssl" }

query-engine/connectors/sql-query-connector/examples/mysql_async.rs

@@ -0,0 +1,21 @@
+use mysql_async::prelude::*;
+use std::time::Instant;
+
+#[tokio::main]
+async fn main() -> () {
+    let url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set to a mysql URL");
+    let start = Instant::now();
+
+    let now = Instant::now();
+    let pool = mysql_async::Pool::from_url(&url).unwrap();
+    println!("Building pool: {:?}", now.elapsed());
+
+    let mut conn = pool.get_conn().await.unwrap();
+    println!("Checking out connection: {:?}", now.elapsed());
+
+    let now = Instant::now();
+    let _: Vec<u8> = conn.query("SELECT 1").await.unwrap();
+    println!("Query: {:?}", now.elapsed());
+
+    println!("Total: {:?}", start.elapsed());
+}

query-engine/connectors/sql-query-connector/examples/quaint_query.rs

@@ -0,0 +1,26 @@
+use std::time::Instant;
+
+use quaint::{pooled::Quaint, prelude::Queryable};
+
+#[tokio::main]
+async fn main() -> () {
+    let url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set to a quaint-compatible database URL");
+    let start = Instant::now();
+
+    let now = Instant::now();
+    let mut builder = Quaint::builder(&url).expect("should connect");
+    builder.health_check_interval(std::time::Duration::from_secs(15));
+    builder.test_on_check_out(true);
+    let pool = builder.build();
+    println!("Building pool: {:?}", now.elapsed());
+
+    let now = Instant::now();
+    let conn = pool.check_out().await.unwrap();
+    println!("Checking out connection: {:?}", now.elapsed());
+
+    let now = Instant::now();
+    conn.execute_raw("SELECT 1", &[]).await.unwrap();
+    println!("Query: {:?}", now.elapsed());
+
+    println!("Total: {:?}", start.elapsed());
+}

query-engine/connectors/sql-query-connector/examples/tokio_pg.rs

@@ -0,0 +1,25 @@
+use std::time::Instant;
+
+#[tokio::main]
+async fn main() -> () {
+    let url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set to a postgres URL");
+    let start = Instant::now();
+
+    let now = Instant::now();
+    let (client, connection) = tokio_postgres::connect(&url, postgres::NoTls).await.unwrap();
+    println!("Connect: {:?}", now.elapsed());
+
+    tokio::spawn(async move {
+        let now = Instant::now();
+        if let Err(e) = connection.await {
+            eprintln!("connection error: {}", e);
+        }
+        eprintln!("connection time: {:?}", now.elapsed());
+    });
+
+    let now = Instant::now();
+    client.query("SELECT 1;", &[]).await.unwrap();
+    println!("Query: {:?}", now.elapsed());
+
+    println!("Total: {:?}", start.elapsed());
+}

script/.profile-shell/Dockerfile

@@ -0,0 +1,33 @@
+ARG RUST_VERSION
+FROM rust:$RUST_VERSION-buster
+RUN printf "deb http://ftp.de.debian.org/debian buster main" >> /etc/apt/sources.list 
+RUN apt update
+RUN apt-get install -y git valgrind linux-perf strace
+
+RUN curl -fsSL https://fnm.vercel.app/install | bash
+RUN source ~/.bashrc
+RUN fnm install v16
+
+# Symlink caches
+RUN mkdir -p /.cargo/registry
+RUN mkdir .cargo/git
+RUN rm -rf /usr/local/cargo/git \
+    /usr/local/cargo/registry \
+    /usr/local/cargo/.package-cache \
+    /usr/local/cargo/.crates.toml \
+    /usr/local/cargo/.crates2.json
+RUN ln -s /.cargo/git /usr/local/cargo
+RUN ln -s /.cargo/registry /usr/local/cargo
+RUN ln -s /.cargo/.package-cache /usr/local/cargo
+RUN ln -s /.cargo/.crates.toml /usr/local/cargo
+RUN ln -s /.cargo/.crates2.json /usr/local/cargo
+
+# Set env
+ENV PATH="${PATH}:/prisma-engines/script/.profile-shell/bin"
+ENV VALGRIND_LIB=/usr/lib/aarch64-linux-gnu/valgrind/
+ENV CARGO_TARGET_DIR=target-alternatives
+
+# Install cargo-flamegraph
+RUN cargo install flamegraph
+
+CMD ["/bin/bash"]

script/.profile-shell/bin/build

@@ -0,0 +1,4 @@
+#/bin/bash
+CMD="cargo build --profile=profiling --examples --benches"
+echo "==> Running command: $CMD"
+cd /prisma-engines && eval "$CMD"