com.godmonth.util.cache.CacheIntercepter Maven / Gradle / Ivy

Go to download
package com.godmonth.util.cache;

import java.util.List;

import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Required;
import org.springframework.cache.Cache;
import org.springframework.cache.Cache.ValueWrapper;
import org.springframework.expression.Expression;
import org.springframework.expression.ExpressionParser;
import org.springframework.expression.spel.standard.SpelExpressionParser;

/**
 * 无法区分overload方法
 * 
 * @author shenyue
 */
public class CacheIntercepter implements MethodInterceptor {

	private static final Logger logger = LoggerFactory.getLogger(CacheIntercepter.class);

	private static final ExpressionParser parser = new SpelExpressionParser();

	private List methodCacheDefinitions;
	private Cache cache;

	@Override
	public Object invoke(MethodInvocation invocation) throws Throwable {
		for (MethodCacheDefinition methodCacheDefinition : methodCacheDefinitions) {
			if (methodCacheDefinition.getMethodName().equals(invocation.getMethod().getName())) {
				Expression parseExpression = parser.parseExpression(methodCacheDefinition.getCacheKeyExpression());
				String cacheKey = (String) parseExpression.getValue(invocation.getArguments());
				ValueWrapper valueWrapper = cache.get(cacheKey);
				if (valueWrapper != null) {
					logger.trace("cache {}, hit: {}", cache.getName(), cacheKey);
					return valueWrapper.get();
				} else {
					logger.trace("cache {}, miss: {}", cache.getName(), cacheKey);
					Object result = invocation.proceed();
					if (result != null) {
						cache.put(cacheKey, result);
						logger.trace("cache {}, put: {}", cache.getName(), cacheKey);
					} else {
						logger.trace("cache {}, result null, skipped: {}", cache.getName(), cacheKey);
					}
					return result;
				}
			}
		}
		return invocation.proceed();
	}

	@Required
	public void setMethodCacheDefinitions(List methodCacheDefinitions) {
		this.methodCacheDefinitions = methodCacheDefinitions;
	}

	@Required
	public void setCache(Cache cache) {
		this.cache = cache;
	}

}