Enum의 목적
JAVA에서 Enum을 통해 상태와 행위를 한곳에서 관리할 수 있다.
예를 들어 DB에 저장된 code의 값이 “CALC_A"일 경우엔 값 그대로, “CALC_B"일 경우엔 10 한 값을, “CALC_C"일 경우엔 3을 계산하여 전달하는 경우를 생각해보자.
이 때 “DB의 테이블에서 뽑은 특정 값은 지정된 메소드와 관계가 있다.“는 사실을 코드로 나타내기 위해 Enum을 사용할 수 있다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
import java.util.function.Function;
public enum CalculatorType {
CALC_A(value -> value),
CALC_B(value -> value * 10),
CALC_C(value -> value * 30),
CALC_ETC(value -> 0L);
private Function<Long, Long> expression;
CalculatorType(Function<Long, Long> expression) {
this.expression = expression;
}
public long calculate(long value) {
return expression.apply(value);
}
}
|
1
2
3
4
5
6
7
8
9
10
|
import javax.persistence.Column;
import javax.persistence.EnumType;
import javax.persistence.Enumerated;
public class Entity {
@Column
//@Enumerated(EnumType.STRING)를 선언하면 Enum 필드가 테이블에 저장시 숫자형인 1,2,3이 아닌, Enum의 name이 저장된다.
@Enumerated(EnumType.STRING)
private CalculatorType calculatorType;
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import org.testng.annotations.Test;
import static org.junit.jupiter.api.Assertions.assertEquals;
public class CalculatorTypeTest {
private CalculatorType selectType() {
return CalculatorType.CALC_B;
}
@Test
public void 코드에_따라_서로다른_계산하기_Enum() {
CalculatorType code = selectType();
long originValue = 10000L;
long result = code.calculate(originValue);
assertEquals(result, 100000L);
}
}
|
값(상태)과 메소드(행위)가 어떤 관계가 있는지에 대해 더이상 다른 곳을 찾을 필요가 없게 되었다.
코드내에 전부 표현되어 있고, Enum 상수에게 직접 물어보면 되기 때문이다.
Enum 활용
결제를 예로 들어 Enum의 활용법을 알아보고자 한다.
결제라는 데이터는 결제 종류와 결제 수단이라는 2가지 형태로 표현된다.
예를 들어 신용카드 결제는 신용카드 결제라는 결제 수단이며, 카드라는 결제 종류에 포함된다.
이 카드 결제는 페이코, 카카오페이등 여러 결제 수단이 포함되어 있다.
결제종류, 결제수단등의 관계를 명확히 표현하며,
각 타입은 본인이 수행해야할 기능과 책임만 가질 수 있도록 하기 위해 Enum을 사용할 수 있다.
이 때 DB 테이블의 결제수단 컬럼에 잘못된 값을 등록하거나,
파라미터로 전달된 값이 잘못되었을 경우에도 관리할 수 있도록 결제수단 역시 Enum으로 등록할 수 있다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
public enum PayType {
ACCOUNT_TRANSFER("계좌이체"),
REMITTANCE("무통장입금"),
ON_SITE_PAYMENT("현장결제"),
TOSS("토스"),
PAYCO("페이코"),
CARD("신용카드"),
KAKAO_PAY("카카오페이"),
BAEMIN_PAY("배민페이"),
POINT("포인트"),
COUPON("쿠폰");
private final String title;
PayType(String title) {
this.title = title;
}
public String getTitle() {
return title;
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
import lombok.Getter;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
@Getter
public enum PayGroup {
CASH("현금", Arrays.asList(PayType.ACCOUNT_TRANSFER, PayType.REMITTANCE, PayType.ON_SITE_PAYMENT, PayType.TOSS)),
CARD("카드", Arrays.asList(PayType.PAYCO, PayType.CARD, PayType.KAKAO_PAY, PayType.BAEMIN_PAY)),
ETC("기타", Arrays.asList(PayType.POINT, PayType.COUPON)),
EMPTY("없음", Collections.EMPTY_LIST);
private final String title;
private final List<PayType> payList;
PayGroup(String title, List<PayType> payList) {
this.title = title;
this.payList = payList;
}
public static PayGroup findByPayType(PayType payType){
return Arrays.stream(PayGroup.values())
.filter(payGroup -> payGroup.hasPayType(payType))
.findAny()
.orElse(EMPTY);
}
private boolean hasPayType(PayType payType){
return payList.stream()
.anyMatch(pay -> pay == payType);
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;
class PayGroupTest {
@Test
public void PayGroup에게_직접_결제종류_물어보기_PayType () {
PayType payType = selectPayType();
PayGroup payGroup = PayGroup.findByPayType(payType);
assertEquals(payGroup.name(), "CARD");
assertEquals(payGroup.getTitle(), "카드");
}
private PayType selectPayType(){
return PayType.BAEMIN_PAY;
}
}
|
DB 혹은 API에서 PayType으로 데이터를 받아, 타입 안전성까지 확보하여 PayGroup 관련된 처리를 진행할 수 있게 되었다.
참고
https://github.com/jojoldu/blog-code/tree/master/enum-settler