Use lease for DCA leader election if k8s.version >= 1.21

Author: zhuminyi

components/datadog/agent/kubernetes_helm.go

@@ -13,6 +13,7 @@ import (
 	"github.com/DataDog/test-infra-definitions/components/datadog/apps/etcd"
 	"github.com/DataDog/test-infra-definitions/components/datadog/fakeintake"
 	"github.com/DataDog/test-infra-definitions/resources/helm"
+	"github.com/Masterminds/semver"
 
 	"github.com/pulumi/pulumi-kubernetes/sdk/v4/go/kubernetes"
 	corev1 "github.com/pulumi/pulumi-kubernetes/sdk/v4/go/kubernetes/core/v1"
@@ -79,6 +80,10 @@ func NewHelmInstallation(e config.Env, args HelmInstallationArgs, opts ...pulumi
 	appKey := e.AgentAPPKey()
 	baseName := "dda"
 	opts = append(opts, pulumi.Providers(args.KubeProvider), e.WithProviders(config.ProviderRandom), pulumi.DeletedWith(args.KubeProvider))
+	kubeVersion, err := semver.NewVersion(e.KubernetesVersion())
+	if err != nil {
+		kubeVersion, _ = semver.NewVersion("1.21.0")
+	}
 
 	helmComponent := &HelmComponent{}
 	if err := e.Ctx().RegisterComponentResource("dd:agent", "dda", helmComponent, opts...); err != nil {
@@ -156,7 +161,7 @@ func NewHelmInstallation(e config.Env, args HelmInstallationArgs, opts ...pulumi
 	if args.GKEAutopilot {
 		values = buildLinuxHelmValuesAutopilot(baseName, agentImagePath, agentImageTag, clusterAgentImagePath, clusterAgentImageTag, randomClusterAgentToken.Result)
 	} else {
-		values = buildLinuxHelmValues(baseName, agentImagePath, agentImageTag, clusterAgentImagePath, clusterAgentImageTag, randomClusterAgentToken.Result, !args.DisableLogsContainerCollectAll, e.TestingWorkloadDeploy())
+		values = buildLinuxHelmValues(baseName, agentImagePath, agentImageTag, clusterAgentImagePath, clusterAgentImageTag, randomClusterAgentToken.Result, !args.DisableLogsContainerCollectAll, e.TestingWorkloadDeploy(), kubeVersion)
 	}
 	values.configureImagePullSecret(imgPullSecret)
 	values.configureFakeintake(e, args.Fakeintake, args.DualShipping)
@@ -240,7 +245,13 @@ func NewHelmInstallation(e config.Env, args HelmInstallationArgs, opts ...pulumi
 
 type HelmValues pulumi.Map
 
-func buildLinuxHelmValues(baseName, agentImagePath, agentImageTag, clusterAgentImagePath, clusterAgentImageTag string, clusterAgentToken pulumi.StringInput, logsContainerCollectAll bool, testingWorkloadsEnabled bool) HelmValues {
+func buildLinuxHelmValues(
+	baseName, agentImagePath, agentImageTag, clusterAgentImagePath, clusterAgentImageTag string,
+	clusterAgentToken pulumi.StringInput,
+	logsContainerCollectAll bool,
+	testingWorkloadsEnabled bool,
+	kubeVersion *semver.Version,
+) HelmValues {
 	var containerRegistry, imageName string
 	if strings.Contains(agentImagePath, "/") {
 		agentImageElem := strings.Split(agentImagePath, "/")
@@ -250,11 +261,19 @@ func buildLinuxHelmValues(baseName, agentImagePath, agentImageTag, clusterAgentI
 		containerRegistry = ""
 		imageName = agentImagePath
 	}
+	// Determine leader election mechanism based on kubeVersion
+	kubeThresholdVersion, _ := semver.NewVersion("1.21.0")
+	leaderElectionType := "" // Default empty string will be mapped to "lease" for versions >= 1.14.0
+	if kubeVersion.LessThan(kubeThresholdVersion) {
+		leaderElectionType = "configmap"
+	}
+
 	helmValues := HelmValues{
 		"datadog": pulumi.Map{
-			"apiKeyExistingSecret": pulumi.String(baseName + "-datadog-credentials"),
-			"appKeyExistingSecret": pulumi.String(baseName + "-datadog-credentials"),
-			"checksCardinality":    pulumi.String("high"),
+			"apiKeyExistingSecret":   pulumi.String(baseName + "-datadog-credentials"),
+			"appKeyExistingSecret":   pulumi.String(baseName + "-datadog-credentials"),
+			"leaderElectionResource": pulumi.String(leaderElectionType),
+			"checksCardinality":      pulumi.String("high"),
 			"namespaceLabelsAsTags": pulumi.Map{
 				"related_team": pulumi.String("team"),
 			},