go語言io和ioutil包的學習和使用

io包

package main;

import (
	"errors"
	"fmt"
	"io"
)

//io包中定義了非常多的interface
//只要實現了接口中的方法
//那么io包中的導出方法就可以傳入我們自定義的對象然后進行處理
//像什么文件數據，網絡數據，數據庫數據都可以統一操作接口

type MyReWr struct {
	//保存的數據
	data string;
	//指向當前數據讀取的位置下標
	rix int;
	//指向當前數據寫入的位置下標
	wix int;
}

func MyReWrNew(s string) *MyReWr {
	return &MyReWr{s, 0, 0};
}

//讀取數據到p中
func (m *MyReWr) Read(p []byte) (n int, err error) {
	l := len(p);
	num := 0;
	if m.rix >= len(m.data) {
		return 0, errors.New("EOF");
	}
	tmp := []byte(m.data);
	//判斷當前數據讀取的下標
	if (m.rix + l) < len(m.data) {
		num = l;
	} else {
		num = len(m.data) - m.rix;
	}
	//循環給p設置數據
	for ix := 0; ix < num; ix++ {
		p[ix] = tmp[m.rix+ix];
	}
	//增加讀取下標
	m.rix = m.rix + num;
	return num, nil;
}

//將p中數據寫入
func (m *MyReWr) Write(p []byte) (n int, err error) {
	l := len(p);
	num := 0;
	if m.wix >= len(m.data) {
		return 0, errors.New("EOF");
	}
	tmp := []byte(m.data);
	//判斷當前數據寫入的下標
	if (m.wix + l) < len(m.data) {
		num = l;
	} else {
		num = len(m.data) - m.wix;
	}
	//循環寫入數據
	for ix := 0; ix < num; ix++ {
		tmp[m.wix+ix] = p[ix];
	}
	m.data = string(tmp);
	//增加寫入下標
	m.wix = m.wix + num;
	return num, nil;
}

func main() {
	//我們自定義的一個結構，實現了Read和Write方法
	m := MyReWrNew("12345678910");
	p := make([]byte, 3);
	//循環讀取數據
	for {
		n, _ := m.Read(p);
		if n == 0 {
			break;
		}
		fmt.Println(n, string(p[0:n]));
	}
	//循環寫入數據
	for {
		n, _ := m.Write([]byte("111"));
		if n == 0 {
			break;
		}
		fmt.Println(n, m.data);
	}

	//MyReWr結構就可以使用如下方法
	m2 := MyReWrNew("666666666");
	m3 := MyReWrNew("999999999");
	tee := io.TeeReader(m2, m3);
	for {
		//循環從m2中讀取數據到p中，再寫入到m3中。
		n, _ := tee.Read(p);
		if n == 0 {
			break;
		}
		fmt.Println(m2, m3);
	}

	//向m4中拷貝m5中的數據
	m4 := MyReWrNew("aaaaaaaaa");
	m5 := MyReWrNew("mmmmmmm");
	io.Copy(m4, m5);
	fmt.Println(m4, m5);

	//從m6中讀取數據放入p2中
	m6 := MyReWrNew("abcdefghijklmo");
	p2 := make([]byte, len(m6.data));
	io.ReadFull(m6, p2);
	fmt.Println(string(p2));

	//向m7中寫入字符串，如果實現了WriteString方法會直接調用。
	m7 := MyReWrNew("hello");
	io.WriteString(m7, "world123");
	fmt.Println(m7);
}

ioutil包

package main;

import (
	"strings"
	"io/ioutil"
	"fmt"
)

func main() {
	//NopCloser返回一個讀取對象的ReadCloser接口
	//用於提供Close方法
	r := strings.NewReader("hello");
	rcl := ioutil.NopCloser(r);
	defer rcl.Close();

	//ReadAll讀取所有數據
	r2 := strings.NewReader("1234567890");
	p, _ := ioutil.ReadAll(r2);
	fmt.Println(string(p));

	//讀取目錄下信息
	fileInfo, _ := ioutil.ReadDir("./");
	for _, v := range fileInfo {
		fmt.Println(v.Name());
	}

	//讀取整個文件數據
	data, _ := ioutil.ReadFile("./1.txt");
	fmt.Println(string(data));

	//向指定文件寫入數據，如果文件不存在，則創建文件，寫入數據之前清空文件
	ioutil.WriteFile("./xxx.txt", []byte("hello,world"), 0655);

	//在當前目錄下，創建一個以test為前綴的臨時文件夾，並返回文件夾路徑
	name, _ := ioutil.TempDir("./", "test");
	fmt.Println(name);

	//在當前目錄下，創建一個以test為前綴的文件，並以讀寫模式打開文件，並返回os.File指針
	file, _ := ioutil.TempFile("./", "test");
	file.WriteString("寫入字符串");
	file.Close();
}