用golang实现替换某个文件中的字符串

╰+攻爆jí腚メ 2022-03-16 00:54 454阅读 0赞

用golang实现了某个文件中字符的替换，替换为按行替换，执行后会生成新文件，如a.txt，执行后生成a.txt.mdf。新文件即修改后的内容。

主要用来练习文件的读取与写入。

package main
    
    import (
    	"bufio"
    	"fmt"
    	"io"
    	"os"
    	"strings"
    )
    
    func main() {
    	if len(os.Args) != 4 {
    		fmt.Println("lack of config file, eg: go run main.go ${path_of_file} ${old_string} ${new_string}")
    		os.Exit(-1)
    	}
    	fileName := os.Args[1]
    	in, err := os.Open(fileName)
    	if err != nil {
    		fmt.Println("open file fail:", err)
    		os.Exit(-1)
    	}
    	defer in.Close()
    
    	out, err := os.OpenFile(fileName+".mdf", os.O_RDWR|os.O_CREATE, 0766)
    	if err != nil {
    		fmt.Println("Open write file fail:", err)
    		os.Exit(-1)
    	}
    	defer out.Close()
    
    	br := bufio.NewReader(in)
    	index := 1
    	for {
    		line, _, err := br.ReadLine()
    		if err == io.EOF {
    			break
    		}
    		if err != nil {
    			fmt.Println("read err:", err)
    			os.Exit(-1)
    		}
    		newLine := strings.Replace(string(line), os.Args[2], os.Args[3], -1)
    		_, err = out.WriteString(newLine + "\n")
    		if err != nil {
    			fmt.Println("write to file fail:", err)
    			os.Exit(-1)
    		}
    		fmt.Println("done ", index)
    		index++
    	}
    	fmt.Println("FINISH!")
    }

执行结果：

源文件：

![20190225202858460.png][]

将空格替换为逗号：

![20190225203034747.png][]

新文件：

![20190225203105317.png][]

[20190225202858460.png]: /images/20220316/a2babb28f3df442abffc233829825a22.png
[20190225203034747.png]: /images/20220316/7960a3d4220f4c889868c8c53d151c6a.png
[20190225203105317.png]: /images/20220316/8d24d28cf128410fb43762f5f817fdb4.png