Feat: quoter submits bulk quotes

synapsecns · Jul 5, 2024 · cc5624c · cc5624c
1 parent 7eb7089
commit cc5624c
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 14 deletions.
diff --git a/services/rfq/relayer/quoter/quoter.go b/services/rfq/relayer/quoter/quoter.go
@@ -270,21 +270,31 @@ func (m *Manager) prepareAndSubmitQuotes(ctx context.Context, inv map[int]map[co
 	span.SetAttributes(attribute.Int("num_quotes", len(allQuotes)))
 
 	// Now, submit all the generated quotes
-	for _, quote := range allQuotes {
-		if err := m.submitQuote(ctx, quote); err != nil {
-			span.AddEvent("error submitting quote; setting relayPaused to true", trace.WithAttributes(
-				attribute.String("error", err.Error()),
-				attribute.Int(metrics.Origin, quote.OriginChainID),
-				attribute.Int(metrics.Destination, quote.DestChainID),
-				attribute.String("origin_token_addr", quote.OriginTokenAddr),
-				attribute.String("dest_token_addr", quote.DestTokenAddr),
-				attribute.String("max_origin_amount", quote.MaxOriginAmount),
-				attribute.String("dest_amount", quote.DestAmount),
-			))
+	if m.config.SubmitSingleQuotes {
+		for _, quote := range allQuotes {
+			if err := m.submitQuote(ctx, quote); err != nil {
+				span.AddEvent("error submitting quote; setting relayPaused to true", trace.WithAttributes(
+					attribute.String("error", err.Error()),
+					attribute.Int(metrics.Origin, quote.OriginChainID),
+					attribute.Int(metrics.Destination, quote.DestChainID),
+					attribute.String("origin_token_addr", quote.OriginTokenAddr),
+					attribute.String("dest_token_addr", quote.DestTokenAddr),
+					attribute.String("max_origin_amount", quote.MaxOriginAmount),
+					attribute.String("dest_amount", quote.DestAmount),
+				))
+				m.relayPaused.Store(true)
+
+				// Suppress error so that we can continue submitting quotes
+				return nil
+			}
+		}
+	} else {
+		err = m.submitBulkQuotes(ctx, allQuotes)
+		if err != nil {
+			span.AddEvent("error submitting bulk quotes; setting relayPaused to true", trace.WithAttributes(
+				attribute.String("error", err.Error())))
 			m.relayPaused.Store(true)
-
-			// Suppress error so that we can continue submitting quotes
-			return nil
+			return fmt.Errorf("error submitting bulk quotes: %w", err)
 		}
 	}
 
@@ -600,3 +610,18 @@ func (m *Manager) submitQuote(ctx context.Context, quote model.PutQuoteRequest)
 	}
 	return nil
 }
+
+// Submits a single quote.
+func (m *Manager) submitBulkQuotes(ctx context.Context, quotes []model.PutQuoteRequest) error {
+	quoteCtx, quoteCancel := context.WithTimeout(ctx, m.config.GetQuoteSubmissionTimeout())
+	defer quoteCancel()
+
+	req := model.PutBulkQuotesRequest{
+		Quotes: quotes,
+	}
+	err := m.rfqClient.PutBulkQuotes(quoteCtx, &req)
+	if err != nil {
+		return fmt.Errorf("error submitting bulk quotes: %w", err)
+	}
+	return nil
+}
diff --git a/services/rfq/relayer/relconfig/config.go b/services/rfq/relayer/relconfig/config.go
@@ -55,6 +55,8 @@ type Config struct {
 	EnableAPIWithdrawals bool `yaml:"enable_api_withdrawals"`
 	// WithdrawalWhitelist is a list of addresses that are allowed to withdraw.
 	WithdrawalWhitelist []string `yaml:"withdrawal_whitelist"`
+	// SubmitSingleQuotes enables submitting single quotes.
+	SubmitSingleQuotes bool `yaml:"submit_single_quotes"`
 }
 
 // ChainConfig represents the configuration for a chain.