快捷導(dǎo)航

SpringBoot業(yè)務(wù)策略模式多實現(xiàn)類注入過程(Map注入)

更新時間：2026年06月09日 09:16:19 作者：小胖-Vi

這段文章主要討論了在業(yè)務(wù)處理過程中遇到的登錄場景,通過策略模式解決代碼冗余和維護(hù)困難的問題,具體介紹了策略模式的應(yīng)用場景、實現(xiàn)方式以及帶來的好處,適用于需要根據(jù)不同條件選擇不同實現(xiàn)邏輯的場景

前言

在業(yè)務(wù)處理的過程中遇見一個場景，登錄分A,B,C,他們都是對應(yīng)的實現(xiàn)都是增刪改查，但是不同的類型對應(yīng)不同的平臺登錄方式，同時對應(yīng)的增刪改查業(yè)務(wù)邏輯有存在差異，這個時候如果不想辦法設(shè)計代碼就會出現(xiàn)A,B,C三種類型都要寫一個對應(yīng)接口（XXXserver），然后分別再去實現(xiàn)他們（XXXserverImpl），如果后來再增加一個平臺登錄，我們又要去重復(fù)相關(guān)接口和實現(xiàn)，不僅顯得代碼冗余，更是不好維護(hù)和管理，所以利用設(shè)計模式---策略模式就解決了這個問題。

策略模式的引入

一個接口有多個實現(xiàn)

public interface LoginService {


    String passwordSalt = "test";


    /**
     * @param postAccessToken 請求體
     */
    ExecResult<String> prop(Login postAccessToken);

    /**
     * 是否支持特定業(yè)務(wù)信息的處理
     *
     * @return
     */
    Boolean isSupported(Login postAccessToken);
}

//實現(xiàn)一
@Slf4j
@Service
public class DingdingServiceImpl implements LoginService {



    @Override
    public ExecResult<String> prop(Login postAccessToken) {
 
      System.out.println("釘釘用戶");
        return ExecResultUtil.success(JwtUtil.build());
    }

    @Override
    public Boolean isSupported(Login postAccessToken) {
        return Objects.equals(GrantTypeEnum.DINGDING.getCode(), postAccessToken.getGrantType());
    }
}

//實現(xiàn)二
@Service
public class PasswordLoginServiceImpl implements LoginService {


    @Autowired
    private UserAccountService userAccountService;

    @Override
    public ExecResult<String> prop(Login postAccessToken) {
      
      System.out.println("普通用戶");
    return ExecResultUtil.success(JwtUtil.build());
    }

    @Override
    public Boolean isSupported(Login postAccessToken) {
        return Objects.equals(GrantTypeEnum.PASSWORD.getCode(), postAccessToken.getGrantType());
    }
}

//實現(xiàn)三
@Service
public class SysPasswordLoginServiceImpl implements LoginService {


    @Autowired
    private UserAccountService userAccountService;

    @Override
    public ExecResult<String> prop(Login postAccessToken) {
       
      System.out.println("系統(tǒng)用戶");
return ExecResultUtil.success(JwtUtil.build());
    }

    @Override
    public Boolean isSupported(Login postAccessToken) {
        return Objects.equals(GrantTypeEnum.SYS_PASSWORD.getCode(), postAccessToken.getGrantType());
    }
}

枚舉類型

@AllArgsConstructor
public enum GrantTypeEnum {

    PASSWORD("password", "普通用戶密碼"),
    PHONE("phone", "普通手機(jī)驗證碼"),
    WE_CHAT("wechat", "普通微信"),
    DINGDING("dingding", "普通釘釘"),

    SYS_PASSWORD("sys_password", "系統(tǒng)用戶密碼"),
    SYS_PHONE("sys_phone", "系統(tǒng)手機(jī)驗證碼"),
    SYS_WE_CHAT("sys_wechat", "系統(tǒng)微信"),
    SYS_DINGDING("sys_dingding", "系統(tǒng)釘釘"),
    ;
    private String code;

    private String message;

    public String getCode() {
        return this.code;
    }

    public String getMessage() {
        return this.message;
    }}

寫一個公共的類來判斷獲取對應(yīng)的實現(xiàn)類

@Service
public class LoginServiceImpl {

//關(guān)鍵在這個，原理：當(dāng)一個接口有多個實現(xiàn)類的時候，key為實現(xiàn)類名，value為實現(xiàn)類對象
    private Map<String, LoginService> procMap = new ConcurrentHashMap<>();

    @Autowired(required = false)//這個注入了多個實現(xiàn)類對象
    private List<LoginService> procList = Collections.emptyList();

    public ExecResult<String> prop(Login login) {

        LoginService proc = getProc(login);
        if (null == proc) {
            throw new RuntimeException("no grant type:" + login.getGrantType());
        }
        return proc.prop(login);
    }

    private LoginService getProc(Login grantType) {
        LoginService querier = procMap.get(grantType.getGrantType());
        if (querier != null) {
            return querier;
        }
        for (LoginService item : procList) {
            if (item.isSupported(grantType)) {
                querier = item;
                procMap.put(grantType.getGrantType(), querier);
                return querier;
            }
        }
        return null;
    }

}

控制層（為了測試用，根據(jù)業(yè)務(wù)使用，不一定用在控制層哈）

    @Autowired
    private LoginServiceImpl loginService;


    @ApiOperation(value = "登錄")
    @PostMapping("/login")
    public ExecResult<String> postAccessToken(@RequestBody Login login) {
        return loginService.prop(login);
    }