使用 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<ContextRefreshedEvent> {
@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 学习笔记:以JSON格式返回对象》
《Spring Boot 学习笔记:请求数据的解析绑定与返回数据的处理》