Go语言编程中对文件读写的基本方法整理

Go  /  管理员 发布于 5年前   313

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)
}