Add support for Kubernetes metadata (labels/annotations) (#189)

iterative · Oct 12, 2021 · 82d1b60 · 82d1b60
1 parent 4194d33
commit 82d1b60
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/iterative/kubernetes/provider.go b/iterative/kubernetes/provider.go
@@ -42,6 +42,12 @@ func ResourceMachineCreate(ctx context.Context, d *terraform_schema.ResourceData
 	jobName := d.Id()
 	jobNamespace := namespace
 
+	// Define the metadata
+	jobMetadata := map[string]string{}
+	for key, value := range d.Get("metadata").(map[string]interface{}) {
+		jobMetadata[key] = value.(string)
+	}
+
 	// Define the accelerator settings (i.e. GPU type, model, ...)
 	jobNodeSelector := map[string]string{}
 	jobAccelerator := instanceType["accelerator"]["model"]
@@ -88,8 +94,10 @@ func ResourceMachineCreate(ctx context.Context, d *terraform_schema.ResourceData
 
 	job := kubernetes_batch.Job{
 		ObjectMeta: kubernetes_meta.ObjectMeta{
-			Name:      jobName,
-			Namespace: jobNamespace,
+			Name:        jobName,
+			Namespace:   jobNamespace,
+			Labels:      jobMetadata,
+			Annotations: jobMetadata,
 		},
 		Spec: kubernetes_batch.JobSpec{
 			BackoffLimit:            &jobBackoffLimit,