Skip to content

Commit

Permalink
Refactor metadata service access
Browse files Browse the repository at this point in the history
Only pull the metadata that is needed.
  • Loading branch information
gottwald committed Jul 12, 2021
1 parent db266f4 commit baf778f
Showing 1 changed file with 9 additions and 6 deletions.
15 changes: 9 additions & 6 deletions driver/driver.go
Original file line number Diff line number Diff line change
Expand Up @@ -104,16 +104,19 @@ func NewDriver(ep, token, url, region, doTag, driverName, debugAddr string) (*Dr
})
oauthClient := oauth2.NewClient(context.Background(), tokenSource)

var hostID string
mdClient := metadata.NewClient()
if region == "" {
all, err := metadata.NewClient().Metadata()
var err error
region, err = mdClient.Region()
if err != nil {
return nil, fmt.Errorf("couldn't get metadata: %s (are you running outside of a DigitalOcean droplet and possibly forgot to specify the 'region' flag?)", err)
return nil, fmt.Errorf("couldn't get region from metadata: %s (are you running outside of a DigitalOcean droplet and possibly forgot to specify the 'region' flag?)", err)
}

region = all.Region
hostID = strconv.Itoa(all.DropletID)
}
hostIDInt, err := mdClient.DropletID()
if err != nil {
return nil, fmt.Errorf("couldn't get droplet ID from metadata: %s (are you running outside of a DigitalOcean droplet?)", err)
}
hostID := strconv.Itoa(hostIDInt)

opts := []godo.ClientOpt{}
opts = append(opts, godo.SetBaseURL(url))
Expand Down

0 comments on commit baf778f

Please sign in to comment.