MongoDB学习-->Spring Data Mongodb框架之Repository

application-dev.yml

server:
port: 8888

mongo:
host: localhost
port: 27017
timeout: 60000
db: mamabike

配置类

package com.tangzhe.configuration;

import com.mongodb.MongoClient;
import com.mongodb.MongoClientOptions;
import com.mongodb.ServerAddress;
import com.tangzhe.autoid.SaveMongoEventListener;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.data.mongodb.core.convert.DefaultMongoTypeMapper;
import org.springframework.data.mongodb.core.convert.MappingMongoConverter;
import org.springframework.data.mongodb.core.mapping.MongoMappingContext;
import org.springframework.data.mongodb.gridfs.GridFsTemplate;

/**
 * Created by 唐哲
 * 2018-03-13 17:20
 * MongoDB配置
 */
@Configuration
@ConfigurationProperties(prefix = "mongo")
@Data
public class MongoDBConfiguration {

    //mongodb服务地址
    private String host;

    //mongodb服务端口号
    private Integer port;

    //连接超时
    private Integer timeout;

    //mongodb数据库名
    private String db;

    /**
     * 配置MongoDB模板
     */
    @Bean
    public MongoTemplate mongoTemplate(SimpleMongoDbFactory mongoDbFactory,
                                       MappingMongoConverter mappingMongoConverter) {
        return new MongoTemplate(mongoDbFactory, mappingMongoConverter);
    }

    /**
     * 配置自增ID监听器
     */
    @Bean
    public SaveMongoEventListener saveMongoEventListener() {
        return new SaveMongoEventListener();
    }

    /**
     * 配置GridFs模板，实现文件上传下载
     */
    @Bean
    public GridFsTemplate gridFsTemplate(SimpleMongoDbFactory mongoDbFactory,
                                         MappingMongoConverter mappingMongoConverter) {
        return new GridFsTemplate(mongoDbFactory, mappingMongoConverter);
    }

    /**
     * 配置mongoDbFactory
     */
    @Bean
    public SimpleMongoDbFactory mongoDbFactory() {
        MongoClientOptions options = MongoClientOptions.builder().connectTimeout(timeout).build();
        MongoClient client = new MongoClient(new ServerAddress(host, port), options);
        return new SimpleMongoDbFactory(client, db);
    }

    /**
     * 配置mongoMappingContext
     */
    @Bean
    public MongoMappingContext mongoMappingContext() {
        return new MongoMappingContext();
    }

    /**
     * 配置defaultMongoTypeMapper
     */
    @Bean
    public DefaultMongoTypeMapper defaultMongoTypeMapper() {
        //去掉_class字段
        return new DefaultMongoTypeMapper(null);
    }

    /**
     * 配置mappingMongoConverter
     */
    @Bean
    public MappingMongoConverter mappingMongoConverter(SimpleMongoDbFactory mongoDbFactory,
                                                       MongoMappingContext mongoMappingContext,
                                                       DefaultMongoTypeMapper defaultMongoTypeMapper) {
        MappingMongoConverter mappingMongoConverter = new MappingMongoConverter(mongoDbFactory, mongoMappingContext);
        mappingMongoConverter.setTypeMapper(defaultMongoTypeMapper);
        return mappingMongoConverter;
    }

}

实体类

package com.tangzhe.repository;

import lombok.Data;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;
import org.springframework.data.mongodb.core.mapping.Field;

import java.util.Date;
import java.util.List;

/**
 * Created by 唐哲
 * 2018-03-13 17:40
 */
@Document(collection = "article_info")
@Data
public class Article {

    @Id
    private String id;

    @Field("title")
    private String title;

    @Field("v_count")
    private Long visitCount;

    @Field("tags")
    private List<String> tags;

    @Field("add_time")
    private Date addTime;

}

repository类

package com.tangzhe.repository;

import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.repository.CrudRepository;

import java.util.List;

/**
 * Created by 唐哲
 * 2018-03-14 22:04
 */
public interface ArticleRepository extends CrudRepository<Article, String> {

    List<Article> findByTitle(String title);

    List<Article> findByTitleAndVisitCount(String title, Long visitCount);

    /** 分页查询 */
    List<Article> findByTitle(String title, Pageable pageable);

    /** 排序 */
    List<Article> findByTitle(String title, Sort sort);

    /** 查询头一条 */
    List<Article> findTop1ByTitle(String title);

    /** 查询头二条 */
    List<Article> findTop2ByTitle(String title);

}

测试repository

package com.tangzhe.repository;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.Date;
import java.util.Iterator;
import java.util.List;

/**
 * Created by 唐哲
 * 2018-03-14 22:07
 */
@RunWith(SpringRunner.class)
@SpringBootTest
public class ArticleRepositoryTest {

    @Autowired
    private ArticleRepository articleRepository;

    //@Test
    public void repositoryTest() {
        System.out.println(articleRepository.count());

        Iterator<Article> iterator = articleRepository.findAll().iterator();
        while (iterator.hasNext()) {
            Article article = iterator.next();
            System.out.println(article);
        }
    }

    //@Test
    public void testFindByTitle() {
        List<Article> list = articleRepository.findByTitle("猿天地实战课程1");
        System.out.println(list.size());
    }
    
    //@Test
    public void testSave() {
        Article article = new Article();
        article.setTitle("猿天地实战课程1");
        article.setAddTime(new Date());
        article.setVisitCount(100L);
        article = articleRepository.save(article);
        System.out.println(article);
    }

    //@Test
    public void testFindByTitleAndVisitCount() {
        List<Article> list = articleRepository.findByTitleAndVisitCount("猿天地实战课程1", 10L);
        System.out.println(list.size());
    }

    //@Test
    public void testPageable() {
        List<Article> list = articleRepository.findByTitle("猿天地实战课程1", new PageRequest(0, 2));
        System.out.println(list.size());
    }

    //@Test
    public void testSort() {
        List<Article> list = articleRepository.findByTitle("猿天地实战课程1", new Sort(new Sort.Order(Sort.Direction.DESC, "visitCount")));
        for(Article article : list) {
            System.out.println(article.getVisitCount());
        }
    }

    //@Test
    public void testFindTop1ByTitle() {
        List<Article> list = articleRepository.findTop1ByTitle("猿天地实战课程1");
        System.out.println(list.size());
    }

    //@Test
    public void testFindTop2ByTitle() {
        List<Article> list = articleRepository.findTop2ByTitle("猿天地实战课程1");
        System.out.println(list.size());
    }

}