serialize.go

// Copyright 2015 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package cookiejar

import (
	"bytes"
	"encoding/gob"
	"encoding/json/v2"
	"sort"
	"time"
)

// MarshalJSON implements json.Marshaler by encoding all cookies currently in the jar.
func (j *Jar) MarshalJSON() ([]byte, error) {
	j.mu.Lock()
	defer j.mu.Unlock()

	return json.Marshal(j.allEntries())
}

// UnmarshalJSON 解组行为并不会完全重写Jar中的cookie, 它是将传入的cookie与当前的Jar中的cookie进行合并.
func (j *Jar) UnmarshalJSON(p []byte) error {
	j.mu.Lock()
	defer j.mu.Unlock()

	var entries []entry
	err := json.Unmarshal(p, &entries)
	if err != nil {
		return err
	}

	j.merge(entries)
	j.deleteExpired(time.Now())
	return nil
}

func (j *Jar) GobEncode() ([]byte, error) {
	j.mu.Lock()
	defer j.mu.Unlock()

	buf := new(bytes.Buffer)
	encoder := gob.NewEncoder(buf)
	err := encoder.Encode(j.allEntries())
	return buf.Bytes(), err
}

// GobDecode 解组行为并不会完全重写Jar中的cookie, 它是将传入的cookie与当前的Jar中的cookie进行合并.
func (j *Jar) GobDecode(p []byte) error {
	j.mu.Lock()
	defer j.mu.Unlock()

	var entries []entry
	decoder := gob.NewDecoder(bytes.NewReader(p))
	err := decoder.Decode(&entries)
	if err != nil {
		return err
	}

	j.merge(entries)
	j.deleteExpired(time.Now())
	return nil
}

// allEntries returns all the entries in the jar, sorted by primarly by canonical host
// name and secondarily by path length.
func (j *Jar) allEntries() []entry {
	j.deleteExpired(time.Now())
	var entries []entry
	for _, submap := range j.entries {
		for _, e := range submap {
			entries = append(entries, e)
		}
	}
	sort.Sort(byCanonicalHost{entries})
	return entries
}