impl gen hash for enums
This commit is contained in:
parent
e652545472
commit
ec2535e9ce
@ -366,6 +366,33 @@ impl Default for Foo {
|
|||||||
Self { }
|
Self { }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn add_custom_impl_hash_enum() {
|
||||||
|
check_assist(
|
||||||
|
replace_derive_with_manual_impl,
|
||||||
|
r#"
|
||||||
|
//- minicore: hash
|
||||||
|
#[derive(Has$0h)]
|
||||||
|
enum Foo {
|
||||||
|
Bar,
|
||||||
|
Baz,
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
r#"
|
||||||
|
enum Foo {
|
||||||
|
Bar,
|
||||||
|
Baz,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl core::hash::Hash for Foo {
|
||||||
|
$0fn hash<H: core::hash::Hasher>(&self, state: &mut H) {
|
||||||
|
core::mem::discriminant(self).hash(state);
|
||||||
|
}
|
||||||
|
}
|
||||||
"#,
|
"#,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
@ -18,6 +18,7 @@ pub(crate) fn gen_trait_fn_body(
|
|||||||
match trait_path.segment()?.name_ref()?.text().as_str() {
|
match trait_path.segment()?.name_ref()?.text().as_str() {
|
||||||
"Debug" => gen_debug_impl(adt, func),
|
"Debug" => gen_debug_impl(adt, func),
|
||||||
"Default" => gen_default_impl(adt, func),
|
"Default" => gen_default_impl(adt, func),
|
||||||
|
"Hash" => gen_hash_impl(adt, func),
|
||||||
_ => None,
|
_ => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -151,3 +152,49 @@ fn gen_default_impl(adt: &ast::Adt, func: &ast::Fn) -> Option<()> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Generate a `Hash` impl based on the fields and members of the target type.
|
||||||
|
fn gen_hash_impl(adt: &ast::Adt, func: &ast::Fn) -> Option<()> {
|
||||||
|
let body = match adt {
|
||||||
|
// `Hash` cannot be derived for unions, so no default impl can be provided.
|
||||||
|
ast::Adt::Union(_) => return None,
|
||||||
|
|
||||||
|
// => std::mem::discriminant(self).hash(state);
|
||||||
|
ast::Adt::Enum(_) => {
|
||||||
|
let root = make::ext::ident_path("core");
|
||||||
|
let submodule = make::ext::ident_path("mem");
|
||||||
|
let fn_name = make::ext::ident_path("discriminant");
|
||||||
|
let fn_name = make::path_concat(submodule, fn_name);
|
||||||
|
let fn_name = make::expr_path(make::path_concat(root, fn_name));
|
||||||
|
|
||||||
|
let arg = make::expr_path(make::ext::ident_path("self"));
|
||||||
|
let fn_call = make::expr_call(fn_name, make::arg_list(Some(arg)));
|
||||||
|
|
||||||
|
let method = make::name_ref("hash");
|
||||||
|
let arg = make::expr_path(make::ext::ident_path("state"));
|
||||||
|
let expr = make::expr_method_call(fn_call, method, make::arg_list(Some(arg)));
|
||||||
|
let stmt = make::expr_stmt(expr);
|
||||||
|
|
||||||
|
make::block_expr(Some(stmt.into()), None).indent(ast::edit::IndentLevel(1))
|
||||||
|
}
|
||||||
|
ast::Adt::Struct(strukt) => match strukt.field_list() {
|
||||||
|
// => self.<field>.hash(state);*
|
||||||
|
Some(ast::FieldList::RecordFieldList(field_list)) => {
|
||||||
|
// let mut stmts = vec![];
|
||||||
|
for field in field_list.fields() {}
|
||||||
|
todo!();
|
||||||
|
}
|
||||||
|
|
||||||
|
// => self.<field_index>.hash(state);*
|
||||||
|
Some(ast::FieldList::TupleFieldList(field_list)) => {
|
||||||
|
todo!();
|
||||||
|
}
|
||||||
|
|
||||||
|
// No fields in the body means there's nothing to hash.
|
||||||
|
None => make::ext::empty_block_expr(),
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
ted::replace(func.body()?.syntax(), body.clone_for_update().syntax());
|
||||||
|
Some(())
|
||||||
|
}
|
||||||
|
@ -25,6 +25,7 @@
|
|||||||
//! iterator: option
|
//! iterator: option
|
||||||
//! iterators: iterator, fn
|
//! iterators: iterator, fn
|
||||||
//! default: sized
|
//! default: sized
|
||||||
|
//! hash:
|
||||||
//! clone: sized
|
//! clone: sized
|
||||||
//! copy: clone
|
//! copy: clone
|
||||||
//! from: sized
|
//! from: sized
|
||||||
@ -87,6 +88,16 @@ pub mod default {
|
|||||||
}
|
}
|
||||||
// endregion:default
|
// endregion:default
|
||||||
|
|
||||||
|
// region:hash
|
||||||
|
pub mod hash {
|
||||||
|
pub trait Hasher {}
|
||||||
|
|
||||||
|
pub trait Hash {
|
||||||
|
fn hash<H: Hasher>(&self, state: &mut H);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// endregion:hash
|
||||||
|
|
||||||
// region:clone
|
// region:clone
|
||||||
pub mod clone {
|
pub mod clone {
|
||||||
#[lang = "clone"]
|
#[lang = "clone"]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user