1
0
Fork 0
mirror of https://github.com/eosswedenorg/thalos synced 2026-07-03 11:53:41 +02:00

nternal/config/cli.go: move overrideCliFlags() to cmd/thalos/server.go as its pretty specific code.

This commit is contained in:
Henrik Hautakoski 2024-02-18 18:00:26 +01:00
parent 53baae8a7f
commit 7f1f186aa0
4 changed files with 15 additions and 24 deletions

View file

@ -7,6 +7,7 @@ import (
"io" "io"
"os" "os"
"os/signal" "os/signal"
"path"
"syscall" "syscall"
"time" "time"
@ -204,10 +205,21 @@ func GetConfig(flags *pflag.FlagSet) (*config.Config, error) {
return nil, err return nil, err
} }
return config.NewBuilder(). cfg, err := config.NewBuilder().
SetConfigFile(filename). SetConfigFile(filename).
SetFlags(flags). SetFlags(flags).
Build() Build()
if err != nil {
return nil, err
}
logFile, _ := flags.GetString("log")
if len(logFile) > 0 {
cfg.Log.Directory = path.Dir(logFile)
cfg.Log.Filename = path.Base(logFile)
}
return cfg, nil
} }
func serverCmd(cmd *cobra.Command, args []string) { func serverCmd(cmd *cobra.Command, args []string) {

View file

@ -116,12 +116,5 @@ func (b *Builder) Build() (*Config, error) {
return nil, err return nil, err
} }
// Call custom handler.
if b.flags != nil {
if err := overrideCliFlags(&conf, b.flags); err != nil {
return nil, err
}
}
return &conf, nil return &conf, nil
} }

View file

@ -110,8 +110,8 @@ name: "ship-reader-1"
api: "http://127.0.0.1:8080" api: "http://127.0.0.1:8080"
message_codec: "mojibake" message_codec: "mojibake"
log: log:
filename: some_file.log filename: mylog.log
directory: /path/to/whatever directory: /var/log
maxtime: 30m maxtime: 30m
maxfilesize: 200b maxfilesize: 200b
ship: ship:
@ -135,7 +135,6 @@ redis:
require.NoError(t, flags.Set("url", "https://api.example.com")) require.NoError(t, flags.Set("url", "https://api.example.com"))
require.NoError(t, flags.Set("codec", "msgpack")) require.NoError(t, flags.Set("codec", "msgpack"))
require.NoError(t, flags.Set("log", "/var/log/mylog.log"))
require.NoError(t, flags.Set("redis-user", "userfromcli")) require.NoError(t, flags.Set("redis-user", "userfromcli"))
builder.SetFlags(flags) builder.SetFlags(flags)
@ -164,7 +163,6 @@ func TestBuilder_Flags(t *testing.T) {
require.NoError(t, flags.Set("redis-prefix", "custom-prefix")) require.NoError(t, flags.Set("redis-prefix", "custom-prefix"))
require.NoError(t, flags.Set("telegram-id", "72983126312982618")) require.NoError(t, flags.Set("telegram-id", "72983126312982618"))
require.NoError(t, flags.Set("telegram-channel", "-293492332")) require.NoError(t, flags.Set("telegram-channel", "-293492332"))
require.NoError(t, flags.Set("log", "/path/to/logs/mylog"))
require.NoError(t, flags.Set("log-max-filesize", "25mb")) require.NoError(t, flags.Set("log-max-filesize", "25mb"))
require.NoError(t, flags.Set("log-max-time", "10m")) require.NoError(t, flags.Set("log-max-time", "10m"))
require.NoError(t, flags.Set("ship-url", "ws://myship.com:7823")) require.NoError(t, flags.Set("ship-url", "ws://myship.com:7823"))
@ -183,8 +181,6 @@ func TestBuilder_Flags(t *testing.T) {
Api: "https://myapi", Api: "https://myapi",
MessageCodec: "binary", MessageCodec: "binary",
Log: log.Config{ Log: log.Config{
Filename: "mylog",
Directory: "/path/to/logs",
MaxFileSize: 25 * 1000 * 1000, // 25 mb MaxFileSize: 25 * 1000 * 1000, // 25 mb
MaxTime: time.Minute * 10, MaxTime: time.Minute * 10,
}, },

View file

@ -1,7 +1,6 @@
package config package config
import ( import (
"path"
"time" "time"
shipclient "github.com/eosswedenorg-go/antelope-ship-client" shipclient "github.com/eosswedenorg-go/antelope-ship-client"
@ -45,12 +44,3 @@ func GetFlags() *pflag.FlagSet {
return &flags return &flags
} }
func overrideCliFlags(cfg *Config, flags *pflag.FlagSet) error {
logFile, _ := flags.GetString("log")
if len(logFile) > 0 {
cfg.Log.Directory = path.Dir(logFile)
cfg.Log.Filename = path.Base(logFile)
}
return nil
}