feat: add support for batch v1 cronjob

Signed-off-by: Nico Braun <[email protected]>
Shopify · Jan 31, 2023 · 7308fe2 · 7308fe2
1 parent ad40a0c
commit 7308fe2
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 0 deletions.
diff --git a/internal/k8sinternal/client_test.go b/internal/k8sinternal/client_test.go
@@ -86,6 +86,7 @@ func TestGetAllResources(t *testing.T) {
 		k8s.NewStatefulSet(),
 		k8s.NewPodTemplate(),
 		k8s.NewCronJob(),
+		k8s.NewCronJobV1(),
 		k8s.NewServiceAccount(),
 		k8s.NewService(),
 		k8s.NewJob(),

diff --git a/pkg/k8s/helpers.go b/pkg/k8s/helpers.go
@@ -111,6 +111,8 @@ func GetPodTemplateSpec(resource Resource) *PodTemplateSpecV1 {
 	switch kubeType := resource.(type) {
 	case *CronJobV1Beta1:
 		return &kubeType.Spec.JobTemplate.Spec.Template
+	case *CronJobV1:
+		return &kubeType.Spec.JobTemplate.Spec.Template
 	case *DaemonSetV1:
 		return &kubeType.Spec.Template
 	case *DeploymentV1:

diff --git a/pkg/k8s/resources.go b/pkg/k8s/resources.go
@@ -127,6 +127,24 @@ func NewCronJob() *CronJobV1Beta1 {
 	}
 }
 
+// NewCronJobV1 creates a new CronJob resource
+func NewCronJobV1() *CronJobV1 {
+	return &CronJobV1{
+		TypeMeta: TypeMetaV1{
+			Kind:       "CronJob",
+			APIVersion: "batch/v1",
+		},
+		ObjectMeta: ObjectMetaV1{},
+		Spec: CronJobSpecV1{
+			JobTemplate: JobTemplateSpecV1{
+				Spec: JobSpecV1{
+					Template: podTemplateSpec,
+				},
+			},
+		},
+	}
+}
+
 // NewServiceAccount creates a new ServiceAccount resource
 func NewServiceAccount() *ServiceAccountV1 {
 	return &ServiceAccountV1{

diff --git a/pkg/k8s/types.go b/pkg/k8s/types.go
@@ -25,6 +25,12 @@ type CronJobV1Beta1 = batchv1beta1.CronJob
 // CronJobSpecV1Beta1 is a type alias for the v1beta1 version of the k8s batch API.
 type CronJobSpecV1Beta1 = batchv1beta1.CronJobSpec
 
+// CronJobV1 is a type alias for the v1 version of the k8s batch API.
+type CronJobV1 = batchv1.CronJob
+
+// CronJobSpecV1 is a type alias for the v1 version of the k8s batch API.
+type CronJobSpecV1 = batchv1.CronJobSpec
+
 // DaemonSetSpecV1 is a type alias for the v1 version of the k8s apps API.
 type DaemonSetSpecV1 = appsv1.DaemonSetSpec
 
@@ -40,6 +46,9 @@ type DeploymentV1 = appsv1.Deployment
 // JobTemplateSpecV1Beta1 is a type alias for the v1beta1 version of the k8s batch API.
 type JobTemplateSpecV1Beta1 = batchv1beta1.JobTemplateSpec
 
+// JobTemplateSpecV1 is a type alias for the v1 version of the k8s batch API.
+type JobTemplateSpecV1 = batchv1.JobTemplateSpec
+
 // JobSpecV1 is a type alias for the v1 version of the k8s batch API.
 type JobSpecV1 = batchv1.JobSpec