mirror of
https://github.com/tgdrive/teldrive.git
synced 2024-09-20 08:15:55 +08:00
42 lines
673 B
Go
42 lines
673 B
Go
package kv
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/gotd/td/telegram"
|
|
)
|
|
|
|
type Session struct {
|
|
kv KV
|
|
key string
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func NewSession(kv KV, key string) telegram.SessionStorage {
|
|
return &Session{kv: kv, key: key}
|
|
}
|
|
|
|
func (s *Session) LoadSession(_ context.Context) ([]byte, error) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
b, err := s.kv.Get(s.key)
|
|
if err != nil {
|
|
if errors.Is(err, ErrNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
data := make([]byte, len(b))
|
|
copy(data, b)
|
|
return data, nil
|
|
}
|
|
|
|
func (s *Session) StoreSession(_ context.Context, data []byte) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
return s.kv.Set(s.key, data)
|
|
}
|