FilesExpand file tree

 master

/

controller.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

155 lines (128 loc) · 5.18 KB

 master

/

controller.go

Copy path

Top

File metadata and controls

• Code
• Blame

155 lines (128 loc) · 5.18 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

/*

Copyright 2018 The Kubernetes Authors.

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 controller

import (

"context"

"fmt"

"time"

"github.com/go-logr/logr"

"k8s.io/client-go/util/workqueue"

"k8s.io/klog/v2"

"sigs.k8s.io/controller-runtime/pkg/handler"

"sigs.k8s.io/controller-runtime/pkg/internal/controller"

"sigs.k8s.io/controller-runtime/pkg/manager"

"sigs.k8s.io/controller-runtime/pkg/predicate"

"sigs.k8s.io/controller-runtime/pkg/ratelimiter"

"sigs.k8s.io/controller-runtime/pkg/reconcile"

"sigs.k8s.io/controller-runtime/pkg/source"

)

// Options are the arguments for creating a new Controller.

type Options struct {

// MaxConcurrentReconciles is the maximum number of concurrent Reconciles which can be run. Defaults to 1.

MaxConcurrentReconciles int

// Reconciler reconciles an object

Reconciler reconcile.Reconciler

// RateLimiter is used to limit how frequently requests may be queued.

// Defaults to MaxOfRateLimiter which has both overall and per-item rate limiting.

// The overall is a token bucket and the per-item is exponential.

RateLimiter ratelimiter.RateLimiter

// LogConstructor is used to construct a logger used for this controller and passed

// to each reconciliation via the context field.

LogConstructor func(request *reconcile.Request) logr.Logger

// CacheSyncTimeout refers to the time limit set to wait for syncing caches.

// Defaults to 2 minutes if not set.

CacheSyncTimeout time.Duration

// RecoverPanic indicates whether the panic caused by reconcile should be recovered.

RecoverPanic bool

}

// Controller implements a Kubernetes API. A Controller manages a work queue fed reconcile.Requests

// from source.Sources. Work is performed through the reconcile.Reconciler for each enqueued item.

// Work typically is reads and writes Kubernetes objects to make the system state match the state specified

// in the object Spec.

type Controller interface {

// Reconciler is called to reconcile an object by Namespace/Name

reconcile.Reconciler

// Watch takes events provided by a Source and uses the EventHandler to

// enqueue reconcile.Requests in response to the events.

//

// Watch may be provided one or more Predicates to filter events before

// they are given to the EventHandler. Events will be passed to the

// EventHandler if all provided Predicates evaluate to true.

Watch(src source.Source, eventhandler handler.EventHandler, predicates ...predicate.Predicate) error

// Start starts the controller. Start blocks until the context is closed or a

// controller has an error starting.

Start(ctx context.Context) error

// GetLogger returns this controller logger prefilled with basic information.

GetLogger() logr.Logger

}

// New returns a new Controller registered with the Manager. The Manager will ensure that shared Caches have

// been synced before the Controller is Started.

func New(name string, mgr manager.Manager, options Options) (Controller, error) {

c, err := NewUnmanaged(name, mgr, options)

if err != nil {

return nil, err

}

// Add the controller as a Manager components

return c, mgr.Add(c)

}

// NewUnmanaged returns a new controller without adding it to the manager. The

// caller is responsible for starting the returned controller.

func NewUnmanaged(name string, mgr manager.Manager, options Options) (Controller, error) {

if options.Reconciler == nil {

return nil, fmt.Errorf("must specify Reconciler")

}

if len(name) == 0 {

return nil, fmt.Errorf("must specify Name for Controller")

}

if options.LogConstructor == nil {

log := mgr.GetLogger().WithValues(

"controller", name,

)

options.LogConstructor = func(req *reconcile.Request) logr.Logger {

log := log

if req != nil {

log = log.WithValues(

"object", klog.KRef(req.Namespace, req.Name),

"namespace", req.Namespace, "name", req.Name,

)

}

return log

}

}

if options.MaxConcurrentReconciles <= 0 {

options.MaxConcurrentReconciles = 1

}

if options.CacheSyncTimeout == 0 {

options.CacheSyncTimeout = 2 * time.Minute

}

if options.RateLimiter == nil {

options.RateLimiter = workqueue.DefaultControllerRateLimiter()

}

// Inject dependencies into Reconciler

if err := mgr.SetFields(options.Reconciler); err != nil {

return nil, err

}

// Create controller with dependencies set

return &controller.Controller{

Do: options.Reconciler,

MakeQueue: func() workqueue.RateLimitingInterface {

return workqueue.NewNamedRateLimitingQueue(options.RateLimiter, name)

},

MaxConcurrentReconciles: options.MaxConcurrentReconciles,

CacheSyncTimeout: options.CacheSyncTimeout,

SetFields: mgr.SetFields,

Name: name,

LogConstructor: options.LogConstructor,

RecoverPanic: options.RecoverPanic,

}, nil

}