Set CacheControl for static resources from properties file

Author: tinexw

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/ResourceProperties.java

@@ -16,7 +16,13 @@
 
 package org.springframework.boot.autoconfigure.web;
 
+import java.util.concurrent.TimeUnit;
+
+import javax.annotation.PostConstruct;
+
 import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.context.properties.NestedConfigurationProperty;
+import org.springframework.util.Assert;
 
 /**
  * Properties used to configure resource handling.
@@ -25,14 +31,15 @@
  * @author Brian Clozel
  * @author Dave Syer
  * @author Venil Noronha
+ * @author Kristine Jetzke
  * @since 1.1.0
  */
 @ConfigurationProperties(prefix = "spring.resources", ignoreUnknownFields = false)
 public class ResourceProperties {
 
 	private static final String[] CLASSPATH_RESOURCE_LOCATIONS = {
 			"classpath:/META-INF/resources/", "classpath:/resources/",
-			"classpath:/static/", "classpath:/public/" };
+			"classpath:/static/", "classpath:/public/"};
 
 	/**
 	 * Locations of static resources. Defaults to classpath:[/META-INF/resources/,
@@ -41,10 +48,17 @@ public class ResourceProperties {
 	private String[] staticLocations = CLASSPATH_RESOURCE_LOCATIONS;
 
 	/**
-	 * Cache period for the resources served by the resource handler, in seconds.
+	 * Cache period for the resources served by the resource handler, in seconds. Either
+	 * {@link #cachePeriod} or {@link #cacheControl} can be set.
 	 */
 	private Integer cachePeriod;
 
+	/**
+	 * Cache control headers. Either {@link #cachePeriod} or {@link #cacheControl} can be set.
+	 */
+	@NestedConfigurationProperty
+	private CacheControl cacheControl;
+
 	/**
 	 * Enable default resource handling.
 	 */
@@ -77,6 +91,15 @@ public void setCachePeriod(Integer cachePeriod) {
 		this.cachePeriod = cachePeriod;
 	}
 
+	public CacheControl getCacheControl() {
+		return this.cacheControl;
+	}
+
+	public void setCacheControl(CacheControl cacheControl) {
+		this.cacheControl = cacheControl;
+	}
+
+
 	public boolean isAddMappings() {
 		return this.addMappings;
 	}
@@ -89,6 +112,36 @@ public Chain getChain() {
 		return this.chain;
 	}
 
+	public org.springframework.http.CacheControl createCacheControl() {
+		if (this.cachePeriod != null) {
+			return org.springframework.http.CacheControl.maxAge(this.cachePeriod, TimeUnit.SECONDS);
+		}
+		if (this.cacheControl != null) {
+			return CacheControl.createCacheControl(this.cacheControl);
+		}
+		return null;
+	}
+
+	@PostConstruct
+	public void checkIncompatibleCacheOptions() {
+		Assert.state(this.cachePeriod == null || this.cacheControl == null,
+				"Only one of cache-period or cache-control may be set.");
+		if (this.cacheControl != null) {
+			if (this.cacheControl.getMaxAge() != null) {
+				Assert.state(!Boolean.TRUE.equals(this.cacheControl.getNoCache()), "no-cache may not be set if max-age is set.");
+				Assert.state(!Boolean.TRUE.equals(this.cacheControl.getNoStore()), "no-store may not be set if max-age is set.");
+			}
+			if (this.cacheControl.getNoCache() != null) {
+				Assert.state(this.cacheControl.getMaxAge() == null, "max-age may not be set if no-cache is set.");
+				Assert.state(!Boolean.TRUE.equals(this.cacheControl.getNoStore()), "no-store may not be set if no-cache is set.");
+			}
+			if (this.cacheControl.getNoStore() != null) {
+				Assert.state(this.cacheControl.getMaxAge() == null, "max-age may not be set if no-store is set.");
+				Assert.state(!Boolean.TRUE.equals(this.cacheControl.getNoCache()), "no-cache may not be set if no-store is set.");
+			}
+		}
+	}
+
 	/**
 	 * Configuration for the Spring Resource Handling chain.
 	 */
