surrealdb-sql 1.1.0

Full type definitions for the SurrealQL query language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use crate::number::Number;
use std::cmp::Reverse;
use std::collections::BinaryHeap;

pub trait Top {
	/// Find the greatest `k` records from the collection in arbetrary order
	/// O(n*k*log(k)) time complex
	fn top(self, k: i64) -> Vec<Number>;
}

impl Top for Vec<Number> {
	fn top(self, k: i64) -> Vec<Number> {
		// Convert to usize
		let k = k as usize;
		// Create a heap to store the numbers
		let mut heap = BinaryHeap::new();
		// Iterate and store the top numbers
		for (i, v) in self.into_iter().enumerate() {
			heap.push(Reverse(v));
			if i >= k {
				heap.pop();
			}
		}
		// Return the numbers as a vector
		heap.into_iter().map(|x| x.0).collect()
	}
}