Re-intorduce GetAllByOwner in intent store

Author: Jeff Yanta

pkg/code/data/intent/memory/store.go

@@ -3,10 +3,12 @@ package memory
 import (
 	"context"
 	"errors"
+	"sort"
 	"sync"
 	"time"
 
 	"github.com/code-payments/code-server/pkg/code/data/intent"
+	"github.com/code-payments/code-server/pkg/database/query"
 )
 
 type store struct {
@@ -73,6 +75,28 @@ func (s *store) findByState(state intent.State) []*intent.Record {
 	return res
 }
 
+func (s *store) findByOwner(owner string) []*intent.Record {
+	res := make([]*intent.Record, 0)
+	for _, item := range s.records {
+		if item.InitiatorOwnerAccount == owner {
+			res = append(res, item)
+			continue
+		}
+
+		if item.SendPublicPaymentMetadata != nil && item.SendPublicPaymentMetadata.DestinationOwnerAccount == owner {
+			res = append(res, item)
+			continue
+		}
+
+		if item.ExternalDepositMetadata != nil && item.ExternalDepositMetadata.DestinationOwnerAccount == owner {
+			res = append(res, item)
+			continue
+		}
+	}
+
+	return res
+}
+
 func (s *store) findByDestination(destination string) []*intent.Record {
 	res := make([]*intent.Record, 0)
 	for _, item := range s.records {
@@ -119,6 +143,38 @@ func (s *store) findByInitiatorAndType(intentType intent.Type, owner string) []*
 	return res
 }
 
+func (s *store) filter(items []*intent.Record, cursor query.Cursor, limit uint64, direction query.Ordering) []*intent.Record {
+	var start uint64
+
+	start = 0
+	if direction == query.Descending {
+		start = s.last + 1
+	}
+	if len(cursor) > 0 {
+		start = cursor.ToUint64()
+	}
+
+	var res []*intent.Record
+	for _, item := range items {
+		if item.Id > start && direction == query.Ascending {
+			res = append(res, item)
+		}
+		if item.Id < start && direction == query.Descending {
+			res = append(res, item)
+		}
+	}
+
+	if direction == query.Descending {
+		sort.Sort(sort.Reverse(ById(res)))
+	}
+
+	if len(res) >= int(limit) {
+		return res[:limit]
+	}
+
+	return res
+}
+
 func (s *store) filterByState(items []*intent.Record, include bool, states ...intent.State) []*intent.Record {
 	var res []*intent.Record
 
@@ -197,6 +253,23 @@ func (s *store) Get(ctx context.Context, intentID string) (*intent.Record, error
 	return nil, intent.ErrIntentNotFound
 }
 
+func (s *store) GetAllByOwner(ctx context.Context, owner string, cursor query.Cursor, limit uint64, direction query.Ordering) ([]*intent.Record, error) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	if items := s.findByOwner(owner); len(items) > 0 {
+		res := s.filter(items, cursor, limit, direction)
+
+		if len(res) == 0 {
+			return nil, intent.ErrIntentNotFound
+		}
+
+		return res, nil
+	}
+
+	return nil, intent.ErrIntentNotFound
+}
+
 func (s *store) GetLatestByInitiatorAndType(ctx context.Context, intentType intent.Type, owner string) (*intent.Record, error) {
 	s.mu.Lock()
 	defer s.mu.Unlock()

pkg/code/data/intent/postgres/model.go

@@ -13,6 +13,7 @@ import (
 	"github.com/code-payments/code-server/pkg/currency"
 
 	pgutil "github.com/code-payments/code-server/pkg/database/postgres"
+	q "github.com/code-payments/code-server/pkg/database/query"
 )
 
 const (
@@ -211,6 +212,29 @@ func dbGetIntent(ctx context.Context, db *sqlx.DB, intentID string) (*intentMode
 	return res, nil
 }
 
+func dbGetAllByOwner(ctx context.Context, db *sqlx.DB, owner string, cursor q.Cursor, limit uint64, direction q.Ordering) ([]*intentModel, error) {
+	res := []*intentModel{}
+
+	query := `SELECT id, intent_id, intent_type, owner, source, destination_owner, destination, quantity, exchange_currency, exchange_rate, native_amount, usd_market_value, is_withdraw, is_deposit, is_remote_send, is_returned, is_issuer_voiding_gift_card, is_micro_payment, extended_metadata, state, created_at
+		FROM ` + intentTableName + `
+		WHERE owner = $1 OR destination_owner = $1
+	`
+
+	opts := []any{owner}
+	query, opts = q.PaginateQuery(query, opts, cursor, limit, direction)
+
+	err := db.SelectContext(ctx, &res, query, opts...)
+	if err != nil {
+		return nil, pgutil.CheckNoRows(err, intent.ErrIntentNotFound)
+	}
+
+	if len(res) == 0 {
+		return nil, intent.ErrIntentNotFound
+	}
+
+	return res, nil
+}
+
 func dbGetLatestByInitiatorAndType(ctx context.Context, db *sqlx.DB, intentType intent.Type, owner string) (*intentModel, error) {
 	res := &intentModel{}
 

pkg/code/data/intent/postgres/store.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 
 	"github.com/code-payments/code-server/pkg/code/data/intent"
+	"github.com/code-payments/code-server/pkg/database/query"
 	"github.com/jmoiron/sqlx"
 )
 
@@ -49,6 +50,23 @@ func (s *store) Get(ctx context.Context, intentID string) (*intent.Record, error
 	return fromIntentModel(obj), nil
 }
 
+// GetAllByOwner returns all records for a given owner (as both a source and destination).
+//
+// Returns ErrNotFound if no records are found.
+func (s *store) GetAllByOwner(ctx context.Context, owner string, cursor query.Cursor, limit uint64, direction query.Ordering) ([]*intent.Record, error) {
+	models, err := dbGetAllByOwner(ctx, s.db, owner, cursor, limit, direction)
+	if err != nil {
+		return nil, err
+	}
+
+	intents := make([]*intent.Record, len(models))
+	for i, model := range models {
+		intents[i] = fromIntentModel(model)
+	}
+
+	return intents, nil
+}
+
 // GetLatestByInitiatorAndType gets the latest record by intent type and initiating owner
 //
 // Returns ErrNotFound if no records are found.

pkg/code/data/intent/store.go

@@ -2,6 +2,8 @@ package intent
 
 import (
 	"context"
+
+	"github.com/code-payments/code-server/pkg/database/query"
 )
 
 type Store interface {
@@ -13,6 +15,11 @@ type Store interface {
 	// Returns ErrNotFound if no record is found.
 	Get(ctx context.Context, intentID string) (*Record, error)
 
+	// GetAllByOwner returns all records for a given owner (as both a source and destination).
+	//
+	// Returns ErrNotFound if no records are found.
+	GetAllByOwner(ctx context.Context, owner string, cursor query.Cursor, limit uint64, direction query.Ordering) ([]*Record, error)
+
 	// GetLatestByInitiatorAndType gets the latest record by initiating owner and intent type
 	//
 	// Returns ErrNotFound if no records are found.

pkg/code/data/internal.go

@@ -171,6 +171,7 @@ type DatabaseData interface {
 	SaveIntent(ctx context.Context, record *intent.Record) error
 	GetIntent(ctx context.Context, intentID string) (*intent.Record, error)
 	GetIntentBySignature(ctx context.Context, signature string) (*intent.Record, error)
+	GetAllIntentsByOwner(ctx context.Context, owner string, opts ...query.Option) ([]*intent.Record, error)
 	GetLatestIntentByInitiatorAndType(ctx context.Context, intentType intent.Type, owner string) (*intent.Record, error)
 	GetOriginalGiftCardIssuedIntent(ctx context.Context, giftCardVault string) (*intent.Record, error)
 	GetGiftCardClaimedIntent(ctx context.Context, giftCardVault string) (*intent.Record, error)
@@ -635,6 +636,14 @@ func (dp *DatabaseProvider) GetIntentBySignature(ctx context.Context, signature
 
 	return dp.intents.Get(ctx, fulfillmentRecord.Intent)
 }
+func (dp *DatabaseProvider) GetAllIntentsByOwner(ctx context.Context, owner string, opts ...query.Option) ([]*intent.Record, error) {
+	req, err := query.DefaultPaginationHandler(opts...)
+	if err != nil {
+		return nil, err
+	}
+
+	return dp.intents.GetAllByOwner(ctx, owner, req.Cursor, req.Limit, req.SortBy)
+}
 func (dp *DatabaseProvider) GetLatestIntentByInitiatorAndType(ctx context.Context, intentType intent.Type, owner string) (*intent.Record, error) {
 	return dp.intents.GetLatestByInitiatorAndType(ctx, intentType, owner)
 }