mirror of https://github.com/databricks/cli.git
72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package root
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/databricks/bricks/libs/flags"
|
|
"github.com/databricks/bricks/libs/log"
|
|
"golang.org/x/exp/slog"
|
|
)
|
|
|
|
const (
|
|
envBricksLogFile = "BRICKS_LOG_FILE"
|
|
envBricksLogLevel = "BRICKS_LOG_LEVEL"
|
|
envBricksLogFormat = "BRICKS_LOG_FORMAT"
|
|
)
|
|
|
|
func initializeLogger(ctx context.Context) (context.Context, error) {
|
|
opts := slog.HandlerOptions{}
|
|
opts.Level = logLevel.Level()
|
|
opts.AddSource = true
|
|
opts.ReplaceAttr = log.ReplaceAttrFunctions{
|
|
log.ReplaceLevelAttr,
|
|
log.ReplaceSourceAttr,
|
|
}.ReplaceAttr
|
|
|
|
// Open the underlying log file if the user configured an actual file to log to.
|
|
err := logFile.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var handler slog.Handler
|
|
switch logOutput {
|
|
case flags.OutputJSON:
|
|
handler = opts.NewJSONHandler(logFile.Writer())
|
|
case flags.OutputText:
|
|
handler = opts.NewTextHandler(logFile.Writer())
|
|
default:
|
|
return nil, fmt.Errorf("invalid log output: %s", logOutput)
|
|
}
|
|
|
|
slog.SetDefault(slog.New(handler))
|
|
return log.NewContext(ctx, slog.Default()), nil
|
|
}
|
|
|
|
var logFile = flags.NewLogFileFlag()
|
|
var logLevel = flags.NewLogLevelFlag()
|
|
var logOutput = flags.OutputText
|
|
|
|
func init() {
|
|
// Configure defaults from environment, if applicable.
|
|
// If the provided value is invalid it is ignored.
|
|
if v, ok := os.LookupEnv(envBricksLogFile); ok {
|
|
logFile.Set(v)
|
|
}
|
|
if v, ok := os.LookupEnv(envBricksLogLevel); ok {
|
|
logLevel.Set(v)
|
|
}
|
|
if v, ok := os.LookupEnv(envBricksLogFormat); ok {
|
|
logOutput.Set(v)
|
|
}
|
|
|
|
RootCmd.PersistentFlags().Var(&logFile, "log-file", "file to write logs to")
|
|
RootCmd.PersistentFlags().Var(&logLevel, "log-level", "log level")
|
|
RootCmd.PersistentFlags().Var(&logOutput, "log-format", "log output format (text or json)")
|
|
RootCmd.RegisterFlagCompletionFunc("log-file", logFile.Complete)
|
|
RootCmd.RegisterFlagCompletionFunc("log-level", logLevel.Complete)
|
|
RootCmd.RegisterFlagCompletionFunc("log-format", logOutput.Complete)
|
|
}
|