Merge pull request #8259 from paulmey/azure-chroot/user-agent

[azure] Set correct user agent for Azure client set
pull/8265/head
Megan Marsh 6 years ago committed by GitHub
commit ea3e9b67a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -8,6 +8,7 @@ import (
"github.com/Azure/azure-sdk-for-go/profiles/latest/compute/mgmt/compute"
"github.com/Azure/azure-sdk-for-go/profiles/latest/compute/mgmt/compute/computeapi"
"github.com/Azure/go-autorest/autorest"
"github.com/hashicorp/packer/helper/useragent"
)
type AzureClientSet interface {
@ -33,6 +34,10 @@ type azureClientSet struct {
}
func New(c Config, say func(string)) (AzureClientSet, error) {
return new(c, say)
}
func new(c Config, say func(string)) (*azureClientSet, error) {
token, err := c.GetServicePrincipalToken(say, c.CloudEnvironment().ResourceManagerEndpoint)
if err != nil {
return nil, err
@ -46,12 +51,16 @@ func New(c Config, say func(string)) (AzureClientSet, error) {
}
func (s azureClientSet) configureAutorestClient(c *autorest.Client) {
c.AddToUserAgent(useragent.String())
c.Authorizer = s.authorizer
c.Sender = s.sender
}
func (s azureClientSet) MetadataClient() MetadataClientAPI {
return metadataClient{s.sender}
return metadataClient{
s.sender,
useragent.String(),
}
}
func (s azureClientSet) DisksClient() computeapi.DisksClientAPI {

@ -27,6 +27,7 @@ type ComputeInfo struct {
// metadataClient implements MetadataClient
type metadataClient struct {
autorest.Sender
UserAgent string
}
var _ MetadataClientAPI = metadataClient{}
@ -38,6 +39,7 @@ func (client metadataClient) GetComputeInfo() (*ComputeInfo, error) {
req, err := autorest.CreatePreparer(
autorest.AsGet(),
autorest.WithHeader("Metadata", "true"),
autorest.WithUserAgent(client.UserAgent),
autorest.WithBaseURL(imdsURL),
).Prepare((&http.Request{}))
if err != nil {

Loading…
Cancel
Save