H-2008: Fix root sorting by embedding distance (#3934)

hashintel · Jan 25, 2024 · 6c15374 · 6c15374
1 parent 57be107
commit 6c15374
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/apps/hash-graph/libs/graph/src/store/postgres/query/compile.rs b/apps/hash-graph/libs/graph/src/store/postgres/query/compile.rs
@@ -471,9 +471,11 @@ impl<'p, R: PostgresRecord> SelectCompiler<'p, R> {
                         });
                     }
 
-                    self.statement
-                        .order_by_expression
-                        .push(distance_column, Ordering::Ascending);
+                    self.statement.order_by_expression.insert(
+                        0,
+                        distance_column,
+                        Ordering::Ascending,
+                    );
                     self.statement
                         .selects
                         .push(SelectExpression::from_column(distance_column, None));

diff --git a/apps/hash-graph/libs/graph/src/store/postgres/query/expression/order_clause.rs b/apps/hash-graph/libs/graph/src/store/postgres/query/expression/order_clause.rs
@@ -18,6 +18,10 @@ impl OrderByExpression {
         self.columns.push((column, ordering));
     }
 
+    pub fn insert(&mut self, index: usize, column: AliasedColumn, ordering: Ordering) {
+        self.columns.insert(index, (column, ordering));
+    }
+
     pub fn is_empty(&self) -> bool {
         self.columns.is_empty()
     }