feat: changed on the repository to fetch multiple quotes

Author: gsoares85

internal/adapters/dataproviders/database/mongo/pegin.go

@@ -68,6 +68,39 @@ func (repo *peginMongoRepository) GetQuote(ctx context.Context, hash string) (*q
 	return &result.PeginQuote, nil
 }
 
+func (repo *peginMongoRepository) GetQuotes(ctx context.Context, hashes []string) ([]quote.PeginQuote, error) {
+	var result StoredPeginQuote
+	dbCtx, cancel := context.WithTimeout(ctx, repo.conn.timeout)
+	defer cancel()
+
+	for _, hash := range hashes {
+		if err := quote.ValidateQuoteHash(hash); err != nil {
+			return nil, err
+		}
+	}
+
+	collection := repo.conn.Collection(PeginQuoteCollection)
+	filter := bson.M{"hash": bson.M{"$in": hashes}}
+
+	quotesReturn := make([]quote.PeginQuote, 0)
+
+	cursor, err := collection.Find(dbCtx, filter)
+	if errors.Is(err, mongo.ErrNoDocuments) {
+		return nil, nil
+	} else if err != nil {
+		return nil, err
+	}
+	for cursor.Next(ctx) {
+		err := cursor.Decode(&result)
+		if err != nil {
+			return nil, err
+		}
+		quotesReturn = append(quotesReturn, result.PeginQuote)
+	}
+	logDbInteraction(Read, quotesReturn)
+	return quotesReturn, nil
+}
+
 func (repo *peginMongoRepository) GetRetainedQuote(ctx context.Context, hash string) (*quote.RetainedPeginQuote, error) {
 	var result quote.RetainedPeginQuote
 	dbCtx, cancel := context.WithTimeout(ctx, repo.conn.timeout)

internal/adapters/dataproviders/database/mongo/pegin_test.go

@@ -338,3 +338,45 @@ func TestPeginMongoRepository_DeleteQuotes(t *testing.T) {
 		assert.Zero(t, count)
 	})
 }
+
+func TestPeginMongoRepository_GetQuotes(t *testing.T) {
+	t.Run("Successfully retrieves quotes", func(t *testing.T) {
+		client, collection := getClientAndCollectionMocks(mongo.PeginQuoteCollection)
+		log.SetLevel(log.DebugLevel)
+		hashes := []string{testRetainedPegoutQuote.QuoteHash}
+		repo := mongo.NewPeginMongoRepository(mongo.NewConnection(client, time.Duration(1)))
+		collection.On("Find", mock.Anything,
+			bson.M{"hash": bson.M{"$in": hashes}},
+		).Return(mongoDb.NewCursorFromDocuments([]any{testPeginQuote}, nil, nil)).Once()
+		result, err := repo.GetQuotes(context.Background(), hashes)
+		collection.AssertExpectations(t)
+		require.NoError(t, err)
+		assert.Equal(t, []quote.PeginQuote{testPeginQuote}, result)
+	})
+
+	t.Run("Fails validation for hashes", func(t *testing.T) {
+		client, _ := getClientAndCollectionMocks(mongo.PeginQuoteCollection)
+
+		invalidHashes := []string{"invalidHash"}
+		conn := mongo.NewConnection(client, time.Duration(1))
+		repo := mongo.NewPeginMongoRepository(conn)
+
+		_, err := repo.GetQuotes(context.Background(), invalidHashes)
+		require.Error(t, err)
+		assert.Equal(t, "invalid quote hash length: expected 64 characters, got 11", err.Error())
+	})
+
+	t.Run("No quotes found", func(t *testing.T) {
+		client, collection := getClientAndCollectionMocks(mongo.PeginQuoteCollection)
+
+		expectedHashes := []string{testRetainedPegoutQuote.QuoteHash}
+		collection.On("Find", mock.Anything, bson.M{"hash": bson.M{"$in": expectedHashes}}).Return(nil, mongoDb.ErrNoDocuments).Once()
+
+		conn := mongo.NewConnection(client, time.Duration(1))
+		repo := mongo.NewPeginMongoRepository(conn)
+
+		quotes, err := repo.GetQuotes(context.Background(), expectedHashes)
+		require.NoError(t, err)
+		assert.Nil(t, quotes)
+	})
+}

internal/entities/quote/pegin_quote.go

@@ -27,6 +27,7 @@ const (
 type PeginQuoteRepository interface {
 	InsertQuote(ctx context.Context, hash string, quote PeginQuote) error
 	GetQuote(ctx context.Context, hash string) (*PeginQuote, error)
+	GetQuotes(ctx context.Context, hashes []string) ([]PeginQuote, error)
 	GetRetainedQuote(ctx context.Context, hash string) (*RetainedPeginQuote, error)
 	InsertRetainedQuote(ctx context.Context, quote RetainedPeginQuote) error
 	UpdateRetainedQuote(ctx context.Context, quote RetainedPeginQuote) error

internal/usecases/pegin/get_pegin_report.go

@@ -53,16 +53,14 @@ func (useCase *GetPeginReportUseCase) Run(ctx context.Context) (GetPeginReportRe
 		}, nil
 	}
 
-	quotes := make([]quote.PeginQuote, 0)
+	quoteHashes := make([]string, 0)
 	for _, q := range retained {
-		var peginQuote *quote.PeginQuote
-		peginQuote, err = useCase.peginQuoteRepository.GetQuote(ctx, q.QuoteHash)
-		if err != nil {
-			return GetPeginReportResult{}, usecases.WrapUseCaseError(usecases.GetPeginReportId, err)
-		}
-		if peginQuote != nil {
-			quotes = append(quotes, *peginQuote)
-		}
+		quoteHashes = append(quoteHashes, q.QuoteHash)
+	}
+
+	quotes, err := useCase.peginQuoteRepository.GetQuotes(ctx, quoteHashes)
+	if err != nil {
+		return GetPeginReportResult{}, usecases.WrapUseCaseError(usecases.GetPeginReportId, err)
 	}
 
 	minimumQuoteValue = useCase.calculateMinimumQuoteValue(quotes)

internal/usecases/pegin/get_pegin_report_test.go

@@ -27,6 +27,8 @@ func TestGetPeginReportUseCase_Run(t *testing.T) {
 		{QuoteHash: "hash10"},
 	}
 
+	quoteHashes := []string{"hash1", "hash2", "hash3", "hash4", "hash5", "hash6", "hash7", "hash8", "hash9", "hash10"}
+
 	peginQuotes := []quote.PeginQuote{
 		{Value: entities.NewWei(1000), CallFee: entities.NewWei(10)},
 		{Value: entities.NewWei(2000), CallFee: entities.NewWei(20)},
@@ -51,9 +53,7 @@ func TestGetPeginReportUseCase_Run(t *testing.T) {
 	peginQuoteRepository.On("GetRetainedQuoteByState", ctx, quote.PeginStateRegisterPegInSucceeded).
 		Return(retainedQuotes, nil).Once()
 
-	for i, retainedQuote := range retainedQuotes {
-		peginQuoteRepository.On("GetQuote", ctx, retainedQuote.QuoteHash).Return(&peginQuotes[i], nil).Once()
-	}
+	peginQuoteRepository.On("GetQuotes", ctx, quoteHashes).Return(peginQuotes, nil).Once()
 
 	useCase := pegin.NewGetPeginReportUseCase(peginQuoteRepository)