diff --git a/http.go b/http.go index bea8f68..7e01df0 100644 --- a/http.go +++ b/http.go @@ -32,7 +32,7 @@ func HttpGet(url string, client *http.Client) string { req.Header.Set("User-Agent", randomUserAgent()) req.Header.Set("Accept-Language", "en-us") // req.Header.Set("X-Requested-With", "XMLHttpRequest") - // req.Header.Set("Referer", "https://cart.tinydeal.com/index.php?__page_id=checkout&order_source=shopping_cart") + req.Header.Set("Referer", "https://www.papajohns.ru/") resp, err := client.Do(req) if err != nil { @@ -45,6 +45,20 @@ func HttpGet(url string, client *http.Client) string { return string(contents) } +func HttpHead(url string, client *http.Client) int { + req, _ := http.NewRequest("HEAD", url, nil) + req.Header.Set("User-Agent", randomUserAgent()) + req.Header.Set("Accept-Language", "en-us") + + resp, err := client.Do(req) + if err != nil { + fmt.Printf("Cannot get url %s, err: %s", url, err.Error()) + return 0 + } + + return resp.StatusCode +} + func HttpGetDump(url string, client *http.Client, file string) (string, string) { req, _ := http.NewRequest("GET", url, nil) req.Header.Set("User-Agent", randomUserAgent()) diff --git a/random.go b/random.go index 3f7f7b6..919fbfd 100644 --- a/random.go +++ b/random.go @@ -6,6 +6,10 @@ import ( ) func Random(min, max int) int { + if min == max { + return min + } + rand.Seed(time.Now().Unix()) return rand.Intn(max-min) + min }