Update/add references to ExternalTrafficPolicy in structure_service_spec

hashicorp · Apr 3, 2018 · f169008 · f169008
1 parent 229245b
commit f169008
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/kubernetes/structure_service_spec.go b/kubernetes/structure_service_spec.go
@@ -58,6 +58,9 @@ func flattenServiceSpec(in v1.ServiceSpec) []interface{} {
 	if in.ExternalName != "" {
 		att["external_name"] = in.ExternalName
 	}
+	if in.ExternalTrafficPolicy != "" {
+		att["external_traffic_policy"] = string(in.ExternalTrafficPolicy)
+	}
 	return []interface{}{att}
 }
 
@@ -139,7 +142,7 @@ func expandServiceSpec(l []interface{}) v1.ServiceSpec {
 		obj.ExternalName = v
 	}
 	if v, ok := in["external_traffic_policy"].(string); ok {
-		obj.ExternalName = v
+		obj.ExternalTrafficPolicy = v1.ServiceExternalTrafficPolicyType(v)
 	}
 	return obj
 }
@@ -209,5 +212,11 @@ func patchServiceSpec(keyPrefix, pathPrefix string, d *schema.ResourceData, v *v
 			Value: d.Get(keyPrefix + "external_name").(string),
 		})
 	}
+	if d.HasChange(keyPrefix + "external_traffic_policy") {
+		ops = append(ops, &ReplaceOperation{
+			Path:  pathPrefix + "externalTrafficPolicy",
+			Value: d.Get(keyPrefix + "external_traffic_policy").(string),
+		})
+	}
 	return ops, nil
 }