Remove Receive 'Msg' type, add actor trait bound

riker-rs · Jun 1, 2020 · 113178b · 113178b
1 parent d43f5d3
commit 113178b
Show file tree

Hide file tree

Showing 16 changed files with 1 addition and 102 deletions.
diff --git a/examples/basic_panic.rs b/examples/basic_panic.rs
@@ -38,8 +38,6 @@ impl Actor for PanicActor {
 }
 
 impl Receive<Panic> for PanicActor {
-    type Msg = PanicActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         panic!("// TEST PANIC // TEST PANIC // TEST PANIC //");
     }

diff --git a/examples/channel.rs b/examples/channel.rs
@@ -41,8 +41,6 @@ impl Actor for GpsActor {
 }
 
 impl Receive<PowerStatus> for GpsActor {
-    type Msg = GpsActorMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, msg: PowerStatus, _sender: Sender) {
         println!("{}: -> got msg: {:?}", ctx.myself.name(), msg);
     }
@@ -82,8 +80,6 @@ impl Actor for NavigationActor {
 }
 
 impl Receive<PowerStatus> for NavigationActor {
-    type Msg = NavigationActorMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, msg: PowerStatus, _sender: Sender) {
         println!("{}: -> got msg: {:?}", ctx.myself.name(), msg);
     }

diff --git a/examples/channel_dead_letters.rs b/examples/channel_dead_letters.rs
@@ -19,8 +19,6 @@ impl Actor for DumbActor {
 }
 
 impl Receive<SomeMessage> for DumbActor {
-    type Msg = DumbActorMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, msg: SomeMessage, _sender: Sender) {
         println!("{}: -> got msg: {:?} ", ctx.myself.name(), msg);
     }
@@ -55,8 +53,6 @@ impl Actor for DeadLetterActor {
 }
 
 impl Receive<DeadLetter> for DeadLetterActor {
-    type Msg = DeadLetterActorMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, msg: DeadLetter, _sender: Sender) {
         println!("{}: -> got msg: {:?} ", ctx.myself.name(), msg);
     }

diff --git a/examples/channel_system.rs b/examples/channel_system.rs
@@ -19,8 +19,6 @@ impl Actor for DumbActor {
 }
 
 impl Receive<Panic> for DumbActor {
-    type Msg = DumbActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         panic!("// TEST PANIC // TEST PANIC // TEST PANIC //");
     }
@@ -65,8 +63,6 @@ impl Actor for SystemActor {
 }
 
 impl Receive<SystemEvent> for SystemActor {
-    type Msg = SystemActorMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, msg: SystemEvent, _sender: Sender) {
         print!("{}: -> got system msg: {:?} ", ctx.myself.name(), msg);
         match msg {

diff --git a/examples/messaging.rs b/examples/messaging.rs
@@ -37,24 +37,18 @@ impl Actor for Counter {
 }
 
 impl Receive<Add> for Counter {
-    type Msg = CounterMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Add, _sender: Sender) {
         self.count += 1;
     }
 }
 
 impl Receive<Sub> for Counter {
-    type Msg = CounterMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Sub, _sender: Sender) {
         self.count -= 1;
     }
 }
 
 impl Receive<Print> for Counter {
-    type Msg = CounterMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Print, _sender: Sender) {
         println!("Total counter value: {}", self.count);
     }

diff --git a/examples/supervision_escalate.rs b/examples/supervision_escalate.rs
@@ -38,8 +38,6 @@ impl Actor for PanicActor {
 }
 
 impl Receive<Panic> for PanicActor {
-    type Msg = PanicActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         panic!("// TEST PANIC // TEST PANIC // TEST PANIC //");
     }
@@ -73,8 +71,6 @@ impl Actor for EscalateSup {
 }
 
 impl Receive<Panic> for EscalateSup {
-    type Msg = EscalateSupMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         self.actor_to_fail.as_ref().unwrap().tell(Panic, None);
     }
@@ -108,8 +104,6 @@ impl Actor for EscRestartSup {
 }
 
 impl Receive<Panic> for EscRestartSup {
-    type Msg = EscRestartSupMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         self.escalator.as_ref().unwrap().tell(Panic, None);
     }

diff --git a/examples/supervision_restart.rs b/examples/supervision_restart.rs
@@ -38,8 +38,6 @@ impl Actor for PanicActor {
 }
 
 impl Receive<Panic> for PanicActor {
-    type Msg = PanicActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         panic!("// TEST PANIC // TEST PANIC // TEST PANIC //");
     }
@@ -69,8 +67,6 @@ impl Actor for RestartSup {
 }
 
 impl Receive<Panic> for RestartSup {
-    type Msg = RestartSupMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         self.actor_to_fail.as_ref().unwrap().tell(Panic, None);
     }

diff --git a/examples/supervision_stop.rs b/examples/supervision_stop.rs
@@ -38,8 +38,6 @@ impl Actor for PanicActor {
 }
 
 impl Receive<Panic> for PanicActor {
-    type Msg = PanicActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         panic!("// TEST PANIC // TEST PANIC // TEST PANIC //");
     }
