Support for experiment env vars (#1214)

* feat: experiments env var handling

* feat: read experiments from dotenv files

* docs: removed experimental cli flags

* feat: use vars instead of struct with reflection
This commit is contained in:
Pete Davison
2023-06-18 02:04:05 +01:00
committed by GitHub
parent c79378f380
commit d8a12fe56d
2 changed files with 58 additions and 9 deletions

View File

@@ -0,0 +1,43 @@
package experiments
import (
"errors"
"fmt"
"os"
"strings"
"github.com/joho/godotenv"
)
const envPrefix = "TASK_X_"
var TestExperiment bool
func init() {
if err := readDotEnv(); err != nil {
panic(err)
}
TestExperiment = parseEnv("TestExperiment")
}
func parseEnv(xName string) bool {
envName := fmt.Sprintf("%s%s", envPrefix, xName)
return os.Getenv(envName) == "1"
}
func readDotEnv() error {
env, err := godotenv.Read()
if errors.Is(err, os.ErrNotExist) {
return nil
}
if err != nil {
return err
}
// If the env var is an experiment, set it.
for key, value := range env {
if strings.HasPrefix(key, envPrefix) {
os.Setenv(key, value)
}
}
return nil
}