Cache operation invocation hook point
This commit adds a invokeOperation protected method in case one needs a hook point in the way the underlying cache method is invoked, and how exceptions that might be thrown by that invocation are handled. Issue: SPR-11540
This commit is contained in:
136
spring-context/src/test/java/org/springframework/cache/config/CustomInterceptorTests.java
vendored
Normal file
136
spring-context/src/test/java/org/springframework/cache/config/CustomInterceptorTests.java
vendored
Normal file
@@ -0,0 +1,136 @@
|
||||
/*
|
||||
* Copyright 2002-2014 the original author or authors.
|
||||
*
|
||||
* 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 org.springframework.cache.config;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
import org.junit.After;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
import org.springframework.cache.CacheManager;
|
||||
import org.springframework.cache.CacheTestUtils;
|
||||
import org.springframework.cache.annotation.EnableCaching;
|
||||
import org.springframework.cache.interceptor.CacheInterceptor;
|
||||
import org.springframework.cache.interceptor.CacheOperationInvoker;
|
||||
import org.springframework.cache.interceptor.CacheOperationSource;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
/**
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
*/
|
||||
public class CustomInterceptorTests {
|
||||
|
||||
protected ConfigurableApplicationContext ctx;
|
||||
|
||||
protected CacheableService<?> cs;
|
||||
|
||||
@Before
|
||||
public void setup() {
|
||||
ctx = new AnnotationConfigApplicationContext(EnableCachingConfig.class);
|
||||
cs = ctx.getBean("service", CacheableService.class);
|
||||
}
|
||||
|
||||
@After
|
||||
public void tearDown() {
|
||||
ctx.close();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onlyOneInterceptorIsAvailable() {
|
||||
Map<String, CacheInterceptor> interceptors = ctx.getBeansOfType(CacheInterceptor.class);
|
||||
assertEquals("Only one interceptor should be defined", 1, interceptors.size());
|
||||
CacheInterceptor interceptor = interceptors.values().iterator().next();
|
||||
assertEquals("Custom interceptor not defined", TestCacheInterceptor.class, interceptor.getClass());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void customInterceptorAppliesWithRuntimeException() {
|
||||
Object o = cs.throwUnchecked(0L);
|
||||
assertEquals(55L, o); // See TestCacheInterceptor
|
||||
}
|
||||
|
||||
@Test
|
||||
public void customInterceptorAppliesWithCheckedException() {
|
||||
try {
|
||||
cs.throwChecked(0L);
|
||||
fail("Should have failed");
|
||||
}
|
||||
catch (RuntimeException e) {
|
||||
assertNotNull("missing original exception", e.getCause());
|
||||
assertEquals(Exception.class, e.getCause().getClass());
|
||||
}
|
||||
catch (Exception e) {
|
||||
fail("Wrong exception type " + e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Configuration
|
||||
@EnableCaching
|
||||
static class EnableCachingConfig {
|
||||
|
||||
@Bean
|
||||
public CacheManager cacheManager() {
|
||||
return CacheTestUtils.createSimpleCacheManager("default", "primary", "secondary");
|
||||
}
|
||||
|
||||
@Bean
|
||||
public CacheableService<?> service() {
|
||||
return new DefaultCacheableService();
|
||||
}
|
||||
|
||||
@Bean
|
||||
public CacheInterceptor cacheInterceptor(CacheOperationSource cacheOperationSource) {
|
||||
CacheInterceptor cacheInterceptor = new TestCacheInterceptor();
|
||||
cacheInterceptor.setCacheManager(cacheManager());
|
||||
cacheInterceptor.setCacheOperationSources(cacheOperationSource);
|
||||
return cacheInterceptor;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* A test {@link CacheInterceptor} that handles special exception
|
||||
* types.
|
||||
*/
|
||||
static class TestCacheInterceptor extends CacheInterceptor {
|
||||
|
||||
@Override
|
||||
protected Object invokeOperation(CacheOperationInvoker invoker) {
|
||||
try {
|
||||
return super.invokeOperation(invoker);
|
||||
}
|
||||
catch (CacheOperationInvoker.ThrowableWrapper e) {
|
||||
Throwable original = e.getOriginal();
|
||||
if (original.getClass() == UnsupportedOperationException.class) {
|
||||
return 55L;
|
||||
}
|
||||
else {
|
||||
throw new CacheOperationInvoker.ThrowableWrapper(
|
||||
new RuntimeException("wrapping original", original));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user