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

DATAREDIS-605 - Polishing

Browse Source 
Some minor code and documentation changes. More interestingly obtain the RedisConverter instead of the KeyValueAdapter for IndexResolver resolution in  QueryByExampleRedisExecutor and stick to findOne contract by throwing IncorrectResultSizeDataAccessException for queries returning non unique results.

Original Pull Request: #301
This commit is contained in:
Christoph Strobl
2018-01-10 13:32:54 +01:00
parent a3f876f926
commit 323cb32861
5 changed files with 51 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/java/org/springframework/data/redis/core/RedisKeyValueTemplate.java
View File

@@ -22,6 +22,7 @@ import java.util.List;
import org.springframework.data.keyvalue.core.KeyValueAdapter;
import org.springframework.data.keyvalue.core.KeyValueCallback;
import org.springframework.data.keyvalue.core.KeyValueTemplate;
import org.springframework.data.redis.core.convert.RedisConverter;
import org.springframework.data.redis.core.mapping.RedisMappingContext;
import org.springframework.util.Assert;
import org.springframework.util.ClassUtils;
@@ -49,11 +50,13 @@ public class RedisKeyValueTemplate extends KeyValueTemplate {
}
/**
* @return the {@link RedisKeyValueAdapter}.
* Obtain the underlying redis specific {@link org.springframework.data.convert.EntityConverter}.
*
* @return never {@literal null}.
* @since 2.1
*/
public RedisKeyValueAdapter getAdapter() {
return adapter;
public RedisConverter getConverter() {
return adapter.getConverter();
}
/*
@@ -65,7 +68,6 @@ public class RedisKeyValueTemplate extends KeyValueTemplate {
return (RedisMappingContext) super.getMappingContext();
}
/**
* Retrieve entities by resolving their {@literal id}s and converting them into required type. <br />
* The callback provides either a single {@literal id} or an {@link Iterable} of {@literal id}s, used for retrieving
@@ -104,15 +106,13 @@ public class RedisKeyValueTemplate extends KeyValueTemplate {
}
Iterable<?> ids = ClassUtils.isAssignable(Iterable.class, callbackResult.getClass())
? (Iterable<?>) callbackResult
: Collections.singleton(callbackResult);
? (Iterable<?>) callbackResult : Collections.singleton(callbackResult);
List<T> result = new ArrayList<>();
for (Object id : ids) {
String idToUse = adapter.getConverter().getConversionService().canConvert(id.getClass(), String.class)
? adapter.getConverter().getConversionService().convert(id, String.class)
: id.toString();
? adapter.getConverter().getConversionService().convert(id, String.class) : id.toString();
findById(idToUse, type).ifPresent(result::add);
}

44
src/main/java/org/springframework/data/redis/repository/support/QueryByExampleRedisExecutor.java
View File

@@ -20,6 +20,7 @@ import java.util.Iterator;
import java.util.List;
import java.util.Optional;
import org.springframework.dao.IncorrectResultSizeDataAccessException;
import org.springframework.data.domain.Example;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
@@ -29,7 +30,6 @@ import org.springframework.data.keyvalue.core.query.KeyValueQuery;
import org.springframework.data.redis.core.RedisKeyValueTemplate;
import org.springframework.data.redis.core.convert.IndexResolver;
import org.springframework.data.redis.core.convert.PathIndexResolver;
import org.springframework.data.redis.core.convert.RedisConverter;
import org.springframework.data.redis.repository.query.ExampleQueryMapper;
import org.springframework.data.redis.repository.query.RedisOperationChain;
import org.springframework.data.repository.core.EntityInformation;
@@ -43,6 +43,7 @@ import org.springframework.util.Assert;
* methods.
*
* @author Mark Paluch
* @author Christoph Strobl
* @since 2.1
*/
@SuppressWarnings("unchecked")
@@ -62,17 +63,9 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
public QueryByExampleRedisExecutor(EntityInformation<T, ?> entityInformation,
RedisKeyValueTemplate keyValueTemplate) {
Assert.notNull(entityInformation, "EntityInformation must not be null!");
Assert.notNull(keyValueTemplate, "RedisKeyValueTemplate must not be null!");
this.entityInformation = entityInformation;
this.keyValueTemplate = keyValueTemplate;
RedisConverter converter = keyValueTemplate.getAdapter().getConverter();
IndexResolver indexResolver = converter.getIndexResolver();
this.mapper = new ExampleQueryMapper(keyValueTemplate.getMappingContext(),
indexResolver != null ? indexResolver : new PathIndexResolver(converter.getMappingContext()));
this(entityInformation, keyValueTemplate,
keyValueTemplate.getConverter().getIndexResolver() != null ? keyValueTemplate.getConverter().getIndexResolver()
: new PathIndexResolver(keyValueTemplate.getMappingContext()));
}
/**
@@ -91,8 +84,7 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
this.entityInformation = entityInformation;
this.keyValueTemplate = keyValueTemplate;
this.mapper = new ExampleQueryMapper(keyValueTemplate.getMappingContext(),
new PathIndexResolver(keyValueTemplate.getMappingContext()));
this.mapper = new ExampleQueryMapper(keyValueTemplate.getMappingContext(), indexResolver);
}
/*
@@ -102,13 +94,21 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
@Override
public <S extends T> Optional<S> findOne(Example<S> example) {
RedisOperationChain operationChain = getQuery(example);
RedisOperationChain operationChain = createQuery(example);
KeyValueQuery<RedisOperationChain> query = new KeyValueQuery<>(operationChain);
Iterable<T> result = keyValueTemplate.find(query.limit(1), entityInformation.getJavaType());
Iterator<T> iterator = result.iterator();
Iterator<T> iterator = keyValueTemplate.find(query.limit(2), entityInformation.getJavaType()).iterator();
return iterator.hasNext() ? Optional.of((S) iterator.next()) : Optional.empty();
Optional result = Optional.empty();
if (iterator.hasNext()) {
result = Optional.of((S) iterator.next());
if (iterator.hasNext()) {
throw new IncorrectResultSizeDataAccessException(1);
}
}
return result;
}
/*
@@ -118,7 +118,7 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
@Override
public <S extends T> Iterable<S> findAll(Example<S> example) {
RedisOperationChain operationChain = getQuery(example);
RedisOperationChain operationChain = createQuery(example);
return (Iterable<S>) keyValueTemplate.find(new KeyValueQuery<>(operationChain), entityInformation.getJavaType());
}
@@ -141,7 +141,7 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
Assert.notNull(pageable, "Pageable must not be null!");
RedisOperationChain operationChain = getQuery(example);
RedisOperationChain operationChain = createQuery(example);
KeyValueQuery<RedisOperationChain> query = new KeyValueQuery<>(operationChain);
Iterable<T> result = keyValueTemplate.find(
@@ -166,7 +166,7 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
@Override
public <S extends T> long count(Example<S> example) {
RedisOperationChain operationChain = getQuery(example);
RedisOperationChain operationChain = createQuery(example);
return keyValueTemplate.count(new KeyValueQuery<>(operationChain), entityInformation.getJavaType());
}
@@ -180,7 +180,7 @@ public class QueryByExampleRedisExecutor<T> implements QueryByExampleExecutor<T>
return count(example) > 0;
}
private <S extends T> RedisOperationChain getQuery(Example<S> example) {
private <S extends T> RedisOperationChain createQuery(Example<S> example) {
Assert.notNull(example, "Example must not be null!");