...
1
16
17 package rest
18
19 import (
20 nodev1 "k8s.io/api/node/v1"
21 "k8s.io/apiserver/pkg/registry/generic"
22 "k8s.io/apiserver/pkg/registry/rest"
23 genericapiserver "k8s.io/apiserver/pkg/server"
24 serverstorage "k8s.io/apiserver/pkg/server/storage"
25 "k8s.io/kubernetes/pkg/api/legacyscheme"
26 nodeinternal "k8s.io/kubernetes/pkg/apis/node"
27 runtimeclassstorage "k8s.io/kubernetes/pkg/registry/node/runtimeclass/storage"
28 )
29
30
31 type RESTStorageProvider struct{}
32
33
34 func (p RESTStorageProvider) NewRESTStorage(apiResourceConfigSource serverstorage.APIResourceConfigSource, restOptionsGetter generic.RESTOptionsGetter) (genericapiserver.APIGroupInfo, error) {
35 apiGroupInfo := genericapiserver.NewDefaultAPIGroupInfo(nodeinternal.GroupName, legacyscheme.Scheme, legacyscheme.ParameterCodec, legacyscheme.Codecs)
36
37 if storageMap, err := p.v1Storage(apiResourceConfigSource, restOptionsGetter); err != nil {
38 return genericapiserver.APIGroupInfo{}, err
39 } else if len(storageMap) > 0 {
40 apiGroupInfo.VersionedResourcesStorageMap[nodev1.SchemeGroupVersion.Version] = storageMap
41 }
42
43 return apiGroupInfo, nil
44 }
45
46 func (p RESTStorageProvider) v1Storage(apiResourceConfigSource serverstorage.APIResourceConfigSource, restOptionsGetter generic.RESTOptionsGetter) (map[string]rest.Storage, error) {
47 storage := map[string]rest.Storage{}
48
49 if resource := "runtimeclasses"; apiResourceConfigSource.ResourceEnabled(nodev1.SchemeGroupVersion.WithResource(resource)) {
50 s, err := runtimeclassstorage.NewREST(restOptionsGetter)
51 if err != nil {
52 return storage, err
53 }
54 storage[resource] = s
55 }
56
57 return storage, nil
58 }
59
60
61 func (p RESTStorageProvider) GroupName() string {
62 return nodeinternal.GroupName
63 }
64
View as plain text