summaryrefslogtreecommitdiff
path: root/src/main/java/com/orbekk/protobuf/ServiceHolder.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/com/orbekk/protobuf/ServiceHolder.java')
-rw-r--r--src/main/java/com/orbekk/protobuf/ServiceHolder.java11
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);
}