无需HibernateOGM：Spring Boot启动时自动重置MongoDB的方法

对于SQL数据库我们可以用spring.jpa.hibernate.ddl-auto: create来自动重置库表，MongoDB场景下，不用依赖已废弃的HibernateOGM，用Spring Data MongoDB自带的组件就能实现，下面是几种可行方案：

方案1：通过CommandLineRunner/ApplicationRunner在启动时清空数据库 #

创建一个Spring Bean，实现CommandLineRunner或ApplicationRunner接口，利用MongoTemplate提供的方法在应用启动时执行数据库重置操作：

import org.springframework.boot.CommandLineRunner;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;

@Component
// 可选：仅在开发环境生效，避免生产环境误操作
// @Profile("dev")
public class MongoDbResetRunner implements CommandLineRunner {

    private final MongoTemplate mongoTemplate;

    public MongoDbResetRunner(MongoTemplate mongoTemplate) {
        this.mongoTemplate = mongoTemplate;
    }

    @Override
    public void run(String... args) throws Exception {
        // 方案A：直接删除整个数据库，会自动重建空库
        mongoTemplate.getDb().drop();
        
        // 方案B：如果只想清空所有集合（保留数据库本身），可以遍历集合删除
        // mongoTemplate.getCollectionNames().forEach(collectionName -> {
        //     mongoTemplate.dropCollection(collectionName);
        // });
    }
}

方案2：结合配置开关控制重置逻辑 #

如果需要通过配置文件决定是否执行重置，可以新增配置项，在Runner中判断：

1. 在application.properties中添加开关：

# 仅在开发环境开启
spring.data.mongodb.reset-on-startup=true

2. 修改Runner类读取配置：

import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;

@Component
@ConfigurationProperties(prefix = "spring.data.mongodb")
public class MongoDbResetRunner implements CommandLineRunner {

    private final MongoTemplate mongoTemplate;
    private boolean resetOnStartup;

    public MongoDbResetRunner(MongoTemplate mongoTemplate) {
        this.mongoTemplate = mongoTemplate;
    }

    @Override
    public void run(String... args) throws Exception {
        if (resetOnStartup) {
            mongoTemplate.getDb().drop();
        }
    }

    public void setResetOnStartup(boolean resetOnStartup) {
        this.resetOnStartup = resetOnStartup;
    }
}

注意事项 #

• 环境隔离：务必通过@Profile或配置开关限制仅在开发/测试环境启用，绝对不要在生产环境执行数据库删除操作。
• 权限检查：确保MongoDB用户拥有dropDatabase或dropCollection的权限，否则会抛出权限异常。
• 数据初始化：如果重置后需要初始化基础数据，可以在run方法中继续添加数据插入逻辑，比如调用mongoTemplate.insert()。

内容的提问来源于stack exchange，提问作者gs97ahn