Ryan McKinley
119 lines
3.5 KiB
Go
119 lines
3.5 KiB
Go
package example
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"slices"
|
|
|
|
"k8s.io/apimachinery/pkg/apis/meta/internalversion"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
genericregistry "k8s.io/apiserver/pkg/registry/generic/registry"
|
|
"k8s.io/apiserver/pkg/registry/rest"
|
|
|
|
example "github.com/grafana/grafana/pkg/apis/example/v0alpha1"
|
|
grafanarequest "github.com/grafana/grafana/pkg/services/grafana-apiserver/endpoints/request"
|
|
grafanaregistry "github.com/grafana/grafana/pkg/services/grafana-apiserver/registry/generic"
|
|
)
|
|
|
|
var (
|
|
_ rest.Storage = (*dummyStorage)(nil)
|
|
_ rest.Scoper = (*dummyStorage)(nil)
|
|
_ rest.SingularNameProvider = (*dummyStorage)(nil)
|
|
_ rest.Getter = (*dummyStorage)(nil)
|
|
_ rest.Lister = (*dummyStorage)(nil)
|
|
)
|
|
|
|
type dummyStorage struct {
|
|
store *genericregistry.Store
|
|
names []string
|
|
creationTimestamp metav1.Time
|
|
}
|
|
|
|
func newDummyStorage(gv schema.GroupVersion, scheme *runtime.Scheme, names ...string) *dummyStorage {
|
|
strategy := grafanaregistry.NewStrategy(scheme)
|
|
store := &genericregistry.Store{
|
|
NewFunc: func() runtime.Object { return &example.DummyResource{} }, // getter not supported
|
|
NewListFunc: func() runtime.Object { return &example.DummyResourceList{} }, // both list and get return the same thing
|
|
PredicateFunc: grafanaregistry.Matcher,
|
|
DefaultQualifiedResource: gv.WithResource("dummy").GroupResource(),
|
|
SingularQualifiedResource: gv.WithResource("dummy").GroupResource(),
|
|
CreateStrategy: strategy,
|
|
UpdateStrategy: strategy,
|
|
DeleteStrategy: strategy,
|
|
}
|
|
store.TableConvertor = rest.NewDefaultTableConvertor(store.DefaultQualifiedResource)
|
|
|
|
return &dummyStorage{
|
|
store: store,
|
|
names: names,
|
|
creationTimestamp: metav1.Now(),
|
|
}
|
|
}
|
|
|
|
func (s *dummyStorage) New() runtime.Object {
|
|
return s.store.New()
|
|
}
|
|
|
|
func (s *dummyStorage) Destroy() {}
|
|
|
|
func (s *dummyStorage) NamespaceScoped() bool {
|
|
return true
|
|
}
|
|
|
|
func (s *dummyStorage) GetSingularName() string {
|
|
return "dummy"
|
|
}
|
|
|
|
func (s *dummyStorage) NewList() runtime.Object {
|
|
return s.store.NewListFunc()
|
|
}
|
|
|
|
func (s *dummyStorage) ConvertToTable(ctx context.Context, object runtime.Object, tableOptions runtime.Object) (*metav1.Table, error) {
|
|
return s.store.TableConvertor.ConvertToTable(ctx, object, tableOptions)
|
|
}
|
|
|
|
func (s *dummyStorage) Get(ctx context.Context, name string, options *metav1.GetOptions) (runtime.Object, error) {
|
|
info, err := grafanarequest.NamespaceInfoFrom(ctx, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
idx := slices.Index(s.names, name)
|
|
if idx < 0 {
|
|
return nil, fmt.Errorf("dummy not found")
|
|
}
|
|
|
|
return &example.DummyResource{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: name,
|
|
Namespace: info.Value,
|
|
CreationTimestamp: s.creationTimestamp,
|
|
ResourceVersion: "1",
|
|
},
|
|
Spec: fmt.Sprintf("dummy: %s", name),
|
|
}, nil
|
|
}
|
|
|
|
func (s *dummyStorage) List(ctx context.Context, options *internalversion.ListOptions) (runtime.Object, error) {
|
|
info, err := grafanarequest.NamespaceInfoFrom(ctx, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
res := &example.DummyResourceList{}
|
|
for _, name := range s.names {
|
|
res.Items = append(res.Items, example.DummyResource{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: name,
|
|
Namespace: info.Value,
|
|
CreationTimestamp: s.creationTimestamp,
|
|
ResourceVersion: "1",
|
|
},
|
|
Spec: fmt.Sprintf("dummy: %s", name),
|
|
})
|
|
}
|
|
return res, nil
|
|
}
|