repository: set user agent for mirror requests (#1342)

pingcap · May 6, 2021 · c7536d9 · c7536d9
1 parent 54ac4d5
commit c7536d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/repository/mirror.go b/pkg/repository/mirror.go
@@ -39,6 +39,7 @@ import (
 	"github.com/pingcap/tiup/pkg/utils/mock"
 	"github.com/pingcap/tiup/pkg/utils/rand"
 	"github.com/pingcap/tiup/pkg/verbose"
+	"github.com/pingcap/tiup/pkg/version"
 )
 
 const (
@@ -293,6 +294,7 @@ func (l *httpMirror) download(url string, to string, maxSize int64) (io.ReadClos
 	}(time.Now())
 
 	client := grab.NewClient()
+	client.UserAgent = fmt.Sprintf("tiup/%s", version.NewTiUPVersion().SemVer())
 	req, err := grab.NewRequest(to, url)
 	if err != nil {
 		return nil, errors.Trace(err)