登录
首页 >  Golang >  Go问答

在 GO 中如何创建一个文件,其中包含特定部分的接口名称,并解析另一个文件以修改数据?

来源:stackoverflow

时间:2024-03-04 15:42:25 419浏览 收藏

IT行业相对于一般传统行业,发展更新速度更快,一旦停止了学习,很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习,精进自己的技术,尤其是初学者。今天golang学习网给大家整理了《在 GO 中如何创建一个文件,其中包含特定部分的接口名称,并解析另一个文件以修改数据?》,聊聊,我们一起来看看吧!

问题内容

我正在构建一个程序来更改 ubuntu 18.04.1 lts 中机器的 ip 地址。所以基本上我正在使用

创建一个 yaml 文件模板
network:
   ethernets:
      {interface}:
          dhcp4: no
          addresses: [{ip}/24]
          gateway4: {gateway}
          optional: true
          nameservers:
           addresses: [{dns}]
     {interface2}:
          dhcp4: no
          addresses: [{ip2}/24]
          gateway4: {gateway2}
          optional: true
          nameservers:
           addresses: [{dns2}]

现在我需要扫描此模板文件,并通过 json 调用或作为 cli 中的参数添加另一个文件中的值。

因此应用程序将采用预先准备好的模板文件。如yaml文件的示例模板。

需要3个参数 该模板文件、翻译文件(json)和输出

翻译文件也可以在命令行上解析

它基于 2 列,搜索变量如 ip 及其新值 该实用程序将搜索 {ip} 变量并将其替换为第二列值。

在 cli 上,它可能如下所示:

./searchreplace -intemplate='templateip.yaml' -translation='{ip},xxx.x.x.x; {子网},yyy.yyy.yyy.y} -outfile=/tmp/newipfile.yaml

我当前的代码对数据进行硬编码并更改 ip,但我希望它位于单独的模板文件中。这是我当前的代码

package main

import (
    "bufio"
    "fmt"
    "log"
    "net"
    "os"
    "os/exec"
    "regexp"
    "strings"
)

var pathdelete = "/etc/netplan/01-network-manager-all.yaml"
var ipv4Regex = regexp.MustCompile(`^(?:[0-9]{1,3}\.){3}[0-9]{1,3}/[0-9]+$`)
var gatewayRegex = regexp.MustCompile(`^(?:[0-9]{1,3}\.){3}[0-9]{1,3}$`)
var nsRegex = regexp.MustCompile(`^(?:[0-9]\.){3}[0-9]{1,3}\,(?:[0-9]\.){3}[0-9]{1,3}$`)
var nicRegex = regexp.MustCompile(`e([a-z]+)`)

