Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 5e8196d

Browse files
committed
加上 UA
1 parent 57b214d commit 5e8196d

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

logic/article.go

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ import (
1010
"context"
1111
"errors"
1212
"fmt"
13+
"net/http"
1314
"net/url"
1415
"regexp"
1516
"strconv"
@@ -89,8 +90,22 @@ func (self ArticleLogic) ParseArticle(ctx context.Context, articleUrl string, au
8990
// }
9091

9192
var doc *goquery.Document
92-
if doc, err = goquery.NewDocument(articleUrl); err != nil {
93-
logger.Errorln("goquery newdocument error:", err)
93+
94+
ua := `Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.116 Safari/537.36`
95+
req, err := http.NewRequest("GET", articleUrl, nil)
96+
if err != nil {
97+
logger.Errorln("new request error:", err)
98+
return nil, err
99+
}
100+
req.Header.Add("User-Agent", ua)
101+
resp, err := http.DefaultClient.Do(req)
102+
if err != nil {
103+
logger.Errorln("get response error:", err)
104+
return nil, err
105+
}
106+
defer resp.Body.Close()
107+
if doc, err = goquery.NewDocumentFromReader(resp.Body); err != nil {
108+
logger.Errorln("goquery NewDocumentFromReader error:", err)
94109
return nil, err
95110
}
96111

0 commit comments

Comments
 (0)