@@ -121,8 +174,9 @@ public static class Chain {
 		/**
 		 * Return whether the resource chain is enabled. Return {@code null} if no
 		 * specific settings are present.
-		 * @return whether the resource chain is enabled or {@code null} if no specified
-		 * settings are present.
+		 *
+		 * @return whether the resource chain is enabled or {@code null} if no specified settings are
+		 * present.
 		 */
 		public Boolean getEnabled() {
 			return getEnabled(getStrategy().getFixed().isEnabled(),
@@ -200,7 +254,7 @@ public static class Content {
 		/**
 		 * Comma-separated list of patterns to apply to the Version Strategy.
 		 */
-		private String[] paths = new String[] { "/**" };
+		private String[] paths = new String[]{"/**"};
 
 		public boolean isEnabled() {
 			return this.enabled;
@@ -233,7 +287,7 @@ public static class Fixed {
 		/**
 		 * Comma-separated list of patterns to apply to the Version Strategy.
 		 */
-		private String[] paths = new String[] { "/**" };
+		private String[] paths = new String[]{"/**"};
 
 		/**
 		 * Version string to use for the Version Strategy.
@@ -266,4 +320,173 @@ public void setVersion(String version) {
 
 	}
 
+	/**
+	 * Configuration for the Cache Control header.
+	 */
+
+	public static class CacheControl {
+
+		private Long maxAge;
+
+		private Boolean noCache;
+
+		private Boolean noStore;
+
+		private Boolean mustRevalidate;
+
+		private Boolean noTransform;
+
+		private Boolean cachePublic;
+
+		private Boolean cachePrivate;
+
+		private Boolean proxyRevalidate;
+
+		private Long staleWhileRevalidate;
+
+		private Long staleIfError;
+
+		private Long sMaxAge;
+
+		public Long getMaxAge() {
+			return this.maxAge;
+		}
+
+		public void setMaxAge(Long maxAge) {
+			this.maxAge = maxAge;
+		}
+
+		public Boolean getNoCache() {
+			return this.noCache;
+		}
+
+		public void setNoCache(Boolean noCache) {
+			this.noCache = noCache;
+		}
+
+		public Boolean getNoStore() {
+			return this.noStore;
+		}
+
+		public void setNoStore(Boolean noStore) {
+			this.noStore = noStore;
+		}
+
+		public Boolean getMustRevalidate() {
+			return this.mustRevalidate;
+		}
+
+		public void setMustRevalidate(Boolean mustRevalidate) {
+			this.mustRevalidate = mustRevalidate;
+		}
+
+		public Boolean getNoTransform() {
+			return this.noTransform;
+		}
+
+		public void setNoTransform(Boolean noTransform) {
+			this.noTransform = noTransform;
+		}
+
+		public Boolean getCachePublic() {
+			return this.cachePublic;
+		}
+
+		public void setCachePublic(Boolean cachePublic) {
+			this.cachePublic = cachePublic;
+		}
+
+		public Boolean getCachePrivate() {
+			return this.cachePrivate;
+		}
+
+		public void setCachePrivate(Boolean cachePrivate) {
+			this.cachePrivate = cachePrivate;
+		}
+
+		public Boolean getProxyRevalidate() {
+			return this.proxyRevalidate;
+		}
+
+		public void setProxyRevalidate(Boolean proxyRevalidate) {
+			this.proxyRevalidate = proxyRevalidate;
+		}
+
+		public Long getStaleWhileRevalidate() {
+			return this.staleWhileRevalidate;
+		}
+
+		public void setStaleWhileRevalidate(Long staleWhileRevalidate) {
+			this.staleWhileRevalidate = staleWhileRevalidate;
+		}
+
+		public Long getStaleIfError() {
+			return this.staleIfError;
+		}
+
+		public void setStaleIfError(Long staleIfError) {
+			this.staleIfError = staleIfError;
+		}
+
+		public Long getsMaxAge() {
+			return this.sMaxAge;
+		}
+
+		public void setsMaxAge(Long sMaxAge) {
+			this.sMaxAge = sMaxAge;
+		}
+
+		static org.springframework.http.CacheControl createCacheControl(CacheControl cacheControlProperties) {
+			org.springframework.http.CacheControl cacheControl;
+			if (cacheControlProperties.getMaxAge() != null) {
+				cacheControl = org.springframework.http.CacheControl.maxAge(cacheControlProperties.getMaxAge(), TimeUnit.SECONDS);
+			}
+			else if (Boolean.TRUE.equals(cacheControlProperties.getNoCache())) {
+				cacheControl = org.springframework.http.CacheControl.noCache();
+			}
+			else if (Boolean.TRUE.equals(cacheControlProperties.getNoStore())) {
+				cacheControl = org.springframework.http.CacheControl.noStore();
+			}
+			else {
+				cacheControl = org.springframework.http.CacheControl.empty();
+			}
+			if (Boolean.TRUE.equals(cacheControlProperties.getMustRevalidate())) {
+				cacheControl = cacheControl.mustRevalidate();
+			}
+			if (Boolean.TRUE.equals(cacheControlProperties.getNoTransform())) {
+				cacheControl = cacheControl.noTransform();
+			}
+			if (Boolean.TRUE.equals(cacheControlProperties.getCachePublic())) {
+				cacheControl = cacheControl.cachePublic();
+			}
+			if (Boolean.TRUE.equals(cacheControlProperties.getCachePrivate())) {
+				cacheControl = cacheControl.cachePrivate();
+			}
+			if (Boolean.TRUE.equals(cacheControlProperties.getProxyRevalidate())) {
+				cacheControl = cacheControl.proxyRevalidate();
+			}
+			if (cacheControlProperties.getStaleWhileRevalidate() != null) {
+				cacheControl = cacheControl.staleWhileRevalidate(
+						cacheControlProperties.getStaleWhileRevalidate(),
+						TimeUnit.SECONDS);
+			}
+			if (cacheControlProperties.getStaleIfError() != null) {
+				cacheControl = cacheControl.staleIfError(
+						cacheControlProperties.getStaleIfError(),
+						TimeUnit.SECONDS);
+			}
+			if (cacheControlProperties.getsMaxAge() != null) {
+				cacheControl = cacheControl.sMaxAge(
+						cacheControlProperties.getsMaxAge(),
+						TimeUnit.SECONDS);
+			}
+			return cacheControl;
+		}
+
+
+
+
+	}
+
+
 }

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/servlet/WebMvcAutoConfiguration.java

@@ -72,6 +72,7 @@
 import org.springframework.format.Formatter;
 import org.springframework.format.FormatterRegistry;
 import org.springframework.format.datetime.DateFormatter;
+import org.springframework.http.CacheControl;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.MediaType;
 import org.springframework.http.converter.HttpMessageConverter;
@@ -132,6 +133,7 @@
  * @author Sébastien Deleuze
  * @author Eddú Meléndez
  * @author Stephane Nicoll
+ * @author Kristine Jetzke
  */
 @Configuration
 @ConditionalOnWebApplication(type = Type.SERVLET)
@@ -305,21 +307,21 @@ public void addResourceHandlers(ResourceHandlerRegistry registry) {
 				logger.debug("Default resource handling disabled");
 				return;
 			}
-			Integer cachePeriod = this.resourceProperties.getCachePeriod();
+			CacheControl cacheControl = this.resourceProperties.createCacheControl();
 			if (!registry.hasMappingForPattern("/webjars/**")) {
 				customizeResourceHandlerRegistration(
 						registry.addResourceHandler("/webjars/**")
 								.addResourceLocations(
 										"classpath:/META-INF/resources/webjars/")
-						.setCachePeriod(cachePeriod));
+						.setCacheControl(cacheControl));
 			}
 			String staticPathPattern = this.mvcProperties.getStaticPathPattern();
 			if (!registry.hasMappingForPattern(staticPathPattern)) {
 				customizeResourceHandlerRegistration(
 						registry.addResourceHandler(staticPathPattern)
 								.addResourceLocations(getResourceLocations(
 										this.resourceProperties.getStaticLocations()))
-						.setCachePeriod(cachePeriod));
+						.setCacheControl(cacheControl));
 			}
 		}