Better server User-Agent string

This commit is contained in:
Richard Patel
2019-02-03 12:23:21 +01:00
parent 5239af08f7
commit 9d5f549774

View File

@@ -17,8 +17,11 @@ import (
var serverClient = http.Client {
Timeout: config.ServerTimeout,
Transport: new(ServerTripper),
}
var serverUserAgent = "od-database-crawler/" + rootCmd.Version
func FetchTask() (t *Task, err error) {
res, err := serverClient.PostForm(
config.ServerUrl + "/task/get",
@@ -176,3 +179,10 @@ func CancelTask(websiteId uint64) (err error) {
return
}
type ServerTripper struct{}
func (t *ServerTripper) RoundTrip(req *http.Request) (res *http.Response, err error) {
req.Header.Set("User-Agent", serverUserAgent)
return http.DefaultClient.Transport.RoundTrip(req)
}