-
Notifications
You must be signed in to change notification settings - Fork 3
/
reader.go
56 lines (48 loc) · 1.17 KB
/
reader.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package ioKit
import (
"bufio"
"bytes"
"github.com/richelieu-yang/chimera/v3/src/file/fileKit"
"io"
"os"
"strings"
)
var (
// NewReader
/*
PS: bytes.Reader 结构体实现了 io.Reader 接口.
*/
NewReader func(b []byte) *bytes.Reader = bytes.NewReader
// NewReaderFromString
/*
PS: strings.Reader 结构体实现了 io.Reader 接口.
*/
NewReaderFromString func(s string) *strings.Reader = strings.NewReader
)
// NewReaderFromPath
/*
!!!: 要在外部手动调用 *os.File 的Close方法.
PS: os.File 结构体 实现了 io.Reader 接口.
@param path 文件(或目录)的路径
*/
func NewReaderFromPath(path string) (*os.File, error) {
if err := fileKit.AssertExist(path); err != nil {
return nil, err
}
return os.Open(path)
}
// NewBufferedReader 带缓冲的Reader.
/*
PS: bufio.Reader 结构体 实现了 io.Reader 接口.
*/
func NewBufferedReader(reader io.Reader, bufSizeArgs ...int) *bufio.Reader {
if bufferedReader, ok := reader.(*bufio.Reader); ok {
// 已经带缓冲了
return bufferedReader
}
if len(bufSizeArgs) == 0 {
// 默认缓冲大小: 4096
return bufio.NewReader(reader)
}
return bufio.NewReaderSize(reader, bufSizeArgs[0])
}