tx_per_address.go

package tangle

import (
	"fmt"

	"github.com/iotaledger/goshimmer/packages/database"
	"github.com/iotaledger/hive.go/typeutils"
	"github.com/iotaledger/iota.go/trinary"
)

var (
	transactionsHashesForAddressDatabase database.Database
)

func configureTransactionHashesForAddressDatabase() {
	if db, err := database.Get(database.DBPrefixAddressTransactions, database.GetBadgerInstance()); err != nil {
		panic(err)
	} else {
		transactionsHashesForAddressDatabase = db
	}
}

type TxHashForAddress struct {
	Address trinary.Hash
	TxHash  trinary.Hash
}

func StoreTransactionHashForAddressInDatabase(address *TxHashForAddress) error {
	if err := transactionsHashesForAddressDatabase.Set(database.Entry{
		Key:   databaseKeyForHashPrefixedHash(address.Address, address.TxHash),
		Value: []byte{},
	}); err != nil {
		return fmt.Errorf("%w: failed to store tx for address in database: %s", ErrDatabaseError, err)
	}
	return nil
}

func DeleteTransactionHashForAddressInDatabase(address *TxHashForAddress) error {
	if err := transactionsHashesForAddressDatabase.Delete(
		databaseKeyForHashPrefixedHash(address.Address, address.TxHash),
	); err != nil {
		return fmt.Errorf("%w: failed to delete tx for address: %s", ErrDatabaseError, err)
	}

	return nil
}

func ReadTransactionHashesForAddressFromDatabase(address trinary.Hash) ([]trinary.Hash, error) {
	var transactionHashes []trinary.Hash
	err := transactionsHashesForAddressDatabase.StreamForEachPrefixKeyOnly(databaseKeyForHashPrefix(address), func(key database.KeyOnlyEntry) error {
		transactionHashes = append(transactionHashes, typeutils.BytesToString(key.Key))
		return nil
	})

	if err != nil {
		return nil, fmt.Errorf("%w: failed to read tx per address from database: %s", ErrDatabaseError, err)
	}
	return transactionHashes, nil
}