Return unconfirmed utxos

api-spec/protobuf/ocean/v1/account.proto

@@ -150,6 +150,8 @@ message ListUtxosResponse{
   Utxos spendable_utxos = 1;
   // List of currently locked utxos.
   Utxos locked_utxos = 2;
+  // List of unconfirmed utxos.
+  Utxos unconfirmed_utxos = 3;
 }
 
 message DeleteAccountRequest{

internal/core/application/account_service.go

@@ -172,21 +172,30 @@ func (as *AccountService) ListUtxosForAccount(
 		return nil, err
 	}
 
-	spendableUtxos, err := as.repoManager.UtxoRepository().GetSpendableUtxosForAccount(
+	utxos, err := as.repoManager.UtxoRepository().GetAllUtxosForAccount(
 		ctx, account.Namespace, scripts,
 	)
 	if err != nil {
 		return nil, err
 	}
 
-	lockedUtxos, err := as.repoManager.UtxoRepository().GetLockedUtxosForAccount(
-		ctx, account.Namespace, scripts,
-	)
-	if err != nil {
-		return nil, err
+	spendableUtxos := make([]*domain.Utxo, 0, len(utxos))
+	unconfirmedUtxos := make([]*domain.Utxo, 0, len(utxos))
+	lockedUtxos := make([]*domain.Utxo, 0, len(utxos))
+
+	for _, u := range utxos {
+		if u.IsLocked() {
+			lockedUtxos = append(lockedUtxos, u)
+		} else {
+			if u.IsConfirmed() {
+				spendableUtxos = append(spendableUtxos, u)
+			} else {
+				unconfirmedUtxos = append(unconfirmedUtxos, u)
+			}
+		}
 	}
 
-	return &UtxoInfo{spendableUtxos, lockedUtxos}, nil
+	return &UtxoInfo{spendableUtxos, lockedUtxos, unconfirmedUtxos}, nil
 }
 
 func (as *AccountService) DeleteAccount(

internal/core/application/transaction_service.go

@@ -105,7 +105,7 @@ func (ts *TransactionService) SelectUtxos(
 	}
 
 	utxos, err := ts.repoManager.UtxoRepository().GetSpendableUtxosForAccount(
-		ctx, account.Namespace, nil,
+		ctx, account.Namespace,
 	)
 	if err != nil {
 		return nil, 0, -1, err
@@ -396,7 +396,7 @@ func (ts *TransactionService) Transfer(
 	}
 
 	utxos, err := utxoRepo.GetSpendableUtxosForAccount(
-		ctx, account.Namespace, nil,
+		ctx, account.Namespace,
 	)
 	if err != nil {
 		return "", err
@@ -778,7 +778,7 @@ func (ts *TransactionService) scheduleUtxoUnlocker() {
 
 	for accountName := range w.Accounts {
 		utxos, _ := utxoRepo.GetLockedUtxosForAccount(
-			ctx, accountName, nil,
+			ctx, accountName,
 		)
 		if len(utxos) > 0 {
 			utxosToUnlock := make([]domain.UtxoKey, 0, len(utxos))

internal/core/application/types.go

@@ -55,8 +55,9 @@ type BuildInfo struct {
 }
 
 type UtxoInfo struct {
-	Spendable Utxos
-	Locked    Utxos
+	Spendable   Utxos
+	Locked      Utxos
+	Unconfirmed Utxos
 }
 
 type TransactionInfo domain.Transaction

internal/core/domain/utxo_repository.go

@@ -51,15 +51,14 @@ type UtxoRepository interface {
 	// GetSpendableUtxos returns all unlocked utxo UTXOs.
 	GetSpendableUtxos(ctx context.Context) ([]*Utxo, error)
 	// GetAllUtxosForAccount returns the list of all utxos for the given
-	// account.
-	GetAllUtxosForAccount(ctx context.Context, account string) ([]*Utxo, error)
+	// account. Can be filtered by output scripts.
+	GetAllUtxosForAccount(ctx context.Context, account string, scripts [][]byte) ([]*Utxo, error)
 	// GetSpendableUtxosForAccount returns the list of spendable utxos for the
 	// given account. The list incldues only confirmed and unlocked utxos.
-	// Can be filtered by output scripts.
-	GetSpendableUtxosForAccount(ctx context.Context, account string, scripts [][]byte) ([]*Utxo, error)
+	GetSpendableUtxosForAccount(ctx context.Context, account string) ([]*Utxo, error)
 	// GetLockedUtxosForAccount returns the list of all currently locked utxos
-	// for the given account. Can be filtered by output scripts.
-	GetLockedUtxosForAccount(ctx context.Context, account string, scripts [][]byte) ([]*Utxo, error)
+	// for the given account.
+	GetLockedUtxosForAccount(ctx context.Context, account string) ([]*Utxo, error)
 	// GetBalanceForAccount returns the confirmed, unconfirmed and locked
 	// balances per each asset for the given account.
 	GetBalanceForAccount(ctx context.Context, account string) (map[string]*Balance, error)

internal/infrastructure/storage/db/badger/utxo_repository.go

@@ -76,19 +76,9 @@ func (r *utxoRepository) GetSpendableUtxos(
 }
 
 func (r *utxoRepository) GetAllUtxosForAccount(
-	ctx context.Context, accountName string,
-) ([]*domain.Utxo, error) {
-	query := badgerhold.Where("AccountName").Eq(accountName)
-
-	return r.findUtxos(ctx, query)
-}
-
-func (r *utxoRepository) GetSpendableUtxosForAccount(
 	ctx context.Context, accountName string, scripts [][]byte,
 ) ([]*domain.Utxo, error) {
-	query := badgerhold.Where("SpentStatus").Eq(domain.UtxoStatus{}).
-		And("ConfirmedStatus").Ne(domain.UtxoStatus{}).
-		And("LockTimestamp").Eq(int64(0)).And("AccountName").Eq(accountName)
+	query := badgerhold.Where("AccountName").Eq(accountName)
 
 	utxos, err := r.findUtxos(ctx, query)
 	if err != nil {
@@ -111,37 +101,29 @@ func (r *utxoRepository) GetSpendableUtxosForAccount(
 	return filteredUtxos, nil
 }
 
-func (r *utxoRepository) GetLockedUtxosForAccount(
-	ctx context.Context, accountName string, scripts [][]byte,
+func (r *utxoRepository) GetSpendableUtxosForAccount(
+	ctx context.Context, accountName string,
 ) ([]*domain.Utxo, error) {
 	query := badgerhold.Where("SpentStatus").Eq(domain.UtxoStatus{}).
-		And("LockTimestamp").Gt(int64(0)).And("AccountName").Eq(accountName)
+		And("ConfirmedStatus").Ne(domain.UtxoStatus{}).
+		And("LockTimestamp").Eq(int64(0)).And("AccountName").Eq(accountName)
 
-	utxos, err := r.findUtxos(ctx, query)
-	if err != nil {
-		return nil, err
-	}
+	return r.findUtxos(ctx, query)
+}
 
-	if len(scripts) <= 0 {
-		return utxos, nil
-	}
+func (r *utxoRepository) GetLockedUtxosForAccount(
+	ctx context.Context, accountName string,
+) ([]*domain.Utxo, error) {
+	query := badgerhold.Where("SpentStatus").Eq(domain.UtxoStatus{}).
+		And("LockTimestamp").Gt(int64(0)).And("AccountName").Eq(accountName)
 
-	filteredUtxos := make([]*domain.Utxo, 0, len(utxos))
-	for _, u := range utxos {
-		for _, script := range scripts {
-			if bytes.Equal(u.Script, script) {
-				filteredUtxos = append(filteredUtxos, u)
-				break
-			}
-		}
-	}
-	return filteredUtxos, nil
+	return r.findUtxos(ctx, query)
 }
 
 func (r *utxoRepository) GetBalanceForAccount(
 	ctx context.Context, accountName string,
 ) (map[string]*domain.Balance, error) {
-	utxos, err := r.GetAllUtxosForAccount(ctx, accountName)
+	utxos, err := r.GetAllUtxosForAccount(ctx, accountName, nil)
 	if err != nil {
 		return nil, err
 	}

internal/infrastructure/storage/db/inmemory/utxo_repository.go

@@ -80,30 +80,30 @@ func (r *utxoRepository) GetSpendableUtxos(_ context.Context) ([]*domain.Utxo, e
 }
 
 func (r *utxoRepository) GetAllUtxosForAccount(
-	_ context.Context, account string,
+	_ context.Context, account string, scripts [][]byte,
 ) ([]*domain.Utxo, error) {
 	r.store.lock.RLock()
 	defer r.store.lock.RUnlock()
 
-	return r.getUtxosForAccount(account, false, false, nil)
+	return r.getUtxosForAccount(account, false, false, scripts)
 }
 
 func (r *utxoRepository) GetSpendableUtxosForAccount(
-	_ context.Context, account string, scripts [][]byte,
+	_ context.Context, account string,
 ) ([]*domain.Utxo, error) {
 	r.store.lock.RLock()
 	defer r.store.lock.RUnlock()
 
-	return r.getUtxosForAccount(account, true, false, scripts)
+	return r.getUtxosForAccount(account, true, false, nil)
 }
 
 func (r *utxoRepository) GetLockedUtxosForAccount(
-	_ context.Context, account string, scripts [][]byte,
+	_ context.Context, account string,
 ) ([]*domain.Utxo, error) {
 	r.store.lock.RLock()
 	defer r.store.lock.RUnlock()
 
-	return r.getUtxosForAccount(account, false, true, scripts)
+	return r.getUtxosForAccount(account, false, true, nil)
 }
 
 func (r *utxoRepository) GetBalanceForAccount(

internal/infrastructure/storage/db/postgres/utxo_repository.go

@@ -239,7 +239,7 @@ func (u *utxoRepositoryPg) GetSpendableUtxos(
 }
 
 func (u *utxoRepositoryPg) GetAllUtxosForAccount(
-	ctx context.Context, account string,
+	ctx context.Context, account string, scripts [][]byte,
 ) ([]*domain.Utxo, error) {
 	resp := make([]*domain.Utxo, 0)
 	utxos, err := u.querier.GetUtxosForAccount(ctx, account)
@@ -262,14 +262,23 @@ func (u *utxoRepositoryPg) GetAllUtxosForAccount(
 	}
 
 	for _, v := range utxosByKey {
-		resp = append(resp, v)
+		found := len(scripts) <= 0
+		for _, script := range scripts {
+			if bytes.Equal(v.Script, script) {
+				found = true
+				break
+			}
+		}
+		if found {
+			resp = append(resp, v)
+		}
 	}
 
 	return resp, nil
 }
 
 func (u *utxoRepositoryPg) GetSpendableUtxosForAccount(
-	ctx context.Context, account string, scripts [][]byte,
+	ctx context.Context, account string,
 ) ([]*domain.Utxo, error) {
 	resp := make([]*domain.Utxo, 0)
 	utxos, err := u.querier.GetUtxosForAccount(ctx, account)
@@ -293,24 +302,15 @@ func (u *utxoRepositoryPg) GetSpendableUtxosForAccount(
 
 	for _, v := range utxosByKey {
 		if !v.IsLocked() && v.IsConfirmed() && !v.IsSpent() {
-			found := len(scripts) <= 0
-			for _, script := range scripts {
-				if bytes.Equal(v.Script, script) {
-					found = true
-					break
-				}
-			}
-			if found {
-				resp = append(resp, v)
-			}
+			resp = append(resp, v)
 		}
 	}
 
 	return resp, nil
 }
 
 func (u *utxoRepositoryPg) GetLockedUtxosForAccount(
-	ctx context.Context, account string, scripts [][]byte,
+	ctx context.Context, account string,
 ) ([]*domain.Utxo, error) {
 	resp := make([]*domain.Utxo, 0)
 	utxos, err := u.querier.GetUtxosForAccount(ctx, account)
@@ -334,16 +334,7 @@ func (u *utxoRepositoryPg) GetLockedUtxosForAccount(
 
 	for _, v := range utxosByKey {
 		if v.IsLocked() {
-			found := len(scripts) <= 0
-			for _, script := range scripts {
-				if bytes.Equal(v.Script, script) {
-					found = true
-					break
-				}
-			}
-			if found {
-				resp = append(resp, v)
-			}
+			resp = append(resp, v)
 		}
 	}
 

internal/infrastructure/storage/db/test/utxo_repository_test.go

@@ -70,23 +70,23 @@ func testAddAndGetUtxos(t *testing.T, repo domain.UtxoRepository) {
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
-		utxos, err = repo.GetAllUtxosForAccount(ctx, accountName)
+		utxos, err = repo.GetAllUtxosForAccount(ctx, accountName, nil)
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
-		utxos, err = repo.GetAllUtxosForAccount(ctx, wrongAccountName)
+		utxos, err = repo.GetAllUtxosForAccount(ctx, wrongAccountName, nil)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
 		utxos, err = repo.GetSpendableUtxos(ctx)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
-		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
-		utxos, err = repo.GetLockedUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetLockedUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
@@ -140,7 +140,7 @@ func testConfirmUtxos(t *testing.T, repo domain.UtxoRepository) {
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
-		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
@@ -166,15 +166,15 @@ func testLockUtxos(t *testing.T, repo domain.UtxoRepository) {
 		require.NoError(t, err)
 		require.Zero(t, count)
 
-		utxos, err := repo.GetLockedUtxosForAccount(ctx, accountName, nil)
+		utxos, err := repo.GetLockedUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
 		utxos, err = repo.GetSpendableUtxos(ctx)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
-		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
@@ -200,15 +200,15 @@ func testUnlockUtxos(t *testing.T, repo domain.UtxoRepository) {
 		require.NoError(t, err)
 		require.Zero(t, count)
 
-		utxos, err := repo.GetLockedUtxosForAccount(ctx, accountName, nil)
+		utxos, err := repo.GetLockedUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
 		utxos, err = repo.GetSpendableUtxos(ctx)
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
-		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Len(t, utxos, len(newUtxos))
 
@@ -238,7 +238,7 @@ func testSpendUtxos(t *testing.T, repo domain.UtxoRepository) {
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
-		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
@@ -268,7 +268,7 @@ func testConfirmSpentUtxos(t *testing.T, repo domain.UtxoRepository) {
 		require.NoError(t, err)
 		require.Empty(t, utxos)
 
-		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName, nil)
+		utxos, err = repo.GetSpendableUtxosForAccount(ctx, accountName)
 		require.NoError(t, err)
 		require.Empty(t, utxos)