Fix AMQP and Kafka inputs

Graylog2 · Sep 26, 2017 · 312ead3 · 312ead3
1 parent 6b28be9
commit 312ead3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/main/java/org/graylog/plugins/cef/input/CEFAmqpInput.java b/src/main/java/org/graylog/plugins/cef/input/CEFAmqpInput.java
@@ -5,7 +5,6 @@
 import com.google.inject.assistedinject.AssistedInject;
 import org.graylog.plugins.cef.codec.CEFCodec;
 import org.graylog2.inputs.transports.AmqpTransport;
-import org.graylog2.inputs.transports.KafkaTransport;
 import org.graylog2.plugin.LocalMetricRegistry;
 import org.graylog2.plugin.ServerStatus;
 import org.graylog2.plugin.configuration.Configuration;
@@ -41,9 +40,9 @@ public CEFAmqpInput(@Assisted Configuration configuration,
     }
 
     @FactoryClass
-    public interface Factory extends MessageInput.Factory<CEFTCPInput> {
+    public interface Factory extends MessageInput.Factory<CEFAmqpInput> {
         @Override
-        CEFTCPInput create(Configuration configuration);
+        CEFAmqpInput create(Configuration configuration);
 
         @Override
         Config getConfig();
@@ -62,7 +61,7 @@ public Descriptor() {
     @ConfigClass
     public static class Config extends MessageInput.Config {
         @Inject
-        public Config(KafkaTransport.Factory transport, CEFCodec.Factory codec) {
+        public Config(AmqpTransport.Factory transport, CEFCodec.Factory codec) {
             super(transport.getConfig(), codec.getConfig());
         }
     }

diff --git a/src/main/java/org/graylog/plugins/cef/input/CEFKafkaInput.java b/src/main/java/org/graylog/plugins/cef/input/CEFKafkaInput.java
@@ -40,9 +40,9 @@ public CEFKafkaInput(@Assisted Configuration configuration,
     }
 
     @FactoryClass
-    public interface Factory extends MessageInput.Factory<CEFTCPInput> {
+    public interface Factory extends MessageInput.Factory<CEFKafkaInput> {
         @Override
-        CEFTCPInput create(Configuration configuration);
+        CEFKafkaInput create(Configuration configuration);
 
         @Override
         Config getConfig();