add yaml config file support

README.md

@@ -22,7 +22,8 @@ or download the latest binary from [releases page](https://github.com/cenkalti/d
 Usage
 -----
 
-See [example config file](https://github.com/cenkalti/dalga/blob/v3/dalga.toml) for configuration options.
+See [example config file](https://github.com/cenkalti/dalga/blob/v3/config.toml) for configuration options.
+TOML and YAML file formats are supported.
 Configuration values can also be set via environment variables with `DALGA_` prefix.
 
 First, you must create the table for storing jobs:

cmd/dalga/main.go

@@ -6,13 +6,15 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
+	"path/filepath"
 	"strings"
 	"syscall"
 
 	"github.com/cenkalti/dalga/v3"
 	"github.com/cenkalti/dalga/v3/internal/log"
 	"github.com/knadh/koanf"
 	"github.com/knadh/koanf/parsers/toml"
+	"github.com/knadh/koanf/parsers/yaml"
 	"github.com/knadh/koanf/providers/env"
 	"github.com/knadh/koanf/providers/file"
 )
@@ -84,7 +86,14 @@ func main() {
 func readConfig() (c dalga.Config, err error) {
 	c = dalga.DefaultConfig
 	k := koanf.New(".")
-	err = k.Load(file.Provider(*configFlag), toml.Parser())
+	var parser koanf.Parser
+	ext := filepath.Ext(*configFlag)
+	if ext == ".yaml" || ext == ".yml" {
+		parser = yaml.Parser()
+	} else {
+		parser = toml.Parser()
+	}
+	err = k.Load(file.Provider(*configFlag), parser)
 	if err != nil {
 		return
 	}