Save response cookies

This commit is contained in:
Mikhail Klementyev 2016-11-20 16:18:47 +03:00
parent 386c74a10b
commit f01134eaa4

View file

@ -132,6 +132,13 @@ func Get(db *sql.DB, linkUrl string) {
storage.AddHistoryURL(db, linkUrl) storage.AddHistoryURL(db, linkUrl)
if len(resp.Cookies()) != 0 {
err = storage.AddCookies(db, lastUrl.Host, resp.Cookies())
if err != nil {
log.Fatalln("Add cookies:", err)
}
}
defer resp.Body.Close() defer resp.Body.Close()
utf8, err := charset.NewReader(resp.Body, resp.Header.Get("Content-Type")) utf8, err := charset.NewReader(resp.Body, resp.Header.Get("Content-Type"))
@ -231,6 +238,13 @@ func Form(db *sql.DB, formID int64, formArgs []string) {
var status int64 var status int64
fmt.Sscanf(resp.Status, "%d", &status) fmt.Sscanf(resp.Status, "%d", &status)
if status < 400 && len(resp.Cookies()) != 0 {
err = storage.AddCookies(db, lastUrl.Host, resp.Cookies())
if err != nil {
log.Fatalln("Add cookies:", err)
}
}
if status >= 300 && status < 400 { if status >= 300 && status < 400 {
Get(db, lastUrl.String()) Get(db, lastUrl.String())
} }