mirror of
https://github.com/gofiber/storage.git
synced 2025-11-01 04:02:44 +08:00
28 lines
559 B
Go
28 lines
559 B
Go
package bbolt
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/gofiber/utils/v2"
|
|
"go.etcd.io/bbolt"
|
|
berrors "go.etcd.io/bbolt/errors"
|
|
)
|
|
|
|
func createBucket(cfg Config, conn *bbolt.DB) error {
|
|
return conn.Update(func(tx *bbolt.Tx) error {
|
|
_, err := tx.CreateBucketIfNotExists(utils.UnsafeBytes(cfg.Bucket))
|
|
|
|
return err
|
|
})
|
|
}
|
|
|
|
func removeBucket(cfg Config, conn *bbolt.DB) error {
|
|
return conn.Update(func(tx *bbolt.Tx) error {
|
|
err := tx.DeleteBucket(utils.UnsafeBytes(cfg.Bucket))
|
|
if errors.Is(err, berrors.ErrBucketNotFound) {
|
|
return nil
|
|
}
|
|
|
|
return err
|
|
})
|
|
}
|