49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package badger
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.loyso.art/frx/eway/internal/entity"
|
|
badger "github.com/dgraph-io/badger/v4"
|
|
)
|
|
|
|
var (
|
|
categorySequenceIDKey = []byte("!!cat_seq!!")
|
|
)
|
|
|
|
type client struct {
|
|
db *badger.DB
|
|
nextCategoryIDSeq *badger.Sequence
|
|
}
|
|
|
|
func NewClient(db *badger.DB) (*client, error) {
|
|
categorySeqGen, err := db.GetSequence(categorySequenceIDKey, 10)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("getting sequence for categories: %w", err)
|
|
}
|
|
|
|
return &client{
|
|
db: db,
|
|
nextCategoryIDSeq: categorySeqGen,
|
|
}, nil
|
|
}
|
|
|
|
// Close closes the underlying sequences in the client. Should be called right before
|
|
// underlying *badger.DB closed.
|
|
func (c *client) Close() error {
|
|
err := c.nextCategoryIDSeq.Release()
|
|
if err != nil {
|
|
return fmt.Errorf("releasing next_category_sequence: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *client) Category() entity.CategoryRepository {
|
|
return newCategoryClient(c.db, c.nextCategoryIDSeq)
|
|
}
|
|
|
|
func (c *client) GoodsItem() entity.GoodsItemRepository {
|
|
return newGoodsItemClient(c.db)
|
|
}
|