mox/vendor/github.com/mjl-/bstore/tx.go

package bstore

import (
	"bytes"
	"context"
	"fmt"
	"reflect"

	bolt "go.etcd.io/bbolt"
)

// Mark a tx as botched, mentioning last actual error.
// Used when write operations fail. The transaction can be in inconsistent
// state, e.g. only some of a type's indicies may have been updated. We never
// want to commit such transactions.
func (tx *Tx) markError(err *error) {
	if *err != nil && tx.err == nil {
		tx.err = fmt.Errorf("%w (after %v)", ErrTxBotched, *err)
	}
}

// Return if an error condition is set on on the transaction. To be called before
// starting an operation.
func (tx *Tx) error() error {
	if tx.err != nil {
		return tx.err
	}
	if tx.db == nil {
		return errTxClosed
	}
	if err := tx.ctx.Err(); err != nil {
		tx.err = err
		return err
	}
	return nil
}

func (tx *Tx) structptr(value any) (reflect.Value, error) {
	rv := reflect.ValueOf(value)
	if !rv.IsValid() || rv.Kind() != reflect.Ptr || !rv.Elem().IsValid() || rv.Type().Elem().Kind() != reflect.Struct {
		return reflect.Value{}, fmt.Errorf("%w: value must be non-nil pointer to a struct, is %T", ErrParam, value)
	}
	rv = rv.Elem()
	return rv, nil
}

func (tx *Tx) structOrStructptr(value any) (reflect.Value, error) {
	rv := reflect.ValueOf(value)
	if !rv.IsValid() {
		return reflect.Value{}, fmt.Errorf("%w: value must be non-nil if pointer", ErrParam)
	}
	if rv.Kind() == reflect.Ptr {
		rv = rv.Elem()
		if !rv.IsValid() {
			return rv, fmt.Errorf("%w: value must be non-nil if pointer", ErrParam)
		}
	}
	if rv.Kind() != reflect.Struct {
		return reflect.Value{}, fmt.Errorf("%w: value must be a struct or pointer to a struct, is %T", ErrParam, value)
	}
	return rv, nil
}

// update indices by comparing indexed fields of the ov (old) and v (new). Only if
// the fields changed will the index be updated. Either ov or v may be the
// reflect.Value zero value, indicating there is no old/new value and the index
// should be updated.
func (tx *Tx) updateIndices(tv *typeVersion, pk []byte, ov, v reflect.Value) error {

	changed := func(idx *index) bool {
		for _, f := range idx.Fields {
			ofv := ov.FieldByIndex(f.structField.Index)
			nfv := v.FieldByIndex(f.structField.Index)
			if f.Type.Kind == kindSlice {
				// Index field is a slice type, cannot use direct interface comparison.
				on := ofv.Len()
				nn := nfv.Len()
				if on != nn {
					return true
				}
				for i := 0; i < nn; i++ {
					// Slice elements are comparable.
					if ofv.Index(i) != nfv.Index(i) {
						return true
					}
				}
			} else if ofv.Interface() != nfv.Interface() {
				// note: checking the interface values is enough.
				return true
			}
		}
		return false
	}

	for _, idx := range tv.Indices {
		var add, remove bool
		if !ov.IsValid() {
			add = true
		} else if !v.IsValid() {
			remove = true
		} else if !changed(idx) {
			continue
		} else {
			add, remove = true, true
		}

		ib, err := tx.indexBucket(idx)
		if err != nil {
			return err
		}
		if remove {
			ikl, err := idx.packKey(ov, pk)
			if err != nil {
				return err
			}
			for _, ik := range ikl {
				tx.stats.Index.Delete++
				if sanityChecks {
					tx.stats.Index.Get++
					if ib.Get(ik.full) == nil {
						return fmt.Errorf("%w: key missing from index", ErrStore)
					}
				}
				if err := ib.Delete(ik.full); err != nil {
					return fmt.Errorf("%w: removing from index: %s", ErrStore, err)
				}
			}
		}
		if add {
			ikl, err := idx.packKey(v, pk)
			if err != nil {
				return err
			}
			for _, ik := range ikl {
				if idx.Unique {
					tx.stats.Index.Cursor++
					if xk, _ := ib.Cursor().Seek(ik.pre); xk != nil && bytes.HasPrefix(xk, ik.pre) {
						return fmt.Errorf("%w: %q", ErrUnique, idx.Name)
					}
				}

				tx.stats.Index.Put++
				if err := ib.Put(ik.full, []byte{}); err != nil {
					return fmt.Errorf("inserting into index: %w", err)
				}
			}
		}
	}
	return nil
}

