data_job.go

package jobs

import (
	"context"
	"fmt"

	"github.com/databricks/terraform-provider-databricks/common"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
)

func DataSourceJob() *schema.Resource {
	type queryableJobData struct {
		Id      string `json:"id,omitempty" tf:"computed"`
		JobId   string `json:"job_id,omitempty" tf:"computed"`
		Name    string `json:"name,omitempty" tf:"computed"`
		JobName string `json:"job_name,omitempty" tf:"computed"`
		Job     *Job   `json:"job_settings,omitempty" tf:"computed"`
	}
	return common.DataResource(queryableJobData{}, func(ctx context.Context, e any, c *common.DatabricksClient) error {
		data := e.(*queryableJobData)
		jobsAPI := NewJobsAPI(ctx, c)
		var list []Job
		var err error
		if data.Id == "" {
			data.Id = data.JobId
		}
		if data.Name == "" {
			data.Name = data.JobName
		}
		if data.Name != "" {
			// if name is provided, need to list all jobs ny name
			list, err = jobsAPI.ListByName(data.Name, true)
		} else {
			// otherwise, just read the job
			var job Job
			job, err = jobsAPI.Read(data.Id)
			if err != nil {
				return err
			}
			data.Job = &job
			data.Name = job.Settings.Name
		}
		if err != nil {
			return err
		}
		for _, job := range list {
			currentJob := job // De-referencing the temp variable used by the loop
			currentJobId := currentJob.ID()
			currentJobName := currentJob.Settings.Name
			if currentJobName == data.Name || currentJobId == data.Id {
				data.Job = &currentJob
				data.Name = currentJobName
				data.Id = currentJobId
				data.JobId = currentJobId
				return nil // break the loop after we found the job
			}
		}
		if data.Job == nil {
			return fmt.Errorf("no job found with specified name")
		}
		return nil
	})
}