61 lines
1.9 KiB
Go
61 lines
1.9 KiB
Go
package tflog
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/terraform-plugin-log/internal/logging"
|
|
)
|
|
|
|
// Options are a collection of logging options, useful for collecting arguments
|
|
// to NewSubsystem prior to calling it.
|
|
type Options []logging.Option
|
|
|
|
// WithAdditionalLocationOffset returns an option that allowing implementations
|
|
// to fix location information when implementing helper functions. The default
|
|
// offset of 1 is automatically added to the provided value to account for the
|
|
// tflog logging functions.
|
|
func WithAdditionalLocationOffset(additionalLocationOffset int) logging.Option {
|
|
return logging.WithAdditionalLocationOffset(additionalLocationOffset)
|
|
}
|
|
|
|
// WithLevelFromEnv returns an option that will set the level of the logger
|
|
// based on the string in an environment variable. The environment variable
|
|
// checked will be `name` and `subsystems`, joined by _ and in all caps.
|
|
func WithLevelFromEnv(name string, subsystems ...string) logging.Option {
|
|
return func(l logging.LoggerOpts) logging.LoggerOpts {
|
|
envVar := strings.Join(subsystems, "_")
|
|
if envVar != "" {
|
|
envVar = "_" + envVar
|
|
}
|
|
envVar = strings.ToUpper(name + envVar)
|
|
l.Level = hclog.LevelFromString(os.Getenv(envVar))
|
|
return l
|
|
}
|
|
}
|
|
|
|
// WithLevel returns an option that will set the level of the logger.
|
|
func WithLevel(level hclog.Level) logging.Option {
|
|
return func(l logging.LoggerOpts) logging.LoggerOpts {
|
|
l.Level = level
|
|
return l
|
|
}
|
|
}
|
|
|
|
// WithRootFields enables the copying of root logger fields to a new subsystem
|
|
// logger during creation.
|
|
func WithRootFields() logging.Option {
|
|
return logging.WithRootFields()
|
|
}
|
|
|
|
// WithoutLocation returns an option that disables including the location of
|
|
// the log line in the log output, which is on by default. This has no effect
|
|
// when used with NewSubsystem.
|
|
func WithoutLocation() logging.Option {
|
|
return func(l logging.LoggerOpts) logging.LoggerOpts {
|
|
l.IncludeLocation = false
|
|
return l
|
|
}
|
|
}
|