Spring Boot：根据触发源切换GraphService实现类

你的场景需要运行时动态选择不同的GraphService实现，而条件Bean是启动阶段静态创建Bean的方式，所以无法满足需求。下面提供两种直接可行的实现方案：

方法一：通过方法参数传递调用标识 #

这种方式最简单直接，通过给refreshUsers()添加参数标识调用来源，在方法内部选择对应的服务实现。

步骤1：给两个GraphService实现添加@Qualifier区分 #

@Service
@Qualifier("userGraphService")
public class GraphUserService implements GraphService {
    // 实现getGraphClient()逻辑
}

@Service
@Qualifier("appGraphService")
public class GraphApplicationService implements GraphService {
    // 实现getGraphClient()逻辑
}

步骤2：AdminService注入两个实现并根据参数选择 #

@Service
public class AdminService {
    @Autowired
    @Qualifier("userGraphService")
    private GraphService userGraphService;

    @Autowired
    @Qualifier("appGraphService")
    private GraphService appGraphService;

    public void refreshUsers(boolean isUserInitiated) {
        // 根据调用来源选择对应的服务
        GraphService targetService = isUserInitiated ? userGraphService : appGraphService;
        List<User> users = targetService.getAllUsers();
        // 执行后续的用户刷新逻辑
    }
}

步骤3：不同触发方调用时传递对应参数 #

• 用户触发（比如Controller层）：

@RestController
@RequestMapping("/admin")
public class AdminController {
    @Autowired
    private AdminService adminService;

    @PostMapping("/refresh-users")
    public void triggerRefresh() {
        // 用户主动触发，传true
        adminService.refreshUsers(true);
    }
}

• 定时任务触发：

@Service
public class ScheduleService {
    @Autowired
    private AdminService adminService;

    @Scheduled(fixedRate = 3600000) // 1小时执行一次
    public void scheduledRefresh() {
        // 定时任务触发，传false
        adminService.refreshUsers(false);
    }
}

方法二：用ThreadLocal存储调用上下文（无侵入修改方法参数） #

如果不想修改refreshUsers()的方法签名，可以用ThreadLocal在调用链中传递调用来源标识。

步骤1：定义上下文持有类 #

public class CallContextHolder {
    private static final ThreadLocal<String> CALL_SOURCE = new ThreadLocal<>();

    public static void setCallSource(String source) {
        CALL_SOURCE.set(source);
    }

    public static String getCallSource() {
        return CALL_SOURCE.get();
    }

    // 必须清理，避免内存泄漏
    public static void clear() {
        CALL_SOURCE.remove();
    }
}

步骤2：在调用入口设置上下文 #

• 用户触发的Controller：

@RestController
@RequestMapping("/admin")
public class AdminController {
    @Autowired
    private AdminService adminService;

    @PostMapping("/refresh-users")
    public void triggerRefresh() {
        try {
            CallContextHolder.setCallSource("USER");
            adminService.refreshUsers();
        } finally {
            CallContextHolder.clear();
        }
    }
}

• 定时任务：

@Service
public class ScheduleService {
    @Autowired
    private AdminService adminService;

    @Scheduled(fixedRate = 3600000)
    public void scheduledRefresh() {
        try {
            CallContextHolder.setCallSource("SCHEDULE");
            adminService.refreshUsers();
        } finally {
            CallContextHolder.clear();
        }
    }
}

步骤3：AdminService根据上下文选择服务 #

@Service
public class AdminService {
    @Autowired
    @Qualifier("userGraphService")
    private GraphService userGraphService;

    @Autowired
    @Qualifier("appGraphService")
    private GraphService appGraphService;

    public void refreshUsers() {
        String source = CallContextHolder.getCallSource();
        GraphService targetService = "USER".equals(source) ? userGraphService : appGraphService;
        List<User> users = targetService.getAllUsers();
        // 执行后续的用户刷新逻辑
    }
}

注意：使用ThreadLocal时必须在finally块中清理上下文，防止线程池复用导致的上下文污染。

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