Merge entity and behavior into same object

examples/src/main/scala/zio/actors/examples/sharding/ShoppingCartApp.scala

@@ -22,7 +22,7 @@ object ShoppingCartApp extends ZIOAppDefault {
  for {
  _ <- Sharding.registerEntity(
  ShoppingCartEntity.ShoppingCartEntityType,
- ShoppingCartBehavior.behavior
+ ShoppingCartEntity.behavior
  )
  _ <- Sharding.registerScoped
  cart <- Sharding.messenger(ShoppingCartEntity.ShoppingCartEntityType)

examples/src/main/scala/zio/actors/examples/sharding/ShoppingCartEntity.scala

@@ -1,10 +1,14 @@
 package zio.actors.examples.sharding
 
-import com.devsisters.shardcake.EntityType
 import com.devsisters.shardcake.Messenger.Replier
+import com.devsisters.shardcake.{ EntityType, Sharding }
+import zio.ZIO
+import zio.actors.ActorRef
 import zio.actors.examples.persistence.ShoppingCart
+import zio.actors.persistence.EventSourcedStateful
+import zio.actors.sharding.Behavior
 
-object ShoppingCartEntity {
+object ShoppingCartEntity extends Behavior {
  sealed trait Message
 
  final case class AddItem(
@@ -29,4 +33,41 @@ object ShoppingCartEntity {
  ) extends Message
 
  object ShoppingCartEntityType extends EntityType[Message]("ShoppingCartEntity")
+
+ type BehaviorMessage = Message
+ type State = ShoppingCart.State
+ type Command[+A] = ShoppingCart.Command[A]
+ type Event = ShoppingCart.Event
+
+ def stateEmpty: State = ShoppingCart.State.empty
+
+ def eventSourcedFactory: String => EventSourcedStateful[Any, State, Command, Event] =
+ persistenceId => ShoppingCart(persistenceId)
+
+ def messageHandler(
+ message: Message,
+ actor: ActorRef[Command]
+ ): ZIO[Sharding, Throwable, Unit] =
+ message match {
+ case AddItem(itemId, quantity, replier) =>
+ actor
+ .?(ShoppingCart.AddItem(itemId, quantity))
+ .flatMap(replier.reply)
+ case RemoveItem(itemId, replier) =>
+ actor
+ .?(ShoppingCart.RemoveItem(itemId))
+ .flatMap(replier.reply)
+ case AdjustItemQuantity(itemId, quantity, replier) =>
+ actor
+ .?(ShoppingCart.AdjustItemQuantity(itemId, quantity))
+ .flatMap(replier.reply)
+ case Checkout(replier) =>
+ actor
+ .?(ShoppingCart.Checkout)
+ .flatMap(replier.reply)
+ case Get(replier) =>
+ actor
+ .?(ShoppingCart.Get)
+ .flatMap(replier.reply)
+ }
 }

sharding/src/main/scala/zio/actors/sharding/Behavior.scala

@@ -6,7 +6,8 @@ import zio.actors.persistence.EventSourcedStateful
 import zio.actors.sharding.Layers.ActorSystemZ
 import zio.{ Dequeue, RIO, ZIO }
 
-trait Behavior[BehaviorMessage] {
+trait Behavior {
+ type BehaviorMessage
  type State
  type Command[+_]
  type Event