add user-agent to request for tracking terraform usage

Author: Sean Sain

sumologic/sumologic_client.go

@@ -35,6 +35,13 @@ var endpoints = map[string]string{
 
 var rateLimiter = time.Tick(time.Minute / 240)
 
+func addHeaders(request *http.Request, accessId string, accessKey string) *http.Request {
+	request.Header.Add("Content-Type", "application/json")
+	request.Header.Add("User-Agent", "SumoLogicTerraformProvider/2.0.0")
+	request.SetBasicAuth(accessId, accessKey)
+	return request
+}
+
 func (s *Client) PostWithCookies(urlPath string, payload interface{}) ([]byte, []*http.Cookie, error) {
 	relativeURL, err := url.Parse(urlPath)
 	if err != nil {
@@ -53,8 +60,7 @@ func (s *Client) PostWithCookies(urlPath string, payload interface{}) ([]byte, [
 		return nil, nil, err
 	}
 
-	req.Header.Add("Content-Type", "application/json")
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 
 	<-rateLimiter
 	resp, err := s.httpClient.Do(req)
@@ -90,8 +96,7 @@ func (s *Client) GetWithCookies(urlPath string, cookies []*http.Cookie) ([]byte,
 		return nil, "", err
 	}
 
-	req.Header.Add("Content-Type", "application/json")
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 
 	for _, cookie := range cookies {
 		req.AddCookie(cookie)
@@ -124,8 +129,7 @@ func (s *Client) Post(urlPath string, payload interface{}) ([]byte, error) {
 
 	body, _ := json.Marshal(payload)
 	req, _ := http.NewRequest(http.MethodPost, sumoURL.String(), bytes.NewBuffer(body))
-	req.Header.Add("Content-Type", "application/json")
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 
 	<-rateLimiter
 	resp, err := s.httpClient.Do(req)
@@ -150,8 +154,7 @@ func (s *Client) PostRawPayload(urlPath string, payload string) ([]byte, error)
 	relativeURL, _ := url.Parse(urlPath)
 	sumoURL := s.BaseURL.ResolveReference(relativeURL)
 	req, _ := http.NewRequest(http.MethodPost, sumoURL.String(), bytes.NewBuffer([]byte(payload)))
-	req.Header.Add("Content-Type", "application/json")
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 
 	<-rateLimiter
 	resp, err := s.httpClient.Do(req)
@@ -180,11 +183,9 @@ func (s *Client) Put(urlPath string, payload interface{}) ([]byte, error) {
 
 	body, _ := json.Marshal(payload)
 	req, _ := http.NewRequest(http.MethodPut, sumoURL.String(), bytes.NewBuffer(body))
-	req.Header.Add("Content-Type", "application/json")
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 	req.Header.Add("If-Match", etag)
 
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
-
 	<-rateLimiter
 	resp, err := s.httpClient.Do(req)
 	if err != nil {
@@ -209,8 +210,7 @@ func (s *Client) Get(urlPath string) ([]byte, string, error) {
 	sumoURL := s.BaseURL.ResolveReference(relativeURL)
 
 	req, _ := http.NewRequest(http.MethodGet, sumoURL.String(), nil)
-	req.Header.Add("Content-Type", "application/json")
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 
 	<-rateLimiter
 	resp, err := s.httpClient.Do(req)
@@ -238,8 +238,7 @@ func (s *Client) Delete(urlPath string) ([]byte, error) {
 	sumoURL := s.BaseURL.ResolveReference(relativeURL)
 
 	req, _ := http.NewRequest(http.MethodDelete, sumoURL.String(), nil)
-	req.Header.Add("Content-Type", "application/json")
-	req.SetBasicAuth(s.AccessID, s.AccessKey)
+	req = addHeaders(req, s.AccessID, s.AccessKey)
 
 	<-rateLimiter
 	resp, err := s.httpClient.Do(req)