com.github.developframework.resource.spring.mongo.MongoResourceHandler Maven / Gradle / Ivy

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

import com.github.developframework.resource.ResourceDefinition;
import com.github.developframework.resource.Search;
import com.github.developframework.resource.spring.SpringDataResourceHandler;
import com.github.developframework.resource.spring.mongo.utils.AggregationOperationBuilder;
import develop.toolkit.base.utils.K;
import lombok.Getter;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.repository.MongoRepository;

import java.io.Serializable;
import java.util.List;
import java.util.Optional;

/**
 * MongoDB资源操作器
 *
 * @author qiushui on 2019-08-21.
 */
public class MongoResourceHandler,
        ID extends Serializable,
        REPOSITORY extends MongoRepository
        > extends SpringDataResourceHandler {

    @Getter
    private final MongoOperations mongoOperations;

    public MongoResourceHandler(REPOSITORY repository, ResourceDefinition resourceDefinition, MongoOperations mongoOperations) {
        super(repository, resourceDefinition);
        this.mongoOperations = mongoOperations;
    }

    @Override
    public Optional queryByIdForUpdate(ID id) {
        return queryById(id);
    }

    @Override
    public List query(Search search) {
        Query query = safeSearch(search);
        return query != null ? mongoOperations.find(query, resourceDefinition.getEntityClass()) : mongoOperations.findAll(resourceDefinition.getEntityClass());
    }

    @Override
    public > List query(Sort sort, SEARCH search) {
        Query query = safeSearch(search);
        AggregationOperationBuilder builder = new AggregationOperationBuilder(mongoOperations);
        K.let(query, builder::match);
        return builder.aggregation(sort != null, Aggregation.sort(sort)).list(resourceDefinition.getEntityClass(), resourceDefinition.getEntityClass());
    }

    @Override
    public > Page queryPager(Pageable pageable, SEARCH search) {
        Query query = safeSearch(search);
        AggregationOperationBuilder builder = new AggregationOperationBuilder(mongoOperations);
        K.let(query, builder::match);
        return builder.pager(pageable, resourceDefinition.getEntityClass(), resourceDefinition.getEntityClass());
    }

    private Query safeSearch(Search search) {
        return K.map(search, s -> ((MongoSearch) s).toQuery());
    }
}