All Downloads are FREE. Search and download functionalities are using the official Maven repository.

javax.cache.annotation.impl.AbstractCachePutInterceptor Maven / Gradle / Ivy

/**
 *  Copyright 2011 Terracotta, Inc.
 *  Copyright 2011 Oracle America Incorporated
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package javax.cache.annotation.impl;


import java.lang.annotation.Annotation;

import javax.cache.Cache;
import javax.cache.annotation.CacheInvocationParameter;
import javax.cache.annotation.CacheKey;
import javax.cache.annotation.CacheKeyGenerator;
import javax.cache.annotation.CachePut;
import javax.cache.annotation.CacheResolver;


/**
 * Interceptor for {@link CachePut}
 * 
 * @author Rick Hightower
 * @author Eric Dalquist
 * @param  The intercepted method invocation
 * @since 1.0
 */
public abstract class AbstractCachePutInterceptor extends AbstractKeyedCacheInterceptor {
    
    /**
     * Handles the {@link Cache#put(Object, Object)} as specified for the {@link CachePut} annotation
     * 
     * @param cacheContextSource The intercepted invocation
     * @param invocation The intercepted invocation
     * @return The result from {@link #proceed(Object)}
     * @throws Throwable if {@link #proceed(Object)} threw
     */
    public Object cachePut(CacheContextSource cacheContextSource, I invocation) throws Throwable {
        final InternalCacheKeyInvocationContext cacheKeyInvocationContext = 
                cacheContextSource.getCacheKeyInvocationContext(invocation);
        final CachePutMethodDetails methodDetails = this.getStaticCacheKeyInvocationContext(cacheKeyInvocationContext, InterceptorType.CACHE_PUT);
        
        final CachePut cachePutAnnotation = methodDetails.getCacheAnnotation();
        final boolean afterInvocation = cachePutAnnotation.afterInvocation();
        
        final CacheInvocationParameter valueParameter = cacheKeyInvocationContext.getValueParameter();
        final Object value = valueParameter.getValue();

        if (!afterInvocation) {
            cacheValue(cacheKeyInvocationContext, methodDetails, value);
        }
        
        final Object result;
        try {
            //Call the annotated method
            result = this.proceed(invocation);
        } catch (Throwable t) {
            if (afterInvocation) {
                //If after invocation is true and if the throwable passes the include/exclude filters and then call put
                final Class[] cacheFor = cachePutAnnotation.cacheFor();
                final Class[] noCacheFor = cachePutAnnotation.noCacheFor();
                
                //Check for empty/null here since isIncluded returns true for those cases
                final boolean cache = ClassFilter.isIncluded(t, cacheFor, noCacheFor, false);
                
                //Exception is included
                if (cache) {
                    cacheValue(cacheKeyInvocationContext, methodDetails, value);
                }
            }

            throw t;
        }
        
        if (afterInvocation) {
            cacheValue(cacheKeyInvocationContext, methodDetails, value);
        }
        
        return result;
    }


    /**
     * Lookup the Cache, generate a CacheKey and store the value in the cache.
     * 
     * @param cacheKeyInvocationContext The invocation context 
     * @param methodDetails The details about the cached method
     * @param value The value to cache
     */
    protected void cacheValue(final InternalCacheKeyInvocationContext cacheKeyInvocationContext,
            final CachePutMethodDetails methodDetails, final Object value) {
        
        final Object cachedValue;
        if (value == null) {
            if (methodDetails.getCacheAnnotation().cacheNull()) {
                //Null values are cached, set value to the null placeholder 
                cachedValue = CacheContextSource.NULL_PLACEHOLDER;
            } else {
                //Ignore null values
                return;
            }
        } else {
            cachedValue = value;
        }
        
        
        final CacheResolver cacheResolver = methodDetails.getCacheResolver();
        final Cache cache = cacheResolver.resolveCache(cacheKeyInvocationContext);

        final CacheKeyGenerator cacheKeyGenerator = methodDetails.getCacheKeyGenerator();
        final CacheKey cacheKey = cacheKeyGenerator.generateCacheKey(cacheKeyInvocationContext);
        
        cache.put(cacheKey, cachedValue);
    }
}