Golang之文件操作 | 易学教程

1. 相关api介绍

建立File内存地址api

//返回File的内存地址,错误信息,通过os库调用
func Create(name string) (file *File, err Error)
//返回文件的内存地址,通过os库调用
func NewFile(fd int, name string) *File

打开文件api

//返回File的内存地址,错误信息,通过os库调用
func Open(name string) (file *File, err Error)
//返回File的内存地址,错误信息,通过os库调用
func OpenFile(name string, flag int, perm unit32) (file *File, err Error)

写文件api

//写入一个slice,返回写的个数,错误信息,通过File的内存地址调用
func (file *File) Write(b []byte) (n int, err Error)
//从slice的某个位置开始写入,返回写的个数,错误信息,通过File的内存地址调用
func (file *File) WriteAt(b []byte, off int64) (n int, err Error)
//写入一个字符串,返回写的个数,错误信息,通过File的内存地址调用
func (file *File) WriteString(s string) (ret int, err Error)

读文件api

//读取一个slice,返回读的个数,错误信息,通过File的内存地址调用
func (file *File) Read(b []byte) (n int, err Error)
//从slice的某个位置开始读取,返回读到的个数,错误信息,通过File的内存地址调用
func (file *File) ReadAt(b []byte, off int64) (n int, err Error)

删除文件api

//传入文件的路径来删除文件,返回错误个数
func Remove(name string) Error

2. 写文件的实例代码

编写代码vim main.go

package main

import (
	"fmt"
	"os"
)

func main() {
	myFile := "./abc.txt"
	fout, err := os.Create(myFile) //fout,	err := os.OpenFile(myFile,os.O_CREATE,0644)
	if err != nil {
		fmt.Println(err)
		return
	}
	for i := 0; i < 5; i++ {
		outstr := fmt.Sprintf("%s:%d\n", "Hello world", i)
		fout.WriteString(outstr)
		fout.Write([]byte("abcd\n"))
	}
	fout.Close()
}

执行代码：go run main.go

成功: 进程退出代码 0.

查看输出文件:cat abc.txt

Hello world:0
abcd
Hello world:1
abcd
Hello world:2
abcd
Hello world:3
abcd
Hello world:4
abcd

3. 读文件的实例代码

package main

import (
	"fmt"
	"os"
)

func test_read_file(filename string) {
	fin, err := os.Open(filename)
	if err != nil {
		fmt.Println(err)
	}
	defer fin.Close()
	buf := make([]byte, 1024) //开辟1024个字节的slice作为缓冲
	for {
		n, _ := fin.Read(buf)
		if n == 0 { //0	表示到达EOF
			break
		}
		os.Stdout.Write(buf)
	}
}
func main() {
	test_read_file("./abc.txt")
}

4. 同时使用os.Open和os.Create操作文件

package main

import (
	"io"
	"os"
)

func main() {
	fi, err := os.Open("./abc.txt") //打开输入*File
	if err != nil {
		panic(err)
	}
	defer fi.Close()
	fo, err := os.Create("./abc_new.txt") //创建输出*File
	if err != nil {
		panic(err)
	}
	defer fo.Close()
	buf := make([]byte, 1024)
	for {
		n, err := fi.Read(buf) //从input.txt读取
		if err != nil && err != io.EOF {
			panic(err)
		}
		if n == 0 {
			break
		}
		if n2, err := fo.Write(buf[:n]); err != nil {
			//写入output .txt,直到错误
			panic(err)
		} else if n2 != n {
			panic("error	in	writing")
		}
	}
}

5. 使用bufio库来操作文件读写

package main

import (
	"bufio"
	"io"
	"os"
)

func main() {
	fi, err := os.Open("input.txt") //打开输入*File
	if err != nil {
		panic(err)
	}
	defer fi.Close()
	r := bufio.NewReader(fi)           //创建一个读取缓冲流
	fo, err := os.Create("output.txt") //创建输出*File
	if err != nil {
		panic(err)
	}
	defer fo.Close()
	w := bufio.NewWriter(fo) //创建输出缓冲流
	buf := make([]byte, 1024)
	for {
		n, err := r.Read(buf)
		if err != nil && err != io.EOF {
			panic(err)
		}
		if n == 0 {
			break
		}
		if n2, err := w.Write(buf[:n]); err != nil {
			panic(err)
		} else if n2 != n {
			panic("error	in	writing")
		}
	}
	if err = w.Flush(); err != nil {
		panic(err)
	}
}

6. 使用ioutil来读写文件

package main

import (
	"io/ioutil"
)

func main() {
	b, err := ioutil.ReadFile("input.txt") //读文件
	if err != nil {
		panic(err)
	}
	err = ioutil.WriteFile("output.txt", b, 0644) //写文件
	if err != nil {
		panic(err)
	}
}

7. 遍历文件夹

package main

import (
	"flag"
	"fmt"
	"os"
	"path/filepath"
)

func getFilelist(path string) {
	err := filepath.Walk(path, func(path string, f os.FileInfo, err error) error {
		if f == nil {
			return err
		}
		if f.IsDir() {
			return nil
		}
		println(path)
		return nil
	})

	if err != nil {
		fmt.Printf("filepath.Walk()	returned	%v\n", err)
	}
}
func main() {
	flag.Parse()
	root := flag.Arg(0)
	getFilelist(root)
}

来源：oschina

链接：https://my.oschina.net/u/4243209/blog/3208199

标签

golang