@Override public EnrichmentConfigurations defaultConfigurations() { return new EnrichmentConfigurations(); }
/** * Timeout for this writer. * @param sensorName n/a * @return timeout in ms */ @Override public int getBatchTimeout(String sensorName) { return config.orElse(new EnrichmentConfigurations()).getBatchTimeout(); }
@Override public Map<String, Object> getGlobalConfig() { return config.orElse(new EnrichmentConfigurations()).getGlobalConfig(); }
/** * Batch size for writing. * @param sensorName n/a * @return batch size in # messages */ @Override public int getBatchSize(String sensorName) { return config.orElse(new EnrichmentConfigurations()).getBatchSize(); }
@Override public Map<String, Object> getSensorConfig(String sensorName) { return config.orElse(new EnrichmentConfigurations()).getSensorEnrichmentConfig(sensorName) .getConfiguration(); }
public static EnrichmentConfigurations getSampleEnrichmentConfigs() throws IOException { EnrichmentConfigurations configurations = new EnrichmentConfigurations(); configurations.updateGlobalConfig(ConfigurationsUtils.readGlobalConfigFromFile(TestConstants.SAMPLE_CONFIG_PATH)); Map<String, byte[]> sensorEnrichmentConfigs = ConfigurationsUtils.readSensorEnrichmentConfigsFromFile(TestConstants.SAMPLE_CONFIG_PATH); for(String sensorType: sensorEnrichmentConfigs.keySet()) { configurations.updateSensorEnrichmentConfig(sensorType, sensorEnrichmentConfigs.get(sensorType)); } return configurations; }
@Test public void gets_batch_size_and_timeout_from_global_config() throws IOException { EnrichmentConfigurations configs = new EnrichmentConfigurations(); configs.updateGlobalConfig(globalJson.getBytes()); EnrichmentWriterConfiguration writerConfig = new EnrichmentWriterConfiguration(configs); assertThat("batch timeout should match global config setting", writerConfig.getBatchTimeout(null), equalTo(555)); assertThat("list should have single batch timeout matching global config setting", writerConfig.getAllConfiguredTimeouts(), equalTo(asList(555))); assertThat("batch size should match global config setting", writerConfig.getBatchSize(null), equalTo(12345)); }
@Test public void strategies_build_writer_configs() { assertThat(PARSERS.createWriterConfig(writer, new ParserConfigurations()), instanceOf(ParserWriterConfiguration.class)); assertThat(ENRICHMENT.createWriterConfig(writer, new EnrichmentConfigurations()), instanceOf(EnrichmentWriterConfiguration.class)); assertThat(INDEXING.createWriterConfig(writer, new IndexingConfigurations()), instanceOf(IndexingWriterConfiguration.class)); assertThat(PROFILER.createWriterConfig(writer, new ProfilerConfigurations()), instanceOf(ProfilerWriterConfiguration.class)); }
@Test public void test() throws Exception { EnrichmentConfigurations sampleConfigurations = new EnrichmentConfigurations(); UnitTestHelper.setLog4jLevel(ConfiguredBolt.class, Level.FATAL); try {