qdrant-point-sort.rs

use std::cmp::Ordering;

pub struct SortablePoint<'a>(pub &'a RetrievedPoint);

pub fn get_value(kind: &Kind) -> Option<String> {
    match kind {
        Kind::NullValue(_) => None,
        Kind::DoubleValue(value) => Some(value.to_string()),
        Kind::IntegerValue(value) => Some(value.to_string()),
        Kind::StringValue(value) => Some(value.clone()),
        Kind::BoolValue(value) => Some(value.to_string()),
        Kind::StructValue(value) => Some(format!("{:?}", value)),
        Kind::ListValue(value) => Some(format!("{:?}", value)),
    }
}

impl<'a> Eq for SortablePoint<'a> {}

impl<'a> PartialOrd for SortablePoint<'a> {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

impl<'a> Ord for SortablePoint<'a> {
    fn cmp(&self, other: &Self) -> Ordering {
        let this_id =
            get_value_from_payload(&self.0.payload, "msgId").and_then(|id| id.parse::<i32>().ok());

        let other_id =
            get_value_from_payload(&other.0.payload, "msgId").and_then(|id| id.parse::<i32>().ok());

        dbg!(this_id, other_id);

        match (this_id, other_id) {
            (Some(a), Some(b)) => a.cmp(&b),
            (Some(_), None) => Ordering::Greater, // Treat None as smaller than any i32 value
            (None, Some(_)) => Ordering::Less,    // Treat None as smaller than any i32 value
            (None, None) => Ordering::Equal,
        }
    }
}