bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/query-group-macro/src/queries.rs')
-rw-r--r--crates/query-group-macro/src/queries.rs6
1 files changed, 4 insertions, 2 deletions
diff --git a/crates/query-group-macro/src/queries.rs b/crates/query-group-macro/src/queries.rs
index c151cca072..7698ce5fff 100644
--- a/crates/query-group-macro/src/queries.rs
+++ b/crates/query-group-macro/src/queries.rs
@@ -48,7 +48,8 @@ impl ToTokens for TrackedQuery {
quote!(#(#options),*)
})
.into_iter()
- .chain(self.lru.map(|lru| quote!(lru = #lru)));
+ .chain(self.lru.map(|lru| quote!(lru = #lru)))
+ .chain(Some(quote!(unsafe(non_update_return_type))));
let annotation = quote!(#[salsa_macros::tracked( #(#options),* )]);
let pat_and_tys = &self.pat_and_tys;
@@ -326,7 +327,8 @@ impl ToTokens for Lookup {
let wrapper_struct = self.interned_struct_path.to_token_stream();
let method = quote! {
#sig {
- #wrapper_struct::ingredient(self).data(self.as_dyn_database(), id.as_id()).0.clone()
+ let zalsa = self.zalsa();
+ #wrapper_struct::ingredient(zalsa).data(zalsa, id.as_id()).0.clone()
}
};