snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-09-06 12:46:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

add a new _geoPolygon filter to query the cellulite database

Browse Source
This commit is contained in:
Tamo
2025-07-16 00:50:03 +02:00
parent 05a13f662d
commit e510d4a8a3
4 changed files with 78 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
crates/milli/Cargo.toml
View File

@ -112,6 +112,7 @@ utoipa = { version = "5.4.0", features = [
"openapi_extensions",
] }
lru = "0.14.0"
geo-types = "0.7.16"
[dev-dependencies]
mimalloc = { version = "0.1.47", default-features = false }

33
crates/milli/src/search/facet/filter.rs
View File

@ -11,7 +11,7 @@ use roaring::{MultiOps, RoaringBitmap};
use serde_json::Value;
use super::facet_range_search;
use crate::constants::RESERVED_GEO_FIELD_NAME;
use crate::constants::{RESERVED_GEOJSON_FIELD_NAME, RESERVED_GEO_FIELD_NAME};
use crate::error::{Error, UserError};
use crate::filterable_attributes_rules::{filtered_matching_patterns, matching_features};
use crate::heed_codec::facet::{
@ -778,6 +778,37 @@ impl<'a> Filter<'a> {
))?
}
}
FilterCondition::GeoPolygon { points } => {
if index.is_geojson_enabled(rtxn)? {
let polygon = geo_types::Polygon::new(
geo_types::LineString(
points
.iter()
.map(|p| {
Ok(geo_types::Coord {
x: p[0].parse_finite_float()?,
y: p[1].parse_finite_float()?,
})
})
.collect::<Result<_, filter_parser::Error>>()?,
),
Vec::new(),
);
let cellulite = cellulite::Writer::new(index.cellulite);
let result = cellulite
.in_shape(rtxn, &polygon.into(), &mut |_| ())
.map_err(InternalError::CelluliteError)?;
Ok(result)
} else {
Err(points[0][0].as_external_error(FilterError::AttributeNotFilterable {
attribute: RESERVED_GEOJSON_FIELD_NAME,
filterable_patterns: filtered_matching_patterns(
filterable_attribute_rules,
&|features| features.is_filterable(),
),
}))?
}
}
}
}
}