Skip to content

Commit e04f1f1

Browse files
authored
feat(ops): sync Rc<RefCell<OpState>> (#14438)
1 parent ca134d2 commit e04f1f1

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

ops/lib.rs

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -185,13 +185,15 @@ fn codegen_v8_async(core: &TokenStream2, f: &syn::ItemFn) -> TokenStream2 {
185185
/// Generate the body of a v8 func for a sync op
186186
fn codegen_v8_sync(core: &TokenStream2, f: &syn::ItemFn) -> TokenStream2 {
187187
let arg0 = f.sig.inputs.first();
188-
let uses_opstate = arg0.map(is_mut_ref_opstate).unwrap_or_default();
189-
let args_head = if uses_opstate {
190-
quote! { op_state, }
191-
} else {
192-
quote! {}
188+
let (rust_i0, args_head) = match arg0 {
189+
Some(arg0) if is_rc_refcell_opstate(arg0) => {
190+
(1, quote! { ctx.state.clone(), })
191+
}
192+
Some(arg0) if is_mut_ref_opstate(arg0) => {
193+
(1, quote! { &mut ctx.state.borrow_mut(), })
194+
}
195+
_ => (0, quote! {}),
193196
};
194-
let rust_i0 = if uses_opstate { 1 } else { 0 };
195197
let (arg_decls, args_tail) = codegen_args(core, f, rust_i0, 0);
196198
let ret = codegen_sync_ret(core, &f.sig.output);
197199
let type_params = &f.sig.generics.params;
@@ -205,9 +207,9 @@ fn codegen_v8_sync(core: &TokenStream2, f: &syn::ItemFn) -> TokenStream2 {
205207

206208
#arg_decls
207209

208-
let op_state = &mut ctx.state.borrow_mut();
209210
let result = Self::call::<#type_params>(#args_head #args_tail);
210211

212+
let op_state = &mut ctx.state.borrow();
211213
op_state.tracker.track_sync(ctx.id);
212214

213215
#ret

0 commit comments

Comments
 (0)