Add default headers to HTTP client

Author: GregEremeev

clients/coinmarketcap.go

@@ -4,6 +4,8 @@ import (
 	"encoding/json"
 	"fmt"
 	"strconv"
+
+	"github.com/Pallinder/go-randomdata"
 )
 
 // coinmarketcap API urls
@@ -63,7 +65,7 @@ func (cac CoinmarketcapAPIClient) GetSymbols(
 		"sortBy":   sortBy,
 		"sortType": sortType,
 	}
-	res, err := cac.httpClient.MakeGetRequest(CryptoListingAPIURL, queryParams)
+	res, err := cac.httpClient.MakeGetRequest(CryptoListingAPIURL, queryParams, getDefaultHeaders())
 	if err != nil {
 		return nil, err
 	}
@@ -81,7 +83,7 @@ func (cac CoinmarketcapAPIClient) GetCoinIDBySymbol(symbol string) (int, error)
 	queryParams := map[string]string{
 		"symbol": symbol,
 	}
-	res, err := cac.httpClient.MakeGetRequest(CryptoMapAPIURL, queryParams)
+	res, err := cac.httpClient.MakeGetRequest(CryptoMapAPIURL, queryParams, getDefaultHeaders())
 	if err != nil {
 		return -1, err
 	}
@@ -103,7 +105,7 @@ func (cac CoinmarketcapAPIClient) GetSymbolChart(
 		"id":    fmt.Sprintf("%d", symbolID),
 		"range": rangeType,
 	}
-	res, err := cac.httpClient.MakeGetRequest(CryptoChartAPIURL, queryParams)
+	res, err := cac.httpClient.MakeGetRequest(CryptoChartAPIURL, queryParams, getDefaultHeaders())
 	if err != nil {
 		return map[string]valuesChart{}, err
 	}
@@ -115,6 +117,12 @@ func (cac CoinmarketcapAPIClient) GetSymbolChart(
 	return responseData.Data.Points, nil
 }
 
+func getDefaultHeaders() map[string]string {
+	return map[string]string{
+		"User-Agent": randomdata.UserAgentString(),
+	}
+}
+
 // CryptoChartResponse is a data structure with points to draw a chart
 type CryptoChartResponse struct {
 	Data timestampsChart `json:"data"`

clients/http.go

@@ -21,6 +21,7 @@ type customHTTPClient struct {
 func (chc customHTTPClient) MakeGetRequest(
 	urlPath string,
 	params map[string]string,
+	headers map[string]string,
 ) (*http.Response, error) {
 	finalURL, err := addQueryParams(urlPath, params)
 	if err != nil {
@@ -35,6 +36,9 @@ func (chc customHTTPClient) MakeGetRequest(
 	if err != nil {
 		return nil, err
 	}
+	for k, v := range headers {
+		req.Header.Set(k, v)
+	}
 	res, err := chc.httpClient.Do(req)
 	if err != nil {
 		return nil, err

clients/http_test.go

@@ -89,7 +89,7 @@ func TestMakeGetRequest(t *testing.T) {
 	defer patches.Reset()
 
 	httpClient := getHTTPClient()
-	res, err := httpClient.MakeGetRequest(expectedURL, expectedParams)
+	res, err := httpClient.MakeGetRequest(expectedURL, expectedParams, getDefaultHeaders())
 
 	if err != nil {
 		t.Fatalf("Err has to be nil but err is %v", err)

go.mod

@@ -19,10 +19,10 @@ require (
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.16 // indirect
-	github.com/mattn/go-runewidth v0.0.13 // indirect
+	github.com/mattn/go-runewidth v0.0.15 // indirect
 	github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
-	github.com/rivo/uniseg v0.2.0 // indirect
+	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	golang.org/x/sys v0.5.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect

go.sum

@@ -22,16 +22,18 @@ github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovk
 github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
 github.com/mattn/go-isatty v0.0.16 h1:bq3VjFmv/sOjHtdEhmkEV4x1AJtvUvOJ2PFAZ5+peKQ=
 github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
-github.com/mattn/go-runewidth v0.0.13 h1:lTGmDsbAYt5DmK6OnoV7EuIF1wEIFAcxld6ypU4OSgU=
 github.com/mattn/go-runewidth v0.0.13/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/mattn/go-runewidth v0.0.15 h1:UNAjwbU9l54TA3KzvqLGxwWjHmMgBUVhBiTjelZgg3U=
+github.com/mattn/go-runewidth v0.0.15/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
 github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d h1:5PJl274Y63IEHC+7izoQE9x6ikvDFZS2mDVS3drnohI=
 github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE=
 github.com/nbio/st v0.0.0-20140626010706-e9e8d9816f32 h1:W6apQkHrMkS0Muv8G/TipAy/FJl/rCYT0+EuS8+Z0z4=
 github.com/nbio/st v0.0.0-20140626010706-e9e8d9816f32/go.mod h1:9wM+0iRr9ahx58uYLpLIr5fm8diHn0JbqRycJi6w0Ms=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
 github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
+github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
+github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
 github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=