【轉】消除代碼中的 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;
}

//各個 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;
    }
}

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

• 1、定義一個註解，用來完全消除 if-else：

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

• 2、先定義了個介面，所有 tab 都要實現該介面。其中 list 是 tab 數據展示的方法。

public interface UserRelated {

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

•  3、定義具體的各個 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;
    }
}

•  3、定義一個從 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;
    }
}

•  4、定義一個簡單工廠，用來生產各種 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 );
    }
}

 5、調用的代碼（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，並加上相應的註解即可。