Skip to content
Permalink
Browse files

Allowing to provide cache policies on ResourcesRoute

  • Loading branch information
fcamblor committed Jan 23, 2015
1 parent c8f6ed6 commit f32586eabaeb653aa35c5424d855dedbcfb4fb8e
Showing with 61 additions and 1 deletion.
  1. +61 −1 restx-core/src/main/java/restx/ResourcesRoute.java
@@ -1,6 +1,7 @@
package restx;

import com.google.common.base.Optional;
import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableMap;
import com.google.common.io.ByteStreams;
import com.google.common.io.Resources;
@@ -10,6 +11,8 @@

import java.io.IOException;
import java.net.URL;
import java.util.Collections;
import java.util.List;

import static com.google.common.base.Preconditions.checkNotNull;

@@ -35,17 +38,60 @@
*/
private final String baseResourcePath;
private final ImmutableMap<String, String> aliases;
private final List<CachedResourcePolicy> cachedResourcePolicies;

public static class ResourceInfo {
String contentType;
String path;

public ResourceInfo(String contentType, String path) {
this.contentType = contentType;
this.path = path;
}

public String getContentType() {
return contentType;
}

public String getPath() {
return path;
}
}

public static class CachedResourcePolicy {
Predicate<ResourceInfo> matcher;
String cacheValue;

public CachedResourcePolicy(Predicate<ResourceInfo> matcher, String cacheValue) {
this.matcher = matcher;
this.cacheValue = cacheValue;
}

public boolean matches(String contentType, String path) {
return matcher.apply(new ResourceInfo(contentType, path));
}

public String getCacheValue() {
return cacheValue;
}
}


public ResourcesRoute(String name, String baseRestPath, String baseResourcePath) {
this(name, baseRestPath, baseResourcePath, ImmutableMap.<String, String>of());
}

public ResourcesRoute(String name, String baseRestPath, String baseResourcePath, ImmutableMap<String, String> aliases) {
this(name, baseRestPath, baseResourcePath, aliases, Collections.<CachedResourcePolicy>emptyList());
}

public ResourcesRoute(String name, String baseRestPath, String baseResourcePath, ImmutableMap<String, String> aliases, List<CachedResourcePolicy> cachedResourcePolicies) {
this.name = checkNotNull(name);
this.baseRestPath = ("/" + checkNotNull(baseRestPath) + "/").replaceAll("/+", "/");
this.baseResourcePath = checkNotNull(baseResourcePath)
.replace('.', '/').replaceAll("^/", "").replaceAll("/$", "") + "/";
this.aliases = checkNotNull(aliases);
this.cachedResourcePolicies = cachedResourcePolicies;
}

@Override
@@ -71,13 +117,27 @@ public void handle(RestxRequestMatch match, RestxRequest req, RestxResponse resp
);
resp.setLogLevel(RestxLogLevel.QUIET);
resp.setStatus(HttpStatus.OK);
resp.setContentType(HTTP.getContentTypeFromExtension(relativePath).or("application/octet-stream"));
String contentType = HTTP.getContentTypeFromExtension(relativePath).or("application/octet-stream");
Optional<CachedResourcePolicy> cachedResourcePolicy = cachePolicyMatching(contentType, relativePath);
if(cachedResourcePolicy.isPresent()) {
resp.setHeader("Cache-Control", cachedResourcePolicy.get().getCacheValue());
}
resp.setContentType(contentType);
Resources.asByteSource(resource).copyTo(resp.getOutputStream());
} catch (IllegalArgumentException e) {
notFound(resp, relativePath);
}
}

private Optional<CachedResourcePolicy> cachePolicyMatching(String contentType, String path) {
for(CachedResourcePolicy cachedResourcePolicy : cachedResourcePolicies){
if(cachedResourcePolicy.matches(contentType, path)){
return Optional.of(cachedResourcePolicy);
}
}
return Optional.absent();
}

private void notFound(RestxResponse resp, String relativePath) throws IOException {
resp.setStatus(HttpStatus.NOT_FOUND);
resp.setContentType("text/plain");

0 comments on commit f32586e

Please sign in to comment.
You can’t perform that action at this time.