Skip to content

Commit

Permalink
DATACASS-7 - Polish.
Browse files Browse the repository at this point in the history
Origin pull request: #74.
  • Loading branch information
jxblum committed Jul 25, 2016
1 parent 43fc751 commit f96e938
Show file tree
Hide file tree
Showing 22 changed files with 566 additions and 561 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,6 @@ public class BasicCassandraMappingContext
* Creates a new {@link BasicCassandraMappingContext}.
*/
public BasicCassandraMappingContext() {

setCustomConversions(new CustomConversions(Collections.EMPTY_LIST));
setSimpleTypeHolder(CassandraSimpleTypeHolder.HOLDER);
}
Expand All @@ -95,7 +94,6 @@ public BasicCassandraMappingContext() {
* @since 1.5
*/
public void setCustomConversions(CustomConversions customConversions) {

Assert.notNull(customConversions, "CustomConversions must not be null");

this.customConversions = customConversions;
Expand Down Expand Up @@ -127,6 +125,7 @@ public Collection<CassandraPersistentEntity<?>> getPersistentEntities(boolean in
if (includePrimaryKeyTypes) {
return super.getPersistentEntities();
}

return Collections.unmodifiableSet(nonPrimaryKeyEntities);
}

Expand Down Expand Up @@ -296,55 +295,50 @@ public DataType getDataType(CassandraPersistentProperty property) {
public DataType getDataType(Class<?> type) {

return (customConversions.hasCustomWriteTarget(type)
? getDataTypeFor(customConversions.getCustomWriteTarget(type))
: getDataTypeFor(type));
? getDataTypeFor(customConversions.getCustomWriteTarget(type)) : getDataTypeFor(type));
}

public void setMapping(Mapping mapping) {

Assert.notNull(mapping, "Mapping must not be null");

this.mapping = mapping;
}

protected void processMappingOverrides() {

if (mapping != null) {
for (EntityMapping entityMapping : mapping.getEntityMappings()) {

if (entityMapping == null) {
continue;
}

String entityClassName = entityMapping.getEntityClassName();
if (entityMapping != null) {
String entityClassName = entityMapping.getEntityClassName();

try {
Class<?> entityClass = ClassUtils.forName(entityClassName, beanClassLoader);
try {
Class<?> entityClass = ClassUtils.forName(entityClassName, beanClassLoader);

CassandraPersistentEntity<?> entity = getPersistentEntity(entityClass);
CassandraPersistentEntity<?> entity = getPersistentEntity(entityClass);

if (entity == null) {
throw new IllegalStateException(String.format("unknown persistent entity class name [%s]", entityClassName));
}
if (entity == null) {
throw new IllegalStateException(String.format(
"Unknown persistent entity class name [%s]", entityClassName));
}

String tableName = entityMapping.getTableName();
String tableName = entityMapping.getTableName();

if (StringUtils.hasText(tableName)) {
entity.setTableName(cqlId(tableName, Boolean.valueOf(entityMapping.getForceQuote())));
}
if (StringUtils.hasText(tableName)) {
entity.setTableName(cqlId(tableName, Boolean.valueOf(entityMapping.getForceQuote())));
}

processMappingOverrides(entity, entityMapping);
processMappingOverrides(entity, entityMapping);

} catch (ClassNotFoundException e) {
throw new IllegalStateException(String.format(
"unknown persistent entity name [%s]", entityClassName), e);
} catch (ClassNotFoundException e) {
throw new IllegalStateException(String.format(
"unknown persistent entity name [%s]", entityClassName), e);
}
}
}
}
}

protected void processMappingOverrides(CassandraPersistentEntity<?> entity, EntityMapping entityMapping) {

for (PropertyMapping mapping : entityMapping.getPropertyMappings().values()) {
processMappingOverride(entity, mapping);
}
Expand All @@ -356,7 +350,7 @@ protected void processMappingOverride(CassandraPersistentEntity<?> entity, Prope

if (property == null) {
throw new IllegalArgumentException(String.format("Entity class [%s] has no persistent property named [%s]",
entity.getType().getName(), mapping.getPropertyName()));
entity.getType().getName(), mapping.getPropertyName()));
}

boolean forceQuote = false;
Expand Down
Loading

0 comments on commit f96e938

Please sign in to comment.