// 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);
}
}
ContentResponse fragment = cachedEntry != null ? cachedEntry.contentRef.getContent() : null;
// If no valid fragment we must invoke
if (fragment == null || cachedEntry.expirationTimeMillis < System.currentTimeMillis())
{
// Set validation token for revalidation only we have have a fragment
if (fragment != null)
{
renderInvocation.setValidationToken(cachedEntry.validationToken);
}
// Invoke
PortletInvocationResponse response = super.invoke(invocation);
// Try to cache any fragment result
CacheControl control = null;
if (response instanceof ContentResponse)
{
fragment = (ContentResponse)response;
control = fragment.getCacheControl();
}
else if (response instanceof RevalidateMarkupResponse)
{
RevalidateMarkupResponse revalidate = (RevalidateMarkupResponse)response;
control = revalidate.getCacheControl();
}
// Compute expiration time, i.e when it will expire
long expirationTimeMillis = 0;
String validationToken = null;
if (control != null)
{
if (control.getExpirationSecs() == -1)
{
expirationTimeMillis = Long.MAX_VALUE;
}
else if (control.getExpirationSecs() > 0)
{
expirationTimeMillis = System.currentTimeMillis() + control.getExpirationSecs() * 1000;
}
if (control.getValidationToken() != null)
{
validationToken = control.getValidationToken();
}
else if (cachedEntry != null)
{
validationToken = cachedEntry.validationToken;
}
}
// Cache if we can
if (expirationTimeMillis > 0)
{
CacheEntry cacheEntry = new CacheEntry(
navigationalState,
publicNavigationalState,
windowState,
mode,
fragment,
expirationTimeMillis,
validationToken);
userContext.setAttribute(scopeKey, cacheEntry);
}
//
return response;
}
else
{
// Use the cached fragment
return fragment;
}
}
else
{
// Invalidate
userContext.setAttribute(scopeKey, null);
// Invoke
return super.invoke(invocation);
}
}