8ab643dc10
Co-authored-by: marbar3778 <marbar3778@yahoo.com> Co-authored-by: Marko <marko@baricevic.me> Co-authored-by: Matt Kocubinski <mkocubinski@gmail.com> Co-authored-by: Likhita Polavarapu <78951027+likhita-809@users.noreply.github.com> Co-authored-by: unknown unknown <unknown@unknown> Co-authored-by: Julien Robert <julien@rbrt.fr>
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package serverv2
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
corectx "cosmossdk.io/core/context"
|
|
"cosmossdk.io/log"
|
|
)
|
|
|
|
// SetCmdServerContext sets a command's Context value to the provided argument.
|
|
// If the context has not been set, set the given context as the default.
|
|
func SetCmdServerContext(cmd *cobra.Command, viper *viper.Viper, logger log.Logger) error {
|
|
var cmdCtx context.Context
|
|
if cmd.Context() == nil {
|
|
cmdCtx = context.Background()
|
|
} else {
|
|
cmdCtx = cmd.Context()
|
|
}
|
|
|
|
cmd.SetContext(context.WithValue(cmdCtx, corectx.LoggerContextKey{}, logger))
|
|
cmd.SetContext(context.WithValue(cmdCtx, corectx.ViperContextKey{}, viper))
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetViperFromCmd(cmd *cobra.Command) *viper.Viper {
|
|
value := cmd.Context().Value(corectx.ViperContextKey{})
|
|
v, ok := value.(*viper.Viper)
|
|
if !ok {
|
|
return viper.New()
|
|
}
|
|
return v
|
|
}
|
|
|
|
func GetLoggerFromCmd(cmd *cobra.Command) log.Logger {
|
|
v := cmd.Context().Value(corectx.LoggerContextKey{})
|
|
logger, ok := v.(log.Logger)
|
|
if !ok {
|
|
return log.NewLogger(os.Stdout)
|
|
}
|
|
return logger
|
|
}
|