Перейти к разбору файла yaml

Я пытаюсь разобрать файл yaml с помощью Go. К сожалению не могу понять как. У меня есть файл yaml:

---
firewall_network_rules:
  rule1:
    src:       blablabla-host
    dst:       blabla-hostname
...

У меня есть этот код Go, но он не работает:

package main

import (
    "fmt"
    "io/ioutil"
    "path/filepath"

    "gopkg.in/yaml.v2"
)

type Config struct {
    Firewall_network_rules map[string][]string
}

func main() {
    filename, _ := filepath.Abs("./fruits.yml")
    yamlFile, err := ioutil.ReadFile(filename)

    if err != nil {
        panic(err)
    }

    var config Config

    err = yaml.Unmarshal(yamlFile, &config)
    if err != nil {
        panic(err)
    }

    fmt.Printf("Value: %#v\n", config.Firewall_network_rules)
}

Когда я запускаю это, я получаю сообщение об ошибке. Я думаю, это потому, что я не создал структуру для ключей/значений src и dst. К вашему сведению: когда я изменяю это на список, это работает.

Итак, приведенный выше код анализирует это:

---
firewall_network_rules:
  rule1:
    - value1
    - value2
...

go yaml struct
person Rogier Lommers    schedule 23.02.2015    source источник

Ответы (4)


arrow_upward
27
arrow_downward

Если вы работаете с облаком Google или kubernetes и хотите проанализировать service.yaml следующим образом:

apiVersion: v1
kind: Service
metadata:
  name: myName
  namespace: default
  labels:
    router.deis.io/routable: "true"
  annotations:
    router.deis.io/domains: ""
spec:
  type: NodePort
  selector:
    app: myName
  ports:
    - name: http
      port: 80
      targetPort: 80
    - name: https
      port: 443
      targetPort: 443

Предоставление примера из реального мира, чтобы вы поняли, как можно написать вложение.

type Service struct {
    APIVersion string `yaml:"apiVersion"`
    Kind       string `yaml:"kind"`
    Metadata   struct {
        Name      string `yaml:"name"`
        Namespace string `yaml:"namespace"`
        Labels    struct {
            RouterDeisIoRoutable string `yaml:"router.deis.io/routable"`
        } `yaml:"labels"`
        Annotations struct {
            RouterDeisIoDomains string `yaml:"router.deis.io/domains"`
        } `yaml:"annotations"`
    } `yaml:"metadata"`
    Spec struct {
        Type     string `yaml:"type"`
        Selector struct {
            App string `yaml:"app"`
        } `yaml:"selector"`
        Ports []struct {
            Name       string `yaml:"name"`
            Port       int    `yaml:"port"`
            TargetPort int    `yaml:"targetPort"`
            NodePort   int    `yaml:"nodePort,omitempty"`
        } `yaml:"ports"`
    } `yaml:"spec"`
}

Существует удобный сервис yaml-to-go https://yaml.to-go.online/. который преобразует YAML в структуры go, просто введите свой YAML в эту службу, и вы получите автоматически сгенерированную структуру.

И последний unmarshal как предыдущий плакат написал:

var service Service

err = yaml.Unmarshal(yourFile, &service)
if err != nil {
    panic(err)
}

fmt.Print(service.Metadata.Name)
person JazzCat    schedule 23.03.2017

arrow_upward
9
arrow_downward

Почему бы не организовать файл yaml, как показано ниже, если вам не важно имя правила?

---
firewall_network_rules:
  - 
    name:      rule1
    src:       blablabla-host
    dst:       blabla-hostname
  - 
    name:      rule2
    src:       bla-host
    dst:       bla-hostname

Таким образом, код будет таким, он чистый и расширяемый:

type Rule struct {
    Name  string  `yaml:"name"`
    Src   string  `yaml:"src"`
    Dst   string  `yaml:"dst"`
}

type Config struct {
   FirewallNetworkRules []Rule  `yaml:"firewall_network_rules"`
}
person boyal    schedule 03.10.2016

arrow_upward
8
arrow_downward

Ну, я думаю, что я понял это сам. Следующий фрагмент кода работает нормально. Любые предложения/улучшения?

package main

import (
    "fmt"
    "io/ioutil"
    "path/filepath"

    "gopkg.in/yaml.v2"
)

type Config struct {
    Firewall_network_rules map[string]Options
}

type Options struct {
    Src string
    Dst string
}

func main() {
    filename, _ := filepath.Abs("./fruits.yml")
    yamlFile, err := ioutil.ReadFile(filename)

    if err != nil {
        panic(err)
    }

    var config Config

    err = yaml.Unmarshal(yamlFile, &config)
    if err != nil {
        panic(err)
    }

    fmt.Printf("Value: %#v\n", config.Firewall_network_rules)
}
person Rogier Lommers    schedule 23.02.2015
comment
Попробуйте более идиоматичное FirewallNetworkRules и добавьте тег структуры, чтобы зафиксировать форматирование YAML - например. ` yaml:"firewall_network_rules" ` Документы по использованию тегов struct в библиотеке YAML см. здесь: godoc.org /gopkg.in/yaml.v2#Маршал - person elithrar; 24.02.2015
comment
Спасибо за ваше предложение, оно действительно проясняет мой код. - person Rogier Lommers; 24.02.2015

arrow_upward
2
arrow_downward

Если ваш файл YAML прост (одиночное вложение), как показано ниже

mongo:
    DB: database
    COL: collection
log:
    error: log/error/error.log
api:
    key: jhgwewbcjwefwjfg

Здесь вы можете использовать интерфейс вместо объявления структуры.

main(){
  config := Config()
  mongoConfig := config["mongo"]

  mongo.MongoDial(
    String(
        Get(mongoConfig, "DB")
    ), 
    String(
        Get(mongoConfig, "COL")
    )
  )
}

func Config() map[string]interface{} {
    filename, _ := filepath.Abs("configs/config.yaml")
    yamlFile, err := ioutil.ReadFile(filename)

    if err != nil {
        panic(err)
    }

    var config map[string]interface{}

    err = yaml.Unmarshal(yamlFile, &config)
    if err != nil {
        panic(err)
    }

    return config
}
func Get(this interface{}, key string) interface{}  {
    return this.(map[interface{}]interface{})[key]
}
func String(payload interface{}) string  {
    var load string
    if pay, oh := payload.(string); oh {
        load = pay
    }else{
        load = ""
    }
    return load
}

Это отлично работает для вложенности уровня 1, если у вас сложная вложенность, рекомендуется использовать struct.

person reoxey    schedule 13.05.2018