[Go] enum 사용하기

 

들어가며

Go는 공식적으로 enum 키워드가 없기 때문에 별도로 정의해서 사용해야 합니다. 방법은 쉬우니 몇개 소개해드리겠습니다. 대표적으로 const + iota를 활용하거나 타입 정의(type alias)로 열거형과 유사하게 구현합니다.

 

기본 iota를 이용한 Enum

package main

import "fmt"

// Weekday 타입 정의
type Weekday int

// iota를 이용한 열거형 상수 선언
const (
    Sunday Weekday = iota
    Monday
    Tuesday
    Wednesday
    Thursday
    Friday
    Saturday
)

func main() {
    fmt.Println(Sunday, Monday, Tuesday) // 0 1 2
    fmt.Printf("%T\n", Sunday)           // main.Weekday
}

• iota는 const 블록 내에서 0부터 자동 증가하는 숫자를 생성합니다.
• Weekday 타입을 선언해서, 타입 안전성도 확보 가능합니다.
• 다른 int 값과 구분할 수 있습니다(type wrapping).

 

String 출력용 Enum

package main

import "fmt"

type Weekday int

const (
    Sunday Weekday = iota
    Monday
    Tuesday
    Wednesday
    Thursday
    Friday
    Saturday
)

// Weekday를 문자열로 변환
func (w Weekday) String() string {
    return [...]string{"Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"}[w]
}

func main() {
    day := Wednesday
    fmt.Println(day)         // Wednesday
    fmt.Println(int(day))    // 3
}

• [...string]{} 배열을 활용해서 enum 값을 문자열로 매핑합니다.
• String() 메서드 구현으로 fmt.Println에서 자동으로 호출합니다.

 

bitmask Enum 예제 (Flags)

package main

import "fmt"

type Permission uint8

const (
    Read Permission = 1 << iota  // 1
    Write                        // 2
    Execute                      // 4
)

func main() {
    var p Permission = Read | Write
    fmt.Println(p&Read != 0)   // true
    fmt.Println(p&Execute != 0) // false
}

• iota + 비트 연산으로 권한 플래그(enum flags) 구현이 가능합니다.
• | 연산으로 조합하고, & 연산으로 검사합니다.
• 플래그 논리 연산이 자주 필요한 경우에 사용됩니다.

 

실무형 예제: 상태값 Enum

package main

import "fmt"

// Order 상태 정의
type OrderStatus int

const (
    Pending OrderStatus = iota
    Paid
    Shipped
    Completed
    Cancelled
)

func (s OrderStatus) String() string {
    return [...]string{"Pending", "Paid", "Shipped", "Completed", "Cancelled"}[s]
}

func main() {
    var status OrderStatus = Shipped

    switch status {
    case Pending, Paid:
        fmt.Println("Order not delivered yet")
    case Shipped:
        fmt.Println("Order is on the way")
    case Completed:
        fmt.Println("Order delivered")
    case Cancelled:
        fmt.Println("Order cancelled")
    }

    fmt.Println(status) // Shipped
}

 

• 다른 int 값과 혼용 가능성을 줄임으로써 타입 안전성을 보장합니다.
• String() 메서드로 디버깅/로그 출력이 쉽도록 합니다.
• Switch-case와 조합 가능하기 때문에 상태 관리에 유용합니다.