Go语言编程中对文件读写的基本方法整理
投稿：goldensun
1.func Copy(dst Writer, src Reader) (written int64, err error)这个函数是从一个文件读取拷贝到另外一个文件，一直拷贝到读取文件的EOF，所以不会返回io.EOF错误，参数是写入目标器和读取目标器，返回int64的拷贝字节数和err信息
import (
"fmt"
"io"
"os"
)
func main() {
r, _ := os.Open("test.txt")
w, _ := os.Create("write.txt")
num, err := io.Copy(w, w)
if err != nil {
fmt.Println(err)
}
fmt.Println(num) //返回int64的11 打开我的write.txt正是test.txt里边的hello widuu
}
2.func CopyN(dst Writer, src Reader, n int64) (written int64, err error)看函数就知道了跟上述的是一样的，只是多加了一个读取数的限制，然后我们看下代码
import (
"fmt"
"io"
"io/ioutil"
"os"
)
func main() {
r, _ := os.Open("test.txt")
w, _ := os.Create("write1.txt")
num, err := io.CopyN(w, r, 5)
if err != nil {
fmt.Println(err)
}
defer r.Close()
b, _ := ioutil.ReadFile("write1.txt")
fmt.Println(string(b)) //输出 hello
fmt.Println(num) //5
}
3.func ReadAtLeast(r Reader, buf []byte, min int) (n int, err error)这个函数就是从读取器中读取数据放到我们的buf中，限定了最小的读取字节数，如果我们读取的数据小于最小读取器，譬如你设定min的值是8，但是你读取的数据字节数是5就会返回一个`io.ErrUnexpectedEOF`,如果大于就会返回`io.ErrShortBuffer`,读取完毕会有`io.EOF`~~，多讲一些哈，这个Reader只要我们满足这个interface就可以用这个
type Reader interface {
Read(p []byte) (n int, err error)
}
其中*File就支持func (f *File) Read(b []byte) (n int, err error)
import (
"fmt"
"io"
"os"
)
func main() {
r, _ := os.Open("write1.txt")
b := make([]byte, 20)
defer r.Close()
var total int
for {
n, err := io.ReadAtLeast(r, b, 8)
if err == nil {
fmt.Println("Read enough value:", string(b)) // Read enough value: hello widuu
}
if err == io.ErrUnexpectedEOF { //读取了的数据小于我们限定的最小读取数据8
fmt.Println("Read fewer value:", string(b[0:n]))
}
if err == io.ErrShortBuffer{ //这个是我们设定的buf也就是b小于我们限定8
fmt.Println("buf too Short")
os.Exit(1)
}
if err == io.EOF { //读完了 输出
fmt.Println("Read end total", total) //Read end total 11
break
}
total = total + n
}
}
4.func ReadFull(r Reader, buf []byte) (n int, err error)这个函数和上边的函数是相似，只不过是读取len(buf)个，放在buf中
import (
"fmt"
"io"
"os"
)
func main() {
r, _ := os.Open("write.txt")
b := make([]byte, 20)
num, err := io.ReadFull(r, b)
defer r.Close()
if err == io.EOF {
fmt.Println("Read end total", num)
}
if err == io.ErrUnexpectedEOF {
fmt.Println("Read fewer value:", string(b[:num])) //Read fewer value: hello widuu，依然是buf长度大于读取的长度
return
}
fmt.Println("Read value:", string(b)) //如果b是5 就出现这里
}
5.func WriteString(w Writer, s string) (n int, err error)弄完读了，当然带要写了，这个函数主要是向写入目标中写入字符创，返回是写入的字节数还有error错误，主要是权限的错误，其中写入呀！都是writer这个结构就可以写入
type Writer interface {
Write(p []byte) (n int, err error)
}
跟read一样我们的*File是有func (f *File) Write(b []byte) (n int, err error)，当然其实我们的*File中就已经有WirteString了func (f *File) WriteString(s string) (ret int, err error)
import (
"fmt"
"io"
"io/ioutil"
"os"
)
func main() {
w, _ := os.OpenFile("write1.txt", os.O_RDWR, os.ModePerm)
n, err := io.WriteString(w, "ni hao ma")
if err != nil {
fmt.Println(err) //当我用os.open()的时候木有权限 悲催的~~输出write write1.txt: Access is denied.
}
defer w.Close()
b, _ := ioutil.ReadFile("write1.txt")
fmt.Println("write total", n) //write total 9
fmt.Println(string(b)) // ni hao ma
}
6.type LimitedReader
type LimitedReader struct {
R Reader // 读取器了
N int64 // 最大字节限制
}
只实现了一个方法func (l *LimitedReader) Read(p []byte) (n int, err error)其实我们不难发现这个跟我们的ReadAtLast()就是亲兄弟的节奏
import (
"fmt"
"io"
"os"
)
func main() {
reader, _ := os.Open("test.txt")
limitedreader := io.LimitedReader{
R: reader,
N: 20,
}
p := make([]byte, 10)
var total int
for {
n, err := limitedreader.Read(p)
if err == io.EOF {
fmt.Println("read total", total) //read total 11
fmt.Println("read value", string(p)) //read value hello widuu
break
}
total = total + n
}
}
7.type PipeReader
type PipeReader struct {
// contains filtered or unexported fields
}
(1)func Pipe() (*PipeReader, *PipeWriter)创建一个管道，并返回它的读取器和写入器，这个会在内存中进行管道同步，它的开启会io.Reader然后等待io.Writer的输入，没有内部缓冲，它是安全的调用Read和Write彼此和并行调用写
import (
"fmt"
"io"
"reflect"
)
func main() {
r, w := io.Pipe()
fmt.Println(reflect.TypeOf(r)) //*io.PipeReader
fmt.Println(reflect.TypeOf(w)) //*io.PipeWriter
}
(2)func (r *PipeReader) Close() error管道关闭后，正在进行或后续的写入Write操作返回ErrClosedPipe
import (
"fmt"
"io"
)
func main() {
r, w := io.Pipe()
r.Close()
_, err := w.Write([]byte("hello widuu"))
if err == io.ErrClosedPipe {
fmt.Println("管道已经关闭无法写入") //管道已经关闭无法写入
}
}
(3)func (r *PipeReader) CloseWithError(err error) error这个就是上边的r.Close关闭的时候，写入器会返回错误的信息
import (
"errors"
"fmt"
"io"
)
func main() {
r, w := io.Pipe()
r.Close()
err := errors.New("管道符关闭了") //errors这个包我们前边已经说过了，就一个方法New不会的可以看看前边的
r.CloseWithError(err)
_, err = w.Write([]byte("test"))
if err != nil {
fmt.Println(err) //管道符关闭了
}
}
(4)func (r *PipeReader) Read(data []byte) (n int, err error)标准的阅读接口，它从管道中读取数据、阻塞一直到一个写入接口关闭，如果写入端发生错误，它就会返回错误，否则返回的EOF
import (
"fmt"
"io"
)
func main() {
r, w := io.Pipe()
go w.Write([]byte("hello widuu"))
d := make([]byte, 11)
n, _ := r.Read(d) //从管道里读取数据
fmt.Println(string(d))
fmt.Println(n)
}