【轉】消除代碼中的 if-else/switch-case

  在不少時候，咱們代碼中會有不少分支，並且分支下面的代碼又有一些複雜的邏輯，相信不少人都喜歡用 if-else/switch-case 去實現。作的很差的會直接把實現的代碼放在 if-else/switch-case 的分支之下：

switch ( type ) {
    case case1:
        ...
        ...
        break;
    case case2:
        ...
        ...
        break;
    case case3:
        ...
        ...
        break
    default:
        return null;
}

 這樣的代碼不只冗長，讀起來也很是困難。作的好一點的會把這些邏輯封裝成函數而後在分支中調用：

switch ( type ) {
    case case1:
        return case1Func();
    case case2:
        return case2Func();
    case case3:
        return case3Func();
    default:
        return null;
}

 

即便這樣也是面向過程思惟的寫法，之前寫 C 程序的時候也總喜歡這樣寫，毫無設計模式可言。不只違背開閉原則，並且隨着 switch-case 分支的增多，該段代碼只會愈來愈冗長。其實這種代碼已經有成熟的模式去消除諸多的 if-else/switch-case 分支。本文就教你們在 Spring 中如何用註解+策略模式+簡單工廠的方式消除 if-else/switch-case 。咱們就拿 QQ 空間的我的中心舉例子，假如 QQ 空間我的中心有四個 tab 分別是列出個人說說、個人日誌、個人照片和個人訪客。通常的後臺代碼頗有可能以下：

 

//各個 tab 名稱的枚舉：
public enum UserRelatedType {
    /**
     * 說說
     */
    SHUOSHUO("說說"),

    /**
     * 日誌
     */
    RIZHI("日誌"),

    /**
     * 發佈
     */
    ZHAOPIAN("照片"),

    /**
     * 訪客
     */
    FANGKE("");

    private String desc;

    UserRelatedType(String desc) {
        this.desc = desc;
    }

    public String getDesc() {
        return desc;
    }

    public void setDesc(String desc) {
        this.desc = desc;
    }
}

列出 QQ 用戶我的中心相關 tab 的代碼：

public List<UserRelatedVO> listRelated(UserRelatedQuery query){

    UserRelatedType relatedType = UserRelatedType.valueOf(StringUtils.upperCase(query.getType()) );
    switch ( relatedType ) {
        case SHUOSHUO:
            return listRelatedShuoshuo( query );
        case RIZHI:
            return listRelatedRizhi( query );
        case ZHAOPIAN:
            return listRelatedZhaopian( query );
        case FANGKE:
            return listRelatedFangke( query );
        default:
            return null;
    }
}

 而採用註解+策略模式+簡單工廠，重構後代碼以下：

• 一、定義一個註解，用來徹底消除 if-else：

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface RelatedTypeAnnotation {
    /**
     * 用戶相關類型名稱
     */
    UserRelatedType value();
}

• 二、先定義了個接口，全部 tab 都要實現該接口。其中 list 是 tab 數據展現的方法。

public interface UserRelated {

    /**
     * 列出詳細信息
     *
     * @param query
     * @return
     */
    List<UserRelatedVO> list(UserRelatedQuery query);
}

•  三、定義具體的各個 tab 的實現，繼承 UserRelated 策略接口

   個人說說

@Component("userRelatedShuoshuo")
@RelatedTypeAnnotation( value = UserRelatedType.SHUOSHUO )
public class UserRelatedShuoshuo implements UserRelated {
    @Override
    public List<UserRelatedVO> list(UserRelatedQuery query) {
        System.out.println("個人說說！");
        return list;
    }
}

 個人日誌

@Component("userRelatedRizhi")
@RelatedTypeAnnotation( value = UserRelatedType.RIZHI )
public class UserRelatedRizhi implements UserRelated {
    @Override
    public List<UserRelatedVO> list(UserRelatedQuery query) {
        System.out.println("個人日誌！");
        return list;
    }
}

 個人照片

@Component("userRelatedZhaopian")
@RelatedTypeAnnotation( value = UserRelatedType.ZHAOPIAN )
public class UserRelatedZhaopian implements UserRelated {
    @Override
    public List<UserRelatedVO> list(UserRelatedQuery query) {
        System.out.println("個人照片！");
        return list;
    }
}

 個人訪客

@Component("userRelatedFangke")
@RelatedTypeAnnotation( value = UserRelatedType.FANGKE )
public class UserRelatedFangke implements UserRelated {
    @Override
    public List<UserRelatedVO> list(UserRelatedQuery query) {
        System.out.println("個人訪客！");
        return list;
    }
}

•  三、定義一個從 Spring context 獲取 bean 的工具類

@Component
public class SpringContextUtil implements ApplicationContextAware {

    private ApplicationContext context;

    public ApplicationContext getContext() {
        return context;
    }

    @Override
    public void setApplicationContext(ApplicationContext context)throws BeansException {
        this.context = context;
    }
}

•  四、定義一個簡單工廠，用來生產各類 tab 對象。

 

@Component
public class UserRelatedFactory {

    @Autowired
    SpringContextUtil springContextUtil;
    
    private static Map<UserRelatedType, UserRelated> userRelatedMap = Maps.newConcurrentMap();

    //工廠將 Spring 裝配的相關的 Bean 用 Map 保存起來
    public UserRelatedFactory(){
        Map<String, Object> beanMap = springContextUtil.getContext().getBeansWithAnnotation(RelatedTypeAnnotation.class);

        for(Object userRelated : beanMap.values()) {
            RelatedTypeAnnotation annotation = userRelated.getClass().getAnnotation(RelatedTypeAnnotation.class);
            userRelatedMap.put(annotation.value(), (UserRelated)userRelated);
        }
    }

    public static UserRelated createRelated(UserRelatedType relatedType) {
        return userRelatedMap.get( relatedType );
    }
}

 五、調用的代碼（listRelated 會在 controller 中被調用）。

public List<UserRelatedVO> listRelated(UserRelatedQuery query){

    UserRelatedType relatedType = UserRelatedType.valueOf(StringUtils.upperCase(query.getType()) );
    UserRelated related = UserRelatedFactory.createRelated( relatedType );
    if( related != null ) {
        return related.list( query );
    } else {
        return null;
    }
}

 

 重構後的代碼若是須要再新增一種 tab，好比個人好友，只須要新增一種類型繼承 UserRelated 實現其中的 list，並加上相應的註解便可。

 

做者：水目沾 連接：https://juejin.im/post/5ca9f113e51d452b5e458ec3 來源：掘金