knadh / koanf

Simple, extremely lightweight, extensible, configuration management library for Go. Support for JSON, TOML, YAML, env, command line, file, S3 etc. Alternative to viper.
MIT License
2.79k stars 153 forks source link
config config-loader configuration configuration-file configuration-management etcd-client go golang golang-package s3-bucket toml viper yaml

koanf

koanf is a library for reading configuration from different sources in different formats in Go applications. It is a cleaner, lighter alternative to spf13/viper with better abstractions and extensibility and far fewer dependencies.

koanf v2 has modules (Providers) for reading configuration from a variety of sources such as files, command line flags, environment variables, Vault, and S3 and for parsing (Parsers) formats such as JSON, YAML, TOML, Hashicorp HCL. It is easy to plug in custom parsers and providers.

All external dependencies in providers and parsers are detached from the core and can be installed separately as necessary.

Run Tests GoDoc

Installation

# Install the core.
go get -u github.com/knadh/koanf/v2

# Install the necessary Provider(s).
# Available: file, env, posflag, basicflag, confmap, rawbytes,
#            structs, fs, s3, appconfig/v2, consul/v2, etcd/v2, vault/v2, parameterstore/v2
# eg: go get -u github.com/knadh/koanf/providers/s3
# eg: go get -u github.com/knadh/koanf/providers/consul/v2

go get -u github.com/knadh/koanf/providers/file

# Install the necessary Parser(s).
# Available: toml, toml/v2, json, yaml, dotenv, hcl, hjson, nestedtext
# go get -u github.com/knadh/koanf/parsers/$parser

go get -u github.com/knadh/koanf/parsers/toml

See the list of all bundled Providers and Parsers.

Contents

Concepts

With these two interface implementations, koanf can obtain configuration in any format from any source, parse it, and make it available to an application.

Reading config from files

package main

import (
    "fmt"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/parsers/yaml"
    "github.com/knadh/koanf/providers/file"
)

// Global koanf instance. Use "." as the key path delimiter. This can be "/" or any character.
var k = koanf.New(".")

