避免大量ifelse(枚举、工厂模式、策略模式)

枚举

public class Test {
    public static void main(String[] args) {
        System.out.println(judge("ROLE_ROOT_ADMIN"));
    }
 
    public static String judge(String roleName){
        return RoleEnum.valueOf(roleName).operation();
    }
}
 
interface RoleOperation {
    String operation();
}
 
enum RoleEnum implements RoleOperation {
    ROLE_ROOT_ADMIN {
        @Override
        public String operation() {
            return "root";
        }
    },
    ROLE_ORDER_ADMIN {
        @Override
        public String operation() {
            return "order";
        }
    },
    ROLE_NORMAL_ADMIN {
        @Override
        public String operation() {
            return "normal";
        }
    };
}

工厂模式

import java.util.HashMap;
import java.util.Map;
 
public class Test {
    public static void main(String[] args) {
        System.out.println(judge("ROLE_ROOT_ADMIN"));
    }
 
    public static String judge(String roleName) {
        return RoleFactory.getOperation(roleName).operation();
    }
}
 
interface RoleOperation {
    String operation();
}
 
 
class RoleFactory {
    static Map<String, RoleOperation> roleOperationMap = new HashMap<>();
 
    static {
        roleOperationMap.put("ROLE_ROOT_ADMIN", new RootAdminRole("ROLE_ROOT_ADMIN"));
        roleOperationMap.put("ROLE_ORDER_ADMIN", new RootAdminRole("ROLE_ORDER_ADMIN"));
        roleOperationMap.put("ROLE_NORMAL_ADMIN", new RootAdminRole("ROLE_NORMAL_ADMIN"));
    }
 
    public static RoleOperation getOperation(String name) {
        return roleOperationMap.get(name);
    }
}
 
class RootAdminRole implements RoleOperation {
    private final String roleName;
 
    public RootAdminRole(String roleName) {
        this.roleName = roleName;
    }
 
    @Override
    public String operation() {
        return roleName + " root permission";
    }
}
 
class OrderAdminRole implements RoleOperation {
    private final String roleName;
 
    public OrderAdminRole(String roleName) {
        this.roleName = roleName;
    }
 
    @Override
    public String operation() {
        return roleName + " order permission";
    }
}
 
class NormalRole implements RoleOperation {
    private final String roleName;
 
    public NormalRole(String roleName) {
        this.roleName = roleName;
    }
 
    @Override
    public String operation() {
        return roleName + " normal permission";
    }
}

策略模式

package test;
 
public class Test {
    public static void main(String[] args) {
        System.out.println(judge(new RootAdminRole("ROLE_ROOT_ADMIN")));
    }
 
    public static String judge(RoleOperation roleOperation) {
        RoleContext roleContext = new RoleContext();
        roleContext.setOperation(roleOperation);
        return roleContext.execute();
    }
}
 
// 策略
interface RoleOperation {
    // 策略方法
    String operation();
}
 
// 环境类:持有一个策略类的引用，最终给客户端调用
class RoleContext {
    private RoleOperation roleOperation;
 
    public void setOperation(RoleOperation roleOperation) {
        this.roleOperation = roleOperation;
    }
 
    public String execute() {
        return roleOperation.operation();
    }
}
 
// 具体策略类
class RootAdminRole implements RoleOperation {
    private final String roleName;
 
    public RootAdminRole(String roleName) {
        this.roleName = roleName;
    }
 
    @Override
    public String operation() {
        return roleName + " root permission";
    }
}
 
class OrderAdminRole implements RoleOperation {
    private final String roleName;
 
    public OrderAdminRole(String roleName) {
        this.roleName = roleName;
    }
 
    @Override
    public String operation() {
        return roleName + " order permission";
    }
}
 
class NormalRole implements RoleOperation {
    private final String roleName;
 
    public NormalRole(String roleName) {
        this.roleName = roleName;
    }
 
    @Override
    public String operation() {
        return roleName + " normal permission";
    }
}