update
This commit is contained in:
parent
c33b59bcab
commit
53756c3ef9
4 changed files with 5 additions and 78 deletions
5
defaults.go
Normal file
5
defaults.go
Normal file
|
@ -0,0 +1,5 @@
|
|||
package securecookie
|
||||
|
||||
var Gob = GobEncoder{}
|
||||
var JSON = JSONEncoder{}
|
||||
var Nop = NopEncoder{}
|
56
litecodec.go
56
litecodec.go
|
@ -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
|
||||
}
|
|
@ -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)
|
||||
}
|
||||
}
|
|
@ -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)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue