// Compute the cache key
String scopeKey = "cached_markup." + invocation.getWindowContext().getId();
// We use the principal scope to avoid security issues like a user loggedout seeing a cached entry
// by a previous logged in user
UserContext userContext = invocation.getUserContext();
//
if (invocation instanceof RenderInvocation)
{
RenderInvocation renderInvocation = (RenderInvocation)invocation;
//
StateString navigationalState = renderInvocation.getNavigationalState();
Map<String, String[]> publicNavigationalState = renderInvocation.getPublicNavigationalState();
WindowState windowState = renderInvocation.getWindowState();
Mode mode = renderInvocation.getMode();
//
CacheEntry cachedEntry = (CacheEntry)userContext.getAttribute(scopeKey);
//
if (cachedEntry != null)
{
// Check time validity for fragment
boolean useEntry = false;
StateString entryNavigationalState = cachedEntry.navigationalState;
Map<String, String[]> entryPublicNavigationalState = cachedEntry.publicNavigationalState;
// Then check nav state equality
if (navigationalState == null)
{
if (entryNavigationalState == null)
{
useEntry = true;
}
else if (entryNavigationalState instanceof ParametersStateString)
{
// We consider a parameters state string empty equivalent to a null value
useEntry = ((ParametersStateString)entryNavigationalState).getSize() == 0;
}
}
else if (entryNavigationalState == null)
{
if (navigationalState instanceof ParametersStateString)
{
useEntry = ((ParametersStateString)navigationalState).getSize() == 0;
}
}
else
{
useEntry = navigationalState.equals(entryNavigationalState);
}
// Check public nav state equality
if (useEntry)
{
if (publicNavigationalState == null)
{
if (entryPublicNavigationalState == null)
{
//
}
else
{
useEntry = entryPublicNavigationalState.size() == 0;
}
}
else if (entryPublicNavigationalState == null)
{
useEntry = publicNavigationalState.size() == 0;
}
else
{
ParameterMap publicPM = ParameterMap.wrap(publicNavigationalState);
ParameterMap entryPM = ParameterMap.wrap(entryPublicNavigationalState);
useEntry = publicPM.equals(entryPM);
}
}
// Then check window state equality
useEntry &= windowState.equals(cachedEntry.windowState);
// Then check mode equality
useEntry &= mode.equals(cachedEntry.mode);
// Clean if it is null
if (!useEntry)
{
cachedEntry = null;
userContext.setAttribute(scopeKey, null);
}
}
//
final ContentResponse cachedContent = cachedEntry != null ? cachedEntry.contentRef.getContent() : null;
// If no valid content we must invoke
long now = System.currentTimeMillis();
if (cachedContent == null || cachedEntry.expirationTimeMillis < now)
{
// Set validation token for revalidation only we have have a content
if (cachedContent != null)
{
renderInvocation.setValidationToken(cachedEntry.validationToken);
}
// Invoke
PortletInvocationResponse response = super.invoke(invocation);
// If we had a cached content that was revalidated we substitute
if (response instanceof RevalidateMarkupResponse && cachedContent != null)
{
// Normally we receive such response when the validation token was set which implies we have an existing content
// We substitute with the appropriate content response
RevalidateMarkupResponse revalidate = (RevalidateMarkupResponse)response;
CacheControl control = revalidate.getCacheControl();
if (cachedContent instanceof FragmentResponse)
{
response = new FragmentResponse((FragmentResponse)cachedContent, control);
}
else
{
response = new ContentResponse(cachedContent, control);
}
}
// If we have a content cache it whenever it is possible
if (response instanceof ContentResponse)
{
ContentResponse contentResponse = (ContentResponse)response;
CacheControl control = contentResponse.getCacheControl();
//
if (control != null)
{
// Compute expiration time, i.e when it will expire
long expirationTimeMillis = 0;
if (control.getExpirationSecs() == -1)
{
expirationTimeMillis = Long.MAX_VALUE;
}
else if (control.getExpirationSecs() > 0)
{
expirationTimeMillis = System.currentTimeMillis() + control.getExpirationSecs() * 1000;
}
// Cache if we can
if (expirationTimeMillis > 0)
{
// Use validation token if any
String validationToken = null;
if (control.getValidationToken() != null)
{
validationToken = control.getValidationToken();
}
else if (cachedEntry != null)
{
validationToken = cachedEntry.validationToken;
}
CacheEntry cacheEntry = new CacheEntry(
navigationalState,
publicNavigationalState,
windowState,
mode,
contentResponse,
expirationTimeMillis,
validationToken);
userContext.setAttribute(scopeKey, cacheEntry);
}
}
}
//
return response;
}
else
{
// Use the cached content
return cachedContent;
}
}
else
{
// Invalidate
userContext.setAttribute(scopeKey, null);
// Invoke
return super.invoke(invocation);
}
}