70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
package users
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
var keyUsers = []byte("users")
|
|
|
|
type BoltUserStore struct {
|
|
db *bbolt.DB
|
|
}
|
|
|
|
func NewBoltUserStore(path string) (*BoltUserStore, error) {
|
|
db, err := bbolt.Open(path, 0666, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := db.Update(func(tx *bbolt.Tx) error {
|
|
_, err := tx.CreateBucketIfNotExists(keyUsers)
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &BoltUserStore{db: db}, nil
|
|
}
|
|
|
|
func (s *BoltUserStore) Close() error {
|
|
return s.db.Close()
|
|
}
|
|
|
|
func (s *BoltUserStore) Get(username string) (*User, error) {
|
|
var user User
|
|
err := s.db.View(func(tx *bbolt.Tx) error {
|
|
bkt := tx.Bucket(keyUsers)
|
|
rawUser := bkt.Get([]byte(username))
|
|
if err := json.Unmarshal(rawUser, &user); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (s *BoltUserStore) Store(user *User) error {
|
|
return s.db.Update(func(tx *bbolt.Tx) error {
|
|
bkt := tx.Bucket(keyUsers)
|
|
|
|
data, err := json.Marshal(user)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return bkt.Put([]byte(user.Username), data)
|
|
})
|
|
}
|
|
|
|
func (s *BoltUserStore) Delete(username string) error {
|
|
return s.db.Update(func(tx *bbolt.Tx) error {
|
|
bkt := tx.Bucket(keyUsers)
|
|
return bkt.Delete([]byte(username))
|
|
})
|
|
}
|