Make TerminationGracePeriodSeconds configurable

- related to issue #419

Author: ChunyiLyu

api/v1beta1/rabbitmqcluster_types.go

@@ -63,6 +63,12 @@ type RabbitmqClusterSpec struct {
 	// Has no effect if the cluster only consists of one node.
 	// For more information, see https://www.rabbitmq.com/rabbitmq-queues.8.html#rebalance
 	SkipPostDeploySteps bool `json:"skipPostDeploySteps,omitempty"`
+	// TerminationGracePeriodSeconds is the timeout that each rabbitmqcluster pod will have to terminate gracefully
+	// It defaults to 604800 seconds ( a week long) to ensure that the container preStop lifecycle hook can finish running
+	// For more information, see: https://github.com/rabbitmq/cluster-operator/blob/main/docs/design/20200520-graceful-pod-termination.md
+	// +kubebuilder:validation:Minimum:=0
+	// +kubebuilder:default:=604800
+	TerminationGracePeriodSeconds *int64 `json:"terminationGracePeriodSeconds,omitempty"`
 }
 
 type RabbitmqClusterOverrideSpec struct {

api/v1beta1/rabbitmqcluster_types_test.go

@@ -179,6 +179,7 @@ var _ = Describe("RabbitmqCluster", func() {
 
 			When("CR is fully populated", func() {
 				It("outputs the CR", func() {
+					zero := int64(0)
 					storage := k8sresource.MustParse("987Gi")
 					storageClassName := "some-class"
 					rmqClusterInstance = RabbitmqCluster{
@@ -187,9 +188,10 @@ var _ = Describe("RabbitmqCluster", func() {
 							Namespace: "default",
 						},
 						Spec: RabbitmqClusterSpec{
-							Replicas:         &three,
-							Image:            "rabbitmq-image-from-cr",
-							ImagePullSecrets: []corev1.LocalObjectReference{{Name: "my-super-secret"}},
+							Replicas:                      &three,
+							Image:                         "rabbitmq-image-from-cr",
+							ImagePullSecrets:              []corev1.LocalObjectReference{{Name: "my-super-secret"}},
+							TerminationGracePeriodSeconds: &zero,
 							Service: RabbitmqClusterServiceSpec{
 								Type: "NodePort",
 								Annotations: map[string]string{
@@ -470,14 +472,16 @@ func getKey(cluster *RabbitmqCluster) types.NamespacedName {
 
 func generateRabbitmqClusterObject(clusterName string) *RabbitmqCluster {
 	storage := k8sresource.MustParse("10Gi")
+	terminationPeriod := int64(604800)
 	return &RabbitmqCluster{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      clusterName,
 			Namespace: "default",
 		},
 		Spec: RabbitmqClusterSpec{
-			Replicas: pointer.Int32Ptr(1),
-			Image:    "rabbitmq:3.8.9",
+			Replicas:                      pointer.Int32Ptr(1),
+			Image:                         "rabbitmq:3.8.9",
+			TerminationGracePeriodSeconds: &terminationPeriod,
 			Service: RabbitmqClusterServiceSpec{
 				Type: "ClusterIP",
 			},

api/v1beta1/zz_generated.deepcopy.go

@@ -3723,6 +3723,15 @@ spec:
                   the cluster only consists of one node. For more information, see
                   https://www.rabbitmq.com/rabbitmq-queues.8.html#rebalance
                 type: boolean
+              terminationGracePeriodSeconds:
+                default: 604800
+                description: 'TerminationGracePeriodSeconds is the timeout that each
+                  rabbitmqcluster pod will have to terminate gracefully It defaults
+                  to 604800 seconds ( a week long) to ensure that the container preStop
+                  lifecycle hook can finish running For more information, see: https://github.com/rabbitmq/cluster-operator/blob/main/docs/design/20200520-graceful-pod-termination.md'
+                format: int64
+                minimum: 0
+                type: integer
               tls:
                 properties:
                   caCertName:

config/crd/bases/rabbitmq.com_rabbitmqclusters.yaml

@@ -245,8 +245,6 @@ func (builder *StatefulSetBuilder) podTemplateSpec(annotations, labels map[strin
 	rabbitmqGID := int64(999)
 	rabbitmqUID := int64(999)
 
-	terminationGracePeriod := defaultGracePeriodTimeoutSeconds
-
 	volumes := []corev1.Volume{
 		{
 			Name: "server-conf",
@@ -510,7 +508,7 @@ func (builder *StatefulSetBuilder) podTemplateSpec(annotations, labels map[strin
 				RunAsUser:  &rabbitmqUID,
 			},
 			ImagePullSecrets:              builder.Instance.Spec.ImagePullSecrets,
-			TerminationGracePeriodSeconds: &terminationGracePeriod,
+			TerminationGracePeriodSeconds: builder.Instance.Spec.TerminationGracePeriodSeconds,
 			ServiceAccountName:            builder.Instance.ChildResourceName(serviceAccountName),
 			AutomountServiceAccountToken:  &automountServiceAccountToken,
 			Affinity:                      builder.Instance.Spec.Affinity,

internal/resource/statefulset.go

@@ -1011,12 +1011,19 @@ var _ = Describe("StatefulSet", func() {
 			}))
 		})
 
-		It("adds the required terminationGracePeriodSeconds", func() {
+		It("sets TerminationGracePeriodSeconds in podTemplate as provided in instance spec", func() {
+			ten := int64(10)
+			instance.Spec.TerminationGracePeriodSeconds = &ten
+			builder = &resource.RabbitmqResourceBuilder{
+				Instance: &instance,
+				Scheme:   scheme,
+			}
+
 			stsBuilder := builder.StatefulSet()
 			Expect(stsBuilder.Update(statefulSet)).To(Succeed())
 
 			gracePeriodSeconds := statefulSet.Spec.Template.Spec.TerminationGracePeriodSeconds
-			expectedGracePeriodSeconds := int64(60 * 60 * 24 * 7)
+			expectedGracePeriodSeconds := int64(10)
 			Expect(gracePeriodSeconds).To(Equal(&expectedGracePeriodSeconds))
 		})