验证yaml文件中是否包含golang文件中的字符串

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《验证yaml文件中是否包含golang文件中的字符串》，聊聊，希望可以帮助到正在努力赚钱的你。

我对使用 yaml 和 golang 相当陌生。目前，我正在创建一个 golang 程序，它解析 rpm 包以检查子系统依赖性。它扩展了 go-rpmutils 库。

到目前为止，这是我在主函数中处理条件的代码：

func main() {
    // parse the rpm
    rpm, err := rpmutils.readrpm("file.rpm")
    if err != nil {
        panic(err)
    }

    // get rpm deps
    dependencies, err := rpm.header.getstrings(rpmutils.requirename)
    if err != nil {
        panic(err)
    }

    // check for specific dep condition
    for _, p := range dependencies {
        if strings.hasprefix(p, "prefix1") && p != "string-including-prefix1" {
            fmt.printf("\t%s\n", p)
            defer os.exit(1)
        }
    }
}

我能够输出依赖项，但希望为特定子系统依赖项存在时设置几个 if else 条件。

在一个单独的 yaml 文件中，我有：

allowed-deps:
    -dep1
    -dep2
    -dep3

third-party-deps:
    -dep4
    -dep5
    -dep6

internal-deps:
    -dep7
    -dep8
    -dep9

我想将 for 循环中的 var p 值与 yaml 文件中的值进行比较。例如：

• 如果 p 仅等于 allowed-deps 中的值，则打印“成功构建 rpm”并且不提示 os.exit(1)
• 如果 p 等于任何第三方 deps，则打印“err msg forthird-party deps”并 os.exit(1)
• 如果 p 等于任何内部 deps，则打印“another err mssg”和 os.exit(1)

我怎样才能做到这一点？

解决方案

您可以使用 yaml 包（例如 https://github.com/go-yaml/yaml），将文件加载到变量中，并在您建议的 ifs 中的每个步骤中检查它。我会使用地图，因为看起来您会非常频繁地检查这些集合。

这里有一个我使用该包制作的简单示例，以便您可以了解如何解组文件、转换为地图并检查地图：https://play.golang.org/p/t1GhUPvAQNQ

package main

import (
    "fmt"
     "github.com/go-yaml/yaml"
)

const str = `
allowed-deps:
    - dep1
    - dep2
    - dep3

third-party-deps:
    - dep4
    - dep5
    - dep6

internal-deps:
    - dep7
    - dep8
    - dep9
`

type MyYAML struct {
    AllowedDeps    []string `yaml:"allowed-deps"`
    ThirdPartyDeps []string `yaml:"third-party-deps"`
    InternalDeps   []string `yaml:"internal-deps"`
}

func main() {
    var a MyYAML
    err := yaml.Unmarshal([]byte(str), &a)
    if err != nil {
        panic(err)
    }

    // Build a map for every section.
    allowedDeps    := map[string]struct{}{}
    thirdPartyDeps := map[string]struct{}{}
    internalDeps   := map[string]struct{}{}

    for _, dep := range a.AllowedDeps {
        allowedDeps[dep] = struct{}{}
    }
    for _, dep := range a.ThirdPartyDeps {
        thirdPartyDeps[dep] = struct{}{}
    }
    for _, dep := range a.InternalDeps {
        internalDeps[dep] = struct{}{}
    }

    // Some checking examples.
    if _, ok := allowedDeps["dep1"]; ok {
        fmt.Println("dep1 found")
    }
    if _, ok := thirdPartyDeps["dep1"]; ok {
        fmt.Println("dep1 found")
    }
    if _, ok := internalDeps["dep8"]; ok {
        fmt.Println("dep8 found")
    }

}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《验证yaml文件中是否包含golang文件中的字符串》文章吧，也可关注golang学习网公众号了解相关技术文章。