Spring boot初始化Mongo数据库(将.json文件持久化到Mongo数据库)

秒速五厘米 2022-09-25 11:19 378阅读 0赞

这篇文章教你如何在项目启动的时候,将你配置在项目下的json内容存储在MongoDB数据库中。  
众所周知，Spring Boo项目t是通过一个main函数来启动的。所以我们想能不能在其启动时某些操作(也就是监听起启动事件)。通过查看Spring boot 文档([http://docs.spring.io/spring-boot/docs/current-SNAPSHOT/reference/htmlsingle/][http_docs.spring.io_spring-boot_docs_current-SNAPSHOT_reference_htmlsingle])我们知道Spring Boot提供了一个接口用来监听Spring Boot的启动:

@Component
    class SystemInitializer implements ApplicationListener<ContextRefreshedEvent> {
        
    
        @Override
        public void onApplicationEvent(ContextRefreshedEvent arg0) {
            // TODO Auto-generated method stub
            System.out.println("Spring Boot启动！");
        }
    
    }

而对于我们持久化json数据到Mongo数据库,一开始我的想法是自己来实现该接口来操作，后面发现Spring 专门提供了这样一个类:

Jackson2RepositoryPopulatorFactoryBean

**使用方法:**  
首先我们看.json文件:

[
    {
      "_class":"com.wang.domain.Student",
      "name":"wang",
      "code":"java",
      "email":"alibaba@gmail.com"
     },
     {
      "_class":"com.wang.domain.Student",
      "name":"xi",
      "code":"Android",
      "email":"123456@qq.com"
     }
    ]

除了\_class自动字段比较特殊外。其余的大家都理解吧，这个\_class字段必不可少制定了该数据属于哪个类。

package com.wang.domain;
    
    import org.springframework.data.annotation.Id;
    
    public class Student {
        
        @Id
        private String _id;
        private String name;
        private String email;
        private String code;
    
        //getter setter省
    }

这下大家一目了然了吧。现在还有一个也是必不可少的操作就是给Student这个类定义一个Repository(即我们常说的DAO),如果不配置则会抛出No Repository异常。

public interface StudentRepository extends MongoRepository<Student, String>{
        
    
    }

现在就剩下最后一步了,为Jackson2RepositoryPopulatorFactoryBean配置这个.json文件。

//初始化数据
        @Bean
        public Jackson2RepositoryPopulatorFactoryBean repositoryPopulator(){
            Jackson2RepositoryPopulatorFactoryBean factory = new Jackson2RepositoryPopulatorFactoryBean();
            //判断是否需要重新生成数据库内容
            if(db_rebuildMongoData){
                //加载.json文件
                Resource initData = new ClassPathResource("init/initTestData.json");
                factory.setResources(new Resource[]{initData});
            }else{
                factory.setResources(new Resource[]{});
            }
            return factory;
        }

大家可能有疑问，这个db\_rebuildMongoData变量是什么。因为这个类是在Spring Boot启动的时候来执行的而且是用来初始化数据库的，没有特殊需求生产环境中不可能每次重新发布项目的时候都初始化一下数据库吧。所以我们在Spring Boot的配置文件application.properties中加入这么一行:

db_rebuild=true

在main类中:

@Value("${db_rebuild}")
    private Boolean db_rebuildMongoData;

现在不难看出上述代码的意义了，这样是否初始化数据库就在自己的掌握之中了。

如果对Spring Boot不太熟悉的话可以参考我的关于Spring Boot的其他博客：  
[http://blog.csdn.net/canot/article/details/51449589][http_blog.csdn.net_canot_article_details_51449589]  
[http://blog.csdn.net/canot/article/details/51571343][http_blog.csdn.net_canot_article_details_51571343]

[http_docs.spring.io_spring-boot_docs_current-SNAPSHOT_reference_htmlsingle]: http://docs.spring.io/spring-boot/docs/current-SNAPSHOT/reference/htmlsingle/
[http_blog.csdn.net_canot_article_details_51449589]: http://blog.csdn.net/canot/article/details/51449589
[http_blog.csdn.net_canot_article_details_51571343]: http://blog.csdn.net/canot/article/details/51571343