程式碼優化-多型代替IF條件判斷

小碼農薛堯發表於2019-12-01
優化多型

場景描述

在開發的場景中，常常會遇到打折的業務需求，每個使用者對應的等級，他們的打折情況也是不一樣的。例如普通會員打9折，青銅會員打8.5折，黃金會員打8折等等。在一般開發中最簡單的就是判斷使用者的等級，然後對訂單作對應的打折處理。

場景示例

寫了一個簡單的小示例，如下所示：

//1 代表學生 2老師   3校長
int type = 1;
if (1 == type) {
    System.out.println("學生笑嘻嘻的說話");
} else if (2 == type) {
    System.out.println("老師開心的說話");
} else {
    System.out.println("校長嚴肅的說話");
}
複製程式碼

上面的程式碼，是我們經常的做法，程式碼少的時候，看起來非常清晰，但是程式碼多起來或者有了更多的判斷條件，那上面的程式碼會更加的混亂，如果每次有修改，都要改動這部分程式碼。

解決方法

可以把上面的程式碼改成多型方式，建立三個類，學生Student,老師Teacher,校長HeadMater，父類為Person，這三個類都實現父類的方法say，如下所示:

Person.class

package me.xueyao.service;

/**
 * @author Simon.Xue
 * @date 2019-12-01 14:31
 **/
public interface Person {
    void say();
}
複製程式碼

Student.class

package me.xueyao.service.impl;

import me.xueyao.service.Person;
import org.springframework.stereotype.Service;

/**
 * @author Simon.Xue
 * @date 2019-12-01 14:34
 **/
@Service
public class Student implements Person {
    @Override
    public void say() {
        System.out.println("學生笑嘻嘻的說話");
    }
}
複製程式碼

Teacher.class

package me.xueyao.service.impl;

import me.xueyao.service.Person;
import org.springframework.stereotype.Service;

/**
 * @author Simon.Xue
 * @date 2019-12-01 14:37
 **/
@Service
public class Teacher implements Person {
    @Override
    public void say() {
        System.out.println("老師開心的說話");
    }
}
複製程式碼

HeadMaster.class

package me.xueyao.service.impl;

import me.xueyao.service.Person;
import org.springframework.stereotype.Service;

/**
 * @author Simon.Xue
 * @date 2019-12-01 14:41
 **/
@Service
public class HeadMaster implements Person {

    @Override
    public void say() {
        System.out.println("校長嚴肅的說話");
    }
}
複製程式碼

測試方法

@Test
public void testSay() {
    Person student = new Student();
    student.say();

    Person teacher = new Teacher();
    teacher.say();

    Person headMaster = new HeadMaster();
    headMaster.say();
}
複製程式碼

優化

上面的這種做法，基本上是完成了優化，但是我們還會發現了一個問題，就是每次我們還是要建立對應的物件。上面有三個類，我們就要建立有三個物件，能否再次優化一下？

因為現在專案用Sping框架，所以可以用注入來完成優化。

首先，建立一個Person列舉類,如下所示：

package me.xueyao.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;
import me.xueyao.service.impl.HeadMaster;
import me.xueyao.service.impl.Student;
import me.xueyao.service.impl.Teacher;

/**
 * @author Simon.Xue
 * @date 2019-12-01 15:55
 **/
@AllArgsConstructor
@Getter
public enum  PersonEnums {
    STUDENT(1, "學生", Student.class),
    TEACHER(2, "老師", Teacher.class),
    HEADMASTER(3, "校長", HeadMaster.class);

    Integer code;
    String msg;
    Class clazz;

    /**
     * 獲得類的名稱，因為Spring自動注入時，預設名稱是類名(首字母小寫)
     * @param code
     * @return
     */
    public static String className(Integer code) {
        for (PersonEnums value : values()) {
            if (value.getCode().equals(code)) {
                String simpleName = value.getClazz().getSimpleName();
                simpleName.substring(1);
                return String.valueOf(simpleName.charAt(0)).toLowerCase() + simpleName.substring(1);
            }
        }
        return "";
    }

}
複製程式碼

使用方式 ：

@Autowired
private Map<String, Person> personMap = new HashMap<>();
@Test
public void testSay() {
    personMap.get(PersonEnums.className(2)).say();
}
複製程式碼

相關文章