simon987/od-database-crawler
1
0
Fork 0
mirror of https://github.com/terorie/od-database-crawler.git synced 2025-04-19 02:16:43 +00:00
Code Issues Projects Releases Wiki Activity

Better server User-Agent string

Browse Source
This commit is contained in:
Richard Patel 2019-02-03 12:23:21 +01:00
parent 5239af08f7
commit 9d5f549774
No known key found for this signature in database
GPG Key ID: C268B2BBDA2ABECB
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server.go
View File

@ -17,8 +17,11 @@ import (
var serverClient = http.Client { var serverClient = http.Client {
Timeout: config.ServerTimeout, Timeout: config.ServerTimeout,
Transport: new(ServerTripper),
} }
var serverUserAgent = "od-database-crawler/" + rootCmd.Version
func FetchTask() (t *Task, err error) { func FetchTask() (t *Task, err error) {
res, err := serverClient.PostForm( res, err := serverClient.PostForm(
config.ServerUrl + "/task/get", config.ServerUrl + "/task/get",
@ -176,3 +179,10 @@ func CancelTask(websiteId uint64) (err error) {
return 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)
}