mirror of
https://github.com/wneessen/logranger.git
synced 2024-10-06 00:55:57 +02:00
Winni Neessen
f02d5be680
The 'parser' struct in config.go has been updated with Fig configuration annotation. This allows ease of configuration as it sources the configuration options from the environment, making the code cleaner and more flexible.
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
// SPDX-FileCopyrightText: 2023 Winni Neessen <wn@neessen.dev>
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package logranger
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/kkyr/fig"
|
|
"github.com/wneessen/go-parsesyslog"
|
|
"github.com/wneessen/go-parsesyslog/rfc3164"
|
|
"github.com/wneessen/go-parsesyslog/rfc5424"
|
|
)
|
|
|
|
// Config holds all the global configuration settings that are parsed by fig
|
|
type Config struct {
|
|
// Server holds server specific configuration values
|
|
Server struct {
|
|
PIDFile string `fig:"pid_file" default:"/var/run/logranger.pid"`
|
|
}
|
|
Listener struct {
|
|
ListenerUnix struct {
|
|
Path string `fig:"path" default:"/var/tmp/logranger.sock"`
|
|
} `fig:"unix"`
|
|
ListenerTCP struct {
|
|
Addr string `fig:"addr" default:"0.0.0.0"`
|
|
Port uint `fig:"port" default:"9099"`
|
|
} `fig:"tcp"`
|
|
ListenerTLS struct {
|
|
Addr string `fig:"addr" default:"0.0.0.0"`
|
|
Port uint `fig:"port" default:"9099"`
|
|
CertPath string `fig:"cert_path"`
|
|
KeyPath string `fig:"key_path"`
|
|
} `fig:"tls"`
|
|
Type ListenerType `fig:"type" default:"unix"`
|
|
} `fig:"listener"`
|
|
Log struct {
|
|
Level string `fig:"level" default:"info"`
|
|
Extended bool `fig:"extended"`
|
|
} `fig:"log"`
|
|
Parser struct {
|
|
Type string `fig:"type" validate:"required"`
|
|
Timeout time.Duration `fig:"timeout" default:"500ms"`
|
|
} `fig:"parser"`
|
|
internal struct {
|
|
ParserType parsesyslog.ParserType
|
|
}
|
|
}
|
|
|
|
// NewConfig creates a new instance of the Config object by reading and loading
|
|
// configuration values. It takes in the file path and file name of the configuration
|
|
// file as parameters. It returns a pointer to the Config object and an error if
|
|
// there was a problem reading or loading the configuration.
|
|
func NewConfig(p, f string) (*Config, error) {
|
|
co := Config{}
|
|
_, err := os.Stat(fmt.Sprintf("%s/%s", p, f))
|
|
if err != nil {
|
|
return &co, fmt.Errorf("failed to read config: %w", err)
|
|
}
|
|
|
|
if err := fig.Load(&co, fig.Dirs(p), fig.File(f), fig.UseEnv("logranger")); err != nil {
|
|
return &co, fmt.Errorf("failed to load config: %w", err)
|
|
}
|
|
|
|
switch {
|
|
case strings.EqualFold(co.Parser.Type, "rfc3164"):
|
|
co.internal.ParserType = rfc3164.Type
|
|
case strings.EqualFold(co.Parser.Type, "rfc5424"):
|
|
co.internal.ParserType = rfc5424.Type
|
|
default:
|
|
return nil, fmt.Errorf("unknown parser type: %s", co.Parser.Type)
|
|
}
|
|
|
|
return &co, nil
|
|
}
|