diff --git a/spring-data-geode/src/test/java/org/springframework/data/gemfire/cache/CacheErrorHandlerIntegrationTests.java b/spring-data-geode/src/test/java/org/springframework/data/gemfire/cache/CacheErrorHandlerIntegrationTests.java new file mode 100644 index 00000000..d8fee97b --- /dev/null +++ b/spring-data-geode/src/test/java/org/springframework/data/gemfire/cache/CacheErrorHandlerIntegrationTests.java @@ -0,0 +1,141 @@ +/* + * Copyright 2019 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 + * + * https://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.data.gemfire.cache; + +import static org.assertj.core.api.Assertions.assertThat; + +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; + +import org.apache.geode.cache.DataPolicy; +import org.apache.geode.cache.Region; +import org.apache.geode.cache.client.ClientCache; +import org.apache.geode.cache.client.NoAvailableServersException; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.cache.Cache; +import org.springframework.cache.annotation.Cacheable; +import org.springframework.cache.annotation.CachingConfigurerSupport; +import org.springframework.cache.interceptor.CacheErrorHandler; +import org.springframework.cache.interceptor.SimpleCacheErrorHandler; +import org.springframework.context.annotation.Bean; +import org.springframework.data.gemfire.cache.config.EnableGemfireCaching; +import org.springframework.data.gemfire.config.annotation.ClientCacheApplication; +import org.springframework.data.gemfire.config.annotation.EnableCachingDefinedRegions; +import org.springframework.lang.Nullable; +import org.springframework.stereotype.Service; +import org.springframework.test.context.ContextConfiguration; +import org.springframework.test.context.junit4.SpringRunner; + +/** + * Integration Tests demonstrating the use of Spring's {@link CacheErrorHandler} with Apache Geode + * as the caching provider. + * + * @author John Blum + * @see org.junit.Test + * @see org.apache.geode.cache.Region + * @see org.apache.geode.cache.client.ClientCache + * @see org.springframework.cache.annotation.Cacheable + * @see org.springframework.cache.annotation.CachingConfigurerSupport + * @see org.springframework.cache.interceptor.CacheErrorHandler + * @see org.springframework.data.gemfire.cache.config.EnableGemfireCaching + * @see org.springframework.data.gemfire.config.annotation.ClientCacheApplication + * @see org.springframework.data.gemfire.config.annotation.EnableCachingDefinedRegions + * @see org.springframework.test.context.ContextConfiguration + * @see org.springframework.test.context.junit4.SpringRunner + * @since 2.3.0 + */ +@RunWith(SpringRunner.class) +@ContextConfiguration +@SuppressWarnings("unused") +public class CacheErrorHandlerIntegrationTests { + + private static final String GEMFIRE_LOG_LEVEL = "error"; + + @Autowired + private ClientCache clientCache; + + @Autowired + private TestCacheableService cacheableService; + + @Before + public void assertSetup() { + + assertThat(this.clientCache).isNotNull(); + assertThat(this.cacheableService).isNotNull(); + + Region testCache = this.clientCache.getRegion("/TestCache"); + + assertThat(testCache).isNotNull(); + assertThat(testCache.getName()).isEqualTo("TestCache"); + assertThat(testCache.getAttributes()).isNotNull(); + assertThat(testCache.getAttributes().getDataPolicy()).isEqualTo(DataPolicy.EMPTY); + assertThat(testCache.getAttributes().getPoolName()).isEqualTo("DEFAULT"); + } + + @Test + public void cacheServiceOperationIsSuccessful() { + assertThat(this.cacheableService.processCacheableOperation("one")).isEqualTo("TEST"); + } + + @ClientCacheApplication(logLevel = GEMFIRE_LOG_LEVEL) + @EnableCachingDefinedRegions + @EnableGemfireCaching + static class GeodeClientConfiguration extends CachingConfigurerSupport { + + @Nullable @Override + public CacheErrorHandler errorHandler() { + return new TestCacheErrorHandler(); + } + + @Bean + TestCacheableService testCacheableService() { + return new TestCacheableService(); + } + } + + @Service + static class TestCacheableService { + + @Cacheable(cacheNames = "TestCache") + public Object processCacheableOperation(String input) { + return "TEST"; + } + } + + static class TestCacheErrorHandler extends SimpleCacheErrorHandler { + + @Override + public void handleCacheGetError(RuntimeException exception, Cache cache, Object key) { + handleCacheGetAndPutError(exception, cache, key, null); + } + + @Override + public void handleCachePutError(RuntimeException exception, Cache cache, Object key, @Nullable Object value) { + handleCacheGetAndPutError(exception, cache, key, value); + } + + private void handleCacheGetAndPutError(RuntimeException exception, Cache cache, Object key, @Nullable Object value) { + + // If the Apache Geode cluster of servers is not available, then effectively disable caching + // by not rethrowing the Exception! + if (!(exception instanceof NoAvailableServersException)) { + throw exception; + } + } + } +}