首次提交:初始化项目代码
This commit is contained in:
@@ -0,0 +1,59 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
)
|
||||
|
||||
// PrefixReader 自定义Reader,读取固定前缀数据
|
||||
type PrefixReader struct {
|
||||
data []byte
|
||||
offset int // 读取偏移量
|
||||
}
|
||||
|
||||
// NewPrefixReader 创建自定义Reader实例
|
||||
func NewPrefixReader(s string) *PrefixReader {
|
||||
return &PrefixReader{
|
||||
data: []byte(s),
|
||||
}
|
||||
}
|
||||
|
||||
// Read 实现io.Reader接口
|
||||
func (r *PrefixReader) Read(p []byte) (int, error) {
|
||||
if r.offset >= len(r.data) {
|
||||
return 0, io.EOF
|
||||
}
|
||||
n := copy(p, r.data[r.offset:])
|
||||
r.offset += n
|
||||
return n, nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
// 初始化自定义Reader
|
||||
reader := NewPrefixReader("Go IO Reader 最佳实践")
|
||||
buf := make([]byte, 16)
|
||||
|
||||
// 打开文件写入自定义Reader读取的数据
|
||||
file, err := os.Create("io/reader.txt")
|
||||
if err != nil {
|
||||
fmt.Printf("创建文件失败: %v\n", err)
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// 流式读取并写入文件
|
||||
for {
|
||||
n, err := reader.Read(buf)
|
||||
if err == io.EOF {
|
||||
break
|
||||
}
|
||||
if err != nil {
|
||||
errors.New("读取失败")
|
||||
return
|
||||
}
|
||||
_, _ = file.Write(buf[:n])
|
||||
}
|
||||
fmt.Println("自定义Reader读写完成")
|
||||
}
|
||||
Reference in New Issue
Block a user