This commit is contained in:
Hank Shen 2023-12-06 12:34:46 +08:00
parent c33b59bcab
commit 53756c3ef9
4 changed files with 5 additions and 78 deletions

5
defaults.go Normal file
View file

@ -0,0 +1,5 @@
package securecookie
var Gob = GobEncoder{}
var JSON = JSONEncoder{}
var Nop = NopEncoder{}

View file

@ -1,56 +0,0 @@
package securecookie
import "fmt"
var _ Codec = (*LiteCodec)(nil)
func NewLiteCodec() *LiteCodec {
s := &LiteCodec{
sz: GobEncoder{},
}
return s
}
// LiteCodec encodes and decodes
type LiteCodec struct {
err error
sz Serializer
}
// Encoding sets the encoding/serialization method for cookies.
//
// Default is encoding/gob. To encode special structures using encoding/gob,
// they must be registered first using gob.Register().
func (s *LiteCodec) SetSerializer(sz Serializer) *LiteCodec {
s.sz = sz
return s
}
// Encode encodes a value.
func (s *LiteCodec) Encode(name string, value interface{}) (string, error) {
if s.err != nil {
return "", s.err
}
var err error
var b []byte
// Serialize.
if b, err = s.sz.Serialize(value); err != nil {
return "", cookieError{cause: fmt.Errorf(`%w: %s`, err, name), typ: usageError}
}
// Done.
return string(b), nil
}
// Decode decodes a value. The dst argument must be a pointer.
func (s *LiteCodec) Decode(name, value string, dst interface{}, _ ...int) error {
if s.err != nil {
return s.err
}
// Deserialize.
if err := s.sz.Deserialize([]byte(value), dst); err != nil {
return cookieError{cause: fmt.Errorf(`%w: %s`, err, name), typ: decodeError}
}
// Done.
return nil
}

View file

@ -1,20 +0,0 @@
package securecookie
import "testing"
func TestLiteCodec(t *testing.T) {
c := NewLiteCodec()
str := `TestLiteCodec`
encoded, err := c.Encode(`name`, str)
if err != nil {
t.Error(err)
}
var decoded string
err = c.Decode(`name`, encoded, &decoded)
if err != nil {
t.Error(err)
}
if decoded != str {
t.Fatalf("expected %s, got: %#v", str, decoded)
}
}

View file

@ -49,8 +49,6 @@ func SetSerializer(codecs []Codec, sz Serializer) {
for _, c := range codecs {
if codec, ok := c.(*SecureCookie); ok {
codec.SetSerializer(sz)
} else if codec, ok := c.(*LiteCodec); ok {
codec.SetSerializer(sz)
}
}
}