mirror of
https://github.com/maplibre/martin.git
synced 2024-12-20 13:21:59 +03:00
def4a6f1ec
* Detect all unrecognized config file values, and report them. Ideally we want to use `serde-ignored` crate, but it doesn't work with flattened structs (yet). So using a bad workaround. * CI test has been using all sorts of somewhat duplicated temporary pbf files - cleaned up to `tmp.pbf`, and made sure curl only shows errors, not download stats. * In CI, crash psql instead of silently ignoring errors * Don't serialize optional config values as nulls * Tiny error message cleanup
120 lines
3.6 KiB
Rust
120 lines
3.6 KiB
Rust
use criterion::{criterion_group, criterion_main, Criterion};
|
|
use martin::pg::composite_source::CompositeSource;
|
|
use martin::pg::dev::make_pool;
|
|
use martin::pg::function_source::FunctionSource;
|
|
use martin::pg::table_source::TableSource;
|
|
use martin::source::{Source, Xyz};
|
|
use std::collections::HashMap;
|
|
|
|
fn mock_table_source(schema: &str, table: &str) -> TableSource {
|
|
TableSource {
|
|
id: format!("{schema}.{table}"),
|
|
schema: schema.to_owned(),
|
|
table: table.to_owned(),
|
|
id_column: None,
|
|
geometry_column: "geom".to_owned(),
|
|
minzoom: None,
|
|
maxzoom: None,
|
|
bounds: None,
|
|
srid: 3857,
|
|
extent: Some(4096),
|
|
buffer: Some(64),
|
|
clip_geom: Some(true),
|
|
geometry_type: None,
|
|
properties: HashMap::new(),
|
|
unrecognized: HashMap::new(),
|
|
}
|
|
}
|
|
|
|
fn mock_function_source(schema: &str, function: &str) -> FunctionSource {
|
|
FunctionSource {
|
|
id: format!("{schema}.{function}"),
|
|
schema: schema.to_owned(),
|
|
function: function.to_owned(),
|
|
minzoom: None,
|
|
maxzoom: None,
|
|
bounds: None,
|
|
unrecognized: HashMap::new(),
|
|
}
|
|
}
|
|
|
|
async fn get_table_source() {
|
|
let source = mock_table_source("public", "table_source");
|
|
let _tilejson = source.get_tilejson();
|
|
}
|
|
|
|
async fn get_table_source_tile() {
|
|
let pool = make_pool().await;
|
|
let mut connection = pool.get().await.unwrap();
|
|
|
|
let source = mock_table_source("public", "table_source");
|
|
let xyz = Xyz { z: 0, x: 0, y: 0 };
|
|
|
|
let _tile = source.get_tile(&mut connection, &xyz, &None).await.unwrap();
|
|
}
|
|
|
|
async fn get_composite_source() {
|
|
let points1 = mock_table_source("public", "points1");
|
|
let points2 = mock_table_source("public", "points2");
|
|
|
|
let source = CompositeSource {
|
|
id: "public.points1,public.points2".to_owned(),
|
|
table_sources: vec![points1, points2],
|
|
};
|
|
|
|
let _tilejson = source.get_tilejson();
|
|
}
|
|
|
|
async fn get_composite_source_tile() {
|
|
let pool = make_pool().await;
|
|
let mut connection = pool.get().await.unwrap();
|
|
|
|
let points1 = mock_table_source("public", "points1");
|
|
let points2 = mock_table_source("public", "points2");
|
|
|
|
let source = CompositeSource {
|
|
id: "public.points1,public.points2".to_owned(),
|
|
table_sources: vec![points1, points2],
|
|
};
|
|
|
|
let xyz = Xyz { z: 0, x: 0, y: 0 };
|
|
let _tile = source.get_tile(&mut connection, &xyz, &None).await.unwrap();
|
|
}
|
|
|
|
async fn get_function_source() {
|
|
let source = mock_function_source("public", "function_source");
|
|
let _tilejson = source.get_tilejson();
|
|
}
|
|
|
|
async fn get_function_source_tile() {
|
|
let pool = make_pool().await;
|
|
let mut connection = pool.get().await.unwrap();
|
|
|
|
let source = mock_function_source("public", "function_source");
|
|
let xyz = Xyz { z: 0, x: 0, y: 0 };
|
|
|
|
let _tile = source.get_tile(&mut connection, &xyz, &None).await.unwrap();
|
|
}
|
|
|
|
fn table_source(c: &mut Criterion) {
|
|
c.bench_function("get_table_source", |b| b.iter(get_table_source));
|
|
c.bench_function("get_table_source_tile", |b| b.iter(get_table_source_tile));
|
|
}
|
|
|
|
fn composite_source(c: &mut Criterion) {
|
|
c.bench_function("get_composite_source", |b| b.iter(get_composite_source));
|
|
c.bench_function("get_composite_source_tile", |b| {
|
|
b.iter(get_composite_source_tile);
|
|
});
|
|
}
|
|
|
|
fn function_source(c: &mut Criterion) {
|
|
c.bench_function("get_function_source", |b| b.iter(get_function_source));
|
|
c.bench_function("get_function_source_tile", |b| {
|
|
b.iter(get_function_source_tile);
|
|
});
|
|
}
|
|
|
|
criterion_group!(benches, table_source, composite_source, function_source);
|
|
criterion_main!(benches);
|