Chapter2. Sub command - Dead line

[robert-min]

데드라인을 사용한 사용자 입력

• 프로그램이 사용자에게 입력하도록 요청하고 사용자가 5초 이내로 입력을 마치고 엔터를 누르도록 함.
• 입력을 하지 않을 경우 기본값을 하도록 코드 구현

[robert-min]

기본 main 동작

• context.WithTimeOut : Context의 Timeout을 할 콘텍스트 생성
  • 별도로 사용할 콘텍스트가 없기 때문에 default context인 Background() 사용
  • 콘텍스트를 취소하는 함수인 cancel은 main 함수가 반환되기 직접 호출되도록 defer

var totalDuration time.Duration = 5

func main() {
    allowedDuration := totalDuration * time.Second
    ctx, cancel := context.WithTimeout(context.Background(), allowedDuration)
    defer cancel()

    name, err := getNameContext(ctx)

    if err != nil && !errors.Is(err, context.DeadlineExceeded) {
        fmt.Fprintf(os.Stdout, "%v\n", err)
        os.Exit(1)
    }
    fmt.Fprintln(os.Stdout, name)
}

[robert-min]

go routine을 활용한 사용자 입력 확인

• go routine 채널 : 채널을 통해 데이터를 주고 받는 통로 역할 make() 함수를 통해 미리 생성 참고
• select 패턴 참고

func getName(r io.Reader, w io.Writer) (string, error) {
    scanner := bufio.NewScanner(r)
    msg := "Your name please? Press the Enter key when done"
    fmt.Fprintln(w, msg)

    scanner.Scan()
    if err := scanner.Err(); err != nil {
        return "", err
    }
    name := scanner.Text()
    if len(name) == 0 {
        return "", errors.New("You entered an empty name")
    }
    return name, nil
}

func getNameContext(ctx context.Context) (string, error) {
    var err error
    name := "Default Name"
    c := make(chan error, 1)

    go func() {
        name, err = getName(os.Stdin, os.Stdout)
        c <- err
    }()

    select {
    case <-ctx.Done():
        return name, ctx.Err()
    case err := <-c:
        return name, err
    }
}