Go语言实现在文件中插入内容

逃离我推掉我的手 2023-01-01 02:59 98阅读 0赞

## 导语 ##

此脚本写的还很稚嫩，实现插入的功能也很简单，仅供大家参考，有问题可以留言或者私信。

## 代码实现 ##

**目录结构：**

**123.txt(大家准备个此名的文件，内容随便)**

123456
    abcefg

**main.go**

package main
    
    import (
    	"fmt"
    	"io"
    	"os"
    )
    
    func editFile() {
    	// 要实现追加的方式插入值而不是覆盖
    	// 源文件
    	srcfile, err := os.OpenFile("./123.txt", os.O_RDWR, 0644)
    	if err != nil {
    		fmt.Printf("open file failed, err:%v\n", err)
    		return
    	}
    	// 临时文件
    	tmpfile, err := os.OpenFile("./123.tmp", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
    	if err != nil {
    		fmt.Printf("open file failed, err:%v\n", err)
    		return
    	}
    	// 指定在什么位置插入内容
    	// 读取光标前的内容写到临时文件
    	var b [3]byte
    	// Read函数传入一个数组并把内容写到数组里
    	n, err := srcfile.Read(b[:])
    	if err != nil {
    		fmt.Printf("read file failed, err:%v\n", err)
    		return
    	}
    	tmpfile.Write(b[:n])
    	// 把插入的内容也写到临时文件
    	tmpfile.WriteString("hello")
    	// 把源文件光标后的内容也写入到临时文件
    	var x [1024]byte
    	for {
    		n, err := srcfile.Read(x[:])
    		if err == io.EOF {
    			tmpfile.Write(x[:n])
    			break
    		}
    		if err != nil {
    			fmt.Printf("read from file failed, err:%v\n", err)
    			return
    		}
    		tmpfile.Write(x[:n])
    	}
    	tmpfile.Close()
    	srcfile.Close()
    	// 使用os.Rename不能使用defer语句去关闭文件句柄，因为defer语句是在函数指定完去执行，但是Rename操作需要时需要先关闭句柄
    	ok := os.Rename("./123.tmp", "./123.txt")
    	if ok == nil {
    		fmt.Println("文件操作完成")
    	} else {
    		fmt.Printf("err:%v", ok)
    	}
    }
    
    func main() {
    	editFile()
    }