如何根据Spring激活的profile控制CommandLineRunner的run方法执行

方案1：使用@Profile注解控制Bean注册（最推荐） #

直接给你的CommandLineRunner实现类加上@Profile("development")注解即可，Spring 只会在激活development profile 时才会将这个类实例化为Bean注册到容器中，生产环境下该Bean不会存在，自然不会执行run方法。
示例代码：

import org.springframework.context.annotation.Profile;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

@Component
@Profile("development") // 仅激活development profile时才注册该Bean
public class H2DataInitializer implements CommandLineRunner {

    @Override
    public void run(String... args) throws Exception {
        // 你的H2测试数据预置逻辑
    }
}

如果你的CommandLineRunner是在配置类中通过@Bean声明的，也可以把@Profile加在@Bean方法上：

@Configuration
public class AppConfig {

    @Bean
    @Profile("development")
    public CommandLineRunner h2DataInitializer() {
        return args -> {
            // 测试数据预置逻辑
        };
    }
}

方案2：编程式判断激活的Profile（适配多条件场景） #

如果你有特殊需求需要这个Bean在所有环境都注册，仅区分run方法内部的执行逻辑，可以注入Environment对象判断当前激活的profile：

import org.springframework.core.env.Environment;
import org.springframework.core.env.Profiles;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

@Component
public class DataInitializer implements CommandLineRunner {

    private final Environment env;

    // 构造注入Environment
    public DataInitializer(Environment env) {
        this.env = env;
    }

    @Override
    public void run(String... args) throws Exception {
        // 判断是否激活了development profile
        if (env.acceptsProfiles(Profiles.of("development"))) {
            // 仅开发环境执行的测试数据预置逻辑
        }
        // 全环境都要执行的逻辑可以放在判断外
    }
}

补充说明：如果存在同时激活多个profile的场景，@Profile也支持多值配置，比如@Profile({"development", "test"})就可以同时适配开发和测试环境。

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