func (tx *Tx) checkReferences(tv *typeVersion, pk []byte, ov, rv reflect.Value) error {
	for _, f := range tv.Fields {
		if len(f.References) == 0 {
			continue
		}
		frv := rv.FieldByIndex(f.structField.Index)
		if frv.IsZero() || (ov.IsValid() && ov.FieldByIndex(f.structField.Index).Interface() == frv.Interface()) {
			continue
		}
		k, err := packPK(frv)
		if err != nil {
			return err
		}
		for _, name := range f.References {
			rb, err := tx.recordsBucket(name, tv.fillPercent)
			if err != nil {
				return err
			}
			if rb.Get(k) == nil {
				return fmt.Errorf("%w: value %v from %q to %q", ErrReference, frv.Interface(), tv.name+"."+f.Name, name)
			}
		}
	}
	return nil
}

func (tx *Tx) addStats() {
	tx.db.statsMutex.Lock()
	tx.db.stats.add(tx.stats)
	tx.db.statsMutex.Unlock()
	tx.stats = Stats{}
}

// Get fetches records by their primary key from the database. Each value must
// be a pointer to a struct.
//
// ErrAbsent is returned if the record does not exist.
func (tx *Tx) Get(values ...any) error {
	if err := tx.error(); err != nil {
		return err
	}

	for _, value := range values {
		tx.stats.Get++
		rv, err := tx.structptr(value)
		if err != nil {
			return err
		}
		st, err := tx.db.storeType(rv.Type())
		if err != nil {
			return err
		}
		rb, err := tx.recordsBucket(st.Current.name, st.Current.fillPercent)
		if err != nil {
			return err
		}
		k, _, _, err := st.Current.keyValue(tx, rv, false, rb)
		if err != nil {
			return err
		}
		tx.stats.Records.Get++
		bv := rb.Get(k)
		if bv == nil {
			return ErrAbsent
		}
		if err := st.parse(rv, bv); err != nil {
			return err
		}
	}
	return nil
}

// Delete removes values by their primary key from the database. Each value
// must be a struct or pointer to a struct. Indices are automatically updated
// and referential integrity is maintained.
//
// ErrAbsent is returned if the record does not exist.
// ErrReference is returned if another record still references this record.
func (tx *Tx) Delete(values ...any) error {
	if err := tx.error(); err != nil {
		return err
	}

	for _, value := range values {
		tx.stats.Delete++
		rv, err := tx.structOrStructptr(value)
		if err != nil {
			return err
		}
		st, err := tx.db.storeType(rv.Type())
		if err != nil {
			return err
		}
		rb, err := tx.recordsBucket(st.Current.name, st.Current.fillPercent)
		if err != nil {
			return err
		}
		k, _, _, err := st.Current.keyValue(tx, rv, false, rb)
		if err != nil {
			return err
		}
		tx.stats.Records.Get++
		bv := rb.Get(k)
		if bv == nil {
			return ErrAbsent
		}
		rov, err := st.parseNew(k, bv)
		if err != nil {
			return fmt.Errorf("parsing current value: %w", err)
		}
		if err := tx.delete(rb, st, k, rov); err != nil {
			return err
		}
	}
	return nil
}

func (tx *Tx) delete(rb *bolt.Bucket, st storeType, k []byte, rov reflect.Value) (rerr error) {
	// Check that anyone referencing this type does not reference this record.
	for _, refBy := range st.Current.referencedBy {
		if ib, err := tx.indexBucket(refBy); err != nil {
			return err
		} else {
			tx.stats.Index.Cursor++
			if xk, _ := ib.Cursor().Seek(k); xk != nil && bytes.HasPrefix(xk, k) {
				return fmt.Errorf("%w: index %q", ErrReference, refBy.Name)
			}
		}
	}

	// Delete value from indices.
	defer tx.markError(&rerr)
	if err := tx.updateIndices(st.Current, k, rov, reflect.Value{}); err != nil {
		return fmt.Errorf("removing from indices: %w", err)
	}

	tx.stats.Records.Delete++
	return rb.Delete(k)
}

// Update updates records represented by values by their primary keys into the
// database. Each value must be a pointer to a struct. Indices are
// automatically updated.
//
// ErrAbsent is returned if the record does not exist.
func (tx *Tx) Update(values ...any) error {
	if err := tx.error(); err != nil {
		return err
	}

	for _, value := range values {
		tx.stats.Update++
		rv, err := tx.structptr(value)
		if err != nil {
			return err
		}

		st, err := tx.db.storeType(rv.Type())
		if err != nil {
			return err
		}

		if err := tx.put(st, rv, false); err != nil {
			return err
		}
	}
	return nil
}

