sqlx/sqlx-postgres-protocol/benches/encode.rs

#[macro_use]
extern crate criterion;

use criterion::Criterion;
use sqlx_postgres_protocol::{Encode, PasswordMessage, Query, StartupMessage, Terminate};

fn criterion_benchmark(c: &mut Criterion) {
    c.bench_function("encode PasswordMessage::cleartext", |b| {
        let mut dst = Vec::with_capacity(1024);
        b.iter(|| {
            dst.clear();
            PasswordMessage::cleartext("8e323AMF9YSE9zftFnuhQcvhz7Vf342W4cWU")
                .encode(&mut dst)
                .unwrap();
        })
    });

    c.bench_function("encode Query", |b| {
        let mut dst = Vec::with_capacity(1024);
        b.iter(|| {
            dst.clear();
            Query::new("SELECT 1, 2, 3").encode(&mut dst).unwrap();
        })
    });

    c.bench_function("encode Terminate", |b| {
        let mut dst = Vec::with_capacity(1024);
        b.iter(|| {
            dst.clear();
            Terminate.encode(&mut dst).unwrap();
        })
    });

    c.bench_function("encode StartupMessage", |b| {
        let mut dst = Vec::with_capacity(1024);
        b.iter(|| {
            dst.clear();
            StartupMessage::new(&[
                ("user", "postgres"),
                ("database", "postgres"),
                ("DateStyle", "ISO, MDY"),
                ("IntervalStyle", "iso_8601"),
                ("TimeZone", "UTC"),
                ("extra_float_digits", "3"),
                ("client_encoding", "UTF-8"),
            ])
            .encode(&mut dst)
            .unwrap();
        })
    });

    c.bench_function("encode Password::md5", |b| {
        let mut dst = Vec::with_capacity(1024);
        b.iter(|| {
            dst.clear();
            PasswordMessage::md5(
                "8e323AMF9YSE9zftFnuhQcvhz7Vf342W4cWU",
                "postgres",
                [10, 41, 20, 150],
            )
            .encode(&mut dst)
            .unwrap();
        })
    });
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);