mirror of
https://github.com/eosswedenorg/thalos
synced 2026-06-17 04:30:03 +02:00
nternal/config/cli.go: move overrideCliFlags() to cmd/thalos/server.go as its pretty specific code.
This commit is contained in:
parent
53baae8a7f
commit
7f1f186aa0
4 changed files with 15 additions and 24 deletions
|
|
@ -7,6 +7,7 @@ import (
|
|||
"io"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
|
|
@ -204,10 +205,21 @@ func GetConfig(flags *pflag.FlagSet) (*config.Config, error) {
|
|||
return nil, err
|
||||
}
|
||||
|
||||
return config.NewBuilder().
|
||||
cfg, err := config.NewBuilder().
|
||||
SetConfigFile(filename).
|
||||
SetFlags(flags).
|
||||
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) {
|
||||
|
|
|
|||
|
|
@ -116,12 +116,5 @@ func (b *Builder) Build() (*Config, error) {
|
|||
return nil, err
|
||||
}
|
||||
|
||||
// Call custom handler.
|
||||
if b.flags != nil {
|
||||
if err := overrideCliFlags(&conf, b.flags); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return &conf, nil
|
||||
}
|
||||
|
|
|
|||
|
|
@ -110,8 +110,8 @@ name: "ship-reader-1"
|
|||
api: "http://127.0.0.1:8080"
|
||||
message_codec: "mojibake"
|
||||
log:
|
||||
filename: some_file.log
|
||||
directory: /path/to/whatever
|
||||
filename: mylog.log
|
||||
directory: /var/log
|
||||
maxtime: 30m
|
||||
maxfilesize: 200b
|
||||
ship:
|
||||
|
|
@ -135,7 +135,6 @@ redis:
|
|||
|
||||
require.NoError(t, flags.Set("url", "https://api.example.com"))
|
||||
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"))
|
||||
|
||||
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("telegram-id", "72983126312982618"))
|
||||
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-time", "10m"))
|
||||
require.NoError(t, flags.Set("ship-url", "ws://myship.com:7823"))
|
||||
|
|
@ -183,8 +181,6 @@ func TestBuilder_Flags(t *testing.T) {
|
|||
Api: "https://myapi",
|
||||
MessageCodec: "binary",
|
||||
Log: log.Config{
|
||||
Filename: "mylog",
|
||||
Directory: "/path/to/logs",
|
||||
MaxFileSize: 25 * 1000 * 1000, // 25 mb
|
||||
MaxTime: time.Minute * 10,
|
||||
},
|
||||
|
|
|
|||
|
|
@ -1,7 +1,6 @@
|
|||
package config
|
||||
|
||||
import (
|
||||
"path"
|
||||
"time"
|
||||
|
||||
shipclient "github.com/eosswedenorg-go/antelope-ship-client"
|
||||
|
|
@ -45,12 +44,3 @@ func GetFlags() *pflag.FlagSet {
|
|||
|
||||
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
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue