/
runtime_client.go
101 lines (84 loc) · 3.17 KB
/
runtime_client.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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// Copyright (c) 2020 SAP SE or an SAP affiliate company. All rights reserved. This file is licensed under the Apache Software License, v. 2 except as noted otherwise in the LICENSE file
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package kubernetes
import (
"fmt"
"time"
"golang.org/x/time/rate"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/runtime/schema"
"k8s.io/client-go/rest"
"k8s.io/utils/pointer"
"sigs.k8s.io/controller-runtime/pkg/cache"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
kubernetescache "github.com/gardener/gardener/pkg/client/kubernetes/cache"
)
const (
defaultCacheResyncPeriod = 6 * time.Hour
)
func setClientOptionsDefaults(config *rest.Config, options *client.Options) error {
if options.Mapper == nil {
// default the client's REST mapper to a dynamic REST mapper (automatically rediscovers resources on NoMatchErrors)
mapper, err := apiutil.NewDynamicRESTMapper(
config,
apiutil.WithLazyDiscovery,
apiutil.WithLimiter(rate.NewLimiter(rate.Every(5*time.Second), 1)),
)
if err != nil {
return fmt.Errorf("failed to create new DynamicRESTMapper: %w", err)
}
options.Mapper = mapper
}
return nil
}
// AggregatorCacheFunc returns a `cache.NewCacheFunc` which creates a cache that holds different cache implementations depending on the objects' GVKs.
func AggregatorCacheFunc(newCache cache.NewCacheFunc, typeToNewCache map[client.Object]cache.NewCacheFunc, scheme *runtime.Scheme) cache.NewCacheFunc {
return func(config *rest.Config, options cache.Options) (cache.Cache, error) {
if err := setCacheOptionsDefaults(&options); err != nil {
return nil, err
}
fallbackCache, err := newCache(config, options)
if err != nil {
return nil, err
}
gvkToCache := make(map[schema.GroupVersionKind]cache.Cache)
for object, fn := range typeToNewCache {
gvk, err := apiutil.GVKForObject(object, scheme)
if err != nil {
return nil, err
}
cache, err := fn(config, options)
if err != nil {
return nil, err
}
gvkToCache[gvk] = cache
}
return kubernetescache.NewAggregator(fallbackCache, gvkToCache, scheme), nil
}
}
// NewRuntimeCache creates a new cache.Cache with the given config and options. It can be used
// for creating new controller-runtime clients with caches.
func NewRuntimeCache(config *rest.Config, options cache.Options) (cache.Cache, error) {
if err := setCacheOptionsDefaults(&options); err != nil {
return nil, err
}
return cache.New(config, options)
}
func setCacheOptionsDefaults(options *cache.Options) error {
if options.Resync == nil {
options.Resync = pointer.Duration(defaultCacheResyncPeriod)
}
return nil
}