GO语言实战十四 输入和输出

输入和输出

writer 和 reader 接口

package main

import (
	"bytes"
	"fmt"
	"os"
)

func main() {
	var b bytes.Buffer
	b.Write([]byte("hello "))
	fmt.Fprintf(&b, "word")
	b.WriteTo(os.Stdout)
}

b.Write

//Write将p的内容追加到缓冲区，如果需要，
则将缓冲区增长。返回值n是p的长度；err总是nil。
//如果缓冲区太大，Write会引起崩溃
func (b *Buffer) Write(p []byte) (n int, err error) {
	b.lastRead = opInvalid
	m, ok := b.tryGrowByReslice(len(p))
	if !ok {
		m = b.grow(len(p))
	}
	return copy(b.buf[m:], p), nil
}
// tryGrowByReslice是一种用于快速情况下的无限生长
 //内部缓冲区只需要重新分配。
//它返回应该写入字节的索引以及是否成功。
 func (b *Buffer) tryGrowByReslice(n int) (int, bool) {
	if l := len(b.buf); n <= cap(b.buf)-l {
		b.buf = b.buf[:l+n]
		return l, true
	}
	return 0, false
}

b.WriteTo(os.Stdout)

os包的Stdout值
var (
	Stdin  = NewFile(uintptr(syscall.Stdin), "/dev/stdin")
	Stdout = NewFile(uintptr(syscall.Stdout), "/dev/stdout")
	Stderr = NewFile(uintptr(syscall.Stderr), "/dev/stderr")
)

//WriteTo将数据写入w，直到缓冲区耗尽或发生错误。
//返回值n是写入的字节数；它总是适合于int
//但它是int64来匹配io.WriterTo接口。任何错误
//写入过程中遇到的问题也将返回。
func (b *Buffer) WriteTo(w io.Writer) (n int64, err error) {
	b.lastRead = opInvalid  //非读取操作
	if nBytes := b.Len(); nBytes > 0 {
		m, e := w.Write(b.buf[b.off:])
		if m > nBytes {
			panic("bytes.Buffer.WriteTo: invalid Write count")
		}
		b.off += m
		n = int64(m)
		if e != nil {
			return n, e
		}
		// all bytes should have been written, by definition of
		// Write method in io.Writer
		if m != nBytes {
			return n, io.ErrShortWrite
		}
	}
	// Buffer is now empty; reset.
	b.Reset()
	return n, nil
}

简单的CURL

package main

import (
	"fmt"
	"io"
	"net/http"
	"os"
)

/**
 * 读取远程链接 输出并写入文件
 */
func main() {
	response, err := http.Get("http://www.baidu.com")
	if err != nil {
		fmt.Println(err)
		return
	}
	//创建保存文档
	file, err := os.Create("curl.log")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer file.Close()

	//使用MultiWriter，这样就可以同时向文件和标准输出设备
	dest := io.MultiWriter(os.Stdout, file)
	io.Copy(dest, response.Body)
	if err := response.Body.Close(); err != nil {
		fmt.Println(err)
	}
}