How to use

org.apache.metron.common.configuration.EnrichmentConfigurations

constructor

@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 {