Fix operational point search performance issues

editoast/editoast_derive/src/search.rs

@@ -16,6 +16,8 @@ use syn::DeriveInput;
 )]
 struct SearchParams {
     table: String,
+    #[darling(default)]
+    distinct_on: Option<String>,
     migration: Option<Migration>,
     #[darling(default)]
     joins: String,
@@ -165,6 +167,9 @@ pub fn expand_search(input: &DeriveInput) -> Result<TokenStream> {
 
     let struct_name = &input.ident;
     let table = params.table;
+    let distinct_on = params
+        .distinct_on
+        .map_or_else(|| quote! { None }, |d| quote! { Some(#d.to_owned()) });
     let joins = match params.joins {
         ref joins if !joins.is_empty() => quote! { Some(#joins.to_owned()) },
         _ => quote! { None },
@@ -286,6 +291,7 @@ pub fn expand_search(input: &DeriveInput) -> Result<TokenStream> {
             fn search_config() -> editoast_search::SearchConfig {
                 editoast_search::SearchConfig {
                     table: #table.to_owned(),
+                    distinct_on: #distinct_on,
                     joins: #joins,
                     criterias: Vec::from([#criterias]),
                     properties: Vec::from([#properties]),

editoast/editoast_derive/src/snapshots/editoast_derive__search__tests__search_construction.snap

@@ -26,6 +26,7 @@ impl editoast_search::SearchObject for Track {
     fn search_config() -> editoast_search::SearchConfig {
         editoast_search::SearchConfig {
             table: "search_track".to_owned(),
+            distinct_on: None,
             joins: None,
             criterias: Vec::from([
                 editoast_search::Criteria {

editoast/editoast_search/src/lib.rs

@@ -71,13 +71,17 @@ pub fn query_into_sql(
     }
     let where_expression = context.search_ast_to_sql(&ast)?;
     let table = &search_config.table;
+    let select = search_config.distinct_on.as_ref().map_or_else(
+        || "SELECT".to_owned(),
+        |columns| format!("SELECT DISTINCT ON {columns}"),
+    );
     let joins = search_config.joins.as_ref().cloned().unwrap_or_default();
     let result_columns = search_config.result_columns();
     let mut bindings = Default::default();
     let constraints = where_expression.to_sql(&mut bindings);
     let sql_code = format!(
         "WITH _RESULT AS (
-            SELECT {result_columns}
+            {select} {result_columns}
             FROM {table}
             {joins}
             WHERE {constraints}

editoast/editoast_search/src/search_object.rs

@@ -18,6 +18,7 @@ pub struct Property {
 
 pub struct SearchConfig {
     pub table: String,
+    pub distinct_on: Option<String>,
     pub criterias: Vec<Criteria>,
     pub properties: Vec<Property>,
     pub joins: Option<String>,

editoast/src/views/search.rs

@@ -430,11 +430,11 @@ pub(super) struct SearchResultItemTrack {
 #[derive(Search, Serialize, ToSchema)]
 #[search(
     table = "search_operational_point",
+    distinct_on = "(infra_id, obj_id)",
     migration(src_table = "infra_object_operational_point"),
     joins = "
         INNER JOIN infra_object_operational_point AS OP ON OP.id = search_operational_point.id
-        INNER JOIN (SELECT DISTINCT ON (infra_id, obj_id) * FROM infra_layer_operational_point)
-            AS lay ON OP.obj_id = lay.obj_id AND OP.infra_id = lay.infra_id",
+        INNER JOIN infra_layer_operational_point AS lay ON OP.obj_id = lay.obj_id AND OP.infra_id = lay.infra_id",
     column(
         name = "obj_id",
         data_type = "varchar(255)",