Mariusz Klochowicz
3 years ago
8 changed files with 4 additions and 239 deletions
@ -1,17 +0,0 @@ |
|||
[package] |
|||
name = "xtra_productivity" |
|||
version = "0.1.0" |
|||
edition = "2018" |
|||
|
|||
[lib] |
|||
proc-macro = true |
|||
|
|||
[dependencies] |
|||
quote = "1" |
|||
syn = { version = "1", features = ["full"] } |
|||
|
|||
[dev-dependencies] |
|||
async-trait = "0.1.51" |
|||
tokio = { version = "1", features = ["rt-multi-thread", "macros"] } |
|||
trybuild = "1" |
|||
xtra = { version = "0.6", features = ["with-tokio-1"] } |
|||
@ -1,107 +0,0 @@ |
|||
use proc_macro::TokenStream; |
|||
use quote::quote; |
|||
use syn::{FnArg, GenericParam, ImplItem, ItemImpl, MetaNameValue, ReturnType}; |
|||
|
|||
#[proc_macro_attribute] |
|||
pub fn xtra_productivity(attribute: TokenStream, item: TokenStream) -> TokenStream { |
|||
let block = syn::parse::<ItemImpl>(item).unwrap(); |
|||
let want_message_impl = if attribute.is_empty() { |
|||
true |
|||
} else { |
|||
let attribute = syn::parse::<MetaNameValue>(attribute).unwrap(); |
|||
if !attribute.path.is_ident("message_impl") { |
|||
panic!( |
|||
"Unexpected attribute {:?}", |
|||
attribute.path.get_ident().unwrap() |
|||
) |
|||
} |
|||
|
|||
matches!( |
|||
attribute.lit, |
|||
syn::Lit::Bool(syn::LitBool { value: true, .. }) |
|||
) |
|||
}; |
|||
|
|||
let actor = block.self_ty; |
|||
|
|||
let generic_params = &block.generics.params; |
|||
|
|||
let generic_types = block |
|||
.generics |
|||
.params |
|||
.iter() |
|||
.filter_map(|param| match param { |
|||
GenericParam::Type(ty) => Some(ty.ident.clone()), |
|||
_ => None, |
|||
}) |
|||
.collect::<Vec<_>>(); |
|||
|
|||
let additional_bounds = block |
|||
.generics |
|||
.where_clause |
|||
.map(|bounds| { |
|||
let predicates = bounds.predicates; |
|||
|
|||
quote! { |
|||
#predicates |
|||
} |
|||
}) |
|||
.unwrap_or_default(); |
|||
|
|||
let code = block |
|||
.items |
|||
.into_iter() |
|||
.filter_map(|block_item| match block_item { |
|||
ImplItem::Method(method) => Some(method), |
|||
_ => None, |
|||
}) |
|||
.map(|method| { |
|||
let message_arg = method.sig.inputs[1].clone(); |
|||
|
|||
let message_type = match message_arg { |
|||
// receiver represents self
|
|||
FnArg::Receiver(_) => unreachable!("cannot have receiver on second position"), |
|||
FnArg::Typed(ref typed) => typed.ty.clone() |
|||
}; |
|||
|
|||
let method_return = method.sig.output; |
|||
let method_block = method.block; |
|||
|
|||
let context_arg = method.sig.inputs.iter().nth(2).map(|fn_arg| quote! { #fn_arg }).unwrap_or_else(|| quote! { |
|||
_ctx: &mut xtra::Context<Self> |
|||
}); |
|||
|
|||
let result_type = match method_return { |
|||
ReturnType::Default => quote! { () }, |
|||
ReturnType::Type(_, ref t) => quote! { #t } |
|||
}; |
|||
|
|||
let message_impl = if want_message_impl { |
|||
quote! { |
|||
impl xtra::Message for #message_type { |
|||
type Result = #result_type; |
|||
} |
|||
} |
|||
} else { |
|||
quote! {} |
|||
}; |
|||
|
|||
quote! { |
|||
#message_impl |
|||
|
|||
#[async_trait::async_trait] |
|||
impl<#generic_params> xtra::Handler<#message_type> for #actor |
|||
where |
|||
#additional_bounds |
|||
#(#generic_types: Send + 'static),* |
|||
{ |
|||
async fn handle(&mut self, #message_arg, #context_arg) #method_return #method_block |
|||
} |
|||
} |
|||
}).collect::<Vec<_>>(); |
|||
|
|||
(quote! { |
|||
#(#code)* |
|||
}) |
|||
.into() |
|||
} |
|||
@ -1,27 +0,0 @@ |
|||
use std::marker::PhantomData; |
|||
use xtra_productivity::xtra_productivity; |
|||
|
|||
struct ActorWithParam<C> { |
|||
ty: PhantomData<C>, |
|||
} |
|||
|
|||
struct DummyMessage; |
|||
|
|||
trait Foo {} |
|||
|
|||
impl<C: 'static + Send> xtra::Actor for ActorWithParam<C> {} |
|||
|
|||
// Dummy actor, xtra::Handler and xtra::Message impls generated by xtra_productivity
|
|||
#[xtra_productivity] |
|||
impl<C> ActorWithParam<C> |
|||
where |
|||
C: Foo, |
|||
{ |
|||
pub fn handle_dummy_message(&mut self, _message: DummyMessage) { |
|||
assert_impls_foo::<C>(); |
|||
} |
|||
} |
|||
|
|||
fn assert_impls_foo<T: Foo>() {} |
|||
|
|||
fn main() {} |
|||
@ -1,62 +0,0 @@ |
|||
use xtra::spawn::TokioGlobalSpawnExt; |
|||
use xtra::Actor; |
|||
use xtra_productivity::xtra_productivity; |
|||
|
|||
struct DummyActor; |
|||
|
|||
impl xtra::Actor for DummyActor {} |
|||
|
|||
#[derive(Clone)] |
|||
struct DummyMessage; |
|||
|
|||
struct DummyMessageWithContext; |
|||
|
|||
// Dummy actor, xtra::Handler and xtra::Message impls generated by xtra_productivity
|
|||
#[xtra_productivity] |
|||
impl DummyActor { |
|||
pub fn handle_dummy_message(&mut self, message: DummyMessage) -> i32 { |
|||
let _ = message.clone(); |
|||
0 |
|||
} |
|||
|
|||
pub fn handle_dummy_message_with_context( |
|||
&mut self, |
|||
_message: DummyMessageWithContext, |
|||
context: &mut xtra::Context<Self>, |
|||
) { |
|||
let _ = context.address(); |
|||
} |
|||
} |
|||
|
|||
fn is_i32(_: i32) {} |
|||
|
|||
struct DummyMessageWithoutMessageImpl; |
|||
|
|||
#[xtra_productivity(message_impl = false)] |
|||
impl DummyActor { |
|||
pub fn handle_dummy_message_without_message_impl( |
|||
&mut self, |
|||
_message: DummyMessageWithoutMessageImpl, |
|||
) { |
|||
} |
|||
} |
|||
|
|||
impl xtra::Message for DummyMessageWithoutMessageImpl { |
|||
type Result = (); |
|||
} |
|||
|
|||
#[tokio::main] |
|||
async fn main() { |
|||
// Create dummy actor
|
|||
let dummy_actor = DummyActor.create(None).spawn_global(); |
|||
|
|||
// Send message to dummy actor
|
|||
let i32 = dummy_actor.send(DummyMessage).await.unwrap(); |
|||
is_i32(i32); |
|||
dummy_actor.send(DummyMessageWithContext).await.unwrap(); |
|||
|
|||
dummy_actor |
|||
.send(DummyMessageWithoutMessageImpl) |
|||
.await |
|||
.unwrap(); |
|||
} |
|||
@ -1,6 +0,0 @@ |
|||
#[test] |
|||
fn ui() { |
|||
let t = trybuild::TestCases::new(); |
|||
t.compile_fail("tests/ui/*.rs"); |
|||
t.pass("tests/pass/*.rs"); |
|||
} |
|||
Loading…
Reference in new issue