Go HTTP POST 요청에서 업로드 진행 상황을 추적하는 방법

Go를 사용하여 Linux용 ShareX 복제본을 개발할 때 다음이 필요합니다. HTTP POST 요청을 통해 전송된 대용량 파일 업로드의 진행 상황을 모니터링합니다. 수동으로 TCP 연결을 열고 HTTP 요청을 청크로 작성하는 것은 옵션이지만 가장 효율적이고 안정적인 솔루션은 아닙니다.

ProgressReader: 사용자 정의 io.Reader

보다 효과적인 접근 방식은 실제 판독기를 둘러싸는 사용자 정의 io.Reader를 만드는 것입니다. ProgressReader라고 하는 이 사용자 지정 리더에는 Read 메서드가 호출될 때마다 업로드 진행 상황을 출력하여 실시간 모니터링을 제공하는 Reporter 기능이 있습니다.

이 아이디어를 구현하는 방법에 대한 예는 다음과 같습니다.

type ProgressReader struct {
    io.Reader
    Reporter func(r int64)
}

func (pr *ProgressReader) Read(p []byte) (n int, err error) {
    n, err = pr.Reader.Read(p)
    pr.Reporter(int64(n))
    return
}

이 사용자 정의 리더는 업로드되는 파일을 래핑하는 데 사용됩니다.

file, _ := os.Open("/tmp/blah.go")
total := int64(0)
pr := &ProgressReader{file, func(r int64) {
    total  = r
    if r > 0 {
        fmt.Println("progress", r)
    } else {
        fmt.Println("done", r)
    }
}}

마지막으로 데이터는 ioutil.Discard에 io.Copy를 사용하여 복사됩니다. 이는 진행률 출력을 인쇄하는 동시에 데이터 싱크 역할을 합니다.

io.Copy(ioutil.Discard, pr)

이 방법은 Go에서 HTTP POST 요청에 대한 업로드 진행 상황을 추적하는 간단하고 안정적인 방법을 제공하여 사용자가 진행 상황을 모니터링할 수 있도록 합니다. 대용량 파일을 실시간으로 업로드합니다.