io.beanmapper.spring.web.SpringDataEntityFinder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of beanmapper-spring Show documentation

Spring support for the Bean Mapper

There is a newer version: 6.0.0

package io.beanmapper.spring.web;

import javax.persistence.EntityNotFoundException;

import org.springframework.context.ApplicationContext;
import org.springframework.data.repository.CrudRepository;
import org.springframework.data.repository.support.Repositories;

/**
 * The default implementation is based on the Repositories class of Spring Data. It
 * first looks up the repository. If found, it then calls findOne on the repository.
 * @author Robert Bor
 */
public class SpringDataEntityFinder implements EntityFinder {

    private final Repositories repositories;

    public SpringDataEntityFinder(ApplicationContext applicationContext) {
        this.repositories = new Repositories(applicationContext);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    @SuppressWarnings("unchecked")
    public Object find(Long id, Class entityClass) throws EntityNotFoundException {
        CrudRepository repository = (CrudRepository) repositories.getRepositoryFor(entityClass);
        if (repository == null) {
            throw new EntityNotFoundException(
                    "No repository found for " + entityClass.getName());
        }
        Object entity = repository.findOne(id);
        if (entity == null) {
            throw new EntityNotFoundException(
                    "Entity with ID " + id + "was not found in repository " + repository.getClass().getName());
        }
        return entity;
    }

}