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());
     }

 }