mirror of
https://github.com/gofiber/storage.git
synced 2025-09-27 12:52:25 +08:00
3a8b8d4f71
* Add DB() support for Redis driver * Added support for DB() to all drivers * Fixed typo in README and Lint issue * Fix lint issue with ristretto db * Fix lint issue with bbolt db * Rename DB() to Conn() * Replace all instances of _DB with _Conn * Update all the README files * Return ArangoDB Client instead of DB
112 lines
2.0 KiB
Go
112 lines
2.0 KiB
Go
package bbolt
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/gofiber/utils"
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
// Storage interface that is implemented by storage providers
|
|
type Storage struct {
|
|
conn *bbolt.DB
|
|
bucket string
|
|
}
|
|
|
|
// New creates a new storage
|
|
func New(config ...Config) *Storage {
|
|
// Set default config
|
|
cfg := configDefault(config...)
|
|
|
|
conn, err := bbolt.Open(cfg.Database, 0666, &bbolt.Options{
|
|
Timeout: cfg.Timeout,
|
|
ReadOnly: cfg.ReadOnly,
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Reset bucket if field selected
|
|
if cfg.Reset {
|
|
if err := removeBucket(cfg, conn); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// Create bucket if not exists
|
|
if err := createBucket(cfg, conn); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return &Storage{
|
|
conn: conn,
|
|
bucket: cfg.Bucket,
|
|
}
|
|
|
|
}
|
|
|
|
// Get value by key
|
|
func (s *Storage) Get(key string) ([]byte, error) {
|
|
if len(key) <= 0 {
|
|
return nil, nil
|
|
}
|
|
|
|
var value []byte
|
|
|
|
err := s.conn.View(func(tx *bbolt.Tx) error {
|
|
b := tx.Bucket(utils.UnsafeBytes(s.bucket))
|
|
value = b.Get(utils.UnsafeBytes(key))
|
|
|
|
return nil
|
|
})
|
|
|
|
return value, err
|
|
}
|
|
|
|
// Set key with value
|
|
func (s *Storage) Set(key string, value []byte, exp time.Duration) error {
|
|
if len(key) <= 0 || len(value) <= 0 {
|
|
return nil
|
|
}
|
|
|
|
return s.conn.Update(func(tx *bbolt.Tx) error {
|
|
b := tx.Bucket(utils.UnsafeBytes(s.bucket))
|
|
|
|
return b.Put(utils.UnsafeBytes(key), value)
|
|
})
|
|
}
|
|
|
|
// Delete entry by key
|
|
func (s *Storage) Delete(key string) error {
|
|
if len(key) <= 0 {
|
|
return nil
|
|
}
|
|
|
|
return s.conn.Update(func(tx *bbolt.Tx) error {
|
|
b := tx.Bucket(utils.UnsafeBytes(s.bucket))
|
|
|
|
return b.Delete(utils.UnsafeBytes(key))
|
|
})
|
|
}
|
|
|
|
// Reset all entries
|
|
func (s *Storage) Reset() error {
|
|
return s.conn.Update(func(tx *bbolt.Tx) error {
|
|
b := tx.Bucket(utils.UnsafeBytes(s.bucket))
|
|
|
|
return b.ForEach(func(k, _ []byte) error {
|
|
return b.Delete(k)
|
|
})
|
|
})
|
|
}
|
|
|
|
// Close the database
|
|
func (s *Storage) Close() error {
|
|
return s.conn.Close()
|
|
}
|
|
|
|
// Return database client
|
|
func (s *Storage) Conn() *bbolt.DB {
|
|
return s.conn
|
|
}
|