roc-lang/roc
1
1
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2024-09-11 18:58:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6878 from roc-lang/glue-borrow

Browse Source 
Update rust glue to allow for borrowing payloads from tags
This commit is contained in:
Brendan Hansknecht 2024-07-07 15:23:14 -07:00 committed by GitHub
commit 69d670d27e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
crates/glue/src/RustGlue.roc
View File

@ -644,6 +644,30 @@ generateDestructorFunction = \buf, types, tagUnionType, name, optPayload ->
else
"unsafe { core::mem::ManuallyDrop::take(&mut self.payload.$(name)) }"
(borrow, borrowType) =
if canDeriveCopy types shape then
("unsafe { self.payload.$(name) }", payloadType)
else
(
"""
use core::borrow::Borrow;
unsafe { self.payload.$(name).borrow() }
""",
"&$(payloadType)",
)
(borrowMut, borrowMutType) =
if canDeriveCopy types shape then
("unsafe { &mut self.payload.$(name) }", "&mut $(payloadType)")
else
(
"""
use core::borrow::BorrowMut;
unsafe { self.payload.$(name).borrow_mut() }
""",
"&mut $(payloadType)",
)
"""
$(buf)
@ -652,6 +676,16 @@ generateDestructorFunction = \buf, types, tagUnionType, name, optPayload ->
$(take)
}
pub fn borrow_$(name)(&self) -> $(borrowType) {
debug_assert_eq!(self.discriminant, discriminant_$(tagUnionType)::$(name));
$(borrow)
}
pub fn borrow_mut_$(name)(&mut self) -> $(borrowMutType) {
debug_assert_eq!(self.discriminant, discriminant_$(tagUnionType)::$(name));
$(borrowMut)
}
pub fn is_$(name)(&self) -> bool {
matches!(self.discriminant, discriminant_$(tagUnionType)::$(name))
}