>백엔드 개발 >Golang >io.Reader에서 여러 번 읽는 방법은 무엇입니까?

io.Reader에서 여러 번 읽는 방법은 무엇입니까?

Barbara Streisand
Barbara Streisand원래의
2024-12-16 14:26:18335검색

How to Read from an io.Reader Multiple Times?

동일한 io.Reader에서 여러 번 읽는 방법

request.Body와 같은 io.Reader로 작업할 때 이미지와 같은 데이터가 포함되어 있으면 콘텐츠에 여러 번 액세스해야 할 수도 있습니다. 그러나 ioutil.ReadAll()을 사용하면 전체 콘텐츠가 소모되어 후속 읽기가 불가능해집니다.

과제

표시된 것처럼 리더의 여러 인스턴스 생성 시도 아래 코드에서 두 번째 호출에서는 nil 값이 반환됩니다.

package main

import (
    "io/ioutil"
    "log"
    "strings"
)

func main() {
    r := strings.NewReader("some io.Reader stream to be read\n")
    a := &r
    b := &r
    log.Println(ioutil.ReadAll(*a))
    log.Println(ioutil.ReadAll(*b))
}

해결 방법: TeeReader

동일한 io.Reader에서 여러 번 읽으려면 io.TeeReader를 사용할 수 있습니다. 스트림을 복제하여 여러 소스에서 동시에 읽을 수 있습니다.

package main

import (
    "bytes"
    "io"
    "io/ioutil"
    "log"
    "strings"
)

func main() {
    r := strings.NewReader("some io.Reader stream to be read\n")
    var buf bytes.Buffer
    tee := io.TeeReader(r, &buf)

    log.Println(ioutil.ReadAll(tee))
    log.Println(ioutil.ReadAll(&buf))
}

예:

위의 예에서는 먼저 다음을 읽는 TeeReader를 만듭니다. 원래 판독기와 버퍼에서. 그런 다음 TeeReader와 버퍼에서 읽어 동일한 콘텐츠를 두 번 가져옵니다.

위 내용은 io.Reader에서 여러 번 읽는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.