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

There is a newer version: 1.5.0

Show newest version

package com.hubspot.singularity.runner.base.config;

import java.io.File;
import java.util.Set;

import javax.validation.ConstraintViolation;
import javax.validation.Validator;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.base.Optional;
import com.google.common.base.Throwables;
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;

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 Throwables.propagate(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.or(configuration.filename()), violations);
      }

      return config;
    } catch (Exception e) {
      throw Throwables.propagate(e);
    }
  }
}