brian/boltease
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge branch 'main' of ssh://git.bullercodeworks.com:2200/brian/boltease

Browse Source
This commit is contained in:
Brian Buller 2025-02-06 12:23:26 -06:00
commit 4d3794a057
2 changed files with 56 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
boltease.go
View File

@ -37,7 +37,7 @@ func Create(fn string, m os.FileMode, opts *bolt.Options) (*DB, error) {
// SetStruct takes a database, a path, a struct, and a function that takes a // SetStruct takes a database, a path, a struct, and a function that takes a
// path and an element and processes it. // path and an element and processes it.
func SetStruct[T any](b *DB, path []string, value T, fn func([]string, T) error) error { func SetStructInDB[T any](b *DB, path []string, value T, fn func([]string, T) error) error {
var err error var err error
if !b.dbIsOpen { if !b.dbIsOpen {
if err = b.OpenDB(); err != nil { if err = b.OpenDB(); err != nil {
@ -69,7 +69,7 @@ func SetStruct[T any](b *DB, path []string, value T, fn func([]string, T) error)
// SetStructList takes a path, a slice of structs, and a function that takes // SetStructList takes a path, a slice of structs, and a function that takes
// a bolt bucket and one element of the slice and processes it. // a bolt bucket and one element of the slice and processes it.
func SetStructList[T any](b *DB, path []string, values []T, fn func(*bolt.Bucket, T) error) error { func SetStructListDB[T any](b *DB, path []string, values []T, fn func(*bolt.Bucket, T) error) error {
var err error var err error
if !b.dbIsOpen { if !b.dbIsOpen {
if err = b.OpenDB(); err != nil { if err = b.OpenDB(); err != nil {

57
bucket_funcs.go
View File

@ -143,11 +143,12 @@ func GetBool(bkt *bolt.Bucket, key string) (bool, error) {
if err != nil { if err != nil {
return false, err return false, err
} }
if r == "true" || r == "1" { switch r {
case "true", "1":
return true, nil return true, nil
} else if r == "false" || r == "0" { case "false", "0":
return false, nil return false, nil
} else { default:
return false, fmt.Errorf("cannot parse as a boolean") return false, fmt.Errorf("cannot parse as a boolean")
} }
} }
@ -384,3 +385,53 @@ func SetKeyValueStringMap(bkt *bolt.Bucket, kvMap map[string]string) error {
} }
return SetKeyValueMap(bkt, chain) return SetKeyValueMap(bkt, chain)
} }
func GetStruct[T any](bkt *bolt.Bucket, fn func(*bolt.Bucket) (T, error)) (T, error) {
return fn(bkt)
}
func SetStruct[T any](bkt *bolt.Bucket, value T, fn func(*bolt.Bucket, T) error) error {
return fn(bkt, value)
}
func GetStructList[T any](bkt *bolt.Bucket, fn func(*bolt.Bucket) (T, error)) ([]T, error) {
var ret []T
if bkt == nil {
return nil, fmt.Errorf("bucket must not be nil")
}
keyList, err := GetBucketList(bkt)
if err != nil {
return nil, err
}
for _, k := range keyList {
sBkt, err := GetBucket(bkt, k)
if err != nil {
return nil, err
}
v, err := fn(sBkt)
if err != nil {
return nil, err
}
ret = append(ret, v)
}
return ret, nil
}
func SetStructList[T any](bkt *bolt.Bucket, values []T, fn func(*bolt.Bucket, T) error) error {
bkt.SetSequence(0)
for _, v := range values {
id, _ := bkt.NextSequence()
bId := make([]byte, 8)
binary.BigEndian.PutUint64(bId, id)
// Delete the bucket if it already exists
bkt.DeleteBucket(bId)
structBkt, err := bkt.CreateBucketIfNotExists(bId)
if err != nil {
return err
}
if err = fn(structBkt, v); err != nil {
return err
}
}
return nil
}