Add labels functionality to managed DNS

Author: chrisst

google/resource_dns_managed_zone.go

@@ -53,6 +53,12 @@ func resourceDnsManagedZone() *schema.Resource {
 				Computed: true,
 				ForceNew: true,
 			},
+
+			"labels": &schema.Schema{
+				Type:     schema.TypeMap,
+				Optional: true,
+				Elem:     &schema.Schema{Type: schema.TypeString},
+			},
 		},
 	}
 }
@@ -72,7 +78,18 @@ func resourceDnsManagedZoneCreate(d *schema.ResourceData, meta interface{}) erro
 		Description: d.Get("description").(string),
 	}
 
+	if clabels, ok := d.GetOk("labels"); ok {
+		labels := map[string]string{}
+
+		for k, v := range clabels.(map[string]interface{}) {
+			labels[k] = v.(string)
+		}
+
+		zone.Labels = labels
+	}
+
 	log.Printf("[DEBUG] DNS ManagedZone create request: %#v", zone)
+
 	zone, err = config.clientDns.ManagedZones.Create(project, zone).Do()
 	if err != nil {
 		return fmt.Errorf("Error creating DNS ManagedZone: %s", err)
@@ -102,6 +119,7 @@ func resourceDnsManagedZoneRead(d *schema.ResourceData, meta interface{}) error
 	d.Set("dns_name", zone.DnsName)
 	d.Set("description", zone.Description)
 	d.Set("project", project)
+	d.Set("labels", zone.Labels)
 
 	return nil
 }
@@ -120,6 +138,16 @@ func resourceDnsManagedZoneUpdate(d *schema.ResourceData, meta interface{}) erro
 		Description: d.Get("description").(string),
 	}
 
+	if clabels, ok := d.GetOk("labels"); ok {
+		labels := map[string]string{}
+
+		for k, v := range clabels.(map[string]interface{}) {
+			labels[k] = v.(string)
+		}
+
+		zone.Labels = labels
+	}
+
 	op, err := config.clientDnsBeta.ManagedZones.Patch(project, d.Id(), zone).Do()
 	if err != nil {
 		return err

google/resource_dns_managed_zone_test.go

@@ -85,5 +85,8 @@ resource "google_dns_managed_zone" "foobar" {
 	name = "mzone-test-%s"
 	dns_name = "tf-acctest-%s.hashicorptest.com."
 	description = "%s"
+	labels = {
+		foo = "bar"
+	}
 }`, suffix, suffix, description)
 }