@@ -69,8 +67,6 @@ impl Actor for RestartSup {
 }
 
 impl Receive<Panic> for RestartSup {
-    type Msg = RestartSupMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         self.actor_to_fail.as_ref().unwrap().tell(Panic, None);
     }

diff --git a/riker-macros/src/lib.rs b/riker-macros/src/lib.rs
@@ -108,8 +108,6 @@ fn receive(aname: &Ident, name: &Ident, types: &MsgTypes) -> TokenStream {
     });
     quote! {
         impl Receive<#name> for #aname {
-            type Msg = #name;
-
             fn receive(&mut self,
                         ctx: &Context<Self::Msg>,
                         msg: #name,

diff --git a/src/actor/actor.rs b/src/actor/actor.rs
@@ -113,8 +113,6 @@ impl<A: Actor + ?Sized> Actor for Box<A> {
 /// }
 ///
 /// impl Receive<Foo> for MyActor {
-///     type Msg = MyActorMsg;
-///
 ///     fn receive(&mut self,
 ///                 ctx: &Context<Self::Msg>,
 ///                 msg: Foo, // <-- receive Foo
@@ -124,8 +122,6 @@ impl<A: Actor + ?Sized> Actor for Box<A> {
 /// }
 ///
 /// impl Receive<Bar> for MyActor {
-///     type Msg = MyActorMsg;
-///
 ///     fn receive(&mut self,
 ///                 ctx: &Context<Self::Msg>,
 ///                 msg: Bar, // <-- receive Bar
@@ -141,8 +137,7 @@ impl<A: Actor + ?Sized> Actor for Box<A> {
 /// actor.tell(Foo, None);
 /// actor.tell(Bar, None);
 /// ```
-pub trait Receive<Msg: Message> {
-    type Msg: Message;
+pub trait Receive<Msg: Message> : Actor {
 
     /// Invoked when an actor receives a message
     ///

diff --git a/src/actor/channel.rs b/src/actor/channel.rs
@@ -74,8 +74,6 @@ impl<Msg> Receive<ChannelMsg<Msg>> for Channel<Msg>
 where
     Msg: Message,
 {
-    type Msg = ChannelMsg<Msg>;
-
     fn receive(&mut self, ctx: &ChannelCtx<Msg>, msg: Self::Msg, sender: Sender) {
         match msg {
             ChannelMsg::Publish(p) => self.receive(ctx, p, sender),
@@ -90,8 +88,6 @@ impl<Msg> Receive<Subscribe<Msg>> for Channel<Msg>
 where
     Msg: Message,
 {
-    type Msg = ChannelMsg<Msg>;
-
     fn receive(&mut self, ctx: &ChannelCtx<Msg>, msg: Subscribe<Msg>, sender: Sender) {
         let subs = self.subs.entry(msg.topic).or_default();
         subs.push(msg.actor);
@@ -102,8 +98,6 @@ impl<Msg> Receive<Unsubscribe<Msg>> for Channel<Msg>
 where
     Msg: Message,
 {
-    type Msg = ChannelMsg<Msg>;
-
     fn receive(&mut self, ctx: &ChannelCtx<Msg>, msg: Unsubscribe<Msg>, sender: Sender) {
         unsubscribe(&mut self.subs, &msg.topic, &msg.actor);
     }
@@ -113,8 +107,6 @@ impl<Msg> Receive<UnsubscribeAll<Msg>> for Channel<Msg>
 where
     Msg: Message,
 {
-    type Msg = ChannelMsg<Msg>;
-
     fn receive(&mut self, ctx: &ChannelCtx<Msg>, msg: UnsubscribeAll<Msg>, sender: Sender) {
         let subs = self.subs.clone();
 
@@ -128,8 +120,6 @@ impl<Msg> Receive<Publish<Msg>> for Channel<Msg>
 where
     Msg: Message,
 {
-    type Msg = ChannelMsg<Msg>;
-
     fn receive(&mut self, ctx: &ChannelCtx<Msg>, msg: Publish<Msg>, sender: Sender) {
         // send system event to actors subscribed to all topics
         if let Some(subs) = self.subs.get(&All.into()) {
@@ -187,8 +177,6 @@ impl Actor for EventsChannel {
 }
 
 impl Receive<ChannelMsg<SystemEvent>> for EventsChannel {
-    type Msg = ChannelMsg<SystemEvent>;
-
     fn receive(&mut self, ctx: &ChannelCtx<SystemEvent>, msg: Self::Msg, sender: Sender) {
         // Publish variant uses specialized EventsChannel Receive
         // All other variants use the wrapped Channel (self.0) Receive(s)
@@ -202,8 +190,6 @@ impl Receive<ChannelMsg<SystemEvent>> for EventsChannel {
 }
 
 impl Receive<Publish<SystemEvent>> for EventsChannel {
-    type Msg = ChannelMsg<SystemEvent>;
-
     fn receive(
         &mut self,
         ctx: &ChannelCtx<SystemEvent>,

diff --git a/src/system/system.rs b/src/system/system.rs
@@ -818,8 +818,6 @@ impl Actor for ShutdownActor {
 }
 
 impl Receive<ActorTerminated> for ShutdownActor {
-    type Msg = SystemEvent;
-
     fn receive(
         &mut self,
         ctx: &Context<Self::Msg>,

diff --git a/tests/actors.rs b/tests/actors.rs
@@ -29,16 +29,12 @@ impl Actor for Counter {
 }
 
 impl Receive<TestProbe> for Counter {
-    type Msg = CounterMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, msg: TestProbe, _sender: Sender) {
         self.probe = Some(msg)
     }
 }
 
 impl Receive<Add> for Counter {
-    type Msg = CounterMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Add, _sender: Sender) {
         self.count += 1;
         if self.count == 1_000_000 {

diff --git a/tests/channels.rs b/tests/channels.rs
@@ -50,17 +50,13 @@ impl Actor for Subscriber {
 }
 
 impl Receive<TestProbe> for Subscriber {
-    type Msg = SubscriberMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, msg: TestProbe, _sender: Sender) {
         msg.0.event(());
         self.probe = Some(msg);
     }
 }
 
 impl Receive<SomeMessage> for Subscriber {
-    type Msg = SubscriberMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: SomeMessage, _sender: Sender) {
         self.probe.as_ref().unwrap().0.event(());
     }
@@ -167,16 +163,12 @@ impl Actor for DumbActor {
 }
 
 impl Receive<Panic> for DumbActor {
-    type Msg = DumbActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: Panic, _sender: Sender) {
         panic!("// TEST PANIC // TEST PANIC // TEST PANIC //");
     }
 }
 
 impl Receive<SomeMessage> for DumbActor {
-    type Msg = DumbActorMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: SomeMessage, _sender: Sender) {
 
         // Intentionally left blank
@@ -222,17 +214,13 @@ impl Actor for EventSubscriber {
 }
 
 impl Receive<TestProbe> for EventSubscriber {
-    type Msg = EventSubscriberMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, msg: TestProbe, _sender: Sender) {
         msg.0.event(());
         self.probe = Some(msg);
     }
 }
 
 impl Receive<SystemEvent> for EventSubscriber {
-    type Msg = EventSubscriberMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, msg: SystemEvent, _sender: Sender) {
         match msg {
             SystemEvent::ActorCreated(created) => {
@@ -311,17 +299,13 @@ impl Actor for DeadLetterSub {
 }
 
 impl Receive<TestProbe> for DeadLetterSub {
-    type Msg = DeadLetterSubMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, msg: TestProbe, _sender: Sender) {
         msg.0.event(());
         self.probe = Some(msg);
     }
 }
 
 impl Receive<DeadLetter> for DeadLetterSub {
-    type Msg = DeadLetterSubMsg;
-
     fn receive(&mut self, _ctx: &Context<Self::Msg>, _msg: DeadLetter, _sender: Sender) {
         self.probe.as_ref().unwrap().0.event(());
     }

diff --git a/tests/scheduling.rs b/tests/scheduling.rs
@@ -31,8 +31,6 @@ impl Actor for ScheduleOnce {
 }
 
 impl Receive<TestProbe> for ScheduleOnce {
-    type Msg = ScheduleOnceMsg;
-
     fn receive(&mut self, ctx: &Context<ScheduleOnceMsg>, msg: TestProbe, _sender: Sender) {
         self.probe = Some(msg);
         // reschedule an Empty to be sent to myself()
@@ -41,8 +39,6 @@ impl Receive<TestProbe> for ScheduleOnce {
 }
 
 impl Receive<SomeMessage> for ScheduleOnce {
-    type Msg = ScheduleOnceMsg;
-
     fn receive(&mut self, _ctx: &Context<ScheduleOnceMsg>, _msg: SomeMessage, _sender: Sender) {
         self.probe.as_ref().unwrap().0.event(());
     }
@@ -93,8 +89,6 @@ impl Actor for ScheduleRepeat {
 }
 
 impl Receive<TestProbe> for ScheduleRepeat {
-    type Msg = ScheduleRepeatMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, msg: TestProbe, _sender: Sender) {
         self.probe = Some(msg);
         // schedule Message to be repeatedly sent to myself
@@ -111,8 +105,6 @@ impl Receive<TestProbe> for ScheduleRepeat {
 }
 
 impl Receive<SomeMessage> for ScheduleRepeat {
-    type Msg = ScheduleRepeatMsg;
-
     fn receive(&mut self, ctx: &Context<Self::Msg>, _msg: SomeMessage, _sender: Sender) {
         if self.counter == 5 {
             ctx.cancel_schedule(self.schedule_id.unwrap());