Revert "Remove sampling_percentage until SDK issue fixed"

This reverts commit a6ede6c.
hashicorp · Jan 20, 2021 · fbac625 · fbac625
1 parent f864eef
commit fbac625
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/azurerm/internal/services/apimanagement/api_management_api_diagnostic_resource.go b/azurerm/internal/services/apimanagement/api_management_api_diagnostic_resource.go
@@ -129,6 +129,13 @@ func resourceApiManagementApiDiagnosticAdditionalContentSchema() *schema.Schema
 				},
 			},
 
+			"sampling_percentage": {
+				Type:         schema.TypeFloat,
+				Optional:     true,
+				Computed:     true,
+				ValidateFunc: validation.FloatBetween(0.0, 100.0),
+			},
+
 			"always_log_errors": {
 				Type:     schema.TypeBool,
 				Optional: true,
@@ -229,6 +236,15 @@ func resourceApiManagementApiDiagnosticCreateUpdate(d *schema.ResourceData, meta
 		},
 	}
 
+	if samplingPercentage, ok := d.GetOk("sampling_percentage"); ok {
+		parameters.Sampling = &apimanagement.SamplingSettings{
+			SamplingType: apimanagement.Fixed,
+			Percentage:   utils.Float(samplingPercentage.(float64)),
+		}
+	} else {
+		parameters.Sampling = nil
+	}
+
 	if alwaysLogErrors, ok := d.GetOk("always_log_errors"); ok && alwaysLogErrors.(bool) {
 		parameters.AlwaysLog = apimanagement.AllErrors
 	}
@@ -326,6 +342,9 @@ func resourceApiManagementApiDiagnosticRead(d *schema.ResourceData, meta interfa
 	d.Set("api_management_name", diagnosticId.ServiceName)
 	if props := resp.DiagnosticContractProperties; props != nil {
 		d.Set("api_management_logger_id", props.LoggerID)
+		if props.Sampling != nil && props.Sampling.Percentage != nil {
+			d.Set("sampling_percentage", props.Sampling.Percentage)
+		}
 		d.Set("always_log_errors", props.AlwaysLog == apimanagement.AllErrors)
 		d.Set("verbosity", props.Verbosity)
 		d.Set("log_client_ip", props.LogClientIP)

diff --git a/azurerm/internal/services/apimanagement/api_management_api_diagnostic_resource_test.go b/azurerm/internal/services/apimanagement/api_management_api_diagnostic_resource_test.go
@@ -260,6 +260,7 @@ resource "azurerm_api_management_api_diagnostic" "test" {
   api_management_name       = azurerm_api_management.test.name
   api_name                  = azurerm_api_management_api.test.name
   api_management_logger_id  = azurerm_api_management_logger.test.id
+  sampling_percentage       = 10
   always_log_errors         = true
   log_client_ip             = true
   http_correlation_protocol = "W3C"

diff --git a/website/docs/r/api_management_api_diagnostic.html.markdown b/website/docs/r/api_management_api_diagnostic.html.markdown
@@ -69,6 +69,7 @@ resource "azurerm_api_management_api_diagnostic" "example" {
   api_name                 = azurerm_api_management_api.example.name
   api_management_logger_id = azurerm_api_management_logger.example.id
 
+  sampling_percentage       = 5.0
   always_log_errors         = true
   log_client_ip             = true
   verbosity                 = "Verbose"
@@ -142,6 +143,8 @@ The following arguments are supported:
 
 * `log_client_ip` - (Optional) Log client IP address.
 
+* `sampling_percentage` - (Optional) Sampling (%). For high traffic APIs, please read this [documentation](https://docs.microsoft.com/azure/api-management/api-management-howto-app-insights#performance-implications-and-log-sampling) to understand performance implications and log sampling. Valid values are between `0.0` and `100.0`.
+
 * `verbosity` - (Optional) Logging verbosity. Possible values are `verbose`, `information` or `error`.
 
 ---