GO语言的IO方法实例小结
type PipeWriter
type PipeWriter struct {
// contains filtered or unexported fields
}
(1)func (w *PipeWriter) Close() error关闭管道，关闭时正在进行的Read操作将返回EOF，若管道内仍有未读取的数据，后续仍可正常读取
import ( "fmt" "io" )func main() { r, w := io.Pipe() go w.Write([]byte("hello word"))
data := make([]byte, 10) n, err := r.Read(data) w.Close() if err == io.EOF { fmt.Println("executing read return EOF") fmt.Println("executing read reads number", n) } n, _ = r.Read(data) fmt.Println(string(data)) //hello word fmt.Println("next read number", n) //next read number 0 }
(2)func (w *PipeWriter) CloseWithError(err error) error这个函数和read里边的CloseWithError是大同小异的，关闭管道，关闭时正在进行的Read操作将返回参数传入的异常，若管道内仍有未读取的数据，后续仍可正常读取
import ( "errors" "fmt" "io" )func main() { r, w := io.Pipe() go w.Write([]byte("hello widuu")) newerr := errors.New("your daye 突然关闭了") w.CloseWithError(newerr) data := make([]byte, 10) _, err := r.Read(data) if err != nil { fmt.Println(err) //your daye 突然关闭了 } }
(3)func (w *PipeWriter) Write(data []byte) (n int, err error)终于来打write了，这个是把字节切片写入管道，返回的是写入字节数和error,前边用到的太多了，随便哪一个吧
import ( "fmt" "io" )func main() { r, w := io.Pipe() go w.Write([]byte("hello widuu")) //写入的是[]byte,注意官方文档写的是，写入管道阻塞，一直到所有数据的读取结束 data := make([]byte, 11) n, _ := r.Read(data) fmt.Println(string(data)) //hello widuu fmt.Println("read number", n) //read number 10 }
type Reader
type Reader interface {
Read(p []byte) (n int, err error)
}
(1)func LimitReader(r Reader, n int64) Reader，我们之前就说了Reader这个结构，其实这就是对Reader的一次封装，限定了它读取字节数，其实他实现的就是io.LimitedReader{}这个结构
import ( "fmt" "io" "os" "reflect" )func main() { f, _ := os.Open("test.txt") defer f.Close() reader := io.LimitReader(f, 5) p := make([]byte, 5) fmt.Println(reflect.TypeOf(reader)) //*io.LimitedReader var total int for { n, err := reader.Read(p) if err == io.EOF { fmt.Println("read value", string(p[:total])) //read value hello fmt.Println(total) //5 break } total = total + n }
}
(2)func MultiReader(readers ...Reader) Reader这个函数一看就知道是封装了多个readers,跟上边的方法差不多，只是封装了多个而已，当然还去除了读取的限制，我们代码给大家测试一下
import ( "fmt" "io" "os" "reflect" )func main() { f1, _ := os.Open("test1.txt") f2, _ := os.Open("test.txt") defer f1.Close() defer f2.Close() reader := io.MultiReader(f1, f2) //*io.multiReader fmt.Println(reflect.TypeOf(reader)) p := make([]byte, 10) var total int var data string for { n, err := reader.Read(p) if err == io.EOF { fmt.Println("read end", total) //read end 17 break } total = total + n data = data + string(p[:n]) } fmt.Println("read value", data) //read value widuu2hello widuu fmt.Println("read count", total) // read count 17 }
(3)既然上边介绍读了，我这介绍个写吧type Write`func MultiWriter(writers ...Writer) Writer一样的作用只不过是这次换成写了
import ( "fmt" "io" "io/ioutil" "os" )func main() { f1, _ := os.Create("1.txt") f2, _ := os.Create("2.txt") writer := io.MultiWriter(f1, f2) writer.Write([]byte("widuu")) //千万别这么逻辑来 ，我这是测试用的哈 r1, _ := ioutil.ReadFile("1.txt") r2, _ := ioutil.ReadFile("2.txt") fmt.Println(string(r1)) //widuu fmt.Println(string(r2)) //widuu }
(4)func TeeReader(r Reader, w Writer) Reader这个方法有意思是从r中读取数据然后写入到w中，这个没有内部缓冲区，看下代码
import ( "fmt" "io" "os" "reflect" )func main() { r, _ := os.Open("test.txt") w, _ := os.Create("test2.txt") reader := io.TeeReader(r, w) fmt.Println(reflect.TypeOf(reader)) //*io.teeReader p := make([]byte, 10) n, _ := reader.Read(p) fmt.Println(string(p[:n])) //hello widu }
type SectionReader{}
type SectionReader struct {
// contains filtered or unexported fields
}
(1)func NewSectionReader(r ReaderAt, off int64, n int64) *SectionReader，你一看就知道了，其实就是通过这个方法获取到io.SectionReader,第一个参数读取器，第二个参数偏移量，第三个参数是读取多少
import ( "fmt" "io" "os" "reflect" )func main() { f, _ := os.Open("test.txt") sr := io.NewSectionReader(f, 2, 5) fmt.Println(reflect.TypeOf(sr)) //*io.SectionReader }
(2)func (s *SectionReader) Read(p []byte) (n int, err error)熟悉的read()其实就是读取数据用的，大家看函数就可以理解了，因为咱们经常遇到这个上两个都写这个了~~
import ( "fmt" "io" "os" )func main() { f, _ := os.Open("test.txt") defer f.Close() sr := io.NewSectionReader(f, 2, 5) p := make([]byte, 10) n, err := sr.Read(p) if err != nil { fmt.Println(err) } fmt.Println(string(p[:n])) //llo w }
(3)func (s *SectionReader) ReadAt(p []byte, off int64) (n int, err error)额这个跟之前的ReadAt是一样的，只不过只有一个偏移量，少了截取数，但是你要知道SectionReader做的是什么就把数据截取了，所以就不需要截取数了
import ( "fmt" "io" "os" )func main() { f, _ := os.Open("test.txt") defer f.Close() sr := io.NewSectionReader(f, 2, 5) p := make([]byte, 10) n, err := sr.ReadAt(p, 1) if err == io.EOF { fmt.Println(string(p[:n])) // lo w }
}
(4)func (s *SectionReader) Seek(offset int64, whence int) (int64, error)这个是设置文件指针的便宜量的，之前我们的os里边也是有个seek的，对SectionReader的读取起始点、当前读取点、结束点进行偏移，offset 偏移量，whence 设定选项 0:读取起始点，1:当前读取点，2:结束点(不好用)，其他：将抛出Seek: invalid whence异常
import ( "fmt" "io" "os" )func main() { f, _ := os.Open("test.txt") defer f.Close() sr := io.NewSectionReader(f, 2, 5) p := make([]byte, 10) sr.Seek(1, 0) //相当于起始的地址偏移1 n, err := sr.Read(p) if err != nil { fmt.Println(err) } fmt.Println(string(p[:n])) //lo w 是不是达到了前边的ReadAt() }
(5)func (s *SectionReader) Size() int64返回的是可以读取的字节数，这个不受偏移指针的影响，也不受当前读取的影响，我们具体看下代码
import ( "fmt" "io" "os" )func main() { f, _ := os.Open("test.txt") defer f.Close() sr := io.NewSectionReader(f, 2, 5) fmt.Println(sr.Size()) //5 p := make([]byte, 10) sr.Seek(1, 0) //相当于起始的地址偏移1 n, err := sr.Read(p) if err != nil { fmt.Println(err) } fmt.Println(string(p[:n])) //lo w fmt.Println(sr.Size()) //5 }
