코드 생성 개요

열거형(enum)과 유사한 타입을 위한 String() 메서드를 생성하는 코드 생성기를 만들어 봅시다! 이 환경에서는 실제 go generate를 실행할 수는 없지만, 타입 정의를 기반으로 Go 소스 코드를 프로그래밍 방식으로 생성하는 코드 생성 도구의 역할을 시뮬레이션하게 됩니다.

코드는 다음 두 파일로 구성됩니다:

• generator.go: 코드 생성 로직을 작성합니다.

  Name (문자열)과 Values (상수 이름을 나타내는 문자열 슬라이스)를 가진 열거형 타입 정의를 나타내는 EnumType 구조체를 만듭니다.

  String() 메서드에 대한 Go 소스 코드를 생성하는 GenerateStringMethod(e EnumType) string 함수를 만듭니다. 생성된 코드는 다음 패턴을 따라야 합니다:

  func (e TypeName) String() string {
      switch e {
      case Value1:
          return "Value1"
      case Value2:
          return "Value2"
      default:
          return "Unknown"
      }
  }

  또한 iota를 사용하여 상수 선언을 생성하는 GenerateConstants(e EnumType) string 함수를 만듭니다:

  const (
      Value1 TypeName = iota
      Value2
      Value3
  )

  마지막으로, 이 코드가 생성된 코드임을 나타내는 헤더 주석과 함께 두 출력물을 결합하는 GenerateFullCode(e EnumType) string을 만듭니다:

  // Code generated by enum generator. DO NOT EDIT.
  
  [constants]
  
  [string method]
• main.go: 생성기를 사용하여 다양한 열거형 타입에 대한 코드를 생성합니다.

  열거형 타입 이름을 읽은 다음, 값의 개수와 각 값의 이름을 읽습니다. 이 정보를 생성기에 전달하고 생성된 전체 코드를 출력합니다.

다음 입력이 제공됩니다:

• 1행: 열거형 타입 이름
• 2행: 값의 개수
• 다음 행들: 각 값의 이름

예를 들어, 다음과 같이 입력되면:

Status
3
Pending
Active
Completed

출력은 다음과 같아야 합니다:

// Code generated by enum generator. DO NOT EDIT.

const (
    Pending Status = iota
    Active
    Completed
)

func (e Status) String() string {
    switch e {
    case Pending:
        return "Pending"
    case Active:
        return "Active"
    case Completed:
        return "Completed"
    default:
        return "Unknown"
    }
}

그리고 다음과 같이 입력되면:

Priority
2
Low
High

출력은 다음과 같아야 합니다:

// Code generated by enum generator. DO NOT EDIT.

const (
    Low Priority = iota
    High
)

func (e Priority) String() string {
    switch e {
    case Low:
        return "Low"
    case High:
        return "High"
    default:
        return "Unknown"
    }
}

이 연습은 stringer와 같은 도구 뒤에 숨겨진 핵심 개념을 보여줍니다. 즉, 런타임에 리플렉션을 사용하는 대신 컴파일 타임에 반복적이지만 타입 안전한 코드를 생성하는 것입니다. 여러분의 생성기는 파일로 저장되어 프로젝트의 나머지 부분과 함께 컴파일될 수 있는 유효한 Go 코드를 생성합니다.