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

Refactor userTypeResolver construction into separate method.

Browse Source 
This allows for easy overriding, for example in case you
wish to use a different implementation.

See #1297
This commit is contained in:
Ammar Khaku
2022-08-20 14:51:22 -07:00
committed by Mark Paluch
parent c33addd934
commit fffd27b9d1
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
spring-data-cassandra/src/main/java/org/springframework/data/cassandra/config/AbstractCassandraConfiguration.java
View File

@@ -49,6 +49,7 @@ import com.datastax.oss.driver.api.core.CqlSession;
* @author Matthew T. Adams
* @author John Blum
* @author Mark Paluch
* @author Ammar Khaku
*/
@Configuration
@SuppressWarnings("unused")
@@ -71,14 +72,11 @@ public abstract class AbstractCassandraConfiguration extends AbstractSessionConf
CqlSession cqlSession = getRequiredSession();
UserTypeResolver userTypeResolver =
new SimpleUserTypeResolver(cqlSession, CqlIdentifier.fromCql(getKeyspaceName()));
MappingCassandraConverter converter =
new MappingCassandraConverter(requireBeanOfType(CassandraMappingContext.class));
converter.setCodecRegistry(cqlSession.getContext().getCodecRegistry());
converter.setUserTypeResolver(userTypeResolver);
converter.setUserTypeResolver(userTypeResolver(cqlSession));
converter.setCustomConversions(requireBeanOfType(CassandraCustomConversions.class));
return converter;
@@ -96,11 +94,8 @@ public abstract class AbstractCassandraConfiguration extends AbstractSessionConf
CqlSession cqlSession = getRequiredSession();
UserTypeResolver userTypeResolver =
new SimpleUserTypeResolver(cqlSession, CqlIdentifier.fromCql(getKeyspaceName()));
CassandraMappingContext mappingContext =
new CassandraMappingContext(userTypeResolver, SimpleTupleTypeFactory.DEFAULT);
new CassandraMappingContext(userTypeResolver(cqlSession), SimpleTupleTypeFactory.DEFAULT);
CustomConversions customConversions = requireBeanOfType(CassandraCustomConversions.class);
@@ -260,4 +255,8 @@ public abstract class AbstractCassandraConfiguration extends AbstractSessionConf
protected ByteArrayResource scriptOf(String content) {
return new ByteArrayResource(content.getBytes());
}
protected UserTypeResolver userTypeResolver(CqlSession cqlSession) {
return new SimpleUserTypeResolver(cqlSession, CqlIdentifier.fromCql(getKeyspaceName()));
}
}