[go] Flag 패키지를 활용한 명령줄 플래그 처리 기능 확장 방법

12 Dec 2023

Go 언어를 사용하는 개발자들은 Flag 패키지를 사용하여 명령줄에서 인수를 처리하고 플래그 기능을 확장할 수 있습니다. Flag 패키지는 명령줄 프로그램에 다양한 옵션을 추가하는 효과적인 도구입니다. 이번 블로그에서는 Flag 패키지를 사용하여 명령줄 플래그 처리 기능을 확장하는 방법에 대해 알아보겠습니다.

Flag 패키지의 기본 기능

Flag 패키지를 사용하면 명령줄에서 인수를 처리하고 쉽게 플래그를 정의할 수 있습니다. 다음은 간단한 예제 코드입니다.

package main

import (
	"flag"
	"fmt"
)

func main() {
	wordPtr := flag.String("word", "foo", "a string")
	numPtr := flag.Int("num", 42, "an int")
	boolPtr := flag.Bool("fork", false, "a bool")

	flag.Parse()

	fmt.Println("word:", *wordPtr)
	fmt.Println("num:", *numPtr)
	fmt.Println("fork:", *boolPtr)
}

위 예제에서 flag.String, flag.Int, flag.Bool 함수를 사용하여 각각의 플래그를 정의하고, flag.Parse 함수를 호출하여 명령줄 플래그를 처리합니다.

Flag 패키지의 확장 방법

명령줄 플래그 처리 기능을 확장하려면 Flag 패키지의 기본 기능을 활용하여 필요한 추가 기능을 구현할 수 있습니다. 예를 들어, 사용자 정의 타입을 플래그로 사용하거나, 사용자 정의 동작을 정의할 수 있습니다. 다음은 몇 가지 확장 방법 예시입니다.

사용자 정의 타입을 플래그로 사용

type CustomType int

func (c *CustomType) String() string {
	return fmt.Sprintf("%d", *c)
}

func (c *CustomType) Set(value string) error {
	num, err := strconv.Atoi(value)
	if err != nil {
		return err
	}
	*c = CustomType(num)
	return nil
}

func main() {
	var custom CustomType
	flag.Var(&custom, "custom", "a custom type")
	flag.Parse()

	fmt.Println("custom:", custom)
}

사용자 정의 동작 정의

var (
	verbose bool
)

func init() {
	flag.BoolVar(&verbose, "verbose", false, "Verbose mode")
}

func main() {
	flag.Parse()
	if verbose {
		fmt.Println("Running in verbose mode")
	}
}

위 예제에서 flag.Var를 사용하여 사용자 정의 타입을 플래그로 사용하거나, flag.BoolVar를 사용하여 사용자 정의 동작을 정의할 수 있습니다.

마무리

Go 언어의 Flag 패키지를 사용하여 명령줄 플래그 처리 기능을 확장하고 다양한 사용자 정의 동작을 구현할 수 있습니다. Flag 패키지를 활용하여 명령줄 프로그램을 보다 유연하고 사용자 친화적으로 개발할 수 있습니다.

더 많은 정보를 원하시면 Go 공식 문서를 참조하세요.