-
Notifications
You must be signed in to change notification settings - Fork 926
/
feature_flag.go
63 lines (49 loc) · 1.9 KB
/
feature_flag.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package v7action
import (
"code.cloudfoundry.org/cli/actor/actionerror"
"code.cloudfoundry.org/cli/api/cloudcontroller/ccerror"
"code.cloudfoundry.org/cli/api/cloudcontroller/ccv3"
)
type FeatureFlag ccv3.FeatureFlag
// GetFeatureFlagByName returns a featureFlag with the provided name.
func (actor Actor) GetFeatureFlagByName(featureFlagName string) (FeatureFlag, Warnings, error) {
var (
ccv3FeatureFlag ccv3.FeatureFlag
warnings ccv3.Warnings
err error
)
ccv3FeatureFlag, warnings, err = actor.CloudControllerClient.GetFeatureFlag(featureFlagName)
if err != nil {
if _, ok := err.(ccerror.FeatureFlagNotFoundError); ok {
return FeatureFlag{}, Warnings(warnings), actionerror.FeatureFlagNotFoundError{FeatureFlagName: featureFlagName}
}
return FeatureFlag{}, Warnings(warnings), err
}
return FeatureFlag(ccv3FeatureFlag), Warnings(warnings), err
}
func (actor Actor) GetFeatureFlags() ([]FeatureFlag, Warnings, error) {
var (
featureFlags []FeatureFlag
)
ccv3FeatureFlags, warnings, err := actor.CloudControllerClient.GetFeatureFlags()
if err != nil {
return nil, Warnings(warnings), err
}
for _, flag := range ccv3FeatureFlags {
featureFlags = append(featureFlags, FeatureFlag(flag))
}
return featureFlags, Warnings(warnings), nil
}
func (actor Actor) EnableFeatureFlag(flagName string) (Warnings, error) {
return actor.updateFeatureFlag(FeatureFlag{Name: flagName, Enabled: true})
}
func (actor Actor) DisableFeatureFlag(flagName string) (Warnings, error) {
return actor.updateFeatureFlag(FeatureFlag{Name: flagName, Enabled: false})
}
func (actor Actor) updateFeatureFlag(flag FeatureFlag) (Warnings, error) {
_, warnings, err := actor.CloudControllerClient.UpdateFeatureFlag(ccv3.FeatureFlag(flag))
if _, ok := err.(ccerror.FeatureFlagNotFoundError); ok {
err = actionerror.FeatureFlagNotFoundError{FeatureFlagName: flag.Name}
}
return Warnings(warnings), err
}