@Override
public void initialize(ConfigurableWebApplicationContext applicationContext) {
Resource resource = null;
ServletContext servletContext = applicationContext.getServletContext();
WebApplicationContextUtils.initServletPropertySources(
applicationContext.getEnvironment().getPropertySources(),
servletContext,
applicationContext.getServletConfig());
ServletConfig servletConfig = applicationContext.getServletConfig();
String locations =
servletConfig == null
? null
: servletConfig.getInitParameter(PROFILE_CONFIG_FILE_LOCATIONS);
resource = getResource(servletContext, applicationContext, locations);
if (resource == null) {
servletContext.log(
"No YAML environment properties from servlet. Defaulting to servlet context.");
locations = servletContext.getInitParameter(PROFILE_CONFIG_FILE_LOCATIONS);
resource = getResource(servletContext, applicationContext, locations);
}
try {
servletContext.log("Loading YAML environment properties from location: " + resource);
YamlMapFactoryBean factory = new YamlMapFactoryBean();
factory.setResolutionMethod(ResolutionMethod.OVERRIDE_AND_IGNORE);
List<Resource> resources = new ArrayList<Resource>();
String defaultLocation =
servletConfig == null
? null
: servletConfig.getInitParameter(PROFILE_CONFIG_FILE_DEFAULT);
if (defaultLocation != null) {
Resource defaultResource = new ClassPathResource(defaultLocation);
if (defaultResource.exists()) {
resources.add(defaultResource);
}
}
resources.add(resource);
factory.setResources(resources.toArray(new Resource[resources.size()]));
Map<String, Object> map = factory.getObject();
String yamlStr = (new Yaml()).dump(map);
map.put(rawYamlKey, yamlStr);
NestedMapPropertySource properties = new NestedMapPropertySource("servletConfigYaml", map);
applicationContext.getEnvironment().getPropertySources().addLast(properties);
applySpringProfiles(applicationContext.getEnvironment(), servletContext);
applyLog4jConfiguration(applicationContext.getEnvironment(), servletContext);
} catch (Exception e) {
servletContext.log("Error loading YAML environment properties from location: " + resource, e);
}
}
protected void configureAndRefreshWebApplicationContext(ConfigurableWebApplicationContext wac) {
if (ObjectUtils.identityToString(wac).equals(wac.getId())) {
// The application context id is still set to its original default value
// -> assign a more useful id based on available information
if (this.contextId != null) {
wac.setId(this.contextId);
} else {
// Generate default id...
wac.setId(
ConfigurableWebApplicationContext.APPLICATION_CONTEXT_ID_PREFIX
+ ObjectUtils.getDisplayString(getServletContext().getContextPath())
+ "/"
+ getServletName());
}
}
wac.setServletContext(getServletContext());
wac.setServletConfig(getServletConfig());
wac.setNamespace(getNamespace());
wac.addApplicationListener(new SourceFilteringListener(wac, new ContextRefreshListener()));
// The wac environment's #initPropertySources will be called in any case when the context
// is refreshed; do it eagerly here to ensure servlet property sources are in place for
// use in any post-processing or initialization that occurs below prior to #refresh
ConfigurableEnvironment env = wac.getEnvironment();
if (env instanceof ConfigurableWebEnvironment) {
((ConfigurableWebEnvironment) env)
.initPropertySources(getServletContext(), getServletConfig());
}
postProcessWebApplicationContext(wac);
applyInitializers(wac);
wac.refresh();
}
@Override
public void initialize(ConfigurableWebApplicationContext applicationContext) {
// The profile can be read from property file or environment variable
applicationContext.getEnvironment().setActiveProfiles("dev");
}
private Resource getResource(
ServletContext servletContext,
ConfigurableWebApplicationContext applicationContext,
String locations) {
Resource resource = null;
String[] configFileLocations =
locations == null
? DEFAULT_PROFILE_CONFIG_FILE_LOCATIONS
: StringUtils.commaDelimitedListToStringArray(locations);
for (String location : configFileLocations) {
location = applicationContext.getEnvironment().resolvePlaceholders(location);
servletContext.log("Testing for YAML resources at: " + location);
resource = applicationContext.getResource(location);
if (resource != null && resource.exists()) {
break;
}
}
return resource;
}
@Override
public void initialize(ConfigurableWebApplicationContext ctx) {
ConfigurableEnvironment environment = ctx.getEnvironment();
environment.setActiveProfiles("prod");
}
