use criterion::{Criterion, black_box, criterion_group, criterion_main};
use jsonwebtoken::{Algorithm, DecodingKey, EncodingKey, Header, Validation, decode, encode};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
struct Claims {
sub: String,
company: String,
}
fn bench_encode(c: &mut Criterion) {
let claim = Claims { sub: "b@b.com".to_owned(), company: "ACME".to_owned() };
let key = EncodingKey::from_secret("secret".as_ref());
c.bench_function("bench_encode", |b| {
b.iter(|| encode(black_box(&Header::default()), black_box(&claim), black_box(&key)))
});
}
fn bench_encode_custom_extra_headers(c: &mut Criterion) {
let claim = Claims { sub: "b@b.com".to_owned(), company: "ACME".to_owned() };
let key = EncodingKey::from_secret("secret".as_ref());
let mut extras = HashMap::with_capacity(1);
extras.insert("custom".to_string(), "header".to_string());
let header = &Header { extras, ..Default::default() };
c.bench_function("bench_encode", |b| {
b.iter(|| encode(black_box(header), black_box(&claim), black_box(&key)))
});
}
fn bench_decode(c: &mut Criterion) {
let token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ";
let key = DecodingKey::from_secret("secret".as_ref());
c.bench_function("bench_decode", |b| {
b.iter(|| {
decode::<Claims>(
black_box(token),
black_box(&key),
black_box(&Validation::new(Algorithm::HS256)),
)
})
});
}
criterion_group!(benches, bench_encode, bench_encode_custom_extra_headers, bench_decode);
criterion_main!(benches);