策略模式-实现方式三

一枚举类

@Getter
public enum AuthTypeEnum {QCT_PASSWORD("qct_password", "密码"),MOBILE("mobile", "验证码");public final String code;public final String desc;AuthTypeEnum(String code, String desc) {this.code = code;this.desc = desc;}public static AuthTypeEnum getByCode(String codeVal) {for (AuthTypeEnum resultCodeEnum : AuthTypeEnum.values()) {if (resultCodeEnum.code.equals(codeVal)) {return resultCodeEnum;}}return null;}
}

二抽象策略类

public abstract class AbstractLoginService {public static String login( String authType) {Map<String, AbstractLoginService> LoginServices = ApplicationConfiguration.getBeansOfType(AbstractLoginService.class);for (AbstractLoginService loginService : LoginServices.values()) {if (loginService.getType().equals(authType)) {loginService.doLogin(authType);return loginService.loginFinal(authType);}}throw new RuntimeException("无效的登录方式");}public abstract String getType();protected abstract void doLogin(String authType);protected abstract String loginFinal(String authType);
}

三具体策略实现类

@Slf4j
@Service
public class CaptchaLoginService extends AbstractLoginService {@Overridepublic String getType() {return AuthTypeEnum.MOBILE.getCode();}@Overrideprotected void doLogin(String authType) {log.info("验证码登录");}@Overrideprotected String loginFinal(String authType) {return authType;}
}

@Slf4j
@Component
public class PasswordLoginService extends AbstractLoginService {@Overridepublic String getType() {return AuthTypeEnum.QCT_PASSWORD.getCode();}@Overrideprotected void doLogin(String authType) {log.info("密码登录");}@Overrideprotected String loginFinal(String authType) {return authType;}}

四应用上下文配置

@Configuration
public class ApplicationConfiguration implements ApplicationContextAware {private static ApplicationContext applicationContext;public static <T> T getBean(Class<T> clazz) {return applicationContext.getBean(clazz);}public static Object getBean(String name) {return applicationContext.getBean(name);}public static <T> T getBean(String name, Class<T> clazz) {return applicationContext.getBean(name, clazz);}public static <T> Map<String, T> getBeansOfType(Class<T> clazz) {return applicationContext.getBeansOfType(clazz);}@Overridepublic void setApplicationContext(@NonNull ApplicationContext applicationContext) throws BeansException {ApplicationConfiguration.applicationContext = applicationContext;}
}

五单元测试

@SpringBootTest
class LoginTest {@Testvoid loginPasswordTest() {String loginType = AuthTypeEnum.QCT_PASSWORD.code;String login = AbstractLoginService.login(loginType);System.out.println("login = " + login);}@Testvoid loginCaptchaTest() {String loginType = AuthTypeEnum.MOBILE.code;String login = AbstractLoginService.login(loginType);System.out.println("login = " + login);}}