add beta support

Uniswap · Nov 8, 2023 · f2f1c89 · f2f1c89
1 parent cb7528e
commit f2f1c89
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/constants.ts b/lib/constants.ts
@@ -7,6 +7,7 @@ export const PRODUCTION_S3_KEY = 'production.json';
 export const BETA_S3_KEY = 'beta.json';
 export const FADE_RATE_S3_KEY = 'fade-rate.json';
 export const PROD_COMPLIANCE_S3_KEY = 'production.json';
+export const BETA_COMPLIANCE_S3_KEY = 'beta.json';
 
 export const DYNAMO_TABLE_NAME = {
   FADES: 'Fades',

diff --git a/lib/handlers/quote/injector.ts b/lib/handlers/quote/injector.ts
@@ -4,6 +4,7 @@ import { APIGatewayProxyEvent, Context } from 'aws-lambda';
 import { default as bunyan, default as Logger } from 'bunyan';
 
 import {
+  BETA_COMPLIANCE_S3_KEY,
   BETA_S3_KEY,
   COMPLIANCE_CONFIG_BUCKET,
   FADE_RATE_BUCKET,
@@ -52,10 +53,11 @@ export class QuoteInjector extends ApiInjector<ContainerInjected, RequestInjecte
       FADE_RATE_S3_KEY
     );
 
+    const complianceKey = stage === STAGE.BETA ? BETA_COMPLIANCE_S3_KEY : PROD_COMPLIANCE_S3_KEY;
     const fillerComplianceProvider = new S3FillerComplianceConfigurationProvider(
       log,
       `${COMPLIANCE_CONFIG_BUCKET}-${stage}-1`,
-      PROD_COMPLIANCE_S3_KEY
+      complianceKey
     );
 
     const quoters: Quoter[] = [new WebhookQuoter(log, webhookProvider, circuitBreakerProvider, fillerComplianceProvider)];