blob: 8e55d2598f133201c1a86f0ed4def125ae6f1a89 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
package bbolt
import (
"bytes"
"context"
"encoding/gob"
"log/slog"
"go.etcd.io/bbolt"
"go.neonxp.ru/middleware/session"
)
func New(db *bbolt.DB, bucketName []byte) session.Store {
return &Store{
db: db,
bucketName: bucketName,
}
}
type Store struct {
db *bbolt.DB
bucketName []byte
}
func (s *Store) Load(ctx context.Context, sessionID string) session.Value {
v := session.Value{}
err := s.db.View(func(tx *bbolt.Tx) error {
bucket := tx.Bucket(s.bucketName)
if bucket == nil {
// no bucket -- normal situation
return nil
}
vb := bucket.Get([]byte(sessionID))
if vb == nil {
// no session -- no error
return nil
}
rdr := bytes.NewBuffer(vb)
return gob.NewDecoder(rdr).Decode(&v)
})
if err != nil {
slog.WarnContext(ctx, "failed load session", slog.Any("error", err))
}
return v
}
func (s *Store) Save(ctx context.Context, sessionID string, value session.Value) error {
return s.db.Update(func(tx *bbolt.Tx) error {
bucket, err := tx.CreateBucketIfNotExists(s.bucketName)
if err != nil {
return err
}
wrt := bytes.NewBuffer([]byte{})
if err := gob.NewEncoder(wrt).Encode(value); err != nil {
return err
}
return bucket.Put([]byte(sessionID), wrt.Bytes())
})
}
|
