正在检查yaml文件中是否存在golang文件中的字符串

Question

我对使用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%sn", 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仅等于允许deps的值，则打印"成功构建的rpm"，并且不提示os。出口(1(
• 如果p等于任何第三方deps，则打印"第三方deps的错误消息"和os。出口(1(
• 如果p等于任何内部deps，则打印"另一个错误mssg"和os。出口(1(

我该怎么做？

Answer 1

您可以使用YAML包(如https://github.com/go-yaml/yaml)，将你的文件加载到一个变量中，并在你提出的if中的每一步都对其进行检查。我会使用地图，因为看起来你会非常频繁地检查集合。

这里有一个我使用该包制作的简单示例，这样您就可以了解如何解组文件、转换为映射以及检查映射：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")
}
}