akka - 访问 Actor 邮箱的消息

Question

我想阅读 ShardRegions 邮箱中的消息。在以前的akka​​版本中，我们可以使用下面的代码只获取邮箱的大小:

getContext().getMailboxSize();

有没有办法获取邮箱中的消息类型？

Answer 1

解决方法

像这样为分片区域定义一个包装邮箱:

class UnboundedMailboxWrapper extends MailboxType with ProducesMessageQueue[UnboundedMailboxWrapper.MessageQueue] {
  def this(settings: ActorSystem.Settings, config: Config) = this()

  final override def create(owner: Option[ActorRef], system: Option[ActorSystem]): MessageQueue =
    new UnboundedMailboxWrapper.MessageQueue
}

object UnboundedMailboxWrapper {
  class MessageQueue extends ConcurrentLinkedQueue[Envelope] with UnboundedQueueBasedMessageQueue {
    val runtime = RuntimeManager.runtime
    final def queue: Queue[Envelope] = this
    override def enqueue(receiver: ActorRef, handle: Envelope): Unit = {
      runtime.queue.add(handle)
      queue add handle
    }
    override def dequeue(): Envelope = {
      if (!runtime.queue.isEmpty) runtime.mailbox.queue.poll()
      queue.poll()
    }
  }
}

在这个邮箱中，我们将添加的元素复制到另一个队列中，因此在计算大小和对其进行其他操作时，对邮箱队列的性能没有任何影响

在重复队列中，我们可以计算消息的数量并对它们进行排序:

def getElemets(): String = {
  runtime.queue.asScala.toList.groupBy(_.message.getClass.getName)
    .map(e ⇒ (e._1, e._2.length)).toSeq
    .sortBy(_._2).foldLeft("") { (a, b) ⇒
    b._1 + ":" + b._2 + "\n" + a
  }
}

并且使用 JMX 或任何其他方式我们可以在运行时调用此方法

最后将这个邮箱分配给 ShardRegion 调度程序:

monit-dispatcher {
  mailbox-type = "im.actor.server.cluster.UnboundedMailboxWrapper"
}

akka.cluster.sharding {
  use-dispatcher = "monit-dispatcher"
}