func main() {
    reader := bufio.NewReader(os.Stdin)
    fmt.Print("Enter the IP address: ")
    IPaddress, _ := reader.ReadString('\n')
    IPaddress = strings.TrimSuffix(IPaddress, "\n")

    //sanity check
    testInput := net.ParseIP(IPaddress)
    if testInput.To4() == nil {
        fmt.Printf("%v is not a valid IPv4 address\n", testInput)
    } else {
        fmt.Printf("It is a valid IP address\n")
        writeFile()
        readFile()
    }
}
func writeFile() {
    err01 := os.Truncate(pathdelete, 0)
    if err01 != nil {
        log.Fatal(err01)
    }

    //reading Network Cards
    reader := bufio.NewReader(os.Stdin)
    fmt.Print("Enter the Ethernet Card (for example : 'ens0N' or 'eth0') ")
    nic, _ := reader.ReadString('\n')
    nic = strings.TrimSuffix(nic, "\n")
    if nicRegex.MatchString(nic) {
    } else {
        log.Fatal("Please check the name of the NIC Card you have entered")
        os.Exit(1)
    }

    //reading IP Address and DNS
    fmt.Print("Enter the IP address along with the DNS in the format xxx.xxx.x.xxx/yy ")
    IP, _ := reader.ReadString('\n')
    IP = strings.TrimSuffix(IP, "\n")

    if ipv4Regex.MatchString(IP) {
    } else {
        log.Fatal("Please check the IP address you have entered")
        os.Exit(1)
    }

    //reading Gateway
    fmt.Print("Enter the gateway in the format xxx.xxx.x.x ")
    gateway, _ := reader.ReadString('\n')
    gateway = strings.TrimSuffix(gateway, "\n")
    if gatewayRegex.MatchString(gateway) {
    } else {
        log.Fatal("Please check the Gateway you have entered")
        os.Exit(1)
    }

    //reading Name servers
    fmt.Print("Enter the nameservers seperated by comma(,) in the format x.x.x.x,y.y.y.y ")
    ns, _ := reader.ReadString('\n')
    ns = strings.TrimSuffix(ns, "\n")
    if nsRegex.MatchString(ns) {
    } else {
        log.Fatal("Please check the Name Servers you have entered")
        os.Exit(1)
    }

    // open file using READ & WRITE permission
    var file, err = os.OpenFile(pathdelete, os.O_RDWR, 0644)
    checkError(err)
    defer file.Close()

    // write some text to file
    _, err = file.WriteString("# Let NetworkManager manage all devices on this system\nnetwork:\n  version: 2\n  renderer: NetworkManager\n")
    if err != nil {
        fmt.Println(err.Error())
        return //must return here for defer statements to be called
    }
    _, err = file.WriteString("  ethernets:\n    " + nic + ":\n      dhcp4: no\n      dhcp6: no\n")
    if err != nil {
        fmt.Println(err.Error())
        return //same as above
    }
    _, err = file.WriteString("      addresses: [" + IP + "]\n      gateway4: " + gateway + "\n      nameservers:\n        addresses: [" + ns + "]")
    if err != nil {
        fmt.Println(err.Error())
        return //same as above
    }

    // save changes
    err = file.Sync()
    if err != nil {
        fmt.Println(err.Error())
        return //same as above
    }
    applyCmd := exec.Command("bash", "-c", "sudo netplan apply")
    _, err2 := applyCmd.Output()
    if err2 != nil {
        panic(err2)
    }
}

func readFile() {
    // re-open file
    var file, err = os.OpenFile(pathdelete, os.O_RDWR, 0644)
    checkError(err)
    defer file.Close()

    // read file
    var text = make([]byte, 1024)
    n, err := file.Read(text)
    if n > 0 {
        fmt.Println("\nThe data you have entered is \n ")
        fmt.Println(string(text))
    }
    //if there is an error while reading
    //just print however much was read if any
    //at return file will be closed
}

func checkError(err error) {
    if err != nil {
        fmt.Println(err.Error())
        os.Exit(1)
    }
}

解决方案


我能够通过转换命令行输入来执行此操作

./program --xyz.yaml --{接口} eth0 --ip地址 192.168.1.233/24 --{网关} 192.168.0.1 --dns 8.8.3.8,8.8.8.3

我使用映射函数将 cli 映射到字符串

func manyrandomarg() map[string]string {
    rv := make(map[string]string)
    for ix, x := range os.args {
        if x[:2] == "--" {
            rv[x] = os.args[ix+1]
        }
    }
    return rv
}

然后我只对键进行排序并执行搜索替换功能

keys := make([]string, 0, len(rv))
    for key := range rv {
        keys = append(keys, key)
    }
    sort.strings(keys) //sort keys alphabetically

必须修剪键中的“--”部分

key0 = trimleftchars(keys[0], 2)

使用的字符串替换函数是

r := strings.NewReplacer(key0, rv[keys[0]], key1, rv[keys[1]], key2, rv[keys[2]], key3, rv[keys[3]], key4, rv[keys[4]])

本篇关于《在 GO 中如何创建一个文件,其中包含特定部分的接口名称,并解析另一个文件以修改数据?》的介绍就到此结束啦,但是学无止境,想要了解学习更多关于Golang的相关知识,请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>