Examples with CompositePropertySource org.springframework.core.env.CompositePropertySource used on opensource projects

Example 21 with CompositePropertySource

use of org.springframework.core.env.CompositePropertySource in project spring-cloud-consul by spring-cloud.

the class ConsulPropertySourceLocator method locate.

@Override
@Retryable(interceptor = "consulRetryInterceptor")
public PropertySource<?> locate(Environment environment) {
    if (environment instanceof ConfigurableEnvironment) {
        ConfigurableEnvironment env = (ConfigurableEnvironment) environment;
        String appName = properties.getName();
        if (appName == null) {
            appName = env.getProperty("spring.application.name");
        }
        List<String> profiles = Arrays.asList(env.getActiveProfiles());
        String prefix = this.properties.getPrefix();
        List<String> suffixes = new ArrayList<>();
        if (this.properties.getFormat() != FILES) {
            suffixes.add("/");
        } else {
            suffixes.add(".yml");
            suffixes.add(".yaml");
            suffixes.add(".properties");
        }
        String defaultContext = getContext(prefix, this.properties.getDefaultContext());
        for (String suffix : suffixes) {
            this.contexts.add(defaultContext + suffix);
        }
        for (String suffix : suffixes) {
            addProfiles(this.contexts, defaultContext, profiles, suffix);
        }
        String baseContext = getContext(prefix, appName);
        for (String suffix : suffixes) {
            this.contexts.add(baseContext + suffix);
        }
        for (String suffix : suffixes) {
            addProfiles(this.contexts, baseContext, profiles, suffix);
        }
        Collections.reverse(this.contexts);
        CompositePropertySource composite = new CompositePropertySource("consul");
        for (String propertySourceContext : this.contexts) {
            try {
                ConsulPropertySource propertySource = null;
                if (this.properties.getFormat() == FILES) {
                    Response<GetValue> response = this.consul.getKVValue(propertySourceContext, this.properties.getAclToken());
                    addIndex(propertySourceContext, response.getConsulIndex());
                    if (response.getValue() != null) {
                        ConsulFilesPropertySource filesPropertySource = new ConsulFilesPropertySource(propertySourceContext, this.consul, this.properties);
                        filesPropertySource.init(response.getValue());
                        propertySource = filesPropertySource;
                    }
                } else {
                    propertySource = create(propertySourceContext, contextIndex);
                }
                if (propertySource != null) {
                    composite.addPropertySource(propertySource);
                }
            } catch (Exception e) {
                if (this.properties.isFailFast()) {
                    log.error("Fail fast is set and there was an error reading configuration from consul.");
                    ReflectionUtils.rethrowRuntimeException(e);
                } else {
                    log.warn("Unable to load consul config from " + propertySourceContext, e);
                }
            }
        }
        return composite;
    }
    return null;
}

Example 22 with CompositePropertySource

use of org.springframework.core.env.CompositePropertySource in project spring-cloud-consul by spring-cloud.

the class ConsulPropertySourceLocatorFilesTests method propertySourcesFound.

@Test
public void propertySourcesFound() throws Exception {
    String foo = this.environment.getProperty("foo");
    assertThat("foo was wrong", foo, is(equalTo("bar-app-dev")));
    String myBaz = this.environment.getProperty("my.baz");
    assertThat("my.baz was wrong", myBaz, is(equalTo("bar-app-dev")));
    MutablePropertySources propertySources = this.environment.getPropertySources();
    PropertySource<?> bootstrapProperties = propertySources.get("bootstrapProperties");
    assertThat("bootstrapProperties was null", bootstrapProperties, is(notNullValue()));
    assertThat("bootstrapProperties was wrong type", bootstrapProperties.getClass(), is(typeCompatibleWith(CompositePropertySource.class)));
    Collection<PropertySource<?>> consulSources = ((CompositePropertySource) bootstrapProperties).getPropertySources();
    assertThat("consulSources was wrong size", consulSources, hasSize(1));
    PropertySource<?> consulSource = consulSources.iterator().next();
    assertThat("consulSource was wrong type", consulSource.getClass(), is(typeCompatibleWith(CompositePropertySource.class)));
    Collection<PropertySource<?>> fileSources = ((CompositePropertySource) consulSource).getPropertySources();
    assertThat("fileSources was wrong size", fileSources, hasSize(4));
    assertFileSourceNames(fileSources, APP_NAME_DEV_PROPS, APP_NAME_PROPS, APPLICATION_DEV_YML, APPLICATION_YML);
}

