Spring Boot 学习笔记：MongoTemplate 移除 _class 字段

 

使用 Spring Boot 中 MongoTemplate 插入数据的时候，会将对象类型以 _class 字段一起存入MongoDB数据库。

若不想存入 _class 数据，就需要从MongoTemplate入手，移除相应逻辑。

通过Google和Baidu搜索到的方法，都是通过自定义MongoDbFactory和MongoTemplate的方式，在生成相应Bean的时候，最终以null作为参数，构造一个DefaultMongoTypeMapper对象，并赋值给MongoConverter的TypeMapper字段。

由于不想对原有的Spring Boot生成MongoTemplate Bean的逻辑做太多干扰，原始生成逻辑的每一段代码可能都有其背后的原因。所以，希望以最小的改动，来达到移除 "_class" 字段的目的。经过一番源码研究，得到一种方法，自测有效，若各位大神发现有任何问题，还望指正。

 

首先，通过Spring Boot启动过程了解到，可以通过ApplicationListener监听“完成容器刷新事件”；

接着，通过MongoTemplate源码了解到，可以通过MappingMongoConverter.setTypeMapper设置新的TypeMapper对象；

结合上述两个情况，可以通过监听事件，在事件发生之后，重新设置TypeMapper对象，即可达到需求。

 

代码如下：

@Configuration
public class ApplicationReadyListener implements ApplicationListener {

  @Autowired
  MongoTemplate mongoTemplate;

  @Override
  public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {
    MongoConverter converter = mongoTemplate.getConverter();
    if (converter.getTypeMapper().isTypeKey("_class")) {
      ((MappingMongoConverter) converter).setTypeMapper(new DefaultMongoTypeMapper(null));
    }
  }
}

 

相关文章

《Spring Boot 学习笔记：常见注解》

《Spring Boot 学习笔记：以JSON格式返回对象》

《Spring Boot 学习笔记：Schedule》

《Spring Boot 学习笔记：启动过程方法》

《Spring Boot 学习笔记：请求数据的解析绑定与返回数据的处理》