Spring data mongodb 替换 Repository 实现类,findAll 排除 字段

因文档比较大,有时候findAll 不想返回所有数据.没有找到默认的findAll 能够include 或者 exclude 的方法,所以想办法扩展一下实现类

query.fields().include()
query.fields().exclude()

调试结果 SimpleMongoRepository 为 MongoRepository 的默认实现类,那么继承 SimpleMongoRepository 做一点修改

/**

 * Created by laizhenwei on 2017/10/25

 */

public class MySimpleMongoRepository <T, ID extends Serializable> extends SimpleMongoRepository<T, ID> {

    private final MongoOperations mongoOperations;

    private final MongoEntityInformation<T, ID> entityInformation;

    public MySimpleMongoRepository(MongoEntityInformation<T, ID> metadata, MongoOperations mongoOperations) {

        super(metadata, mongoOperations);

        this.entityInformation = metadata;

        this.mongoOperations = mongoOperations;

    }

    public <S extends T> Page<S> findAll(final Example<S> example,final Query query, Pageable pageable) {

        Assert.notNull(example, "Sample must not be null!");

        query.addCriteria((new Criteria()).alike(example)).with(pageable);

        List<S> list = this.mongoOperations.find(query, example.getProbeType(), this.entityInformation.getCollectionName());

        return PageableExecutionUtils.getPage(list, pageable, () -> mongoOperations.count(query, example.getProbeType(),entityInformation.getCollectionName()));

    }

}

OrderInfoRepository

/**

 * Created by laizhenwei on 2017/10/25

 */

public interface OrderInfoRepository extends MongoRepository<OrderInfo,ObjectId> {

    Page<OrderInfo> findAll(final Example<OrderInfo> example, final Query query, Pageable pageable);

}

MongoConfig

/**

 * Created by laizhenwei on 2017/10/25

 */

@Configuration

@EnableMongoRepositories(basePackages = "com.example.mongo.Repository",repositoryBaseClass = MySimpleMongoRepository.class)

public class MongoConfig {

}

JunitTest

 @Test

    public void findPage(){

        ExampleMatcher matcher = ExampleMatcher.matching()

                .withMatcher("buyer", ExampleMatcher.GenericPropertyMatchers.exact())

                .withMatcher("phone",ExampleMatcher.GenericPropertyMatchers.exact())

        OrderInfo orderInfo = new OrderInfo();

        orderInfo.setBuyer("A君");

        orderInfo.setPhone("13422981024");

        Example<OrderInfo> example = Example.of(orderInfo,matcher);

        Query query = new Query();

        query.fields().include("orderInfo");

        Page<OrderInfo> page = orderInfoRepository.findAll(example,query,new PageRequest(0,10));

        List<OrderInfo> orderInfos = page.getContent();

        Assert.assertTrue(orderInfos.stream().filter(val->val.getPhone().equals("13422981024")).count()==1);

        System.out.println(orderInfos.size());

        System.out.println(orderInfos.get(0).getOrderDetails());

    }