文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Go实现替换(覆盖)文件某一行内容的示例代码

2024-04-02 19:55

关注

1、前言

有这样一个需求,我们查找到文件中带有某个关键词的一行内容后,对该行内容进行替换,替换成我们需要的新内容,比如修改网络配置文件、修改图片地址、修改代码中所有关键词等,类似于编辑器中的关键词替换功能,只不过我们是直接判断文件而已。

2、实现覆盖某一行文件内容的思路

1、打开文件
2、读取文件每一行
3、根据关键词判断是否是需要覆盖的行,是的话则从行开头写内容,使其覆盖该行旧内容
由于是覆盖,所以我们有一个前提是新写入的内容长度需要大于等于旧内容的长度,至于新内容小于旧内容的情况下,我们在扩展中再做尝试,基本思路包括两个:1、写入空内容覆盖多出的位置(应该不行,可以试一下);2、新内容写入后直接加入换行,然后将之前文件剩余内容覆盖多出来的这部分,还是覆盖的思想。

3、实现覆盖某一行内容的代码示例

我们这里以修改我的虚拟机中的网络配置文件为例做一个简单的示例(记得先备份):

原本的Ubuntu配置文件内容:

$ cat /etc/network/interfaces
# interfaces(5) file used by ifup(8) and ifdown(8)
auto lo
iface lo inet loopback

auto ens33
iface ens33 inet static
address 40.40.40.210
gateway 40.40.40.1
netmask 255.255.255.0

然后我们通过address、gateway、netmask等关键词来修改最后三行的内容,以此来修改配置文件中的ip地址、网关和子网掩码。

代码内容:

package main

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

func main() {
    //读写方式打开文件
    file, err := os.OpenFile("/etc/network/interfaces", os.O_RDWR, 0666)
    if err != nil {
        fmt.Println("open file filed.", err)
        return
    }
    //defer关闭文件
    defer file.Close()

    //获取文件大小
    stat, err := file.Stat()
    if err != nil {
        panic(err)
    }
    var size = stat.Size()
    fmt.Println("file size:", size)

    //读取文件内容到io中
    reader := bufio.NewReader(file)
    pos := int64(0)
    ip := "40.40.40.220"
    gateway := "40.40.40.1"
    netmask := "255.255.255.0"
    for {
        //读取每一行内容
        line, err := reader.ReadString('\n')
        if err != nil {
            //读到末尾
            if err == io.EOF {
                fmt.Println("File read ok!")
                break
            } else {
                fmt.Println("Read file error!", err)
                return
            }
        }
        fmt.Println(line)

        //根据关键词覆盖当前行
        if strings.Contains(line, "address") {
            bytes := []byte("address " + ip + "\n")
            file.WriteAt(bytes, pos)
        } else if strings.Contains(line, "gateway") {
            bytes := []byte("gateway " + gateway + "\n")
            file.WriteAt(bytes, pos)
        } else if strings.Contains(line, "netmask") {
            bytes := []byte("netmask " + netmask + "\n")
            file.WriteAt(bytes, pos)
        }

        //每一行读取完后记录位置
        pos += int64(len(line))
    }
}

结果:

$ sudo ./go_build_test_go_linux 
[sudo] xx 的密码: 
file size: 180
# interfaces(5) file used by ifup(8) and ifdown(8)

auto lo

iface lo inet loopback

auto ens33

iface ens33 inet static

address 40.40.40.210

gateway 40.40.40.1

netmask 255.255.255.0

File read ok!
$ cat /etc/network/interfaces
# interfaces(5) file used by ifup(8) and ifdown(8)
auto lo
iface lo inet loopback

auto ens33
iface ens33 inet static
address 40.40.40.220
gateway 40.40.40.1
netmask 255.255.255.0

关键点:每读取一行记录目前的移动位置,然后调用file.WriteAt进行内容覆盖写入(不能是追加方式),因为找到这一行的时候记录的位置正好是上一行的末尾,所以正好覆盖。

4、扩展

其实最方便的方式其实是shell脚本的方式,然后通过各种语言都可以调用,并且某些时候也可以单独执行脚本。

其次,对于新内容长度少于旧内容的长度时无法做到全部覆盖,这个时候就稍微麻烦一点,再下下一行的时候或者其余的内容全部读取然后覆盖写入即可。

不明白接口使用的查看一下官方的io包,后面我们也会再专注整理一下io标准库的接口。

到此这篇关于Go实现替换(覆盖)文件某一行内容的示例代码的文章就介绍到这了,更多相关Go 替换文件某一行内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