Spring/spring-data-geode
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

DATAGEODE-323 - Add ClientCacheDefaultPoolResolver implementation.

Browse Source
This commit is contained in:
John Blum
2020-04-01 12:37:56 -07:00
parent b7c0f16d53
commit 75701480ea
2 changed files with 212 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
spring-data-geode/src/main/java/org/springframework/data/gemfire/client/support/ClientCacheDefaultPoolResolver.java Normal file
View File

@@ -0,0 +1,86 @@
/*
* Copyright 2020 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.client.support;
import java.util.Optional;
import org.apache.geode.cache.client.ClientCache;
import org.apache.geode.cache.client.Pool;
import org.springframework.data.gemfire.CacheResolver;
import org.springframework.data.gemfire.client.PoolResolver;
import org.springframework.lang.NonNull;
import org.springframework.lang.Nullable;
import org.springframework.util.Assert;
/**
* {@link PoolResolver} implementation used to resolve the {@literal DEFAULT} {@link Pool} from a {@link ClientCache}
* instance by lazily resolving the {@link ClientCache} instance and calling {@link ClientCache#getDefaultPool()}
* on {@literal DEFAULT} {@link Pool} resolution.
*
* @author John Blum
* @see org.apache.geode.cache.client.ClientCache
* @see org.apache.geode.cache.client.Pool
* @see org.springframework.data.gemfire.CacheResolver
* @see org.springframework.data.gemfire.client.PoolResolver
* @since 2.3.0
*/
public class ClientCacheDefaultPoolResolver implements PoolResolver {
private final CacheResolver<ClientCache> clientCacheResolver;
/**
* Constructs a new instance of {@link ClientCacheDefaultPoolResolver} initialized with a {@link CacheResolver}
* used to lazily resolve the {@link ClientCache} instance on {@link Pool} resolution.
*
* @param clientCacheResolver {@link CacheResolver} used to lazily resolve the {@link ClientCache} instance;
* must not be {@literal null}.
* @throws IllegalArgumentException if {@link CacheResolver} is {@literal null}.
* @see org.apache.geode.cache.client.ClientCache
* @see org.springframework.data.gemfire.CacheResolver
*/
public ClientCacheDefaultPoolResolver(@NonNull CacheResolver<ClientCache> clientCacheResolver) {
Assert.notNull(clientCacheResolver, "CacheResolver for ClientCache must not be null");
this.clientCacheResolver = clientCacheResolver;
}
/**
* Returns a reference to the configured {@link CacheResolver} used to (lazily) resolve
* the {@link ClientCache} instance.
*
* @return the configured {@link CacheResolver} used to resolve the {@link ClientCache} instance.
* @see org.apache.geode.cache.client.ClientCache
* @see org.springframework.data.gemfire.CacheResolver
*/
protected @NonNull CacheResolver<ClientCache> getClientCacheResolver() {
return this.clientCacheResolver;
}
/**
* @inheritDoc
*/
@Nullable @Override
public Pool resolve(@Nullable String poolName) {
return Optional.of(getClientCacheResolver())
.filter(it -> DEFAULT_POOL_NAME.equals(poolName))
.map(CacheResolver::resolve)
.map(ClientCache::getDefaultPool)
.orElse(null);
}
}

126
spring-data-geode/src/test/java/org/springframework/data/gemfire/client/support/ClientCacheDefaultPoolResolverUnitTests.java Normal file
View File

@@ -0,0 +1,126 @@
/*
* Copyright 2020 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.client.support;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoInteractions;
import static org.mockito.Mockito.when;
import org.junit.Test;
import org.apache.geode.cache.client.ClientCache;
import org.apache.geode.cache.client.Pool;
import org.springframework.data.gemfire.CacheResolver;
import org.springframework.data.gemfire.client.PoolResolver;
/**
* Unit Tests for {@link ClientCacheDefaultPoolResolver}.
*
* @author John Blum
* @see org.junit.Test
* @see org.mockito.Mock
* @see org.mockito.Mockito
* @see org.apache.geode.cache.client.ClientCache
* @see org.apache.geode.cache.client.Pool
* @see org.springframework.data.gemfire.CacheResolver
* @see org.springframework.data.gemfire.client.PoolResolver
* @see org.springframework.data.gemfire.client.support.ClientCacheDefaultPoolResolver
* @since 2.3.0
*/
public class ClientCacheDefaultPoolResolverUnitTests {
@Test
@SuppressWarnings("unchecked")
public void constructClientCacheDefaultPoolResolver() {
CacheResolver<ClientCache> mockClientCacheResolver = mock(CacheResolver.class);
ClientCacheDefaultPoolResolver poolResolver = new ClientCacheDefaultPoolResolver(mockClientCacheResolver);
assertThat(poolResolver).isNotNull();
assertThat(poolResolver.getClientCacheResolver()).isEqualTo(mockClientCacheResolver);
verifyNoInteractions(mockClientCacheResolver);
}
@Test(expected = IllegalArgumentException.class)
public void constructClientCacheDefaultPoolResolverWithNullCacheResolver() {
try {
new ClientCacheDefaultPoolResolver(null);
}
catch (IllegalArgumentException expected) {
assertThat(expected).hasMessage("CacheResolver for ClientCache must not be null");
assertThat(expected).hasNoCause();
throw expected;
}
}
@Test
public void resolvePoolFromClientCacheReturnsDefaultPool() {
ClientCache mockClientCache = mock(ClientCache.class);
Pool mockPool = mock(Pool.class, PoolResolver.DEFAULT_POOL_NAME);
when(mockClientCache.getDefaultPool()).thenReturn(mockPool);
CacheResolver<ClientCache> clientCacheResolver = () -> mockClientCache;
ClientCacheDefaultPoolResolver poolResolver = new ClientCacheDefaultPoolResolver(clientCacheResolver);
assertThat(poolResolver).isNotNull();
assertThat(poolResolver.getClientCacheResolver()).isEqualTo(clientCacheResolver);
assertThat(poolResolver.resolve(PoolResolver.DEFAULT_POOL_NAME)).isEqualTo(mockPool);
verify(mockClientCache, times(1)).getDefaultPool();
verifyNoInteractions(mockPool);
}
@Test
public void resolvePoolWhenClientCacheResolvesToNullIsNullSafe() {
CacheResolver<ClientCache> clientCacheResolver = () -> null;
ClientCacheDefaultPoolResolver poolResolver = new ClientCacheDefaultPoolResolver(clientCacheResolver);
assertThat(poolResolver).isNotNull();
assertThat(poolResolver.getClientCacheResolver()).isEqualTo(clientCacheResolver);
assertThat(poolResolver.resolve(PoolResolver.DEFAULT_POOL_NAME)).isNull();
}
@Test
public void resolvePoolWithNonDefaultPool() {
ClientCache mockClientCache = mock(ClientCache.class);
CacheResolver<ClientCache> clientCacheResolver = () -> mockClientCache;
ClientCacheDefaultPoolResolver poolResolver = new ClientCacheDefaultPoolResolver(clientCacheResolver);
assertThat(poolResolver).isNotNull();
assertThat(poolResolver.getClientCacheResolver()).isEqualTo(clientCacheResolver);
assertThat(poolResolver.resolve("CUSTOM")).isNull();
verifyNoInteractions(mockClientCache);
}
}