cookiejar.go

package httplib

import (
	"net/http"
	"net/url"
	"sync"
)

var _ http.CookieJar = (*customCookieJar)(nil)

type customCookieJar struct {
	mu   sync.Mutex
	data map[string]string
}

func (c *customCookieJar) SetCookies(u *url.URL, cookies []*http.Cookie) {
	c.mu.Lock()
	defer c.mu.Unlock()
	for i := range cookies {
		name := cookies[i].Name
		value := cookies[i].Value
		c.data[name] = value
	}
}

func (c *customCookieJar) Cookies(u *url.URL) []*http.Cookie {
	c.mu.Lock()
	defer c.mu.Unlock()
	cookies := make([]*http.Cookie, 0, len(c.data))
	for k, v := range c.data {
		cookie := http.Cookie{Value: v, Name: k}
		cookies = append(cookies, &cookie)
	}
	return cookies
}