Example 23 with CompositePropertySource

use of org.springframework.core.env.CompositePropertySource in project gravitee-management-rest-api by gravitee-io.

the class EnvironmentUtils method getAllProperties.

public static Map<String, Object> getAllProperties(PropertySource<?> aPropSource) {
    Map<String, Object> result = new HashMap<>();
    if (aPropSource instanceof CompositePropertySource) {
        CompositePropertySource cps = (CompositePropertySource) aPropSource;
        cps.getPropertySources().forEach(ps -> addAll(result, getAllProperties(ps)));
        return result;
    }
    if (aPropSource instanceof EnumerablePropertySource<?>) {
        EnumerablePropertySource<?> ps = (EnumerablePropertySource<?>) aPropSource;
        Arrays.asList(ps.getPropertyNames()).forEach(key -> result.put(key, ps.getProperty(key)));
        return result;
    }
    // note: Most descendants of PropertySource are EnumerablePropertySource. There are some
    // few others like JndiPropertySource or StubPropertySource
    LOGGER.debug("Given PropertySource is instanceof " + aPropSource.getClass().getName() + " and cannot be iterated");
    return result;
}

Example 24 with CompositePropertySource

use of org.springframework.core.env.CompositePropertySource in project spring-cloud-config by spring-cloud.

the class ConfigServicePropertySourceLocator method locate.

@Override
@Retryable(interceptor = "configServerRetryInterceptor")
public org.springframework.core.env.PropertySource<?> locate(org.springframework.core.env.Environment environment) {
    ConfigClientProperties properties = this.defaultProperties.override(environment);
    CompositePropertySource composite = new CompositePropertySource("configService");
    RestTemplate restTemplate = this.restTemplate == null ? getSecureRestTemplate(properties) : this.restTemplate;
    Exception error = null;
    String errorBody = null;
    logger.info("Fetching config from server at: " + properties.getRawUri());
    try {
        String[] labels = new String[] { "" };
        if (StringUtils.hasText(properties.getLabel())) {
            labels = StringUtils.commaDelimitedListToStringArray(properties.getLabel());
        }
        String state = ConfigClientStateHolder.getState();
        // Try all the labels until one works
        for (String label : labels) {
            Environment result = getRemoteEnvironment(restTemplate, properties, label.trim(), state);
            if (result != null) {
                log(result);
                if (result.getPropertySources() != null) {
                    // result.getPropertySources() can be null if using xml
                    for (PropertySource source : result.getPropertySources()) {
                        @SuppressWarnings("unchecked") Map<String, Object> map = (Map<String, Object>) source.getSource();
                        composite.addPropertySource(new MapPropertySource(source.getName(), map));
                    }
                }
                if (StringUtils.hasText(result.getState()) || StringUtils.hasText(result.getVersion())) {
                    HashMap<String, Object> map = new HashMap<>();
                    putValue(map, "config.client.state", result.getState());
                    putValue(map, "config.client.version", result.getVersion());
                    composite.addFirstPropertySource(new MapPropertySource("configClient", map));
                }
                return composite;
            }
        }
    } catch (HttpServerErrorException e) {
        error = e;
        if (MediaType.APPLICATION_JSON.includes(e.getResponseHeaders().getContentType())) {
            errorBody = e.getResponseBodyAsString();
        }
    } catch (Exception e) {
        error = e;
    }
    if (properties.isFailFast()) {
        throw new IllegalStateException("Could not locate PropertySource and the fail fast property is set, failing", error);
    }
    logger.warn("Could not locate PropertySource: " + (errorBody == null ? error == null ? "label not found" : error.getMessage() : errorBody));
    return null;
}

Example 25 with CompositePropertySource

use of org.springframework.core.env.CompositePropertySource in project cas by apereo.

the class MongoDbPropertySourceLocator method locate.

@Override
public PropertySource<?> locate(final Environment environment) {
    if (environment instanceof ConfigurableEnvironment) {
        val sourceName = MongoDbPropertySource.class.getSimpleName();
        val composite = new CompositePropertySource(sourceName);
        val source = new MongoDbPropertySource(sourceName, mongo);
        composite.addFirstPropertySource(source);
        return composite;
    }
    return null;
}