Add tracing spans to azquery (#21986)

* Add tracing spans to azquery * fix moduleName const * remove vestigial setting
Azure · Nov 15, 2023 · 4acec80 · 4acec80
1 parent 5df5a71
commit 4acec80
Show file tree

Hide file tree

Showing 6 changed files with 15 additions and 4 deletions.
diff --git a/sdk/monitor/azquery/CHANGELOG.md b/sdk/monitor/azquery/CHANGELOG.md
@@ -9,6 +9,7 @@
 ### Bugs Fixed
 
 ### Other Changes
+* Enabled spans for distributed tracing.
 
 ## 1.1.0 (2023-05-09)
 

diff --git a/sdk/monitor/azquery/autorest.md b/sdk/monitor/azquery/autorest.md
@@ -3,7 +3,6 @@
 ``` yaml
 title: Logs Query Client
 clear-output-folder: false
-export-clients: true
 go: true
 input-file: 
     - https://github.com/Azure/azure-rest-api-specs/blob/72427ef3ff5875bd8409ef112ef5e6f3cf2b8795/specification/operationalinsights/data-plane/Microsoft.OperationalInsights/stable/2022-10-27/OperationalInsights.json
@@ -13,6 +12,7 @@ openapi-type: "data-plane"
 output-folder: ../azquery
 security: "AADToken"
 use: "@autorest/[email protected]"
+inject-spans: true
 version: "^3.0.0"
 
 directive:

diff --git a/sdk/monitor/azquery/custom_client.go b/sdk/monitor/azquery/custom_client.go
@@ -46,7 +46,7 @@ func NewLogsClient(credential azcore.TokenCredential, options *LogsClientOptions
 	}
 
 	authPolicy := runtime.NewBearerTokenPolicy(credential, []string{c.Audience + "/.default"}, nil)
-	azcoreClient, err := azcore.NewClient("azquery.LogsClient", version, runtime.PipelineOptions{PerRetry: []policy.Policy{authPolicy}}, &options.ClientOptions)
+	azcoreClient, err := azcore.NewClient(moduleName, version, runtime.PipelineOptions{PerRetry: []policy.Policy{authPolicy}}, &options.ClientOptions)
 	if err != nil {
 		return nil, err
 	}
@@ -67,7 +67,7 @@ func NewMetricsClient(credential azcore.TokenCredential, options *MetricsClientO
 	}
 
 	authPolicy := runtime.NewBearerTokenPolicy(credential, []string{c.Audience + "/.default"}, nil)
-	azcoreClient, err := azcore.NewClient("azquery.MetricsClient", version, runtime.PipelineOptions{PerRetry: []policy.Policy{authPolicy}}, &options.ClientOptions)
+	azcoreClient, err := azcore.NewClient(moduleName, version, runtime.PipelineOptions{PerRetry: []policy.Policy{authPolicy}}, &options.ClientOptions)
 	if err != nil {
 		return nil, err
 	}

diff --git a/sdk/monitor/azquery/logs_client.go b/sdk/monitor/azquery/logs_client.go
diff --git a/sdk/monitor/azquery/metrics_client.go b/sdk/monitor/azquery/metrics_client.go
diff --git a/sdk/monitor/azquery/version.go b/sdk/monitor/azquery/version.go
@@ -7,6 +7,6 @@
 package azquery
 
 const (
-	moduleName = "azquery"
+	moduleName = "github.com/Azure/azure-sdk-for-go/sdk/monitor/azquery"
 	version    = "v1.1.1"
 )