mirror of
https://github.com/grafana/grafana.git
synced 2025-08-03 06:22:13 +08:00
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package featuretoggle
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
|
|
"k8s.io/apimachinery/pkg/apis/meta/internalversion"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apiserver/pkg/registry/rest"
|
|
|
|
common "github.com/grafana/grafana/pkg/apimachinery/apis/common/v0alpha1"
|
|
"github.com/grafana/grafana/pkg/apis/featuretoggle/v0alpha1"
|
|
"github.com/grafana/grafana/pkg/services/apiserver/utils"
|
|
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
|
)
|
|
|
|
var (
|
|
_ rest.Storage = (*featuresStorage)(nil)
|
|
_ rest.Scoper = (*featuresStorage)(nil)
|
|
_ rest.SingularNameProvider = (*featuresStorage)(nil)
|
|
_ rest.Lister = (*featuresStorage)(nil)
|
|
_ rest.Getter = (*featuresStorage)(nil)
|
|
)
|
|
|
|
type featuresStorage struct {
|
|
resource *common.ResourceInfo
|
|
tableConverter rest.TableConvertor
|
|
features *v0alpha1.FeatureList
|
|
featuresOnce sync.Once
|
|
}
|
|
|
|
// NOTE! this does not depend on config or any system state!
|
|
// In the future, the existence of features (and their properties) can be defined dynamically
|
|
func NewFeaturesStorage() *featuresStorage {
|
|
resourceInfo := v0alpha1.FeatureResourceInfo
|
|
return &featuresStorage{
|
|
resource: &resourceInfo,
|
|
tableConverter: utils.NewTableConverter(
|
|
resourceInfo.GroupResource(),
|
|
[]metav1.TableColumnDefinition{
|
|
{Name: "Name", Type: "string", Format: "name"},
|
|
{Name: "Stage", Type: "string", Format: "string", Description: "Where is the flag in the dev cycle"},
|
|
{Name: "Owner", Type: "string", Format: "string", Description: "Which team owns the feature"},
|
|
},
|
|
func(obj any) ([]interface{}, error) {
|
|
r, ok := obj.(*v0alpha1.Feature)
|
|
if ok {
|
|
return []interface{}{
|
|
r.Name,
|
|
r.Spec.Stage,
|
|
r.Spec.Owner,
|
|
}, nil
|
|
}
|
|
return nil, fmt.Errorf("expected resource or info")
|
|
}),
|
|
}
|
|
}
|
|
|
|
func (s *featuresStorage) New() runtime.Object {
|
|
return s.resource.NewFunc()
|
|
}
|
|
|
|
func (s *featuresStorage) Destroy() {}
|
|
|
|
func (s *featuresStorage) NamespaceScoped() bool {
|
|
return false
|
|
}
|
|
|
|
func (s *featuresStorage) GetSingularName() string {
|
|
return s.resource.GetSingularName()
|
|
}
|
|
|
|
func (s *featuresStorage) NewList() runtime.Object {
|
|
return s.resource.NewListFunc()
|
|
}
|
|
|
|
func (s *featuresStorage) ConvertToTable(ctx context.Context, object runtime.Object, tableOptions runtime.Object) (*metav1.Table, error) {
|
|
return s.tableConverter.ConvertToTable(ctx, object, tableOptions)
|
|
}
|
|
|
|
func (s *featuresStorage) init() {
|
|
s.featuresOnce.Do(func() {
|
|
rv := "1"
|
|
features, _ := featuremgmt.GetEmbeddedFeatureList()
|
|
for _, feature := range features.Items {
|
|
if feature.ResourceVersion > rv {
|
|
rv = feature.ResourceVersion
|
|
}
|
|
}
|
|
features.ResourceVersion = rv
|
|
s.features = &features
|
|
})
|
|
}
|
|
|
|
func (s *featuresStorage) List(ctx context.Context, options *internalversion.ListOptions) (runtime.Object, error) {
|
|
s.init()
|
|
if s.features == nil {
|
|
return nil, fmt.Errorf("error loading embedded features")
|
|
}
|
|
return s.features, nil
|
|
}
|
|
|
|
func (s *featuresStorage) Get(ctx context.Context, name string, options *metav1.GetOptions) (runtime.Object, error) {
|
|
s.init()
|
|
|
|
for idx, flag := range s.features.Items {
|
|
if flag.Name == name {
|
|
return &s.features.Items[idx], nil
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("not found")
|
|
}
|