-
Notifications
You must be signed in to change notification settings - Fork 4.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Maksymilian Boguń
committed
Jun 17, 2019
1 parent
4320131
commit 9afcc80
Showing
6 changed files
with
616 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,260 @@ | ||
package azurerm | ||
|
||
import ( | ||
"fmt" | ||
"regexp" | ||
"strings" | ||
|
||
"github.com/Azure/azure-sdk-for-go/services/preview/iothub/mgmt/2018-12-01-preview/devices" | ||
"github.com/hashicorp/terraform/helper/schema" | ||
"github.com/hashicorp/terraform/helper/validation" | ||
"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/azure" | ||
"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/tf" | ||
"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/validate" | ||
"github.com/terraform-providers/terraform-provider-azurerm/azurerm/utils" | ||
) | ||
|
||
func resourceArmIotHubRoute() *schema.Resource { | ||
return &schema.Resource{ | ||
Create: resourceArmIotHubRouteCreateUpdate, | ||
Read: resourceArmIotHubRouteRead, | ||
Update: resourceArmIotHubRouteCreateUpdate, | ||
Delete: resourceArmIotHubRouteDelete, | ||
Importer: &schema.ResourceImporter{ | ||
State: schema.ImportStatePassthrough, | ||
}, | ||
|
||
Schema: map[string]*schema.Schema{ | ||
"name": { | ||
Type: schema.TypeString, | ||
Required: true, | ||
ValidateFunc: validation.StringMatch( | ||
regexp.MustCompile("^[-_.a-zA-Z0-9]{1,64}$"), | ||
"Route Name name can only include alphanumeric characters, periods, underscores, hyphens, has a maximum length of 64 characters, and must be unique.", | ||
), | ||
}, | ||
|
||
"resource_group_name": azure.SchemaResourceGroupName(), | ||
|
||
"iothub_name": { | ||
Type: schema.TypeString, | ||
Required: true, | ||
ForceNew: true, | ||
ValidateFunc: validate.IoTHubName, | ||
}, | ||
|
||
"source": { | ||
Type: schema.TypeString, | ||
Required: true, | ||
ValidateFunc: validation.StringInSlice([]string{ | ||
"DeviceJobLifecycleEvents", | ||
"DeviceLifecycleEvents", | ||
"DeviceMessages", | ||
"Invalid", | ||
"TwinChangeEvents", | ||
}, false), | ||
}, | ||
"condition": { | ||
// The condition is a string value representing device-to-cloud message routes query expression | ||
// https://docs.microsoft.com/en-us/azure/iot-hub/iot-hub-devguide-query-language#device-to-cloud-message-routes-query-expressions | ||
Type: schema.TypeString, | ||
Optional: true, | ||
Default: "true", | ||
}, | ||
"endpoint_names": { | ||
Type: schema.TypeList, | ||
Elem: &schema.Schema{ | ||
Type: schema.TypeString, | ||
}, | ||
Required: true, | ||
}, | ||
"enabled": { | ||
Type: schema.TypeBool, | ||
Required: true, | ||
}, | ||
}, | ||
} | ||
} | ||
|
||
func resourceArmIotHubRouteCreateUpdate(d *schema.ResourceData, meta interface{}) error { | ||
client := meta.(*ArmClient).iothub.ResourceClient | ||
ctx := meta.(*ArmClient).StopContext | ||
|
||
iothubName := d.Get("iothub_name").(string) | ||
resourceGroup := d.Get("resource_group_name").(string) | ||
|
||
azureRMLockByName(iothubName, iothubResourceName) | ||
defer azureRMUnlockByName(iothubName, iothubResourceName) | ||
|
||
iothub, err := client.Get(ctx, resourceGroup, iothubName) | ||
if err != nil { | ||
if utils.ResponseWasNotFound(iothub.Response) { | ||
return fmt.Errorf("IotHub %q (Resource Group %q) was not found", iothubName, resourceGroup) | ||
} | ||
|
||
return fmt.Errorf("Error loading IotHub %q (Resource Group %q): %+v", iothubName, resourceGroup, err) | ||
} | ||
|
||
routeName := d.Get("name").(string) | ||
|
||
resourceId := fmt.Sprintf("%s/Routes/%s", *iothub.ID, routeName) | ||
|
||
source := devices.RoutingSource(d.Get("source").(string)) | ||
condition := d.Get("condition").(string) | ||
endpointNamesRaw := d.Get("endpoint_names").([]interface{}) | ||
isEnabled := d.Get("enabled").(bool) | ||
|
||
route := devices.RouteProperties{ | ||
Name: &routeName, | ||
Source: source, | ||
Condition: &condition, | ||
EndpointNames: utils.ExpandStringSlice(endpointNamesRaw), | ||
IsEnabled: &isEnabled, | ||
} | ||
|
||
routing := iothub.Properties.Routing | ||
|
||
if routing == nil { | ||
routing = &devices.RoutingProperties{} | ||
} | ||
|
||
if routing.Routes == nil { | ||
routes := make([]devices.RouteProperties, 0) | ||
routing.Routes = &routes | ||
} | ||
|
||
routes := make([]devices.RouteProperties, 0) | ||
|
||
alreadyExists := false | ||
for _, existingRoute := range *routing.Routes { | ||
if strings.EqualFold(*existingRoute.Name, routeName) { | ||
if d.IsNewResource() && requireResourcesToBeImported { | ||
return tf.ImportAsExistsError("azurerm_iothub_route", resourceId) | ||
} | ||
routes = append(routes, route) | ||
alreadyExists = true | ||
|
||
} else { | ||
routes = append(routes, existingRoute) | ||
} | ||
} | ||
|
||
if d.IsNewResource() { | ||
routes = append(routes, route) | ||
} else if !alreadyExists { | ||
return fmt.Errorf("Unable to find Route %q defined for IotHub %q (Resource Group %q)", routeName, iothubName, resourceGroup) | ||
} | ||
|
||
routing.Routes = &routes | ||
|
||
future, err := client.CreateOrUpdate(ctx, resourceGroup, iothubName, iothub, "") | ||
if err != nil { | ||
return fmt.Errorf("Error creating/updating IotHub %q (Resource Group %q): %+v", iothubName, resourceGroup, err) | ||
} | ||
|
||
if err = future.WaitForCompletionRef(ctx, client.Client); err != nil { | ||
return fmt.Errorf("Error waiting for the completion of the creating/updating of IotHub %q (Resource Group %q): %+v", iothubName, resourceGroup, err) | ||
} | ||
|
||
d.SetId(resourceId) | ||
|
||
return resourceArmIotHubRouteRead(d, meta) | ||
} | ||
|
||
func resourceArmIotHubRouteRead(d *schema.ResourceData, meta interface{}) error { | ||
client := meta.(*ArmClient).iothub.ResourceClient | ||
ctx := meta.(*ArmClient).StopContext | ||
|
||
parsedIothubRouteId, err := parseAzureResourceID(d.Id()) | ||
|
||
if err != nil { | ||
return err | ||
} | ||
|
||
resourceGroup := parsedIothubRouteId.ResourceGroup | ||
iothubName := parsedIothubRouteId.Path["IotHubs"] | ||
routeName := parsedIothubRouteId.Path["Routes"] | ||
|
||
iothub, err := client.Get(ctx, resourceGroup, iothubName) | ||
if err != nil { | ||
return fmt.Errorf("Error loading IotHub %q (Resource Group %q): %+v", iothubName, resourceGroup, err) | ||
} | ||
|
||
d.Set("name", routeName) | ||
d.Set("iothub_name", iothubName) | ||
d.Set("resource_group_name", resourceGroup) | ||
|
||
if iothub.Properties == nil || iothub.Properties.Routing == nil { | ||
return nil | ||
} | ||
|
||
if routes := iothub.Properties.Routing.Routes; routes != nil { | ||
for _, route := range *routes { | ||
if strings.EqualFold(*route.Name, routeName) { | ||
|
||
d.Set("source", route.Source) | ||
d.Set("condition", route.Condition) | ||
d.Set("enabled", route.IsEnabled) | ||
d.Set("endpoint_names", route.EndpointNames) | ||
} | ||
} | ||
} | ||
|
||
return nil | ||
} | ||
|
||
func resourceArmIotHubRouteDelete(d *schema.ResourceData, meta interface{}) error { | ||
client := meta.(*ArmClient).iothub.ResourceClient | ||
ctx := meta.(*ArmClient).StopContext | ||
|
||
parsedIothubRouteId, err := parseAzureResourceID(d.Id()) | ||
|
||
if err != nil { | ||
return err | ||
} | ||
|
||
resourceGroup := parsedIothubRouteId.ResourceGroup | ||
iothubName := parsedIothubRouteId.Path["IotHubs"] | ||
routeName := parsedIothubRouteId.Path["Routes"] | ||
|
||
azureRMLockByName(iothubName, iothubResourceName) | ||
defer azureRMUnlockByName(iothubName, iothubResourceName) | ||
|
||
iothub, err := client.Get(ctx, resourceGroup, iothubName) | ||
if err != nil { | ||
if utils.ResponseWasNotFound(iothub.Response) { | ||
return fmt.Errorf("IotHub %q (Resource Group %q) was not found", iothubName, resourceGroup) | ||
} | ||
|
||
return fmt.Errorf("Error loading IotHub %q (Resource Group %q): %+v", iothubName, resourceGroup, err) | ||
} | ||
|
||
if iothub.Properties == nil || iothub.Properties.Routing == nil { | ||
return nil | ||
} | ||
routes := iothub.Properties.Routing.Routes | ||
|
||
if routes == nil { | ||
return nil | ||
} | ||
|
||
updatedRoutes := make([]devices.RouteProperties, 0) | ||
for _, route := range *routes { | ||
if !strings.EqualFold(*route.Name, routeName) { | ||
updatedRoutes = append(updatedRoutes, route) | ||
} | ||
} | ||
|
||
iothub.Properties.Routing.Routes = &updatedRoutes | ||
|
||
future, err := client.CreateOrUpdate(ctx, resourceGroup, iothubName, iothub, "") | ||
if err != nil { | ||
return fmt.Errorf("Error updating IotHub %q (Resource Group %q) with Route %q: %+v", iothubName, resourceGroup, routeName, err) | ||
} | ||
|
||
if err = future.WaitForCompletionRef(ctx, client.Client); err != nil { | ||
return fmt.Errorf("Error waiting for IotHub %q (Resource Group %q) to finish updating Route %q: %+v", iothubName, resourceGroup, routeName, err) | ||
} | ||
|
||
return nil | ||
} |
Oops, something went wrong.