refactor: use one method to add endpoints only (#17)

Hapag-Lloyd · May 12, 2022 · 1cc1cb8 · 1cc1cb8
1 parent dd4d399
commit 1cc1cb8
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/src/main/java/com/hlag/tools/commvis/analyzer/model/CommunicationModel.java b/src/main/java/com/hlag/tools/commvis/analyzer/model/CommunicationModel.java
@@ -51,16 +51,16 @@ public class CommunicationModel {
     @SerializedName(value="jms_consumers")
     private Collection<ISenderReceiverCommunication> jmsConsumers = new HashSet<>();
 
-    public void addHttpConsumer(ISenderReceiverCommunication consumer) {
-        httpConsumers.add(consumer);
-    }
-
-    public void addHttpProducer(ISenderReceiverCommunication producer) {
-        httpProducers.add(producer);
-    }
-
-    public void addJmsConsumer(ISenderReceiverCommunication consumer) {
-        jmsConsumers.add(consumer);
+    public <T extends ISenderReceiverCommunication> void addSenderReceiver(T endpoint) {
+        if (endpoint instanceof HttpConsumer) {
+            httpConsumers.add(endpoint);
+        } else if (endpoint instanceof HttpProducer) {
+            httpProducers.add(endpoint);
+        } else if (endpoint instanceof JmsReceiver) {
+            jmsConsumers.add(endpoint);
+        } else {
+            throw new IllegalStateException(String.format("We have no endpoints of type %s", endpoint.getClass().getCanonicalName()));
+        }
     }
 
     public void visit(AbstractCommunicationModelVisitor visitor) {