golang IO COPY

golang_yh · · 17458 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

使用golang实现 IO copy,测试程序如下:

package main

import (
	"fmt"
	"io"
	"os"
	"syscall"
)

func main() {

	var errCh chan error
	errCh = Go(func() error {
		return test_stdout()
	})

	if err := <-errCh; err != nil {
		fmt.Printf("End io_copy: %s\n", err)
		//return err
	}
}

//异步执行
func Go(f func() error) chan error {
	ch := make(chan error, 1)
	go func() {
		ch <- f()
	}()
	return ch
}

//输入流复制到文件里面
func test_stdin() error {
	var out io.Writer
	var in io.Reader

	in = os.NewFile(uintptr(syscall.Stdin), "/dev/stdin")

	out, _ = os.OpenFile("D:/golang/project/docker/env.txt", os.O_RDWR, 0666)

	n, err := io.Copy(out, in)

	fmt.Printf("\n write %d err %v \n", n, err)

	return err

}

//文件里面复制到输出流
func test_stdout() error {
	var out io.Writer
	var in io.Reader

	out = os.NewFile(uintptr(syscall.Stdout), "/dev/stdout")

	in, _ = os.OpenFile("D:/golang/project/docker/env.txt", os.O_RDWR, 0666)

	n, err := io.Copy(out, in)

	fmt.Printf("\n write %d err %v \n", n, err)

	return err

}

再看看io.Copy的代码:

func Copy(dst Writer, src Reader) (written int64, err error) {
	// If the reader has a WriteTo method, use it to do the copy.
	// Avoids an allocation and a copy.
	if wt, ok := src.(WriterTo); ok {
		return wt.WriteTo(dst)
	}
	// Similarly, if the writer has a ReadFrom method, use it to do the copy.
	if rt, ok := dst.(ReaderFrom); ok {
		return rt.ReadFrom(src)
	}
	buf := make([]byte, 32*1024)
	for {
		nr, er := src.Read(buf)
		if nr > 0 {
			nw, ew := dst.Write(buf[0:nr])
			if nw > 0 {
				written += int64(nw)
			}
			if ew != nil {
				err = ew
				break
			}
			if nr != nw {
				err = ErrShortWrite
				break
			}
		}
		if er == EOF {
			break
		}
		if er != nil {
			err = er
			break
		}
	}
	return written, err
}

阻塞式的从src输入流读数据到dst输出流


有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:golang_yh

查看原文:golang IO COPY

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

17458 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传