...
1
16
17 package rest
18
19 import (
20 coordinationv1 "k8s.io/api/coordination/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 "k8s.io/kubernetes/pkg/apis/coordination"
27 leasestorage "k8s.io/kubernetes/pkg/registry/coordination/lease/storage"
28 )
29
30 type RESTStorageProvider struct{}
31
32 func (p RESTStorageProvider) NewRESTStorage(apiResourceConfigSource serverstorage.APIResourceConfigSource, restOptionsGetter generic.RESTOptionsGetter) (genericapiserver.APIGroupInfo, error) {
33 apiGroupInfo := genericapiserver.NewDefaultAPIGroupInfo(coordination.GroupName, legacyscheme.Scheme, legacyscheme.ParameterCodec, legacyscheme.Codecs)
34
35
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[coordinationv1.SchemeGroupVersion.Version] = storageMap
41 }
42 return apiGroupInfo, nil
43 }
44
45 func (p RESTStorageProvider) v1Storage(apiResourceConfigSource serverstorage.APIResourceConfigSource, restOptionsGetter generic.RESTOptionsGetter) (map[string]rest.Storage, error) {
46 storage := map[string]rest.Storage{}
47
48
49 if resource := "leases"; apiResourceConfigSource.ResourceEnabled(coordinationv1.SchemeGroupVersion.WithResource(resource)) {
50 leaseStorage, err := leasestorage.NewREST(restOptionsGetter)
51 if err != nil {
52 return storage, err
53 }
54 storage[resource] = leaseStorage
55 }
56 return storage, nil
57 }
58
59 func (p RESTStorageProvider) GroupName() string {
60 return coordination.GroupName
61 }
62
View as plain text