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:
parent
53baae8a7f
commit
7f1f186aa0
4 changed files with 15 additions and 24 deletions
|
|
@ -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) {
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -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,
|
||||||
},
|
},
|
||||||
|
|
|
||||||
|
|
@ -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
|
|
||||||
}
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue