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

Add shared Spring @Configuration class for enabling (client/Pool) subscriptions.

Browse Source 
This class will be used by both the ContinuousQueryAutoConfiguration class and the SpringSessionAutoConfiguration class.
This commit is contained in:
John Blum
2019-09-12 16:54:55 -07:00
parent 8e30b4ddd2
commit edb12fc523
2 changed files with 211 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
spring-geode-autoconfigure/src/main/java/org/springframework/geode/boot/autoconfigure/support/EnableSubscriptionConfiguration.java Normal file
View File

@@ -0,0 +1,63 @@
/*
* 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.geode.boot.autoconfigure.support;
import java.util.Optional;
import java.util.Set;
import org.apache.geode.cache.client.Pool;
import org.apache.shiro.util.CollectionUtils;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.gemfire.config.annotation.ClientCacheConfigurer;
import org.springframework.data.gemfire.config.annotation.PoolConfigurer;
/**
* A Spring {@link Configuration} class used to enable subscription on the Apache Geode & Pivotal GemFire
* {@literal DEFAULT} {@link Pool} as well as the SDG {@literal gemfirePool} {@link Pool}, only.
*
* @author John Blum
* @see org.apache.geode.cache.client.Pool
* @see org.springframework.context.annotation.Bean
* @see org.springframework.context.annotation.Configuration
* @see org.springframework.data.gemfire.config.annotation.ClientCacheConfigurer
* @see org.springframework.data.gemfire.config.annotation.PoolConfigurer
* @since 1.2.0
*/
@Configuration
@SuppressWarnings("unused")
public class EnableSubscriptionConfiguration {
private static final String DEFAULT_POOL_NAME = "DEFAULT";
private static final String GEMFIRE_POOL_NAME = "gemfirePool";
private static final Set<String> POOL_NAMES = CollectionUtils.asSet(DEFAULT_POOL_NAME, GEMFIRE_POOL_NAME);
@Bean
public ClientCacheConfigurer enableSubscriptionClientCacheConfigurer() {
return (beanName, clientCacheFactoryBean) -> clientCacheFactoryBean.setSubscriptionEnabled(true);
}
@Bean
public PoolConfigurer enableSubscriptionPoolConfigurer() {
return (beanName, poolFactoryBean) -> Optional.ofNullable(beanName)
.filter(POOL_NAMES::contains)
.ifPresent(poolName -> poolFactoryBean.setSubscriptionEnabled(true));
}
}

148
spring-geode-autoconfigure/src/test/java/org/springframework/geode/boot/autoconfigure/support/EnableSubscriptionConfigurationUnitTests.java Normal file
View File

@@ -0,0 +1,148 @@
/*
* 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.geode.boot.autoconfigure.support;
import static org.assertj.core.api.Assertions.assertThat;
import static org.springframework.data.gemfire.util.RuntimeExceptionFactory.newIllegalArgumentException;
import java.util.Optional;
import org.junit.Test;
import org.springframework.data.gemfire.client.ClientCacheFactoryBean;
import org.springframework.data.gemfire.client.PoolFactoryBean;
import org.springframework.util.ReflectionUtils;
/**
* Unit Tests for {@link EnableSubscriptionConfiguration}.
*
* @author John Blum
* @see org.junit.Test
* @see org.springframework.data.gemfire.client.ClientCacheFactoryBean
* @see org.springframework.data.gemfire.client.PoolFactoryBean
* @since 1.2.0
*/
public class EnableSubscriptionConfigurationUnitTests {
private EnableSubscriptionConfiguration configuration = new EnableSubscriptionConfiguration();
@SuppressWarnings("all")
private <T> T getFieldValue(Object target, String fieldName) {
return Optional.ofNullable(target)
.map(Object::getClass)
.map(targetType -> ReflectionUtils.findField(targetType, fieldName))
.map(field -> {
ReflectionUtils.makeAccessible(field);
return field;
})
.<T>map(field -> (T) ReflectionUtils.getField(field, target))
.orElseThrow(() -> newIllegalArgumentException("Unable to get value of field [%s] on object of type [%s]",
fieldName, target != null ? target.getClass().getName() : null));
}
@Test
public void subscriptionEnabledForClientCachePool() {
ClientCacheFactoryBean clientCacheFactoryBean = new ClientCacheFactoryBean();
assertThat(clientCacheFactoryBean).isNotNull();
assertThat(Boolean.TRUE.equals(clientCacheFactoryBean.getSubscriptionEnabled())).isFalse();
this.configuration.enableSubscriptionClientCacheConfigurer()
.configure("testClientCache", clientCacheFactoryBean);
assertThat(clientCacheFactoryBean.getSubscriptionEnabled()).isTrue();
}
@Test
public void subscriptionEnabledForDefaultPool() {
PoolFactoryBean poolFactoryBean = new PoolFactoryBean();
assertThat(poolFactoryBean).isNotNull();
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
this.configuration.enableSubscriptionPoolConfigurer().configure("DEFAULT", poolFactoryBean);
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isTrue();
}
@Test
public void subscriptionEnabledForGemFirePool() {
PoolFactoryBean poolFactoryBean = new PoolFactoryBean();
assertThat(poolFactoryBean).isNotNull();
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
this.configuration.enableSubscriptionPoolConfigurer().configure("gemfirePool", poolFactoryBean);
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isTrue();
}
@Test
public void subscriptionEnabledOnGemFirePool() {
PoolFactoryBean poolFactoryBean = new PoolFactoryBean();
assertThat(poolFactoryBean).isNotNull();
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
this.configuration.enableSubscriptionPoolConfigurer().configure("gemfirePool", poolFactoryBean);
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isTrue();
}
@Test
public void subscriptionNotEnabledForGeodePool() {
PoolFactoryBean poolFactoryBean = new PoolFactoryBean();
assertThat(poolFactoryBean).isNotNull();
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
this.configuration.enableSubscriptionPoolConfigurer().configure("geodePool", poolFactoryBean);
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
}
@Test
public void subscriptionNotEnabledForNonPreciseDefaultPoolCase() {
PoolFactoryBean poolFactoryBean = new PoolFactoryBean();
assertThat(poolFactoryBean).isNotNull();
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
this.configuration.enableSubscriptionPoolConfigurer().configure("Default", poolFactoryBean);
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
}
@Test
public void subscriptionNotEnabledForTestPool() {
PoolFactoryBean poolFactoryBean = new PoolFactoryBean();
assertThat(poolFactoryBean).isNotNull();
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
this.configuration.enableSubscriptionPoolConfigurer().configure("TEST", poolFactoryBean);
assertThat(this.<Boolean>getFieldValue(poolFactoryBean, "subscriptionEnabled")).isFalse();
}
}