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/network/client.go
View File

@@ -11,9 +11,9 @@ import (
)
const (
baseURI = "https://management.azure.com"
userAgent = "virtual-kubelet/azure-arm-networking/2018-07-01"
apiVersion = "2018-07-01"
baseURI = "https://management.azure.com"
defaultUserAgent = "virtual-kubelet/azure-arm-network/2018-08-01"
apiVersion = "2018-08-01"
)
// Client is a client for interacting with Azure networking
@@ -25,11 +25,16 @@ type Client struct {
}
// NewClient creates a new client for interacting with azure networking
func NewClient(azAuth *azure.Authentication) (*Client, error) {
func NewClient(azAuth *azure.Authentication, extraUserAgent string) (*Client, error) {
if azAuth == 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(azAuth, baseURI, userAgent)
if err != nil {
return nil, fmt.Errorf("Creating Azure client failed: %v", err)

4
providers/azure/client/network/main_test.go
View File

@@ -27,7 +27,7 @@ func TestMain(m *testing.M) {
os.Exit(1)
}
c, err := resourcegroups.NewClient(testAuth)
c, err := resourcegroups.NewClient(testAuth, "unit-test")
if err != nil {
os.Exit(1)
}
@@ -68,7 +68,7 @@ func newTestClient(t *testing.T) *Client {
if err := setupAuth(); err != nil {
t.Fatal(err)
}
c, err := NewClient(testAuth)
c, err := NewClient(testAuth, "unit-test")
if err != nil {
t.Fatal(err)
}