Go中的文件操作

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

文件操作

读取文件的内容并显示在终端（带缓冲区的方式），使用os.Open，file.Close,bufio.NewReader(),reader.ReadString函数和方法。

package main

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

func main() {
	// 打开一个文件
	file, err := os.Open("/home/gocode/src/filedemo/test")
	if err != nil{
		fmt.Println("open file err=",err)
	}

	defer file.Close()  // 要及时关闭文件句柄，否则发有内存泄漏

    // 创建一个*Reader,带缓冲区的，默认的缓冲区为4096
	reader := bufio.NewReader(file)
    // 循环读取文件内容
	for{
		str, err := reader.ReadString('\n')
		if err == io.EOF{  // io.EOF表示文件的末尾
			break
		}
		fmt.Print(str)
	}
	fmt.Println("文件读取结束")
}

一次将整个文件读取到内存中，这种方式适用于文件不大的情况，相关方法和函数（ioutil.ReadFile）

package main

import (
	"fmt"
	"io/ioutil"
)


func main() {
	file := "/home/gocode/src/filedemo/test"
    
    // 使用ioutil.ReadFile一次性读取
    content, err := ioutil.ReadFile(file)  // 返回值是[]byte{}
	if err != nil{
		fmt.Println("read file err:",err)
	}

	fmt.Println(string(content))  // 需要转换成字符串显示
   
    // 我们没有显示的打开文件，所以也不需要显示的关闭
    // Open()和Close()被封装到ReadFile 函数内部
}

写文件操作

(1)创建一个新文件，写入内容5句“hello Gardon”

// 写入时，使用带缓存的*Writer
package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	// 创建一个新文件，写入内容5句“hello Gardon”
	filePath := "test.txt"
	file, err := os.OpenFile(filePath,os.O_WRONLY|os.O_CREATE,0666)
	if err != nil{
		fmt.Println("open file err=",err)
		return
	}

	defer file.Close()

	// 写入内容5句“hello Gardon”
	str := "hello Gardon\n"
	// 写入时，使用带缓存的*Writer
	writer := bufio.NewWriter(file)
	for i:=0;i<5;i++{
		writer.WriteString(str)
	}
	// 因为writer是带缓存，因此在调用WriteString方法时，其实内容是先写到缓存的
	writer.Flush()
}

(2)打开一个存在的文件，将原来的内容覆盖成新的内容10句“你好Christian”

package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	filePath := "test.txt"
	file,err := os.OpenFile(filePath,os.O_WRONLY|os.O_TRUNC,0666)
	if err != nil{
		fmt.Println("open file err:",err)
		return
	}

	defer file.Close()

	writer := bufio.NewWriter(file)

	str := "你好Christian\n"
	for i:=0;i<10;i++{
		writer.WriteString(str)
	}
	writer.Flush()
}

(3)打开一个存在的文件，在原来的内容追加内容“ABC!ENGLISH”

package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	filePath := "test.txt"
	file,err := os.OpenFile(filePath,os.O_WRONLY|os.O_APPEND,0666)
	if err != nil{
		fmt.Println("open file err:",err)
		return
	}

	defer file.Close()

	writer := bufio.NewWriter(file)

	str := "ABC!ENGLISH\n"
	writer.WriteString(str)
	writer.Flush()
}

(4)打开一个存在的文件，将原来的内容读出显示在终端，并追加5句“hello 西安”

package main

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

func main() {
	filePath := "test.txt"
	file,err := os.OpenFile(filePath,os.O_RDWR|os.O_APPEND,0666)
	if err != nil{
		fmt.Println("open file err:",err)
		return
	}

	defer file.Close()

	reader := bufio.NewReader(file)
	for{
		str,err:=reader.ReadString('\n')
		if err == io.EOF{
			break
		}
		fmt.Print(str)
	}
	writer := bufio.NewWriter(file)

	str := "hello 西安\n"
	for i := 0; i<5;i++{
		writer.WriteString(str)
	}
	writer.Flush()
}

(5)将一个文件中的内容写入到另外一个文件中。(判断要写入的文件是否存在)

package main

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

// 判断文件以及文件夹是否存在
func pathExists(path string) (bool,error){
	_,err := os.Stat(path)
	if err == nil{
		return true,nil
	}
	if os.IsNotExist(err){
		return false,nil
	}
	return false,err
}


func main() {
	file1 := "test.txt"
	file2 := "kkk.txt"

	content, err := ioutil.ReadFile(file1)
	if err != nil{
		fmt.Printf("read file err=%v\n",err)
		return
	}

	b,_:= pathExists(file2)  // 判断文件是否存在，不存在则创建
	if !b{
		file,err := os.OpenFile(file2,os.O_RDWR|os.O_CREATE,0666)
		if err != nil{
			fmt.Println("create file err:",err)
			return
		}
		defer file.Close()
	}

	err = ioutil.WriteFile(file2,content,0666)
	if err != nil{
		fmt.Println("write file error=",err)
		return
	}
}

拷贝文件

package main

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

func CopyFile(dstFileName, srcFileName string) (written int64,err error){
	srcFile, err := os.Open(srcFileName)
	if err != nil{
		fmt.Println("open file err:",err)
		return
	}

	defer srcFile.Close()

	reader := bufio.NewReader(srcFile)

	dstFile, err := os.OpenFile(dstFileName,os.O_RDWR|os.O_CREATE,0666)
	if err != nil{
		fmt.Println("open file err:",err)
		return
	}

	defer dstFile.Close()

	writer := bufio.NewWriter(dstFile)

    return io.Copy(writer,reader)  // 调用io.Copy()    
}

func main() {
	srcFile := "kkk.txt"
	dstFile := "kkkk.txt"
	_, err := CopyFile(dstFile,srcFile)
	if err != nil{
		fmt.Println("copy err:",err)
	}
}

统计字符出现的次数

// 打开一个文件，创建一个Reader,每取一行，就去统计有多少个英文，数字，空格和其他字符
// 将结果保存在结构体中

package main

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

type Charcount struct{
	ChCount int
	NumCount int
	SpaceCount int
	OtherCount int
}

func main() {
	fileName := "kkk.txt"
	file,err := os.Open(fileName)
	if err != nil{
		fmt.Println("open file err:",err)
		return
	}
	defer file.Close()

	count := Charcount{}

	reader := bufio.NewReader(file)
	for {
		str,err := reader.ReadString('\n')
		if err == io.EOF{
			break
		}
		for _,v:=range str{
			switch  {
			case v >= 'a' && v <= 'z':
				fallthrough
			case v >= 'A' && v <= 'Z':
				count.ChCount += 1
			case v == ' ' || v == '\t':
				count.SpaceCount +=1
			case v >= '0' && v <= '9':
				count.NumCount += 1
			default:
				count.OtherCount += 1
			}
		}
	}
	fmt.Println(count)
}