diff options
Diffstat (limited to 'src/main/java/com/orbekk/protobuf/ServiceHolder.java')
-rw-r--r-- | src/main/java/com/orbekk/protobuf/ServiceHolder.java | 11 |
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); } |