go/task
1
0
Fork 0
mirror of https://github.com/go-task/task.git synced 2025-11-25 22:32:55 +02:00
Code Issues Releases Activity

feat: support multiple experiment values

Browse Source
This commit is contained in:
Pete Davison
2023-12-23 02:33:12 +00:00
parent f6a24fe925
commit dfe39bfb5d
4 changed files with 44 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
internal/experiments/experiments.go
View File

@@ -8,29 +8,55 @@ import (
"text/tabwriter"
"github.com/joho/godotenv"
"golang.org/x/exp/slices"
"github.com/go-task/task/v3/internal/logger"
)
const envPrefix = "TASK_X_"
type Experiment struct {
Name string
Enabled bool
Value string
}
// A list of experiments.
var (
GentleForce bool
RemoteTaskfiles bool
AnyVariables bool
GentleForce Experiment
RemoteTaskfiles Experiment
AnyVariables Experiment
)
func init() {
readDotEnv()
GentleForce = parseEnv("GENTLE_FORCE")
RemoteTaskfiles = parseEnv("REMOTE_TASKFILES")
AnyVariables = parseEnv("ANY_VARIABLES")
GentleForce = New("GENTLE_FORCE")
RemoteTaskfiles = New("REMOTE_TASKFILES")
AnyVariables = New("ANY_VARIABLES")
}
func parseEnv(xName string) bool {
func New(xName string, enabledValues ...string) Experiment {
if len(enabledValues) == 0 {
enabledValues = []string{"1"}
}
value := getEnv(xName)
return Experiment{
Name: xName,
Enabled: slices.Contains(enabledValues, value),
Value: value,
}
}
func (x Experiment) String() string {
if x.Enabled {
return fmt.Sprintf("on (%s)", x.Value)
}
return "off"
}
func getEnv(xName string) string {
envName := fmt.Sprintf("%s%s", envPrefix, xName)
return os.Getenv(envName) == "1"
return os.Getenv(envName)
}
func readDotEnv() {
@@ -43,16 +69,16 @@ func readDotEnv() {
}
}
func printExperiment(w io.Writer, l *logger.Logger, name string, value bool) {
func printExperiment(w io.Writer, l *logger.Logger, x Experiment) {
l.FOutf(w, logger.Yellow, "* ")
l.FOutf(w, logger.Green, name)
l.FOutf(w, logger.Default, ": \t%t\n", value)
l.FOutf(w, logger.Green, x.Name)
l.FOutf(w, logger.Default, ": \t%s\n", x.String())
}
func List(l *logger.Logger) error {
w := tabwriter.NewWriter(os.Stdout, 0, 8, 0, ' ', 0)
printExperiment(w, l, "GENTLE_FORCE", GentleForce)
printExperiment(w, l, "REMOTE_TASKFILES", RemoteTaskfiles)
printExperiment(w, l, "ANY_VARIABLES", AnyVariables)
printExperiment(w, l, GentleForce)
printExperiment(w, l, RemoteTaskfiles)
printExperiment(w, l, AnyVariables)
return w.Flush()
}