func main() {
    // Load JSON config.
    if err := k.Load(file.Provider("mock/mock.json"), json.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    // Load YAML config and merge into the previously loaded config (because we can).
    k.Load(file.Provider("mock/mock.yml"), yaml.Parser())

    fmt.Println("parent's name is = ", k.String("parent1.name"))
    fmt.Println("parent's ID is = ", k.Int("parent1.id"))
}

Watching file for changes

Some providers expose a Watch() method that makes the provider watch for changes in configuration and trigger a callback to reload the configuration. This is not goroutine safe if there are concurrent *Get() calls happening on the koanf object while it is doing a Load(). Such scenarios will need mutex locking.

file, appconfig, vault, consul providers have a Watch() method.

package main

import (
    "fmt"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/parsers/yaml"
    "github.com/knadh/koanf/providers/file"
)

// Global koanf instance. Use "." as the key path delimiter. This can be "/" or any character.
var k = koanf.New(".")

func main() {
    // Load JSON config.
    f := file.Provider("mock/mock.json")
    if err := k.Load(f, json.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    // Load YAML config and merge into the previously loaded config (because we can).
    k.Load(file.Provider("mock/mock.yml"), yaml.Parser())

    fmt.Println("parent's name is = ", k.String("parent1.name"))
    fmt.Println("parent's ID is = ", k.Int("parent1.id"))

    // Watch the file and get a callback on change. The callback can do whatever,
    // like re-load the configuration.
    // File provider always returns a nil `event`.
    f.Watch(func(event interface{}, err error) {
        if err != nil {
            log.Printf("watch error: %v", err)
            return
        }

        // Throw away the old config and load a fresh copy.
        log.Println("config changed. Reloading ...")
        k = koanf.New(".")
        k.Load(f, json.Parser())
        k.Print()
    })

    // To stop a file watcher, call:
    // f.Unwatch()

    // Block forever (and manually make a change to mock/mock.json) to
    // reload the config.
    log.Println("waiting forever. Try making a change to mock/mock.json to live reload")
    <-make(chan bool)
}

Reading from command line

The following example shows the use of posflag.Provider, a wrapper over the spf13/pflag library, an advanced commandline lib. For Go's built in flag package, use basicflag.Provider.

package main

import (
    "fmt"
    "log"
    "os"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/toml"

    // TOML version 2 is available at:
    // "github.com/knadh/koanf/parsers/toml/v2"

    "github.com/knadh/koanf/providers/file"
    "github.com/knadh/koanf/providers/posflag"
    flag "github.com/spf13/pflag"
)

// Global koanf instance. Use "." as the key path delimiter. This can be "/" or any character.
var k = koanf.New(".")

func main() {
    // Use the POSIX compliant pflag lib instead of Go's flag lib.
    f := flag.NewFlagSet("config", flag.ContinueOnError)
    f.Usage = func() {
        fmt.Println(f.FlagUsages())
        os.Exit(0)
    }
    // Path to one or more config files to load into koanf along with some config params.
    f.StringSlice("conf", []string{"mock/mock.toml"}, "path to one or more .toml config files")
    f.String("time", "2020-01-01", "a time string")
    f.String("type", "xxx", "type of the app")
    f.Parse(os.Args[1:])

    // Load the config files provided in the commandline.
    cFiles, _ := f.GetStringSlice("conf")
    for _, c := range cFiles {
        if err := k.Load(file.Provider(c), toml.Parser()); err != nil {
            log.Fatalf("error loading file: %v", err)
        }
    }

    // "time" and "type" may have been loaded from the config file, but
    // they can still be overridden with the values from the command line.
    // The bundled posflag.Provider takes a flagset from the spf13/pflag lib.
    // Passing the Koanf instance to posflag helps it deal with default command
    // line flag values that are not present in conf maps from previously loaded
    // providers.
    if err := k.Load(posflag.Provider(f, ".", k), nil); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    fmt.Println("time is = ", k.String("time"))
}

Reading environment variables

package main

import (
    "fmt"
    "log"
    "strings"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/providers/env"
    "github.com/knadh/koanf/providers/file"
)

// Global koanf instance. Use . as the key path delimiter. This can be / or anything.
var k = koanf.New(".")

func main() {
    // Load JSON config.
    if err := k.Load(file.Provider("mock/mock.json"), json.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    // Load environment variables and merge into the loaded config.
    // "MYVAR" is the prefix to filter the env vars by.
    // "." is the delimiter used to represent the key hierarchy in env vars.
    // The (optional, or can be nil) function can be used to transform
    // the env var names, for instance, to lowercase them.
    //
    // For example, env vars: MYVAR_TYPE and MYVAR_PARENT1_CHILD1_NAME
    // will be merged into the "type" and the nested "parent1.child1.name"
    // keys in the config file here as we lowercase the key, 
    // replace `_` with `.` and strip the MYVAR_ prefix so that 
    // only "parent1.child1.name" remains.
    k.Load(env.Provider("MYVAR_", ".", func(s string) string {
        return strings.Replace(strings.ToLower(
            strings.TrimPrefix(s, "MYVAR_")), "_", ".", -1)
    }), nil)

    fmt.Println("name is = ", k.String("parent1.child1.name"))
}

You can also use the env.ProviderWithValue with a callback that supports mutating both the key and value to return types other than a string. For example, here, env values separated by spaces are returned as string slices or arrays. eg: MYVAR_slice=a b c becomes slice: [a, b, c].

    k.Load(env.ProviderWithValue("MYVAR_", ".", func(s string, v string) (string, interface{}) {
        // Strip out the MYVAR_ prefix and lowercase and get the key while also replacing
        // the _ character with . in the key (koanf delimeter).
        key := strings.Replace(strings.ToLower(strings.TrimPrefix(s, "MYVAR_")), "_", ".", -1)

        // If there is a space in the value, split the value into a slice by the space.
        if strings.Contains(v, " ") {
            return key, strings.Split(v, " ")
        }

        // Otherwise, return the plain string.
        return key, v
    }), nil)

Reading from an S3 bucket

// Load JSON config from s3.
if err := k.Load(s3.Provider(s3.Config{
    AccessKey: os.Getenv("AWS_S3_ACCESS_KEY"),
    SecretKey: os.Getenv("AWS_S3_SECRET_KEY"),
    Region:    os.Getenv("AWS_S3_REGION"),
    Bucket:    os.Getenv("AWS_S3_BUCKET"),
    ObjectKey: "dir/config.json",
}), json.Parser()); err != nil {
    log.Fatalf("error loading config: %v", err)
}

Reading raw bytes

The bundled rawbytes Provider can be used to read arbitrary bytes from a source, like a database or an HTTP call.

package main

import (
    "fmt"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/providers/rawbytes"
)

// Global koanf instance. Use . as the key path delimiter. This can be / or anything.
var k = koanf.New(".")

func main() {
    b := []byte(`{"type": "rawbytes", "parent1": {"child1": {"type": "rawbytes"}}}`)
    k.Load(rawbytes.Provider(b), json.Parser())
    fmt.Println("type is = ", k.String("parent1.child1.type"))
}

Unmarshalling and marshalling

Parsers can be used to unmarshal and scan the values in a Koanf instance into a struct based on the field tags, and to marshal a Koanf instance back into serialized bytes, for example to JSON or YAML files

package main

import (
    "fmt"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/providers/file"
)

// Global koanf instance. Use . as the key path delimiter. This can be / or anything.
var (
    k      = koanf.New(".")
    parser = json.Parser()
)

func main() {
    // Load JSON config.
    if err := k.Load(file.Provider("mock/mock.json"), parser); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    // Structure to unmarshal nested conf to.
    type childStruct struct {
        Name       string            `koanf:"name"`
        Type       string            `koanf:"type"`
        Empty      map[string]string `koanf:"empty"`
        GrandChild struct {
            Ids []int `koanf:"ids"`
            On  bool  `koanf:"on"`
        } `koanf:"grandchild1"`
    }

    var out childStruct

    // Quick unmarshal.
    k.Unmarshal("parent1.child1", &out)
    fmt.Println(out)

    // Unmarshal with advanced config.
    out = childStruct{}
    k.UnmarshalWithConf("parent1.child1", &out, koanf.UnmarshalConf{Tag: "koanf"})
    fmt.Println(out)

    // Marshal the instance back to JSON.
    // The parser instance can be anything, eg: json.Parser(), yaml.Parser() etc.
    b, _ := k.Marshal(parser)
    fmt.Println(string(b))
}

Unmarshalling with flat paths

Sometimes it is necessary to unmarshal an assortment of keys from various nested structures into a flat target structure. This is possible with the UnmarshalConf.FlatPaths flag.

package main

import (
    "fmt"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/providers/file"
)

// Global koanf instance. Use . as the key path delimiter. This can be / or anything.
var k = koanf.New(".")

func main() {
    // Load JSON config.
    if err := k.Load(file.Provider("mock/mock.json"), json.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    type rootFlat struct {
        Type                        string            `koanf:"type"`
        Empty                       map[string]string `koanf:"empty"`
        Parent1Name                 string            `koanf:"parent1.name"`
        Parent1ID                   int               `koanf:"parent1.id"`
        Parent1Child1Name           string            `koanf:"parent1.child1.name"`
        Parent1Child1Type           string            `koanf:"parent1.child1.type"`
        Parent1Child1Empty          map[string]string `koanf:"parent1.child1.empty"`
        Parent1Child1Grandchild1IDs []int             `koanf:"parent1.child1.grandchild1.ids"`
        Parent1Child1Grandchild1On  bool              `koanf:"parent1.child1.grandchild1.on"`
    }

    // Unmarshal the whole root with FlatPaths: True.
    var o1 rootFlat
    k.UnmarshalWithConf("", &o1, koanf.UnmarshalConf{Tag: "koanf", FlatPaths: true})
    fmt.Println(o1)

    // Unmarshal a child structure of "parent1".
    type subFlat struct {
        Name                 string            `koanf:"name"`
        ID                   int               `koanf:"id"`
        Child1Name           string            `koanf:"child1.name"`
        Child1Type           string            `koanf:"child1.type"`
        Child1Empty          map[string]string `koanf:"child1.empty"`
        Child1Grandchild1IDs []int             `koanf:"child1.grandchild1.ids"`
        Child1Grandchild1On  bool              `koanf:"child1.grandchild1.on"`
    }

    var o2 subFlat
    k.UnmarshalWithConf("parent1", &o2, koanf.UnmarshalConf{Tag: "koanf", FlatPaths: true})
    fmt.Println(o2)
}

Reading from nested maps

The bundled confmap provider takes a map[string]interface{} that can be loaded into a koanf instance.

package main

import (
    "fmt"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/providers/confmap"
    "github.com/knadh/koanf/providers/file"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/parsers/yaml"
)

// Global koanf instance. Use "." as the key path delimiter. This can be "/" or any character.
var k = koanf.New(".")

func main() {
    // Load default values using the confmap provider.
    // We provide a flat map with the "." delimiter.
    // A nested map can be loaded by setting the delimiter to an empty string "".
    k.Load(confmap.Provider(map[string]interface{}{
        "parent1.name": "Default Name",
        "parent3.name": "New name here",
    }, "."), nil)

    // Load JSON config on top of the default values.
    if err := k.Load(file.Provider("mock/mock.json"), json.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    // Load YAML config and merge into the previously loaded config (because we can).
    k.Load(file.Provider("mock/mock.yml"), yaml.Parser())

    fmt.Println("parent's name is = ", k.String("parent1.name"))
    fmt.Println("parent's ID is = ", k.Int("parent1.id"))
}

Reading from struct

The bundled structs provider can be used to read data from a struct to load into a koanf instance.

package main

import (
    "fmt"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/providers/structs"
)

// Global koanf instance. Use "." as the key path delimiter. This can be "/" or any character.
var k = koanf.New(".")

type parentStruct struct {
    Name   string      `koanf:"name"`
    ID     int         `koanf:"id"`
    Child1 childStruct `koanf:"child1"`
}
type childStruct struct {
    Name        string            `koanf:"name"`
    Type        string            `koanf:"type"`
    Empty       map[string]string `koanf:"empty"`
    Grandchild1 grandchildStruct  `koanf:"grandchild1"`
}
type grandchildStruct struct {
    Ids []int `koanf:"ids"`
    On  bool  `koanf:"on"`
}
type sampleStruct struct {
    Type    string            `koanf:"type"`
    Empty   map[string]string `koanf:"empty"`
    Parent1 parentStruct      `koanf:"parent1"`
}

func main() {
    // Load default values using the structs provider.
    // We provide a struct along with the struct tag `koanf` to the
    // provider.
    k.Load(structs.Provider(sampleStruct{
        Type:  "json",
        Empty: make(map[string]string),
        Parent1: parentStruct{
            Name: "parent1",
            ID:   1234,
            Child1: childStruct{
                Name:  "child1",
                Type:  "json",
                Empty: make(map[string]string),
                Grandchild1: grandchildStruct{
                    Ids: []int{1, 2, 3},
                    On:  true,
                },
            },
        },
    }, "koanf"), nil)

    fmt.Printf("name is = `%s`\n", k.String("parent1.child1.name"))
}

Merge behavior

Default behavior

The default behavior when you create Koanf this way is: koanf.New(delim) that the latest loaded configuration will merge with the previous one.

For example: first.yml

key: [1,2,3]

second.yml

key: 'string'

When second.yml is loaded it will override the type of the first.yml.

If this behavior is not desired, you can merge 'strictly'. In the same scenario, Load will return an error.

package main

import (
    "errors"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/maps"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/parsers/yaml"
    "github.com/knadh/koanf/providers/file"
)

var conf = koanf.Conf{
    Delim:       ".",
    StrictMerge: true,
}
var k = koanf.NewWithConf(conf)

func main() {
    yamlPath := "mock/mock.yml"
    if err := k.Load(file.Provider(yamlPath), yaml.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    jsonPath := "mock/mock.json"
    if err := k.Load(file.Provider(jsonPath), json.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }
}

Note: When merging different extensions, each parser can treat his types differently, meaning even though you the load same types there is a probability that it will fail with StrictMerge: true.

For example: merging JSON and YAML will most likely fail because JSON treats integers as float64 and YAML treats them as integers.

Order of merge and key case sensitivity

Custom Providers and Parsers

A Provider returns a nested map[string]interface{} config that can be loaded directly into koanf with koanf.Load() or it can return raw bytes that can be parsed with a Parser (again, loaded using koanf.Load(). Writing Providers and Parsers are easy. See the bundled implementations in the providers and parsers directories.

Custom merge strategies

By default, when merging two config sources using Load(), koanf recursively merges keys of nested maps (map[string]interface{}), while static values are overwritten (slices, strings, etc). This behaviour can be changed by providing a custom merge function with the WithMergeFunc option.

package main

import (
    "errors"
    "log"

    "github.com/knadh/koanf/v2"
    "github.com/knadh/koanf/maps"
    "github.com/knadh/koanf/parsers/json"
    "github.com/knadh/koanf/parsers/yaml"
    "github.com/knadh/koanf/providers/file"
)

var conf = koanf.Conf{
    Delim:       ".",
    StrictMerge: true,
}
var k = koanf.NewWithConf(conf)

func main() {
    yamlPath := "mock/mock.yml"
    if err := k.Load(file.Provider(yamlPath), yaml.Parser()); err != nil {
        log.Fatalf("error loading config: %v", err)
    }

    jsonPath := "mock/mock.json"
    if err := k.Load(file.Provider(jsonPath), json.Parser(), koanf.WithMergeFunc(func(src, dest map[string]interface{}) error {
     // Your custom logic, copying values from src into dst
     return nil
    })); err != nil {
        log.Fatalf("error loading config: %v", err)
    }
}

API

See the full API documentation of all available methods at https://pkg.go.dev/github.com/knadh/koanf/v2#section-documentation

Bundled Providers

Install with go get -u github.com/knadh/koanf/providers/$provider

Package Provider Description
file file.Provider(filepath string) Reads a file and returns the raw bytes to be parsed.
fs fs.Provider(f fs.FS, filepath string) (Experimental) Reads a file from fs.FS and returns the raw bytes to be parsed. The provider requires go v1.16 or higher.
basicflag basicflag.Provider(f *flag.FlagSet, delim string) Takes an stdlib flag.FlagSet
posflag posflag.Provider(f *pflag.FlagSet, delim string) Takes an spf13/pflag.FlagSet (advanced POSIX compatible flags with multiple types) and provides a nested config map based on delim.
env env.Provider(prefix, delim string, f func(s string) string) Takes an optional prefix to filter env variables by, an optional function that takes and returns a string to transform env variables, and returns a nested config map based on delim.
confmap confmap.Provider(mp map[string]interface{}, delim string) Takes a premade map[string]interface{} conf map. If delim is provided, the keys are assumed to be flattened, thus unflattened using delim.
structs structs.Provider(s interface{}, tag string) Takes a struct and struct tag.
s3 s3.Provider(s3.S3Config{}) Takes a s3 config struct.
rawbytes rawbytes.Provider(b []byte) Takes a raw []byte slice to be parsed with a koanf.Parser
vault/v2 vault.Provider(vault.Config{}) Hashicorp Vault provider
appconfig/v2 vault.AppConfig(appconfig.Config{}) AWS AppConfig provider
etcd/v2 etcd.Provider(etcd.Config{}) CNCF etcd provider
consul/v2 consul.Provider(consul.Config{}) Hashicorp Consul provider
parameterstore/v2 parameterstore.Provider(parameterstore.Config{}) AWS Systems Manager Parameter Store provider

Bundled Parsers

Install with go get -u github.com/knadh/koanf/parsers/$parser

Package Parser Description
json json.Parser() Parses JSON bytes into a nested map
yaml yaml.Parser() Parses YAML bytes into a nested map
toml toml.Parser() Parses TOML bytes into a nested map
toml/v2 toml.Parser() Parses TOML bytes into a nested map (using go-toml v2)
dotenv dotenv.Parser() Parses DotEnv bytes into a flat map
hcl hcl.Parser(flattenSlices bool) Parses Hashicorp HCL bytes into a nested map. flattenSlices is recommended to be set to true. Read more.
nestedtext nestedtext.Parser() Parses NestedText bytes into a flat map
hjson hjson.Parser() Parses HJSON bytes into a nested map

Third-party Providers

Package Provider Description
github.com/defensestation/koanf/providers/secretsmanager vault.SecretsMananger(secretsmanager.Config{}, f func(s string) string) AWS Secrets Manager provider, takes map or string as a value from store
github.com/defensestation/koanf/providers/parameterstore vault.ParameterStore(parameterstore.Config{}, f func(s string) string) AWS ParameterStore provider, an optional function that takes and returns a string to transform env variables

Alternative to viper

koanf is a lightweight alternative to the popular spf13/viper. It was written as a result of multiple stumbling blocks encountered with some of viper's fundamental flaws.