golang 代码实现修改配置文件

最新推荐文章于 2024-07-01 15:40:52 发布

whatday

最新推荐文章于 2024-07-01 15:40:52 发布

阅读量2k

点赞数 2

原文链接：https://blog.csdn.net/whatday/article/details/109703096

版权

本文介绍了在Golang中如何修改配置文件，特别是数据库连接配置。通过打开文件、读取内容、替换特定参数并写回文件，实现了配置的动态更新。示例代码展示了如何查找并替换配置文件中address、port和dbname的值。

摘要由CSDN通过智能技术生成

我们在使用golang的场景中，经常会有修改某个配置文件的需求，那么怎么处理呢？
下面我用一个例子给大家演示一下：
现在，我有一个连接数据库的配置文件如下（配置文件路径为ConfigPath）：

[Config]
address = localhost
port = 9527
dbname = XXX

如果我要修改这几个参数，那么可以使用下面函数处理：

func ChangeTestConfig(address, port ,dbname string) error{
    file, err:= os.OpenFile(ConfigPath, os.O_RDWR, 0666)
    if err != nil {
        glog.Error("open config file fail, err: ", err)
        return err
    }
    defer file.Close()

    buf := bufio.NewReader(file)
    output := make([]byte, 0)
    for {
        line, _, c := buf.ReadLine()
        if c == io.EOF {
            break
        }
        if strings.Contains(string(line),"address"){
            newline := "address = " + address
            line = []byte(newline)
        }
        if strings.Contains(string(line),"port"){
            newline := "port = " + port
            line = []byte(newline)
        }
        if strings.Contains(string(line),"dbname"){
            newline := "dbname = " + dbname
            line = []byte(newline)
        }
        output = append(output, line...)
        output = append(output, []byte("\n")...)
    }

    if err := writeToFile(ConfigPath ,output);err != nil{
        glog.Errorf("write config file err: %v", err)
        return err
    }
    return nil
}

func writeToFile(filePath string, outPut []byte) error {
    f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_TRUNC, 0600)
    defer f.Close()
    if err != nil {
        return err
    }
    writer := bufio.NewWriter(f)
    _, err = writer.Write(outPut)
    if err != nil {
        return err
    }
    writer.Flush()
    return nil
}