datadog-operator/internal/controller/datadogagent/controller.go at 87e8b9dae0f3c0f2f60f85a61eafa56d3baf822b · DataDog/datadog-operator · GitHub

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
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2016-present Datadog, Inc.

package datadogagent

import (
	"context"
	"time"

	"github.com/go-logr/logr"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/util/managedfields"
	"k8s.io/client-go/tools/record"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/reconcile"

	"github.com/DataDog/datadog-operator/api/datadoghq/v2alpha1"
	componentagent "github.com/DataDog/datadog-operator/internal/controller/datadogagent/component/agent"
	"github.com/DataDog/datadog-operator/pkg/controller/utils/datadog"
	"github.com/DataDog/datadog-operator/pkg/kubernetes"

	// Use to register features
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/admissioncontroller"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/apm"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/appsec"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/asm"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/autoscaling"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/clusterchecks"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/controlplanemonitoring"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/cspm"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/cws"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/dataplane"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/dogstatsd"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/dummy"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/ebpfcheck"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/enabledefault"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/eventcollection"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/externalmetrics"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/flightrecorder"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/gpu"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/helmcheck"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/hostprofiler"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/kubernetesstatecore"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/livecontainer"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/liveprocess"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/logcollection"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/npm"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/oomkill"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/orchestratorexplorer"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/otelagentgateway"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/otelcollector"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/otlp"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/privateactionrunner"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/processdiscovery"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/prometheusscrape"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/remoteconfig"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/sbom"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/servicediscovery"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/tcpqueuelength"
	_ "github.com/DataDog/datadog-operator/internal/controller/datadogagent/feature/usm"
)

const (
	defaultRequeuePeriod    = 15 * time.Second
	defaultErrRequeuePeriod = 5 * time.Second
)

// ReconcilerOptions provides options read from command line
type ReconcilerOptions struct {
	ExtendedDaemonsetOptions   componentagent.ExtendedDaemonsetOptions
	SupportCilium              bool
	OperatorMetricsEnabled     bool
	IntrospectionEnabled       bool
	DatadogAgentProfileEnabled bool
	UntaintControllerEnabled   bool
	DatadogCSIDriverEnabled    bool
	CreateControllerRevisions  bool
	// ExperimentTimeout overrides ExperimentDefaultTimeout. Zero means use the default.
	ExperimentTimeout time.Duration
}

// Reconciler is the internal reconciler for Datadog Agent
type Reconciler struct {
	options           ReconcilerOptions
	client            client.Client
	platformInfo      kubernetes.PlatformInfo
	scheme            *runtime.Scheme
	log               logr.Logger
	recorder          record.EventRecorder
	forwarders        datadog.MetricsForwardersManager
	fieldManager      *managedfields.FieldManager
	componentRegistry *ComponentRegistry
}

func (r *Reconciler) initializeComponentRegistry() {
	r.componentRegistry = NewComponentRegistry(r)
	// Register all components
	r.componentRegistry.Register(NewClusterAgentComponent(r))
	r.componentRegistry.Register(NewClusterChecksRunnerComponent(r))
	r.componentRegistry.Register(NewOtelAgentGatewayComponent(r))
}

// NewReconciler returns a reconciler for DatadogAgent
func NewReconciler(options ReconcilerOptions, client client.Client, platformInfo kubernetes.PlatformInfo,
	scheme *runtime.Scheme, log logr.Logger, recorder record.EventRecorder, metricForwardersMgr datadog.MetricsForwardersManager,
) (*Reconciler, error) {
	r := &Reconciler{
		options:      options,
		client:       client,
		platformInfo: platformInfo,
		scheme:       scheme,
		log:          log,
		recorder:     recorder,
		forwarders:   metricForwardersMgr,
	}

	// Initialize component registry
	r.initializeComponentRegistry()

	return r, nil
}

// Reconcile is similar to reconciler.Reconcile interface, but taking a context
func (r *Reconciler) Reconcile(ctx context.Context, dda *v2alpha1.DatadogAgent) (reconcile.Result, error) {
	var resp reconcile.Result
	var err error

	resp, err = r.internalReconcileV2(ctx, dda)

	r.metricsForwarderProcessError(dda, err)
	return resp, err
}

// metricsForwarderProcessError convert the reconciler errors into metrics if metrics forwarder is enabled
func (r *Reconciler) metricsForwarderProcessError(dda *v2alpha1.DatadogAgent, err error) {
	if r.options.OperatorMetricsEnabled {
		r.forwarders.ProcessError(dda, err)
	}
}