feat: add SerializerAdapter support custom serializer

This commit is contained in:
weloe
2023-05-17 01:16:14 +08:00
parent 900e86123a
commit 2d9f8124cc
5 changed files with 148 additions and 3 deletions

View File

@@ -449,14 +449,57 @@ func (e *Enforcer) AddTokenGenerateFun(tokenStyle string, f model.GenerateFunc)
func (e *Enforcer) GetSession(id string) *model.Session {
if v := e.adapter.Get(e.spliceSessionKey(id)); v != nil {
session := v.(*model.Session)
return session
if s := e.sessionUnSerialize(v); s != nil {
return s
} else {
session, ok := v.(*model.Session)
if !ok {
return nil
}
return session
}
}
return nil
}
func (e *Enforcer) sessionUnSerialize(v interface{}) *model.Session {
// get serializer
serializer, ok := e.adapter.(persist.SerializerAdapter)
if !ok {
return nil
}
// to bytes
bytes, err := util.InterfaceToBytes(v)
if err != nil {
return nil
}
session, err := serializer.UnSerialize(bytes)
if err != nil {
return nil
}
return session
}
func (e *Enforcer) sessionSerialize(v *model.Session) ([]byte, error) {
serializer, ok := e.adapter.(persist.SerializerAdapter)
if !ok {
return nil, nil
}
return serializer.Serialize(v)
}
func (e *Enforcer) SetSession(id string, session *model.Session, timeout int64) error {
err := e.adapter.Set(e.spliceSessionKey(id), session, timeout)
bytes, err := e.sessionSerialize(session)
if err != nil {
return err
}
if bytes != nil {
err = e.adapter.Set(e.spliceSessionKey(id), bytes, timeout)
} else {
err = e.adapter.Set(e.spliceSessionKey(id), session, timeout)
}
if err != nil {
return err
}
@@ -470,3 +513,19 @@ func (e *Enforcer) deleteSession(id string) error {
}
return nil
}
func (e *Enforcer) updateSession(id string, session *model.Session) error {
bytes, err := e.sessionSerialize(session)
if err != nil {
return err
}
if bytes != nil {
err = e.adapter.Update(e.spliceSessionKey(id), bytes)
} else {
err = e.adapter.Update(e.spliceSessionKey(id), session)
}
if err != nil {
return err
}
return nil
}