65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package storm
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/boltdb/bolt"
|
|
"github.com/fatih/structs"
|
|
)
|
|
|
|
// Remove removes a structure from the associated bucket
|
|
func (n *Node) Remove(data interface{}) error {
|
|
if !structs.IsStruct(data) {
|
|
return ErrBadType
|
|
}
|
|
|
|
info, err := extract(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
id, err := toBytes(info.ID.Value, n.s.Codec)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if n.tx != nil {
|
|
return n.remove(n.tx, info, id)
|
|
}
|
|
|
|
return n.s.Bolt.Update(func(tx *bolt.Tx) error {
|
|
return n.remove(tx, info, id)
|
|
})
|
|
}
|
|
|
|
func (n *Node) remove(tx *bolt.Tx, info *modelInfo, id []byte) error {
|
|
bucket := n.GetBucket(tx, info.Name)
|
|
if bucket == nil {
|
|
return fmt.Errorf("bucket %s doesn't exist", info.Name)
|
|
}
|
|
|
|
for fieldName, idxInfo := range info.Indexes {
|
|
idx, err := getIndex(bucket, idxInfo.Type, fieldName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = idx.RemoveID(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
raw := bucket.Get(id)
|
|
if raw == nil {
|
|
return ErrNotFound
|
|
}
|
|
|
|
return bucket.Delete(id)
|
|
}
|
|
|
|
// Remove removes a structure from the associated bucket
|
|
func (s *DB) Remove(data interface{}) error {
|
|
return s.root.Remove(data)
|
|
}
|