com.datadoghq.trace.resolver.FactoryUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dd-trace Show documentation
Datadog core library
There is a newer version: 0.2.12
package com.datadoghq.trace.resolver;

import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class FactoryUtils {

  private static final ObjectMapper objectMapper = new ObjectMapper(new YAMLFactory());

  public static  A loadConfigFromFilePropertyOrResource(
      final String systemProperty, final String resourceName, final Class targetClass) {
    final String filePath = System.getProperty(systemProperty);
    if (filePath != null) {
      try {
        log.info("Loading config from file " + filePath);
        return objectMapper.readValue(new File(filePath), targetClass);
      } catch (final Exception e) {
        log.error(
            "Cannot read provided configuration file " + filePath + ". Using the default one.", e);
      }
    }

    return loadConfigFromResource(resourceName, targetClass);
  }

  public static  A loadConfigFromFilePropertyOrResource(
      final String systemProperty, final String resourceName, final TypeReference type) {
    final String filePath = System.getProperty(systemProperty);
    if (filePath != null) {
      try {
        log.info("Loading config from file " + filePath);
        return objectMapper.readValue(new File(filePath), type);
      } catch (final Exception e) {
        log.error(
            "Cannot read provided configuration file " + filePath + ". Using the default one.", e);
      }
    }

    return loadConfigFromResource(resourceName, type);
  }

  public static  A loadConfigFromResource(
      final String resourceName, final Class targetClass) {
    A config = null;

    // Try loading both suffixes
    if (!resourceName.endsWith(".yaml") && !resourceName.endsWith(".yml")) {
      config = loadConfigFromResource(resourceName + ".yaml", targetClass);
      if (config == null) {
        config = loadConfigFromResource(resourceName + ".yml", targetClass);
      }
      if (config != null) {
        return config;
      }
    }

    try {
      final ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
      final URL resource = classLoader.getResource(resourceName);
      if (resource != null) {
        log.info("Loading config from resource " + resource);
        config = objectMapper.readValue(resource.openStream(), targetClass);
      }
    } catch (final IOException e) {
      log.warn("Could not load configuration file {}.", resourceName);
      log.error("Error when loading config file", e);
    }
    return config;
  }

  public static  A loadConfigFromResource(
      final String resourceName, final TypeReference type) {
    A config = null;

    // Try loading both suffixes
    if (!resourceName.endsWith(".yaml") && !resourceName.endsWith(".yml")) {
      config = loadConfigFromResource(resourceName + ".yaml", type);
      if (config == null) {
        config = loadConfigFromResource(resourceName + ".yml", type);
      }
      if (config != null) {
        return config;
      }
    }

    try {
      final ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
      final URL resource = classLoader.getResource(resourceName);
      if (resource != null) {
        log.info("Loading config from resource " + resource);
        config = objectMapper.readValue(resource.openStream(), type);
      }
    } catch (final IOException e) {
      log.warn("Could not load configuration file {}.", resourceName);
      log.error("Error when loading config file", e);
    }
    return config;
  }
}