// Insert inserts values as new records into the database. Each value must be a
// pointer to a struct. If the primary key field is zero and autoincrement is not
// disabled, the next sequence is assigned. Indices are automatically updated.
//
// ErrUnique is returned if the record already exists.
// ErrSeq is returned if no next autoincrement integer is available.
// ErrZero is returned if a nonzero constraint would be violated.
// ErrReference is returned if another record is referenced that does not exist.
func (tx *Tx) Insert(values ...any) error {
	if err := tx.error(); err != nil {
		return err
	}

	for _, value := range values {
		tx.stats.Insert++
		rv, err := tx.structptr(value)
		if err != nil {
			return err
		}

		st, err := tx.db.storeType(rv.Type())
		if err != nil {
			return err
		}

		// todo optimize: should track per field whether it (or a child) has a default value, and only applyDefault if so.
		if err := st.Current.applyDefault(rv); err != nil {
			return err
		}

		if err := tx.put(st, rv, true); err != nil {
			return err
		}
	}
	return nil
}

func (tx *Tx) put(st storeType, rv reflect.Value, insert bool) error {
	rb, err := tx.recordsBucket(st.Current.name, st.Current.fillPercent)
	if err != nil {
		return err
	}
	k, krv, seq, err := st.Current.keyValue(tx, rv, insert, rb)
	if err != nil {
		return err
	}
	if insert {
		tx.stats.Records.Get++
		bv := rb.Get(k)
		if bv != nil {
			return fmt.Errorf("%w: record already exists", ErrUnique)
		}
		err := tx.insert(rb, st, rv, krv, k)
		if err != nil && seq {
			// Zero out the generated sequence.
			krv.Set(reflect.Zero(krv.Type()))
		}
		return err
	} else {
		tx.stats.Records.Get++
		bv := rb.Get(k)
		if bv == nil {
			return ErrAbsent
		}
		ov, err := st.parseNew(k, bv)
		if err != nil {
			return fmt.Errorf("parsing current value: %w", err)
		}
		return tx.update(rb, st, rv, ov, k)
	}
}

func (tx *Tx) insert(rb *bolt.Bucket, st storeType, rv, krv reflect.Value, k []byte) (rerr error) {
	v, err := st.pack(rv)
	if err != nil {
		return err
	}
	if err := tx.checkReferences(st.Current, k, reflect.Value{}, rv); err != nil {
		return err
	}
	defer tx.markError(&rerr)
	if err := tx.updateIndices(st.Current, k, reflect.Value{}, rv); err != nil {
		return fmt.Errorf("updating indices for inserted value: %w", err)
	}
	tx.stats.Records.Put++
	if err := rb.Put(k, v); err != nil {
		return err
	}
	rv.Field(0).Set(krv)
	return nil
}

func (tx *Tx) update(rb *bolt.Bucket, st storeType, rv, rov reflect.Value, k []byte) (rerr error) {
	if st.Current.equal(rov, rv) {
		return nil
	}

	v, err := st.pack(rv)
	if err != nil {
		return err
	}
	if err := tx.checkReferences(st.Current, k, rov, rv); err != nil {
		return err
	}
	defer tx.markError(&rerr)
	if err := tx.updateIndices(st.Current, k, rov, rv); err != nil {
		return fmt.Errorf("updating indices for updated record: %w", err)
	}
	tx.stats.Records.Put++
	return rb.Put(k, v)
}

// Begin starts a transaction.
//
// If writable is true, the transaction allows modifications.  Only one writable
// transaction can be active at a time on a DB. No read-only transactions can be
// active at the same time. Attempting to begin a read-only transaction from a
// writable transaction leads to deadlock.
//
// A writable Tx can be committed or rolled back. A read-only transaction must
// always be rolled back.
func (db *DB) Begin(ctx context.Context, writable bool) (*Tx, error) {
	if err := ctx.Err(); err != nil {
		return nil, err
	}
	btx, err := db.bdb.Begin(writable)
	if err != nil {
		return nil, err
	}
	db.typesMutex.RLock()
	tx := &Tx{ctx: ctx, db: db, btx: btx}
	if writable {
		tx.stats.Writes++
	} else {
		tx.stats.Reads++
	}
	return tx, nil
}

// Rollback aborts and cancels any changes made in this transaction.
// Statistics are added to its DB.
func (tx *Tx) Rollback() error {
	if tx.db == nil {
		return errTxClosed
	}

	tx.addStats()
	tx.db.typesMutex.RUnlock()
	err := tx.btx.Rollback()
	tx.db = nil
	return err
}

// Commit commits changes made in the transaction to the database.
// Statistics are added to its DB.
// If the commit fails, or the transaction was botched, the transaction is
// rolled back.
func (tx *Tx) Commit() error {
	if tx.db == nil {
		return errTxClosed
	} else if tx.err != nil {
		tx.Rollback()
		return tx.err
	}

	tx.addStats()
	tx.db.typesMutex.RUnlock()
	err := tx.btx.Commit()
	if err != nil {
		tx.btx.Rollback() // Nothing to do for error.
	}
	tx.db = nil
	return err
}