Add machine_type to dataflow_job

GoogleCloudPlatform · May 31, 2019 · 913f76a · 913f76a
1 parent 5385389
commit 913f76a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/third_party/terraform/resources/resource_dataflow_job.go b/third_party/terraform/resources/resource_dataflow_job.go
@@ -110,6 +110,12 @@ func resourceDataflowJob() *schema.Resource {
 				ForceNew:         true,
 				DiffSuppressFunc: compareSelfLinkOrResourceName,
 			},
+
+			"machine_type": {
+				Type:     schema.TypeString,
+				Optional: true,
+				ForceNew: true,
+			},
 		},
 	}
 }
@@ -140,6 +146,7 @@ func resourceDataflowJobCreate(d *schema.ResourceData, meta interface{}) error {
 		ServiceAccountEmail: d.Get("service_account_email").(string),
 		Subnetwork:          d.Get("subnetwork").(string),
 		TempLocation:        d.Get("temp_gcs_location").(string),
+		MachineType:         d.Get("machine_type").(string),
 		Zone:                zone,
 	}
 

diff --git a/third_party/terraform/tests/resource_dataflow_job_test.go b/third_party/terraform/tests/resource_dataflow_job_test.go
@@ -332,6 +332,7 @@ resource "google_dataflow_job" "big_data" {
 
 	template_gcs_path = "gs://dataflow-templates/wordcount/template_file"
 	temp_gcs_location = "${google_storage_bucket.temp.url}"
+	machine_type = "n1-standard-2"
 
 	parameters = {
 		inputFile = "gs://dataflow-samples/shakespeare/kinglear.txt"

diff --git a/third_party/terraform/website/docs/r/dataflow_job.html.markdown b/third_party/terraform/website/docs/r/dataflow_job.html.markdown
@@ -52,6 +52,7 @@ The following arguments are supported:
 * `service_account_email` - (Optional) The Service Account email used to create the job.
 * `network` - (Optional) The network to which VMs will be assigned. If it is not provided, "default" will be used.
 * `subnetwork` - (Optional) The subnetwork to which VMs will be assigned. Should be of the form "regions/REGION/subnetworks/SUBNETWORK".
+* `machine_type` - (Optional) The machine type to use for the job.
 
 
 ## Attributes Reference