Search in sources :

Example 1 with KinesisCloudWatchFlowLogCodec

use of org.graylog.integrations.aws.codecs.KinesisCloudWatchFlowLogCodec in project graylog-plugin-integrations by Graylog2.

the class AWSTestingUtils method buildTestCodecs.

public static Map<String, Codec.Factory<? extends Codec>> buildTestCodecs() {
    // Prepare test codecs. These have to be manually instantiated for the test context.
    Map<String, Codec.Factory<? extends Codec>> availableCodecs = new HashMap<>();
    ObjectMapper objectMapper = new ObjectMapperProvider().get();
    availableCodecs.put(KinesisRawLogCodec.NAME, new KinesisRawLogCodec.Factory() {

        @Override
        public KinesisRawLogCodec create(Configuration configuration) {
            return new KinesisRawLogCodec(configuration, objectMapper);
        }

        @Override
        public KinesisRawLogCodec.Config getConfig() {
            return null;
        }

        @Override
        public Codec.Descriptor getDescriptor() {
            return null;
        }
    });
    availableCodecs.put(KinesisCloudWatchFlowLogCodec.NAME, new KinesisCloudWatchFlowLogCodec.Factory() {

        @Override
        public KinesisCloudWatchFlowLogCodec create(Configuration configuration) {
            return new KinesisCloudWatchFlowLogCodec(configuration, objectMapper);
        }

        @Override
        public KinesisCloudWatchFlowLogCodec.Config getConfig() {
            return null;
        }

        @Override
        public Codec.Descriptor getDescriptor() {
            return null;
        }
    });
    return availableCodecs;
}
Also used : Configuration(org.graylog2.plugin.configuration.Configuration) HashMap(java.util.HashMap) ObjectMapperProvider(org.graylog2.shared.bindings.providers.ObjectMapperProvider) KinesisCloudWatchFlowLogCodec(org.graylog.integrations.aws.codecs.KinesisCloudWatchFlowLogCodec) Codec(org.graylog2.plugin.inputs.codecs.Codec) KinesisRawLogCodec(org.graylog.integrations.aws.codecs.KinesisRawLogCodec) KinesisCloudWatchFlowLogCodec(org.graylog.integrations.aws.codecs.KinesisCloudWatchFlowLogCodec) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) KinesisRawLogCodec(org.graylog.integrations.aws.codecs.KinesisRawLogCodec)

Aggregations

ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)1 HashMap (java.util.HashMap)1 KinesisCloudWatchFlowLogCodec (org.graylog.integrations.aws.codecs.KinesisCloudWatchFlowLogCodec)1 KinesisRawLogCodec (org.graylog.integrations.aws.codecs.KinesisRawLogCodec)1 Configuration (org.graylog2.plugin.configuration.Configuration)1 Codec (org.graylog2.plugin.inputs.codecs.Codec)1 ObjectMapperProvider (org.graylog2.shared.bindings.providers.ObjectMapperProvider)1