是否可以通过委托给struct成员来创建实现Ord的宏?

时间:2017-08-13 19:59:56

标签: rust rust-macros

我有一个结构:

struct Student {
    first_name: String,
    last_name: String,
}

我想创建一个Vec<Student>,可以按last_name排序。我需要实施OrdPartialOrdPartialEq

use std::cmp::Ordering;

impl Ord for Student {
    fn cmp(&self, other: &Student) -> Ordering {
        self.last_name.cmp(&other.last_name)
    }
}

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

impl PartialEq for Student {
    fn eq(&self, other: &Student) -> bool {
        self.last_name == other.last_name
    }
}

如果你有很多带有明显字段的结构,那么这可能是相当单调和重复的。是否可以创建一个宏来自动实现它?

类似的东西:

impl_ord!(Student, Student.last_name)

我找到了Automatically implement traits of enclosed type for Rust newtypes (tuple structs with one field),但这并不是我想要的。

2 个答案:

答案 0 :(得分:3)

是的,你可以,但首先:请阅读为什么你不应该

为什么不呢?

当类型实现OrdPartialOrd时,这意味着此类型具有自然排序,这反过来意味着实现的排序是唯一的逻辑排序。取整数:3自然小于4.还有其他有用的排序,当然。您可以按递减顺序对整数进行排序,而不是使用反向排序,但只有一个自然排序。

现在你有一个由两个字符串组成的类型。有自然的顺序吗?我声称:不!有很多有用的排序,但是使用姓氏排序比使用名字排序更自然吗?我不这么认为。

那怎么办?

还有另外两种排序方法:

两者都允许您修改排序算法比较值的方式。按姓氏排序可以像这样(full code):

students.sort_by(|a, b| a.last_name.cmp(&b.last_name));

这样,您可以指定如何对每个方法调用进行排序。有时您可能希望按姓氏和其他时间按名字排序。由于没有明显和自然的排序方式,你不应该&#34;附加&#34;任何特定的排序方式本身。

但严重的是,我想要一个宏......

当然,Rust可以编写这样的宏。一旦你理解了宏系统,它实际上很容易。但是,请不要为你的Student例子做这件事,因为 - 正如我希望你现在明白的那样 - 这是一个坏主意。

什么时候是好主意?当只有一个字段语义是该类型的一部分时。采用这种数据结构:

struct Foo {
    actual_data: String,
    _internal_cache: String,
}

此处,_internal_cache在语义上不属于您的类型。它只是一个实现细节,因此应该忽略EqOrd。简单的宏是:

macro_rules! impl_ord {
    ($type_name:ident, $field:ident) => {
        impl Ord for $type_name {
            fn cmp(&self, other: &$type_name) -> Ordering {
                self.$field.cmp(&other.$field)
            }
        }

        impl PartialOrd for $type_name {
            fn partial_cmp(&self, other: &$type_name) -> Option<Ordering> {
                Some(self.cmp(other))
            }
        }

        impl PartialEq for $type_name {
            fn eq(&self, other: &$type_name) -> bool {
                self.$field == other.$field
            }
        }

        impl Eq for $type_name {}
    }
}

为什么我要问这么大的代码简单?嗯,绝大多数代码正是您已编写的代码:impls。我执行了两个简单的步骤:

  1. 在代码周围添加宏定义并考虑我们需要的参数(type_namefield
  2. Student的所有提及替换为$type_name,并将last_name的所有提及替换为$field
  3. 这就是为什么它被称为&#34;宏示例&#34;:你基本上只是编写你的普通代码作为例子,但可以使每个参数的部分变量。

    您可以测试整件事here

答案 1 :(得分:1)

我创建了一个宏,该宏允许通过定义用于比较元素ord_by_key::ord_eq_by_key_selector的表达式来实现Ord,类似于您要的内容。

use ord_by_key::ord_eq_by_key_selector;

#[ord_eq_by_key_selector(|s| &s.last_name)]
struct Student {
    first_name: String,
    last_name: String,
}

如果必须在不同情况下按不同条件进行排序,则可以为结构引入一个容器,该容器将实现不同的排序策略:

use ord_by_key::ord_eq_by_key_selector;

struct Student {
    first_name: String,
    last_name: String,
}

#[ord_eq_by_key_selector(|(s)| &s.first_name)]
struct StudentByFirstName(Student);

#[ord_eq_by_key_selector(|(s)| &s.last_name, &s.first_name)]
struct StudentByLastNameAndFirstName(Student);