[CRDB-43619] Added support for terminationGracePeriod in crdbcluster CR

Author: NishanthNalluri

WORKSPACE

@@ -57,12 +57,14 @@ load("@bazel_gazelle//:deps.bzl", "gazelle_dependencies")
 
 go_rules_dependencies()
 
+# gazelle:repository_macro hack/build/repos.bzl%_go_dependencies
+go_dependencies()
+
 go_register_toolchains(version = "1.17")
 
 gazelle_dependencies()
 
-# gazelle:repository_macro hack/build/repos.bzl%_go_dependencies
-go_dependencies()
+
 
 ################################
 # begin rules_oci dependencies #

apis/v1alpha1/cluster_types.go

@@ -145,6 +145,11 @@ type CrdbClusterSpec struct {
 	// Default: false
 	// +optional
 	AutomountServiceAccountToken bool `json:"automountServiceAccountToken,omitempty"`
+	// (Optional) Amount of grace time prior to the container being forcibly terminated
+	// when marked for deletion or restarted
+	// Default : 300
+	// +optional
+	TerminationGracePeriodSecs int32 `json:"terminationGracePeriodSecs,omitempty"`
 }
 
 // +k8s:openapi-gen=true

config/crd/bases/crdb.cockroachlabs.com_crdbclusters.yaml

@@ -1393,6 +1393,12 @@ spec:
                 description: '(Optional) The SQL Port number Default: 26257'
                 format: int32
                 type: integer
+              terminationGracePeriodSecs:
+                description: '(Optional) Amount of grace time prior to the container
+                  being forcibly terminated when marked for deletion or restarted
+                  Default : 300'
+                format: int32
+                type: integer
               tlsEnabled:
                 description: (Optional) TLSEnabled determines if TLS is enabled for
                   your CockroachDB Cluster

e2e/upgrades/upgrades_test.go

@@ -221,6 +221,7 @@ func TestUpgradesMinorVersionThenRollback(t *testing.T) {
 
 	builder := testutil.NewBuilder("crdb").
 		WithAutomountServiceAccountToken(true).
+		WithTerminationGracePeriodSeconds(5).
 		WithNodeCount(3).
 		WithTLS().
 		WithImage(e2e.MinorVersion1).

examples/3-nodes-insecure-no-topology.yaml

@@ -24,7 +24,7 @@ spec:
         - ReadWriteOnce
         resources:
           requests:
-            storage: 60Gi
+            storage: 1Gi
         volumeMode: Filesystem
   image:
     name: cockroachdb/cockroach:v19.2.6

hack/dev.sh

@@ -21,7 +21,7 @@ APP_VERSION="${APP_VERSION:-v$(cat version.txt)}"
 CLUSTER_NAME="dev"
 NODE_IMAGE="rancher/k3s:v1.23.3-k3s1"
 REGISTRY_NAME="registry.localhost"
-REGISTRY_PORT=5000
+REGISTRY_PORT=5002
 DEV_REGISTRY="${REGISTRY_NAME}:${REGISTRY_PORT}"
 
 main() {

install/crds.yaml

@@ -1391,6 +1391,12 @@ spec:
                 description: '(Optional) The SQL Port number Default: 26257'
                 format: int32
                 type: integer
+              terminationGracePeriodSecs:
+                description: '(Optional) Amount of grace time prior to the container
+                  being forcibly terminated when marked for deletion or restarted
+                  Default : 300'
+                format: int32
+                type: integer
               tlsEnabled:
                 description: (Optional) TLSEnabled determines if TLS is enabled for
                   your CockroachDB Cluster

pkg/resource/cluster.go

@@ -330,6 +330,13 @@ func (cluster Cluster) SecureMode() string {
 	return "--insecure"
 }
 
+func (cluster Cluster) GetTerminationGracePeriod() int32 {
+	if cluster.Spec().TerminationGracePeriodSecs == 0 {
+		return terminationGracePeriodSecs
+	}
+	return cluster.Spec().TerminationGracePeriodSecs
+}
+
 func (cluster Cluster) LoggingConfiguration(fetcher Fetcher) (string, error) {
 	if cluster.Spec().LogConfigMap != "" {
 		cm := &corev1.ConfigMap{

pkg/resource/statefulset.go

@@ -209,7 +209,7 @@ func (b StatefulSetBuilder) makePodTemplate() corev1.PodTemplateSpec {
 				RunAsUser: ptr.Int64(1000581000),
 				FSGroup:   ptr.Int64(1000581000),
 			},
-			TerminationGracePeriodSeconds: ptr.Int64(terminationGracePeriodSecs),
+			TerminationGracePeriodSeconds: ptr.Int64(int64(b.GetTerminationGracePeriod())),
 			Containers:                    b.MakeContainers(),
 			AutomountServiceAccountToken:  ptr.Bool(b.Spec().AutomountServiceAccountToken),
 			ServiceAccountName:            b.ServiceAccountName(),

pkg/resource/testdata/TestStatefulSetBuilder/automount_sa.golden

@@ -26,6 +26,7 @@ spec:
         car: koenigsegg
     spec:
       automountServiceAccountToken: true
+      terminationGracePeriodSecs: 5
       containers:
       - command:
         - /bin/bash