feat(downloadLimit): release download limit feature - close #2

benjohns1 · Feb 20, 2024 · 4462534 · 4462534
1 parent f5d8e5c
commit 4462534
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/cmd/main.go b/cmd/main.go
@@ -50,7 +50,10 @@ func run(ctx context.Context) (err error) {
 	l := log.New(log.Config{GetRequestID: request.GetID})
 	l.Printf(ctx, "Running build %q", build)
 
-	ff, err := featureflag.New(featureflag.WithFeaturesFromEnvironment("FEATURE_FLAG_"))
+	ff, err := featureflag.New(
+		featureflag.WithFeaturesFromEnvironment("FEATURE_FLAG_"),
+		featureflag.WithReleasedFeatures("DownloadLimit"),
+	)
 	if err != nil {
 		return err
 	}

diff --git a/featureflag/featureflag.go b/featureflag/featureflag.go
@@ -60,6 +60,15 @@ func WithFeaturesFromEnvironment(prefix string) ConfigOption {
 	}
 }
 
+func WithReleasedFeatures(features ...string) ConfigOption {
+	return func(ff *FeatureFlags) error {
+		for _, feature := range features {
+			ff.features[Name(feature)] = true
+		}
+		return nil
+	}
+}
+
 func New(opts ...ConfigOption) (*FeatureFlags, error) {
 	ff := FeatureFlags{
 		features: make(map[Name]bool),