mirror of
https://gitgud.io/yats/libkiwi.git
synced 2026-06-28 06:55:34 -04:00
refactoring
This commit is contained in:
+10
-8
@@ -6,12 +6,13 @@ import (
|
||||
)
|
||||
|
||||
// http.CookieJar interface wrapper to add convenience functions.
|
||||
type jar struct {
|
||||
http.CookieJar
|
||||
type cookieJar struct {
|
||||
jar *http.CookieJar
|
||||
}
|
||||
|
||||
func (j *jar) GetCookie(u *url.URL, name string) *http.Cookie {
|
||||
cookies := j.Cookies(u)
|
||||
func (j *cookieJar) GetCookie(u *url.URL, name string) *http.Cookie {
|
||||
jar := (*j.jar)
|
||||
cookies := jar.Cookies(u)
|
||||
for _, c := range cookies {
|
||||
if c.Name == name {
|
||||
return c
|
||||
@@ -21,18 +22,19 @@ func (j *jar) GetCookie(u *url.URL, name string) *http.Cookie {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (j *jar) SetCookie(u *url.URL, newCookie *http.Cookie) {
|
||||
cookies := j.Cookies(u)
|
||||
func (j *cookieJar) SetCookie(u *url.URL, newCookie *http.Cookie) {
|
||||
jar := (*j.jar)
|
||||
cookies := jar.Cookies(u)
|
||||
|
||||
for i, c := range cookies {
|
||||
if c.Name == newCookie.Name {
|
||||
cookies[i] = newCookie
|
||||
j.SetCookies(u, cookies)
|
||||
jar.SetCookies(u, cookies)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Append if not already existing.
|
||||
cookies = append(cookies, newCookie)
|
||||
j.SetCookies(u, cookies)
|
||||
jar.SetCookies(u, cookies)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user