Browse Source
We also pick up the ID attribute to be used to determine isNew(…) if the RepositoryFactoryBean gets a MappingContext injected. Collection name of mapping defaults to the simple class name.pull/1/head
13 changed files with 322 additions and 103 deletions
@ -0,0 +1,80 @@ |
|||||||
|
/* |
||||||
|
* Copyright (c) 2011 by the original author(s). |
||||||
|
* |
||||||
|
* 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 |
||||||
|
* |
||||||
|
* http://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.document.mongodb.repository; |
||||||
|
|
||||||
|
import java.io.Serializable; |
||||||
|
import org.springframework.data.document.mongodb.mapping.MongoPersistentEntity; |
||||||
|
import org.springframework.data.mapping.MappingBeanHelper; |
||||||
|
import org.springframework.data.mapping.model.PersistentProperty; |
||||||
|
import org.springframework.data.repository.support.AbstractEntityInformation; |
||||||
|
|
||||||
|
/** |
||||||
|
* {@link MongoEntityInformation} implementation using a {@link MongoPersistentEntity} instance to lookup the necessary |
||||||
|
* information. |
||||||
|
* |
||||||
|
* @author Oliver Gierke |
||||||
|
*/ |
||||||
|
public class MappingMongoEntityInformation<T, ID extends Serializable> extends AbstractEntityInformation<T, ID> |
||||||
|
implements MongoEntityInformation<T, ID> { |
||||||
|
|
||||||
|
private final MongoPersistentEntity<T> entityMetadata; |
||||||
|
|
||||||
|
/** |
||||||
|
* Creates a new {@link MappingMongoEntityInformation} for the given {@link MongoPersistentEntity}. |
||||||
|
* |
||||||
|
* @param domainClass |
||||||
|
* @param entity |
||||||
|
*/ |
||||||
|
public MappingMongoEntityInformation(MongoPersistentEntity<T> entity) { |
||||||
|
super(entity.getType()); |
||||||
|
this.entityMetadata = entity; |
||||||
|
} |
||||||
|
|
||||||
|
/* (non-Javadoc) |
||||||
|
* @see org.springframework.data.repository.support.EntityInformation#getId(java.lang.Object) |
||||||
|
*/ |
||||||
|
public ID getId(T entity) { |
||||||
|
|
||||||
|
PersistentProperty idProperty = entityMetadata.getIdProperty(); |
||||||
|
|
||||||
|
try { |
||||||
|
return (ID) MappingBeanHelper.getProperty(entity, idProperty, idProperty.getType(), false); |
||||||
|
} catch (Exception e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/* (non-Javadoc) |
||||||
|
* @see org.springframework.data.repository.support.EntityInformation#getIdType() |
||||||
|
*/ |
||||||
|
public Class<ID> getIdType() { |
||||||
|
return (Class<ID>) entityMetadata.getIdProperty().getType(); |
||||||
|
} |
||||||
|
|
||||||
|
/* (non-Javadoc) |
||||||
|
* @see org.springframework.data.document.mongodb.repository.MongoEntityInformation#getCollectionName() |
||||||
|
*/ |
||||||
|
public String getCollectionName() { |
||||||
|
return entityMetadata.getCollection(); |
||||||
|
} |
||||||
|
|
||||||
|
/* (non-Javadoc) |
||||||
|
* @see org.springframework.data.document.mongodb.repository.MongoEntityInformation#getIdAttribute() |
||||||
|
*/ |
||||||
|
public String getIdAttribute() { |
||||||
|
return "_id"; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,111 @@ |
|||||||
|
/* |
||||||
|
* Copyright (c) 2011 by the original author(s). |
||||||
|
* |
||||||
|
* 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 |
||||||
|
* |
||||||
|
* http://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.document.mongodb.repository; |
||||||
|
|
||||||
|
import java.io.Serializable; |
||||||
|
import java.lang.reflect.Field; |
||||||
|
import java.util.Arrays; |
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
import org.springframework.data.repository.support.AbstractEntityInformation; |
||||||
|
import org.springframework.util.ReflectionUtils; |
||||||
|
import org.springframework.util.StringUtils; |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* Expects the domain class to contain a field with a name out of the following |
||||||
|
* {@value #FIELD_NAMES}. |
||||||
|
* |
||||||
|
* @author Oliver Gierke |
||||||
|
*/ |
||||||
|
class SimpleMongoEntityInformation<T extends Object, ID extends Serializable> extends AbstractEntityInformation<T, ID> implements MongoEntityInformation<T, ID> { |
||||||
|
|
||||||
|
private static final List<String> FIELD_NAMES = Arrays.asList("ID", "id", "_id"); |
||||||
|
private Field field; |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* Creates a new {@link SimpleMongoEntityInformation}. |
||||||
|
* |
||||||
|
* @param domainClass |
||||||
|
*/ |
||||||
|
public SimpleMongoEntityInformation(Class<T> domainClass) { |
||||||
|
|
||||||
|
super(domainClass); |
||||||
|
|
||||||
|
for (String name : FIELD_NAMES) { |
||||||
|
|
||||||
|
Field candidate = ReflectionUtils.findField(domainClass, name); |
||||||
|
|
||||||
|
if (candidate != null) { |
||||||
|
ReflectionUtils.makeAccessible(candidate); |
||||||
|
this.field = candidate; |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if (this.field == null) { |
||||||
|
throw new IllegalArgumentException(String.format( |
||||||
|
"Given domain class %s does not contain an id property!", |
||||||
|
domainClass.getName())); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* (non-Javadoc) |
||||||
|
* @see org.springframework.data.document.mongodb.repository.MongoEntityInformation#getCollectionName() |
||||||
|
*/ |
||||||
|
public String getCollectionName() { |
||||||
|
|
||||||
|
return StringUtils.uncapitalize(getJavaType().getSimpleName()); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* (non-Javadoc) |
||||||
|
* @see org.springframework.data.document.mongodb.repository.MongoEntityInformation#getIdAttribute() |
||||||
|
*/ |
||||||
|
public String getIdAttribute() { |
||||||
|
|
||||||
|
return "_id"; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* |
||||||
|
* (non-Javadoc) |
||||||
|
* |
||||||
|
* @see |
||||||
|
* org.springframework.data.repository.support.IdAware#getId(java.lang.Object |
||||||
|
* ) |
||||||
|
*/ |
||||||
|
@SuppressWarnings("unchecked") |
||||||
|
public ID getId(Object entity) { |
||||||
|
|
||||||
|
return (ID) ReflectionUtils.getField(field, entity); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* |
||||||
|
* (non-Javadoc) |
||||||
|
* |
||||||
|
* @see |
||||||
|
* org.springframework.data.repository.support.EntityInformation#getIdType() |
||||||
|
*/ |
||||||
|
@SuppressWarnings("unchecked") |
||||||
|
public Class<ID> getIdType() { |
||||||
|
|
||||||
|
return (Class<ID>) field.getType(); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue