com.github.developframework.resource.spring.jpa.JpaResourceHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of resource-manager-jpa Show documentation
The newest version!
package com.github.developframework.resource.spring.jpa;

import com.github.developframework.resource.ResourceDefinition;
import com.github.developframework.resource.Search;
import com.github.developframework.resource.spring.SpringDataResourceHandler;
import develop.toolkit.base.utils.K;
import lombok.Getter;
import org.apache.commons.collections4.IterableUtils;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
import org.springframework.data.repository.PagingAndSortingRepository;

import javax.persistence.EntityManager;
import javax.persistence.LockModeType;
import java.io.Serializable;
import java.util.List;
import java.util.Optional;

/**
 * 基于spring-data-jpa的资源操作器
 *
 * @author qiushui on 2019-08-15.
 */
public class JpaResourceHandler<
        PO extends com.github.developframework.resource.spring.jpa.PO,
        ID extends Serializable,
        REPOSITORY extends PagingAndSortingRepository & JpaSpecificationExecutor
        > extends SpringDataResourceHandler {

    @Getter
    private final EntityManager entityManager;

    public JpaResourceHandler(REPOSITORY repository, ResourceDefinition resourceDefinition, EntityManager entityManager) {
        super(repository, resourceDefinition);
        this.entityManager = entityManager;
    }

    @Override
    public Optional queryByIdForUpdate(ID id) {
        return Optional.ofNullable(entityManager.find(resourceDefinition.getEntityClass(), id, LockModeType.PESSIMISTIC_WRITE));
    }

    @Override
    public List query(Search search) {
        Specification specification = safeSearch(search);
        return specification != null ? repository.findAll(specification) : IterableUtils.toList(repository.findAll());
    }

    @Override
    public > List query(Sort sort, SEARCH search) {
        Specification specification = safeSearch(search);
        return specification != null ? repository.findAll(specification, sort) : IterableUtils.toList(repository.findAll(sort));
    }

    @Override
    public > Page queryPager(Pageable pageable, SEARCH search) {
        Specification specification = safeSearch(search);
        return specification != null ? repository.findAll(specification, pageable) : repository.findAll(pageable);
    }

    private Specification safeSearch(Search search) {
        return K.map(search, s -> ((JpaSearch) s).toSpecification());
    }
}