带有毒蛇的动态组名

Question

I'm trying to get some groups from the config, which names are not known (will get more later on).

I'm new to golang and struggling a bit. I'm using Viper, because of its support for yaml, json and toml.

json config:

{
    "database": {
        "db": "mydb",
        "host": "localhost",
        "pass": "mypassword",
        "port": 3306,
        "user": "username"
    },
    "customers": {
        "company_one": {
            "address": "66 Great Queen St, London WC2B 5BX, UK",
            "contacts": [
                {
                    "email": "[e-mail]",
                    "name": "Joe Doe"
                },
                {
                    "email": "[e-mail]",
                    "name": "Jane Doe"
                }
            ]
        },
        "company_two": {
            "address": "Irish Town Pl, Gibraltar GX11 1AA, Gibraltar",
            "contacts": [
                {
                    "email": "[e-mail]",
                    "name": "Lucky Luke"
                }
            ]
        }
    }
}

and the source:

package main

import (
    "fmt"
    "log"
    "github.com/spf13/viper"
)

type Config struct {
    database Database
    customers Customer
}

type Database struct {
    host string
    port uint16
    user string
    pass string
    db string
}

type Customer struct {
    company Company
}

type Company struct {
    contacts Contact
    address string
}

type Contact struct {
    name string
    email string
}

func main() {
    viper.SetConfigName("a")
    viper.AddConfigPath(".")
    if err := viper.ReadInConfig(); err != nil {
        log.Fatal("Unable to read config file", err)
    }
    var conf Config
    database := viper.GetStringMap("database")
    for key, i := range database {
        switch key {
        case "host":
            conf.database.host = i.(string)
        case "port":
            conf.database.port = i.(uint16)
        case "user":
            conf.database.user = i.(string)
        case "pass":
            conf.database.pass = i.(string)
        case "db":
            conf.database.db = i.(string)
        }
    }
    fmt.Printf("%v
", conf)
}

I have no idea how to cycle through the customers.

I have tried this one already (converted from go-toml to viper), but it didn't work as expected:

var contMap = map[string]Contacts

cust := config.Get("customers")

for _, c := range cust.Keys() {
    sub := cust.Get(c).([]interface{})
    for _,d := range sub{
        address := d.([]interface{})[0].(string)
        hostMap[host] = Contacts{
            email: email,
            name: name,
        }
    }
    log.Printf("Customers: %s contact: %q", c, sub)
}

Answer 1

I started down the road of answering your initial question but it is so cumbersome and probably not what you actually want.

Your code as you've pasted it has a bunch of errors in it so instead let me offer a simpler solution like I was talking about in my comment.

package main

import (
    "log"

    "github.com/spf13/viper"
)

type Config struct {
    Database  Database           `mapstructure:"database"`
    Customers map[string]Company `mapstructure:"customers"`
}

type Database struct {
    Host string `mapstructure:"host"`
    Port uint16 `mapstructure:"port"`
    User string `mapstructure:"user"`
    Pass string `mapstructure:"pass"`
    Db   string `mapstructure:"db"`
}

type Company struct {
    Address  string    `mapstructure:"address"`
    Contacts []Contact `mapstructure:"contacts"`
}

type Contact struct {
    Name  string `mapstructure:"name"`
    Email string `mapstructure:"email"`
}

func main() {
    viper.SetConfigName("config")
    viper.AddConfigPath(".")
    if err := viper.ReadInConfig(); err != nil {
        log.Fatal("Unable to read config file", err)
    }
    var conf Config
    err := viper.Unmarshal(&conf)
    if err != nil {
        panic(err)
    }

    log.Printf("%#v", conf)
}

If you run this code with your JSON config file (named config) it returns the following output:

2018/04/28 14:47:54 main.Config{Database:main.Database{Host:"localhost", Port:0xcea, User:"username", Pass:"mypassword", Db:"mydb"}, Customers:map[string]main.Company{"company_two":main.Company{Address:"Irish Town Pl, Gibraltar GX11 1AA, Gibraltar", Contacts:[]main.Contact{main.Contact{Name:"Lucky Luke", Email:"[e-mail]"}}}, "company_one":main.Company{Address:"66 Great Queen St, London WC2B 5BX, UK", Contacts:[]main.Contact{main.Contact{Name:"Joe Doe", Email:"[e-mail]"}, main.Contact{Name:"Jane Doe", Email:"[e-mail]"}}}}}

And here it is reformatted as it would be written if you were creating this whole structure in code:

 Config{
    Database{
        Host: "localhost",
        Port: 0xcea,
        User: "username",
        Pass: "mypassword",
        Db:   "mydb"},
    Customers: map[string]Company{
        "company_two": Company{Address: "Irish Town Pl, Gibraltar GX11 1AA, Gibraltar",
            Contacts: []Contact{
                Contact{Name: "Lucky Luke", Email: "[e-mail]"}}},
        "company_one": Company{Address: "66 Great Queen St, London WC2B 5BX, UK",
            Contacts: []Contact{
                Contact{Name: "Joe Doe", Email: "[e-mail]"},
                Contact{Name: "Jane Doe", Email: "[e-mail]"}}},
    },
}