d8dfun/virtual-kubelet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

[ACI] Support to add extra user agent to the request (#378)

Browse Source 
* Add UserAgent

* Change to join the user agent with a whitespace and set in the header

* Remove empty entry from user agent
This commit is contained in:
Robbie Zhang
2018-10-18 14:28:43 -07:00
committed by GitHub
parent 0dbc88826c
commit 5991b29f76
8 changed files with 49 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
providers/azure/client/resourcegroups/client.go
View File

@@ -9,9 +9,9 @@ import (
const (
// BaseURI is the default URI used for compute services.
BaseURI = "https://management.azure.com"
userAgent = "virtual-kubelet/azure-arm-resourcegroups/2017-12-01"
apiVersion = "2017-08-01"
BaseURI = "https://management.azure.com"
defaultUserAgent = "virtual-kubelet/azure-arm-resourcegroups/2017-12-01"
apiVersion = "2017-08-01"
resourceGroupURLPath = "subscriptions/{{.subscriptionId}}/resourcegroups/{{.resourceGroupName}}"
)
@@ -26,11 +26,16 @@ type Client struct {
}
// NewClient creates a new Azure resource groups client.
func NewClient(auth *azure.Authentication) (*Client, error) {
func NewClient(auth *azure.Authentication, extraUserAgent string) (*Client, error) {
if auth == nil {
return nil, fmt.Errorf("Authentication is not supplied for the Azure client")
}
userAgent := []string{defaultUserAgent}
if extraUserAgent != "" {
userAgent = append(userAgent, extraUserAgent)
}
client, err := azure.NewClient(auth, BaseURI, userAgent)
if err != nil {
return nil, fmt.Errorf("Creating Azure client failed: %v", err)

2
providers/azure/client/resourcegroups/client_test.go
View File

@@ -25,7 +25,7 @@ func TestNewClient(t *testing.T) {
t.Fatalf("Failed to load Azure authentication file: %v", err)
}
c, err := NewClient(auth)
c, err := NewClient(auth, "unit-test")
if err != nil {
t.Fatal(err)
}