接口多实现类动态调用的两种方式

╰+攻爆jí腚メ 2022-12-26 11:18 243阅读 0赞

### 接口多实现类动态调用的两种方式 ###

**本篇内容：** 一个接口在不同场景，需要有不同的实现类，实现动态调用

> **模拟场景：** 学生、教师部分在中国，部分在泰国，处理方式需要区分，接口国际化支持会传入一个来源，根据来源区分服务实现。

代码库：[https://github.com/xpwi/springboot-multi-example][https_github.com_xpwi_springboot-multi-example]

### 方式一：利用 @Autowired 把多实现类注入到一个 Map ###

利用 @Autowired 注解，可以把同一接口的实现类，注入到集合类型中，比如 List，Map，这里使用 Map 介绍

#### （1）定义接口 ####

public interface StudentService { 
        /** * 根据id 查询学生名字 * * @param stuId Integer * @return String */
        String findStudentName(Integer stuId);
    }

#### （2）定义一个常量管理服务映射 ####

public class ServiceMapConstants { 
        /** * 学生服务 */
        public static final String STUDENT_SERVICE_PREFIX = "studentService";
    
    
        public static class StudentServiceConstants { 
            /** * 学生服务 - 中国站 */
            public static final String REQUEST_SITE_CN = "studentServiceCN";
    
            /** * 学生服务 - 泰国站 */
            public static final String REQUEST_SITE_TH = "studentServiceTH";
        }
    }

#### （3）定义两个实现类 ####

中国区

@Service(ServiceMapConstants.StudentServiceConstants.REQUEST_SITE_CN)
    public class StudentServiceCnImpl implements StudentService { 
    
        @Override
        public String findStudentName(Integer stuId) { 
            return "小明 --from cn";
        }
    }

泰国区

@Service(ServiceMapConstants.StudentServiceConstants.REQUEST_SITE_TH)
    public class StudentServiceThImpl implements StudentService { 
    
        @Override
        public String findStudentName(Integer stuId) { 
            return "XiaoMing --from th";
        }
    }

#### （4）注入与调用服务 ####

@Autowired
        private Map<String, StudentService> studentServiceMap;
    
        @ApiOperation(value = "根据 ID 获取学生名称")
        @GetMapping("/getStudentName")
        public String getStudentById(Integer studentId, String source) { 
            String key = ServiceMapConstants.STUDENT_SERVICE_PREFIX + source;
            // 如果没找到默认为中国
            StudentService studentService = Optional.ofNullable(studentServiceMap.get(key))
                    .orElse(studentServiceMap.get(ServiceMapConstants.StudentServiceConstants.REQUEST_SITE_CN));
            return studentService.findStudentName(studentId);
        }

### 方式二：利用 ApplicationContext ###

通过实现 ApplicationContextAware 接口，获取 ApplicationContext 对象，再通过名称或类型去获取具体的实现

#### （1）定义接口 ####

public interface TeacherService { 
    
        /** * 根据id 查询教师名字 * * @param teacherId Integer * @return String */
        String findTeacherName(Integer teacherId);
    }

#### （2）定义一个常量管理服务映射 ####

public class ServiceMapConstants { 
    
        /** * 学生服务 */
        public static final String TEACHER_SERVICE_PREFIX = "teacherService";
    
        public static class TeacherServiceConstants { 
            /** * 教师服务 - 中国站 */
            public static final String REQUEST_SITE_CN = "teacherServiceCN";
    
            /** * 学生服务 - 泰国站 */
            public static final String REQUEST_SITE_TH = "teacherServiceTH";
        }
    }

#### （3）定义两个实现类 ####

中国区

@Service(ServiceMapConstants.TeacherServiceConstants.REQUEST_SITE_CN)
    public class TeacherServiceCnImpl implements TeacherService { 
    
        @Override
        public String findTeacherName(Integer teacherId) { 
            return "张老师 --from cn";
        }
    }

泰国区

@Service(ServiceMapConstants.TeacherServiceConstants.REQUEST_SITE_TH)
    public class TeacherServiceThImpl implements TeacherService { 
    
        @Override
        public String findTeacherName(Integer teacherId) { 
            return "Teacher Zhang. --from th";
        }
    }

#### （4）定义获取服务的 Context ####

@Component
    public class ServiceBeanContext implements ApplicationContextAware { 
    
        private static ApplicationContext context;
    
        @Override
        public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { 
            context = applicationContext;
        }
    
        public static Object getProvider(String providerKey) { 
            Assert.notNull(providerKey, "provider key not null");
            return context.getBean(providerKey);
        }
    }

#### （5）注入与调用服务 ####

@ApiOperation(value = "根据 ID 获取教师名称")
        @GetMapping("/getTeacherName")
        public String getStudentById(Integer teacherId, String source) { 
            try { 
                String key = ServiceMapConstants.TEACHER_SERVICE_PREFIX + source;
    
                TeacherService teacherService = (TeacherService) ServiceBeanContext.getProvider(key);
                if (teacherService != null) { 
                    return teacherService.findTeacherName(teacherId);
                }
            } catch (Exception e) { 
                log.error("TeacherController.getStudentById 发生异常!", e);
            }
            return "";
        }

### 其他 ###

本篇只是说一下思路，有很多点都可以去自定义，如果有其他方式也欢迎评论

完整代码已上传代码库：[https://github.com/xpwi/springboot-multi-example][https_github.com_xpwi_springboot-multi-example]

[https_github.com_xpwi_springboot-multi-example]: https://github.com/xpwi/springboot-multi-example/tree/master/springboot-multi-service-impl