在Go中解析動態YAML的習慣用法是什麼？

我有一些處理YAML配置文件的代碼，這個文件有點失控w/type斷言，我覺得必須有更好的方法來做到這一點。在Go中解析動態YAML的習慣用法是什麼？

下面是我的配置文件中的相關片段：

plugins: 
    taxii20: 
    default: default 
    api_roots: 
     default: 
     auth: 
      - ldap 
      - mutualtls 
     collections: 
      all: 
      selector: g.V().Save("<type>").Save("<created>").All() 
      selector_query_lang: gizmo

這是我的解析代碼：

func parseTaxiiConfig() { 
    plg.ConfigMutex.Lock() 
    taxiiConfig := plg.ConfigData.Plugins["taxii20"].(map[interface{}]interface{}) 
    ConfigData = &Config{} 
    if taxiiConfig["default"] != nil { 
     ConfigData.DefaultRoot = taxiiConfig["default"].(string) 
    } 
    if taxiiConfig["api_roots"] != nil { 
     ConfigData.APIRoots = make([]model.APIRoot, 0) 
     iroots := taxiiConfig["api_roots"].(map[interface{}]interface{}) 
     for iname, iroot := range iroots { 
      root := model.APIRoot{Name: iname.(string)} 
      authMethods := iroot.(map[interface{}]interface{})["auth"].([]interface{}) 
      root.AuthMethods = make([]string, 0) 
      for _, method := range authMethods { 
       root.AuthMethods = append(root.AuthMethods, method.(string)) 
      } 
      collections := iroot.(map[interface{}]interface{})["collections"].(map[interface{}]interface{}) 
      root.Collections = make([]model.Collection, 0) 
      for icolName, icollection := range collections { 
       collection := model.Collection{Name: icolName.(string)} 
       collection.Selector = icollection.(map[interface{}]interface{})["selector"].(string) 
       collection.SelectorQueryLang = icollection.(map[interface{}]interface{})["selector_query_lang"].(string) 
       root.Collections = append(root.Collections, collection) 
      } 
      ConfigData.APIRoots = append(ConfigData.APIRoots, root) 
     } 
    } 
    plg.ConfigMutex.Unlock() 

    // debug 
    fmt.Println(ConfigData) 
}

的代碼按預期工作，但在這裏只是這麼多類型的斷言和我可以不會動搖我錯過了更好的方式的感覺。

作爲配置暗示的一個可能的關鍵項目，這是配置Caddy風格的插件系統，所以主配置分析器不能提前知道插件配置的外觀形狀。它必須將配置文件的插件部分的處理委託給插件本身。

來源

2017-08-25 Bob Aman

實際上，發現我的一些同事的代碼，真實解決這個問題http://github.com/mitchellh/mapstructure - 稍後我會更新這個問題，如果我能得到那個工作。 –

這是我想出來的。更可讀。

// Config represents TAXII 2.0 plugin structure 
type Config struct { 
    DefaultRoot string 
    APIRoots []model.APIRoot 
} 

// Intermediate config for mapstructure 
type configRaw struct { 
    DefaultRoot string    `mapstructure:"default"` 
    APIRoots map[string]apiRootRaw `mapstructure:"api_roots"` 
} 
type apiRootRaw struct { 
    AuthMethods []string     `mapstructure:"auth"` 
    Collections map[string]collectionRaw `mapstructure:"collections"` 
} 
type collectionRaw struct { 
    Selector   string `mapstructure:"selector"` 
    SelectorQueryLang string `mapstructure:"selector_query_lang"` 
} 

func parseTaxiiConfig() error { 
    plg.ConfigMutex.Lock() 
    defer plg.ConfigMutex.Unlock() 

    taxiiConfig := plg.ConfigData.Plugins["taxii20"].(map[interface{}]interface{}) 
    fmt.Println(taxiiConfig) 
    ConfigData = &Config{} 
    raw := &configRaw{} 
    err := mapstructure.Decode(taxiiConfig, raw) 
    if err != nil { 
     return err 
    } 

    ConfigData.DefaultRoot = raw.DefaultRoot 
    ConfigData.APIRoots = make([]model.APIRoot, 0) 
    for name, root := range raw.APIRoots { 
     apiRoot := model.APIRoot{Name: name} 
     apiRoot.AuthMethods = root.AuthMethods 
     apiRoot.Collections = make([]model.Collection, 0) 
     for colName, col := range root.Collections { 
      collection := model.Collection{Name: colName} 
      collection.Selector = col.Selector 
      collection.SelectorQueryLang = col.SelectorQueryLang 
      apiRoot.Collections = append(apiRoot.Collections, collection) 
     } 
     ConfigData.APIRoots = append(ConfigData.APIRoots, apiRoot) 
    } 

    return nil 
}

來源

2017-08-26 00:31:54

在Go中解析動態YAML的習慣用法是什麼？

回答

相關問題