com.hubspot.singularity.runner.base.config.SingularityRunnerConfigurationProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of SingularityRunnerBase Show documentation
The newest version!
package com.hubspot.singularity.runner.base.config;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.name.Named;
import com.hubspot.singularity.runner.base.configuration.BaseRunnerConfiguration;
import com.hubspot.singularity.runner.base.configuration.Configuration;
import io.dropwizard.configuration.ConfigurationValidationException;
import java.io.File;
import java.util.Optional;
import java.util.Set;
import javax.validation.ConstraintViolation;
import javax.validation.Validator;

public class SingularityRunnerConfigurationProvider
  implements Provider {
  private final Class clazz;
  private final Optional filename;

  @Inject
  @Named(SingularityRunnerBaseModule.YAML)
  private ObjectMapper objectMapper;

  @Inject
  private Validator validator;

  public SingularityRunnerConfigurationProvider(
    Class clazz,
    Optional filename
  ) {
    this.clazz = clazz;
    this.filename = filename;
  }

  private JsonNode loadYamlField(String filename, String field) {
    final File yamlFile = new File(filename);

    if (!yamlFile.exists()) {
      return objectMapper.createObjectNode();
    }

    try {
      final JsonNode baseTree = objectMapper.readTree(yamlFile);

      return baseTree.hasNonNull(field)
        ? baseTree.get(field)
        : objectMapper.createObjectNode();
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  @Override
  public T get() {
    final Configuration configuration = clazz.getAnnotation(Configuration.class);

    try {
      final File baseFile = new File(configuration.filename());
      final T baseConfig = baseFile.exists()
        ? objectMapper.readValue(baseFile, clazz)
        : clazz.newInstance();

      final JsonNode overrideNode = filename.isPresent()
        ? loadYamlField(filename.get(), configuration.consolidatedField())
        : objectMapper.createObjectNode();

      final T config = objectMapper.readerForUpdating(baseConfig).readValue(overrideNode);

      final Set> violations = validator.validate(config);
      if (!violations.isEmpty()) {
        throw new ConfigurationValidationException(
          filename.orElse(configuration.filename()),
          violations
        );
      }

      return config;
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }
}