asciigoat's core library
https://asciigoat.org/core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1015 B
62 lines
1015 B
package env |
|
|
|
import ( |
|
"bytes" |
|
"io" |
|
"os" |
|
) |
|
|
|
const ( |
|
MinimumBufferSize = 32 |
|
DefaultBufferSize = 4096 |
|
) |
|
|
|
type Reader struct { |
|
in io.Reader |
|
buf *bytes.Buffer |
|
out *bytes.Buffer |
|
} |
|
|
|
func (rd *Reader) Read(b []byte) (int, error) { |
|
return rd.in.Read(b) |
|
} |
|
|
|
func (rd *Reader) Close() error { |
|
if f, ok := rd.in.(io.Closer); ok { |
|
return f.Close() |
|
} |
|
return nil |
|
} |
|
|
|
// |
|
// Constructors |
|
// |
|
func NewReaderSize(in io.Reader, size int) *Reader { |
|
if size < MinimumBufferSize { |
|
size = DefaultBufferSize |
|
} |
|
|
|
return &Reader{ |
|
in: in, |
|
buf: bytes.NewBuffer(make([]byte, 0, size)), |
|
out: bytes.NewBuffer(make([]byte, 0, size)), |
|
} |
|
} |
|
|
|
func NewReader(in io.Reader) *Reader { |
|
return NewReaderSize(in, DefaultBufferSize) |
|
} |
|
|
|
func NewReaderFileSize(filename string, size int) (*Reader, error) { |
|
f, err := os.Open(filename) |
|
if err != nil { |
|
return nil, err |
|
} |
|
|
|
rd := NewReaderSize(f, size) |
|
return rd, nil |
|
} |
|
|
|
func NewReaderFile(filename string) (*Reader, error) { |
|
return NewReaderFileSize(filename, DefaultBufferSize) |
|
}
|
|
|