Add removeService() to SimpleProtobufServer.

1 files changed, 10 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);
     }