Go에서 입력이 STDIN으로 파이프되는지 어떻게 감지할 수 있나요?

Golang의 STDIN 입력 감지

명령줄 유틸리티를 개발할 때 다른 프로그램에서 파이프로 전달되는 입력과 대화형으로 입력되는 입력을 구별해야 하는 경우가 많습니다. 이 문서에서는 Go에서 STDIN에 데이터가 있는지 확인하는 방법을 설명합니다.

예

다음 코드를 고려하세요.

package main

import (
    "fmt"
    "io/ioutil"
    "os"
)

func main() {
    bytes, _ := ioutil.ReadAll(os.Stdin)

    if len(bytes) > 0 {
        fmt.Println("Something on STDIN: " + string(bytes))
    } else {
        fmt.Println("Nothing on STDIN")
    }
}

파이프 입력으로 호출하는 경우(예: echo foo | go run test.go), 이 코드는 입력을 올바르게 식별합니다. 그러나 파이프 입력 없이 대화형으로 호출되면 프로그램은 ioutil.ReadAll(os.Stdin)에서 무기한 대기합니다.

해결책

파이프 입력과 파이프 입력이 아닌 것을 구별하려면 os를 활용하세요. .ModeCharDevice. 이 모드는 입력이 터미널에서 오는지 여부를 식별합니다. 터미널에서 나온 것이 아니라면 파이프로 전송되었을 가능성이 높습니다.

stat, _ := os.Stdin.Stat()
if (stat.Mode() & os.ModeCharDevice) == 0 {
    fmt.Println("data is being piped to stdin")
} else {
    fmt.Println("stdin is from a terminal")
}

위 내용은 Go에서 입력이 STDIN으로 파이프되는지 어떻게 감지할 수 있나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!