diff options
author | Kjetil Ørbekk <kjetil.orbekk@gmail.com> | 2012-04-30 11:43:29 +0200 |
---|---|---|
committer | Kjetil Ørbekk <kjetil.orbekk@gmail.com> | 2012-04-30 11:43:29 +0200 |
commit | e77a91ea3842e46e93e03ce89fb6b4b0d90c2999 (patch) | |
tree | 16ecff34ab16facc000bd29bf8933a98491929f1 /src/main/java | |
parent | 02711b2f96bded722e46fb98a693055297e47926 (diff) |
Add removeService() to SimpleProtobufServer.
Diffstat (limited to 'src/main/java')
-rw-r--r-- | src/main/java/com/orbekk/protobuf/ServiceHolder.java | 11 | ||||
-rw-r--r-- | src/main/java/com/orbekk/protobuf/SimpleProtobufServer.java | 8 |
2 files changed, 18 insertions, 1 deletions
diff --git a/src/main/java/com/orbekk/protobuf/ServiceHolder.java b/src/main/java/com/orbekk/protobuf/ServiceHolder.java index cab9a5e..e43c7a9 100644 --- a/src/main/java/com/orbekk/protobuf/ServiceHolder.java +++ b/src/main/java/com/orbekk/protobuf/ServiceHolder.java @@ -14,7 +14,7 @@ public class ServiceHolder { public ServiceHolder() { } - public void registerService (Service service) { + public void registerService(Service service) { String serviceName = service.getDescriptorForType().getFullName(); Service previousService = services.put(serviceName, service); if (previousService != null) { @@ -23,6 +23,15 @@ public class ServiceHolder { } } + public void removeService(Service service) { + String serviceName = service.getDescriptorForType().getFullName(); + removeService(serviceName); + } + + public void removeService(String serviceName) { + services.remove(serviceName); + } + public Service get(String fullServiceName) { return services.get(fullServiceName); } diff --git a/src/main/java/com/orbekk/protobuf/SimpleProtobufServer.java b/src/main/java/com/orbekk/protobuf/SimpleProtobufServer.java index 14887d8..6ea427d 100644 --- a/src/main/java/com/orbekk/protobuf/SimpleProtobufServer.java +++ b/src/main/java/com/orbekk/protobuf/SimpleProtobufServer.java @@ -82,6 +82,14 @@ public class SimpleProtobufServer extends Thread { public void registerService(Service service) { services.registerService(service); } + + public void removeService(Service service) { + services.removeService(service); + } + + public void removeService(String fullServiceName) { + services.removeService(fullServiceName); + } private synchronized void handleConnection(Socket connection) { if (serverSocket.isClosed()) { |