From 4f0bb01fd2664f7d113696f6020669b0a19fe4e1 Mon Sep 17 00:00:00 2001
From: Viktor Klang <viktor.klang@gmail.com>
Date: Wed, 22 Sep 2010 11:37:23 +0200
Subject: [PATCH] Preparing to add UUIDs to RemoteServer as well

---
 akka-remote/src/main/scala/remote/RemoteServer.scala | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/akka-remote/src/main/scala/remote/RemoteServer.scala b/akka-remote/src/main/scala/remote/RemoteServer.scala
index 8784a8c81f..fadc5ce60a 100644
--- a/akka-remote/src/main/scala/remote/RemoteServer.scala
+++ b/akka-remote/src/main/scala/remote/RemoteServer.scala
@@ -314,7 +314,7 @@ class RemoteServer extends Logging with ListenerManagement {
     }
   }
 
-  private def register(id: String, actorRef: ActorRef, registry: ConcurrentHashMap[String, ActorRef]) {
+  private def register[Key](id: Key, actorRef: ActorRef, registry: ConcurrentHashMap[Key, ActorRef]) {
     if (_isRunning) {
       if (!registry.contains(id)) {
         if (!actorRef.isRunning) actorRef.start
@@ -323,7 +323,7 @@ class RemoteServer extends Logging with ListenerManagement {
     }
   }
 
-  private def registerTypedActor(id: String, typedActor: AnyRef, registry: ConcurrentHashMap[String, AnyRef]) {
+  private def registerTypedActor[Key](id: Key, typedActor: AnyRef, registry: ConcurrentHashMap[Key, AnyRef]) {
     if (_isRunning) {
       if (!registry.contains(id)) {
         registry.put(id, typedActor)
@@ -337,8 +337,7 @@ class RemoteServer extends Logging with ListenerManagement {
   def unregister(actorRef: ActorRef):Unit = synchronized {
     if (_isRunning) {
       log.debug("Unregistering server side remote actor [%s] with id [%s:%s]", actorRef.actorClass.getName, actorRef.id, actorRef.uuid)
-      val actorMap = actors()
-      actorMap remove actorRef.id
+      actors() remove actorRef.id
       if (actorRef.registeredInRemoteNodeDuringSerialization) actorsByUuid() remove actorRef.uuid
     }
   }