mirror of
https://github.com/prometheus/prometheus.git
synced 2024-11-13 09:04:06 -08:00
21 lines
403 B
Go
21 lines
403 B
Go
|
package utilities
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
)
|
||
|
|
|
||
|
|
// IOReaderFactory takes in an io.Reader and returns a function that will allow you to create a new reader that begins
|
||
|
|
// at the start of the stream
|
||
|
|
func IOReaderFactory(r io.Reader) (func() io.Reader, error) {
|
||
|
|
b, err := ioutil.ReadAll(r)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return func() io.Reader {
|
||
|
|
return bytes.NewReader(b)
|
||
|
|
}, nil
|
||
|
|
}
|